summaryrefslogtreecommitdiff
path: root/tools/roi-visualize.php
blob: 65067305fb9cb6452df13cce1ff785832d3d13c3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#!/usr/bin/env php
<?php

function fatal(string $message) {
     fprintf(STDERR, $message);
     exit(1);
}

function parse_roi_input(string $file): array {
    if (!file_exists($file))
        throw new Error("file $file does not exists");

    $lines = file($file);
    $lines = array_map('trim', $lines);
    $lines = array_filter($lines, fn($line) => $line != '' && $line[0] != '#');
    $lines = array_map(fn($line) => array_map('intval', explode(' ', $line)), $lines);
    foreach ($lines as $points) {
        if (count($points) != 4)
            throw new Exception(__METHOD__.": invalid line: ".implode(' ', $points));
    }

    return $lines;
}

function hex2rgb(int $color): array {
    $r = ($color >> 16) & 0xff;
    $g = ($color >> 8) & 0xff;
    $b = $color & 0xff;
    return [$r, $g, $b];
}

function imageopen(string $filename) {
    $size = getimagesize($filename);
    $types = [
        2 => 'jpeg',
        3 => 'png'
    ];
    if (!$size || !isset($types[$size[2]]))
        return false;

    $f = 'imagecreatefrom'.$types[$size[2]];
    return call_user_func($f, $filename);
}

error_reporting(E_ALL);
ini_set('display_errors', 1);

$colors = [
    0xff0000,
    0x00ff00,
    0x0000ff,
    0xffff00,
    0xff00ff,
    0x00ffff,
];

if ($argc < 2)
    fatal("usage: {$argv[0]} --roi FILE --input PATH --output PATH\n");

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 '--output':
                array_shift($argv);
                $output = array_shift($argv);
                break;

            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");
}