CaToolsWFExpression.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 CaToolsWFExpression_h
#define CaToolsWFExpression_h 1
 
#include "CaToolsDef.h"
#include "TCut.h"
#include "TString.h"
 
#include <cassert>
#include <tuple>
#include <vector>
 
class TTree;
class TH2F;
class TPad;
 
namespace cbm::ca::tools
{
  class WFExpression {
   public:
    static constexpr int kNpars = 1;  
 
    WFExpression() = delete;
 
    WFExpression(TTree* pTree, const char* exprDist, const char* exprParam);
 
    WFExpression(const WFExpression& other) = delete;
 
    WFExpression(WFExpression&& other) = delete;
 
    ~WFExpression() = default;
 
    WFExpression& operator=(const WFExpression& other) = delete;
 
    WFExpression& operator=(WFExpression&& other) = delete;
 
    std::tuple<std::vector<float>, std::vector<float>> CalculateParameters();
 
    void SetCut(const TCut& cut) { fCut = cut; }
 
    void SetEps(float eps);
 
    void SetNslices(int nSlices);
 
    void SetNbins(int nBinsDist, int nBinsParam);
 
    void SetPadBase(TPad* pad);
 
    void SetPadSlices(TPad* pad);
 
    void SetTitle(const char* title) { fsTitle = title; }
 
   private:
    // *****************************
    // ** Private class functions **
    // *****************************
 
    std::tuple<float, float, float> ProcessSlice(int iBinMin, int iBinMax);
 
    void GetConstWindowParams();
    // TODO: use other functions for other window shapes: GetParabWindowParams, GetEllipticWindowParams etc.
 
 
    // *********************
    // ** Class variables **
    // *********************
    TTree* fpTree    = nullptr;  
    TH2F* fpHistBase = nullptr;  
 
    TString fsExprDist  = "";  
    TString fsExprParam = "";  
 
    TCut fCut    = "";      
    int fNslices = 8;       
    float fEps   = 0.0005;  
 
 
    std::vector<float> fvUpSBoundaries = std::vector<float>(fNslices);  
    std::vector<float> fvLoSBoundaries = std::vector<float>(fNslices);  
    std::vector<float> fvSCenters      = std::vector<float>(fNslices);  
    std::vector<float> fvUpParams      = std::vector<float>(kNpars);    
    std::vector<float> fvLoParams      = std::vector<float>(kNpars);    
 
    // ----- Plotting options
    int fNbinsParam   = 400;      
    int fNbinsDist    = 400;      
    TString fsTitle   = "";       
    TString fsName    = "";       
    TPad* fpPadBase   = nullptr;  
    TPad* fpPadSlices = nullptr;  
  };
}  // namespace cbm::ca::tools
 
#endif  // CaToolsWFExpression_h