NtpTools.h

Go to the documentation of this file.

#ifndef NtpTools_h
#define NtpTools_h

#include <vector>

#include "TClonesArray.h"
#include "TCanvas.h"
#include "TH2F.h"
#include "TMarker.h"
#include "TMath.h"

#include "MessageService/MsgService.h"
#include "BeamDataNtuple/NtpBDLiteRecord.h"
#include "CandNtupleSR/NtpSREvent.h"
#include "CandNtupleSR/NtpSRTrack.h"
#include "CandNtupleSR/NtpSRShower.h"
#include "CandNtupleSR/NtpSRSlice.h"
#include "CandNtupleSR/NtpSRStrip.h"
#include "StandardNtuple/NtpStRecord.h"
#include "MCNtuple/NtpMCTruth.h"

#include "NtpConstants.h"
#include "RecoE.h"
#include "MyShwInfo.h"
//
//This class has a collection of general event handling tools that I use.
//

class Zfluk;

class NtpTools{
 public:

  typedef enum EBeamType{
    kAny = -1,  //use this to say "I dont care what beam type it is"
    kUnknown = 0,
    kLE10 = 1,
    kME = 2,
    kHE = 3,
    kpME = 4,
    pHE = 5
  }BeamType_t;


  NtpTools();
  ~NtpTools();
  static const NtpSREvent* GetEvent(const NtpStRecord* record, const int eventno);  
  static VldContext GetVldContext(const NtpStRecord* record);
  static Detector::Detector_t GetDetector(const NtpStRecord* record);
  

  static bool VtxIsContained(const NtpStRecord* record, const int eventno);  
  static bool IsInFidVolumeNear(const float x, const float y, const float z);
  static bool IsInFidVolumeFar(const float x, const float y, const float z);
  static bool PassBeamCuts(const BeamType_t beamtype, const NtpBDLiteRecord* beamntp, float& pot , bool isMC=false); 
  static float RadialDistFromBeam(const float x, const float y);


  static const NtpSRTrack* GetPrimaryTrack(const NtpStRecord* record, const int eventno);  
  static bool FillRecoEInfo(const NtpStRecord* record, const int eventno, RecoE* recoe);  
  static Int_t GetPittContainmentFlag(const NtpSREvent* evt);
  
  static bool FillMyShwInfo(const NtpStRecord* record, const int eventno, MyShwInfo* shwinfo);

  static bool GetTrueEnergy(const NtpStRecord* record, const int eventno, float& mceneu);  
  
  static const NtpMCTruth* GetTruth(const NtpStRecord* record, const int eventno);

  static int  GetNEventsPerSlice(const NtpStRecord* record, const int eventno);

  float GetMCEventWeight(const NtpStRecord* record, const int eventno);

  static float GetTotHadPt(const NtpStRecord* record, const NtpMCTruth* truth);

  //plotting stuff
  bool              MakeEventPict(const char* title, const NtpStRecord* record, const int eventno);
 private:
  Zfluk* fFluxModel;
   
  TCanvas* fCanvas;
  TH2F*  fUZAxis;
  TH2F*  fVZAxis;
  std::vector<TMarker> fUZMarkers;
  std::vector<TMarker> fVZMarkers;
  bool              ResetCanvas(Detector::Detector_t det);
  bool              PrintCanvas(const char* title);

};

inline const NtpSREvent* NtpTools::GetEvent(const NtpStRecord* record, const int eventno){
  return (dynamic_cast<const NtpSREvent*>(record->evt->At(eventno)));
}

inline const VldContext GetVldContext(const NtpStRecord* record){
  return(record->GetHeader().GetVldContext());
}

inline Detector::Detector_t NtpTools::GetDetector(const NtpStRecord* record){
  return(record->GetHeader().GetVldContext().GetDetector());
}

inline bool NtpTools::VtxIsContained(const NtpStRecord* record, const int eventno){
  if(!record) return false;
  const NtpSREvent* event = GetEvent(record, eventno);
  if(!event) return false;
  if(GetDetector(record)==Detector::kNear){
    return (IsInFidVolumeNear(event->vtx.x, event->vtx.y, event->vtx.z) );
  }else{
    return (IsInFidVolumeFar(event->vtx.x, event->vtx.y, event->vtx.z) );
  }
}

inline float NtpTools::RadialDistFromBeam(const float x, const float y){
  float radius = sqrt((x - NtpConstants::kNearBeamCentreX)*
                      (x - NtpConstants::kNearBeamCentreX) + 
                      (y - NtpConstants::kNearBeamCentreY)*
                      (y - NtpConstants::kNearBeamCentreY)  
                      ); 

  return radius;
  
}


inline bool NtpTools::IsInFidVolumeNear(const float x, const float y, const float z){  
  float radius = RadialDistFromBeam(x, y);
  if( z>1. &&
      z<5. &&
      radius<1. ) return true;
  else return false;
}

inline bool NtpTools::IsInFidVolumeFar(const float /*x*/, const float /*y*/, const float /*z*/){  
  return true;//<--
}



#endif

---