/*
* svrt is the ``Synthetic Visual Reasoning Test'', an image
* generator for evaluating classification performance of machine
* learning systems, humans and primates.
*
* Copyright (c) 2009 Idiap Research Institute, http://www.idiap.ch/
* Written by Francois Fleuret
*
* This file is part of svrt.
*
* svrt is free software: you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* svrt 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 svrt. If not, see .
*
*/
#include "vision_problem_22.h"
#include "shape.h"
VisionProblem_22::VisionProblem_22() { }
void VisionProblem_22::sample_shapes_positions_aligned(int nb_shapes, int *xs, int *ys) {
for(int n = 0; n < nb_shapes - 1; n++) {
xs[n] = int(drand48() * (Vignette::width - part_size + 1)) + part_size/2;
ys[n] = int(drand48() * (Vignette::height - part_size + 1)) + part_size/2;
}
scalar_t alpha = drand48();
xs[nb_shapes - 1] = int(alpha * xs[0] + (1 - alpha) * xs[1]);
ys[nb_shapes - 1] = int(alpha * ys[0] + (1 - alpha) * ys[1]);
}
void VisionProblem_22::generate(int label, Vignette *vignette) {
int nb_shapes = 3;
int xs[nb_shapes], ys[nb_shapes];
int error;
do {
sample_shapes_positions_aligned(nb_shapes, xs, ys);
vignette->clear();
Shape shape;
error = 0;
for(int n = 0; n < nb_shapes; n++) {
if(n == 0 || label == 0) {
shape.randomize(part_size/2, hole_size/2);
}
error |= shape.overwrites(vignette, xs[n], ys[n]);
shape.draw(vignette, xs[n], ys[n]);
}
} while(error);
}