TOPreco.h

//*****************************************************************************
//-----------------------------------------------------------------------------
// Fast simulation and reconstruction package for TOP counter (F77 core)
// M. Staric, April-2009
//-----------------------------------------------------------------------------
//
// TOPreco.h
// C++ interface to F77 functions: reconstruction
//-----------------------------------------------------------------------------
//*****************************************************************************
 
#pragma once
 
#include <top/reconstruction/TOPtrack.h>
#include <top/dbobjects/TOPCalChannelMask.h>
#include <top/dbobjects/TOPCalChannelT0.h>
#include <top/dbobjects/TOPCalTimebase.h>
#include <framework/database/DBObjPtr.h>
#include <top/dataobjects/TOPAsicMask.h>
 
namespace Belle2 {
  namespace TOP {
 
    enum {c_Local = 0, c_Global = 1};
 
    class TOPreco {
    public:
      enum PDFoption {c_Rough = 0, c_Fine, c_Optimal};
 
      enum StoreOption {c_Reduced, c_Full};
 
      TOPreco(int NumHyp, double Masses[], int pdgCodes[],
              double BkgPerModule = 0, double ScaleN0 = 1);
 
      static void setChannelMask(const DBObjPtr<TOPCalChannelMask>& mask,
                                 const TOPAsicMask& asicMask);
 
      static void setUncalibratedChannelsOff(const DBObjPtr<TOPCalChannelT0>& channelT0);
 
      static void setUncalibratedChannelsOff(const DBObjPtr<TOPCalTimebase>& timebase);
 
      static void printChannelMask();
 
      static void setChannelEffi();
 
      void setMass(double mass, int pdg);
 
      void setTimeWindow(double Tmin, double Tmax);
 
      void getTimeWindow(double& Tmin, double& Tmax);
 
      void setPDFoption(PDFoption opt, int NP = 0, int NC = 0);
 
      void setStoreOption(StoreOption opt);
 
      void clearData();
 
      int addData(int moduleID, int pixelID, double time, double timeError);
 
      int getDataSize();
 
      void reconstruct(const TOPtrack& trk, int pdg = 0);
 
      int getFlag();
 
      double getExpectedPhotons(int i = 0);
 
      double getExpectedBG();
 
      int getNumOfPhotons();
 
      double getLogL(int i = 0);
 
      void getLogL(int Size, double LogL[], double ExpNphot[], int& Nphot);
 
      double getLogL(double timeShift, double timeMin, double timeMax, double sigma = 0.0);
 
      void getLogL(double timeShift, double timeMin, double timeMax, double sigma,
                   float* logL);
 
      void getTrackHit(int LocGlob, double R[3], double Dir[3], double& Len,
                       double& Tlen, double& Mom, int& moduleID);
 
      void dumpLogL(int NumHyp);
 
      void dumpTrackHit(int LocGlob);
 
      int getPullSize();
 
      void getPull(int K, int& pixelID, float& T, float& T0, float& Wid,
                   float& PhiCer, float& Wt);
 
      double getPDF(int pixelID, double t, double mass, int PDG, double jitter = 0);
 
      void setBeta(double beta);
 
      double getBeta() const {return m_beta;};
 
      void redoPDF(double mass, int PDG);
 
      int getNumofPDFPeaks(int pixelID) const;
 
      void getPDFPeak(int pixelID, int k,
                      float& position, float& width, float& numPhotons) const;
 
      float getBkgLevel(int pixelID) const;
 
      int getPDFPeakType(int pixelID, int k) const;
 
      float getPDFPeakFic(int pixelID, int k) const;
 
      float getPDFPeakE(int pixelID, int k) const;
 
      float getPDFPeakSigE(int pixelID, int k) const;
 
      int getPDFPeakNx(int pixelID, int k) const;
 
      int getPDFPeakNy(int pixelID, int k) const;
 
      int getPDFPeakNxm(int pixelID, int k) const;
 
      int getPDFPeakNym(int pixelID, int k) const;
 
      int getPDFPeakNxe(int pixelID, int k) const;
 
      int getPDFPeakNye(int pixelID, int k) const;
 
      float getPDFPeakXD(int pixelID, int k) const;
 
      float getPDFPeakYD(int pixelID, int k) const;
 
      float getPDFPeakKxe(int pixelID, int k) const;
 
      float getPDFPeakKye(int pixelID, int k) const;
 
      float getPDFPeakKze(int pixelID, int k) const;
 
      float getPDFPeakKxd(int pixelID, int k) const;
 
      float getPDFPeakKyd(int pixelID, int k) const;
 
      float getPDFPeakKzd(int pixelID, int k) const;
 
    private:
 
      void reconstruct(double X, double Y, double Z, double Tlen,
                       double Px, double Py, double Pz, int Q,
                       int pdg = 0, int moduleID = 0);
 
      int m_hypID = 0;    
      double m_beta = 0;  
    };
 
  } // TOP namespace
} // Belle2 namespace