HitMatrix.h

Go to the documentation of this file.

#ifndef HitMatrix_h
#define HitMatrix_h

#include <iostream>
#include <vector>
#include <TH2.h>
#include <TCanvas.h>

using std::vector;
using std::cout;
using std::endl;

enum { PE, DR, CT, AP };
enum { X, Y, TIME, TYPE, AMPLITUDE, PROCESSED };

class HitMatrix : public vector<vector<double> >
{
        public:

                HitMatrix();
                ~HitMatrix();

                void            Init();                                                                 
                void            Clear( int type );                                                      
                int             AddHit( int x, int y, double time, int type );                          
                void            HitProcessed( unsigned int iHit );                                      
                vector<double>  GetHit( unsigned int iHit );                                            
                bool            GetProcessed( unsigned int iHit );                                      
                double          GetPreviousTime( unsigned int iHit );                                   
                int             FindHit( vector<double> aHit );                                         
                void            EraseHit( vector<double> aHit );                                        
                int             nHits( int type );                                                      
                int             nHits();                                                                
                TH2D*           DrawMatrix();                                                           
                void            PrintMatrix();                                                          
                void            SetAmplitude( unsigned int iHit, double amplitude );                    
                void            SetGate( double g, bool gCut=true ){ gate = g; gateCut = gCut; };
                void            SetGeometry( TH2I* geometry );                                          
        private:

                double          gate;
                bool            gateCut;
                TH2I            *h_geometry;
                int             Nx;
                int             Ny;
                vector<double>  hit;
                TH2D            *h_hits;
                TCanvas         *c_hitMatrix;
};

#endif