Oups.
[pysvrt.git] / svrt_generator.cc
1 /*
2  *  svrt is the ``Synthetic Visual Reasoning Test'', an image
3  *  generator for evaluating classification performance of machine
4  *  learning systems, humans and primates.
5  *
6  *  Copyright (c) 2016 Idiap Research Institute, http://www.idiap.ch/
7  *  Written by Francois Fleuret <francois.fleuret@idiap.ch>
8  *
9  *  This file is part of svrt.
10  *
11  *  svrt is free software: you can redistribute it and/or modify it
12  *  under the terms of the GNU General Public License version 3 as
13  *  published by the Free Software Foundation.
14  *
15  *  svrt is distributed in the hope that it will be useful, but
16  *  WITHOUT ANY WARRANTY; without even the implied warranty of
17  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18  *  General Public License for more details.
19  *
20  *  You should have received a copy of the GNU General Public License
21  *  along with svrt.  If not, see <http://www.gnu.org/licenses/>.
22  *
23  */
24
25 #include <iostream>
26 #include <fstream>
27 #include <cmath>
28 #include <stdio.h>
29 #include <stdlib.h>
30
31 using namespace std;
32
33 #include "random.h"
34
35 #include "vision_problem_1.h"
36 #include "vision_problem_2.h"
37 #include "vision_problem_3.h"
38 #include "vision_problem_4.h"
39 #include "vision_problem_5.h"
40 #include "vision_problem_6.h"
41 #include "vision_problem_7.h"
42 #include "vision_problem_8.h"
43 #include "vision_problem_9.h"
44 #include "vision_problem_10.h"
45 #include "vision_problem_11.h"
46 #include "vision_problem_12.h"
47 #include "vision_problem_13.h"
48 #include "vision_problem_14.h"
49 #include "vision_problem_15.h"
50 #include "vision_problem_16.h"
51 #include "vision_problem_17.h"
52 #include "vision_problem_18.h"
53 #include "vision_problem_19.h"
54 #include "vision_problem_20.h"
55 #include "vision_problem_21.h"
56 #include "vision_problem_22.h"
57 #include "vision_problem_23.h"
58
59 #define NB_PROBLEMS 23
60
61 VignetteGenerator *new_generator(int nb) {
62   VignetteGenerator *generator;
63
64   switch(nb) {
65   case 1:
66     generator = new VisionProblem_1();
67     break;
68   case 2:
69     generator = new VisionProblem_2();
70     break;
71   case 3:
72     generator = new VisionProblem_3();
73     break;
74   case 4:
75     generator = new VisionProblem_4();
76     break;
77   case 5:
78     generator = new VisionProblem_5();
79     break;
80   case 6:
81     generator = new VisionProblem_6();
82     break;
83   case 7:
84     generator = new VisionProblem_7();
85     break;
86   case 8:
87     generator = new VisionProblem_8();
88     break;
89   case 9:
90     generator = new VisionProblem_9();
91     break;
92   case 10:
93     generator = new VisionProblem_10();
94     break;
95   case 11:
96     generator = new VisionProblem_11();
97     break;
98   case 12:
99     generator = new VisionProblem_12();
100     break;
101   case 13:
102     generator = new VisionProblem_13();
103     break;
104   case 14:
105     generator = new VisionProblem_14();
106     break;
107   case 15:
108     generator = new VisionProblem_15();
109     break;
110   case 16:
111     generator = new VisionProblem_16();
112     break;
113   case 17:
114     generator = new VisionProblem_17();
115     break;
116   case 18:
117     generator = new VisionProblem_18();
118     break;
119   case 19:
120     generator = new VisionProblem_19();
121     break;
122   case 20:
123     generator = new VisionProblem_20();
124     break;
125   case 21:
126     generator = new VisionProblem_21();
127     break;
128   case 22:
129     generator = new VisionProblem_22();
130     break;
131   case 23:
132     generator = new VisionProblem_23();
133     break;
134   default:
135     cerr << "Can not find problem "
136          << nb
137          << endl;
138     abort();
139   }
140
141   generator->precompute();
142
143   return generator;
144 }
145
146 extern "C" {
147
148 struct VignetteSet {
149   int n_problem;
150   int nb_vignettes;
151   int width;
152   int height;
153   unsigned char *data;
154 };
155
156 void svrt_generate_vignettes(int n_problem, int nb_vignettes, long *labels,
157                              VignetteSet *result) {
158   Vignette tmp;
159
160   if(n_problem < 1 || n_problem > NB_PROBLEMS) {
161     printf("Problem number should be between 1 and %d. Provided value is %d.\n", NB_PROBLEMS, n_problem);
162     exit(1);
163   }
164
165   VignetteGenerator *vg = new_generator(n_problem);
166   result->n_problem = n_problem;
167   result->nb_vignettes = nb_vignettes;
168   result->width = Vignette::width;
169   result->height = Vignette::height;
170   result->data = (unsigned char *) malloc(sizeof(unsigned char) * result->nb_vignettes * result->width * result->height);
171
172   unsigned char *s = result->data;
173   for(int i = 0; i < nb_vignettes; i++) {
174     if(labels[i] == 0 || labels[i] == 1) {
175       vg->generate(labels[i], &tmp);
176     } else {
177       printf("Vignette class label has to be 0 or 1. Provided value is %ld.\n", labels[i]);
178       exit(1);
179     }
180
181     int *r = tmp.content;
182     for(int k = 0; k < Vignette::width * Vignette::height; k++) {
183       *s++ = *r++;
184     }
185   }
186
187   delete vg;
188 }
189
190 }