1 |
|
#!/usr/bin/env python |
2 |
|
|
3 |
< |
import os, commands |
3 |
> |
import os, commands, sys |
4 |
|
from array import array |
5 |
|
import ROOT |
6 |
|
|
7 |
< |
|
8 |
< |
###Read data from txt file |
9 |
< |
file = open('pressure_twiki_160406-171484.dat') |
10 |
< |
counter = 0 |
11 |
< |
runs = [] |
12 |
< |
pressure = [] |
13 |
< |
for line in file: |
14 |
< |
print line.split() |
15 |
< |
runs.append(float(line.split()[0])) |
16 |
< |
if(len(line.split())>1): |
17 |
< |
pressure.append(float(line.split()[1])) |
18 |
< |
else: |
19 |
< |
pressure.append(0.0) |
20 |
< |
counter+=1 |
21 |
< |
|
22 |
< |
x = array("d",runs) |
23 |
< |
y = array("d",pressure) |
24 |
< |
|
25 |
< |
###Define efficiency map histogram |
26 |
< |
myFile = ROOT.TFile("PressureGraph.root","RECREATE") |
27 |
< |
myGraph = ROOT.TGraph(len(runs),x,y) |
28 |
< |
|
29 |
< |
myGraph.Write() |
30 |
< |
myFile.Write() |
31 |
< |
myFile.Close() |
7 |
> |
if __name__ == "__main__": |
8 |
> |
if len(sys.argv[1:]) != 1: |
9 |
> |
print "\nUsage "+sys.argv[0]+" <file>" |
10 |
> |
print "where <file> is a text file with 2 columns:" |
11 |
> |
print "RUN_NUMBER PRESSURE[mbar]\n" |
12 |
> |
exit() |
13 |
> |
|
14 |
> |
###Read data from txt file |
15 |
> |
#file = open('pressure_twiki_160406-172268.dat') |
16 |
> |
file = open(sys.argv[1]) |
17 |
> |
counter = 0 |
18 |
> |
runs = [] |
19 |
> |
pressure = [] |
20 |
> |
for line in file: |
21 |
> |
print line.split() |
22 |
> |
if(len(line.split())>0): |
23 |
> |
runs.append(float(line.split()[0])) |
24 |
> |
if(len(line.split())>1): |
25 |
> |
pressure.append(float(line.split()[1])) |
26 |
> |
else: |
27 |
> |
pressure.append(0.0) |
28 |
> |
counter+=1 |
29 |
> |
|
30 |
> |
x = array("d",runs) |
31 |
> |
y = array("d",pressure) |
32 |
> |
|
33 |
> |
###Define efficiency map histogram |
34 |
> |
myFile = ROOT.TFile("PressureGraph.root","RECREATE") |
35 |
> |
myGraph = ROOT.TGraph(len(runs),x,y) |
36 |
> |
myGraph.SetName("Graph") |
37 |
> |
|
38 |
> |
myGraph.Write() |
39 |
> |
myFile.Write() |
40 |
> |
myFile.Close() |