1 // Copyright 2003, 2004, 2005, 2006 David Hilvert <dhilvert@auricle.dyndns.org>,
2 // <dhilvert@ugcs.caltech.edu>
4 /* This file is part of the Anti-Lamenessing Engine.
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 2 of the License, or
9 (at your option) any later version.
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.
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
22 * d3/scene.h: Representation of a 3D scene.
31 * View angle multiplier.
33 * Setting this to a value larger than one can be useful for debugging.
36 #define VIEW_ANGLE_MULTIPLIER 1
43 static ale_pos front_clip
;
44 static ale_pos rear_clip
;
47 * Decimation exponents for geometry
49 static int primary_decimation_upper
;
50 static int input_decimation_lower
;
51 static int output_decimation_preferred
;
56 static int output_clip
;
61 static const char *load_model_name
;
62 static const char *save_model_name
;
65 * Occupancy attenuation
68 static double occ_att
;
71 * Normalization of output by weight
74 static int normalize_weights
;
80 static int use_filter
;
81 static const char *d3chain_type
;
87 static double falloff_exponent
;
90 * Third-camera error multiplier
92 static double tc_multiplier
;
95 * Occupancy update iterations
97 static unsigned int ou_iterations
;
102 static unsigned int pairwise_ambiguity
;
105 * Pairwise comparisons
107 static const char *pairwise_comparisons
;
112 static int d3px_count
;
113 static double *d3px_parameters
;
118 static const ale_real nearness
;
121 * Encounter threshold for defined pixels.
123 static double encounter_threshold
;
126 * Median calculation radii.
128 static double depth_median_radius
;
129 static double diff_median_radius
;
132 * Flag for subspace traversal.
134 static int subspace_traverse
;
137 * Structure to hold input frame information at levels of
138 * detail between full detail and full decimation.
142 unsigned int entries
;
143 std::vector
<const d2::image
*> im
;
144 std::vector
<pt
> transformation
;
150 lod_image(unsigned int _f
) {
155 im
.push_back(d2::image_rw::copy(f
, "3D reference image"));
158 _pt
= d3::align::projective(f
);
159 _pt
.scale(1 / _pt
.scale_2d());
160 transformation
.push_back(_pt
);
162 while (im
.back()->height() > 4
163 && im
.back()->width() > 4) {
165 im
.push_back(im
.back()->scale_by_half("3D, reduced LOD"));
168 _pt
.scale(1 / _pt
.scale_2d() / pow(2, entries
));
169 transformation
.push_back(_pt
);
176 * Get the number of scales
178 unsigned int count() {
185 const d2::image
*get_image(unsigned int i
) {
191 int in_bounds(d2::point p
) {
192 return im
[0]->in_bounds(p
);
196 * Get a 'trilinear' color. We currently don't do interpolation
197 * between levels of detail; hence, it's discontinuous in tl_coord.
199 d2::pixel
get_tl(d2::point p
, ale_pos tl_coord
) {
201 assert(in_bounds(p
));
203 tl_coord
= round(tl_coord
);
205 if (tl_coord
>= entries
)
210 p
= p
/ (ale_pos
) pow(2, tl_coord
);
212 unsigned int itlc
= (unsigned int) tl_coord
;
214 if (p
[0] > im
[itlc
]->height() - 1)
215 p
[0] = im
[itlc
]->height() - 1;
216 if (p
[1] > im
[itlc
]->width() - 1)
217 p
[1] = im
[itlc
]->width() - 1;
222 return im
[itlc
]->get_bl(p
);
225 d2::pixel
get_max_diff(d2::point p
, ale_pos tl_coord
) {
226 assert(in_bounds(p
));
228 tl_coord
= round(tl_coord
);
230 if (tl_coord
>= entries
)
235 p
= p
/ (ale_pos
) pow(2, tl_coord
);
237 unsigned int itlc
= (unsigned int) tl_coord
;
239 if (p
[0] > im
[itlc
]->height() - 1)
240 p
[0] = im
[itlc
]->height() - 1;
241 if (p
[1] > im
[itlc
]->width() - 1)
242 p
[1] = im
[itlc
]->width() - 1;
247 return im
[itlc
]->get_max_diff(p
);
251 * Get the transformation
253 pt
get_t(unsigned int i
) {
256 return transformation
[i
];
260 * Get the camera origin in world coordinates
263 return transformation
[0].origin();
270 for (unsigned int i
= 0; i
< entries
; i
++) {
277 * Structure to hold weight information for reference images.
281 std::vector
<d2::image
*> weights
;
287 void set_image(d2::image
*im
, ale_real value
) {
289 for (unsigned int i
= 0; i
< im
->height(); i
++)
290 for (unsigned int j
= 0; j
< im
->width(); j
++)
291 im
->pix(i
, j
) = d2::pixel(value
, value
, value
);
294 d2::image
*make_image(ale_pos sf
, ale_real init_value
= 0) {
295 d2::image
*result
= new d2::image_ale_real(
296 (unsigned int) ceil(transformation
.unscaled_height() * sf
),
297 (unsigned int) ceil(transformation
.unscaled_width() * sf
), 3);
301 set_image(result
, init_value
);
309 * Explicit weight subtree
313 subtree
*children
[2][2];
315 subtree(ale_real nv
, subtree
*a
, subtree
*b
, subtree
*c
, subtree
*d
) {
324 for (int i
= 0; i
< 2; i
++)
325 for (int j
= 0; j
< 2; j
++)
326 delete children
[i
][j
];
333 ref_weights(unsigned int _f
) {
335 transformation
= d3::align::projective(f
);
340 * Check spatial bounds.
342 int in_spatial_bounds(point p
) {
351 if (p
[0] > transformation
.unscaled_height() - 1)
353 if (p
[1] > transformation
.unscaled_width() - 1)
361 int in_spatial_bounds(const space::traverse
&t
) {
362 point p
= transformation
.centroid(t
);
363 return in_spatial_bounds(p
);
367 * Increase resolution to the given level.
369 void increase_resolution(int tc
, unsigned int i
, unsigned int j
) {
370 d2::image
*im
= weights
[tc
- tc_low
];
372 assert(i
<= im
->height() - 1);
373 assert(j
<= im
->width() - 1);
376 * Check for the cases known to have no lower level of detail.
379 if (im
->pix(i
, j
)[0] == -1)
385 increase_resolution(tc
+ 1, i
/ 2, j
/ 2);
388 * Load the lower-level image structure.
391 d2::image
*iim
= weights
[tc
+ 1 - tc_low
];
394 assert(i
/ 2 <= iim
->height() - 1);
395 assert(j
/ 2 <= iim
->width() - 1);
398 * Check for the case where no lower level of detail is
402 if (iim
->pix(i
/ 2, j
/ 2)[0] == -1)
406 * Spread out the lower level of detail among (uninitialized)
410 for (unsigned int ii
= (i
/ 2) * 2; ii
< (i
/ 2) * 2 + 1; ii
++)
411 for (unsigned int jj
= (j
/ 2) * 2; jj
< (j
/ 2) * 2 + 1; jj
++) {
412 assert(ii
<= im
->height() - 1);
413 assert(jj
<= im
->width() - 1);
414 assert(im
->pix(ii
, jj
)[0] == 0);
416 im
->pix(ii
, jj
) = iim
->pix(i
/ 2, j
/ 2);
420 * Set the lower level of detail to point here.
423 iim
->pix(i
/ 2, j
/ 2)[0] = -1;
427 * Add weights to positive higher-resolution pixels where
428 * found when their current values match the given subtree
429 * values; set negative pixels to zero and return 0 if no
430 * positive higher-resolution pixels are found.
432 int add_partial(int tc
, unsigned int i
, unsigned int j
, ale_real weight
, subtree
*st
) {
433 d2::image
*im
= weights
[tc
- tc_low
];
436 if (i
== im
->height() - 1
437 || j
== im
->width() - 1) {
441 assert(i
<= im
->height() - 1);
442 assert(j
<= im
->width() - 1);
445 * Check for positive values.
448 if (im
->pix(i
, j
)[0] > 0) {
449 if (st
&& st
->node_value
== im
->pix(i
, j
)[0])
450 im
->pix(i
, j
)[0] += weight
* (1 - im
->pix(i
, j
)[0]);
455 * Handle the case where there are no higher levels of detail.
459 if (im
->pix(i
, j
)[0] != 0) {
460 fprintf(stderr
, "failing assertion: im[%d]->pix(%d, %d)[0] == %g\n", tc
, i
, j
,
463 assert(im
->pix(i
, j
)[0] == 0);
468 * Handle the case where higher levels of detail are available.
473 for (int ii
= 0; ii
< 2; ii
++)
474 for (int jj
= 0; jj
< 2; jj
++)
475 success
[ii
][jj
] = add_partial(tc
- 1, i
* 2 + ii
, j
* 2 + jj
, weight
,
476 st
? st
->children
[ii
][jj
] : NULL
);
482 im
->pix(i
, j
)[0] = 0;
486 d2::image
*iim
= weights
[tc
- 1 - tc_low
];
489 for (int ii
= 0; ii
< 2; ii
++)
490 for (int jj
= 0; jj
< 2; jj
++)
491 if (success
[ii
][jj
] == 0) {
492 assert(i
* 2 + ii
< iim
->height());
493 assert(j
* 2 + jj
< iim
->width());
495 iim
->pix(i
* 2 + ii
, j
* 2 + jj
)[0] = weight
;
498 im
->pix(i
, j
)[0] = -1;
506 void add_weight(int tc
, unsigned int i
, unsigned int j
, ale_real weight
, subtree
*st
) {
508 assert (weight
>= 0);
510 d2::image
*im
= weights
[tc
- tc_low
];
513 // fprintf(stderr, "[aw tc=%d i=%d j=%d imax=%d jmax=%d]\n",
514 // tc, i, j, im->height(), im->width());
516 assert(i
<= im
->height() - 1);
517 assert(j
<= im
->width() - 1);
520 * Increase resolution, if necessary
523 increase_resolution(tc
, i
, j
);
526 * Attempt to add the weight at levels of detail
527 * where weight is defined.
530 if (add_partial(tc
, i
, j
, weight
, st
))
534 * If no weights are defined at any level of detail,
535 * then set the weight here.
538 im
->pix(i
, j
)[0] = weight
;
541 void add_weight(int tc
, d2::point p
, ale_real weight
, subtree
*st
) {
543 assert (weight
>= 0);
547 unsigned int i
= (unsigned int) floor(p
[0]);
548 unsigned int j
= (unsigned int) floor(p
[1]);
550 add_weight(tc
, i
, j
, weight
, st
);
553 void add_weight(const space::traverse
&t
, ale_real weight
, subtree
*st
) {
555 assert (weight
>= 0);
560 ale_pos tc
= transformation
.trilinear_coordinate(t
);
561 point p
= transformation
.centroid(t
);
562 assert(in_spatial_bounds(p
));
567 * Establish a reasonable (?) upper bound on resolution.
570 if (tc
< input_decimation_lower
) {
571 weight
/= pow(4, (input_decimation_lower
- tc
));
572 tc
= input_decimation_lower
;
576 * Initialize, if necessary.
580 tc_low
= tc_high
= (int) tc
;
582 ale_real sf
= pow(2, -tc
);
584 weights
.push_back(make_image(sf
));
590 * Check resolution bounds
593 assert (tc_low
<= tc_high
);
596 * Generate additional levels of detail, if necessary.
599 while (tc
< tc_low
) {
602 ale_real sf
= pow(2, -tc_low
);
604 weights
.insert(weights
.begin(), make_image(sf
));
607 while (tc
> tc_high
) {
610 ale_real sf
= pow(2, -tc_high
);
612 weights
.push_back(make_image(sf
, -1));
615 add_weight((int) tc
, p
.xy(), weight
, st
);
621 ale_real
get_weight(int tc
, unsigned int i
, unsigned int j
) {
623 // fprintf(stderr, "[gw tc=%d i=%u j=%u tclow=%d tchigh=%d]\n",
624 // tc, i, j, tc_low, tc_high);
626 if (tc
< tc_low
|| !initialized
)
630 return (get_weight(tc
- 1, i
* 2 + 0, j
* 2 + 0)
631 + get_weight(tc
- 1, i
* 2 + 1, j
* 2 + 0)
632 + get_weight(tc
- 1, i
* 2 + 1, j
* 2 + 1)
633 + get_weight(tc
- 1, i
* 2 + 0, j
* 2 + 1)) / 4;
636 assert(weights
.size() > (unsigned int) (tc
- tc_low
));
638 d2::image
*im
= weights
[tc
- tc_low
];
641 if (i
== im
->height())
643 if (j
== im
->width())
646 assert(i
< im
->height());
647 assert(j
< im
->width());
649 if (im
->pix(i
, j
)[0] == -1) {
650 return (get_weight(tc
- 1, i
* 2 + 0, j
* 2 + 0)
651 + get_weight(tc
- 1, i
* 2 + 1, j
* 2 + 0)
652 + get_weight(tc
- 1, i
* 2 + 1, j
* 2 + 1)
653 + get_weight(tc
- 1, i
* 2 + 0, j
* 2 + 1)) / 4;
656 if (im
->pix(i
, j
)[0] == 0) {
659 if (weights
[tc
- tc_low
+ 1]->pix(i
/ 2, j
/ 2)[0] == -1)
661 return get_weight(tc
+ 1, i
/ 2, j
/ 2);
664 return im
->pix(i
, j
)[0];
667 ale_real
get_weight(int tc
, d2::point p
) {
671 unsigned int i
= (unsigned int) floor(p
[0]);
672 unsigned int j
= (unsigned int) floor(p
[1]);
674 return get_weight(tc
, i
, j
);
677 ale_real
get_weight(const space::traverse
&t
) {
678 ale_pos tc
= transformation
.trilinear_coordinate(t
);
679 point p
= transformation
.centroid(t
);
680 assert(in_spatial_bounds(p
));
691 return get_weight((int) tc
, p
.xy());
695 * Check whether a subtree is simple.
697 int is_simple(subtree
*s
) {
700 if (s
->node_value
== -1
701 && s
->children
[0][0] == NULL
702 && s
->children
[0][1] == NULL
703 && s
->children
[1][0] == NULL
704 && s
->children
[1][1] == NULL
)
711 * Get a weight subtree.
713 subtree
*get_subtree(int tc
, unsigned int i
, unsigned int j
) {
716 * tc > tc_high is handled recursively.
720 subtree
*result
= new subtree(-1,
721 get_subtree(tc
- 1, i
* 2 + 0, j
* 2 + 0),
722 get_subtree(tc
- 1, i
* 2 + 0, j
* 2 + 1),
723 get_subtree(tc
- 1, i
* 2 + 1, j
* 2 + 0),
724 get_subtree(tc
- 1, i
* 2 + 1, j
* 2 + 1));
726 if (is_simple(result
)) {
734 assert(tc
>= tc_low
);
735 assert(weights
[tc
- tc_low
]);
737 d2::image
*im
= weights
[tc
- tc_low
];
740 * Rectangular images will, in general, have
741 * out-of-bounds tree sections. Handle this case.
744 if (i
>= im
->height())
746 if (j
>= im
->width())
750 * -1 weights are handled recursively
753 if (im
->pix(i
, j
)[0] == -1) {
754 subtree
*result
= new subtree(-1,
755 get_subtree(tc
- 1, i
* 2 + 0, j
* 2 + 0),
756 get_subtree(tc
- 1, i
* 2 + 0, j
* 2 + 1),
757 get_subtree(tc
- 1, i
* 2 + 1, j
* 2 + 0),
758 get_subtree(tc
- 1, i
* 2 + 1, j
* 2 + 1));
760 if (is_simple(result
)) {
761 im
->pix(i
, j
)[0] = 0;
770 * Zero weights have NULL subtrees.
773 if (im
->pix(i
, j
)[0] == 0)
777 * Handle the remaining case.
780 return new subtree(im
->pix(i
, j
)[0], NULL
, NULL
, NULL
, NULL
);
783 subtree
*get_subtree(int tc
, d2::point p
) {
786 unsigned int i
= (unsigned int) floor(p
[0]);
787 unsigned int j
= (unsigned int) floor(p
[1]);
789 return get_subtree(tc
, i
, j
);
792 subtree
*get_subtree(const space::traverse
&t
) {
793 ale_pos tc
= transformation
.trilinear_coordinate(t
);
794 point p
= transformation
.centroid(t
);
795 assert(in_spatial_bounds(p
));
800 if (tc
< input_decimation_lower
)
801 tc
= input_decimation_lower
;
808 return get_subtree((int) tc
, p
.xy());
815 for (unsigned int i
= 0; i
< weights
.size(); i
++) {
824 static int resolution_ok(pt transformation
, ale_pos tc
) {
826 if (pow(2, tc
) > transformation
.unscaled_height()
827 || pow(2, tc
) > transformation
.unscaled_width())
830 if (tc
< input_decimation_lower
- 1.5)
837 * Structure to hold input frame information at all levels of detail.
845 std::vector
<lod_image
*> images
;
850 images
.resize(d2::image_rw::count(), NULL
);
853 unsigned int count() {
854 return d2::image_rw::count();
857 void open(unsigned int f
) {
858 assert (images
[f
] == NULL
);
860 if (images
[f
] == NULL
)
861 images
[f
] = new lod_image(f
);
865 for (unsigned int f
= 0; f
< d2::image_rw::count(); f
++)
869 lod_image
*get(unsigned int f
) {
870 assert (images
[f
] != NULL
);
874 void close(unsigned int f
) {
875 assert (images
[f
] != NULL
);
881 for (unsigned int f
= 0; f
< d2::image_rw::count(); f
++)
891 * All levels-of-detail
894 static struct lod_images
*al
;
897 * Data structure for storing best encountered subspace candidates.
900 std::vector
<std::vector
<std::pair
<ale_pos
, ale_real
> > > levels
;
906 * Point p is in world coordinates.
908 void generate_subspace(point iw
, ale_pos diagonal
) {
910 // fprintf(stderr, "[gs iw=%f %f %f d=%f]\n",
911 // iw[0], iw[1], iw[2], diagonal);
913 space::traverse st
= space::traverse::root();
915 if (!st
.includes(iw
)) {
924 * Loop until resolutions of interest have been generated.
929 ale_pos current_diagonal
= (st
.get_max() - st
.get_min()).norm();
931 assert(!isnan(current_diagonal
));
934 * Generate any new desired spatial registers.
941 for (int f
= 0; f
< 2; f
++) {
947 if (current_diagonal
< 2 * diagonal
949 if (spatial_info_map
.find(st
.get_node()) == spatial_info_map
.end()) {
950 spatial_info_map
[st
.get_node()];
951 ui::get()->d3_increment_spaces();
960 if (current_diagonal
< diagonal
962 if (spatial_info_map
.find(st
.get_node()) == spatial_info_map
.end()) {
963 spatial_info_map
[st
.get_node()];
964 ui::get()->d3_increment_spaces();
971 * Check for completion
974 if (highres
&& lowres
)
978 * Check precision before analyzing space further.
981 if (st
.precision_wall()) {
982 fprintf(stderr
, "\n\n*** Error: reached subspace precision wall ***\n\n");
987 if (st
.positive().includes(iw
)) {
990 } else if (st
.negative().includes(iw
)) {
994 fprintf(stderr
, "failed iw = (%f, %f, %f)\n",
995 iw
[0], iw
[1], iw
[2]);
1005 height
= (unsigned int) al
->get(f
)->get_t(0).unscaled_height();
1006 width
= (unsigned int) al
->get(f
)->get_t(0).unscaled_width();
1009 * Is this necessary?
1012 levels
.resize(primary_decimation_upper
- input_decimation_lower
+ 1);
1013 for (int l
= input_decimation_lower
; l
<= primary_decimation_upper
; l
++) {
1014 levels
[l
- input_decimation_lower
].resize((unsigned int) (floor(height
/ pow(2, l
))
1015 * floor(width
/ pow(2, l
))
1016 * pairwise_ambiguity
),
1017 std::pair
<ale_pos
, ale_real
>(0, 0));
1022 * Point p is expected to be in local projective coordinates.
1025 void add_candidate(point p
, int tc
, ale_real score
) {
1026 assert(tc
<= primary_decimation_upper
);
1027 assert(tc
>= input_decimation_lower
);
1031 int i
= (unsigned int) floor(p
[0] / pow(2, tc
));
1032 int j
= (unsigned int) floor(p
[1] / pow(2, tc
));
1034 int swidth
= (int) floor(width
/ pow(2, tc
));
1037 assert(i
< (int) floor(height
/ pow(2, tc
)));
1039 for (unsigned int k
= 0; k
< pairwise_ambiguity
; k
++) {
1040 std::pair
<ale_pos
, ale_real
> *pk
=
1041 &(levels
[tc
- input_decimation_lower
][i
* swidth
* pairwise_ambiguity
+ j
* pairwise_ambiguity
+ k
]);
1043 if (pk
->first
!= 0 && score
>= pk
->second
)
1046 if (i
== 1 && j
== 1 && tc
== 4)
1047 fprintf(stderr
, "[ac p2=%f score=%f]\n", p
[2], score
);
1049 ale_pos tp
= pk
->first
;
1050 ale_real tr
= pk
->second
;
1064 * Generate subspaces for candidates.
1067 void generate_subspaces() {
1069 fprintf(stderr
, "+");
1070 for (int l
= input_decimation_lower
; l
<= primary_decimation_upper
; l
++) {
1071 unsigned int sheight
= (unsigned int) floor(height
/ pow(2, l
));
1072 unsigned int swidth
= (unsigned int) floor(width
/ pow(2, l
));
1074 for (unsigned int i
= 0; i
< sheight
; i
++)
1075 for (unsigned int j
= 0; j
< swidth
; j
++)
1076 for (unsigned int k
= 0; k
< pairwise_ambiguity
; k
++) {
1077 std::pair
<ale_pos
, ale_real
> *pk
=
1078 &(levels
[l
- input_decimation_lower
]
1079 [i
* swidth
* pairwise_ambiguity
+ j
* pairwise_ambiguity
+ k
]);
1081 if (pk
->first
== 0) {
1082 fprintf(stderr
, "o");
1085 fprintf(stderr
, "|");
1088 ale_pos si
= i
* pow(2, l
) + ((l
> 0) ? pow(2, l
- 1) : 0);
1089 ale_pos sj
= j
* pow(2, l
) + ((l
> 0) ? pow(2, l
- 1) : 0);
1091 // fprintf(stderr, "[gss l=%d i=%d j=%d d=%g]\n", l, i, j, pk->first);
1093 point p
= al
->get(image_index
)->get_t(0).pw_unscaled(point(si
, sj
, pk
->first
));
1095 generate_subspace(p
,
1096 al
->get(image_index
)->get_t(0).diagonal_distance_3d(pk
->first
, l
));
1103 * List for calculating weighted median.
1110 ale_real
&_w(unsigned int i
) {
1115 ale_real
&_d(unsigned int i
) {
1117 return data
[i
* 2 + 1];
1120 void increase_capacity() {
1127 data
= (ale_real
*) realloc(data
, sizeof(ale_real
) * 2 * (size
* 1));
1130 assert (size
> used
);
1133 fprintf(stderr
, "Unable to allocate %d bytes of memory\n",
1134 sizeof(ale_real
) * 2 * (size
* 1));
1139 void insert_weight(unsigned int i
, ale_real v
, ale_real w
) {
1140 assert(used
< size
);
1142 for (unsigned int j
= used
; j
> i
; j
--) {
1155 unsigned int get_size() {
1159 unsigned int get_used() {
1164 fprintf(stderr
, "[st %p sz %u el", this, size
);
1165 for (unsigned int i
= 0; i
< used
; i
++)
1166 fprintf(stderr
, " (%f, %f)", _d(i
), _w(i
));
1167 fprintf(stderr
, "]\n");
1174 void insert_weight(ale_real v
, ale_real w
) {
1175 for (unsigned int i
= 0; i
< used
; i
++) {
1182 increase_capacity();
1183 insert_weight(i
, v
, w
);
1188 increase_capacity();
1189 insert_weight(used
, v
, w
);
1193 * Finds the median at half-weight, or between half-weight
1194 * and zero-weight, depending on the attenuation value.
1197 ale_real
find_median(double attenuation
= 0) {
1199 assert(attenuation
>= 0);
1200 assert(attenuation
<= 1);
1204 ale_real undefined
= zero1
/ zero2
;
1206 ale_accum weight_sum
= 0;
1208 for (unsigned int i
= 0; i
< used
; i
++)
1209 weight_sum
+= _w(i
);
1211 // if (weight_sum == 0)
1212 // return undefined;
1214 if (used
== 0 || used
== 1)
1217 if (weight_sum
== 0) {
1218 ale_accum data_sum
= 0;
1219 for (unsigned int i
= 0; i
< used
; i
++)
1221 return data_sum
/ used
;
1225 ale_accum midpoint
= weight_sum
* (0.5 - 0.5 * attenuation
);
1227 ale_accum weight_sum_2
= 0;
1229 for (unsigned int i
= 0; i
< used
&& weight_sum_2
< midpoint
; i
++) {
1230 weight_sum_2
+= _w(i
);
1232 if (weight_sum_2
> midpoint
) {
1234 } else if (weight_sum_2
== midpoint
) {
1235 assert (i
+ 1 < used
);
1236 return (_d(i
) + _d(i
+ 1)) / 2;
1243 wml(int initial_size
= 0) {
1245 // if (initial_size == 0) {
1246 // initial_size = (int) (d2::image_rw::count() * 1.5);
1249 size
= initial_size
;
1253 data
= (ale_real
*) malloc(size
* sizeof(ale_real
) * 2);
1261 * copy constructor. This is required to avoid undesired frees.
1267 data
= (ale_real
*) malloc(size
* sizeof(ale_real
) * 2);
1270 memcpy(data
, w
.data
, size
* sizeof(ale_real
) * 2);
1279 * Class for information regarding spatial regions of interest.
1281 * This class is configured for convenience in cases where sampling is
1282 * performed using an approximation of the fine:box:1,triangle:2 chain.
1283 * In this case, the *_1 variables would store the fine data and the
1284 * *_2 variables would store the coarse data. Other uses are also
1288 class spatial_info
{
1290 * Map channel value --> weight.
1292 wml color_weights_1
[3];
1293 wml color_weights_2
[3];
1301 * Map occupancy value --> weight.
1303 wml occupancy_weights_1
;
1304 wml occupancy_weights_2
;
1307 * Current occupancy value.
1315 unsigned int pocc_density
;
1316 unsigned int socc_density
;
1319 * Insert a weight into a list.
1321 void insert_weight(wml
*m
, ale_real v
, ale_real w
) {
1322 m
->insert_weight(v
, w
);
1326 * Find the median of a weighted list. Uses NaN for undefined.
1328 ale_real
find_median(wml
*m
, double attenuation
= 0) {
1329 return m
->find_median(attenuation
);
1337 color
= d2::pixel::zero();
1344 * Accumulate color; primary data set.
1346 void accumulate_color_1(int f
, d2::pixel color
, d2::pixel weight
) {
1347 for (int k
= 0; k
< 3; k
++)
1348 insert_weight(&color_weights_1
[k
], color
[k
], weight
[k
]);
1352 * Accumulate color; secondary data set.
1354 void accumulate_color_2(d2::pixel color
, d2::pixel weight
) {
1355 for (int k
= 0; k
< 3; k
++)
1356 insert_weight(&color_weights_2
[k
], color
[k
], weight
[k
]);
1360 * Accumulate occupancy; primary data set.
1362 void accumulate_occupancy_1(int f
, ale_real occupancy
, ale_real weight
) {
1363 insert_weight(&occupancy_weights_1
, occupancy
, weight
);
1367 * Accumulate occupancy; secondary data set.
1369 void accumulate_occupancy_2(ale_real occupancy
, ale_real weight
) {
1370 insert_weight(&occupancy_weights_2
, occupancy
, weight
);
1372 if (occupancy
== 0 || occupancy_weights_2
.get_size() < 96)
1375 // fprintf(stderr, "%p updated socc with: %f %f\n", this, occupancy, weight);
1376 // occupancy_weights_2.print_info();
1380 * Update color (and clear accumulation structures).
1382 void update_color() {
1383 for (int d
= 0; d
< 3; d
++) {
1384 ale_real c
= find_median(&color_weights_1
[d
]);
1386 c
= find_median(&color_weights_2
[d
]);
1392 color_weights_1
[d
].clear();
1393 color_weights_2
[d
].clear();
1398 * Update occupancy (and clear accumulation structures).
1400 void update_occupancy() {
1401 ale_real o
= find_median(&occupancy_weights_1
, occ_att
);
1403 o
= find_median(&occupancy_weights_2
, occ_att
);
1409 pocc_density
= occupancy_weights_1
.get_used();
1410 socc_density
= occupancy_weights_2
.get_used();
1412 occupancy_weights_1
.clear();
1413 occupancy_weights_2
.clear();
1418 * Get current color.
1420 d2::pixel
get_color() {
1425 * Get current occupancy.
1427 ale_real
get_occupancy() {
1428 assert (finite(occupancy
));
1433 * Get primary color density.
1436 unsigned int get_pocc_density() {
1437 return pocc_density
;
1440 unsigned int get_socc_density() {
1441 return socc_density
;
1446 * Map spatial regions of interest to spatial info structures. XXX:
1447 * This may get very poor cache behavior in comparison with, say, an
1448 * array. Unfortunately, there is no immediately obvious array
1449 * representation. If some kind of array representation were adopted,
1450 * it would probably cluster regions of similar depth from the
1451 * perspective of the typical camera. In particular, for a
1452 * stereoscopic view, depth ordering for two random points tends to be
1453 * similar between cameras, I think. Unfortunately, it is never
1454 * identical for all points (unless cameras are co-located). One
1455 * possible approach would be to order based on, say, camera 0's idea
1459 #if !defined(HASH_MAP_GNU) && !defined(HASH_MAP_STD)
1460 typedef std::map
<struct space::node
*, spatial_info
> spatial_info_map_t
;
1461 #elif defined(HASH_MAP_GNU)
1464 size_t operator()(struct space::node
*n
) const
1466 return __gnu_cxx::hash
<long>()((long) n
);
1469 typedef __gnu_cxx::hash_map
<struct space::node
*, spatial_info
, node_hash
> spatial_info_map_t
;
1470 #elif defined(HASH_MAP_STD)
1471 typedef std::hash_map
<struct space::node
*, spatial_info
> spatial_info_map_t
;
1474 static spatial_info_map_t spatial_info_map
;
1479 * Debugging variables.
1482 static unsigned long total_ambiguity
;
1483 static unsigned long total_pixels
;
1484 static unsigned long total_divisions
;
1485 static unsigned long total_tsteps
;
1491 static void et(double et_parameter
) {
1492 encounter_threshold
= et_parameter
;
1495 static void dmr(double dmr_parameter
) {
1496 depth_median_radius
= dmr_parameter
;
1499 static void fmr(double fmr_parameter
) {
1500 diff_median_radius
= fmr_parameter
;
1503 static void load_model(const char *name
) {
1504 load_model_name
= name
;
1507 static void save_model(const char *name
) {
1508 save_model_name
= name
;
1511 static void fc(ale_pos fc
) {
1515 static void di_upper(ale_pos _dgi
) {
1516 primary_decimation_upper
= (int) round(_dgi
);
1519 static void do_try(ale_pos _dgo
) {
1520 output_decimation_preferred
= (int) round(_dgo
);
1523 static void di_lower(ale_pos _idiv
) {
1524 input_decimation_lower
= (int) round(_idiv
);
1531 static void no_oc() {
1535 static void rc(ale_pos rc
) {
1540 * Initialize 3D scene from 2D scene, using 2D and 3D alignment
1543 static void init_from_d2() {
1546 * Rear clip value of 0 is converted to infinity.
1549 if (rear_clip
== 0) {
1553 rear_clip
= one
/ zero
;
1554 assert(isinf(rear_clip
) == +1);
1558 * Scale and translate clipping plane depths.
1561 ale_pos cp_scalar
= d3::align::projective(0).wc(point(0, 0, 0))[2];
1563 front_clip
= front_clip
* cp_scalar
- cp_scalar
;
1564 rear_clip
= rear_clip
* cp_scalar
- cp_scalar
;
1567 * Allocate image structures.
1570 al
= new lod_images
;
1572 if (tc_multiplier
!= 0) {
1578 * Perform spatial_info updating on a given subspace, for given
1581 static void subspace_info_update(space::iterate si
, int f
, ref_weights
*weights
) {
1585 space::traverse st
= si
.get();
1588 * Skip spaces with no color information.
1590 * XXX: This could be more efficient, perhaps.
1593 if (spatial_info_map
.count(st
.get_node()) == 0) {
1598 ui::get()->d3_increment_space_num();
1602 * Get in-bounds centroid, if one exists.
1605 point p
= al
->get(f
)->get_t(0).centroid(st
);
1613 * Get information on the subspace.
1616 spatial_info
*sn
= &spatial_info_map
[st
.get_node()];
1617 d2::pixel color
= sn
->get_color();
1618 ale_real occupancy
= sn
->get_occupancy();
1621 * Store current weight so we can later check for
1622 * modification by higher-resolution subspaces.
1625 ref_weights::subtree
*tree
= weights
->get_subtree(st
);
1628 * Check for higher resolution subspaces, and
1629 * update the space iterator.
1632 if (st
.get_node()->positive
1633 || st
.get_node()->negative
) {
1636 * Cleave space for the higher-resolution pass,
1637 * skipping the current space, since we will
1638 * process that later.
1641 space::iterate cleaved_space
= si
.cleave();
1643 cleaved_space
.next();
1645 subspace_info_update(cleaved_space
, f
, weights
);
1652 * Add new data on the subspace and update weights.
1655 ale_pos tc
= al
->get(f
)->get_t(0).trilinear_coordinate(st
);
1656 d2::pixel pcolor
= al
->get(f
)->get_tl(p
.xy(), tc
);
1657 d2::pixel colordiff
= (color
- pcolor
) * (ale_real
) 256;
1659 if (falloff_exponent
!= 0) {
1660 d2::pixel max_diff
= al
->get(f
)->get_max_diff(p
.xy(), tc
) * (ale_real
) 256;
1662 for (int k
= 0; k
< 3; k
++)
1663 if (max_diff
[k
] > 1)
1664 colordiff
[k
] /= pow(max_diff
[k
], falloff_exponent
);
1668 * Determine the probability of encounter.
1671 d2::pixel encounter
= d2::pixel(1, 1, 1) * (1 - weights
->get_weight(st
));
1677 weights
->add_weight(st
, occupancy
, tree
);
1680 * Delete the subtree, if necessary.
1686 * Check for cases in which the subspace should not be
1690 if (!resolution_ok(al
->get(f
)->get_t(0), tc
))
1693 if (d2::render::is_excluded_f(p
.xy(), f
))
1700 sn
->accumulate_color_1(f
, pcolor
, encounter
);
1701 d2::pixel channel_occ
= pexp(-colordiff
* colordiff
);
1703 ale_accum occ
= channel_occ
[0];
1705 for (int k
= 1; k
< 3; k
++)
1706 if (channel_occ
[k
] < occ
)
1707 occ
= channel_occ
[k
];
1709 sn
->accumulate_occupancy_1(f
, occ
, encounter
[0]);
1715 * Run a single iteration of the spatial_info update cycle.
1717 static void spatial_info_update() {
1719 * Iterate through each frame.
1721 for (unsigned int f
= 0; f
< d2::image_rw::count(); f
++) {
1723 ui::get()->d3_occupancy_status(f
);
1726 * Open the frame and transformation.
1729 if (tc_multiplier
== 0)
1733 * Allocate weights data structure for storing encounter
1737 ref_weights
*weights
= new ref_weights(f
);
1740 * Call subspace_info_update for the root space.
1743 subspace_info_update(space::iterate(al
->get(f
)->origin()), f
, weights
);
1752 * Close the frame and transformation.
1755 if (tc_multiplier
== 0)
1760 * Update all spatial_info structures.
1762 for (spatial_info_map_t::iterator i
= spatial_info_map
.begin(); i
!= spatial_info_map
.end(); i
++) {
1763 i
->second
.update_color();
1764 i
->second
.update_occupancy();
1766 // d2::pixel color = i->second.get_color();
1768 // fprintf(stderr, "space p=%p updated to c=[%f %f %f] o=%f\n",
1769 // i->first, color[0], color[1], color[2],
1770 // i->second.get_occupancy());
1775 * Support function for view() and depth(). This function
1776 * always performs exclusion.
1779 static const void view_recurse(int type
, d2::image
*im
, d2::image
*weights
, space::iterate si
, pt _pt
,
1780 int prune
= 0, d2::point pl
= d2::point(0, 0), d2::point ph
= d2::point(0, 0)) {
1781 while (!si
.done()) {
1782 space::traverse st
= si
.get();
1785 * Remove excluded regions.
1797 if (prune
&& !_pt
.check_inclusion_scaled(st
, pl
, ph
)) {
1803 * XXX: This could be more efficient, perhaps.
1806 if (spatial_info_map
.count(st
.get_node()) == 0) {
1811 ui::get()->d3_increment_space_num();
1813 spatial_info sn
= spatial_info_map
[st
.get_node()];
1816 * Get information on the subspace.
1819 d2::pixel color
= sn
.get_color();
1820 // d2::pixel color = d2::pixel(1, 1, 1) * (double) (((unsigned int) (st.get_node()) / sizeof(space)) % 65535);
1821 ale_real occupancy
= sn
.get_occupancy();
1824 * Determine the view-local bounding box for the
1830 _pt
.get_view_local_bb_scaled(st
, bb
);
1839 || max
[1] < pl
[1]) {
1860 * Data structure to check modification of weights by
1861 * higher-resolution subspaces.
1864 std::queue
<d2::pixel
> weight_queue
;
1867 * Check for higher resolution subspaces, and
1868 * update the space iterator.
1871 if (st
.get_node()->positive
1872 || st
.get_node()->negative
) {
1875 * Store information about current weights,
1876 * so we will know which areas have been
1877 * covered by higher-resolution subspaces.
1880 for (int i
= (int) ceil(min
[0]); i
<= (int) floor(max
[0]); i
++)
1881 for (int j
= (int) ceil(min
[1]); j
<= (int) floor(max
[1]); j
++)
1882 weight_queue
.push(weights
->get_pixel(i
, j
));
1885 * Cleave space for the higher-resolution pass,
1886 * skipping the current space, since we will
1887 * process that afterward.
1890 space::iterate cleaved_space
= si
.cleave();
1892 cleaved_space
.next();
1894 view_recurse(type
, im
, weights
, cleaved_space
, _pt
, prune
, pl
, ph
);
1902 * Iterate over pixels in the bounding box, finding
1903 * pixels that intersect the subspace. XXX: assume
1904 * for now that all pixels in the bounding box
1905 * intersect the subspace.
1908 for (int i
= (int) ceil(min
[0]); i
<= (int) floor(max
[0]); i
++)
1909 for (int j
= (int) ceil(min
[1]); j
<= (int) floor(max
[1]); j
++) {
1912 * Check for higher-resolution updates.
1915 if (weight_queue
.size()) {
1916 if (weight_queue
.front() != weights
->get_pixel(i
, j
)) {
1924 * Determine the probability of encounter.
1927 d2::pixel encounter
= (d2::pixel(1, 1, 1)
1928 - weights
->get_pixel(i
, j
))
1941 weights
->pix(i
, j
) += encounter
;
1942 im
->pix(i
, j
) += encounter
* color
;
1944 } else if (type
== 1) {
1947 * Weighted (transparent) depth display
1950 ale_pos depth_value
= _pt
.wp_scaled(st
.get_min())[2];
1951 weights
->pix(i
, j
) += encounter
;
1952 im
->pix(i
, j
) += encounter
* depth_value
;
1954 } else if (type
== 2) {
1957 * Ambiguity (ambivalence) measure.
1960 weights
->pix(i
, j
) = d2::pixel(1, 1, 1);
1961 im
->pix(i
, j
) += 0.1 * d2::pixel(1, 1, 1);
1963 } else if (type
== 3) {
1966 * Closeness measure.
1969 ale_pos depth_value
= _pt
.wp_scaled(st
.get_min())[2];
1970 if (weights
->pix(i
, j
)[0] == 0) {
1971 weights
->pix(i
, j
) = d2::pixel(1, 1, 1);
1972 im
->pix(i
, j
) = d2::pixel(1, 1, 1) * depth_value
;
1973 } else if (im
->pix(i
, j
)[2] < depth_value
) {
1974 im
->pix(i
, j
) = d2::pixel(1, 1, 1) * depth_value
;
1979 } else if (type
== 4) {
1982 * Weighted (transparent) contribution display
1985 ale_pos contribution_value
= sn
.get_pocc_density() /* + sn.get_socc_density() */;
1986 weights
->pix(i
, j
) += encounter
;
1987 im
->pix(i
, j
) += encounter
* contribution_value
;
1989 assert (finite(encounter
[0]));
1990 assert (finite(contribution_value
));
1992 } else if (type
== 5) {
1995 * Weighted (transparent) occupancy display
1998 ale_pos contribution_value
= occupancy
;
1999 weights
->pix(i
, j
) += encounter
;
2000 im
->pix(i
, j
) += encounter
* contribution_value
;
2002 } else if (type
== 6) {
2005 * (Depth, xres, yres) triple
2008 ale_pos depth_value
= _pt
.wp_scaled(st
.get_min())[2];
2009 weights
->pix(i
, j
)[0] += encounter
[0];
2010 if (weights
->pix(i
, j
)[1] < encounter
[0]) {
2011 weights
->pix(i
, j
)[1] = encounter
[0];
2012 im
->pix(i
, j
)[0] = weights
->pix(i
, j
)[1] * depth_value
;
2013 im
->pix(i
, j
)[1] = max
[0] - min
[0];
2014 im
->pix(i
, j
)[2] = max
[1] - min
[1];
2017 } else if (type
== 7) {
2020 * (xoff, yoff, 0) triple
2023 weights
->pix(i
, j
)[0] += encounter
[0];
2024 if (weights
->pix(i
, j
)[1] < encounter
[0]) {
2025 weights
->pix(i
, j
)[1] = encounter
[0];
2026 im
->pix(i
, j
)[0] = i
- min
[0];
2027 im
->pix(i
, j
)[1] = j
- min
[1];
2028 im
->pix(i
, j
)[2] = 0;
2031 } else if (type
== 8) {
2034 * Value = 1 for any intersected space.
2037 weights
->pix(i
, j
) = d2::pixel(1, 1, 1);
2038 im
->pix(i
, j
) = d2::pixel(1, 1, 1);
2040 } else if (type
== 9) {
2043 * Number of contributions for the nearest space.
2046 if (weights
->pix(i
, j
)[0] == 1)
2049 weights
->pix(i
, j
) = d2::pixel(1, 1, 1);
2050 im
->pix(i
, j
) = d2::pixel(1, 1, 1) * (sn
.get_pocc_density() * 0.1);
2059 * Generate an depth image from a specified view.
2061 static const d2::image
*depth(pt _pt
, int n
= -1, int prune
= 0,
2062 d2::point pl
= d2::point(0, 0), d2::point ph
= d2::point(0, 0)) {
2063 assert ((unsigned int) n
< d2::image_rw::count() || n
< 0);
2065 _pt
.view_angle(_pt
.view_angle() * VIEW_ANGLE_MULTIPLIER
);
2068 assert((int) floor(d2::align::of(n
).scaled_height())
2069 == (int) floor(_pt
.scaled_height()));
2070 assert((int) floor(d2::align::of(n
).scaled_width())
2071 == (int) floor(_pt
.scaled_width()));
2074 d2::image
*im1
, *im2
, *im3
, *weights
;;
2078 im1
= new d2::image_ale_real((int) floor(ph
[0] - pl
[0]) + 1,
2079 (int) floor(ph
[1] - pl
[1]) + 1, 3);
2081 im2
= new d2::image_ale_real((int) floor(ph
[0] - pl
[0]) + 1,
2082 (int) floor(ph
[1] - pl
[1]) + 1, 3);
2084 im3
= new d2::image_ale_real((int) floor(ph
[0] - pl
[0]) + 1,
2085 (int) floor(ph
[1] - pl
[1]) + 1, 3);
2087 weights
= new d2::image_ale_real((int) floor(ph
[0] - pl
[0]) + 1,
2088 (int) floor(ph
[1] - pl
[1]) + 1, 3);
2092 im1
= new d2::image_ale_real((int) floor(_pt
.scaled_height()),
2093 (int) floor(_pt
.scaled_width()), 3);
2095 im2
= new d2::image_ale_real((int) floor(_pt
.scaled_height()),
2096 (int) floor(_pt
.scaled_width()), 3);
2098 im3
= new d2::image_ale_real((int) floor(_pt
.scaled_height()),
2099 (int) floor(_pt
.scaled_width()), 3);
2101 weights
= new d2::image_ale_real((int) floor(_pt
.scaled_height()),
2102 (int) floor(_pt
.scaled_width()), 3);
2106 * Iterate through subspaces.
2109 space::iterate
si(_pt
.origin());
2111 view_recurse(6, im1
, weights
, si
, _pt
, prune
, pl
, ph
);
2116 weights
= new d2::image_ale_real((int) floor(ph
[0] - pl
[0]) + 1,
2117 (int) floor(ph
[1] - pl
[1]) + 1, 3);
2119 weights
= new d2::image_ale_real((int) floor(_pt
.scaled_height()),
2120 (int) floor(_pt
.scaled_width()), 3);
2124 view_recurse(7, im2
, weights
, si
, _pt
, prune
, pl
, ph
);
2126 view_recurse(8, im2
, weights
, si
, _pt
, prune
, pl
, ph
);
2131 * Normalize depths by weights
2134 if (normalize_weights
)
2135 for (unsigned int i
= 0; i
< im1
->height(); i
++)
2136 for (unsigned int j
= 0; j
< im1
->width(); j
++)
2137 im1
->pix(i
, j
)[0] /= weights
->pix(i
, j
)[1];
2140 for (unsigned int i
= 0; i
< im1
->height(); i
++)
2141 for (unsigned int j
= 0; j
< im1
->width(); j
++) {
2144 * Handle interpolation.
2149 d2::point
res(im1
->pix(i
, j
)[1], im1
->pix(i
, j
)[2]);
2151 for (int d
= 0; d
< 2; d
++) {
2153 if (im2
->pix(i
, j
)[d
] < res
[d
] / 2)
2154 x
[d
] = (ale_pos
) (d
?j
:i
) - res
[d
] / 2 - im2
->pix(i
, j
)[d
];
2156 x
[d
] = (ale_pos
) (d
?j
:i
) + res
[d
] / 2 - im2
->pix(i
, j
)[d
];
2158 blx
[d
] = 1 - ((d
?j
:i
) - x
[d
]) / res
[d
];
2161 ale_real depth_val
= 0;
2162 ale_real depth_weight
= 0;
2164 for (int ii
= 0; ii
< 2; ii
++)
2165 for (int jj
= 0; jj
< 2; jj
++) {
2166 d2::point p
= x
+ d2::point(ii
, jj
) * res
;
2167 if (im1
->in_bounds(p
)) {
2169 ale_real d
= im1
->get_bl(p
)[0];
2174 ale_real w
= ((ii
? (1 - blx
[0]) : blx
[0]) * (jj
? (1 - blx
[1]) : blx
[1]));
2180 ale_real depth
= depth_val
/ depth_weight
;
2183 * Handle encounter thresholds
2186 if (weights
->pix(i
, j
)[0] < encounter_threshold
) {
2187 im3
->pix(i
, j
) = d2::pixel::zero() / d2::pixel::zero();
2189 im3
->pix(i
, j
) = d2::pixel(1, 1, 1) * depth
;
2200 static const d2::image
*depth(unsigned int n
) {
2202 assert (n
< d2::image_rw::count());
2204 pt _pt
= align::projective(n
);
2206 return depth(_pt
, n
);
2211 * This function always performs exclusion.
2214 static space::node
*most_visible_pointwise(d2::pixel
*weight
, space::iterate si
, pt _pt
, d2::point p
) {
2216 space::node
*result
= NULL
;
2218 while (!si
.done()) {
2219 space::traverse st
= si
.get();
2222 * Prune certain regions known to be uninteresting.
2225 if (excluded(st
) || !_pt
.check_inclusion_scaled(st
, p
)) {
2231 * XXX: This could be more efficient, perhaps.
2234 if (spatial_info_map
.count(st
.get_node()) == 0) {
2239 spatial_info sn
= spatial_info_map
[st
.get_node()];
2242 * Get information on the subspace.
2245 ale_real occupancy
= sn
.get_occupancy();
2248 * Preserve current weight in order to check for
2249 * modification by higher-resolution subspaces.
2252 d2::pixel old_weight
= *weight
;
2255 * Check for higher resolution subspaces, and
2256 * update the space iterator.
2259 if (st
.get_node()->positive
2260 || st
.get_node()->negative
) {
2263 * Cleave space for the higher-resolution pass,
2264 * skipping the current space, since we will
2265 * process that afterward.
2268 space::iterate cleaved_space
= si
.cleave();
2270 cleaved_space
.next();
2272 space::node
*r
= most_visible_pointwise(weight
, cleaved_space
, _pt
, p
);
2274 if (old_weight
[1] != (*weight
)[1])
2283 * Check for higher-resolution updates.
2286 if (old_weight
!= *weight
)
2290 * Determine the probability of encounter.
2293 ale_pos encounter
= (1 - (*weight
)[0]) * occupancy
;
2296 * (*weight)[0] stores the cumulative weight; (*weight)[1] stores the maximum.
2299 if (encounter
> (*weight
)[1]) {
2300 result
= st
.get_node();
2301 (*weight
)[1] = encounter
;
2304 (*weight
)[0] += encounter
;
2311 * This function performs exclusion iff SCALED is true.
2313 static void most_visible_generic(std::vector
<space::node
*> &results
, d2::image
*weights
,
2314 space::iterate si
, pt _pt
, int scaled
) {
2316 assert (results
.size() == weights
->height() * weights
->width());
2318 while (!si
.done()) {
2319 space::traverse st
= si
.get();
2321 if (scaled
&& excluded(st
)) {
2327 * XXX: This could be more efficient, perhaps.
2330 if (spatial_info_map
.count(st
.get_node()) == 0) {
2335 spatial_info sn
= spatial_info_map
[st
.get_node()];
2338 * Get information on the subspace.
2341 ale_real occupancy
= sn
.get_occupancy();
2344 * Determine the view-local bounding box for the
2350 _pt
.get_view_local_bb_scaled(st
, bb
);
2356 * Data structure to check modification of weights by
2357 * higher-resolution subspaces.
2360 std::queue
<d2::pixel
> weight_queue
;
2363 * Check for higher resolution subspaces, and
2364 * update the space iterator.
2367 if (st
.get_node()->positive
2368 || st
.get_node()->negative
) {
2371 * Store information about current weights,
2372 * so we will know which areas have been
2373 * covered by higher-resolution subspaces.
2376 for (int i
= (int) ceil(min
[0]); i
<= (int) floor(max
[0]); i
++)
2377 for (int j
= (int) ceil(min
[1]); j
<= (int) floor(max
[1]); j
++)
2378 weight_queue
.push(weights
->get_pixel(i
, j
));
2381 * Cleave space for the higher-resolution pass,
2382 * skipping the current space, since we will
2383 * process that afterward.
2386 space::iterate cleaved_space
= si
.cleave();
2388 cleaved_space
.next();
2390 most_visible_generic(results
, weights
, cleaved_space
, _pt
, scaled
);
2398 * Iterate over pixels in the bounding box, finding
2399 * pixels that intersect the subspace. XXX: assume
2400 * for now that all pixels in the bounding box
2401 * intersect the subspace.
2404 for (int i
= (int) ceil(min
[0]); i
<= (int) floor(max
[0]); i
++)
2405 for (int j
= (int) ceil(min
[1]); j
<= (int) floor(max
[1]); j
++) {
2408 * Check for higher-resolution updates.
2411 if (weight_queue
.size()) {
2412 if (weight_queue
.front() != weights
->get_pixel(i
, j
)) {
2420 * Determine the probability of encounter.
2423 ale_pos encounter
= (1 - weights
->get_pixel(i
, j
)[0]) * occupancy
;
2426 * weights[0] stores the cumulative weight; weights[1] stores the maximum.
2429 if (encounter
> weights
->get_pixel(i
, j
)[1]
2430 || results
[i
* weights
->width() + j
] == NULL
) {
2431 results
[i
* weights
->width() + j
] = st
.get_node();
2432 weights
->chan(i
, j
, 1) = encounter
;
2435 weights
->chan(i
, j
, 0) += encounter
;
2440 static std::vector
<space::node
*> most_visible_scaled(pt _pt
) {
2441 d2::image
*weights
= new d2::image_ale_real((int) floor(_pt
.scaled_height()),
2442 (int) floor(_pt
.scaled_width()), 3);
2443 std::vector
<space::node
*> results
;
2445 results
.resize(weights
->height() * weights
->width(), 0);
2447 most_visible_generic(results
, weights
, space::iterate(_pt
.origin()), _pt
, 1);
2452 static std::vector
<space::node
*> most_visible_unscaled(pt _pt
) {
2453 d2::image
*weights
= new d2::image_ale_real((int) floor(_pt
.unscaled_height()),
2454 (int) floor(_pt
.unscaled_width()), 3);
2455 std::vector
<space::node
*> results
;
2457 results
.resize(weights
->height() * weights
->width(), 0);
2459 most_visible_generic(results
, weights
, space::iterate(_pt
.origin()), _pt
, 0);
2464 static const int visibility_search(const std::vector
<space::node
*> &fmv
, space::node
*mv
) {
2469 if (std::binary_search(fmv
.begin(), fmv
.end(), mv
))
2472 return (visibility_search(fmv
, mv
->positive
)
2473 || visibility_search(fmv
, mv
->negative
));
2478 * Class to generate focal sample views.
2481 class view_generator
{
2484 * Original projective transformation.
2490 * Data type for shared view data.
2495 std::vector
<space::node
*> mv
;
2497 d2::image
*color_weights
;
2498 const d2::image
*_depth
;
2499 d2::image
*median_depth
;
2500 d2::image
*median_diff
;
2503 shared_view(pt _pt
) {
2506 color_weights
= NULL
;
2508 median_depth
= NULL
;
2512 shared_view(const shared_view
©_origin
) {
2513 _pt
= copy_origin
._pt
;
2514 mv
= copy_origin
.mv
;
2516 color_weights
= NULL
;
2518 median_depth
= NULL
;
2525 delete color_weights
;
2527 delete median_depth
;
2530 void get_view_recurse(d2::image
*data
, d2::image
*weights
, int type
) {
2532 * Iterate through subspaces.
2535 space::iterate
si(_pt
.origin());
2537 ui::get()->d3_render_status(0, 0, -1, -1, -1, -1, 0);
2539 view_recurse(type
, data
, weights
, si
, _pt
);
2543 color
= new d2::image_ale_real((int) floor(_pt
.scaled_height()),
2544 (int) floor(_pt
.scaled_width()), 3);
2546 color_weights
= new d2::image_ale_real((int) floor(_pt
.scaled_height()),
2547 (int) floor(_pt
.scaled_width()), 3);
2549 get_view_recurse(color
, color_weights
, 0);
2553 _depth
= depth(_pt
, -1);
2556 void init_medians() {
2562 median_diff
= _depth
->fcdiff_median((int) floor(diff_median_radius
));
2563 median_depth
= _depth
->medians((int) floor(depth_median_radius
));
2565 assert(median_diff
);
2566 assert(median_depth
);
2574 space::node
*get_most_visible(unsigned int i
, unsigned int j
) {
2575 unsigned int height
= (int) floor(_pt
.scaled_height());
2576 unsigned int width
= (int) floor(_pt
.scaled_width());
2578 if (i
< 0 || i
>= height
2579 || j
< 0 || j
>= width
) {
2583 if (mv
.size() == 0) {
2584 mv
= most_visible_scaled(_pt
);
2587 assert (mv
.size() > i
* width
+ j
);
2589 return mv
[i
* width
+ j
];
2592 space::node
*get_most_visible(d2::point p
) {
2593 unsigned int i
= (unsigned int) round (p
[0]);
2594 unsigned int j
= (unsigned int) round (p
[1]);
2596 return get_most_visible(i
, j
);
2599 d2::pixel
get_color(unsigned int i
, unsigned int j
) {
2600 if (color
== NULL
) {
2604 assert (color
!= NULL
);
2606 return color
->get_pixel(i
, j
);
2609 d2::pixel
get_depth(unsigned int i
, unsigned int j
) {
2610 if (_depth
== NULL
) {
2614 assert (_depth
!= NULL
);
2616 return _depth
->get_pixel(i
, j
);
2619 void get_median_depth_and_diff(d2::pixel
*t
, d2::pixel
*f
, unsigned int i
, unsigned int j
) {
2620 if (median_depth
== NULL
&& median_diff
== NULL
)
2623 assert (median_depth
&& median_diff
);
2625 if (i
< 0 || i
>= median_depth
->height()
2626 || j
< 0 || j
>= median_depth
->width()) {
2627 *t
= d2::pixel::undefined();
2628 *f
= d2::pixel::undefined();
2630 *t
= median_depth
->get_pixel(i
, j
);
2631 *f
= median_diff
->get_pixel(i
, j
);
2635 void get_color_and_weight(d2::pixel
*c
, d2::pixel
*w
, d2::point p
) {
2636 if (color
== NULL
) {
2640 assert (color
!= NULL
);
2642 if (!color
->in_bounds(p
)) {
2643 *c
= d2::pixel::undefined();
2644 *w
= d2::pixel::undefined();
2646 *c
= color
->get_bl(p
);
2647 *w
= color_weights
->get_bl(p
);
2651 d2::pixel
get_depth(d2::point p
) {
2652 if (_depth
== NULL
) {
2656 assert (_depth
!= NULL
);
2658 if (!_depth
->in_bounds(p
)) {
2659 return d2::pixel::undefined();
2662 return _depth
->get_bl(p
);
2665 void get_median_depth_and_diff(d2::pixel
*t
, d2::pixel
*f
, d2::point p
) {
2666 if (median_diff
== NULL
&& median_depth
== NULL
)
2669 assert (median_diff
!= NULL
&& median_depth
!= NULL
);
2671 if (!median_diff
->in_bounds(p
)) {
2672 *t
= d2::pixel::undefined();
2673 *f
= d2::pixel::undefined();
2675 *t
= median_depth
->get_bl(p
);
2676 *f
= median_diff
->get_bl(p
);
2683 * Shared view array, indexed by aperture diameter and view number.
2686 std::map
<ale_pos
, std::vector
<shared_view
> > aperture_to_shared_views_map
;
2689 * Method to generate a new stochastic focal view.
2692 pt
get_new_view(ale_pos aperture
) {
2694 ale_pos ofx
= aperture
;
2695 ale_pos ofy
= aperture
;
2697 while (ofx
* ofx
+ ofy
* ofy
> aperture
* aperture
/ 4) {
2698 ofx
= (rand() * aperture
) / RAND_MAX
- aperture
/ 2;
2699 ofy
= (rand() * aperture
) / RAND_MAX
- aperture
/ 2;
2703 * Generate a new view from the given offset.
2706 point new_view
= original_pt
.cw(point(ofx
, ofy
, 0));
2707 pt _pt_new
= original_pt
;
2708 for (int d
= 0; d
< 3; d
++)
2709 _pt_new
.e().set_translation(d
, -new_view
[d
]);
2726 view(shared_view
*sv
, pt _pt
= pt()) {
2729 this->_pt
= sv
->get_pt();
2739 space::node
*get_most_visible(unsigned int i
, unsigned int j
) {
2741 return sv
->get_most_visible(i
, j
);
2744 space::node
*get_most_visible(d2::point p
) {
2746 return sv
->get_most_visible(p
);
2749 d2::pixel
weight(0, 0, 0);
2751 return most_visible_pointwise(&weight
, space::iterate(_pt
.origin()), _pt
, p
);
2755 d2::pixel
get_color(unsigned int i
, unsigned int j
) {
2756 return sv
->get_color(i
, j
);
2759 void get_color_and_weight(d2::pixel
*color
, d2::pixel
*weight
, d2::point p
) {
2761 sv
->get_color_and_weight(color
, weight
, p
);
2766 * Determine weight and color for the given point.
2769 d2::image
*im_point
= new d2::image_ale_real(1, 1, 3);
2770 d2::image
*wt_point
= new d2::image_ale_real(1, 1, 3);
2772 view_recurse(0, im_point
, wt_point
, space::iterate(_pt
.origin()), _pt
, 1, p
, p
);
2774 *color
= im_point
->pix(0, 0);
2775 *weight
= wt_point
->pix(0, 0);
2783 d2::pixel
get_depth(unsigned int i
, unsigned int j
) {
2785 return sv
->get_depth(i
, j
);
2788 void get_median_depth_and_diff(d2::pixel
*depth
, d2::pixel
*diff
, unsigned int i
, unsigned int j
) {
2790 sv
->get_median_depth_and_diff(depth
, diff
, i
, j
);
2793 void get_median_depth_and_diff(d2::pixel
*_depth
, d2::pixel
*_diff
, d2::point p
) {
2795 sv
->get_median_depth_and_diff(_depth
, _diff
, p
);
2800 * Generate a local depth image of required radius.
2805 if (diff_median_radius
+ 1 > radius
)
2806 radius
= diff_median_radius
+ 1;
2807 if (depth_median_radius
> radius
)
2808 radius
= depth_median_radius
;
2810 d2::point pl
= p
- d2::point(radius
, radius
);
2811 d2::point ph
= p
+ d2::point(radius
, radius
);
2812 const d2::image
*local_depth
= depth(_pt
, -1, 1, pl
, ph
);
2815 * Find depth and diff at this point, check for
2816 * undefined values, and generate projections
2817 * of the image corners on the estimated normal
2821 d2::image
*median_diffs
= local_depth
->fcdiff_median((int) floor(diff_median_radius
));
2822 d2::image
*median_depths
= local_depth
->medians((int) floor(depth_median_radius
));
2824 *_depth
= median_depths
->pix((int) radius
, (int) radius
);
2825 *_diff
= median_diffs
->pix((int) radius
, (int) radius
);
2827 delete median_diffs
;
2828 delete median_depths
;
2833 view
get_view(ale_pos aperture
, unsigned index
, unsigned int randomization
) {
2834 if (randomization
== 0) {
2836 while (aperture_to_shared_views_map
[aperture
].size() <= index
) {
2837 aperture_to_shared_views_map
[aperture
].push_back(shared_view(get_new_view(aperture
)));
2840 return view(&(aperture_to_shared_views_map
[aperture
][index
]));
2843 return view(NULL
, get_new_view(aperture
));
2846 view_generator(pt original_pt
) {
2847 this->original_pt
= original_pt
;
2852 * Unfiltered function
2854 static const d2::image
*view_nofilter_focus(pt _pt
, int n
) {
2856 assert ((unsigned int) n
< d2::image_rw::count() || n
< 0);
2859 assert((int) floor(d2::align::of(n
).scaled_height())
2860 == (int) floor(_pt
.scaled_height()));
2861 assert((int) floor(d2::align::of(n
).scaled_width())
2862 == (int) floor(_pt
.scaled_width()));
2865 const d2::image
*depths
= depth(_pt
, n
);
2867 d2::image
*im
= new d2::image_ale_real((int) floor(_pt
.scaled_height()),
2868 (int) floor(_pt
.scaled_width()), 3);
2870 _pt
.view_angle(_pt
.view_angle() * VIEW_ANGLE_MULTIPLIER
);
2872 view_generator
vg(_pt
);
2874 for (unsigned int i
= 0; i
< im
->height(); i
++)
2875 for (unsigned int j
= 0; j
< im
->width(); j
++) {
2877 focus::result _focus
= focus::get(depths
, i
, j
);
2879 if (!finite(_focus
.focal_distance
))
2883 * Data structures for calculating focal statistics.
2886 d2::pixel color
, weight
;
2887 d2::image_weighted_median
*iwm
= NULL
;
2889 if (_focus
.statistic
== 1) {
2890 iwm
= new d2::image_weighted_median(1, 1, 3, _focus
.sample_count
);
2894 * Iterate over views for this focus region.
2897 for (unsigned int v
= 0; v
< _focus
.sample_count
; v
++) {
2899 view_generator::view vw
= vg
.get_view(_focus
.aperture
, v
, _focus
.randomization
);
2901 ui::get()->d3_render_status(0, 1, -1, v
, i
, j
, -1);
2905 * Map the focused point to the new view.
2908 point p
= vw
.get_pt().wp_scaled(_pt
.pw_scaled(point(i
, j
, _focus
.focal_distance
)));
2911 * Determine weight and color for the given point.
2914 d2::pixel view_weight
, view_color
;
2916 vw
.get_color_and_weight(&view_color
, &view_weight
, p
.xy());
2918 if (!color
.finite() || !weight
.finite())
2921 if (_focus
.statistic
== 0) {
2922 color
+= view_color
;
2923 weight
+= view_weight
;
2924 } else if (_focus
.statistic
== 1) {
2925 iwm
->accumulate(0, 0, v
, view_color
, view_weight
);
2930 if (_focus
.statistic
== 1) {
2931 weight
= iwm
->get_weights()->get_pixel(0, 0);
2932 color
= iwm
->get_pixel(0, 0);
2936 if (weight
.min_norm() < encounter_threshold
) {
2937 im
->pix(i
, j
) = d2::pixel::zero() / d2::pixel::zero();
2938 } else if (normalize_weights
)
2939 im
->pix(i
, j
) = color
/ weight
;
2941 im
->pix(i
, j
) = color
;
2950 * Unfiltered function
2952 static const d2::image
*view_nofilter(pt _pt
, int n
) {
2954 if (!focus::is_trivial())
2955 return view_nofilter_focus(_pt
, n
);
2957 assert ((unsigned int) n
< d2::image_rw::count() || n
< 0);
2960 assert((int) floor(d2::align::of(n
).scaled_height())
2961 == (int) floor(_pt
.scaled_height()));
2962 assert((int) floor(d2::align::of(n
).scaled_width())
2963 == (int) floor(_pt
.scaled_width()));
2966 const d2::image
*depths
= depth(_pt
, n
);
2968 d2::image
*im
= new d2::image_ale_real((int) floor(_pt
.scaled_height()),
2969 (int) floor(_pt
.scaled_width()), 3);
2971 _pt
.view_angle(_pt
.view_angle() * VIEW_ANGLE_MULTIPLIER
);
2974 * Use adaptive subspace data.
2977 d2::image
*weights
= new d2::image_ale_real((int) floor(_pt
.scaled_height()),
2978 (int) floor(_pt
.scaled_width()), 3);
2981 * Iterate through subspaces.
2984 space::iterate
si(_pt
.origin());
2986 ui::get()->d3_render_status(0, 0, -1, -1, -1, -1, 0);
2988 view_recurse(0, im
, weights
, si
, _pt
);
2990 for (unsigned int i
= 0; i
< im
->height(); i
++)
2991 for (unsigned int j
= 0; j
< im
->width(); j
++) {
2992 if (weights
->pix(i
, j
).min_norm() < encounter_threshold
2993 || (d3px_count
> 0 && isnan(depths
->pix(i
, j
)[0]))) {
2994 im
->pix(i
, j
) = d2::pixel::zero() / d2::pixel::zero();
2995 weights
->pix(i
, j
) = d2::pixel::zero();
2996 } else if (normalize_weights
)
2997 im
->pix(i
, j
) /= weights
->pix(i
, j
);
3008 * Filtered function.
3010 static const d2::image
*view_filter_focus(pt _pt
, int n
) {
3012 assert ((unsigned int) n
< d2::image_rw::count() || n
< 0);
3015 * Get depth image for focus region determination.
3018 const d2::image
*depths
= depth(_pt
, n
);
3020 unsigned int height
= (unsigned int) floor(_pt
.scaled_height());
3021 unsigned int width
= (unsigned int) floor(_pt
.scaled_width());
3024 * Prepare input frame data.
3027 if (tc_multiplier
== 0)
3030 pt
*_ptf
= new pt
[al
->count()];
3031 std::vector
<space::node
*> *fmv
= new std::vector
<space::node
*>[al
->count()];
3033 for (unsigned int f
= 0; f
< al
->count(); f
++) {
3034 _ptf
[f
] = al
->get(f
)->get_t(0);
3035 fmv
[f
] = most_visible_unscaled(_ptf
[f
]);
3036 std::sort(fmv
[f
].begin(), fmv
[f
].end());
3039 if (tc_multiplier
== 0)
3043 * Open all files for rendering.
3046 d2::image_rw::open_all();
3049 * Prepare data structures for averaging views, as we render
3050 * each view separately. This is spacewise inefficient, but
3051 * is easy to implement given the current operation of the
3055 d2::image_weighted_avg
*iwa
;
3057 if (d3::focus::uses_medians()) {
3058 iwa
= new d2::image_weighted_median(height
, width
, 3, focus::max_samples());
3060 iwa
= new d2::image_weighted_simple(height
, width
, 3, new d2::invariant(NULL
));
3063 _pt
.view_angle(_pt
.view_angle() * VIEW_ANGLE_MULTIPLIER
);
3066 * Prepare view generator.
3069 view_generator
vg(_pt
);
3072 * Render views separately. This is spacewise inefficient,
3073 * but is easy to implement given the current operation of the
3077 for (unsigned int v
= 0; v
< focus::max_samples(); v
++) {
3080 * Generate a new 2D renderer for filtering.
3083 d2::render::reset();
3084 d2::render
*renderer
= d2::render_parse::get(d3chain_type
);
3086 renderer
->init_point_renderer(height
, width
, 3);
3089 * Iterate over output points.
3092 for (unsigned int i
= 0; i
< height
; i
++)
3093 for (unsigned int j
= 0; j
< width
; j
++) {
3095 focus::result _focus
= focus::get(depths
, i
, j
);
3097 if (v
>= _focus
.sample_count
)
3100 if (!finite(_focus
.focal_distance
))
3103 view_generator::view vw
= vg
.get_view(_focus
.aperture
, v
, _focus
.randomization
);
3105 pt _pt_new
= vw
.get_pt();
3107 point p
= _pt_new
.wp_scaled(_pt
.pw_scaled(point(i
, j
, _focus
.focal_distance
)));
3110 * Determine the most-visible subspace.
3113 space::node
*mv
= vw
.get_most_visible(p
.xy());
3119 * Get median depth and diff.
3122 d2::pixel depth
, diff
;
3124 vw
.get_median_depth_and_diff(&depth
, &diff
, p
.xy());
3126 if (!depth
.finite() || !diff
.finite())
3129 point local_points
[3] = {
3130 point(p
[0], p
[1], depth
[0]),
3131 point(p
[0] + 1, p
[1], depth
[0] + diff
[0]),
3132 point(p
[0], p
[1] + 1, depth
[0] + diff
[1])
3136 * Iterate over files.
3139 for (unsigned int f
= 0; f
< d2::image_rw::count(); f
++) {
3141 ui::get()->d3_render_status(1, 1, f
, v
, i
, j
, -1);
3143 if (!visibility_search(fmv
[f
], mv
))
3147 * Determine transformation at (i, j). First
3148 * determine transformation from the output to
3149 * the input, then invert this, as we need the
3150 * inverse transformation for filtering.
3153 d2::point remote_points
[3] = {
3154 _ptf
[f
].wp_unscaled(_pt_new
.pw_scaled(point(local_points
[0]))).xy(),
3155 _ptf
[f
].wp_unscaled(_pt_new
.pw_scaled(point(local_points
[1]))).xy(),
3156 _ptf
[f
].wp_unscaled(_pt_new
.pw_scaled(point(local_points
[2]))).xy()
3160 * Forward matrix for the linear component of the
3164 d2::point forward_matrix
[2] = {
3165 remote_points
[1] - remote_points
[0],
3166 remote_points
[2] - remote_points
[0]
3170 * Inverse matrix for the linear component of
3171 * the transformation. Calculate using the
3175 ale_pos D
= forward_matrix
[0][0] * forward_matrix
[1][1]
3176 - forward_matrix
[0][1] * forward_matrix
[1][0];
3181 d2::point inverse_matrix
[2] = {
3182 d2::point( forward_matrix
[1][1] / D
, -forward_matrix
[1][0] / D
),
3183 d2::point(-forward_matrix
[0][1] / D
, forward_matrix
[0][0] / D
)
3187 * Determine the projective transformation parameters for the
3188 * inverse transformation.
3191 const d2::image
*imf
= d2::image_rw::get_open(f
);
3193 d2::transformation inv_t
= d2::transformation::gpt_identity(imf
, 1);
3195 d2::point local_bounds
[4];
3197 for (int n
= 0; n
< 4; n
++) {
3198 d2::point remote_bound
= d2::point((n
== 1 || n
== 2) ? imf
->height() : 0,
3199 (n
== 2 || n
== 3) ? imf
->width() : 0)
3202 local_bounds
[n
] = d2::point(i
, j
)
3203 + d2::point(remote_bound
[0] * inverse_matrix
[0][0]
3204 + remote_bound
[1] * inverse_matrix
[1][0],
3205 remote_bound
[0] * inverse_matrix
[0][1]
3206 + remote_bound
[1] * inverse_matrix
[1][1]);
3210 if (!local_bounds
[0].finite()
3211 || !local_bounds
[1].finite()
3212 || !local_bounds
[2].finite()
3213 || !local_bounds
[3].finite())
3216 inv_t
.gpt_set(local_bounds
);
3219 * Perform render step for the given frame,
3220 * transformation, and point.
3223 renderer
->point_render(i
, j
, f
, inv_t
);
3227 renderer
->finish_point_rendering();
3229 const d2::image
*im
= renderer
->get_image();
3230 const d2::image
*df
= renderer
->get_defined();
3232 for (unsigned int i
= 0; i
< height
; i
++)
3233 for (unsigned int j
= 0; j
< width
; j
++) {
3234 if (df
->get_pixel(i
, j
).finite()
3235 && df
->get_pixel(i
, j
)[0] > 0)
3236 iwa
->accumulate(i
, j
, v
, im
->get_pixel(i
, j
), d2::pixel(1, 1, 1));
3241 * Close all files and return the result.
3244 d2::image_rw::close_all();
3249 static const d2::image
*view_filter(pt _pt
, int n
) {
3251 if (!focus::is_trivial())
3252 return view_filter_focus(_pt
, n
);
3254 assert ((unsigned int) n
< d2::image_rw::count() || n
< 0);
3257 * Generate a new 2D renderer for filtering.
3260 d2::render::reset();
3261 d2::render
*renderer
= d2::render_parse::get(d3chain_type
);
3264 * Get depth image in order to estimate normals (and hence
3268 const d2::image
*depths
= depth(_pt
, n
);
3270 d2::image
*median_diffs
= depths
->fcdiff_median((int) floor(diff_median_radius
));
3271 d2::image
*median_depths
= depths
->medians((int) floor(depth_median_radius
));
3273 unsigned int height
= (unsigned int) floor(_pt
.scaled_height());
3274 unsigned int width
= (unsigned int) floor(_pt
.scaled_width());
3276 renderer
->init_point_renderer(height
, width
, 3);
3278 _pt
.view_angle(_pt
.view_angle() * VIEW_ANGLE_MULTIPLIER
);
3280 std::vector
<space::node
*> mv
= most_visible_scaled(_pt
);
3282 for (unsigned int f
= 0; f
< d2::image_rw::count(); f
++) {
3284 if (tc_multiplier
== 0)
3287 pt _ptf
= al
->get(f
)->get_t(0);
3289 std::vector
<space::node
*> fmv
= most_visible_unscaled(_ptf
);
3290 std::sort(fmv
.begin(), fmv
.end());
3292 for (unsigned int i
= 0; i
< height
; i
++)
3293 for (unsigned int j
= 0; j
< width
; j
++) {
3295 ui::get()->d3_render_status(1, 0, f
, -1, i
, j
, -1);
3301 int n
= i
* width
+ j
;
3303 if (!visibility_search(fmv
, mv
[n
]))
3307 * Find depth and diff at this point, check for
3308 * undefined values, and generate projections
3309 * of the image corners on the estimated normal
3313 d2::pixel depth
= median_depths
->pix(i
, j
);
3314 d2::pixel diff
= median_diffs
->pix(i
, j
);
3315 // d2::pixel diff = d2::pixel(0, 0, 0);
3317 if (!depth
.finite() || !diff
.finite())
3320 point local_points
[3] = {
3321 point(i
, j
, depth
[0]),
3322 point(i
+ 1, j
, depth
[0] + diff
[0]),
3323 point(i
, j
+ 1, depth
[0] + diff
[1])
3327 * Determine transformation at (i, j). First
3328 * determine transformation from the output to
3329 * the input, then invert this, as we need the
3330 * inverse transformation for filtering.
3333 d2::point remote_points
[3] = {
3334 _ptf
.wp_unscaled(_pt
.pw_scaled(point(local_points
[0]))).xy(),
3335 _ptf
.wp_unscaled(_pt
.pw_scaled(point(local_points
[1]))).xy(),
3336 _ptf
.wp_unscaled(_pt
.pw_scaled(point(local_points
[2]))).xy()
3340 * Forward matrix for the linear component of the
3344 d2::point forward_matrix
[2] = {
3345 remote_points
[1] - remote_points
[0],
3346 remote_points
[2] - remote_points
[0]
3350 * Inverse matrix for the linear component of
3351 * the transformation. Calculate using the
3355 ale_pos D
= forward_matrix
[0][0] * forward_matrix
[1][1]
3356 - forward_matrix
[0][1] * forward_matrix
[1][0];
3361 d2::point inverse_matrix
[2] = {
3362 d2::point( forward_matrix
[1][1] / D
, -forward_matrix
[1][0] / D
),
3363 d2::point(-forward_matrix
[0][1] / D
, forward_matrix
[0][0] / D
)
3367 * Determine the projective transformation parameters for the
3368 * inverse transformation.
3371 const d2::image
*imf
= d2::image_rw::open(f
);
3373 d2::transformation inv_t
= d2::transformation::gpt_identity(imf
, 1);
3375 d2::point local_bounds
[4];
3377 for (int n
= 0; n
< 4; n
++) {
3378 d2::point remote_bound
= d2::point((n
== 1 || n
== 2) ? imf
->height() : 0,
3379 (n
== 2 || n
== 3) ? imf
->width() : 0)
3382 local_bounds
[n
] = local_points
[0].xy()
3383 + d2::point(remote_bound
[0] * inverse_matrix
[0][0]
3384 + remote_bound
[1] * inverse_matrix
[1][0],
3385 remote_bound
[0] * inverse_matrix
[0][1]
3386 + remote_bound
[1] * inverse_matrix
[1][1]);
3389 inv_t
.gpt_set(local_bounds
);
3391 d2::image_rw::close(f
);
3394 * Perform render step for the given frame,
3395 * transformation, and point.
3398 d2::image_rw::open(f
);
3399 renderer
->point_render(i
, j
, f
, inv_t
);
3400 d2::image_rw::close(f
);
3403 if (tc_multiplier
== 0)
3407 renderer
->finish_point_rendering();
3409 return renderer
->get_image();
3415 static const d2::image
*view(pt _pt
, int n
= -1) {
3417 assert ((unsigned int) n
< d2::image_rw::count() || n
< 0);
3420 return view_filter(_pt
, n
);
3422 return view_nofilter(_pt
, n
);
3426 static void tcem(double _tcem
) {
3427 tc_multiplier
= _tcem
;
3430 static void oui(unsigned int _oui
) {
3431 ou_iterations
= _oui
;
3434 static void pa(unsigned int _pa
) {
3435 pairwise_ambiguity
= _pa
;
3438 static void pc(const char *_pc
) {
3439 pairwise_comparisons
= _pc
;
3442 static void d3px(int _d3px_count
, double *_d3px_parameters
) {
3443 d3px_count
= _d3px_count
;
3444 d3px_parameters
= _d3px_parameters
;
3447 static void fx(double _fx
) {
3448 falloff_exponent
= _fx
;
3452 normalize_weights
= 1;
3455 static void no_nw() {
3456 normalize_weights
= 0;
3459 static void nofilter() {
3463 static void filter() {
3467 static void set_filter_type(const char *type
) {
3468 d3chain_type
= type
;
3471 static void set_subspace_traverse() {
3472 subspace_traverse
= 1;
3475 static int excluded(point p
) {
3476 for (int n
= 0; n
< d3px_count
; n
++) {
3477 double *region
= d3px_parameters
+ (6 * n
);
3478 if (p
[0] >= region
[0]
3479 && p
[0] <= region
[1]
3480 && p
[1] >= region
[2]
3481 && p
[1] <= region
[3]
3482 && p
[2] >= region
[4]
3483 && p
[2] <= region
[5])
3491 * This function returns true if a space is completely excluded.
3493 static int excluded(const space::traverse
&st
) {
3494 for (int n
= 0; n
< d3px_count
; n
++) {
3495 double *region
= d3px_parameters
+ (6 * n
);
3496 if (st
.get_min()[0] >= region
[0]
3497 && st
.get_max()[0] <= region
[1]
3498 && st
.get_min()[1] >= region
[2]
3499 && st
.get_max()[1] <= region
[3]
3500 && st
.get_min()[2] >= region
[4]
3501 && st
.get_max()[2] <= region
[5])
3508 static const d2::image
*view(unsigned int n
) {
3510 assert (n
< d2::image_rw::count());
3512 pt _pt
= align::projective(n
);
3514 return view(_pt
, n
);
3517 typedef struct {point iw
; point ip
, is
;} analytic
;
3518 typedef std::multimap
<ale_real
,analytic
> score_map
;
3519 typedef std::pair
<ale_real
,analytic
> score_map_element
;
3524 static std::vector
<pt
> make_pt_list(const char *d_out
[], const char *v_out
[],
3525 std::map
<const char *, pt
> *d3_depth_pt
,
3526 std::map
<const char *, pt
> *d3_output_pt
) {
3528 std::vector
<pt
> result
;
3530 for (unsigned int n
= 0; n
< d2::image_rw::count(); n
++) {
3531 if (d_out
[n
] || v_out
[n
]) {
3532 result
.push_back(align::projective(n
));
3536 for (std::map
<const char *, pt
>::iterator i
= d3_depth_pt
->begin(); i
!= d3_depth_pt
->end(); i
++) {
3537 result
.push_back(i
->second
);
3540 for (std::map
<const char *, pt
>::iterator i
= d3_output_pt
->begin(); i
!= d3_output_pt
->end(); i
++) {
3541 result
.push_back(i
->second
);
3548 * Get a trilinear coordinate for an anisotropic candidate cell.
3550 static ale_pos
get_trilinear_coordinate(point min
, point max
, pt _pt
) {
3552 d2::point local_min
, local_max
;
3554 local_min
= _pt
.wp_unscaled(min
).xy();
3555 local_max
= _pt
.wp_unscaled(min
).xy();
3557 point cell
[2] = {min
, max
};
3560 * Determine the view-local extrema in 2 dimensions.
3563 for (int r
= 1; r
< 8; r
++) {
3564 point local
= _pt
.wp_unscaled(point(cell
[r
>>2][0], cell
[(r
>>1)%2][1], cell
[r
%2][2]));
3566 for (int d
= 0; d
< 2; d
++) {
3567 if (local
[d
] < local_min
[d
])
3568 local_min
[d
] = local
[d
];
3569 if (local
[d
] > local_max
[d
])
3570 local_max
[d
] = local
[d
];
3571 if (isnan(local
[d
]))
3576 ale_pos diameter
= (local_max
- local_min
).norm();
3578 return log(diameter
/ sqrt(2)) / log(2);
3582 * Check whether a cell is visible from a given viewpoint. This function
3583 * is guaranteed to return 1 when a cell is visible, but it is not guaranteed
3584 * to return 0 when a cell is invisible.
3586 static int pt_might_be_visible(const pt
&viewpoint
, point min
, point max
) {
3588 int doc
= (rand() % 100000) ? 0 : 1;
3591 fprintf(stderr
, "checking visibility:\n");
3593 point cell
[2] = {min
, max
};
3596 * Cycle through all vertices of the cell to check certain
3599 int pos
[3] = {0, 0, 0};
3600 int neg
[3] = {0, 0, 0};
3601 for (int i
= 0; i
< 2; i
++)
3602 for (int j
= 0; j
< 2; j
++)
3603 for (int k
= 0; k
< 2; k
++) {
3604 point p
= viewpoint
.wp_unscaled(point(cell
[i
][0], cell
[j
][1], cell
[k
][2]));
3606 if (p
[2] < 0 && viewpoint
.unscaled_in_bounds(p
))
3615 for (int d
= 0; d
< 2; d
++)
3619 fprintf(stderr
, "\t[%f %f %f] --> [%f %f %f]\n",
3620 cell
[i
][0], cell
[j
][1], cell
[k
][2],
3623 for (int d
= 0; d
< 3; d
++)
3627 if (p
[0] <= viewpoint
.unscaled_height() - 1)
3630 if (p
[1] <= viewpoint
.unscaled_width() - 1)
3650 * Check whether a cell is output-visible.
3652 static int output_might_be_visible(const std::vector
<pt
> &pt_outputs
, point min
, point max
) {
3653 for (unsigned int n
= 0; n
< pt_outputs
.size(); n
++)
3654 if (pt_might_be_visible(pt_outputs
[n
], min
, max
))
3660 * Check whether a cell is input-visible.
3662 static int input_might_be_visible(unsigned int f
, point min
, point max
) {
3663 return pt_might_be_visible(align::projective(f
), min
, max
);
3667 * Return true if a cell fails an output resolution bound.
3669 static int fails_output_resolution_bound(point min
, point max
, const std::vector
<pt
> &pt_outputs
) {
3670 for (unsigned int n
= 0; n
< pt_outputs
.size(); n
++) {
3672 point p
= pt_outputs
[n
].centroid(min
, max
);
3677 if (get_trilinear_coordinate(min
, max
, pt_outputs
[n
]) < output_decimation_preferred
)
3685 * Check lower-bound resolution constraints
3687 static int exceeds_resolution_lower_bounds(unsigned int f1
, unsigned int f2
,
3688 point min
, point max
, const std::vector
<pt
> &pt_outputs
) {
3690 pt _pt
= al
->get(f1
)->get_t(0);
3692 if (get_trilinear_coordinate(min
, max
, _pt
) < input_decimation_lower
)
3695 if (fails_output_resolution_bound(min
, max
, pt_outputs
))
3698 if (get_trilinear_coordinate(min
, max
, _pt
) < primary_decimation_upper
)
3705 * Try the candidate nearest to the specified cell.
3707 static void try_nearest_candidate(unsigned int f1
, unsigned int f2
, candidates
*c
, point min
, point max
) {
3708 point centroid
= (max
+ min
) / 2;
3709 pt _pt
[2] = { al
->get(f1
)->get_t(0), al
->get(f2
)->get_t(0) };
3712 // fprintf(stderr, "[tnc n=%f %f %f x=%f %f %f]\n", min[0], min[1], min[2], max[0], max[1], max[2]);
3715 * Reject clipping plane violations.
3718 if (centroid
[2] > front_clip
3719 || centroid
[2] < rear_clip
)
3723 * Calculate projections.
3726 for (int n
= 0; n
< 2; n
++) {
3728 p
[n
] = _pt
[n
].wp_unscaled(centroid
);
3730 if (!_pt
[n
].unscaled_in_bounds(p
[n
]))
3733 // fprintf(stderr, ":");
3740 int tc
= (int) round(get_trilinear_coordinate(min
, max
, _pt
[0]));
3741 int stc
= (int) round(get_trilinear_coordinate(min
, max
, _pt
[1]));
3743 while (tc
< input_decimation_lower
|| stc
< input_decimation_lower
) {
3748 if (tc
> primary_decimation_upper
)
3752 * Calculate score from color match. Assume for now
3753 * that the transformation can be approximated locally
3754 * with a translation.
3758 ale_pos divisor
= 0;
3759 ale_pos l1_multiplier
= 0.125;
3760 lod_image
*if1
= al
->get(f1
);
3761 lod_image
*if2
= al
->get(f2
);
3763 if (if1
->in_bounds(p
[0].xy())
3764 && if2
->in_bounds(p
[1].xy())) {
3765 divisor
+= 1 - l1_multiplier
;
3766 score
+= (1 - l1_multiplier
)
3767 * (if1
->get_tl(p
[0].xy(), tc
) - if2
->get_tl(p
[1].xy(), stc
)).normsq();
3770 for (int iii
= -1; iii
<= 1; iii
++)
3771 for (int jjj
= -1; jjj
<= 1; jjj
++) {
3772 d2::point
t(iii
, jjj
);
3774 if (!if1
->in_bounds(p
[0].xy() + t
)
3775 || !if2
->in_bounds(p
[1].xy() + t
))
3778 divisor
+= l1_multiplier
;
3779 score
+= l1_multiplier
3780 * (if1
->get_tl(p
[0].xy() + t
, tc
) - if2
->get_tl(p
[1].xy() + t
, tc
)).normsq();
3785 * Include third-camera contributions in the score.
3788 if (tc_multiplier
!= 0)
3789 for (unsigned int n
= 0; n
< d2::image_rw::count(); n
++) {
3790 if (n
== f1
|| n
== f2
)
3793 lod_image
*ifn
= al
->get(n
);
3794 pt _ptn
= ifn
->get_t(0);
3795 point pn
= _ptn
.wp_unscaled(centroid
);
3797 if (!_ptn
.unscaled_in_bounds(pn
))
3803 ale_pos ttc
= get_trilinear_coordinate(min
, max
, _ptn
);
3805 divisor
+= tc_multiplier
;
3806 score
+= tc_multiplier
3807 * (if1
->get_tl(p
[0].xy(), tc
) - ifn
->get_tl(pn
.xy(), ttc
)).normsq();
3810 c
->add_candidate(p
[0], tc
, score
/ divisor
);
3814 * Check for cells that are completely clipped.
3816 static int completely_clipped(point min
, point max
) {
3817 return (min
[2] > front_clip
3818 || max
[2] < rear_clip
);
3822 * Update extremum variables for cell points mapped to a particular view.
3824 static void update_extrema(point min
, point max
, pt _pt
, int *extreme_dim
, ale_pos
*extreme_ratio
) {
3826 point local_min
, local_max
;
3828 local_min
= _pt
.wp_unscaled(min
);
3829 local_max
= _pt
.wp_unscaled(min
);
3831 point cell
[2] = {min
, max
};
3833 int near_vertex
= 0;
3836 * Determine the view-local extrema in all dimensions, and
3837 * determine the vertex of closest z coordinate.
3840 for (int r
= 1; r
< 8; r
++) {
3841 point local
= _pt
.wp_unscaled(point(cell
[r
>>2][0], cell
[(r
>>1)%2][1], cell
[r
%2][2]));
3843 for (int d
= 0; d
< 3; d
++) {
3844 if (local
[d
] < local_min
[d
])
3845 local_min
[d
] = local
[d
];
3846 if (local
[d
] > local_max
[d
])
3847 local_max
[d
] = local
[d
];
3850 if (local
[2] == local_max
[2])
3854 ale_pos diameter
= (local_max
.xy() - local_min
.xy()).norm();
3857 * Update extrema as necessary for each dimension.
3860 for (int d
= 0; d
< 3; d
++) {
3862 int r
= near_vertex
;
3864 int p1
[3] = {r
>>2, (r
>>1)%2, r
%2};
3865 int p2
[3] = {r
>>2, (r
>>1)%2, r
%2};
3869 ale_pos local_distance
= (_pt
.wp_unscaled(point(cell
[p1
[0]][0], cell
[p1
[1]][1], cell
[p1
[2]][2])).xy()
3870 - _pt
.wp_unscaled(point(cell
[p2
[0]][0], cell
[p2
[1]][1], cell
[p2
[2]][2])).xy()).norm();
3872 if (local_distance
/ diameter
> *extreme_ratio
) {
3873 *extreme_ratio
= local_distance
/ diameter
;
3880 * Get the next split dimension.
3882 static int get_next_split(int f1
, int f2
, point min
, point max
, const std::vector
<pt
> &pt_outputs
) {
3883 for (int d
= 0; d
< 3; d
++)
3884 if (isinf(min
[d
]) || isinf(max
[d
]))
3885 return space::traverse::get_next_split(min
, max
);
3887 int extreme_dim
= 0;
3888 ale_pos extreme_ratio
= 0;
3890 update_extrema(min
, max
, al
->get(f1
)->get_t(0), &extreme_dim
, &extreme_ratio
);
3891 update_extrema(min
, max
, al
->get(f2
)->get_t(0), &extreme_dim
, &extreme_ratio
);
3893 for (unsigned int n
= 0; n
< pt_outputs
.size(); n
++) {
3894 update_extrema(min
, max
, pt_outputs
[n
], &extreme_dim
, &extreme_ratio
);
3901 * Find candidates for subspace creation.
3903 static void find_candidates(unsigned int f1
, unsigned int f2
, candidates
*c
, point min
, point max
,
3904 const std::vector
<pt
> &pt_outputs
, int depth
= 0) {
3908 if (min
[0] < 20.0001 && max
[0] > 20.0001
3909 && min
[1] < 20.0001 && max
[1] > 20.0001
3910 && min
[2] < 0.0001 && max
[2] > 0.0001)
3914 for (int i
= depth
; i
> 0; i
--) {
3915 fprintf(stderr
, "+");
3917 fprintf(stderr
, "[fc n=%f %f %f x=%f %f %f]\n",
3918 min
[0], min
[1], min
[2], max
[0], max
[1], max
[2]);
3921 if (completely_clipped(min
, max
)) {
3923 fprintf(stderr
, "c");
3927 if (!input_might_be_visible(f1
, min
, max
)
3928 || !input_might_be_visible(f2
, min
, max
)) {
3930 fprintf(stderr
, "v");
3934 if (output_clip
&& !output_might_be_visible(pt_outputs
, min
, max
)) {
3936 fprintf(stderr
, "o");
3940 if (exceeds_resolution_lower_bounds(f1
, f2
, min
, max
, pt_outputs
)) {
3941 if (!(rand() % 100000))
3942 fprintf(stderr
, "([%f %f %f], [%f %f %f]) at %d\n",
3943 min
[0], min
[1], min
[2],
3944 max
[0], max
[1], max
[2],
3948 fprintf(stderr
, "t");
3950 try_nearest_candidate(f1
, f2
, c
, min
, max
);
3954 point new_cells
[2][2];
3956 if (!space::traverse::get_next_cells(get_next_split(f1
, f2
, min
, max
, pt_outputs
), min
, max
, new_cells
)) {
3958 fprintf(stderr
, "n");
3963 fprintf(stderr
, "nc[0][0]=%f %f %f nc[0][1]=%f %f %f nc[1][0]=%f %f %f nc[1][1]=%f %f %f\n",
3975 new_cells
[1][1][2]);
3978 find_candidates(f1
, f2
, c
, new_cells
[0][0], new_cells
[0][1], pt_outputs
, depth
+ 1);
3979 find_candidates(f1
, f2
, c
, new_cells
[1][0], new_cells
[1][1], pt_outputs
, depth
+ 1);
3983 * Generate a map from scores to 3D points for various depths at point (i, j) in f1, at
3984 * lowest resolution.
3986 static score_map
p2f_score_map(unsigned int f1
, unsigned int f2
, unsigned int i
, unsigned int j
) {
3990 pt _pt1
= al
->get(f1
)->get_t(primary_decimation_upper
);
3991 pt _pt2
= al
->get(f2
)->get_t(primary_decimation_upper
);
3993 const d2::image
*if1
= al
->get(f1
)->get_image(primary_decimation_upper
);
3994 const d2::image
*if2
= al
->get(f2
)->get_image(primary_decimation_upper
);
3995 ale_pos pdu_scale
= pow(2, primary_decimation_upper
);
3998 * Get the pixel color in the primary frame
4001 // d2::pixel color_primary = if1->get_pixel(i, j);
4004 * Map two depths to the secondary frame.
4007 point p1
= _pt2
.wp_unscaled(_pt1
.pw_unscaled(point(i
, j
, 1000)));
4008 point p2
= _pt2
.wp_unscaled(_pt1
.pw_unscaled(point(i
, j
, -1000)));
4010 // fprintf(stderr, "%d->%d (%d, %d) point pair: (%d, %d, %d -> %f, %f), (%d, %d, %d -> %f, %f)\n",
4011 // f1, f2, i, j, i, j, 1000, p1[0], p1[1], i, j, -1000, p2[0], p2[1]);
4012 // _pt1.debug_output();
4013 // _pt2.debug_output();
4017 * For cases where the mapped points define a
4018 * line and where points on the line fall
4019 * within the defined area of the frame,
4020 * determine the starting point for inspection.
4021 * In other cases, continue to the next pixel.
4024 ale_pos diff_i
= p2
[0] - p1
[0];
4025 ale_pos diff_j
= p2
[1] - p1
[1];
4026 ale_pos slope
= diff_j
/ diff_i
;
4030 fprintf(stderr
, "%d->%d (%d, %d) has undefined slope\n",
4036 * Make absurdly large/small slopes either infinity, negative infinity, or zero.
4039 if (fabs(slope
) > if2
->width() * 100) {
4042 double inf
= one
/ zero
;
4044 } else if (slope
< 1 / (double) if2
->height() / 100
4045 && slope
> -1/ (double) if2
->height() / 100) {
4049 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4051 ale_pos top_intersect
= p1
[1] - p1
[0] * slope
;
4052 ale_pos lef_intersect
= p1
[0] - p1
[1] / slope
;
4053 ale_pos rig_intersect
= p1
[0] - (p1
[1] - if2
->width() + 2) / slope
;
4056 // fprintf(stderr, "slope == %f\n", slope);
4060 // fprintf(stderr, "case 0\n");
4061 sp_i
= lef_intersect
;
4063 } else if (finite(slope
) && top_intersect
>= 0 && top_intersect
< if2
->width() - 1) {
4064 // fprintf(stderr, "case 1\n");
4066 sp_j
= top_intersect
;
4067 } else if (slope
> 0 && lef_intersect
>= 0 && lef_intersect
<= if2
->height() - 1) {
4068 // fprintf(stderr, "case 2\n");
4069 sp_i
= lef_intersect
;
4071 } else if (slope
< 0 && rig_intersect
>= 0 && rig_intersect
<= if2
->height() - 1) {
4072 // fprintf(stderr, "case 3\n");
4073 sp_i
= rig_intersect
;
4074 sp_j
= if2
->width() - 2;
4076 // fprintf(stderr, "case 4\n");
4077 // fprintf(stderr, "%d->%d (%d, %d) does not intersect the defined area\n",
4083 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4086 * Determine increment values for examining
4087 * point, ensuring that incr_i is always
4091 ale_pos incr_i
, incr_j
;
4093 if (fabs(diff_i
) > fabs(diff_j
)) {
4096 } else if (slope
> 0) {
4100 incr_i
= -1 / slope
;
4104 // fprintf(stderr, "%d->%d (%d, %d) increments are (%f, %f)\n",
4105 // f1, f2, i, j, incr_i, incr_j);
4108 * Examine regions near the projected line.
4111 for (ale_pos ii
= sp_i
, jj
= sp_j
;
4112 ii
<= if2
->height() - 1 && jj
<= if2
->width() - 1 && ii
>= 0 && jj
>= 0;
4113 ii
+= incr_i
, jj
+= incr_j
) {
4115 // fprintf(stderr, "%d->%d (%d, %d) checking (%f, %f)\n",
4116 // f1, f2, i, j, ii, jj);
4120 * Check for higher, lower, and nearby points.
4127 int higher
= 0, lower
= 0, nearby
= 0;
4129 for (int iii
= 0; iii
< 2; iii
++)
4130 for (int jjj
= 0; jjj
< 2; jjj
++) {
4131 d2::pixel p
= if2
->get_pixel((int) floor(ii
) + iii
, (int) floor(jj
) + jjj
);
4133 for (int k
= 0; k
< 3; k
++) {
4134 int bitmask
= (int) pow(2, k
);
4136 if (p
[k
] > color_primary
[k
])
4138 if (p
[k
] < color_primary
[k
])
4140 if (fabs(p
[k
] - color_primary
[k
]) < nearness
)
4146 * If this is not a region of interest,
4151 fprintf(stderr
, "score map (%u, %u) line %u\n", i
, j
, __LINE__
);
4153 // if (((higher & lower) | nearby) != 0x7)
4156 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4158 // fprintf(stderr, "%d->%d (%d, %d) accepted (%f, %f)\n",
4159 // f1, f2, i, j, ii, jj);
4162 * Create an orthonormal basis to
4163 * determine line intersection.
4166 point bp0
= _pt1
.pw_unscaled(point(i
, j
, 0));
4167 point bp1
= _pt1
.pw_unscaled(point(i
, j
, 10));
4168 point bp2
= _pt2
.pw_unscaled(point(ii
, jj
, 0));
4170 point foo
= _pt1
.wp_unscaled(bp0
);
4171 // fprintf(stderr, "(%d, %d, 0) transformed to world and back is: (%f, %f, %f)\n",
4172 // i, j, foo[0], foo[1], foo[2]);
4174 foo
= _pt1
.wp_unscaled(bp1
);
4175 // fprintf(stderr, "(%d, %d, 10) transformed to world and back is: (%f, %f, %f)\n",
4176 // i, j, foo[0], foo[1], foo[2]);
4178 point b0
= (bp1
- bp0
).normalize();
4179 point b1n
= bp2
- bp0
;
4180 point b1
= (b1n
- b1n
.dproduct(b0
) * b0
).normalize();
4181 point b2
= point(0, 0, 0).xproduct(b0
, b1
).normalize(); // Should already have norm=1
4184 foo
= _pt1
.wp_unscaled(bp0
+ 30 * b0
);
4187 * Select a fourth point to define a second line.
4190 point p3
= _pt2
.pw_unscaled(point(ii
, jj
, 10));
4193 * Representation in the new basis.
4196 d2::point nbp0
= d2::point(bp0
.dproduct(b0
), bp0
.dproduct(b1
));
4197 // d2::point nbp1 = d2::point(bp1.dproduct(b0), bp1.dproduct(b1));
4198 d2::point nbp2
= d2::point(bp2
.dproduct(b0
), bp2
.dproduct(b1
));
4199 d2::point np3
= d2::point( p3
.dproduct(b0
), p3
.dproduct(b1
));
4202 * Determine intersection of line
4203 * (nbp0, nbp1), which is parallel to
4204 * b0, with line (nbp2, np3).
4208 * XXX: a stronger check would be
4209 * better here, e.g., involving the
4210 * ratio (np3[0] - nbp2[0]) / (np3[1] -
4211 * nbp2[1]). Also, acceptance of these
4212 * cases is probably better than
4217 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4219 if (np3
[1] - nbp2
[1] == 0)
4223 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4225 d2::point intersection
= d2::point(nbp2
[0]
4226 + (nbp0
[1] - nbp2
[1]) * (np3
[0] - nbp2
[0]) / (np3
[1] - nbp2
[1]),
4229 ale_pos b2_offset
= b2
.dproduct(bp0
);
4232 * Map the intersection back to the world
4236 point iw
= intersection
[0] * b0
+ intersection
[1] * b1
+ b2_offset
* b2
;
4239 * Reject intersection points behind a
4243 point icp
= _pt1
.wc(iw
);
4244 point ics
= _pt2
.wc(iw
);
4247 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4249 if (icp
[2] >= 0 || ics
[2] >= 0)
4253 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4256 * Reject clipping plane violations.
4260 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4262 if (iw
[2] > front_clip
4263 || iw
[2] < rear_clip
)
4267 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4273 point ip
= _pt1
.wp_unscaled(iw
);
4275 point is
= _pt2
.wp_unscaled(iw
);
4277 analytic _a
= { iw
, ip
, is
};
4280 * Calculate score from color match. Assume for now
4281 * that the transformation can be approximated locally
4282 * with a translation.
4286 ale_pos divisor
= 0;
4287 ale_pos l1_multiplier
= 0.125;
4289 if (if1
->in_bounds(ip
.xy())
4290 && if2
->in_bounds(is
.xy())
4291 && !d2::render::is_excluded_f(ip
.xy() * pdu_scale
, f1
)
4292 && !d2::render::is_excluded_f(is
.xy() * pdu_scale
, f2
)) {
4293 divisor
+= 1 - l1_multiplier
;
4294 score
+= (1 - l1_multiplier
)
4295 * (if1
->get_bl(ip
.xy()) - if2
->get_bl(is
.xy())).normsq();
4299 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4301 for (int iii
= -1; iii
<= 1; iii
++)
4302 for (int jjj
= -1; jjj
<= 1; jjj
++) {
4303 d2::point
t(iii
, jjj
);
4305 if (!if1
->in_bounds(ip
.xy() + t
)
4306 || !if2
->in_bounds(is
.xy() + t
)
4307 || d2::render::is_excluded_f(ip
.xy() * pdu_scale
, f1
)
4308 || d2::render::is_excluded_f(is
.xy() * pdu_scale
, f2
))
4311 divisor
+= l1_multiplier
;
4312 score
+= l1_multiplier
4313 * (if1
->get_bl(ip
.xy() + t
) - if2
->get_bl(is
.xy() + t
)).normsq();
4318 * Include third-camera contributions in the score.
4321 if (tc_multiplier
!= 0)
4322 for (unsigned int f
= 0; f
< d2::image_rw::count(); f
++) {
4323 if (f
== f1
|| f
== f2
)
4326 const d2::image
*if3
= al
->get(f
)->get_image(primary_decimation_upper
);
4327 pt _pt3
= al
->get(f
)->get_t(primary_decimation_upper
);
4329 point p
= _pt3
.wp_unscaled(iw
);
4331 if (!if3
->in_bounds(p
.xy())
4332 || !if1
->in_bounds(ip
.xy())
4333 || d2::render::is_excluded_f(p
.xy() * pdu_scale
, f
)
4334 || d2::render::is_excluded_f(ip
.xy() * pdu_scale
, f1
))
4337 divisor
+= tc_multiplier
;
4338 score
+= tc_multiplier
4339 * (if1
->get_bl(ip
.xy()) - if3
->get_bl(p
.xy())).normsq();
4345 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4348 * Reject points with undefined score.
4352 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4354 if (!finite(score
/ divisor
))
4358 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4362 * XXX: reject points not on the z=-27.882252 plane.
4366 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4368 if (_a
.ip
[2] > -27 || _a
.ip
[2] < -28)
4373 // fprintf(stderr, "score map (%u, %u) line %u\n", i, j, __LINE__);
4376 * Add the point to the score map.
4379 // d2::pixel c_ip = if1->in_bounds(ip.xy()) ? if1->get_bl(ip.xy())
4381 // d2::pixel c_is = if2->in_bounds(is.xy()) ? if2->get_bl(is.xy())
4384 // fprintf(stderr, "Candidate subspace: f1=%u f2=%u i=%u j=%u ii=%f jj=%f"
4385 // "cp=[%f %f %f] cs=[%f %f %f]\n",
4386 // f1, f2, i, j, ii, jj, c_ip[0], c_ip[1], c_ip[2],
4387 // c_is[0], c_is[1], c_is[2]);
4389 result
.insert(score_map_element(score
/ divisor
, _a
));
4392 // fprintf(stderr, "Iterating through the score map:\n");
4394 // for (score_map::iterator smi = result.begin(); smi != result.end(); smi++) {
4395 // fprintf(stderr, "%f ", smi->first);
4398 // fprintf(stderr, "\n");
4405 * Attempt to refine space around a point, to high and low resolutions
4406 * resulting in two resolutions in total.
4409 static space::traverse
refine_space(point iw
, ale_pos target_dim
, int use_filler
) {
4411 space::traverse st
= space::traverse::root();
4413 if (!st
.includes(iw
)) {
4418 int lr_done
= !use_filler
;
4421 * Loop until all resolutions of interest have been generated.
4426 point p
[2] = { st
.get_min(), st
.get_max() };
4428 ale_pos dim_max
= 0;
4430 for (int d
= 0; d
< 3; d
++) {
4431 ale_pos d_value
= fabs(p
[0][d
] - p
[1][d
]);
4432 if (d_value
> dim_max
)
4437 * Generate any new desired spatial registers.
4440 for (int f
= 0; f
< 2; f
++) {
4446 if (dim_max
< 2 * target_dim
4448 if (spatial_info_map
.find(st
.get_node()) == spatial_info_map
.end()) {
4449 spatial_info_map
[st
.get_node()];
4450 ui::get()->d3_increment_spaces();
4459 if (dim_max
< target_dim
) {
4460 if (spatial_info_map
.find(st
.get_node()) == spatial_info_map
.end()) {
4461 spatial_info_map
[st
.get_node()];
4462 ui::get()->d3_increment_spaces();
4469 * Check precision before analyzing space further.
4472 if (st
.precision_wall()) {
4473 fprintf(stderr
, "\n\n*** Error: reached subspace precision wall ***\n\n");
4478 if (st
.positive().includes(iw
)) {
4481 } else if (st
.negative().includes(iw
)) {
4485 fprintf(stderr
, "failed iw = (%f, %f, %f)\n",
4486 iw
[0], iw
[1], iw
[2]);
4493 * Calculate target dimension
4496 static ale_pos
calc_target_dim(point iw
, pt _pt
, const char *d_out
[], const char *v_out
[],
4497 std::map
<const char *, pt
> *d3_depth_pt
,
4498 std::map
<const char *, pt
> *d3_output_pt
) {
4500 ale_pos result
= _pt
.distance_1d(iw
, primary_decimation_upper
);
4502 for (unsigned int n
= 0; n
< d2::image_rw::count(); n
++) {
4503 if (d_out
[n
] && align::projective(n
).distance_1d(iw
, 0) < result
)
4504 result
= align::projective(n
).distance_1d(iw
, 0);
4505 if (v_out
[n
] && align::projective(n
).distance_1d(iw
, 0) < result
)
4506 result
= align::projective(n
).distance_1d(iw
, 0);
4509 for (std::map
<const char *, pt
>::iterator i
= d3_output_pt
->begin(); i
!= d3_output_pt
->end(); i
++) {
4510 if (i
->second
.distance_1d(iw
, 0) < result
)
4511 result
= i
->second
.distance_1d(iw
, 0);
4514 for (std::map
<const char *, pt
>::iterator i
= d3_depth_pt
->begin(); i
!= d3_depth_pt
->end(); i
++) {
4515 if (i
->second
.distance_1d(iw
, 0) < result
)
4516 result
= i
->second
.distance_1d(iw
, 0);
4519 assert (result
> 0);
4525 * Calculate level of detail for a given viewpoint.
4528 static int calc_lod(ale_pos depth1
, pt _pt
, ale_pos target_dim
) {
4529 return (int) round(_pt
.trilinear_coordinate(depth1
, target_dim
* sqrt(2)));
4533 * Calculate depth range for a given pair of viewpoints.
4536 static ale_pos
calc_depth_range(point iw
, pt _pt1
, pt _pt2
) {
4538 point ip
= _pt1
.wp_unscaled(iw
);
4540 ale_pos reference_change
= fabs(ip
[2] / 1000);
4542 point iw1
= _pt1
.pw_scaled(ip
+ point(0, 0, reference_change
));
4543 point iw2
= _pt1
.pw_scaled(ip
- point(0, 0, reference_change
));
4545 point is
= _pt2
.wc(iw
);
4546 point is1
= _pt2
.wc(iw1
);
4547 point is2
= _pt2
.wc(iw2
);
4551 ale_pos d1
= (is1
.xy() - is
.xy()).norm();
4552 ale_pos d2
= (is2
.xy() - is
.xy()).norm();
4554 if (is1
[2] < 0 && is2
[2] < 0) {
4557 return reference_change
/ d1
;
4559 return reference_change
/ d2
;
4563 return reference_change
/ d1
;
4566 return reference_change
/ d2
;
4572 * Calculate a refined point for a given set of parameters.
4575 static point
get_refined_point(pt _pt1
, pt _pt2
, int i
, int j
,
4576 int f1
, int f2
, int lod1
, int lod2
, ale_pos depth
,
4577 ale_pos depth_range
) {
4579 d2::pixel comparison_color
= al
->get(f1
)->get_image(lod1
)->get_pixel(i
, j
);
4582 ale_pos best_depth
= depth
;
4584 for (ale_pos d
= depth
- depth_range
; d
< depth
+ depth_range
; d
+= depth_range
/ 10) {
4589 point iw
= _pt1
.pw_unscaled(point(i
, j
, d
));
4590 point is
= _pt2
.wp_unscaled(iw
);
4595 if (!al
->get(f2
)->get_image(lod2
)->in_bounds(is
.xy()))
4598 ale_pos error
= (comparison_color
- al
->get(f2
)->get_image(lod2
)->get_bl(is
.xy())).norm();
4600 if (error
< best
|| best
== -1) {
4606 return _pt1
.pw_unscaled(point(i
, j
, best_depth
));
4610 * Analyze space in a manner dependent on the score map.
4613 static void analyze_space_from_map(const char *d_out
[], const char *v_out
[],
4614 std::map
<const char *, pt
> *d3_depth_pt
,
4615 std::map
<const char *, pt
> *d3_output_pt
,
4616 unsigned int f1
, unsigned int f2
,
4617 unsigned int i
, unsigned int j
, score_map _sm
, int use_filler
) {
4619 int accumulated_ambiguity
= 0;
4620 int max_acc_amb
= pairwise_ambiguity
;
4622 pt _pt1
= al
->get(f1
)->get_t(0);
4623 pt _pt2
= al
->get(f2
)->get_t(0);
4625 if (_pt1
.scale_2d() != 1)
4628 for(score_map::iterator smi
= _sm
.begin(); smi
!= _sm
.end(); smi
++) {
4630 point iw
= smi
->second
.iw
;
4632 if (accumulated_ambiguity
++ >= max_acc_amb
)
4637 ale_pos depth1
= _pt1
.wc(iw
)[2];
4638 ale_pos depth2
= _pt2
.wc(iw
)[2];
4640 ale_pos target_dim
= calc_target_dim(iw
, _pt1
, d_out
, v_out
, d3_depth_pt
, d3_output_pt
);
4642 assert(target_dim
> 0);
4644 int lod1
= calc_lod(depth1
, _pt1
, target_dim
);
4645 int lod2
= calc_lod(depth2
, _pt2
, target_dim
);
4647 while (lod1
< input_decimation_lower
4648 || lod2
< input_decimation_lower
) {
4650 lod1
= calc_lod(depth1
, _pt1
, target_dim
);
4651 lod2
= calc_lod(depth2
, _pt2
, target_dim
);
4655 if (lod1
>= (int) al
->get(f1
)->count()
4656 || lod2
>= (int) al
->get(f2
)->count())
4659 int multiplier
= (unsigned int) floor(pow(2, primary_decimation_upper
- lod1
));
4661 ale_pos depth_range
= calc_depth_range(iw
, _pt1
, _pt2
);
4663 pt _pt1_lod
= al
->get(f1
)->get_t(lod1
);
4664 pt _pt2_lod
= al
->get(f2
)->get_t(lod2
);
4666 int im
= i
* multiplier
;
4667 int jm
= j
* multiplier
;
4669 for (int ii
= 0; ii
< multiplier
; ii
+= 1)
4670 for (int jj
= 0; jj
< multiplier
; jj
+= 1) {
4672 point refined_point
= get_refined_point(_pt1_lod
, _pt2_lod
, im
+ ii
, jm
+ jj
,
4673 f1
, f2
, lod1
, lod2
, depth1
, depth_range
);
4676 * Re-evaluate target dimension.
4679 ale_pos target_dim_
=
4680 calc_target_dim(refined_point
, _pt1
, d_out
, v_out
, d3_depth_pt
, d3_output_pt
);
4682 ale_pos depth1_
= _pt1
.wc(refined_point
)[2];
4683 ale_pos depth2_
= _pt2
.wc(refined_point
)[2];
4685 int lod1_
= calc_lod(depth1_
, _pt1
, target_dim_
);
4686 int lod2_
= calc_lod(depth2_
, _pt2
, target_dim_
);
4688 while (lod1_
< input_decimation_lower
4689 || lod2_
< input_decimation_lower
) {
4691 lod1_
= calc_lod(depth1_
, _pt1
, target_dim_
);
4692 lod2_
= calc_lod(depth2_
, _pt2
, target_dim_
);
4696 * Attempt to refine space around the intersection point.
4699 space::traverse st
=
4700 refine_space(refined_point
, target_dim_
, use_filler
|| _pt1
.scale_2d() != 1);
4702 assert(resolution_ok(al
->get(f1
)->get_t(0), al
->get(f1
)->get_t(0).trilinear_coordinate(st
)));
4703 assert(resolution_ok(al
->get(f2
)->get_t(0), al
->get(f2
)->get_t(0).trilinear_coordinate(st
)));
4711 * Initialize space and identify regions of interest for the adaptive
4714 static void make_space(const char *d_out
[], const char *v_out
[],
4715 std::map
<const char *, pt
> *d3_depth_pt
,
4716 std::map
<const char *, pt
> *d3_output_pt
) {
4718 ui::get()->d3_total_spaces(0);
4721 * Variable indicating whether low-resolution filler space
4722 * is desired to avoid aliased gaps in surfaces.
4725 int use_filler
= d3_depth_pt
->size() != 0
4726 || d3_output_pt
->size() != 0
4727 || output_decimation_preferred
> 0
4728 || input_decimation_lower
> 0
4729 || !focus::is_trivial()
4730 || !strcmp(pairwise_comparisons
, "all");
4732 std::vector
<pt
> pt_outputs
= make_pt_list(d_out
, v_out
, d3_depth_pt
, d3_output_pt
);
4735 * Initialize root space.
4741 * Special handling for experimental option 'subspace_traverse'.
4744 if (subspace_traverse
) {
4746 * Subdivide space to resolve intensity matches between pairs
4750 for (unsigned int f1
= 0; f1
< d2::image_rw::count(); f1
++) {
4752 if (d3_depth_pt
->size() == 0
4753 && d3_output_pt
->size() == 0
4754 && d_out
[f1
] == NULL
4755 && v_out
[f1
] == NULL
)
4758 if (tc_multiplier
== 0)
4761 for (unsigned int f2
= 0; f2
< d2::image_rw::count(); f2
++) {
4766 if (tc_multiplier
== 0)
4769 candidates
*c
= new candidates(f1
);
4771 find_candidates(f1
, f2
, c
, point::neginf(), point::posinf(), pt_outputs
);
4775 c
->generate_subspaces();
4777 if (tc_multiplier
== 0)
4781 if (tc_multiplier
== 0)
4789 * Subdivide space to resolve intensity matches between pairs
4793 for (unsigned int f1
= 0; f1
< d2::image_rw::count(); f1
++)
4794 for (unsigned int f2
= 0; f2
< d2::image_rw::count(); f2
++) {
4798 if (!d_out
[f1
] && !v_out
[f1
] && !d3_depth_pt
->size()
4799 && !d3_output_pt
->size() && strcmp(pairwise_comparisons
, "all"))
4802 if (tc_multiplier
== 0) {
4808 * Iterate over all points in the primary frame.
4811 ale_pos pdu_scale
= pow(2, primary_decimation_upper
);
4813 for (unsigned int i
= 0; i
< al
->get(f1
)->get_image(primary_decimation_upper
)->height(); i
++)
4814 for (unsigned int j
= 0; j
< al
->get(f1
)->get_image(primary_decimation_upper
)->width(); j
++) {
4816 if (d2::render::is_excluded_f(d2::point(i
, j
) * pdu_scale
, f1
))
4819 ui::get()->d3_subdivision_status(f1
, f2
, i
, j
);
4824 * Generate a map from scores to 3D points for
4825 * various depths in f1.
4828 score_map _sm
= p2f_score_map(f1
, f2
, i
, j
);
4831 * Analyze space in a manner dependent on the score map.
4834 analyze_space_from_map(d_out
, v_out
, d3_depth_pt
, d3_output_pt
,
4835 f1
, f2
, i
, j
, _sm
, use_filler
);
4840 * This ordering may encourage image f1 to be cached.
4843 if (tc_multiplier
== 0) {
4852 * Update spatial information structures.
4854 * XXX: the name of this function is horribly misleading. There isn't
4855 * even a 'search depth' any longer, since there is no longer any
4856 * bounded DFS occurring.
4858 static void reduce_cost_to_search_depth(d2::exposure
*exp_out
, int inc_bit
) {
4864 ui::get()->set_steps(ou_iterations
);
4866 for (unsigned int i
= 0; i
< ou_iterations
; i
++) {
4867 ui::get()->set_steps_completed(i
);
4868 spatial_info_update();
4875 * Describe a scene to a renderer
4877 static void describe(render
*r
) {