Files
ferramentas/document_processor.php
T

156 lines
5.0 KiB
PHP

<?php
class DocumentProcessor {
private $srcImage;
private $processedImage;
private $outputPath;
public function __construct() {
if (!extension_loaded('gd')) {
throw new Exception('GD extension is required');
}
}
public function processDocument($imagePath, $outputPath) {
$this->outputPath = $outputPath;
// Detecta o tipo da imagem e carrega
$imageInfo = getimagesize($imagePath);
if ($imageInfo === false) {
throw new Exception('Failed to load image');
}
switch ($imageInfo[2]) {
case IMAGETYPE_JPEG:
$this->srcImage = imagecreatefromjpeg($imagePath);
break;
case IMAGETYPE_PNG:
$this->srcImage = imagecreatefrompng($imagePath);
break;
default:
throw new Exception('Unsupported image type');
}
if ($this->srcImage === false) {
throw new Exception('Failed to create image resource');
}
// Pipeline de processamento
$this->enhanceImage()
->saveResult();
return $this->outputPath;
}
private function enhanceImage() {
// Obtém dimensões originais
$width = imagesx($this->srcImage);
$height = imagesy($this->srcImage);
// Cria nova imagem
$this->processedImage = imagecreatetruecolor($width, $height);
// Preserva transparência para PNGs
imagealphablending($this->processedImage, false);
imagesavealpha($this->processedImage, true);
// Copia imagem original
imagecopy($this->processedImage, $this->srcImage, 0, 0, 0, 0, $width, $height);
// Ajusta contraste
imagefilter($this->processedImage, IMG_FILTER_CONTRAST, -10);
// Ajusta brilho
imagefilter($this->processedImage, IMG_FILTER_BRIGHTNESS, 10);
// Ajusta nitidez
imagefilter($this->processedImage, IMG_FILTER_SMOOTH, -7);
// Ajusta saturação
imagefilter($this->processedImage, IMG_FILTER_COLORIZE, 0, 0, 0, 10);
return $this;
}
private function saveResult() {
if (!isset($this->processedImage)) {
throw new Exception('No processed image to save');
}
$extension = strtolower(pathinfo($this->outputPath, PATHINFO_EXTENSION));
switch ($extension) {
case 'jpg':
case 'jpeg':
// Ordena os pontos para perspectiva
$points = $this->orderPoints($this->documentContour);
// Calcula dimensões do documento
$width = max(
sqrt(pow($points[1][0] - $points[0][0], 2) + pow($points[1][1] - $points[0][1], 2)),
sqrt(pow($points[3][0] - $points[2][0], 2) + pow($points[3][1] - $points[2][1], 2))
);
$height = max(
sqrt(pow($points[2][0] - $points[1][0], 2) + pow($points[2][1] - $points[1][1], 2)),
sqrt(pow($points[3][0] - $points[0][0], 2) + pow($points[3][1] - $points[0][1], 2))
);
// Define pontos de destino
$dst = cv\Mat::zeros([4, 2], CV_32F);
$dst[0] = [0, 0];
$dst[1] = [$width - 1, 0];
$dst[2] = [$width - 1, $height - 1];
$dst[3] = [0, $height - 1];
// Aplica transformação de perspectiva
$matrix = cv\getPerspectiveTransform($points, $dst);
$this->processedImage = cv\warpPerspective($this->srcImage, $matrix, [$width, $height]);
return $this;
}
private function enhanceImage() {
if (!isset($this->processedImage)) {
$this->processedImage = $this->srcImage;
}
// Converte para LAB para melhor processamento
$lab = cv\cvtColor($this->processedImage, cv\COLOR_BGR2LAB);
$l_channel = cv\split($lab)[0];
// Aplica CLAHE para melhorar contraste
$clahe = cv\createCLAHE(3.0, [8, 8]);
$cl = $clahe->apply($l_channel);
// Reconstrói imagem
$limg = cv\merge([$cl, cv\split($lab)[1], cv\split($lab)[2]]);
$this->processedImage = cv\cvtColor($limg, cv\COLOR_LAB2BGR);
return $this;
}
private function saveResult() {
if (!isset($this->processedImage)) {
throw new Exception('No processed image to save');
}
cv\imwrite($this->outputPath, $this->processedImage);
return $this;
}
private function orderPoints($pts) {
// Ordena pontos para transformação de perspectiva
$rect = array_fill(0, 4, [0, 0]);
$s = array_map(function($p) { return $p[0] + $p[1]; }, $pts);
$d = array_map(function($p) { return $p[0] - $p[1]; }, $pts);
$rect[0] = $pts[array_search(min($s), $s)]; // top-left
$rect[2] = $pts[array_search(max($s), $s)]; // bottom-right
$rect[1] = $pts[array_search(min($d), $d)]; // top-right
$rect[3] = $pts[array_search(max($d), $d)]; // bottom-left
return $rect;
}
}