sheetplot.py

import pylab as plt
import numpy as np
import os,asciitable
import haloutils

def get_haloidlist(sheet):
    if sheet==1:
        haloidlist = [95289, 1195448, 1725139,
                      447649, 5320, 581141,
                      94687, 1130025, 1387186]
    elif sheet==2:
        haloidlist = [649861,1599902, 1354437,
                      1631506, 1232164, 264569, 
                      581180, 1422331, 1725372] #1725272, 
    else:
        exit("Invalid sheet number")
    assert len(haloidlist) == 9
    return haloidlist

def sheetplot(sheetnum,Reader,Plotter,figsize=None,save=True,pdf=False):
    #TODO take a list of Plotter objects for the same reader
    figfilename = Plotter.fprefix+'_s'+str(sheetnum)
    if Plotter.fpostfix != '': figfilename += '_'+Plotter.fpostfix
    if pdf: figfilename += '.pdf'
    else:   figfilename += '.png'
    haloidlist = get_haloidlist(sheetnum)
    if figsize==None: figsize=(15,15)
    fig,axarr = plt.subplots(3,3,figsize=figsize,
                             sharex=True,sharey=True)

    plt.subplots_adjust(wspace=0,hspace=0)
    axlist = axarr.reshape(-1)
    for i,(hid,ax) in enumerate(zip(haloidlist,axlist)):
        try:
            data = Reader(hid)
        except IOError as e:
            Plotter(ax,hid,fignum=i)
            continue
        Plotter(ax,data,fignum=i)
    for ax in np.ravel(axarr[0:2,:]):
        ax.set_xlabel('')
    for ax in np.ravel(axarr[:,1:3]):
        ax.set_ylabel('')
    
    if save:
        fig.savefig(figfilename,bbox_inches='tight')
    else:
        plt.show()


def sheetcompute(sheetnum,plugin,lx=None):
    haloidlist = get_haloidlist(sheetnum)
    for hid in haloidlist:
        haloutils.get_available_hpaths(hid)
        #TODO