Geant4InputAction.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_DDG4_GEANT4INPUTACTION_H
#define DD4HEP_DDG4_GEANT4INPUTACTION_H

// Framework include files
#include "DDG4/Geant4Vertex.h"
#include "DDG4/Geant4Particle.h"
#include "DDG4/Geant4GeneratorAction.h"

// C/C++ include files
#include <vector>
#include <memory>

// Forward declarations
class G4Event;

namespace DD4hep  {

  namespace Simulation  {


    class Geant4EventReader  {

    public:
      typedef Geant4Vertex   Vertex;
      typedef Geant4Particle Particle;
      typedef std::vector<Particle*> Particles;
      enum EventReaderStatus {
        EVENT_READER_ERROR=0,
        EVENT_READER_OK=1,
        EVENT_READER_NO_DIRECT=2,
        EVENT_READER_NO_PRIMARIES=4,
        EVENT_READER_NO_FACTORY=6,
        EVENT_READER_IO_ERROR=8
      };
    protected:
      std::string m_name;
      bool m_directAccess;
      int  m_currEvent;
    public:
      Geant4EventReader(const std::string& nam);
      virtual ~Geant4EventReader();
      const std::string& name()  const   {  return m_name;   }
      bool hasDirectAccess() const  {  return m_directAccess; }
      int currentEventNumber() const { return m_currEvent; }

      virtual EventReaderStatus moveToEvent(int event_number);
      virtual EventReaderStatus skipEvent();

      virtual EventReaderStatus readParticles(int event_number, 
                                              Vertex& primary_vertex,
                                              Particles& particles) = 0;
    };


    class Geant4InputAction : public Geant4GeneratorAction {

    public:
      typedef Geant4Vertex   Vertex;
      typedef Geant4Particle Particle;
      typedef std::vector<Particle*> Particles;
    protected:
      std::string         m_input;
      int                 m_firstEvent;
      int                 m_mask;
      double              m_momScale;
      Geant4EventReader*  m_reader;
      int m_currentEventNumber;
      bool m_abort;

    public:
      int readParticles(int event_number, Vertex& primary_vertex, Particles& particles);
      std::string issue(int i) const;

      Geant4InputAction(Geant4Context* context, const std::string& name);
      virtual ~Geant4InputAction();
      Particles* new_particles() const { return new Particles; }
      virtual void operator()(G4Event* event);
    };
  }     /* End namespace Simulation   */
}       /* End namespace DD4hep */
#endif  /* DD4HEP_DDG4_GEANT4INPUTACTION_H  */