code/include/NewWave/RasterisedEvent.hh

Go to the documentation of this file.

#ifndef NEWWAVE_RASTERISED_EVENT_HH
#define NEWWAVE_RASTERISED_EVENT_HH

#include "NewWave/PixelDefinition.hh"
#include "NewWave/MomentumHelpers.hh"

namespace HepMC{
  class GenEvent;
}

namespace NewWave {
  
  using std::size_t;
  
  template<typename T>
  class RasterisedEvent{
    
  public:

    RasterisedEvent(const T &inputParticles,
                    const PixelDefinition &pixelDefn):
    _pixelDefn(pixelDefn),
    _pixels(pixelDefn.makeEmptyPixelArray()),
    _input(T()){
      
      for(auto p: inputParticles){
        if(_pixelDefn.covers(rapidity(p), phi(p) ) ){
          addParticle(p);
        }
      }
    }
    

    RasterisedEvent(const PixelArray &pixelArray,
                    const PixelDefinition &pixelDefn):
            _pixelDefn(pixelDefn),
            _pixels(pixelArray){}
    
    const T &inputParticles()const{return _input;}
    
    

    const PixelArray &pixels()const{return _pixels;}
    

    const PixelDefinition &pixelDefinition()const{return _pixelDefn;}
    
  private:
    
    PixelDefinition _pixelDefn;
    PixelArray _pixels;
    
    T _input;
    
//    void fillFromHepMC(const HepMC::GenEvent *event);
    
    void addParticle(double rapidity, double phi, double pT){
      size_t ybin   = _pixelDefn.yPixelIndex(rapidity);
      size_t phiBin = _pixelDefn.phiPixelIndex(phi);
      _pixels[ybin][phiBin] += pT;
    }
    
    template<typename P>
    void addParticle(const P &particle){
      _input.push_back(particle);
      addParticle(rapidity(particle),
                  phi(particle),
                  pT(particle));
    }
    
  };
  

  template<>
  RasterisedEvent<HepMC::GenEvent*>::RasterisedEvent(HepMC::GenEvent* const &event,
                                                     const PixelDefinition &pixelDefn);
  
}

#endif