#!/usr/bin/env python

import sys, os

import numpy, math

import argparse

import astropy.io.fits

import astropy.stats

import loadingSavingUtils, statsUtils

import spectrumClasses, timeClasses

import scipy.optimize

import copy

import ppgplot

import scipy.signal as signal

def quad(x, a1, a2, a3):

y = a1 * x * x + a2 *x + a3

return y

class object:

self.setHJDs(MJD, HJD)

if __name__ == "__main__":

parser = argparse.ArgumentParser(description='Plots lcurve model alongside observed data.')

parser.add_argument('observed', type=str, help='Observed data.')

parser.add_argument('model', type=str, help='Modelled data.')

parser.add_argument('name', type=str, help='Object name.')

parser.add_argument('--ps', action='store_true', help = "Dump plots to ps files instead of the screen.")

arg = parser.parse_args()

print "Astropy version:", astropy.__version__

objects = []

observedData = object('observed')

dataFile = open(arg.observed, 'rt')

for line in dataFile:

fields = line.strip().split(' ')

data = {}

data['phase'] = float(fields[0])

data['observedtime'] = float(fields[1])

data['flux'] = float(fields[2])

data['flux_err'] = float(fields[3])

# print data

observedData.appendData(data)

dataFile.close()

objects.append(observedData)

modelledData = object('modelled')

dataFile = open(arg.model, 'rt')

for line in dataFile:

fields = line.strip().split(' ')

data = {}

data['phase'] = float(fields[0])

data['observedtime'] = float(fields[1])

data['flux'] = float(fields[2])

data['flux_err'] = float(fields[3])

print data

modelledData.appendData(data)

dataFile.close()

print "%d targets loaded"%len(objects)

observedData.convertFluxMagnitude()

modelledData.convertFluxMagnitude()

if arg.ps: device = arg.name + "_lcurve.ps/ps"

else: device = "/xs"

PGPlotWindow = ppgplot.pgopen(device)

ppgplot.pgask(True)

pgPlotTransform = [0, 1, 0, 0, 0, 1]

ppgplot.pgslct(PGPlotWindow)

ppgplot.pgsci(1)

ppgplot.pgask(False)

phases = observedData.getColumn('phase')

flux = observedData.getColumn('flux')

flux_err = observedData.getColumn('flux_err')

flux_err = [f/10 for f in flux_err]

flux = [f * 3631E3 for f in flux]

flux_err = [f * 3631E3 for f in flux_err]

mag = observedData.getColumn('mag')

err = observedData.getColumn('err')

maxFlux = 0

for f, fe in zip(flux, flux_err):

if f + fe > maxFlux: maxFlux = f + fe

# Duplicate data out to phase 2.0

extendedPhases = copy.deepcopy(phases)

for p in phases:

extendedPhases.append(p + 1.0)

phases = extendedPhases

flux.extend(flux)

flux_err.extend(flux_err)

mag.extend(mag)

err.extend(err)

"""

ppgplot.pgsch(1.6)

ppgplot.pgenv(0, 2, numpy.max(mag), numpy.min(mag), 0, 0)

ppgplot.pglab("Phase", "PTF magnitude", "%s"%(arg.name))

ppgplot.pgsch(1.0)

ppgplot.pgpt(phases, mag)

ppgplot.pgerrb(2, phases, mag, err, 0)

ppgplot.pgerrb(4, phases, mag, err, 0)

ppgplot.pgsci(2)

model = modelledData.getColumn('mag')

model.extend(model)

ppgplot.pgsls(2)

ppgplot.pgslw(7)

ppgplot.pgline(phases, model)

"""

ppgplot.pgsch(1.6)

ppgplot.pgenv(0, 2, 0, maxFlux, 0, 0)

ppgplot.pglab("Phase", "PTF flux (mJy)", "%s"%(arg.name))

ppgplot.pgsch(1.0)

ppgplot.pgpt(phases, flux)

ppgplot.pgerrb(2, phases, flux, flux_err, 0)

ppgplot.pgerrb(4, phases, flux, flux_err, 0)

ppgplot.pgsci(2)

ppgplot.pgsls(2)

ppgplot.pgslw(7)

modelPhases = modelledData.getColumn('phase')

temp = copy.deepcopy(modelPhases)

for p in modelPhases: temp.append(p + 1.0)

modelPhases = temp

model = modelledData.getColumn('flux')

model = [m * 3631E3 for m in model]

model.extend(model)

ppgplot.pgline(modelPhases, model)

ppgplot.pgsci(1)

ppgplot.pgsls(1)

ppgplot.pgslw(1)

# Plot the inset subplot showing the eclipse

subFlux = []

subFluxErr = []

subPhases = []

subModelPhases = []

subModel = []

startPhase = 0.96

endPhase = 1.04

print "maxFlux:", maxFlux

for index, phase in enumerate(phases):

if phase > startPhase and phase < endPhase:

subFlux.append(flux[index])

subPhases.append(phase)

subFluxErr.append(flux_err[index])

for index, phase in enumerate(modelPhases):

if phase > startPhase and phase < endPhase:

subModel.append(model[index])

subModelPhases.append(phase)

print ppgplot.pgqvp(0)

(x1, x2, y1, y2) = ppgplot.pgqvp(0)

print ppgplot.pgqwin(0)

xlower = (x1 + x2) / 2 + 0.05

xupper = x2 - 0.05

yupper = (y1 + y2) / 2

ylower = 0.22

ppgplot.pgsvp(xlower, xupper, ylower, yupper)

ppgplot.pgswin(startPhase, endPhase, 0, numpy.max(subFlux))

print ppgplot.pgqvp(0)

print ppgplot.pgqwin(0)

# ppgplot.pgsubp(4, 3)

# ppgplot.pgpanl(2, 2)

# ppgplot.pgenv(startPhase, endPhase, 0, numpy.max(flux), 0, -1)

# ppgplot.pglab("Phase", "PTF flux", "%s"%(arg.name))

ppgplot.pgbox("BCN", 0, 0, "BC", 0, 0)

ppgplot.pgpt(subPhases, subFlux)

ppgplot.pgerrb(2, subPhases, subFlux, subFluxErr, 0)

ppgplot.pgerrb(4, subPhases, subFlux, subFluxErr, 0)

ppgplot.pgsls(2)

ppgplot.pgline(subModelPhases, subModel)

ppgplot.pgsls(4)

ppgplot.pgline([1, 1], [0, numpy.max(subFlux)])

ppgplot.pgclos()

sys.exit()