Geant4OutputAction.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_GEANT4OUTPUTACTION_H
#define DD4HEP_DDG4_GEANT4OUTPUTACTION_H

// Framework include files
#include "DDG4/Geant4EventAction.h"

// Forward declarations
class G4Run;
class G4Event;
class G4VHitsCollection;


namespace DD4hep {

  namespace Simulation {

    // Forward declarations
    class Geant4ParticleMap;


    class Geant4OutputAction: public Geant4EventAction {
    protected:
      template <typename T> class OutputContext {
      public:
        const T* context;
        void* userData;
        OutputContext(const T* c)
          : context(c), userData(0) {
        }
        template <typename U> U* data() const {
          return (U*) userData;
        }
      };

      std::string m_output;
      bool        m_errorFatal;
      Geant4ParticleMap* m_truth;
    public:
      Geant4OutputAction(Geant4Context* c, const std::string& nam);
      virtual ~Geant4OutputAction();
      virtual void configureFiber(Geant4Context* ctxt);

      virtual void begin(const G4Event* event);
      virtual void end(const G4Event* event);
      virtual void beginRun(const G4Run* run);
      virtual void endRun(const G4Run* run);

      virtual void saveRun(const G4Run* run);
      virtual void saveEvent(OutputContext<G4Event>& ctxt);
      virtual void saveCollection(OutputContext<G4Event>& ctxt, G4VHitsCollection* collection);
      virtual void commit(OutputContext<G4Event>& ctxt);
    };

  }    // End namespace Simulation
}      // End namespace DD4hep
#endif // DD4HEP_DDG4_GEANT4OUTPUTACTION_H