EventHandler.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_EVENTHANDLER_H
#define DDEVE_EVENTHANDLER_H
 
// Framework include files
#include "DDEve/DDEveEventData.h"
#include "TClass.h"
class TH2F;
 
// C/C++ include files
#include <set>
#include <map>
#include <vector>
 
// Forward declarations
#define MEV_2_GEV 0.001
 
/*
 *   dd4hep namespace declaration
 */
namespace dd4hep {
 
  // Forward declarations
  class EventConsumer;
  
 
  struct DDEveHitActor  {
    virtual ~DDEveHitActor() {}
    virtual void operator()(const DDEveHit&) = 0;
    virtual void setSize(size_t /* num_elements */) {}
  };
 
 
  struct DDEveParticleActor  {
    virtual ~DDEveParticleActor() {}
    virtual void operator()(const DDEveParticle&) = 0;
    virtual void setSize(size_t /* num_elements */) {}
  };
 
 
  class EventHandler   {
  public:
    enum CollectionType { 
      NO_COLLECTION=1<<0, 
      PARTICLE_COLLECTION=1<<1, 
      TRACK_COLLECTION=1<<2,
      CALO_HIT_COLLECTION=1<<3,
      TRACKER_HIT_COLLECTION=1<<4,
      HIT_COLLECTION=CALO_HIT_COLLECTION|TRACKER_HIT_COLLECTION
    };
    typedef std::pair<const char*,size_t> Collection;
    typedef std::map<std::string,std::vector<Collection> > TypedEventCollections;
 
  protected:
    bool m_hasFile = false;
    bool m_hasEvent = false;
  public:
    EventHandler() = default;
    virtual ~EventHandler();
    virtual bool hasEvent() const  { return m_hasEvent; }
    virtual bool hasFile() const   { return m_hasFile;  }
    virtual const TypedEventCollections& data()  const = 0;
    virtual long numEvents() const = 0;
    virtual std::string datasourceName() const = 0;
    virtual size_t collectionLoop(const std::string& collection, DDEveHitActor& actor) = 0;
    virtual size_t collectionLoop(const std::string& collection, DDEveParticleActor& actor) = 0;
    virtual CollectionType collectionType(const std::string& collection) const = 0;
    virtual bool Open(const std::string& type, const std::string& file_name) = 0;
    virtual bool NextEvent() = 0;
    virtual bool PreviousEvent() = 0;
    virtual bool GotoEvent(long event_number) = 0;
 
    ClassDef(EventHandler,0);
  };
 
 
  class EventConsumer   {
  public:
    EventConsumer() = default;
    virtual ~EventConsumer();
    virtual void OnNewEvent(EventHandler& /* handler */) = 0;
    virtual void OnFileOpen(EventHandler& /* handler */) = 0;
 
    ClassDef(EventConsumer,0);
  };
}      /* End namespace dd4hep        */
#endif // DDEVE_EVENTHANDLER_H