CutContainer.h

/*
 * CutContainer.h
 *
 *  Created on: 06-08-2013
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
 
#ifndef HALCUTCONTAINER_H_
#define HALCUTCONTAINER_H_
 
#include <TObjArray.h>
#include <TString.h>
 
#include "Cut.h"
#include "CutCollection.h"
#include "CutMonitor.h"
#include "Std.h"
 
//#define SHOW_CUT_INIT
namespace Hal {
  class CutContainer : public TObject {
    Bool_t fInit;
    Int_t fSize;
    // TObjArray *fTempCutContainers;
    TObjArray** fTempCutMonitors;  //[fSize]
    TObjArray** fCutContainers;    //[fSize]
    void VerifyOrder(TObjArray* obj);
    void ExtractComplexMonitor(CutMonitor* mon, TString& opt);
    void MakeComplexAxis(CutMonitor* mon, Int_t axis, Int_t opt);
    Bool_t ExtractRegExp(const CutMonitor& cut, Option_t* opt);
    Bool_t ExtractRegExp2(const CutMonitor& cut, Option_t* opt);
    Bool_t CheckTwoTracksOptions(const CutMonitor& cutmon, Option_t* opt);
    Bool_t CheckTwoTracksOptions(const Cut& cutmon, Option_t* opt);
 
  protected:
    inline TObjArray* GetCutContainer(ECutUpdate upd) const { return (TObjArray*) fCutContainers[static_cast<Int_t>(upd)]; };
 
  public:
    enum class ELinkPolicy { kOneToMany, kEqual, kReplicateLast, kReplicateFirst, kAnyToAny };
    CutContainer();
    CutContainer(const CutContainer& cont);
    CutContainer(ECutUpdate tiers_no);
    void Init(const Int_t task_id = 0);
    void AddCut(const Cut& cut, Option_t* opt = " ");
    void InitReport() const;
    void AddMonitor(const CutMonitor& monitor, Option_t* opt = " ");
    void LinkCollections(ECutUpdate opt_low, Int_t in_low, ECutUpdate opt_high, Int_t in_high);
    void ReplicateCollection(ECutUpdate type, Int_t collection_no, Int_t new_collection_no, Option_t* option = "");
    void RemoveCollection(ECutUpdate update, Int_t collection);
    void MakeDummyCopies(ECutUpdate update, CutContainer* other, Bool_t copy_link);
    void AddVirtualCut(ECutUpdate update, Int_t col);
    Int_t GetCollectionsNo(ECutUpdate update) const;
    Int_t GetEventCollectionsNo() const;
    Int_t GetTrackCollectionsNo() const;
    Int_t GetTwoTrackCollectionsNo() const;
    Int_t GetTwoTrackCollectionsBackgroundNo() const;
    Bool_t LinkCollections(ECutUpdate first, ECutUpdate last, ELinkPolicy policy);
    inline CutCollection* GetEventCollection(Int_t collection) const {
      return (CutCollection*) fCutContainers[static_cast<Int_t>(ECutUpdate::kEvent)]->UncheckedAt(collection);
      ;
    };
    inline CutCollection* GetTrackCollection(Int_t collection) const {
      return (CutCollection*) fCutContainers[static_cast<Int_t>(ECutUpdate::kTrack)]->UncheckedAt(collection);
    };
    inline CutCollection* GetTwoTrackCollection(Int_t collection) const {
      return (CutCollection*) fCutContainers[static_cast<Int_t>(ECutUpdate::kTwoTrack)]->UncheckedAt(collection);
    };
    inline CutCollection* GetTwoTrackBackgroundCollection(Int_t collection) const {
      return (CutCollection*) fCutContainers[static_cast<Int_t>(ECutUpdate::kTwoTrackBackground)]->UncheckedAt(collection);
    };
    inline Int_t GetSize() const { return fSize; };
    inline Bool_t PassEvent(Event* event, const Int_t collection) {
      return ((CutCollection*) fCutContainers[static_cast<Int_t>(ECutUpdate::kEvent)]->UncheckedAt(collection))->PassEvent(event);
    }
    inline Bool_t PassTrack(Track* track, const Int_t collection) {
      return ((CutCollection*) fCutContainers[static_cast<Int_t>(ECutUpdate::kTrack)]->UncheckedAt(collection))->PassTrack(track);
    };
    inline Bool_t PassSignalPair(TwoTrack* pair, const Int_t collection) {
      return ((CutCollection*) fCutContainers[static_cast<Int_t>(ECutUpdate::kTwoTrack)]->UncheckedAt(collection))
        ->PassTwoTrack(pair);
    };
    inline Bool_t PassBackgroundPair(TwoTrack* pair, const Int_t collection) {
      return ((CutCollection*) fCutContainers[static_cast<Int_t>(ECutUpdate::kTwoTrackBackground)]->UncheckedAt(collection))
        ->PassTwoTrack(pair);
    };
    virtual Package* Report() const;
    virtual ~CutContainer();
    ClassDef(CutContainer, 1)
  };
}  // namespace Hal
#endif /* HALCUTCONTAINER_H_ */