コード例 #1
0
ファイル: MichelESimpleToy.py プロジェクト: uboone/larlite
# Make a fake track and shower
# shower:
shrDir = geoalgo.Vector(0.3, 1.0, 0.5)  # direction
shrStart = geoalgo.Vector(10, -10, 30)  # start point
shower = ertool.Shower(shrStart, shrDir, 20, 5)
shower._energy = 100
shower._dedx = 2.2

# track:
track = ertool.Track()
for n in xrange(11):
    track.push_back(
        geoalgo.Vector(-45.1 + (10 + 45) * (n / 10.),
                       15.1 - (10 + 15) * (n / 10.),
                       4.1 + (30 - 4) * (n / 10.)))
track._energy = 435

plt.ion()
viewer = GeoViewer()
viewer.add(shower, 'shower', 'b')
viewer.add(track, 'track', 'r')
viewer.show()
try:
    counter = input('Hit ENTER when you are ready to close the viewer')
except:
    print "Continuing to analys..."

# are these shower and track correlated? is the shower a michel?
dist = shower.Start().Dist(track.back())
print "distance between shower start and track end is: %f" % dist
コード例 #2
0
ファイル: GeoAlgoSphere.py プロジェクト: uboone/larlite
import ROOT
from ROOT import ertool
from ROOT import geoalgo
from basictool.geoviewer import GeoViewer
import matplotlib.pyplot as plt
import random as rnd
import time
import datetime

plt.ion()

tottime = 0

# track:
viewer2 = GeoViewer()
viewer2._use_box = False
geoAlgo = geoalgo.GeoAlgo()
ntries = 1000
npoints = 800
#for x in xrange(ntries):
track = ertool.Track()
#viewer2._fig.clear()
#for n in xrange(npoints):
#time.sleep(0.01)
point = geoalgo.Vector(rnd.random(), rnd.random(), rnd.random())
n = 0
viewer2.add(point, 'pt %i' % n, 'k')
track.push_back(point)
viewer2.add(point, 'pt %i' % n, 'k')
track.push_back(point)
viewer2.add(point, 'pt %i' % n, 'k')
コード例 #3
0
    msg = '\n'
    msg += "Usage 1: %s $INPUT_ROOT_FILE(s)\n" % sys.argv[0]
    msg += '\n'
    sys.stderr.write(msg)
    sys.exit(1)

from larlite import larlite as fmwk

# prepare viewer for GeoAlgo objects
from ROOT import geoalgo
from basictool.geoviewer import GeoViewer
import matplotlib.pyplot as plt

# load viewer
plt.ion()
viewer = GeoViewer(width=4, height=8)
viewer._use_box = False

# Create ana_processor instance
my_proc = fmwk.ana_processor()

# Set input root file
for x in xrange(len(sys.argv) - 1):
    my_proc.add_input_file(sys.argv[x + 1])

# Specify IO mode
my_proc.set_io_mode(fmwk.storage_manager.kREAD)

# Specify output root file name
my_proc.set_ana_output_file("PaddleTrackFilter_output.root")
コード例 #4
0
    msg = '\n'
    msg += "Usage 1: %s $INPUT_ROOT_FILE(s)\n" % sys.argv[0]
    msg += '\n'
    sys.stderr.write(msg)
    sys.exit(1)

from larlite import larlite as fmwk

# prepare viewer for GeoAlgo objects
from ROOT import geoalgo
from basictool.geoviewer import GeoViewer
import matplotlib.pyplot as plt

# load viewer
plt.ion()
viewer = GeoViewer(width=7, height=7)
#viewer = GeoViewer()
viewer._use_box = False
geoAlgo = geoalgo.GeoAlgo()

# Create ana_processor instance
my_proc = fmwk.ana_processor()

# Set input root file
for x in xrange(len(sys.argv) - 1):
    my_proc.add_input_file(sys.argv[x + 1])

# Specify IO mode
my_proc.set_io_mode(fmwk.storage_manager.kREAD)

# Specify output root file name