Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/examples/extended/eventgenerator/exgps/macros/plotfiles.py

Version: [ ReleaseNotes ] [ 1.0 ] [ 1.1 ] [ 2.0 ] [ 3.0 ] [ 3.1 ] [ 3.2 ] [ 4.0 ] [ 4.0.p1 ] [ 4.0.p2 ] [ 4.1 ] [ 4.1.p1 ] [ 5.0 ] [ 5.0.p1 ] [ 5.1 ] [ 5.1.p1 ] [ 5.2 ] [ 5.2.p1 ] [ 5.2.p2 ] [ 6.0 ] [ 6.0.p1 ] [ 6.1 ] [ 6.2 ] [ 6.2.p1 ] [ 6.2.p2 ] [ 7.0 ] [ 7.0.p1 ] [ 7.1 ] [ 7.1.p1 ] [ 8.0 ] [ 8.0.p1 ] [ 8.1 ] [ 8.1.p1 ] [ 8.1.p2 ] [ 8.2 ] [ 8.2.p1 ] [ 8.3 ] [ 8.3.p1 ] [ 8.3.p2 ] [ 9.0 ] [ 9.0.p1 ] [ 9.0.p2 ] [ 9.1 ] [ 9.1.p1 ] [ 9.1.p2 ] [ 9.1.p3 ] [ 9.2 ] [ 9.2.p1 ] [ 9.2.p2 ] [ 9.2.p3 ] [ 9.2.p4 ] [ 9.3 ] [ 9.3.p1 ] [ 9.3.p2 ] [ 9.4 ] [ 9.4.p1 ] [ 9.4.p2 ] [ 9.4.p3 ] [ 9.4.p4 ] [ 9.5 ] [ 9.5.p1 ] [ 9.5.p2 ] [ 9.6 ] [ 9.6.p1 ] [ 9.6.p2 ] [ 9.6.p3 ] [ 9.6.p4 ] [ 10.0 ] [ 10.0.p1 ] [ 10.0.p2 ] [ 10.0.p3 ] [ 10.0.p4 ] [ 10.1 ] [ 10.1.p1 ] [ 10.1.p2 ] [ 10.1.p3 ] [ 10.2 ] [ 10.2.p1 ] [ 10.2.p2 ] [ 10.2.p3 ] [ 10.3 ] [ 10.3.p1 ] [ 10.3.p2 ] [ 10.3.p3 ] [ 10.4 ] [ 10.4.p1 ] [ 10.4.p2 ] [ 10.4.p3 ] [ 10.5 ] [ 10.5.p1 ] [ 10.6 ] [ 10.6.p1 ] [ 10.6.p2 ] [ 10.6.p3 ] [ 10.7 ] [ 10.7.p1 ] [ 10.7.p2 ] [ 10.7.p3 ] [ 10.7.p4 ] [ 11.0 ] [ 11.0.p1 ] [ 11.0.p2 ] [ 11.0.p3, ] [ 11.0.p4 ] [ 11.1 ] [ 11.1.1 ] [ 11.1.2 ] [ 11.1.3 ] [ 11.2 ] [ 11.2.1 ] [ 11.2.2 ] [ 11.3.0 ]

  1 #!/usr/bin/python
  2 
  3 from ROOT import *
  4 from array import array
  5 
  6 def plot_1_file (file):
  7     gROOT.Reset()
  8     input_file_1=TFile(file+'.root','READ')
  9     h1 = input_file_1.Get("histo/1")
 10     h2 = input_file_1.Get("histo/2")
 11     h3 = input_file_1.Get("histo/3")
 12     h4 = input_file_1.Get("histo/4")
 13     h5 = input_file_1.Get("histo/5")
 14     h6 = input_file_1.Get("histo/6")
 15 
 16     c1 = TCanvas('c1', file, 200, 10, 700, 900)
 17     c1.Divide(2,3)
 18 
 19     c1.cd(1)
 20     h1.Draw()
 21     c1.cd(2)
 22     h2.Draw()
 23     c1.cd(3)
 24     h3.Draw()
 25     c1.cd(4)
 26     h4.Draw()
 27     c1.cd(5)
 28     h5.Draw()
 29     c1.cd(6)
 30     h6.Draw()
 31     c1.Update()
 32     c1.Print("./"+file+".png")
 33     
 34     input_file_1.Close()
 35 
 36 #    h_gam.SetLineColor(2)
 37 
 38 def plot_2_files (file):
 39     gROOT.Reset()
 40 
 41     input_file_1=TFile(file+'a.root','READ')
 42     input_file_2=TFile(file+'b.root','READ')
 43 
 44 #input_file_1.cd()
 45 #h_1_1 = input_file_1.Get("h16")
 46 
 47     c1 = TCanvas('c1', file, 200, 10, 700, 500)
 48     c1.SetGridx()
 49     c1.SetGridy()
 50     c1.SetLogx()
 51     c1.SetLogy()
 52 
 53 # histogram for energy spectra
 54     n = 41
 55     bin = array( 'f' )
 56     
 57     for i in range( n ):
 58         bin.append(pow(10,(-2+0.1*i)))
 59 #
 60     h_1 = TH1F('unbiased','Source Spectrum',40,bin)
 61     h_2 = TH1F('biased','Source Spectrum',40,bin)
 62 
 63 #
 64     input_file_1.cd()
 65 # get the tuple t1
 66     t1 = input_file_1.Get('ntuple/MyTuple')
 67     print t1
 68     for i in range(t1.GetEntries()):
 69         t1.GetEntry(i)
 70         h_1.Fill(t1.Ekin,t1.weight)
 71             
 72     input_file_2.cd()
 73 # get the tuple t1
 74     t2 =  input_file_2.Get('ntuple/MyTuple')
 75     for i in range(t2.GetEntries()):
 76         t2.GetEntry(i)
 77         h_2.Fill(t2.Ekin,t2.weight)
 78                 
 79     h_2.SetLineStyle(kDashed);
 80     h_2.SetLineColor(kBlue);
 81     h_2.Draw();
 82     h_1.Draw("same") ;
 83     c1.Update()
 84     c1.Print("./"+file+".png")
 85     
 86     input_file_1.Close()
 87     input_file_2.Close()
 88