aboutsummaryrefslogtreecommitdiff
path: root/noao/twodspec/multispec/t_fitgauss5.x
blob: 146d37b6e6e21b84906163bccfb0c5b314bbea53 (plain) (blame)
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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
include "ms.h"

# T_FIT_GAUSS5 -- Fit the GAUSS5 model.
#
# This task selects the database, the sample lines to be modeled, the
# model fitting algorithm, whether to track models from one sample line
# to the next or model them independently.

procedure t_fit_gauss5 ()

char	image[SZ_FNAME]			# Image
int	lines[3, MAX_RANGES]		# Sample lines to be modeled
bool	track				# Track model solution
int	start				# Starting line for modeling
int	naverage			# Number of image lines to average
real	lower				# Starting point of profile
real	upper				# Ending point of profile

int	i, nsamples, sample_start, sample, line, improved
int	len_line, len_profile, nspectra, nparams
pointer	ms, im
pointer	sp, data, model, profiles, ranges, samples

int	get_sample_line(), get_sample_lines()
int	g5_fit1(), g5_fit2()
int	clgeti(), clgranges(), btoi()
bool	clgetb()
real	clgetr()
pointer	msmap(), immap()

include	"fitgauss5.com"

begin
	# Access the database and the image.
	call clgstr ("image", image, SZ_FNAME)
	ms = msmap (image, READ_WRITE, 0)
	im = immap (image, READ_ONLY, 0)

	# Get the task parameters.
	i = clgranges ("lines", 1, MS_LEN(ms, 2), lines, MAX_RANGES)
	i = clgranges ("spectra", 1, MS_NSPECTRA(ms), spectra, MAX_RANGES)
	track = clgetb ("track")
	start = clgeti ("start")
	naverage = clgeti ("naverage")
	lower = clgetr ("lower")
	upper = clgetr ("upper")
	factor = clgetr ("factor")

	# Algorithm 1 fits the parameters selected in the parameters array
	# simultaneously.  Algorithm 2 does not require the user to specify
	# the parameters.

	algorithm = clgeti ("algorithm")
	if (algorithm == 1) {
	    parameters[I0_INDEX] = btoi (clgetb ("fit_i0"))
	    parameters[X0_INDEX] = btoi (clgetb ("fit_x0"))
	    parameters[S0_INDEX] = btoi (clgetb ("fit_s0"))
	    parameters[S1_INDEX] = btoi (clgetb ("fit_s1"))
	    parameters[S2_INDEX] = btoi (clgetb ("fit_s2"))
	}

	# Select whether to smooth the shape parameters after fitting.
	# If smoothing is desired get the spline smoothing parameters.

	smooth[S0_INDEX] = btoi (clgetb ("smooth_s0"))
	smooth[S1_INDEX] = btoi (clgetb ("smooth_s1"))
	smooth[S2_INDEX] = btoi (clgetb ("smooth_s2"))
	if ((smooth[S0_INDEX] == YES) || (smooth[S1_INDEX] == YES) ||
	    (smooth[S2_INDEX] == YES)) {
	    call ms_set_smooth (1., real(MS_LEN(ms, 1)), MS_NSPECTRA(ms))
	}

	call g5_set_verbose (clgetb ("verbose"))
	call g5_prnt1 (image, naverage, track, start)
	    
	# Set the various array dimensions and allocate memory.
	len_line = MS_LEN(ms, 1)
	len_profile = nint (upper - lower + 2)
	nspectra = MS_NSPECTRA(ms)
	nparams = MS_NGAUSS5
	call smark (sp)
	call salloc (samples, MS_NSAMPLES(ms), TY_INT)
	call salloc (data, len_line, TY_REAL)
	call salloc (model, len_line, TY_REAL)
	call salloc (profiles, len_profile * nspectra * nparams, TY_REAL)
	call salloc (ranges, nspectra * LEN_RANGES, TY_REAL)

	# Convert from image lines to sample lines.
	nsamples = get_sample_lines (ms, lines, Memi[samples])
	sample_start = get_sample_line (ms, start)

	# Initialize forward tracking.  If tracking get the initial parameters,
	# model profiles and model line from the starting line.

	if (track) {
	    call msggauss5 (ms, sample_start)
	    call mod_gauss5 (ms, lower, Memr[profiles], Memr[ranges],
		len_profile, nspectra)
	    call set_model (ms, Memr[model], Memr[profiles], Memr[ranges],
		len_line, len_profile, nspectra)
	}

	# Track forward from the starting line to the specified sample lines.

	do i = 1, nsamples {
	    sample = Memi[samples + i - 1]
	    if (sample < sample_start)
		next
	    line = LINE(ms, sample)

	    # Get the image data line.
	    call msgimage (im, line, naverage, Memr[data])

	    # If not tracking get the initial parameters, model profiles, and
	    # model line for the current line.  Otherwise record the starting
	    # parameters.

	    if (!track) {
	        call msggauss5 (ms, sample)
	        call mod_gauss5 (ms, lower, Memr[profiles], Memr[ranges],
		    len_profile, nspectra)
	        call set_model (ms, Memr[model], Memr[profiles], Memr[ranges],
		    len_line, len_profile, nspectra)
	    } else
		call mspgauss5 (ms, sample)

	    call g5_prnt2 (line, Memr[data], len_line)

	    # Do the model fitting using the selected algorithm.
	    switch (algorithm) {
	    case 1:
	        improved = g5_fit1 (ms, Memr[data], Memr[model], Memr[profiles],
		    Memr[ranges], lower, len_profile)
	    case 2:
	        improved = g5_fit2 (ms, Memr[data], Memr[model], Memr[profiles],
		    Memr[ranges], lower, len_profile)
	    }

	    # If the new model parameters have improved the fit record them in
	    # the database.
	    if (improved == YES)
		call mspgauss5 (ms, sample)
	}

	# Initialize backward tracking.  If tracking get the initial parameters,
	# model profiles and model line from the starting line.

	if (track) {
	    call msggauss5 (ms, sample_start)
	    call mod_gauss5 (ms, lower, Memr[profiles], Memr[ranges],
		len_profile, nspectra)
	    call set_model (ms, Memr[model], Memr[profiles], Memr[ranges],
		len_line, len_profile, nspectra)
	}

	# Track backward from the starting line to the specified sample lines.

	do i = nsamples, 1, -1 {
	    sample = Memi[samples + i - 1]
	    if (sample >= sample_start)
		next
	    line = LINE(ms, sample)

	    # Get the image data line.
	    call msgimage (im, line, naverage, Memr[data])

	    # If not tracking get the initial parameters, model profiles, and
	    # model line for the current line.  Else record the starting
	    # parameters.

	    if (!track) {
	        call msggauss5 (ms, sample)
	        call mod_gauss5 (ms, lower, Memr[profiles], Memr[ranges],
		    len_profile, nspectra)
	        call set_model (ms, Memr[model], Memr[profiles], Memr[ranges],
		    len_line, len_profile, nspectra)
	    } else
		call mspgauss5 (ms, sample)

	    call g5_prnt2 (line, Memr[data], len_line)


	    # Do the model fitting using the selected algorithm.
	    switch (algorithm) {
	    case 1:
	        improved = g5_fit1 (ms, Memr[data], Memr[model], Memr[profiles],
		    Memr[ranges], lower, len_profile)
	    case 2:
	        improved = g5_fit2 (ms, Memr[data], Memr[model], Memr[profiles],
		    Memr[ranges], lower, len_profile)
	    }

	    # If the new model parameters have improved the fit record them in
	    # the database.

	    if (improved == YES)
		call mspgauss5 (ms, sample)
	}

	# Finish up.
	if ((smooth[S0_INDEX] == YES) || (smooth[S1_INDEX] == YES) ||
	    (smooth[S2_INDEX] == YES)) {
	    call ms_free_smooth ()
	}
	call imunmap (im)
	call history (ms, "Fit model")
	call msunmap (ms)
	call sfree (sp)
end