///////////////////////////////////////////////////////////////////////////
//                                                                       //
//                                                                       //
// Authors:
//   * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
//   Julian Book   <Julian.Book@cern.ch>
/*
  Cut class providing cuts for both legs in the PairAnalysisPair
  Add any number of leg cuts using e.g. for leg 1
    GetFilterLeg1().AddCuts(mycut)
  where mycut has to inherit from AnalysisCuts.

  the following cut types are defined and selected via SetCutType():
    kBothLegs, kAnyLeg, kMixLegs, kOneLeg

*/
//                                                                       //
///////////////////////////////////////////////////////////////////////////

#include <TList.h>

#include "PairAnalysisPairLV.h"
#include "PairAnalysisTrack.h"

#include "PairAnalysisPairLegCuts.h"

ClassImp(PairAnalysisPairLegCuts)


PairAnalysisPairLegCuts::PairAnalysisPairLegCuts() :
  PairAnalysisPairLegCuts("pairlegcut","pairlegcut")
{
  //
  // Default contructor
  //
}

//________________________________________________________________________
PairAnalysisPairLegCuts::PairAnalysisPairLegCuts(const char* name, const char* title) :
  AnalysisCuts(name,title),
  fFilterLeg1("PairFilterLeg1","PairFilterLeg1"),
  fFilterLeg2("PairFilterLeg2","PairFilterLeg2")
{
  //
  // Named contructor
  //
}


//________________________________________________________________________
Bool_t PairAnalysisPairLegCuts::IsSelected(TObject* track)
{
  //
  // check if cuts are fulfilled
  //
  
  //check if we have a PairAnalysisPair
  PairAnalysisPair *pair=dynamic_cast<PairAnalysisPair*>(track);
  if (!pair) return kFALSE;

  //get both legs
  PairAnalysisTrack *leg1=pair->GetFirstDaughter();
  PairAnalysisTrack *leg2=pair->GetSecondDaughter();

  //mask used to require that all cuts are fulfilled
  UInt_t selectedMaskLeg1=(1<<fFilterLeg1.GetCuts()->GetEntries())-1;
  UInt_t selectedMaskLeg2=(1<<fFilterLeg2.GetCuts()->GetEntries())-1;
  
  //test cuts
  Bool_t isLeg1selected=(fFilterLeg1.IsSelected(leg1)==selectedMaskLeg1);
  if(fCutType==kBothLegs && !isLeg1selected) {
    SetSelected(isLeg1selected);
    return isLeg1selected;
  }

  Bool_t isLeg2selected=(fFilterLeg2.IsSelected(leg2)==selectedMaskLeg2);
  Bool_t isSelected=isLeg1selected&&isLeg2selected;
  if (fCutType==kAnyLeg) {
    isSelected=isLeg1selected||isLeg2selected;
    SetSelected(isSelected);
    return isSelected;
  }
  if (fCutType==kOneLeg) {
    isSelected=(isLeg1selected!=isLeg2selected);
    SetSelected(isSelected);
    return isSelected;
  }

  if (fCutType==kMixLegs) {
    Bool_t isLeg1selectedMirror=(fFilterLeg1.IsSelected(leg2)==selectedMaskLeg1);
    Bool_t isLeg2selectedMirror=(fFilterLeg2.IsSelected(leg1)==selectedMaskLeg2);
    isSelected=(isLeg1selected&&isLeg2selected)||(isLeg1selectedMirror&&isLeg2selectedMirror);
    SetSelected(isSelected);
  }
  return isSelected;
}