CDCSensitiveDetector.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <simulation/kernel/SensitiveDetectorBase.h>
 
#include <cdc/dataobjects/CDCSimHit.h>
#include <cdc/dataobjects/WireID.h>
 
#include <mdst/dataobjects/MCParticle.h>
#include <framework/datastore/StoreArray.h>
 
#include <vector>
#include <map>
 
 
namespace Belle2 {
  class CDCSimHit;
  namespace CDC {
 
    class CDCGeometryPar;
 
 
    class CDCSensitiveDetector: public Simulation::SensitiveDetectorBase {
 
    public:
 
      CDCSensitiveDetector(G4String name, G4double thresholdEnergyDeposit, G4double thresholdKineticEnergy);
 
      ~CDCSensitiveDetector() {}
 
      void Initialize(G4HCofThisEvent*) override;
 
      bool step(G4Step* aStep, G4TouchableHistory* history) override;
 
      //      void BeginOfEvent(G4HCofThisEvent*);
 
      void EndOfEvent(G4HCofThisEvent*) override;
 
      void saveSimHit(const G4int layerId,
                      const G4int wireId,
                      const G4int trackID,
                      const G4int pid,
                      const G4double distance,
                      const G4double tof,
                      const G4double edep,
                      const G4double stepLength,
                      const G4ThreeVector& mom,
                      const G4ThreeVector& posW,
                      const G4ThreeVector& posIn,
                      const G4ThreeVector& posOut,
                      const G4ThreeVector& posTrack,
                      const G4int lr,
                      const G4int NewLrRaw,
                      const G4int NewLr,
                      const G4double speed,
                      const G4double hitWeight);
 
      //void AddbgOne(bool doit);
 
    private:
 
      G4double ClosestApproach(G4ThreeVector bwp, G4ThreeVector fwp, G4ThreeVector posIn, G4ThreeVector posOut,
                               G4ThreeVector& hitPosition, G4ThreeVector& wirePosition);//,G4double& transferT);
 
 
      void CellBound(const G4int layerId, const G4int ic1, const G4int ic2,
                     const G4double venter[6], const G4double vexit[6],
                     const G4double s1, const G4double s2,
                     G4double xint[6], G4double& sint, G4int& iflag);
 
 
 
      void GCUBS(const G4double x, const G4double y, const G4double d1, const G4double d2, G4double a[4]);
 
      void for_Rotat(const G4double bfld[3]);
 
      void Rotat(G4double& x, G4double& y,
                 G4double& z,
                 const int mode);
 
      void Rotat(G4double x[3], const int mode);
 
 
      void HELWIR(const G4double xwb4,
                  const G4double ywb4,
                  const G4double zwb4,
                  const G4double xwf4,
                  const G4double ywf4,
                  const G4double zwf4,
                  const G4double xp,
                  const G4double yp,
                  const G4double zp,
                  const G4double px,
                  const G4double py,
                  const G4double pz,
                  const G4double B_kG[3],
                  const G4double charge,
                  const G4int ntryMax,
                  G4double& distance,
                  G4double q2[3],
                  G4double q1[3],
                  G4double q3[3],
                  G4int& ntry);
 
 
      void Mvopr(const G4int ndim, const G4double b[3], const G4double m[3][3],
                 const G4double a[3], G4double c[3], const G4int mode);
 
      std::vector<int>  WireId_in_hit_order(int id0, int id1, int nWires);
 
      void setModifiedLeftRightFlag();
 
      void reAssignLeftRightInfo();
 
      unsigned short areNeighbors(const WireID& wireId, const WireID& otherWireId) const;
 
      unsigned short areNeighbors(unsigned short iCLayer, unsigned short iSuperLayer, unsigned short iLayer, unsigned short iWire,
                                  const WireID& otherWireId) const;
 
 
      StoreArray<MCParticle> m_MCParticles;
 
      StoreArray<CDCSimHit> m_CDCSimHits;
 
      G4int m_magneticField;
 
      G4int m_nonUniformField;
 
      G4double m_brot[3][3];  
      //      CDCGeometryPar& m_cdcgp; /**< Reference to CDCGeometryPar object. */
      CDCGeometryPar* m_cdcgp; 
      G4double m_thresholdEnergyDeposit;
 
      G4double m_thresholdKineticEnergy;
 
      G4bool m_wireSag; 
      G4bool m_modifiedLeftRightFlag; 
      G4double m_minTrackLength; 
      int m_hitNumber; 
      std::multimap<unsigned short, CDCSimHit*>
      m_hitWithPosWeight; 
      std::vector<CDCSimHit*> m_hitWithNegWeight; 
      //      std::vector<std::multimap<unsigned short, CDCSimHit*>::iterator> m_posWeightMapItBegin; /**< Iterator showing begin of positive hit map*/
 
      //      std::vector<std::multimap<unsigned short, CDCSimHit*>::iterator> m_posWeightMapItEnd; /**< Iterator showing end of positive hit map*/
 
      //      int m_nPosHits, m_nNegHits;
 
      const signed short CCW = 1; 
      const signed short CW  = -1; 
      const signed short CW_OUT_NEIGHBOR  = 1; 
      const signed short CW_NEIGHBOR      = 3; 
      const signed short CW_IN_NEIGHBOR   = 5; 
      const signed short CCW_IN_NEIGHBOR  = 7; 
      const signed short CCW_NEIGHBOR     = 9; 
      const signed short CCW_OUT_NEIGHBOR = 11; 
    };
  }
} // end of namespace Belle2