DDG4EventHandler.h

Go to the documentation of this file.

//==========================================================================
//  AIDA Detector description implementation 
//--------------------------------------------------------------------------
// 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 DDEVE_DDG4EVENTHANDLER_H
#define DDEVE_DDG4EVENTHANDLER_H
 
// Framework include files
#include "DDEve/EventHandler.h"
 
// C/C++ include files
#include <map>
#include <string>
 
// Forward declarations
class TTree;
class TFile;
class TBranch;
 
namespace dd4hep {
 
  /* I/O handler for generic ROOT files produced by the DDG4 ROOT output stream
   * 
   * \author  M.Frank
   * \version 1.0
   * \ingroup DD4HEP_EVE
   */
  class DDG4EventHandler : public EventHandler  {
  public:
    typedef std::map<std::string,std::pair<TBranch*,void*> > Branches;
    typedef void* (*ParticleAccessor_t)(void*, DDEveParticle*);
    typedef void* (*HitAccessor_t)(void*, DDEveHit*);
  protected:
    std::pair<TFile*,TTree*> m_file;
    Branches m_branches;
    Long64_t m_entry;
    HitAccessor_t m_simhitConverter;
    ParticleAccessor_t m_particleConverter;
    TypedEventCollections m_data;
  public:
    DDG4EventHandler();
    virtual ~DDG4EventHandler();
 
    virtual const TypedEventCollections& data()  const    override  { return m_data;  }
    virtual long numEvents() const  override;
    std::string datasourceName() const  override;
    virtual CollectionType collectionType(const std::string& collection) const  override;
    virtual size_t collectionLoop(const std::string& collection, DDEveHitActor& actor)  override;
    virtual size_t collectionLoop(const std::string& collection, DDEveParticleActor& actor)  override;
    virtual bool Open(const std::string& type, const std::string& file_name)  override;
    virtual bool NextEvent()  override;
    virtual bool PreviousEvent()  override;
    virtual bool GotoEvent(long event_number)  override;
    Int_t ReadEvent(Long64_t n);
 
    ClassDefOverride(DDG4EventHandler,0);
  };
 
}      /* End namespace dd4hep            */
#endif // DDEVE_DDG4EVENTHANDLER_H