EventHandler.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_DDEVE_EVENTHANDLER_H
#define DD4HEP_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;
    bool m_hasEvent;
  public:
    EventHandler();
    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();
    virtual ~EventConsumer();
    virtual void OnNewEvent(EventHandler* /* handler */) {}
    virtual void OnFileOpen(EventHandler* /* handler */) {}

    ClassDef(EventConsumer,0);
  };

} /* End namespace DD4hep   */


#endif /* DD4HEP_DDEVE_EVENTHANDLER_H */