SplitCollectionByLayer.h

#ifndef SplitCollectionByLayer_h
#define SplitCollectionByLayer_h 1

#include "marlin/Processor.h"
#include "marlin/EventModifier.h"

#include "lcio.h"
#include <string>


using namespace lcio ;
using namespace marlin ;


class SplitCollectionByLayer : public Processor {
  
protected:

  struct OutColInfo{
    std::string name ;
    unsigned layer0 ;
    unsigned layer1 ;
    LCCollection* collection ;
  };
  
  enum HitType{
    SimTrackerHitType = 1,
    TrackerHitType,
    SimCalorimeterHitType,
    CalorimeterHitType, 
    UnkownType
  };

 public:
  
  virtual Processor*  newProcessor() { return new SplitCollectionByLayer ; }
  
  
  SplitCollectionByLayer() ;
  
  virtual const std::string & name() const { return Processor::name() ; }
 
  virtual void init() ;
  
  virtual void processRunHeader( LCRunHeader* run ) ;
  
  virtual void processEvent( LCEvent * evt ) ; 
  
  
  virtual void check( LCEvent * evt ) ; 
  
  
  virtual void end() ;
  

 protected:

  std::string _colName ;

  StringVec  _outColAndLayers ;

  std::vector<OutColInfo> _outCols ;

  HitType _type ;

  int _nRun ;
  int _nEvt ;
} ;

#endif