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 PolynomialSearchWindow;
}
 
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::PolynomialSearchWindow 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