TOPtrack.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marko Staric                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <TVector3.h>
#include <framework/gearbox/Const.h>
 
namespace Belle2 {
  class Track;
  class ExtHit;
  class MCParticle;
  class TOPBarHit;
 
  namespace TOP {
 
    class TOPtrack {
    public:
      TOPtrack()
      {}
 
      TOPtrack(double x, double y, double z, double Px, double Py, double Pz,
               double Tlen, int Q, int pdg = 0);
 
 
      explicit TOPtrack(const Track* track,
                        const Const::ChargedStable& chargedStable = Const::pion);
 
      TOPtrack(const Track* track, int moduleID,
               const Const::ChargedStable& chargedStable = Const::pion);
 
      bool isValid() const {return m_valid;}
 
      const TVector3& getPosition() const {return m_position;}
 
      const TVector3& getMomentum() const {return m_momentum;}
 
      double getX() const {return m_position.X();}
 
      double getY() const {return m_position.Y();}
 
      double getZ() const {return m_position.Z();}
 
      double getPx() const {return m_momentum.X();}
 
      double getPy() const {return m_momentum.Y();}
 
      double getPz() const {return m_momentum.Z();}
 
      double getTrackLength() const {return m_trackLength;}
 
      double getTOF(double mass) const;
 
      double getTOF(const Const::ChargedStable& particle) const
      {
        return getTOF(particle.getMass());
      }
 
      void setTrackLength(double tof, double mass);
 
      void setTrackLength(double tof, const Const::ChargedStable& particle)
      {
        setTrackLength(tof, particle.getMass());
      }
 
      double getP() const {return m_momentum.Mag();}
 
      double getTheta() const {return m_momentum.Theta();}
 
      double getPhi() const {return m_momentum.Phi();}
 
      int getPDGcode() const {return m_pdg;}
 
      int getCharge() const {return m_charge;}
 
      int getHypID() const;
 
      int getModuleID() const {return m_moduleID;}
 
      bool atTop() const {return m_atTop;}
 
      const Track* getTrack() const {return m_track;}
 
      const ExtHit* getExtHit() const {return m_extHit;}
 
      const MCParticle* getMCParticle() const {return m_mcParticle;}
 
      const TOPBarHit* getBarHit() const {return m_barHit;}
 
      int toTop();
 
      void smear(double sig_x, double sig_z, double sig_theta, double sig_phi);
 
      void dump() const;
 
    private:
 
      int findModule();
 
      bool m_valid = false;       
      TVector3 m_position;        
      TVector3 m_momentum;        
      double m_trackLength = 0;   
      int m_charge = 0;           
      int m_pdg = 0;              
      bool m_atTop = false;       
      int m_moduleID = 0;         
      const Track* m_track = 0;            
      const ExtHit* m_extHit = 0;          
      const MCParticle* m_mcParticle = 0;  
      const TOPBarHit* m_barHit = 0;       
    };
 
  } // end TOP namespace
} // end Belle2 namespace