PythonPlotting/PlotChicaneLengthsFlat.py at master · khilde/PythonPlotting · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
from ROOT import *
import sys
import os, argparse
import signal

def signal_handler(signal, frame):
        print('You pressed Ctrl+C!')
        sys.exit(0)
signal.signal(signal.SIGINT, signal_handler)
signal.signal(signal.SIGTERM, signal_handler)

def theDeriv(x,par):
    theFit2= TF1("thefit","-[3]*exp(-x*[0]/(2.47*10^-6)*exp(-4.49*10^9/([1]*[2]*299792458*[0])))",0,.03)
    theFit2.FixParameter(0,1.)
    theFit2.FixParameter(1,2.065789)
    theFit2.FixParameter(2,0.875026)
    #theFit.SetParameter(3,10000*dipoleLength)
    theFit2.FixParameter(3,1)
    #print theFit2.Derivative(x[0])
    return par[0]*theFit2.Derivative(x[0])

dipoleLength=0.03
cutLength=.03
#dipoleLength=1
nBins=100


openedFile=open("firstChicaneLength.txt",'r')
#openedFile=open("randomFirst.txt",'r')

#stops each canvas from being physically drawn
gROOT.SetBatch(True)

lines=openedFile.readlines()
oneHisto= TH1F("firstChicane","first chicane",nBins,0,dipoleLength)
counter=0
for line in lines:
    oneHisto.Fill(float(line))
    counter=counter+1

print "number Of particles= %d" %counter
#[0] is mangetic field
#[1] is gamma
#[2] is beta
#[3] is normalization, ie number of particles
#theFit= TF1("thefit","[3]*exp(-x*[0]/(2.47*10^-6)*exp(-4.49*10^9/([1]*[2]*299792458*[0])))",0,.0254)
#theFit= TF1("theFit","[3]*exp(-x*([0]/.0254*x*(x<.0254)+[0]*(x>=.0254))/(2.47*10^-6)*exp(-4.49*10^9/([1]*[2]*299792458*([0]/.0254*x*(x<.0254)+[0]*(x>=.0254)))))",0,.0254)
#theDer= TF1("theDer",theFit.Derivative(x),0,.0254)
theDer= TF1("theDer",theDeriv,0,.03,1)
theDer.SetParameter(0, counter*cutLength/nBins);
#theFit.FixParameter(0,1.2)
#theFit.FixParameter(1,2.065789)
#theFit.FixParameter(2,0.875026)
#theFit.SetParameter(3,10000*dipoleLength)
#theFit.SetParameter(3,1)
#theFit.SetParameter(3,10000*dipoleLength)
#still need to print last graph
theCanvas=TCanvas("TGraph","TGraph",0,0,500,500)
#oneGraph.GetYaxis().SetRangeUser(args.ymin,args.ymax);
#oneGraph.GetXaxis().SetLimits(args.xmin,args.xmax);
#oneGraph.SetMarkerStyle(args.markerStyle)
#oneGraph.SetMarkerSize(args.markerSize)
#oneGraph.SetTitle("theTitle")
oneHisto.Draw("Histo")
#oneHisto.Fit(theDer)
#theFit.Draw("same")
theDer.Draw("same")
#theDer.Draw()
theCanvas.Print("firstChicaneLength.png")
#theCanvas.Print("randomFirst.png")
theCanvas.Clear()

theFile=TFile("firstChangeLength.root","RECREATE")
#theFile=TFile("randomFirst.root","RECREATE")
oneHisto.Write()