CaToolsWindowFinder.h

Go to the documentation of this file.

/* Copyright (C) 2022 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergey Gorbunov, Sergei Zharko [committer] */
 
 
#ifndef CaToolsWindowFinder_h
#define CaToolsWindowFinder_h 1
 
#include "CaIteration.h"
#include "TCut.h"
#include "TObject.h"
 
#include <array>
#include <vector>
 
// TODO: Replace tmp asserts with exceptions
 
class TChain;
class TPad;
class TCanvas;
 
namespace cbm::algo::ca
{
  class SearchWindow;
}
 
namespace cbm::ca::tools
{
  namespace ca = cbm::algo::ca;
 
  enum EExpr
  {
    kDxVsX0,
    kDxVsY0,
    kDyVsX0,
    kDyVsY0,
    kEND
  };
 
  class WindowFinder : public TObject {
   public:
    // TODO: TEMPORARY CONSTANT EXPRESSIONS (TO BE MOVED TO A SEPARATE HEADER)
    static constexpr const char* kTreeName = "t";  
 
   public:
    WindowFinder();
 
    virtual ~WindowFinder() = default;
 
    WindowFinder(const WindowFinder&) = delete;
    WindowFinder(WindowFinder&&)      = delete;
    WindowFinder& operator=(const WindowFinder&) = delete;
    WindowFinder& operator=(WindowFinder&&) = delete;
 
    void AddInputFile(const char* filename);
 
    void DumpCanvasesToPdf(const char* filename = "WFLog") const;
 
    void Process(Option_t* opt = "");
 
    void ReadTrackingIterationsFromYAML(const char* filename);
 
    void SetBinning(int nBinsX, int nBinsY);
 
    void SetEpsilon(float eps);
 
    void SetExtraCut(const TCut& cut) { fExtraCut = cut; }
 
    void SetNslices(int nSlices);
 
    void SetOutputName(const char* filename) { fsOutputName = filename; }
 
    void SetStationIndexes(const std::vector<int>& vStationIndexes);
 
    void SetTarget(double x, double y, double z);
 
 
   private:
    ca::SearchWindow CreateSW(int iStation, const ca::Iteration& caIter);
 
    const char* GetDistExpr(EExpr expr) const;
 
    TCut GetTrackSelectionCut(int iStation, const ca::Iteration& caIter) const;
 
    void PrintCaseInformation(TPad* pPad, int iStation, const ca::Iteration& caIter) const;
 
 
    // *********************
    // ** Class variables **
    // *********************
    int fNparams = 1;  
 
    std::string fsOutputName = "SearchWindows.dat";  
 
    // ----- Input parameters (iterations and stations)
    std::vector<ca::Iteration> fvCaIters = {};   
    std::vector<int> fvStationIndexes    = {};   
    std::array<double, 3> fTargetPos     = {0};  
 
    TCut fExtraCut = TCut("");  
 
    // Window extraction settings
    int fNbinsX  = 400;    
    int fNbinsY  = 400;    
    int fNslices = 1;      
    float fEps   = 0.001;  
 
    TChain* fpMcTripletsTree = nullptr;  
 
    std::vector<TCanvas*> fvpCanvases;  // Vector of pointer to cavnases
 
    ClassDef(WindowFinder, 0);
  };
}  // namespace cbm::ca::tools
 
#endif  // CaToolsWindowsFinder_h