def main(): ar = mastio2.K2Archive() kepid = 206103150 period = 4.16 t0 = 2168.43 #2154.14 fits = ar.getLongCadence(kepid, 3) time = fits['TIME'] cin = fits['CADENCENO'] flags = fits['SAP_QUALITY'] pa = fits['SAP_FLUX'] pdc = fits['PDCSAP_FLUX'] cent1 = fits['MOM_CENTR1'] cent2 = fits['MOM_CENTR2'] badIdx = flags > 0 i0 = np.random.randint(0, len(cin)) fits, hdr = ar.getLongTpf(kepid, 3, header=True) cube = tpf.getTargetPixelArrayFromFits(fits, hdr) gain = hdr['gain'] # cube *= gain #Compute roll phase centColRow = np.vstack((cent1, cent2)).transpose() rot = computeArcLength(centColRow, flags>0) rollPhase = rot[:,0] rollPhase[flags>0] = -9999 #A bad value diffimg.constructK2DifferenceImage(cube, 578, rollPhase, flags) return for i in range(180, len(rollPhase)): mp.clf() try: diffimg.constructK2DifferenceImage(cube, [i], rollPhase, flags) except ValueError: continue mp.savefig('fig-%05i.png' %(i))
def main(): ar = mastio2.K2Archive() kepid = 206103150 period = 4.16 t0 = 2168.43 #2154.14 fits = ar.getLongCadence(kepid, 3) time = fits['TIME'] cin = fits['CADENCENO'] flags = fits['SAP_QUALITY'] pa = fits['SAP_FLUX'] pdc = fits['PDCSAP_FLUX'] cent1 = fits['MOM_CENTR1'] cent2 = fits['MOM_CENTR2'] badIdx = flags > 0 i0 = np.random.randint(0, len(cin)) fits, hdr = ar.getLongTpf(kepid, 3, header=True) cube = tpf.getTargetPixelArrayFromFits(fits, hdr) gain = hdr['gain'] # cube *= gain #Compute roll phase centColRow = np.vstack((cent1, cent2)).transpose() rot = computeArcLength(centColRow, flags > 0) rollPhase = rot[:, 0] rollPhase[flags > 0] = -9999 #A bad value diffimg.constructK2DifferenceImage(cube, 578, rollPhase, flags) return for i in range(180, len(rollPhase)): mp.clf() try: diffimg.constructK2DifferenceImage(cube, [i], rollPhase, flags) except ValueError: continue mp.savefig('fig-%05i.png' % (i))
def oldStuff(): #Compute roll phase centColRow = np.vstack((cent1, cent2)).transpose() rot = computeArcLength(centColRow, flags > 0) rollPhase = rot[:, 0] rollPhase[flags > 0] = -9999 #A bad value #Index of transit mp.clf() time[~np.isfinite(time)] = -1 i = np.argmin(np.fabs(time - t0)) i0 = i indexBefore, indexAfter = diffimg.getIndicesOfOotImages( rollPhase, i0, flags) if True: mp.figure(1) mp.clf() rp0 = rollPhase[i] mp.axvline(time[i], color='r') mp.axhline(rp0, color='grey') mp.axvline(time[indexBefore], color='g') mp.axvline(time[indexAfter], color='g') mp.plot(time[~badIdx], rollPhase[~badIdx], 'k.') mp.plot(time[~badIdx], rot[~badIdx, 1], 'r.') plotThrusterFirings(flags, time) # mp.xlim([2150, 2160]) # mp.savefig('diag%+i.png' %(i0)) mp.xlim(2162, 2172) fits, hdr = ar.getTargetPixelFile(kepid, 3, header=True) cube = tpf.getTargetPixelArrayFromFits(fits, hdr) mp.figure(2) mp.clf() # indexInTransit = np.arange(i-4, i+5) indexInTransit = [i0] diff, oot = diffimg.constructK2DifferenceImage(cube, indexInTransit, rollPhase, flags) diffimg.plotDiffDiagnostic(diff, oot, cube[i])
def oldStuff(): #Compute roll phase centColRow = np.vstack((cent1, cent2)).transpose() rot = computeArcLength(centColRow, flags>0) rollPhase = rot[:,0] rollPhase[flags>0] = -9999 #A bad value #Index of transit mp.clf() time[ ~np.isfinite(time)] = -1 i = np.argmin( np.fabs( time-t0)) i0 = i indexBefore, indexAfter = diffimg.getIndicesOfOotImages(rollPhase, i0, flags) if True: mp.figure(1) mp.clf() rp0 = rollPhase[i] mp.axvline(time[i], color='r') mp.axhline(rp0, color='grey') mp.axvline(time[indexBefore], color='g') mp.axvline(time[indexAfter], color='g') mp.plot(time[~badIdx], rollPhase[~badIdx], 'k.') mp.plot(time[~badIdx], rot[~badIdx, 1], 'r.') plotThrusterFirings(flags,time) # mp.xlim([2150, 2160]) # mp.savefig('diag%+i.png' %(i0)) mp.xlim(2162,2172) fits, hdr = ar.getTargetPixelFile(kepid, 3, header=True) cube = tpf.getTargetPixelArrayFromFits(fits, hdr) mp.figure(2) mp.clf() # indexInTransit = np.arange(i-4, i+5) indexInTransit = [i0] diff, oot = diffimg.constructK2DifferenceImage(cube, indexInTransit, rollPhase, flags) diffimg.plotDiffDiagnostic(diff, oot, cube[i])