import ROOT, glob, math, sys, os, numpy
from ROOT import std


def drawPrelim() :
    prelim=ROOT.TLatex()
    prelim.SetTextColor(ROOT.kGray)
    prelim.SetTextSize(0.07)
    prelim.SetTextAngle(20)
    prelim.SetNDC()
    prelim.DrawLatex(0.38,0.45,"preliminary")
    return

hists={}
canvasopt={}

ROOT.gROOT.ProcessLine('gStyle->SetPalette(1)')
ROOT.gROOT.ProcessLine('gROOT->SetStyle("Plain")')

infile=ROOT.TFile("anafiles/cosmics_100ne74_apr.root")

hists['theta'] = infile.Get("htheta1")
hists['phi']   = infile.Get("hphi1")
hists['trackn']= infile.Get("trknumdist1")
hists['mom']   = infile.Get("hmom")
hists['curv']  = infile.Get("hcurv1")
hists['amp']   = infile.Get("digiamp")

canvasopt['curv']="logz"
canvasopt['mom']="logz"
canvasopt['trackn']="logz"
canvasopt['amp']="logz"

hists['theta'].GetXaxis().SetTitle("Theta [#circ]")
hists['theta'].GetYaxis().SetTitle("Counts")
hists['theta'].GetYaxis().SetTitleOffset(1.2)

hists['phi'].GetXaxis().SetTitle("Phi [#circ]")
hists['phi'].GetYaxis().SetTitle("Counts")

hists['trackn'].GetXaxis().SetTitle("Number of Tracks")
hists['trackn'].GetXaxis().SetRangeUser(0.,20.)
hists['trackn'].GetYaxis().SetTitle("Counts")

hists['mom'].GetXaxis().SetTitle("Momentum [GeV/c]")
hists['mom'].GetYaxis().SetTitle("Counts")

hists['curv'].GetXaxis().SetTitle("Curvature [1/cm]")
hists['curv'].GetXaxis().SetRangeUser(0.,0.3)
hists['curv'].GetYaxis().SetTitle("Counts")

hists['amp'].GetXaxis().SetTitle("Pad Amplitude [ADC Channels]")
hists['amp'].GetYaxis().SetTitle("Counts")

canv={}
for h in hists:
    canv[h]=ROOT.TCanvas("c"+h,h)
    canv[h].cd()
    if canvasopt.get(h,"")=="logz":
        canv[h].GetPad(0).SetLogy()
    hists[h].SetStats(ROOT.kFALSE)
    hists[h].SetTitle("")
    hists[h].Draw()
    canv[h].SaveAs("pics/cosmics/cuts/"+h+".pdf")



nix=raw_input("done?")