#!/usr/bin/env python

"""

Created on Wed Dec 30 16:11:56 2015

@author: John Swoboda

"""

import os, glob,inspect,getopt,sys

import shutil

import pdb

import scipy as sp

import numbers

import matplotlib

import pickle

matplotlib.use('Agg')

from SimISR.IonoContainer import IonoContainer, MakeTestIonoclass,makeionocombined

import SimISR.runsim as runsim

from SimISR.radarData import makeCovmat

from SimISR.analysisplots import analysisdump

from SimISR.utilFunctions import readconfigfile,spect2acf

from SimISR.operators import RadarSpaceTimeOperator

import matplotlib.pyplot as plt

import scipy.fftpack as scfft

import cvxpy as cvx

from PlaneProc import makeline, runradarsims

from PlaneProcPlot import plotinputdata,plotoutput,ploterrors,plotalphaerror,plotLcurve,plotpercenterror

def invertRSTO(RSTO,Iono,alpha_list=1e-2,invtype='tik',rbounds=[100,200],Nlin=0):

return (ionoout,datadif,constval)

def runinversion(basedir,configfile,acfdir='ACF',invtype='tik'):

ionoout.saveh5(outfile)

def mkalphalist(pnamefile):

pickleFile.close()

def parametersweep(basedir,configfile,acfdir='ACF',invtype='tik'):

fig.savefig(os.path.join(costdir,'lcurve{0}-{1}.png'.format(acfdir,invtype)))

def diffmat(dims,order = 'C'):

return tuple(outD)

def cgmat(A,x,b,M=None,max_it=100,tol=1e-8):

return (x,error,max_it,True)

if __name__== '__main__':

from argparse import ArgumentParser

descr = '''

This script will perform the basic run est for ISR sim.

'''

curpath = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))

p = ArgumentParser(description=descr)

p.add_argument('-t','--times',help='Times, as locations in the output time vector array that will be fit.',nargs='+',default=[])

p.add_argument('-i','--idir',help='Base directory',default='all')

p.add_argument('-c','--config',help='Config file for simlation',default = 'planeproc2_stat.ini')

p.add_argument('-r ','--re',help='Remake data True or False.',type=bool,default=False)

p.add_argument("-p", "--path",help='Number of pulses.',default=curpath)

p.add_argument("-l", "--linewid",help='Line Width in number of Samples.',default=1)

p.add_argument('-m', "--mult", help="Multiplication of enhancement.", default=5.)

p.add_argument('-w', "--wtimes", help="Put times at top of plots.",default='n')

p.add_argument('-k', "--ktype",help='The type of constraint can be tik tikd tv.',default='')

p.add_argument('-a', "--acftype",help='The ACF directory that will have the inversions applied to it',default='ACFMat')

p.add_argument('-g', "--gamma",help='The Parameter gamma for the constraints',default=1e-2)

p.add_argument('-f','--funclist',help='Functions to be uses',nargs='+',default=['spectrums','applymat','fittingmat'])#action='append',dest='collection',default=['spectrums','radardata','fitting','analysis'])

args = p.parse_args()

basedir = args.idir

curpath = args.path

configfile = args.config

remakealldata = args.re

funcnamelist = args.funclist

fittimes = args.times

lw =float( args.linewid)

mult = float(args.mult)

wtimes=args.wtimes.lower()=='y'

acffolder=args.acftype

invtype=args.ktype

gamma=float(args.gamma)

if len(fittimes)==0:

fittimes=None

else:

fittimes = [int(i) for i in fittimes]

configlist = ['planeproc2.ini','planeproc2_stat.ini','dishplaneproc.ini','dishplaneproc_stat.ini']

if basedir.lower() == 'all':

basedirlist = glob.glob(os.path.join(curpath,'exp_width_*'))

else:

basedirlist = basedir.split()

if 'paramsweep' in funcnamelist:

parametersweep(basedir,configfile,acfdir=acffolder,invtype=invtype)

funcnamelist.remove('paramsweep')

if 'invertdata' in funcnamelist:

runinversion(basedir,configfile,acfdir=acffolder,invtype=invtype)

funcnamelist.remove('invertdata')

if 'origdata' in funcnamelist:

funcnamelist.remove('origdata')

makedirs = True

(sensdict,simparams) = readconfigfile(configfile)

azangles = [iang[0] for iang in simparams['angles']]

meanaz = sp.mean(azangles)

for ibase in basedirlist:

makeline(ibase,meanaz,linewidth=lw,multval = mult)

if 'all' in funcnamelist:

funcnamelist=['spectrums','applymat','fittingmat','plotting']

plotboolin = False

plotboolout= False

ploterror=False

plotmat=False

plotmatinv=False

if 'plotting' in funcnamelist:

plotboolin=True

plotboolout=True

ploterror=True

funcnamelist.remove('plotting')

if 'plottingin' in funcnamelist:

plotboolin=True

funcnamelist.remove('plottingin')

if 'plottingout' in funcnamelist:

plotboolout=True

funcnamelist.remove('plottingout')

if 'plottingerror' in funcnamelist:

ploterror=True

funcnamelist.remove('plottingerror')

if 'plottingmat' in funcnamelist:

plotmat=True

funcnamelist.remove('plottingmat')

if 'plottingmatinv' in funcnamelist:

plotmatinv=True

funcnamelist.remove('plottingmatinv')

for ibase in basedirlist:

if len(funcnamelist)>0:

runradarsims(ibase,funcnamelist,configfile,remakealldata,fittimes,invtype=invtype)

#save2dropbox(ibase)

if plotboolin:

plotinputdata(ibase,os.path.join(ibase,'Inputimages'),wtimes)

if plotboolout:

#

plotoutput(ibase,os.path.join(ibase,'fittedimages{}'.format(invtype)),configfile,wtimes,fitpath='FittedInv',fitfile='fitteddata{0}.h5'.format(invtype))

if plotmat:

plotoutput(ibase,os.path.join(ibase,'fittedimagesmat'),configfile,wtimes,fitpath='FittedMat')

if plotmatinv:

plotoutput(ibase,os.path.join(ibase,'fittedimagesmat'),configfile,wtimes,fitpath='FittedMatInv',fitfile='fitteddata{0}.h5'.format(invtype))

if ploterror:

ploterrors(ibase,os.path.join(ibase,'fittederroronlyimages'),configfile,wtimes,fitpath='FittedMat')

plotpercenterror(ibase,os.path.join(ibase,'fittederroronlyimages'),configfile,wtimes,fitpath='FittedMat')

#save2dropbox(ibase)