retina_svl_project.c

Go to the documentation of this file.

 
#include "../common/common.h"
 
typedef struct 
{
    long long last_time_ms;
}retina_svl_user_data;
 
static sb_t_err svl_callback(void* const user_data, const sb_t_svl_res* const res, int force)
{
    sb_t_info* info = NULL;
    sb_get_info(&info, 0);
    //Print every 500ms or if "force" flag is set
    retina_svl_user_data *ud = (retina_svl_user_data*)user_data;    
    if (ud->last_time_ms < res->time_ms)
    {
        ud->last_time_ms = res->time_ms + 1000;
        printf("\r\"%s\", TP=%d, FP=%d, FN=%d, goodness=%1.3f, cls=\"%s\", features=\"%s\", time=%llds, time last improvement=%llds (%llds), memory=%lld/%lldMByte                       ",
            res->running_step,
            res->global.tp, res->global.fp, res->global.fn, res->global.goodness,
            res->models.model[0].global.classificator, res->models.model[0].global.features,
            res->time_ms / 1000, res->time_ms_last_improvement / 1000, (res->time_ms - res->time_ms_last_improvement) / 1000,
            info->malloc_size/(1024*1024), info->malloc_size_max/(1024*1024));
        fflush(stdout);
    }
    sb_destroy_info(&info);
    sb_svl_destroy_res(&res);
    return(SB_ERR_NONE);
}
 
#if defined(_WIN32) && !defined(QT_CREATOR)
int svl_run;
static DWORD WINAPI retina_svl(LPVOID svl)
{
sb_t_err err;
    printf(">>> START SVL <<<\n");
    err = sb_svl_run(svl);
    printf("\n>>> STOP SVL <<<\n");
    if (err != SB_ERR_NONE)
    {
        printf("sb_svl_run returns %d - %s", err, sb_err_format(err));
    }
    svl_run = 0;
    return 0;
}
#endif
 
int main(int argc, char* argv[])
{
    sb_t_err err = SB_ERR_NONE;
    sb_t_info* sb_info = NULL;
    char path[256] = "solution";
    int reset = 1;
    float free_memory_percentage = 0.1f;
    SB_HANDLE svl = NULL;
    sb_t_par* par = NULL;
    sb_t_folder* folder_cls = NULL;
    sb_t_svl_res* res = NULL;
    sb_t_solution_info* solution = NULL;
    char* program_name = NULL;
    retina_svl_user_data user_data = {0};
 
#ifdef _WIN32
    program_name = strrchr(argv[0], '\\');
#else
    program_name = strrchr(argv[0], '/');
#endif
    program_name = program_name ? program_name + 1: argv[0];
 
    sb_init("../sb.lic");
 
    printf("wait for license...\n");
    CHECK_FN_GOTO(wait_license())
 
    CHECK_FN_GOTO(sb_get_info(&sb_info, 0));
    printf("---------------------------------------------------------------\n");
    printf("The program run the SVL of the specified project\n");
    printf("Compiled on: %s %s\n", __DATE__, __TIME__);
    printf("sb.dll version: %s, compiled on: %s %s\n\n", sb_info->version_str, sb_info->compile_date, sb_info->compile_time);
    printf("Usage: %s project_path reset free_memory_percentage\n", program_name ? program_name : "");
    printf("   - project_path:\n");
    printf("         path of the project\n");
    printf("   - reset:\n");
    printf("         0 svl incremental\n");
    printf("         1 svl reset\n");
    printf("   - free_memory_percentage\n");
    printf("         When the free physical memory drops below this percentage, the SVL starts to save on disk its data.\n");
    printf("         Value between 0 and 1\n");
    printf("---------------------------------------------------------------\n");
 
    if (argc == 4)
    {
        printf("-------------------------------------------------------------------------------------\n");
        printf("Input parameters:\n");
        strcpy(path, argv[1]);
        reset = atoi(argv[2]);
        free_memory_percentage = (float)atof(argv[3]);
    }
    else
    {
        printf("---------------------------------------------------------------\n");
        printf("No parameters found on command line!\n");
        printf("Runs with default parameter.\n");
    }
    printf("project_path          : \"%s\"\n", path);
    printf("reset                 : %d\n", reset);
    printf("free_memory_percentage: %f\n", free_memory_percentage);
    printf("---------------------------------------------------------------\n");
 
    CHECK_FN_GOTO(sb_folder_load(&folder_cls, path, SB_SOLUTION_EXT, 0, 0))
    if (folder_cls->n == 0)
    {
        printf("No %s file found\n", SB_SOLUTION_EXT);
        err = SB_ERR_FILE_NOT_EXIST;
        goto FnExit;
    }
 
    CHECK_FN_GOTO(sb_solution_get_info(folder_cls->file[0].name, &solution));
 
    CHECK_FN_GOTO(sb_project_load(&svl, folder_cls->file[0].name, solution->projects.info[solution->current_project].uuid, SB_PROJECT_MODE_DETECTION_AND_SVL))
 
    
    CHECK_FN_GOTO(sb_project_get_par(svl, &par))
    strcpy(par->svl.project_path, path);
    par->svl.fp_progress = svl_callback;
    par->svl.free_memory_percentage = free_memory_percentage;
    par->svl.user_data = &user_data;
    char str[8192];
    sb_par_format(par, str, sizeof(str));
    printf("%s\n", str);
    CHECK_FN_GOTO(sb_project_set_par(svl, par))
 
    if(reset)
    {
        CHECK_FN_GOTO(sb_svl_reset(svl))
    }
 
#if defined(_WIN32) && !defined(QT_CREATOR)
    printf("-------------------------------------------------------------------------------------\n");
    printf("CTRL-right + F12 = stop svl\n");                   
    printf("-------------------------------------------------------------------------------------\n");
    svl_run = 1;
    int stop_request = 0;
    CreateThread(NULL, 0, retina_svl, svl, 0, NULL);
    //With WINDOWS Visual Studio it is possibile to check the keyborad without waiting enter to be pressed
    while(1)
    {
        Sleep(100);
        //Check the keyborad without waiting enter to be pressed
        if (!stop_request && (GetAsyncKeyState(VK_RCONTROL) & 0x8000) && (GetAsyncKeyState(VK_F12) & 0x8000))            
        {
            printf("\nSend stop request to svl...\n");
            sb_svl_stop_request(svl);
            stop_request = 1;
        }
        if (!svl_run) { break;}
    }
#else
    CHECK_FN_GOTO(sb_svl_run(svl));
#endif
 
    CHECK_FN_GOTO(sb_svl_get_res(svl, &res))
    if (strcmp(res->running_step, SB_SVL_STEP_FINISHED) == 0)
    {
        printf("\nsaving SVL results....\n");
        CHECK_FN_GOTO(sb_project_save(svl, NULL, SB_PROJECT_MODE_DETECTION_AND_SVL))
    }
 
FnExit:
    sb_destroy_info(&sb_info);
    sb_project_destroy(&svl);
    sb_par_destroy(&par);
    sb_svl_destroy_res(&res);
    sb_folder_destroy(&folder_cls);
    sb_solution_destroy_info(&solution);
 
    printf("Release SqueezeBrains library\n");
    sb_release();
 
    printf("Press ENTER to terminate\n");
    getchar();
    return (int)err;
}