예제 #1
0
def book_maker(data_table, lookup_table, savepath, sort_column, max_i=-1, reverse=True):
    """ A function to make lightcurve-and-period books.
    
    Inputs:
      data_table -- an ATpy table with time-series photometry.
      lookup_table -- an ATpy table with star names, IDs, and other parameters
      savepath -- Where to save output files to
      sort_column -- (str) a column in lookup_table to sort the book by.

    Optional inputs:
      max_i -- how many stars to go through before stopping
               (defaults to all of them)
      reverse -- whether to iterate over the sorted column backwards
                 (defaults to True)
    """

    # First, let's sort our lookup table by the sort column
    lookup_table.sort("%s" % sort_column)

    # This is SO HACKISH
    if reverse:
        look_table = reversed(lookup_table)
    else:
        look_table = lookup_table

    # For each star
    for row, i in zip(look_table, np.arange(len(lookup_table))):

        desig = row["Designation"]
        name = "p%d.%d" % (desig, round(row["fraction"] * 100, 0))
        sid = row["SOURCEID"]
        # For each season

        for season in [0, 1, 2, 3]:
            outname_a = savepath + ("%.3d_%.3d" % (i, desig)) + "." + str(season) + "a"

            outname_b = savepath + ("%.3d_%.3d" % (i, desig)) + "." + str(season) + "b"

            tr.plot_5(data_table, sid, outfile=outname_a, name=name, season=season, png_too=True)
            tr.plot_page_periods(data_table, sid, outfile=outname_b, name=name, season=season, png_too=True)

        print name + " is done"

        if i == max_i:
            break
예제 #2
0
import atpy
import numpy as np
import matplotlib.pyplot as plt
import tr

datapath = '/home/tom/reu/DATA/Merged_Catalogs/'
savepath = '/home/tom/reu/logs/feb_15_2011/huge/'

# Let's load up our data...

wc = atpy.Table(datapath + 'wserv_errbits_combined.fits')

goods = atpy.Table(datapath + '5_sigma_disks_good1.fits')

for row in goods:
    desig = row['Designation']
    name = "p%d.%d" % ( row[4], round(row[3]*100, 0) )
    sid = row[0]
    
    for season in [1,2,3]:
        outname = savepath+ ("%.3d" % desig) + "." + str(season) + ".pdf"
        outname_png = ( savepath+"png/"+ ("%.3d" % desig) + 
                        "." + str(season) + ".png" )
        tr.plot_5(wc, sid, season=season, name=name, outfile=outname)
        tr.plot_5(wc, sid, season=season, name=name, outfile=outname_png)

    print name + " is done"