TwoTrackAna.h

/*
 * TwoTrackAna.h
 *
 *  Created on: 27-08-2013
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
 
#ifndef HALTWOTRACKANA_H_
#define HALTWOTRACKANA_H_
 
#include "MultiTrackAna.h"
 
#include <Rtypes.h>
#include <RtypesCore.h>
#include <TString.h>
 
#include "TwoTrack.h"
 
//#define DISABLE_TWO_TRACK_SWAPPING
namespace Hal {
  class TwoTrackAnaChain;
  class TwoTrackAna : public MultiTrackAna {
    friend class TwoTrackAnaChain;
 
  public:
    enum EAnaMode {
      kPerfectPairs,        
      kPerfectPairsID,      
      kPerfectPairsNID,     
      kMixedPairs,          
      kMixedPairsID,        
      kMixedPairsNID,       
      kRotatedPairs,        
      kRotatedPairsID,      
      kRotatedPairsNID,     
      kHemispherePairs,     
      kHemispherePairsID,   
      kHemispherePairsNID,  
      kNoBackgroundID,      
      kNoBackgroundNID,     
      kNoBackground,        
      kCharged,             
      kChargedID2,          
      kChargedID3,          
      kChargedNID2,         
      kChargedNID3          
    };
 
  private:
    Bool_t fNonIdIsSet;
    Bool_t fSignedBoth;
    Bool_t fIdentical;
    Bool_t fSwapPair;  // variable to swap identical particles
    Bool_t fSkipEmpty;
    Int_t fCurrentTrack2CollectionNo;
    Int_t fCurrentTrack1CollectionNo;
    Track *fCurrentTrack1, *fCurrentTrack2;
    void BuildPairSwapped(TwoTrack* pair);
    void BuildPair(TwoTrack* pair);
    void MakePairs_Mixed();
    void MakePairs2_Mixed();
    void MakePairs();
    void MakePairs2();
    void MakePairs_Perfect();
    void MakePairs2_Perfect();
    void MakePairs_Rotated();
    void MakePairs2_Rotated();
    void MakePairs_Hemisphere();
    void MakePairs2_Hemisphere();
    void MakePairs_Charged2();
    void MakePairs2_Charged2();
    void MakePairs_Charged3();
    void MakePairs2_Charged3();
 
  protected:
    EAnaMode fBackgroundMode;
    Int_t fCurrentPairCollectionID;
    Int_t fTwoTrackCollectionsNo;
    Int_t fTwoTrackCollectionsNoBackground;
    TwoTrack* fCurrentSignalPair;
    TwoTrack* fCurrentBackgroundPair;
    virtual void SetTags();
    Bool_t IdenticalParticles() const { return (fIdentical); };
    virtual Package* Report() const;
    virtual void SetPairHiddenInfo() {};
    virtual void DeleteHiddenPairInfo();
    virtual void ProcessPair() {};
    virtual void ProcessPair2() { ProcessPair(); };
    virtual void ProcessPair_Mixed() {};
    virtual void ProcessPair2_Mixed() { ProcessPair_Mixed(); };
    virtual void ProcessPair_Perfect() {};
    virtual void ProcessPair2_Perfect() { ProcessPair_Perfect(); };
    virtual void ProcessPair_Rotated() {};
    virtual void ProcessPair2_Rotated() { ProcessPair_Rotated(); };
    virtual void ProcessPair_Hemisphere() {};
    virtual void ProcessPair2_Hemisphere() { ProcessPair_Hemisphere(); };
    virtual void ProcessPair_ChargedId() {};
    virtual void ProcessPair_ChargedNId() { ProcessPair_ChargedId(); };
    virtual void ProcessPair_Charged2() { ProcessPair_ChargedNId(); };
    virtual void ProcessPair2_Charged2() { ProcessPair_ChargedId(); };
    virtual void ProcessPair_Charged3() { ProcessPair_ChargedNId(); };
    virtual void ProcessPair2_Charged3() { ProcessPair_ChargedId(); };
    virtual void ProcessEvent();
    virtual void InitNewCutContainer();
    // check collections prepare them to analysis
    virtual void CheckCutContainerCollections();
    // linking collections
    virtual void LinkCollections();
    virtual void FinishEventIdentical();
    virtual void FinishEventNonIdentical();
    virtual Task::EInitFlag Init();
 
  public:
    TwoTrackAna(Bool_t use_background = kFALSE);
    TwoTrackAna(const TwoTrackAna& ana);
    TwoTrackAna& operator=(const TwoTrackAna& other);
    void SetMixSize(Int_t mixsize);
    void ProceedWithEmtyEvents(Bool_t proceed) { fSkipEmpty = !proceed; }
    virtual void FinishTask();
    virtual void SetOption(Option_t* option);
    void EnableNonIdentical() { SetOption("nonid"); };
    static TString BackgroundOptionMixed() { return "background:mixing"; };
    static TString BackgroundOptionPerfect() { return "background:perfect"; };
    static TString BackgroundOptionRotated() { return "background:rotation"; };
    static TString BackgroundOptionHemisphere() { return "background:hemisphere"; };
    static TString BackgroundOptionCharge() { return "background:charge"; };
    static TString BackgroundOptionNoBackground() { return "background:no"; };
    static TwoTrack::PairType GetPairType(EAnaMode mode);
    virtual ~TwoTrackAna();
    ClassDef(TwoTrackAna, 1)
  };
}  // namespace Hal
 
#endif /* HALTWOTRACKANA_H_ */