import pylab, numpy, pyfits, scipy

from pylab import *

from matplotlib import *

from numpy import *

from pyfits import *

import kepmsg, kepio, kepfunc

import sys, glob

#import kepio, kepmsg, kepkey, kepplot, kepfit, keparray, kepfunc

#import sys, time, re, math, glob

#from scipy import interpolate, optimize, ndimage, stats

#from scipy.optimize import fmin_powell

#from scipy.interpolate import RectBivariateSpline

#from scipy.ndimage import interpolation

#from scipy.ndimage.interpolation import shift, rotate

#from scipy.stats import nanmean

# -----------------------------------------------------------

# core code

def kepfake(outfile,prfdir,module,output,column,row,kepmag,background,xdim,ydim,

return

# -----------------------------------------------------------

# plot channel image

def plotimage(imgflux_pl,zminfl,zmaxfl,plmode,row,column,xdim,ydim,tlabel,colmap):

return

# -----------------------------------------------------------

# main

if '--shell' in sys.argv:

import argparse

parser = argparse.ArgumentParser(description='Fitting PRF model to Target Pixel image')

parser.add_argument('--shell', action='store_true', help='Are we running from the shell?')

parser.add_argument('outfile', help='Name of output target pixel file', type=str)

parser.add_argument('--modile', '-b', help='Order of background polynmial fit', default=1, dest='border', type=int)

parser.add_argument('--rownum', '-r', default=2200, help='Row number of image stored in infile', dest='rownum', type=int)

parser.add_argument('columns', help='Column number of each source to be fit', type=str)

parser.add_argument('rows', help='Row number of each source to be fit', type=str)

parser.add_argument('fluxes', help='Relative flux of each source to be fit', type=str)

parser.add_argument('--border', '-b', help='Order of background polynmial fit', default=1, dest='border', type=int)

parser.add_argument('--background', action='store_true', help='Fit background?', default=False)

parser.add_argument('--focus', action='store_true', help='Fit focus changes?', default=False)

parser.add_argument('prfdir', help='Folder containing Point Response Function FITS files', type=str)

parser.add_argument('--xtol', '-x', default=1.0e-4, help='Fit parameter tolerance', dest='xtol', type=float)

parser.add_argument('--ftol', '-f', default=1.0, help='Fit minimization tolerance', dest='ftol', type=float)

parser.add_argument('--imscale', '-i', help='Type of image intensity scale', default='linear', dest='imscale', type=str,choices=['linear','logarithmic','squareroot'])

parser.add_argument('--cmap', '-c', help='Image colormap', default='YlOrBr', dest='cmap', type=str,choices=['Accent','Blues','BrBG','BuGn','BuPu','Dark2','GnBu','Greens','Greys','OrRd','Oranges','PRGn','Paired','Pastel1','Pastel2','PiYG','PuBu','PuBuGn','PuOr','PuRd','Purples','RdBu','RdGy','RdPu','RdYlBu','RdYlGn','Reds','Set1','Set2','Set3','Spectral','YlGn','YlGnBu','YlOrBr','YlOrRd','afmhot','autumn','binary','bone','brg','bwr','cool','copper','flag','gist_earth','gist_gray','gist_heat','gist_ncar','gist_rainbow','gist_yarg','gnuplot','gnuplot2','gray','hot','hsv','jet','ocean','pink','prism','rainbow','seismic','spectral','spring','summer','terrain','winter','browse'])

parser.add_argument('--plot', action='store_true', help='Plot fit results?', default=False)

parser.add_argument('--verbose', action='store_true', help='Write to a log file?')

parser.add_argument('--logfile', '-l', default='kepfakephot.log', help='Name of ascii log file', dest='logfile', type=str)

parser.add_argument('--status', '-e', help='Exit status (0=good)', default=0, dest='status', type=int)

args = parser.parse_args()

cmdLine=True

kepfake(args.infile,args.plotfile,args.rownum,args.columns,args.rows,args.fluxes,args.border,

args.background,args.focus,args.prfdir,args.xtol,args.ftol,args.imscale,args.cmap,

args.plot,args.verbose,args.logfile,args.status,cmdLine)

else:

from pyraf import iraf

parfile = iraf.osfn("kepler$kepfake.par")

t = iraf.IrafTaskFactory(taskname="kepfake", value=parfile, function=kepfake)