BmnNdetClusterFinder.h

Go to the documentation of this file.

 
#ifndef BMNNDETCLUSTERFINDER_H
#define BMNNDETCLUSTERFINDER_H
 
#include "BmnNdetCluster.h"
#include "TObject.h"
 
#include <vector>
 
using namespace std;
 
class BmnNdetClusterFinder : public TObject
{
  public:
    enum eCellStatus
    {
        kIsNotUsed = 0, 
        kIsUsed         
    };
 
    enum eClusterStatus
    {
        kNoCluster = -1, 
        kUnchecked = 0,  
        kIsNeutron,      
        kNotNeutron      
    };
 
  private:
    vector<BmnNdetCluster*> fClusters;
    vector<eClusterStatus> fClusterStatus;
 
    Int_t fNCells;
    Int_t fNClusters;
    Int_t fNUncheckedCellsInEvent;
 
    vector<Int_t> fIndexTime;
    vector<Int_t> fIndexEdep;
    vector<Int_t> fIndexBeta;
 
    vector<eCellStatus> fCellStatus;
 
    vector<Int_t> fCell;
    vector<Int_t> fColumn;
    vector<Int_t> fRow;
    vector<Int_t> fLayer;
    vector<Double_t> fXcoord;
    vector<Double_t> fYcoord;
    vector<Double_t> fZcoord;
    vector<Double_t> fTime;
    vector<Double_t> fEdep;
    vector<Double_t> fBeta;
 
    vector<vector<vector<Int_t>>> fIndexXYZ;
 
  public:
    BmnNdetClusterFinder(const Int_t kNColumns = 11, const Int_t kNRows = 11, const Int_t kNLayers = 16);
    ~BmnNdetClusterFinder();
 
    Int_t GetNClusters();
 
    Int_t GetNCells();
 
    Int_t GetNCells(Double_t threshold);
 
    Int_t GetNColumns();
 
    Int_t GetNRows();
 
    Int_t GetNLayers();
 
    vector<Int_t> GetCells();
 
    vector<Int_t> GetColumns();
 
    vector<Int_t> GetRows();
 
    vector<Int_t> GetLayers();
 
    vector<Double_t> GetX();
 
    vector<Double_t> GetY();
 
    vector<Double_t> GetZ();
 
    vector<Double_t> GetTime();
 
    vector<Double_t> GetEdep();
 
    vector<Double_t> GetBeta();
 
    Int_t GetClusterStatus(Int_t i_cluster);
 
    void SetClusterStatus(Int_t i_cluster, eClusterStatus status);
 
    void Fill(Int_t cell,
              Int_t column,
              Int_t row,
              Int_t layer,
              Double_t x,
              Double_t y,
              Double_t z,
              Double_t t,
              Double_t e_dep);
 
    void Clear();
 
    void Print();
 
    BmnNdetCluster* GetCluster(Int_t i_cluster);
 
    BmnNdetCluster* AddCluster();
 
    void AddCellToCluster(Int_t i_cell, Int_t i_cluster);
 
    Int_t FindClusters1();
 
    Int_t FindClusters1_2();
 
    Int_t FindClusters2();
 
    Int_t FindClusters2_2();
 
    Int_t FindClusters3(Double_t cut_beta = 0.03);
 
    Int_t FindClusters4(Double_t cut_beta = 0.01);
 
    Int_t SelectNeutrons(Double_t cut_angle = 45., Int_t cut_ncells = 3, Bool_t check_border = kFALSE);
 
    Int_t MergeClusters(Int_t i_cluster_1, Int_t i_cluster_2);
 
    Int_t CalculateNumberOfNeutrons(Double_t weight_edep = 1.,
                                    Double_t weight_ncells = 0.01,
                                    Double_t weight_nclusters = 0.05,
                                    Double_t threshold12 = 0.);
 
    virtual const char* GetClassName() { return "BmnNdetClusterFinder"; }
    ClassDef(BmnNdetClusterFinder, 1)
};
 
#endif /* BMNNDETCLUSTERFINDER_H */