1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
//
// Copyright 2007-2009 Johannes Hofmann <Johannes.Hofmann@gmx.de>
//
// This software may be used and distributed according to the terms
// of the GNU General Public License, incorporated herein by reference.
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <algorithm>
#include <gsl/gsl_multifit.h>
#include <Fl/Fl.H>
#include "OutputImage.H"
#include "Stitch.H"
#define MAX_VALUE 65025
static double pi_d = asin(1.0) * 2.0;
static double deg2rad = pi_d / 180.0;
Stitch::Stitch() {
for (int i = 0; i < MAX_PICS; i++)
gipf[i] = NULL;
merged_image = NULL;
num_pics = 0;
}
Stitch::~Stitch() {
for (int i = 0; i < MAX_PICS; i++)
if (gipf[i])
delete gipf[i];
else
break;
}
int
Stitch::load_image(char *file) {
for (int i = 0; i < MAX_PICS; i++) {
if (gipf[i] == NULL) {
gipf[i] = new GipfelWidget(0, 0, 800, 600);
if (gipf[i]->load_image(file) != 0) {
delete gipf[i];
gipf[i] = NULL;
} else {
num_pics++;
}
break;
}
}
return 0;
}
OutputImage*
Stitch::set_output(OutputImage *img) {
OutputImage *ret = merged_image;
merged_image = img;
return ret;
}
int
Stitch::resample(GipfelWidget::sample_mode_t m,
int w, int h, double view_start, double view_end) {
view_start = view_start * deg2rad;
view_end = view_end * deg2rad;
double step_view = (view_end - view_start) / w;
int r, g, b;
int y_off = h / 2;
int merged_pixel_set;
double radius = (double) w / (view_end -view_start);
if (merged_image)
if (merged_image->init(w, h) != 0)
merged_image = NULL;
for (int y = 0; y < h; y++) {
double a_nick = atan((double)(y_off - y)/radius);
for (int x = 0; x < w; x++) {
double a_view;
a_view = view_start + x * step_view;
merged_pixel_set = 0;
for (int i = 0; i < num_pics; i++) {
if (merged_pixel_set)
continue;
if (gipf[i]->get_pixel(m, a_view, a_nick,
&r, &g, &b) == 0) {
r = std::max(std::min(r, MAX_VALUE), 0);
g = std::max(std::min(g, MAX_VALUE), 0);
b = std::max(std::min(b, MAX_VALUE), 0);
if (!merged_pixel_set && merged_image) {
merged_image->set_pixel(x, r, g, b);
merged_pixel_set++;
}
}
}
}
if (merged_image)
merged_image->next_line();
}
if (merged_image)
merged_image->done();
return 0;
}
|
