tools/roi-visualize: upd

This commit is contained in:
Evgeny Zinoviev 2022-06-07 19:57:12 +03:00
parent 49106dd29d
commit 70a622e04b

View File

@ -16,7 +16,7 @@ function parse_roi_input(string $file): array {
$lines = array_map(fn($line) => array_map('intval', explode(' ', $line)), $lines);
foreach ($lines as $points) {
if (count($points) != 4)
throw new Exception("invalid line: ".implode(' ', $points));
throw new Exception(__METHOD__.": invalid line: ".implode(' ', $points));
}
return $lines;
@ -57,58 +57,62 @@ $colors = [
if ($argc < 2)
fatal("usage: {$argv[0]} --roi FILE --input PATH --output PATH\n");
array_shift($argv);
while (count($argv) > 0) {
switch ($argv[0]) {
case '--roi':
array_shift($argv);
$roi_file = array_shift($argv);
break;
try {
array_shift($argv);
while (count($argv) > 0) {
switch ($argv[0]) {
case '--roi':
array_shift($argv);
$roi_file = array_shift($argv);
break;
case '--input':
array_shift($argv);
$input = array_shift($argv);
break;
case '--input':
array_shift($argv);
$input = array_shift($argv);
break;
case '--output':
array_shift($argv);
$output = array_shift($argv);
break;
case '--output':
array_shift($argv);
$output = array_shift($argv);
break;
default:
fatal('unsupported argument: '.$argv[0]);
default:
throw new Exception("unsupported argument: {$argv[0]}");
}
}
if (!$roi_file)
throw new Exception("--roi is not specified");
if (!$input)
throw new Exception('--input is not specified');
if (!$output)
throw new Exception('--output is not specified');
$regions = parse_roi_input($roi_file);
$img = imageopen($input);
if (!$img)
throw new Exception("failed to open image");
$imgw = imagesx($img);
$imgh = imagesy($img);
foreach ($regions as $i => $region) {
list($r, $g, $b) = hex2rgb($colors[$i]);
if ($region[0]+$region[2] > $imgw || $region[1]+$region[3] > $imgh)
throw new Exception('error: invalid region (line '.($i+1).')');
$col = imagecolorallocatealpha($img, $r, $g, $b, 50);
imagerectangle($img, $region[0], $region[1], $region[0]+$region[2], $region[1]+$region[3], $col);
$col = imagecolorallocatealpha($img, $r, $g, $b, 90);
imagefilledrectangle($img, $region[0]+1, $region[1]+1, $region[0]+$region[2]-2, $region[1]+$region[3]-2, $col);
}
imagejpeg($img, $output, 97);
echo "saved to $output\n";
} catch (Exception $e) {
fatal($e->getMessage()."\n");
}
if (!$roi_file)
throw new Exception("--roi is not specified");
if (!$input)
throw new Exception('--input is not specified');
if (!$output)
throw new Exception('--output is not specified');
$regions = parse_roi_input($roi_file);
$img = imageopen($input);
if (!$img)
throw new Exception("failed to open image");
$imgw = imagesx($img);
$imgh = imagesy($img);
foreach ($regions as $i => $region) {
list($r, $g, $b) = hex2rgb($colors[$i]);
if ($region[0]+$region[2] > $imgw || $region[1]+$region[3] > $imgh)
throw new Exception('error: invalid region (line '.($i+1).')');
$col = imagecolorallocatealpha($img, $r, $g, $b, 50);
imagerectangle($img, $region[0], $region[1], $region[0]+$region[2], $region[1]+$region[3], $col);
$col = imagecolorallocatealpha($img, $r, $g, $b, 90);
imagefilledrectangle($img, $region[0]+1, $region[1]+1, $region[0]+$region[2]-2, $region[1]+$region[3]-2, $col);
}
imagejpeg($img, $output, 97);
echo "saved to $output\n";