import matplotlib
matplotlib.use('Agg')
import yt
import numpy as np



BASE = "/home/regan/data/Accretion/Tests/ConvergingFlow/BHFeedback/SMS/Ref18/InitialAccretion/"

Files = [
    BASE + "DD0120/DD0120"]

maxmass = -1
for f in Files:
    NumAPs = 0
    ds = yt.load(f)
    dd = ds.all_data()
    flist = dir(ds.fields)
    ap = flist[1]
    if yt.is_root():
        print "Opening file %s " % (ds.parameter_filename)
    try:
        NumAPs = len(dd[ap, "level"])
        if yt.is_root():
            print "Number of %s active particles = %d" % (ap, NumAPs)
    except:
        if yt.is_root():
            print "No active particles found"
    if(NumAPs > 0):
        for i in range(NumAPs):
            print "ID = ", dd["SmartStar", "identifier"][i]
            timeindex = dd["SmartStar", "TimeIndex"][i].d
            print "timeindex = ", int(timeindex)
            #print "Test: Accretion Array Shape = ", dd["SmartStar", "AccretionRate"][i]
            print "Accretion Rate = ", dd["SmartStar", "AccretionRate"][i]