MdcUtilitySvc.h

Go to the documentation of this file.

#ifndef MDCUTILITYSVC_H
#define MDCUTILITYSVC_H
 
#include "CLHEP/Geometry/Point3D.h"
#include "CLHEP/Geometry/Vector3D.h"
#include "CLHEP/Matrix/SymMatrix.h"
#include "CLHEP/Matrix/Vector.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/IInterface.h"
#include "GaudiKernel/Kernel.h"
#include "GaudiKernel/Service.h"
 
#include "Identifier/MdcID.h"
#include "MagneticFieldSvc/IBesMagFieldSvc.h"
#include "McTruth/McParticle.h"
#include "McTruth/MdcMcHit.h"
#include "MdcGeomSvc/IMdcGeomSvc.h"
#include "MdcRawEvent/MdcDigi.h"
#include "MdcRecEvent/RecMdcTrack.h"
#include "MdcUtilitySvc/IMdcUtilitySvc.h"
#include "RawDataProviderSvc/IRawDataProviderSvc.h"
#include "RawDataProviderSvc/MdcRawDataProvider.h"
#include "TrackUtil/Helix.h"
 
#include <map>
#include <vector>
 
// class MdcDigi;
 
#ifndef ENABLE_BACKWARDS_COMPATIBILITY
//  backwards compatibility will be enabled ONLY in CLHEP 1.9
typedef HepGeom::Point3D<double> HepPoint3D;
#endif
 
#ifndef ENABLE_BACKWARDS_COMPATIBILITY
//  backwards compatibility will be enabled ONLY in CLHEP 1.9
typedef HepGeom::Vector3D<double> HepVector3D;
#endif
 
class MdcUtilitySvc : public extends<Service, IMdcUtilitySvc> {
  // class MdcUtilitySvc: public Service, virtual public IMdcUtilitySvc{
public:
  MdcUtilitySvc( const std::string& name, ISvcLocator* svcloc );
  ~MdcUtilitySvc();
 
  virtual StatusCode initialize();
  virtual StatusCode finalize();
  // StatusCode queryInterface(const InterfaceID& riid, void** ppvUnknown);
 
  int nLayerTrackPassed( const HepVector helix ) const;
  int nLayerTrackPassed( const double helix[5] ) const;
 
  HepVector    patPar2BesPar( const HepVector& helixPar ) const;
  HepSymMatrix patErr2BesErr( const HepSymMatrix& err ) const;
  HepVector    besPar2PatPar( const HepVector& helixPar ) const;
  HepSymMatrix besErr2PatErr( const HepSymMatrix& err ) const;
 
  double doca( int layer, int cell, const HepVector helix, const HepSymMatrix errMat,
               bool passCellRequired = true, bool doSag = true ) const;
  double doca( int layer, int cell, HepPoint3D eastP, HepPoint3D westP,
               const HepVector helixBes, const HepSymMatrix errMatBes,
               bool passCellRequired = true, bool doSag = true ) const;
  double doca( int layer, int cell, const MdcSWire* sWire, const HepVector helixPat,
               const HepSymMatrix errMatPat, bool passCellRequired = true ) const;
  double docaPatPar( int layer, int cell, const HepVector helixPat,
                     const HepSymMatrix errMatPat, bool passCellRequired = true,
                     bool doSag = true ) const;
  double docaPatPar( int layer, int cell, HepPoint3D eastP, HepPoint3D westP,
                     const HepVector helixBes, const HepSymMatrix errMatBes,
                     bool passCellRequired = true, bool doSag = true ) const;
  double docaPatPar( int layer, int cell, const MdcSWire* sWire, const HepVector helixPat,
                     const HepSymMatrix errMatPat, bool passCellRequired = true ) const;
 
  HepPoint3D pointOnHelix( const HepVector helixPar, int lay, int innerOrOuter ) const;
  HepPoint3D pointOnHelixPatPar( const HepVector helixPat, int lay, int innerOrOuter ) const;
  bool       cellTrackPassedByPhi( const HepVector helix, int layer, int& cellId_in,
                                   int& cellId_out ) const;
  bool       cellTrackPassedByPhiPatPar( const HepVector helix, int layer, int& cellId_in,
                                         int& cellId_out ) const;
  bool       cellTrackPassed( const HepVector helix, int layer, int& cellId_in,
                              int& cellId_out ) const;
  bool       cellTrackPassedPatPar( const HepVector helix, int layer, int& cellId_in,
                                    int& cellId_out ) const;
 
  HepPoint3D Hel( HepPoint3D piv, double dr, double phi0, double Alpha_L, double kappa,
                  double dz, double dphi, double tanl ) const;
  double     p_cms( HepVector helix, int runNo, double mass ) const;
  Hep3Vector momentum( const RecMdcTrack* trk ) const;
  double     probab( const int& ndof, const double& chisq ) const;
  // std::vector<MdcDigi*>     getMdcDigiVec() const;
  MdcDigiVec getMdcDigiVec() const;
  // vector<vector<MdcDigi*> > ConnectionHitsGroup(int SameLRange = 1,int DiffLRange = 1)
  // const; vector<vector<MdcDigi*> > ConnectionHitsGroup(vector<MdcDigi*> &MdcHits,int
  // SameLRange = 1,int DiffLRange = 1) const;
 
  void  getHelixOfMcParticle( const Event::McParticle* mcParticle, Helix& helix );
  float getChargeOfMcParticle( const Event::McParticle* mcParticle );
  void  getMomPosOfMcParticle( const Event::McParticle* mcParticle, HepVector3D& pos,
                               HepVector3D& mom );
  void  getMdcDigiOnMcParticle(
       int                                                       trackIndex,
       const std::map<int, std::map<MdcDigi*, Event::MdcMcHit*>> mdcMCAssociation,
       MdcDigiVec& mdcDigiInput, MdcDigiVec& mdcDigiAssociated );
  /// Get association of MdcDigi and MdcMcHit according to track id
  void getMdcMCAssoiciation(
      int trackIndex, const std::vector<MdcDigi*> mdcDigiVecInput,
      std::map<int, std::map<MdcDigi*, Event::MdcMcHit*>>& mdcMCAssociation );
 
private:
  double               Bz() const { return m_pIMF->getReferField() * 1000.; }
  int                  myNWire[43];
  int                  myOuterWire[43][288][2];
  int                  myInnerWire[43][288][2];
  double               myWirePhi[43][288];
  double               dPhi( double phi1, double phi2 );
  IRawDataProviderSvc* m_RawDataProviderSvc;
  IMdcGeomSvc*         m_mdcGeomSvc;
  IBesMagFieldSvc*     m_pIMF;
  int                  m_debug;
  bool                 m_doSag;
};
#endif /* MDCUTILITYSVC_H*/