Eventually including orbit drift corrections

lsctools/config.py

@@ -4,7 +4,7 @@
 options = {'nominalTitle': 'Nominal Position [#mum]'}
 EOSPATH = '/eos/cms/store/group'
 OUTPATH = dirname(__file__)+'/../results'
-OWNPATH = '/afs/cern.ch/work/j/joknolle/store'
+OWNPATH = '/eos/cms/store/user/capalmer/lengthScale'
 
 def PCC():
     """Set common parameters of PCC data sets"""
@@ -276,6 +276,8 @@ def PCC2016ReRecoJan2017():
                             '130_084911'), (5, 4, '213_002332'), (6, 1, \
                             '130_084929'), (7, 3, '213_002401'), (8, 1, \
                             '130_084943'))]
+
+    options['fulltrees'] = ['/comm_luminosity/PCC/VdM/May2016Scans/PCC_Run2016B-17Jan2017-v1_ZeroBiasALL_LengthScale/']
     options['minitrees'] = [s+'_FPix_Feb27' for s in options['fulltrees']]
     # options['dataset'] = ['2016_rereco_jan17', \
     #                       '2016 ReReco Jan 2017']

run2016.py

@@ -0,0 +1,23 @@
+from lsctools import config, prepare, gather, fit, plot, analyze
+
+# Load 2016 configuration
+config.PCC2016ReRecoJan2017()
+
+# Find ROOT files belonging to LS scans
+prepare.findRootFiles('fulltrees')
+
+for scan in config.options['scans']:
+    # Get vertex data from ROOT files
+    gather.vertexPositionPerBxStep(scan, all=True, alternative=True)
+
+    # Fit vertex positions at each step with Gaussian
+    fit.vertexPosition(scan, all=True, alternative=True, fitmethod='L')
+
+    # Create plots of vertex distribution at each step
+    plot.vertexPositionPerBxStep(scan, all=True, alternative=True, fit='L')
+
+    # Analyze full scan
+    analyze.vertexPosition(scan, all=True, alternative=True, fitted='L')
+
+    # Plot full scan results
+    plot.vertexPositionPerDirectionBx(scan, all=True, alternative=True, fitted='L')