import sys from pylab import * import os.path sys.path.insert(0, os.path.join(os.path.expanduser("~"), 'build/Python/modules')) from jg.ctx import wd2jd wasp12 = {'e': 2454508.97605, 'p': 1.0914222} wasp11 = {'e': 2454729.90631, 'p': 3.722469} wd, orig, sub, add = loadtxt("Debug.dat", unpack=True) jd = wd2jd(wd) goodind = orig==orig jd = jd[goodind] orig = orig[goodind] sub = sub[goodind] add = add[goodind] avflux = average(orig) wasp12phase = ((jd - wasp12['e']) / wasp12['p']) % 1 wasp12phase[wasp12phase>0.5] -= 1.0 wasp11phase = ((jd - wasp11['e']) / wasp11['p']) % 1 wasp11phase[wasp11phase>0.5] -= 1.0
def main(args): f = pyfits.open(args.file) Catalogue = f['catalogue'].data #Flux = f['flux'].data #HJD = f['hjd'].data ## get the objects where the width is not 0 Widths = Catalogue.field("TRANS_WIDTH") Epochs = wd2jd(Catalogue.field("TRANS_EPOCH")) Periods = Catalogue.field("TRANS_PERIOD") Depths = Catalogue.field("TRANS_DEPTH") RPlanets = Catalogue.field('TRANS_RP') RStars = Catalogue.field('TRANS_RS') Inclinations = Catalogue.field("TRANS_I") Separations = Catalogue.field("TRANS_A") Names = Catalogue.field('OBJ_ID') #Index, = where(Widths!=0) Index = [] for i in range(Names.size): if "SYNTH" in Names[i]: Index.append(i) Index = array(Index) pp = PdfPages("output.pdf") pp2 = PdfPages('wasp12phase.pdf') ##ion() Reversed = Index[::-1] nObjects = 20 skipStep = 100 for i in Reversed[:nObjects*skipStep:skipStep]: CurrentModel = {'epoch': Epochs[i], 'period': Periods[i], 'rp': RPlanets[i], 'rs': RStars[i], 'i': Inclinations[i], 'a': Separations[i], } print i, CurrentModel cla() Time = wd2jd(f['hjd'].section[i]) Lightcurve = f['flux'].section[i] Errors = f['fluxerr'].section[i] # remove nans GoodIndex = (Lightcurve==Lightcurve) & (Time==Time) & (Errors==Errors) Lightcurve = Lightcurve[GoodIndex] Time = Time[GoodIndex] Errors = Errors[GoodIndex] Phase = ((Time - CurrentModel['epoch']) / (CurrentModel['period'] / secondsInDay)) % 1.0 Phase[Phase>0.5] -= 1.0 MedVal = median(Lightcurve) Lightcurve /= MedVal Errors /= MedVal errorbar(Phase, Lightcurve, Errors, ls="None", color="#AAAAAA") plot(Phase, Lightcurve, 'k,') title("Depth: %f" % (Depths[i],)) #axhline(1. - Depths[i]) #axvline(-Widths[i]/2. / CurrentModel['period'], color='k') #axvline(Widths[i]/2. / CurrentModel['period'], color='k', label='Original') xlim(-0.3, 0.3) ylim(0.5, 1.5) BoxX, BoxY = BoxCoords(Widths[i] / CurrentModel['period'], Depths[i], xlim()) plot(BoxX, BoxY, 'r-') pp.savefig() cla() Phase = ((Time - wasp12['e']) / wasp12['p']) % 1.0 Phase[Phase>0.5] -= 1.0 errorbar(Phase, Lightcurve, Errors, ls="None", color="#AAAAAA") plot(Phase, Lightcurve, 'k,') title("WASP-12b phase") xlim(-0.3, 0.3) pp2.savefig() pp.close() pp2.close()