Geant4Hits.h

Go to the documentation of this file.

// $Id$
//==========================================================================
//  AIDA Detector description implementation for LCD
//--------------------------------------------------------------------------
// Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
// All rights reserved.
//
// For the licensing terms see $DD4hepINSTALL/LICENSE.
// For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
//
// Author     : M.Frank
//
//==========================================================================

#ifndef DD4HEP_GEANT4HITS_H
#define DD4HEP_GEANT4HITS_H

// Framework include files
#include "DD4hep/Objects.h"
#include "DDG4/Geant4StepHandler.h"

// Geant4 include files
#include "G4VHit.hh"
#include "G4Step.hh"
#include "G4StepPoint.hh"

namespace DD4hep {

  namespace Simulation {

    // Forward declarations;
    class Geant4Hit;
    class Geant4TrackerHit;
    class Geant4CalorimeterHit;


    template <class HIT> class HitCompare {
    public:
      virtual ~HitCompare() {}
      virtual bool operator()(const HIT* h) const = 0;
    };


    template <class HIT> struct HitPositionCompare: public HitCompare<HIT> {
      const Position& pos;
      HitPositionCompare(const Position& p) : pos(p) {}
      virtual ~HitPositionCompare() {}
      virtual bool operator()(const HIT* h) const {
        return pos == h->position;
      }
    };


    class Geant4Hit: public G4VHit {
    public:

      // cellID
      unsigned long cellID = 0;

      struct MonteCarloContrib {
        int trackID = -1;
        int pdgID = -1;
        double deposit = 0.0;
        double time = 0.0;
        MonteCarloContrib() = default;
        MonteCarloContrib(const MonteCarloContrib& c) = default;
        MonteCarloContrib(int track_id, double dep, double time_stamp)
          : trackID(track_id), pdgID(-1), deposit(dep), time(time_stamp) {}
        MonteCarloContrib(int track_id, int pdg, double dep, double time_stamp)
          : trackID(track_id), pdgID(pdg), deposit(dep), time(time_stamp) {}
        MonteCarloContrib& operator=(const MonteCarloContrib& c) = default;
        void clear() {
          time = deposit = 0.0;
          pdgID = trackID = -1;
        }
      };
      typedef MonteCarloContrib Contribution;
      typedef std::vector<MonteCarloContrib> Contributions;

    public:
      Geant4Hit() = default;
      virtual ~Geant4Hit() { }
      static bool isGeantino(G4Track* track);
      static Contribution extractContribution(G4Step* step);
    };


    class Geant4TrackerHit: public Geant4Hit {
    public:
      Position position;
      Direction momentum;
      double length;
      Contribution truth;

      double energyDeposit;

    public:
      Geant4TrackerHit();
      Geant4TrackerHit(int track_id, int pdg_id, double deposit, double time_stamp);
      virtual ~Geant4TrackerHit() {}
      Geant4TrackerHit& clear();
      Geant4TrackerHit& storePoint(G4Step* step, G4StepPoint* point);

      Geant4TrackerHit& operator=(const Geant4TrackerHit& c);
      void *operator new(size_t);
      void operator delete(void *ptr);
    };


    class Geant4CalorimeterHit: public Geant4Hit {
    public:
      Position position;
      Contributions truth;
      double energyDeposit;
    public:
      Geant4CalorimeterHit(const Position& cell_pos);
      virtual ~Geant4CalorimeterHit() { }
      void *operator new(size_t);
      void operator delete(void *ptr);
    };

  }    // End namespace Simulation
}      // End namespace DD4hep

#endif // DD4HEP_GEANT4HITS_H