aboutsummaryrefslogtreecommitdiff
path: root/pkg/images/immatch/src/xregister/t_xregister.x
blob: f9fc9b22627a539a67e54e944f7ef68e39332cb1 (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
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
include <imhdr.h>
include <fset.h>
include <gset.h>
include <imset.h>
include "xregister.h"

# T_XREGISTER -- Register a list of images using cross-correlation techniques.

procedure t_xregister()

pointer	freglist		# reference regions list
pointer	database		# the shifts database
int	dformat			# use the database format for the shifts file ?
int	interactive		# interactive mode ?
int	verbose			# verbose mode
pointer	interpstr		# interpolant type
int	boundary		# boundary extension type
real	constant		# constant for boundary extension

int	list1, listr, list2, reglist, reflist, reclist, tfd, stat, nregions
int	c1, c2, l1, l2, ncols, nlines
pointer	sp, image1, image2, imtemp, str, coords
pointer	gd, id, imr, im1, im2, sdb, xc, mw
real	shifts[2]
bool	clgetb()
int	imtopen(), imtlen(), imtgetim(), fntopnb(), clgwrd(), btoi()
int	rg_xregions(), fntlenb(), rg_gxtransform(), rg_xstati()
int	rg_xcorr(), rg_xicorr(), fntgfnb(), access(), open()
pointer	gopen(), immap(), dtmap(), mw_openim()
real	clgetr(), rg_xstatr()
errchk	fntopnb(), gopen()

begin
	# Set STDOUT to flush on a newline character
	call fseti (STDOUT, F_FLUSHNL, YES)

	# Allocate temporary working space.
	call smark (sp)

	call salloc (freglist, SZ_LINE, TY_CHAR)
	call salloc (image1, SZ_FNAME, TY_CHAR)
	call salloc (image2, SZ_FNAME, TY_CHAR)
	call salloc (imtemp, SZ_FNAME, TY_CHAR)
	call salloc (database, SZ_FNAME, TY_CHAR)
	call salloc (coords, SZ_FNAME, TY_CHAR)
	call salloc (interpstr, SZ_FNAME, TY_CHAR)
	call salloc (str, SZ_LINE, TY_CHAR)

	# Get task parameters and open lists.
	call clgstr ("input", Memc[str], SZ_LINE)
	list1 = imtopen (Memc[str])
	call clgstr ("reference", Memc[str], SZ_LINE)
	listr = imtopen (Memc[str])
	call clgstr ("regions", Memc[freglist], SZ_LINE)
	call clgstr ("shifts", Memc[database], SZ_FNAME)
	call clgstr ("output", Memc[str], SZ_LINE)
	list2 = imtopen (Memc[str])
	call clgstr ("records", Memc[str], SZ_LINE)
	if (Memc[str] == EOS)
	    reclist = NULL
	else
	    reclist = fntopnb (Memc[str], NO)
	call clgstr ("coords", Memc[coords], SZ_LINE)

	# Open the cross correlation fitting structure.
	call rg_xgpars (xc)

	# Test the reference image list length.
	if (rg_xstati (xc, CFUNC) != XC_FILE) {
	    if (imtlen (listr) <= 0)
	        call error (0, "The reference image list is empty.")
	    if (imtlen (listr) > 1 && imtlen (listr) != imtlen (list1))
	        call error (0,
	            "The number of reference and input images is not the same.")
	    if (Memc[coords] == EOS)
	        reflist = NULL
	    else {
	        reflist = fntopnb (Memc[coords], NO)
		if (imtlen (listr) != fntlenb (reflist))
	            call error (0,
	      "The number of reference point files and images is not the same.")
	   }
	    iferr {
	        reglist = fntopnb (Memc[freglist], NO)
	    } then {
	        reglist = NULL
	    }
	    call rg_xsets (xc, REGIONS, Memc[freglist])

	} else {
	    call imtclose (listr)
	    listr = NULL
	    reflist = NULL
	    reglist = NULL
	    call rg_xsets (xc, REGIONS, "")
	}

	# Close the output image list if it is empty.
	if (imtlen (list2) == 0) {
	    call imtclose (list2)
	    list2 = NULL
	}

	# Check that the output image list is the same size as the input
	# image list.
	if (list2 != NULL) {
	    if (imtlen (list1) != imtlen (list2)) {
	        call imtclose (list1)
	        if (list2 != NULL)
	            call imtclose (list2)
	       call error (0,
	           "The number of input and output images is not the same.")
	    }
	}

	# Check that the record list is the same length as the input
	# image list length.
	if (reclist != NULL) {
	    if (fntlenb (reclist) != imtlen (list1))
	        call error (0,
	            "Input image and record lists are not the same length.")
	}


	# Open the database file.
	dformat = btoi (clgetb ("databasefmt"))
	if (rg_xstati (xc, CFUNC) == XC_FILE) {
	    if (dformat == YES)
	        sdb = dtmap (Memc[database], READ_ONLY)
	    else
		sdb = open (Memc[database], READ_ONLY, TEXT_FILE)
	} else if (clgetb ("append")) {
	    if (dformat == YES)
	        sdb = dtmap (Memc[database], APPEND)
	    else
		sdb = open (Memc[database], NEW_FILE, TEXT_FILE)
	} else if (access (Memc[database], 0, 0) == YES) {
	    call error (0, "The shifts database file already exists")
	} else {
	    if (dformat == YES)
	        sdb = dtmap (Memc[database], NEW_FILE)
	    else
		sdb = open (Memc[database], NEW_FILE, TEXT_FILE)
	}
	call rg_xsets (xc, DATABASE, Memc[database])

	# Get the boundary extension parameters for the image shift.
	call clgstr ("interp_type", Memc[interpstr], SZ_FNAME)
	boundary = clgwrd ("boundary_type", Memc[str], SZ_LINE,
	   "|constant|nearest|reflect|wrap|")
	constant = clgetr ("constant")

	if (rg_xstati (xc, CFUNC) == XC_FILE)
	    interactive = NO
	else
	    interactive = btoi (clgetb ("interactive"))
	if (interactive == YES) {
	    call clgstr ("graphics", Memc[str], SZ_FNAME)
	    iferr (gd = gopen (Memc[str], NEW_FILE, STDGRAPH))
		gd = NULL
	    call clgstr ("display", Memc[str], SZ_FNAME)
	    iferr (id = gopen (Memc[str], APPEND, STDIMAGE))
		id = NULL
	    verbose = YES
	} else {
	    if (rg_xstati (xc, PFUNC) == XC_MARK)
		call rg_xseti (xc, PFUNC, XC_CENTROID)
	    gd = NULL
	    id = NULL
	    verbose = btoi (clgetb ("verbose"))
	}

	# Initialize the reference image filter descriptors
	imr = NULL
	tfd = NULL

	# Initialize the overlap section.
	c1 = INDEFI
	c2 = INDEFI
	l1 = INDEFI
	l2 = INDEFI
	ncols = INDEFI
	nlines = INDEFI

	# Do each set of input, reference,  and output images.
	while ((imtgetim (list1, Memc[image1], SZ_FNAME) != EOF)) {

	    # Open the reference image, and associated regions and coordinates
	    # files if the correlation function is not file.

	    if (rg_xstati (xc, CFUNC) != XC_FILE) {
	        if (imtgetim (listr, Memc[str], SZ_FNAME) != EOF) {
		    if (imr != NULL)
		        call imunmap (imr)
		    imr = immap (Memc[str], READ_ONLY, 0)
		    if (IM_NDIM(imr) > 2)
		        call error (0, "Reference images must be 1D or 2D")
		    call rg_xsets (xc, REFIMAGE, Memc[str])
		    nregions = rg_xregions (reglist, imr, xc, 1)
		    if (nregions <= 0 && interactive == NO)
		        call error (0, "The regions list is empty.")
		    if (reflist != NULL) {
		        if (tfd != NULL)
			    call close (tfd)
		        tfd = rg_gxtransform (reflist, xc, Memc[str])
		        call rg_xsets (xc, REFFILE, Memc[str])
		    }
	        }
	    } else
		call rg_xsets (xc, REFIMAGE, "reference")

	    # Open the input image.
	    im1 = immap (Memc[image1], READ_ONLY, 0)
	    if (IM_NDIM(im1) > 2) {
	         call error (0, "Input images must be 1D or 2D")
	    } else if (imr != NULL) {
	        if (IM_NDIM(im1) != IM_NDIM(imr))
		    call error (0,
	      "Input images must have same dimensionality as reference images")
	    }
	    call imseti (im1, IM_TYBNDRY, BT_NEAREST)
	    if (IM_NDIM(im1) == 1)
	        call imseti (im1, IM_NBNDRYPIX, IM_LEN(im1,1))
	    else
	        call imseti (im1, IM_NBNDRYPIX,
		    max (IM_LEN(im1,1), IM_LEN(im1,2)))
	    call rg_xsets (xc, IMAGE, Memc[image1])

	    # Open the output image if any.
	    if (list2 == NULL) {
	        im2 = NULL
		Memc[image2] = EOS
	    } else if (imtgetim (list2, Memc[image2], SZ_FNAME) != EOF) {
		call xt_mkimtemp (Memc[image1], Memc[image2], Memc[imtemp],
		    SZ_FNAME)
	        im2 = immap (Memc[image2], NEW_COPY, im1)
	    } else {
	        im2 = NULL
		Memc[image2] = EOS
	    }
	    call rg_xsets (xc, OUTIMAGE, Memc[image2])

	    # Get the image record name for the shifts database.
	    if (reclist == NULL)
		call strcpy (Memc[image1], Memc[str], SZ_FNAME)
	    else if (fntgfnb (reclist, Memc[str], SZ_FNAME) == EOF)
		call strcpy (Memc[image1], Memc[str], SZ_FNAME)
	    call rg_xsets (xc, RECORD, Memc[str])

	    # Compute the initial coordinate shift.
	    if (tfd != NULL)
		call rg_xtransform (tfd, xc)

	    # Perform the cross correlation function.
	    if (interactive == YES) {
		stat = rg_xicorr (imr, im1, im2, sdb, dformat, reglist, tfd,
		    xc, gd, id)
	    } else {
	        stat = rg_xcorr (imr, im1, sdb, dformat, xc)
		if (verbose == YES) {
		    call rg_xstats (xc, REFIMAGE, Memc[str], SZ_LINE)
	            call printf (
		        "Average shift from %s to %s is %g %g pixels\n")
	                call pargstr (Memc[image1])
	                call pargstr (Memc[str])
	                call pargr (rg_xstatr (xc, TXSHIFT))
	                call pargr (rg_xstatr (xc, TYSHIFT))
		}
	    }

	    # Compute the overlap region for the images.
	    call rg_overlap (im1, rg_xstatr (xc, TXSHIFT),
	        rg_xstatr (xc,TYSHIFT), c1, c2, l1, l2, ncols, nlines)

	    # Shift the image and update the wcs.
	    if (im2 != NULL && stat == NO) {
		if (verbose == YES) {
		    call printf (
		        "\tShifting image %s to image %s ...\n")
		        call pargstr (Memc[image1])
			call pargstr (Memc[imtemp])
		}

		call rg_xshiftim (im1, im2, rg_xstatr (xc, TXSHIFT),
		    rg_xstatr (xc, TYSHIFT), Memc[interpstr], boundary,
		        constant)
		mw = mw_openim (im1)
		shifts[1] = rg_xstatr (xc, TXSHIFT)
		shifts[2] = rg_xstatr (xc, TYSHIFT)
		call mw_shift (mw, shifts, 03B)
		call mw_saveim (mw, im2)
		call mw_close (mw)
	    }

	    # Close up the input and output images.
	    call imunmap (im1)
	    if (im2 != NULL) {
		call imunmap (im2)
	        if (stat == YES)
		    call imdelete (Memc[image2])
		else
		    call xt_delimtemp (Memc[image2], Memc[imtemp])
	    }

	    if (stat == YES)
		break
	    call rg_xindefr (xc)
	}

	if (verbose == YES)
	    call rg_poverlap (c1, c2, l1, l2, ncols, nlines)

	call rg_xfree (xc)

	# Close up the lists.
	if (imr != NULL)
	    call imunmap (imr)
	call imtclose (list1)
	if (listr != NULL)
	    call imtclose (listr)
	if (reglist != NULL)
	    call fntclsb (reglist)
	if (list2 != NULL)
	    call imtclose (list2)
	if (tfd != NULL)
	    call close (tfd)
	if (reflist != NULL)
	    call fntclsb (reflist)
	if (reclist != NULL)
	    call fntclsb (reclist)
	if (dformat == YES)
	    call dtunmap (sdb)
	else
	    call close (sdb)

	# Close up the graphics and display devices.
	if (gd != NULL)
	    call gclose (gd)
	if (id != NULL)
	    call gclose (id)

	call sfree (sp)
end


# RG_OVERLAP -- Compute the overlap region of the list of images.

procedure rg_overlap (im1, xshift, yshift, x1, x2, y1, y2, ncols, nlines)

pointer	im1			# pointer to the input image
real	xshift			# the computed x shift of the input image
real	yshift			# the computed y shift of the input image
int	x1, x2			# the input/output column limits
int	y1, y2			# the input/output line limits
int	ncols, nlines		# the input/output size limits

int	ixlo, ixhi, iylo, iyhi
real	xlo, xhi, ylo, yhi

begin
	if (IS_INDEFR(xshift) || IS_INDEFR(yshift))
	    return

	# Compute the limits of the shifted image.
	xlo = 1.0 + xshift
	xhi = IM_LEN(im1,1) + xshift
	ylo = 1.0 + yshift
	yhi = IM_LEN(im1,2) + yshift

	# Round up or down as appropriate.
	ixlo = int (xlo)
	if (xlo > ixlo)
	    ixlo = ixlo + 1
	ixhi = int (xhi)
	if (xhi < ixhi)
	    ixhi = ixhi - 1
	iylo = int (ylo)
	if (ylo > iylo)
	    iylo = iylo + 1
	iyhi = int (yhi)
	if (yhi < iyhi)
	    iyhi = iyhi - 1

	# Determine the new limits.
	if (IS_INDEFI(x1))
	    x1 = ixlo
	else
	    x1 = max (ixlo, x1)
	if (IS_INDEFI(x2))
	    x2 = ixhi
	else
	    x2 = min (ixhi, x2)
	if (IS_INDEFI(y1))
	    y1 = iylo
	else
	    y1 = max (iylo, y1)
	if (IS_INDEFI(y2))
	    y2 = iyhi
	else
	    y2 = min (iyhi, y2)
	if (IS_INDEFI(ncols))
	    ncols = IM_LEN(im1,1)
	else
	    ncols = min (ncols, IM_LEN(im1,1))
	if (IS_INDEFI(nlines))
	    nlines = IM_LEN(im1,2)
	else
	    nlines = min (nlines, IM_LEN(im1,2))
end


# RG_POVERLAP -- Procedure to print the overlap and/or vignetted region.

procedure rg_poverlap (x1, x2, y1, y2, ncols, nlines)

int	x1, x2			# the input column limits
int	y1, y2			# the input line limits
int	ncols, nlines		# the number of lines and columns

int	vx1, vx2, vy1, vy2

begin
	vx1 = max (1, min (x1, ncols))
	vx2 = max (1, min (x2, ncols))
	vy1 = max (1, min (y1, nlines))
	vy2 = max (1, min (y2, nlines))

	call printf ("Overlap region: [%d:%d,%d:%d]\n")
	    call pargi (x1)
	    call pargi (x2)
	    call pargi (y1)
	    call pargi (y2)
	if (vx1 != x1 || vx2 != x2 || vy1 != y1 || vy2 != y2) {
	    call printf ("Vignetted overlap region: [%d:%d,%d:%d]\n")
	        call pargi (vx1)
	        call pargi (vx2)
	        call pargi (vy1)
	        call pargi (vy2)
	}
end