156 lines
5.0 KiB
PHP
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;
|
|
}
|
|
}
|