Initial commit

1 files changed, 128 insertions, 0 deletions

diff --git a/noao/onedspec/ecidentify/ecdofit.x b/noao/onedspec/ecidentify/ecdofit.x
new file mode 100644
index 00000000..14dcea54
--- /dev/null
+++ b/noao/onedspec/ecidentify/ecdofit.x
@@ -0,0 +1,128 @@
+include	<smw.h>
+include	"ecidentify.h"
+
+# EC_DOFIT -- Fit an echelle function to the features.  Eliminate INDEF points.
+
+procedure ec_dofit (ec, interactive, fixedorder)
+
+pointer	ec			# EC pointer
+int	interactive		# Interactive fit?
+int	fixedorder		# Fixed order?
+
+int	i, j, k, nfit
+double	xmin, xmax, ymin, ymax
+pointer	gt1, ecf
+pointer	sp, x, y, z, w, gt_init()
+errchk	ecf_fit
+
+begin
+	# Count number of points and determine the order range.
+	j = ORDER(ec,1)
+	k = ORDER(ec,1)
+	nfit = 0
+	for (i=1; i<=EC_NFEATURES(ec); i=i+1) {
+	    if (IS_INDEFD (PIX(ec,i)) || IS_INDEFD (USER(ec,i)))
+		next
+	    j = min (j, ORDER(ec,i))
+	    k = max (k, ORDER(ec,i))
+	    nfit = nfit + 1
+	}
+
+	# Require at least 4 points and more than one order.
+	if (nfit < 4 || j == k) {
+	    if (EC_ECF(ec) != NULL) {
+	        call dgs_free (EC_ECF(ec))
+		call ecf_setd ("shift", 0.D0)
+	        EC_NEWGRAPH(ec) = YES
+	        EC_NEWECF(ec) = YES
+	    }
+	    return
+	}
+
+	# Allocate arrays for points to be fit and fill them in.
+	call smark (sp)
+	call salloc (x, nfit, TY_DOUBLE)
+	call salloc (y, nfit, TY_DOUBLE)
+	call salloc (z, nfit, TY_DOUBLE)
+	call salloc (w, nfit, TY_DOUBLE)
+
+	nfit = 0
+	do i = 1, EC_NFEATURES(ec) {
+	    if (IS_INDEFD (PIX(ec,i)) || IS_INDEFD (USER(ec,i)))
+		next
+	    Memd[x+nfit] = PIX(ec,i)
+	    Memd[y+nfit] = APN(ec,i)
+	    Memd[z+nfit] = USER(ec,i)
+	    Memd[w+nfit] = 1.
+	    nfit = nfit + 1
+	}
+
+	# Initialize fit limits.
+	ymin = APS(ec,1)
+	ymax = ymin
+	do i = 2, EC_NLINES(ec) {
+	    xmin = APS(ec,i)
+	    if (xmin < ymin)
+		ymin = xmin
+	    if (xmin > ymax)
+		ymax = xmin
+	}
+	xmin = 1
+	xmax = EC_NPTS(ec)
+
+	call ecf_setd ("xmin", xmin)
+	call ecf_setd ("xmax", xmax)
+	call ecf_setd ("ymin", ymin)
+	call ecf_setd ("ymax", ymax)
+
+	# Fit the echelle dispersion function.
+	ecf = EC_ECF(ec)
+	if (interactive == YES) {
+	    gt1 = gt_init()
+	    call ecf_fit (ecf, EC_GP(ec), gt1, Memd[x], Memd[y],
+		Memd[z], Memd[w], nfit, fixedorder)
+	    call gt_free (gt1)
+	} else
+	    call ecf_fit (ecf, NULL, NULL, Memd[x], Memd[y], Memd[z],
+		Memd[w], nfit, fixedorder)
+	EC_ECF(ec) = ecf
+
+	# Remove any deleted points.
+	j = 0
+	k = 0
+	do i = 1, EC_NFEATURES(ec) {
+	    if (IS_INDEFD (PIX(ec,i)) || IS_INDEFD (USER(ec,i))) {
+		j = j + 1
+    	        APN(ec,j) = APN(ec,i)
+    	        LINE(ec,j) = LINE(ec,i)
+    	        ORDER(ec,j) = ORDER(ec,i)
+    	        PIX(ec,j) = PIX(ec,i)
+    	        FIT(ec,j) = FIT(ec,i)
+    	        USER(ec,j) = USER(ec,i)
+    	        FWIDTH(ec,j) = FWIDTH(ec,i)
+    	        FTYPE(ec,j) = abs (FTYPE(ec,i))
+	    } else {
+		if (Memd[w+k] != 0.) {
+		    j = j + 1
+    	            APN(ec,j) = APN(ec,i)
+    	            LINE(ec,j) = LINE(ec,i)
+    	            ORDER(ec,j) = ORDER(ec,i)
+    	            PIX(ec,j) = PIX(ec,i)
+    	            FIT(ec,j) = FIT(ec,i)
+    	            USER(ec,j) = USER(ec,i)
+    	            FWIDTH(ec,j) = FWIDTH(ec,i)
+    	            FTYPE(ec,j) = abs (FTYPE(ec,i))
+		    if (Memd[w+k] < 0.)
+    	                FTYPE(ec,j) = -FTYPE(ec,j)
+		}
+		k = k + 1
+	    }
+	}
+	EC_NFEATURES(ec) = j
+
+	# Set flags.
+	EC_NEWECF(ec) = YES
+	EC_NEWGRAPH(ec) = YES
+
+	call sfree (sp)
+end