/*
 * This file is part of the improclib project.
 * Copyright 2023 Edward V. Emelianov <edward.emelianoff@gmail.com>.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

// detect objects on an image

#include "improclib.h"
#include <usefull_macros.h>
#include <stdio.h>
#include <string.h>

static int help = 0, ndilat = 0, neros = 0;
double bg = -1.;
static char *infile = NULL, *outbg = NULL, *outbin = NULL;

static sl_option_t cmdlnopts[] = {
    {"help",    NO_ARGS,    NULL,   'h',    arg_int,    APTR(&help),    "show this help"},
    {"input",   NEED_ARG,   NULL,   'i',    arg_string, APTR(&infile),  "input file name"},
    {"obg",     NEED_ARG,   NULL,   0,      arg_string, APTR(&outbg),   "input minus bg jpeg filename"},
    {"background",NEED_ARG, NULL,   'b',    arg_double, APTR(&bg),      "background level (default: auto)"},
    {"obin",    NEED_ARG,   NULL,   0,      arg_string, APTR(&outbin),  "--obg after binarizing"},
    {"ndilat",  NEED_ARG,   NULL,   'd',    arg_int,    APTR(&ndilat),  "amount of dilations after erosions"},
    {"neros",   NEED_ARG,   NULL,   'e',    arg_int,    APTR(&neros),   "amount of image erosions"},
    end_option
};

int main(int argc, char **argv){
    sl_init();
    sl_parseargs(&argc, &argv, cmdlnopts);
    if(help) sl_showhelp(-1, cmdlnopts);
    if(!infile) ERRX("Point name of input file");
    il_Image *I = il_Image_read(infile);
    if(!I) ERR("Can't read %s", infile);
    if(bg < 0. && !il_Image_background(I, &bg)) ERRX("Can't calculate background");
    uint8_t ibg = (int)(bg + 0.5);
    printf("Background level: %d\n", ibg);
    int w = I->width, h = I->height, wh = w*h;
    il_Image *Ibg = il_Image_sim(I);
    memcpy(Ibg->data, I->data, wh);
    uint8_t *idata = (uint8_t*) Ibg->data;
    for(int i = 0; i < wh; ++i) idata[i] = (idata[i] > ibg) ? idata[i] - ibg : 0;
    if(outbg) il_write_jpg(outbg, Ibg->width, Ibg->height, 1, idata, 95);
    double t0 = sl_dtime();
    uint8_t *Ibin = il_Image2bin(I, bg);
    if(!Ibin) ERRX("Can't binarize image");
    green("Binarization: %gms\n", 1e3*(sl_dtime()-t0));
    if(neros > 0){
        t0 = sl_dtime();
        uint8_t *eros = il_erosionN(Ibin, w, h, neros);
        FREE(Ibin);
        Ibin = eros;
        green("%d erosions: %gms\n", neros, 1e3*(sl_dtime()-t0));
    }
    if(ndilat > 0){
        t0 = sl_dtime();
        uint8_t *dilat = il_dilationN(Ibin, w, h, ndilat);
        FREE(Ibin);
        Ibin = dilat;
        green("%d dilations: %gms\n", ndilat, 1e3*(sl_dtime()-t0));
    }
    if(outbin){
        il_Image *tmp = il_bin2Image(Ibin, w, h);
        il_write_jpg(outbin, tmp->width, tmp->height, 1, (uint8_t*)tmp->data, 95);
        il_Image_free(&tmp);
    }
    il_ConnComps *comps;
    t0 = sl_dtime();
    size_t *labels = il_CClabel4(Ibin, w, h, &comps);
    green("Labeling: %gms\n", 1e3*(sl_dtime()-t0));
    if(labels && comps->Nobj > 1){
        printf("Detected %zd components\n", comps->Nobj-1);
        il_Box *box = comps->boxes + 1;
        for(size_t i = 1; i < comps->Nobj; ++i, ++box){
            printf("\t%4zd: s=%d, LU=(%d, %d), RD=(%d, %d)\n", i, box->area, box->xmin, box->ymin, box->xmax, box->ymax);
        }
        FREE(labels);
        FREE(comps);
    }
    ;
    return 0;
}