d3/focus.h

   1 // Copyright 2006 David Hilvert <dhilvert@auricle.dyndns.org>,
   2 //                              <dhilvert@ugcs.caltech.edu>
   3
   4 /*  This file is part of the Anti-Lamenessing Engine.
   5
   6     The Anti-Lamenessing Engine is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 3 of the License, or
   9     (at your option) any later version.
  10
  11     The Anti-Lamenessing Engine is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with the Anti-Lamenessing Engine; if not, write to the Free Software
  18     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19 */
  20
  21 /*
  22  * d3/focus.h: Implementation of defocusing logic.
  23  */
  24
  25 #ifndef __focus_h__
  26 #define __focus_h__
  27
  28 class focus {
  29 private:
  30         struct entry {
  31                 int type;
  32                 double distance;
  33                 double px, py;
  34                 double focal_range;
  35                 double vertical_gradient;
  36                 double horizontal_gradient;
  37                 double start_depth;
  38                 double end_depth;
  39                 double start_x;
  40                 double end_x;
  41                 double start_y;
  42                 double end_y;
  43                 double aperture;
  44                 unsigned int sample_count;
  45                 unsigned int focal_statistic;
  46                 unsigned int sample_randomization;
  47         };
  48
  49         static unsigned int _uses_medians;
  50         static unsigned int _max_samples;
  51         static unsigned int camera_index;
  52         static std::vector<std::vector<entry> > focus_list;
  53
  54 public:
  55
  56         struct result {
  57                 double focal_distance;
  58                 double aperture;
  59                 unsigned int sample_count;
  60                 unsigned int statistic;
  61                 unsigned int randomization;
  62         };
  63
  64         static void add_region(unsigned int type, double distance, double px, double py,
  65                         unsigned int ci, double fr, double ht, double vt, double sd, double ed,
  66                         double sx, double ex, double sy, double ey, double ap, unsigned int sc, unsigned int fs,
  67                         unsigned int sr) {
  68
  69                 if (fs)
  70                         _uses_medians = 1;
  71
  72                 if (sc > _max_samples)
  73                         _max_samples = sc;
  74
  75                 if (focus_list.size() <= ci)
  76                         focus_list.resize(ci + 1);
  77
  78                 entry e = { type, distance, px, py, fr, vt, ht, sd, ed, sx, ex, sy, ey, ap, sc, fs, sr };
  79
  80                 focus_list[ci].push_back(e);
  81         }
  82
  83         static int is_trivial() {
  84                 return (focus_list.size() == 0);
  85         }
  86
  87         static int uses_medians() {
  88                 return _uses_medians;
  89         }
  90
  91         static unsigned int max_samples() {
  92                 return _max_samples;
  93         }
  94
  95         static result get(const d2::image *depth, int i, int j) {
  96
  97                 ale_pos d = (double) depth->get_pixel(i, j)[0];
  98
  99                 std::vector<entry> *l = &(focus_list[camera_index]);
 100
 101                 /*
 102                  * Initialize default focus result.
 103                  */
 104
 105                 result r = { d, 0, 1, 0, 0 };
 106
 107                 /*
 108                  * Check for relevant focus regions.
 109                  */
 110
 111                 for (unsigned int n = 0; n < l->size(); n++) {
 112                         entry *e = &((*l)[n]);
 113
 114                         if (i >= e->start_y
 115                          && i <= e->end_y
 116                          && j >= e->start_x
 117                          && j <= e->end_x
 118                          && ((d >= -e->end_depth
 119                            && d <= -e->start_depth)
 120                           || (isnan(d) && (isnan(e->end_depth)
 121                                         || isnan(e->start_depth))))) {
 122                                 d2::point focus_origin;
 123                                 ale_pos distance_at_focus_origin;
 124
 125                                 if (e->type == 0) {
 126                                         /*
 127                                          * Distance at frame center.
 128                                          */
 129                                         focus_origin = d2::point(depth->height() / 2, depth->width() / 2);
 130                                         distance_at_focus_origin = -e->distance;
 131                                 } else if (e->type == 1) {
 132                                         /*
 133                                          * Distance at a given point.
 134                                          */
 135                                         focus_origin = d2::point(e->py, e->px);
 136                                         distance_at_focus_origin = (double) depth->get_bl(d2::point(e->py, e->px))[0];
 137                                 } else {
 138                                         fprintf(stderr, "Bad entry type.\n");
 139                                         assert(0);
 140                                         exit(1);
 141                                 }
 142
 143                                 r.focal_distance = distance_at_focus_origin + (d2::point(i, j) - focus_origin)
 144                                                                      .dproduct(d2::point(-e->vertical_gradient,
 145                                                                                          -e->horizontal_gradient));
 146
 147                                 /*
 148                                  * Adjust according to focal_range.
 149                                  */
 150
 151                                 ale_pos rel_dist = (double) d - r.focal_distance;
 152
 153                                 if (fabs(rel_dist) < e->focal_range / 2) {
 154                                         r.focal_distance = d;
 155                                 } else if (rel_dist > 0) {
 156                                         r.focal_distance += e->focal_range / 2;
 157                                 } else if (rel_dist < 0) {
 158                                         r.focal_distance -= e->focal_range / 2;
 159                                 }
 160
 161                                 r.aperture = e->aperture;
 162                                 r.sample_count = e->sample_count;
 163                                 r.statistic = e->focal_statistic;
 164                                 r.randomization = e->sample_randomization;
 165
 166                                 break;
 167                         }
 168                 }
 169
 170                 return r;
 171         }
 172
 173         static void set_camera(unsigned int c) {
 174                 camera_index = c;
 175         }
 176 };
 177
 178 #endif