Rory cluster

processors/config/roryClusterAna.py

@@ -0,0 +1,56 @@
+import HpstrConf
+import sys
+import baseConfig as base
+from baseConfig import bfield
+
+base.parser.add_argument("-L", "--layer", type=int, dest="layer",
+                  help="Layer Under Investigation", metavar="layer",default=-1)
+base.parser.add_argument("-M", "--module", type=int, dest="module",
+                  help="Module Under Investigation", metavar="module",default=-1)
+base.parser.add_argument("-MC", "--MC", type=int, dest="isMC",
+                  help="IS IT MC", metavar="module",default=0)
+base.parser.add_argument("-doT", "--doT", type=int, dest="doTrack",
+                  help="DO WE DO TRACK STUFF", metavar="doTrack",default=0)
+base.parser.add_argument("-cut", "--cut", type=float, dest="cut",
+                  help="Momentum Cut for NShared Profile", metavar="cut",default=-1.0)
+
+
+options = base.parser.parse_args()
+
+# Use the input file to set the output file name
+root_file = options.inFilename
+ana_file = options.outFilename
+
+#print('LCIO file: %s' % root_file)
+#print('Root file: %s' % ana_file)
+
+p = HpstrConf.Process()
+
+#p.max_events = 1000
+p.run_mode = 1
+p.skip_events = options.skip_events
+p.max_events = options.nevents
+
+# Library containing processors
+p.add_library("libprocessors")
+
+###############################
+#          Processors         #
+###############################
+rclu = HpstrConf.Processor('rclu','RoryClusterAnaProcessor')
+rclu.parameters["debug"] = 0
+rclu.parameters["layer"] = options.layer
+rclu.parameters["module"] = options.module
+rclu.parameters["isMC"] = options.isMC
+rclu.parameters["doTrack"] = options.doTrack
+rclu.parameters["cut"] = options.cut
+
+sequence = [rclu]        
+
+p.sequence = sequence
+
+print("THE INPUT FILE IS "+str(root_file))
+p.input_files = root_file
+p.output_files = [ana_file]
+
+p.printProcess()

processors/config/roryClusterCompareAna.py

@@ -0,0 +1,56 @@
+import HpstrConf
+import sys
+import baseConfig as base
+from baseConfig import bfield
+
+base.parser.add_argument("-L", "--layer", type=int, dest="layer",
+                  help="Layer Under Investigation", metavar="layer",default=-1)
+base.parser.add_argument("-M", "--module", type=int, dest="module",
+                  help="Module Under Investigation", metavar="module",default=-1)
+base.parser.add_argument("-MC", "--MC", type=int, dest="isMC",
+                  help="IS IT MC", metavar="module",default=0)
+base.parser.add_argument("-doT", "--doT", type=int, dest="doTrack",
+                  help="DO WE DO TRACK STUFF", metavar="doTrack",default=0)
+base.parser.add_argument("-cut", "--cut", type=float, dest="cut",
+                  help="Momentum Cut for NShared Profile", metavar="cut",default=-1.0)
+
+
+options = base.parser.parse_args()
+
+# Use the input file to set the output file name
+root_file = options.inFilename
+ana_file = options.outFilename
+
+#print('LCIO file: %s' % root_file)
+#print('Root file: %s' % ana_file)
+
+p = HpstrConf.Process()
+
+#p.max_events = 1000
+p.run_mode = 1
+p.skip_events = options.skip_events
+p.max_events = options.nevents
+
+# Library containing processors
+p.add_library("libprocessors")
+
+###############################
+#          Processors         #
+###############################
+rcclu = HpstrConf.Processor('rclu','RoryClusterCompareAnaProcessor')
+rcclu.parameters["debug"] = 0
+rcclu.parameters["layer"] = options.layer
+rcclu.parameters["module"] = options.module
+rcclu.parameters["isMC"] = options.isMC
+rcclu.parameters["doTrack"] = options.doTrack
+rcclu.parameters["cut"] = options.cut
+
+sequence = [rcclu]        
+
+p.sequence = sequence
+
+print("THE INPUT FILE IS "+str(root_file))
+p.input_files = root_file
+p.output_files = [ana_file]
+
+p.printProcess()

processors/include/RoryClusterAnaProcessor.h

@@ -0,0 +1,157 @@
+#ifndef __RORYCLUSTER_ANAPROCESSOR_H__
+#define __RORYCLUSTER_ANAPROCESSOR_H__
+
+//HPSTR
+#include "HpsEvent.h"
+#include "RawSvtHit.h"
+#include "TrackerHit.h"
+#include "RawSvtHitHistos.h"
+#include "AnaHelpers.h"
+#include "Event.h"
+#include "BaseSelector.h"
+#include "RawSvtHitHistos.h"
+#include "EventHeader.h"
+#include "VTPData.h"
+#include "TSData.h"
+#include "CalCluster.h"
+#include "Track.h"
+#include "TrackerHit.h"
+#include "Collections.h"
+//#include <IMPL/TrackerHitImpl.h>"
+//ROOT
+
+#include "Processor.h"
+#include "TClonesArray.h"
+#include "TBranch.h"
+#include "TTree.h"
+#include "TFile.h"
+#include "TF1.h"
+#include "TGraphErrors.h"
+#include "TAxis.h"
+#include "TROOT.h"
+#include "TPad.h"
+#include "TCanvas.h"
+#include "TLegend.h"
+#include "TProfile.h"
+
+class TTree;
+
+
+class RoryClusterAnaProcessor : public Processor {
+
+    public:
+
+        RoryClusterAnaProcessor(const std::string& name, Process& process);
+
+        ~RoryClusterAnaProcessor();
+
+        virtual bool process(IEvent* ievent);
+
+        virtual void initialize(TTree* tree);
+
+        virtual void Plot1();
+
+        virtual void Plot2();
+
+        virtual void Plot3();
+
+        virtual void Plot4();
+
+        virtual void Plot5();
+
+        virtual void Plot6();
+
+        virtual void Plot7();
+
+        virtual void TrackPlot1();
+
+        //virtual void ClusterFit();
+
+        virtual void fillDeads();
+
+        virtual int GetStrip(int feb, int hyb,int strip);
+
+        //virtual void sample(RawSvtHit* thisHit, std::string word, IEvent* ievent, long t,int i);
+
+        //virtual int hitEff(IEvent* ievent, int L);
+
+        //virtual TF1* fourPoleFitFunction(std::string word, int caser);
+
+        //virtual float str_to_float(std::string word);
+
+        //float reverseEngineerTime(float ti, long t);
+
+        //virtual int maximum(int arr[]);
+
+        virtual void finalize();
+
+        virtual void configure(const ParameterSet& parameters);
+
+    private:
+
+        //Containers to hold histogrammer info
+        //RawSvtHitHistos* histos{nullptr};
+        ModuleMapper * mmapper_;
+
+        TTree* tree_;
+        TBranch* bClusters_{nullptr};
+        TBranch* bClustersKF_{nullptr};
+        TBranch* bsvtraw_{nullptr};
+        TBranch* btracks_{nullptr};
+
+        int layer_{-1};
+        int module_{-1};
+
+        TH1F* layers_;
+        TH1F* layersOnTrk_;
+        TH1F* layersOffTrk_;
+        TH1F* charges_;
+        TH1F* chargesOnTrk_;
+        TH1F* chargesOffTrk_;
+
+        TH1F* layersNTD_;
+        TH1F* layersOnTrkNTD_;
+        TH1F* layersOffTrkNTD_;
+        TH1F* chargesNTD_;
+        TH1F* chargesOnTrkNTD_;
+        TH1F* chargesOffTrkNTD_;
+
+        TH1F* positions_;
+        TH1F* positionsOnTrk_;
+        TH1F* ClusDistances_;
+        TH1F* ClusDistancesNTD_;
+
+        TH1F* times_;
+        TH1F* timesOnTrk_;
+        TH1F* timesOffTrk_;
+        TH1F* timesNTD_;
+        TH1F* timesOnTrkNTD_;
+        TH1F* timesOffTrkNTD_;
+
+        //TRACKING RELATED VARIABLES
+
+        TH2F* Z0VNShare2Hist_;
+        TH2F* Z0VNShare2HistCut_;
+        TH1F* SharedAmplitudes_;
+        TH1F* UnSharedAmplitudes_;
+        TH1F* SharedTimes_;
+        TH1F* UnSharedTimes_;
+
+        float pcut_{-1.0};
+
+        bool doingTracks_{false};
+
+        float Deads_[24576];
+
+        std::vector<TrackerHit*> * Clusters_{};
+        std::vector<TrackerHit*> * ClustersKF_{};
+        std::vector<RawSvtHit*> * svtraw_{};
+        std::vector<Track*> * tracks_{};
+
+
+        //std::string anaName_{"roryClusterAna"};
+        int debug_{0};
+        int isMC_{0};
+};
+
+#endif