vision_problem_tools.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) 2009 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 selector.  If not, see <http://www.gnu.org/licenses/>.
  22  *
  23  */
  24
  25 #include "vision_problem_tools.h"
  26
  27 int cluttered_shapes(int part_size, int nb_shapes, int *xs, int *ys) {
  28   for(int n = 0; n < nb_shapes; n++) {
  29     for(int m = 0; m < n; m++) {
  30       if(abs(xs[n] - xs[m]) < part_size && abs(ys[n] - ys[m]) < part_size)
  31         return 1;
  32     }
  33   }
  34   return 0;
  35 }
  36
  37 scalar_t dist_point_to_segment(scalar_t xp, scalar_t yp,
  38                                scalar_t x1, scalar_t y1, scalar_t x2, scalar_t y2) {
  39   scalar_t s;
  40   s = (xp - x1) * (x2 - x1) + (yp - y1) * (y2 - y1);
  41   if(s < 0) {
  42     return sqrt(sq(xp - x1) + sq(yp - y1));
  43   } else if(s > sq(x2 - x1) + sq(y2 - y1)) {
  44     return sqrt(sq(xp - x2) + sq(yp - y2));
  45   } else {
  46     return abs((xp - x1) * (y2 - y1) - (yp - y1) * (x2 - x1))/sqrt(sq(x2 - x1) + sq(y2 - y1));
  47   }
  48 }
  49
  50 int point_in_band(scalar_t xp, scalar_t yp,
  51                   scalar_t x1, scalar_t y1, scalar_t x2, scalar_t y2,
  52                   scalar_t width) {
  53   scalar_t s;
  54   s = (xp - x1) * (x2 - x1) + (yp - y1) * (y2 - y1);
  55   if(s < 0) {
  56     return 0;
  57   } else if(s > sq(x2 - x1) + sq(y2 - y1)) {
  58     return 0;
  59   } else {
  60     return abs((xp - x1) * (y2 - y1) - (yp - y1) * (x2 - x1))/sqrt(sq(x2 - x1) + sq(y2 - y1)) <= width;
  61   }
  62 }