CutsAndMonitors.h

/*
 * Cuts.h
 *
 *  Created on: 04-05-2022
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
#ifndef HALCUTSANDMONITORS_HALCUTS_H_
#define HALCUTSANDMONITORS_HALCUTS_H_
 
#include "Cut.h"
#include "CutMonitor.h"
#include "CutMonitorRequest.h"
 
#include <TLorentzVector.h>
#include <TString.h>
 
 
class TObjString;
namespace Hal {
  class CutMonitorX;
  class CutMonitorYX;
  class CutMonitorXYZ;
 
  class CutsAndMonitors : public TObject {
    TObjArray* fCuts;
    TObjArray* fCutsOptions;
    TObjArray* fCutMonitors;
    TObjArray* fCutMonitorsOptions;
    TString fGlobalOptionCuts;
    TString fGlobalOptionsCutMonitors;
    std::vector<CutMonitorRequest> fCutMonitorRequests;
 
  protected:
    Cut* CutAt(Int_t i) const;
    CutMonitor* MonitorAt(Int_t i) const;
    void ClearCuts();
    void ClearMonitors();
    void AddRawCut(Cut* cut, TObjString* opt = nullptr);
    void AddRawCutMonitor(CutMonitor* mon, TObjString* opt = nullptr);
    void AddCutMonitorRequest(const CutMonAxisConf& x);
    void AddCutMonitorRequest(const CutMonAxisConf& x, const CutMonAxisConf& y);
    void AddCutMonitorRequest(const CutMonAxisConf& x, const CutMonAxisConf& y, const CutMonAxisConf& z);
    virtual void AddAllCutMonitorRequests(Option_t* opt) = 0;
    CutMonitor* MakeCutMonitor(Int_t request_no) const;
    Int_t GetCutNo(TString classname) const;
 
  public:
    CutsAndMonitors();
    CutsAndMonitors(const CutsAndMonitors& other);
    CutsAndMonitors& operator=(const CutsAndMonitors& other);
    void SetOptionForAllCuts(TString opt) { fGlobalOptionCuts = opt; };
    void SetOptionForAllMonitors(TString opt) { fGlobalOptionsCutMonitors = opt; }
    void AddCut(const Cut& cut, Option_t* opt = "");
    void AddCutMonitor(const CutMonitor& monitor, Option_t* opt = "");
    void SetCollectionID(Int_t id);
    const Cut* GetCut(Int_t i) const;
    TString GetCutOption(Int_t i) const;
    TString GetGlobalCutOption() { return fGlobalOptionCuts; };
    const CutMonitor* GetMonitor(Int_t i) const;
    TString GetGlobalCutMonitorOption() { return fGlobalOptionsCutMonitors; };
    TString GetCutMonitorOption(Int_t i) const;
    Int_t GetNCuts() const;
    Int_t GetNCutMonitors() const;
    Int_t GetNRequest() const { return fCutMonitorRequests.size(); }
    CutMonitorRequest& GetCutMonitorRequest(Int_t i) { return fCutMonitorRequests[i]; }
    void MakeCutMonitors(Option_t* opt = "");
    virtual ~CutsAndMonitors();
    ClassDef(CutsAndMonitors, 1)
  };
}  // namespace Hal
#endif /* HALCUTSANDMONITORS_HALCUTS_H_ */