CutCollection.h

/*
 * CutCollection.h
 *
 *  Created on: 04-05-2022
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
 
#ifndef HALCUTSUBCONTAINER_H_
#define HALCUTSUBCONTAINER_H_
 
#include "Cut.h"
 
#include "Array.h"
 
#include <TObjArray.h>
#include <TString.h>
#include <vector>
 
#include "Std.h"
 
namespace Hal {
  class CutMonitor;
  class Event;
  class Track;
  class TwoTrack;
 
  class CutCollectionLinks : public TObject {
    std::vector<int> fLinks;
    Bool_t fUsed = {kFALSE};
 
  public:
    CutCollectionLinks() {};
    void SafeInit();
    void AddAddr(Int_t addr);
    inline Int_t GetAddr(Int_t index) const { return fLinks[index]; }
    inline Int_t GetSize() const { return fLinks.size(); }
    void MakeUsed() { fUsed = kTRUE; }
    Bool_t IsUsed() const { return fUsed; };
    virtual ~CutCollectionLinks() {};
    ClassDef(CutCollectionLinks, 1)
  };
 
  class CutCollection : public TObject {
  private:
    TObjArray* fCutMonitors;
    TObjArray *fCuts, *fFastCuts;
    ULong64_t fPassedSlow, fFailedSlow, fPassedFast, fFailedFast;
    Int_t fFastCutsNo;
    Int_t fSlowCutsNo;
    Int_t fCutMonitorsNo;
    CutCollectionLinks fPrev, fNext, fPrevBckg, fNextBckg;
    ECutUpdate fMode;  // mode event/ track. two track
    Bool_t fInit, fDummy;
    Int_t fCollectionID, fContainerSize, fStep;
    TObjArray** fCutContainerArr;  //[fContainerSize]
    void Update(Bool_t val, TObject* obj);
    void AdvancedMonitorInitialization(Int_t task_id);
    Int_t MatchCut(Cut* address, CutCollection* input);
    TObjArray* LocateInLowerCollections(TString cut);
    TObjArray* LocateCuts(TString classname);
    Bool_t CheckIfComptatiblie(Cut* cut1, Cut* cut2);
    ECutUpdate GetUpdateFromName(TString cutname) const;
    inline TObjArray* GetCutContainerArray(ECutUpdate upd) const {
      return fCutContainerArr[static_cast<Int_t>(upd)];
      ;
    }
 
  public:
    CutCollection(TObjArray** container, Int_t cont_size, ECutUpdate update, Int_t collectionNo);
    CutCollection();
    CutCollection* MakeNewCopy(TObjArray** container) const;
    void Init(Int_t task_id);
    void PrintInfo() const;
    void SetStep(Int_t step) { fStep = step; };
    void AddCut(Cut* cut, Option_t* opt = " ");
    void MarkAsDummy();
    void AddCutMonitor(CutMonitor* monitor);
    void AddNextAddr(Int_t value, Bool_t background = kFALSE);
    void AddPreviousAddr(Int_t value, Bool_t background = kFALSE);
    void MakeDummyCopy(const CutCollection* copy, Bool_t copy_high_links = kTRUE);
    Int_t Compare(const TObject* obj) const;
    ULong64_t GetFailed(Option_t* opt = "") const;
    ULong64_t GetPassed(Option_t* opt = "") const;
    Int_t GetCutNo() const { return fCuts->GetEntriesFast(); };
    Int_t GetCollectionID() const { return fCollectionID; };
    Int_t GetFastCutNo() const { return fFastCuts->GetEntriesFast(); };
    Int_t GetCutMonitorNo() const { return fCutMonitors->GetEntriesFast(); };
    Bool_t PassEvent(Event* obj);
    Bool_t PassTrack(Track* obj);
    Bool_t PassTwoTrack(TwoTrack* pair);
    Bool_t IsSortable() const { return kTRUE; };
    Bool_t IsDummy() const { return fDummy; };
    Cut* GetCut(Int_t index) const { return (Cut*) fCuts->UncheckedAt(index); };
    Cut* GetFastCut(Int_t index) const { return (Cut*) fFastCuts->UncheckedAt(index); };
    Cut* FindCut(TString name) const;
    CutMonitor* GetCutMonitor(Int_t index) const;
    CutCollection* Replicate(Int_t new_collection) const;
    inline Int_t GetPrevNo() const { return fPrev.GetSize(); };
    inline Int_t GetNextNo() const { return fNext.GetSize(); };
    inline Int_t GetPrevAddr(Int_t index) const { return fPrev.GetAddr(index); };
    inline Int_t GetNextAddr(Int_t index) const { return fNext.GetAddr(index); };
    inline Int_t GetPrevNoBackground() const { return fPrevBckg.GetSize(); };
    inline Int_t GetNextNoBackround() const { return fNextBckg.GetSize(); };
    inline Int_t GetPrevAddBackround(Int_t index) const { return fPrevBckg.GetAddr(index); };
    inline Int_t GetNextAddrBackround(Int_t index) const { return fNextBckg.GetAddr(index); };
    virtual Package* Report() const;
    virtual ~CutCollection();
    ClassDef(CutCollection, 1)
  };
}  // namespace Hal
#endif /* HALCUTSUBCONTAINER_H_ */