import matplotlib.pyplot as plt #archive_path = ('/gscratch/stf/bmmorris/friedrich/chains{0:03d}.hdf5' # .format(int(sys.argv[1]))) # archive_path = ('/media/PASSPORT/friedrich/chains{0:03d}.hdf5' # .format(int(sys.argv[1]))) # # m = MCMCResults(archive_path) # # m.plot_corner() # # m.plot_lnprob() # # m.plot_max_lnp_lc(hat11_params_morris()) # # m.plot_lat_lon() # # m.plot_star() # m.plot_star_projected() # plt.savefig('tmp/{0:03d}.png'.format(int(sys.argv[1]))) # #plt.show() archive_paths = sorted(glob('/local/tmp/friedrich/k17/chains???.hdf5')) #archive_paths = ['chains017.hdf5'] for archive_path in archive_paths: m = MCMCResults(archive_path, k17_params_morris()) m.plot_star_projected() #m.plot_corner() #m.plot_lnprob() # m.plot_max_lnp_lc() transit_number = m.index.split('chains')[1] plt.savefig('tmp/{0:03d}.png'.format(int(transit_number))) # plt.show() plt.close() #plt.show()
# on laptop: light_curve_paths = glob('/Users/bmmorris/data/kepler17/*slc.fits') output_dir = os.path.abspath('/Users/bmmorris/data') elif os.path.exists('/usr/lusers/bmmorris/data/kepler17/'): # on Hyak light_curve_paths = glob('/usr/lusers/bmmorris/data/kepler17/*slc.fits') output_dir = os.path.abspath('/gscratch/stf/bmmorris/friedrich/k17') elif os.path.exists('/local/tmp/kepler17'): # on mist light_curve_paths = glob('/local/tmp/kepler17/*slc.fits') output_dir = os.path.abspath('./') else: raise ValueError('No input files found.') transit_params = k17_params_morris() depth = transit_params.rp**2 # Construct light curve object from the raw data whole_lc = LightCurve.from_raw_fits(light_curve_paths, name='K17') transits = LightCurve(**whole_lc.mask_out_of_transit(transit_params, oot_duration_fraction=0.5) ).get_transit_light_curves(transit_params) transits = [transit for transit in transits if len(transit.fluxes) > 225] # Compute maxes for each quarter available_quarters = whole_lc.get_available_quarters() quarters = [whole_lc.get_quarter(q) for q in whole_lc.get_available_quarters()] quarterly_maxes = {} set_upper_limit = 3e4 for i, quarter_number, lc in zip(range(len(available_quarters)),