PixelMap.h

Go to the documentation of this file.

#ifndef LCVN_PIXELMAP_H
#define LCVN_PIXELMAP_H

#include <ostream>
#include <vector>

#include "TH2F.h"
#include "larrecodnn/CVN/func/Boundary.h"
#include "larrecodnn/CVN/func/HitType.h"

namespace lcvn {

  class PixelMap {
  public:
    PixelMap(unsigned int nWire, unsigned int nTdc, const Boundary& bound);
    PixelMap() { fTotHits = 0; };

    unsigned int NWire() const { return fNWire; };

    unsigned int NTdc() const { return fNTdc; };

    unsigned int NPixel() const { return fPE.size(); };

    Boundary Bound() const { return fBound; };

    unsigned int NInput() const { return NPixel(); };

    void FillInputVector(float* input) const;

    void Add(const unsigned int& wire,
             const double& tdc,
             const unsigned int& view,
             const double& pe);

    unsigned int GlobalToIndex(const unsigned int& wire,
                               const double& tdc,
                               const unsigned int& view);

    unsigned int LocalToIndex(const unsigned int& wire, const unsigned int& tdc) const;

    unsigned int GlobalToIndexSingle(const unsigned int& wire,
                                     const double& tdc,
                                     const unsigned int& view);

    void SetTotHits(unsigned int tothits) { fTotHits = tothits; }
    unsigned int GetTotHits() { return fTotHits; }
    void Print() const;

    TH2F* ToTH2() const;
    TH2F* ToLabTH2() const;
    TH2F* SingleViewToTH2(const unsigned int& view) const;

    std::vector<float> fPE;     
    std::vector<float> fPEX;    
    std::vector<float> fPEY;    
    std::vector<float> fPEZ;    
    std::vector<double> fPur;   
    std::vector<double> fPurX;  
    std::vector<double> fPurY;  
    std::vector<double> fPurZ;  
    std::vector<HitType> fLab;  
    std::vector<HitType> fLabX; 
    std::vector<HitType> fLabY; 
    std::vector<HitType> fLabZ; 

  private:
    unsigned int fNWire;   
    unsigned int fNTdc;    
    unsigned int fTotHits; 

    Boundary fBound; //< Boundary of pixel map
  };

  std::ostream& operator<<(std::ostream& os, const PixelMap& m);

}

#endif // CVN_PIXELMAP_H