interactionmap.py

Go to the documentation of this file.

from optparse import OptionParser
import sys, os, subprocess, shutil, ROOT, array, math

usage = "usage: %prog[options]"
parser = OptionParser(usage=usage)

parser.add_option("-f","--filename",dest="filename",help="filename for histogram creation",default="NA")

(options,args)=parser.parse_args()
pathname = options.filename

output_objects=[]
#collision=[]
#histos=[]

f = ROOT.TFile(str(pathname))
for keys in f.GetListOfKeys():
    output_objects.append(f.Get(keys.GetName()))
           
         
def drawhistos(lists,start,particle,histos):
    xmax=0
    ymax=0
    collision=[]
    histogram=[]
    histcontainer=[]
    ROOT.gStyle.SetPaintTextFormat("4.4f")
    c1=ROOT.TCanvas("c1","c1",750,750)
    c1.cd()
    for objects in lists:
        if objects.GetName()==start:
            for keys in objects.GetListOfKeys():
                collision.append(objects.Get(keys.GetName()))

    for particles in collision:
        if particles.GetName()==particle:
            #print 'Yes i got {}'.format(particle)
            for keys in particles.GetListOfKeys():
                histogram.append(particles.Get(keys.GetName()))

    for hists in histogram:
        if hists.GetName()==histos:
            #print 'Yes I got {}'.format(histos)
            for i in range(1,hists.GetNbinsX()):
                if(hists.GetXaxis()).GetBinLabel(i)=="":
                    xmax=i-1
                    break
            for j in range(1,hists.GetNbinsY()):
                if(hists.GetYaxis()).GetBinLabel(j)=="":
                    ymax=j
                    #print j
                    break                    
                        
            c1.cd()
            (hists.GetXaxis()).SetRangeUser(0,xmax)
            (hists.GetYaxis()).SetRangeUser(0,ymax)
            hists.SetMinimum(0.0001)
            hists.Draw("COLZTEXT")
    raw_input()


def allparticlehistos(lists,start,histos):
    xmax=0
    ymax=0
    collision=[]
    histogram=[]
    histcontainer=[]
    ROOT.gStyle.SetPaintTextFormat("4.4f")
    c1=ROOT.TCanvas("c1","c1",750,750)
    c1.cd()
    for objects in lists:
        if objects.GetName()==start:
            for keys in objects.GetListOfKeys():
                collision.append(objects.Get(keys.GetName()))


    for hists in collision:
        if hists.GetName()==histos:
            #print 'Yes I got {}'.format(histos)
            for i in range(1,hists.GetNbinsX()):
                if(hists.GetXaxis()).GetBinLabel(i)=="":
                    xmax=i-1
                    break
            for j in range(1,hists.GetNbinsY()):
                if(hists.GetYaxis()).GetBinLabel(j)=="":
                    ymax=j
                    #print j
                    break                    
                        
            c1.cd()
            (hists.GetXaxis()).SetRangeUser(0,xmax)
            (hists.GetYaxis()).SetRangeUser(0,ymax)
            hists.SetMinimum(0.0001)
            hists.Draw("COLZTEXT")
    raw_input()


    
#
#drawhistos(output_objects,"Produced","Proton","hmatbkgpi+")
allparticlehistos(output_objects,"Projectile","h_in_vs_mat")