TRooTrackerVtxModuleBase.hxx

Go to the documentation of this file.

#ifndef TRooTrackerVtxModuleBase_hxx_seen
#define TRooTrackerVtxModuleBase_hxx_seen
#include <map>
#include <string>
#include <utility>
#include <vector>

#include "TClonesArray.h"
#include "TFile.h"
#include "TLorentzVector.h"
#include "TNamed.h"
#include "TTree.h"

#include "TND280Event.hxx"

#include "EeventAnalysis.hxx"
#include "TAnalysisTruthModuleBase.hxx"
#include "TG4PrimaryParticle.hxx"
#include "TG4PrimaryVertex.hxx"
#include "TJNuBeamFlux.hxx"

namespace ND {
class TRooTrackerVtxModuleBase;
OA_EXCEPTION(ERooTrackerVerticesModule, EeventAnalysis);
OA_EXCEPTION(EPassThroughMismatch, ERooTrackerVerticesModule);
OA_EXCEPTION(EIncorrectInputGenerator, ERooTrackerVerticesModule);
OA_EXCEPTION(ENoKinematicsPassThroughTrees, ERooTrackerVerticesModule);
};
///
/// Abstract base class for converting the neutrino MC pass-through information
/// into eventAnalysis defined objects which hold the truth information about
/// the interaction vertices in the event. This is so that inherited classes do
/// not have to replicate the funtionality of picking up the pass-through
/// information.
///
class ND::TRooTrackerVtxModuleBase : public TAnalysisTruthModuleBase {
 public:
  TRooTrackerVtxModuleBase();
  virtual ~TRooTrackerVtxModuleBase();
  virtual Bool_t IsEnabledByDefault() const { return kTRUE; }

  // Derived classes share this functionality
  void InitializeBranches();
  void FillConfigTree(TTree* configTree);
  bool FillTree(ND::TND280Event&);
  Bool_t ProcessFirstEvent(ND::TND280Event& event);
  void SetFluxTreeAddresses(ND::JNuBeamFlux* flux);
  // Derived classes have to implement this functionality
  virtual void SetGeneratorTreeAddresses() = 0;
  virtual Bool_t IsKnownGenerator(const char*) = 0;
  virtual Bool_t CheckMismatch(ND::TG4PrimaryVertex vtx) = 0;
  virtual Bool_t IsGeantinoVtx(ND::TG4PrimaryVertex vtx) = 0;
  virtual void FillVtx(int entry) = 0;

  void SetBeginFile(TFile* input) { fLastBeginFile = input; }

 public:
  Int_t fNVtx;         ///< The number of vertices
  TClonesArray* fVtx;  ///< TClonesArray of vertex objects

 protected:
  // File from last time BeginFile was called
  TFile* fLastBeginFile;

  TFile* fCurrInputFile;   ///< Current loaded pass-through file.

  TTree* fRooTrackerTree;   ///< Current loaded rootracker file.

  TTree* fInputFileTree;   ///< Pass-through book keeping trees.

  TTree* fInputKinemTree;
  int fOrigTreeEntryNumber;   ///< Entry in original rootracker tree.

  int fInputTreeEntryNumber;   ///< The entry number in the current rootracker file.

  int fOrigInputTreeEntries;   ///< The number of entries in the original input tree. Important for working  out POTs.
  double fOrigInputTreePOT;
  std::string fOrigInputFileName;   ///< The original rootracker tree POT.

  std::string fGeneratorName;   ///< The generator name. Currently genie or neut.

  std::string fInputTreeName;   ///< The input rootracker tree name. Currently gRooTracker or nRooTracker.

  double fTimeInSpill;   ///< The time within the spill for vertex.

  int fTruthVertexID;   ///< The ID of the truth vertex created from this generator vertex.


 private:
  bool LoadPassThroughInput(const ND::TG4PrimaryVertex& vtx);
  bool UpdateBookKeepingInfo(const ND::TG4PrimaryVertex& vtx);
  bool AreTreesLoaded(const ND::TG4PrimaryVertex& vtx);
  bool GetTreesFromFile(const ND::TG4PrimaryVertex& vtx);
  bool GetTreesFromDirectory(const ND::TG4PrimaryVertex& vtx);

  void ClearInputTrees();
  std::string GetFileName(const ND::TG4PrimaryVertex& vtx);
  std::string GetTreeName(const ND::TG4PrimaryVertex& vtx);
  int GetEntryNum(const ND::TG4PrimaryVertex& vtx);
  std::string GetFileFromPath(const char* path);
  bool ComparePaths(const char* path1, const char* path2);
  void ResetFileInfo();
  void ResetVtxInfo();

  bool fPassThroughPresent;
};
#endif