def b(drizimage, origimage, x=None, y=None, List=None):
# define the output
all_out = []
# Get the parameters from the header
GeoPar = DrizGeoPars(drizimage, origimage)
iraf.unlearn('wtranback')
# Use wtranback, along with all the parameters specified above, to
# transform to the output image
iraf.wtranback.nxin = GeoPar.nxin
iraf.wtranback.nyin = GeoPar.nyin
iraf.wtranback.nxout = GeoPar.nxout
iraf.wtranback.nyout = GeoPar.nyout
iraf.wtranback.scale = GeoPar.scale
iraf.wtranback.xsh = GeoPar.xsh
iraf.wtranback.ysh = GeoPar.ysh
iraf.wtranback.rot = GeoPar.rot
iraf.wtranback.coeffs = GeoPar.coeffs
iraf.wtranback.shft_un = GeoPar.shft_un
iraf.wtranback.shft_fr = GeoPar.shft_fr
iraf.wtranback.align = GeoPar.align
iraf.wtranback.lam = GeoPar.lam
iraf.wtranback.xgeoim = GeoPar.xgeoim
iraf.wtranback.ygeoim = GeoPar.ygeoim
iraf.wtranback.geomode = 'user'
if List != None:
iraf.wtranback.xylist = List
x = 1.0
y = 1.0
str = iraf.wtranback(x, y, mode='h', Stdout=1)
all_out.append(" Xin Yin Xout Yout")
# Just show the lines of interest
for line in str:
if line[0:1] == "!":
print(line)
sys.exit()
if line[0:3] == ' Xi':
xin = float(line.split()[1])
yin = float(line.split()[2])
xout = float(line.split()[4])
yout = float(line.split()[5])
all_out.append("%10.3f %10.3f %10.3f %10.3f" %
(xin, yin, xout, yout))
else:
iraf.wtranback.xylist = ''
# Transform and display the result
str = iraf.wtranback(x, y, mode='h', Stdout=1)
# Just show the lines of interest
for line in str:
if line[0:1] == '!':
all_out.appen(line)
sys.exit()
if line[0:3] == ' Xi':
all_out.append(line)
# return the output
return all_out
def drztranback(drzfile, x=0, y=0, fltlist=None, xylistfile='', verbose=True):
"""
convert a set of coordinates from a drz image into
the coordinate space of each of the contributing _flt files
returns a list of tuples with (fltfile,x,y)
"""
import os
import pyfits
from pyraf import iraf
from iraf import stsdas
from numpy import loadtxt, iterable
stsdas.analysis()
stsdas.dither()
# get output (i.e. drizzled) image size
nxout = pyfits.getval(drzfile, 'NAXIS1')
nyout = pyfits.getval(drzfile, 'NAXIS2')
scaleout = 3600 * (abs(pyfits.getval(drzfile, 'CD1_1')) +
abs(pyfits.getval(drzfile, 'CD2_2')))
xscaleout = 7200 * abs(pyfits.getval(drzfile, 'CD1_1'))
yscaleout = 7200 * abs(pyfits.getval(drzfile, 'CD2_2'))
# if needed, get a list of contributing flt files and/or coeff files
if not fltlist:
fltlist = getfltlist(drzfile)
# build the list of coordinates, starting with the original drzfile x,y coords
returnlist = []
if xylistfile:
xlist, ylist = loadtxt(xylistfile, unpack=True)
elif iterable(x):
xlist, ylist = x, y
else:
xlist, ylist = [float(x)], [float(y)]
for xx, yy in zip(x, y):
returnlist.append((drzfile, 1, xx, yy))
# write out a list of drz-frame x,y positions into a text file.
# this is used as input to wtranback for all flt files
xylistfile = util.naming.chsuffix(os.path.basename(drzfile),
'_fake.xylist')
fout = open(xylistfile, 'w')
for xx, yy in zip(xlist, ylist):
print >> fout, "%15.5f %15.5f" % (xx, yy)
fout.close()
# translate the x,y coords back to _flt coordinates
# TODO : probably need to allow for up to 2 coeff files
# for every flt file, for UVIS and ACS
# for fltfile,coefffile in zip(fltlist,coefflist) :
for fltfile in fltlist:
if verbose: print("translating %s to %s coords" % (drzfile, fltfile))
# find all the sci extensions
flthdulist = pyfits.open(fltfile)
extlist = [
i for i in range(len(flthdulist))
if flthdulist[i].name.lower().startswith('sci')
]
for ext in extlist:
nxin = pyfits.getval(fltfile, 'NAXIS1', ext=ext)
nyin = pyfits.getval(fltfile, 'NAXIS2', ext=ext)
scalein = 3600 * (abs(pyfits.getval(fltfile, 'CD1_1', ext=ext)) +
abs(pyfits.getval(fltfile, 'CD2_2', ext=ext)))
xscalein = 7200 * abs(pyfits.getval(fltfile, 'CD1_1', ext=ext))
yscalein = 7200 * abs(pyfits.getval(fltfile, 'CD2_2', ext=ext))
# slimmed down wtranback call from LS: 2011.04.28
#
# NOTE: we use the xylistfile as input even when
# there is only one pair of coordinates
iraf.wtranback.unlearn()
iraf.flpr()
iraf.flpr()
iraf.gflush()
iraf.gflush()
#coeffile = os.path.join(snworkdir,fltfile[:-5] + '_coeffs1.dat')
coeffile = fltfile[:-5] + '_coeffs1.dat'
output = iraf.wtranback(0,
0,
nxin=nxin,
nxout=nxout,
nyin=nyin,
nyout=nyout,
xylist=xylistfile,
coeffs=coeffile,
geomode='wcs',
refim=drzfile,
inimage=fltfile + '[%i]' % ext,
Stdout=1)
for line in output:
if line.startswith(' Xin,Yin:'):
xin, yin = map(float, line.split()[1:3])
if xin > 0 and xin < nxin and yin > 0 and yin < nyin:
returnlist.append(
(fltfile, ext, float(xin), float(yin)))
return (returnlist)
def drztranback( drzfile, x=0, y=0, xylistfile='', ext='SCI',verbose=False ):
"""
convert a set of coordinates from a drz image into
the coordinate space of each of the contributing _flt files
returns a list of tuples with (fltfile,x,y)
"""
import os
import pyfits
from pyraf import iraf
from iraf import stsdas
from numpy import loadtxt
stsdas.analysis()
stsdas.dither()
# get output (i.e. drizzled) image size
nxout = pyfits.getval( drzfile, 'NAXIS1', ext=ext)
nyout = pyfits.getval( drzfile, 'NAXIS2', ext=ext)
scaleout = 3600*( abs(pyfits.getval(drzfile,'CD1_1',ext=ext)) +
abs(pyfits.getval(drzfile,'CD2_2',ext=ext)) )
xscaleout = 7200*abs(pyfits.getval(drzfile,'CD1_1',ext=ext))
yscaleout = 7200*abs(pyfits.getval(drzfile,'CD2_2',ext=ext))
# get a list of contributing flt files
fltfilelist = getfltlist( drzfile )
scifile = drzfile[:-8]+"sci.fits"
# translate the x,y coords back to _flt coordinates
returnlist = []
if xylistfile :
xlist,ylist = loadtxt( xylistfile, unpack=True )
for x,y in zip(xlist,ylist) :
returnlist.append( (drzfile, x, y))
else :
returnlist.append( (drzfile, x, y) )
for fltfile in fltfilelist :
if verbose: print("translating %s to %s coords"%(drzfile,fltfile))
# import pdb; pdb.set_trace()
nxin = pyfits.getval( fltfile, 'NAXIS1', ext='SCI')
nyin = pyfits.getval( fltfile, 'NAXIS2', ext='SCI')
scalein = 3600*( abs(pyfits.getval(fltfile,'CD1_1',ext='SCI')) +
abs(pyfits.getval(fltfile,'CD2_2',ext='SCI')) )
xscalein = 7200*abs(pyfits.getval(fltfile,'CD1_1',ext='SCI'))
yscalein = 7200*abs(pyfits.getval(fltfile,'CD2_2',ext='SCI'))
coeffile = fltfile[:-5] + '_coeffs1.dat'
# wtranback is better than tranback
iraf.flpr()
iraf.flpr()
iraf.unlearn( iraf.wtranback )
if ext=='SCI' : extstr = '[1]'
else : extstr = ''
output = iraf.wtranback( x, y, nxin=nxin, nxout=nxout,
nyin=nyin, nyout=nyout, xylist=xylistfile,
coeffs=coeffile,
geomode='wcs',
refim=drzfile+extstr,
inimage=fltfile+'[sci,1]',
raref =pyfits.getval(drzfile,'CRVAL1',ext=ext),
decref=pyfits.getval(drzfile,'CRVAL2',ext=ext),
xrefpix=pyfits.getval(drzfile,'CRPIX1',ext=ext),
yrefpix=pyfits.getval(drzfile,'CRPIX2',ext=ext),
orient=pyfits.getval(drzfile,'ORIENTAT',ext=ext),
Stdout=1)
for line in output:
if line.startswith(' Xin,Yin:') :
xin,yin = line.split()[1:3]
returnlist.append( (fltfile,float(xin),float(yin)) )
return( returnlist )
def drztranback(drzfile, x=0, y=0, xylistfile="", ext="SCI", verbose=False):
"""
convert a set of coordinates from a drz image into
the coordinate space of each of the contributing _flt files
returns a list of tuples with (fltfile,x,y)
"""
import os
import pyfits
from pyraf import iraf
from iraf import stsdas
from numpy import loadtxt
stsdas.analysis()
stsdas.dither()
# get output (i.e. drizzled) image size
nxout = pyfits.getval(drzfile, "NAXIS1", ext=ext)
nyout = pyfits.getval(drzfile, "NAXIS2", ext=ext)
scaleout = 3600 * (abs(pyfits.getval(drzfile, "CD1_1", ext=ext)) + abs(pyfits.getval(drzfile, "CD2_2", ext=ext)))
xscaleout = 7200 * abs(pyfits.getval(drzfile, "CD1_1", ext=ext))
yscaleout = 7200 * abs(pyfits.getval(drzfile, "CD2_2", ext=ext))
# get a list of contributing flt files
fltfilelist = getfltlist(drzfile)
scifile = drzfile[:-8] + "sci.fits"
# translate the x,y coords back to _flt coordinates
returnlist = []
if xylistfile:
xlist, ylist = loadtxt(xylistfile, unpack=True)
for x, y in zip(xlist, ylist):
returnlist.append((drzfile, x, y))
else:
returnlist.append((drzfile, x, y))
for fltfile in fltfilelist:
if verbose:
print ("translating %s to %s coords" % (drzfile, fltfile))
# import pdb; pdb.set_trace()
nxin = pyfits.getval(fltfile, "NAXIS1", ext="SCI")
nyin = pyfits.getval(fltfile, "NAXIS2", ext="SCI")
scalein = 3600 * (
abs(pyfits.getval(fltfile, "CD1_1", ext="SCI")) + abs(pyfits.getval(fltfile, "CD2_2", ext="SCI"))
)
xscalein = 7200 * abs(pyfits.getval(fltfile, "CD1_1", ext="SCI"))
yscalein = 7200 * abs(pyfits.getval(fltfile, "CD2_2", ext="SCI"))
coeffile = fltfile[:-5] + "_coeffs1.dat"
# wtranback is better than tranback
iraf.flpr()
iraf.flpr()
iraf.unlearn(iraf.wtranback)
if ext == "SCI":
extstr = "[1]"
else:
extstr = ""
output = iraf.wtranback(
x,
y,
nxin=nxin,
nxout=nxout,
nyin=nyin,
nyout=nyout,
xylist=xylistfile,
coeffs=coeffile,
geomode="wcs",
refim=drzfile + extstr,
inimage=fltfile + "[sci,1]",
raref=pyfits.getval(drzfile, "CRVAL1", ext=ext),
decref=pyfits.getval(drzfile, "CRVAL2", ext=ext),
xrefpix=pyfits.getval(drzfile, "CRPIX1", ext=ext),
yrefpix=pyfits.getval(drzfile, "CRPIX2", ext=ext),
orient=pyfits.getval(drzfile, "ORIENTAT", ext=ext),
Stdout=1,
)
for line in output:
if line.startswith(" Xin,Yin:"):
xin, yin = line.split()[1:3]
returnlist.append((fltfile, float(xin), float(yin)))
return returnlist
def drztranback( drzfile, x=0, y=0, fltlist=None,
xylistfile='', verbose=True ):
"""
convert a set of coordinates from a drz image into
the coordinate space of each of the contributing _flt files
returns a list of tuples with (fltfile,x,y)
"""
import os
import pyfits
from pyraf import iraf
from iraf import stsdas
from numpy import loadtxt, iterable
stsdas.analysis()
stsdas.dither()
# get output (i.e. drizzled) image size
nxout = pyfits.getval( drzfile, 'NAXIS1')
nyout = pyfits.getval( drzfile, 'NAXIS2')
scaleout = 3600*( abs(pyfits.getval(drzfile,'CD1_1')) +
abs(pyfits.getval(drzfile,'CD2_2')) )
xscaleout = 7200*abs(pyfits.getval(drzfile,'CD1_1'))
yscaleout = 7200*abs(pyfits.getval(drzfile,'CD2_2'))
# if needed, get a list of contributing flt files and/or coeff files
if not fltlist :
fltlist = getfltlist( drzfile )
# build the list of coordinates, starting with the original drzfile x,y coords
returnlist = []
if xylistfile :
xlist,ylist = loadtxt( xylistfile, unpack=True )
elif iterable(x) :
xlist, ylist = x, y
else :
xlist,ylist = [float(x)],[float(y)]
for xx,yy in zip(x,y) :
returnlist.append( (drzfile,1,xx,yy) )
# write out a list of drz-frame x,y positions into a text file.
# this is used as input to wtranback for all flt files
xylistfile = util.naming.chsuffix( os.path.basename(drzfile), '_fake.xylist')
fout = open(xylistfile, 'w')
for xx,yy in zip(xlist,ylist) :
print >> fout, "%15.5f %15.5f"%(xx,yy)
fout.close()
# translate the x,y coords back to _flt coordinates
# TODO : probably need to allow for up to 2 coeff files
# for every flt file, for UVIS and ACS
# for fltfile,coefffile in zip(fltlist,coefflist) :
for fltfile in fltlist:
if verbose: print("translating %s to %s coords"%(drzfile,fltfile))
# find all the sci extensions
flthdulist = pyfits.open( fltfile )
extlist = [ i for i in range(len(flthdulist)) if flthdulist[i].name.lower().startswith('sci') ]
for ext in extlist :
nxin = pyfits.getval( fltfile, 'NAXIS1', ext=ext)
nyin = pyfits.getval( fltfile, 'NAXIS2', ext=ext)
scalein = 3600*( abs(pyfits.getval(fltfile,'CD1_1',ext=ext)) +
abs(pyfits.getval(fltfile,'CD2_2',ext=ext)) )
xscalein = 7200*abs(pyfits.getval(fltfile,'CD1_1',ext=ext))
yscalein = 7200*abs(pyfits.getval(fltfile,'CD2_2',ext=ext))
# slimmed down wtranback call from LS: 2011.04.28
#
# NOTE: we use the xylistfile as input even when
# there is only one pair of coordinates
iraf.wtranback.unlearn()
iraf.flpr(); iraf.flpr()
iraf.gflush(); iraf.gflush()
#coeffile = os.path.join(snworkdir,fltfile[:-5] + '_coeffs1.dat')
coeffile = fltfile[:-5] + '_coeffs1.dat'
output = iraf.wtranback( 0, 0, nxin=nxin, nxout=nxout,
nyin=nyin, nyout=nyout, xylist=xylistfile,
coeffs=coeffile,
geomode='wcs', refim=drzfile,
inimage=fltfile+'[%i]'%ext, Stdout=1 )
for line in output:
if line.startswith(' Xin,Yin:') :
xin,yin = map(float,line.split()[1:3])
if xin>0 and xin<nxin and yin>0 and yin<nyin :
returnlist.append( (fltfile,ext,float(xin),float(yin)) )
return( returnlist )