[u/mrichter/AliRoot.git] / CORRFW / AliCFEventRecCuts.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/
// Cut on the Event at reconstructed level: for the moment 
// the requirements on the number of charged tracks and on 
// the vertex position and resolution are implemented
// The argument of IsSelected member function (passed object) is cast into 
// an AliESDEvent. In the future may be modified to use AliVEvent interface
// and include more cut variables.
// The class derives from AliCFCutBase
// Author:S.Arcelli Silvia.Arcelli@cern.ch
//
//
#include "TH1F.h"
#include "TList.h"
#include "TBits.h"
#include "AliLog.h"
#include "AliESDEvent.h"
#include "AliESDVertex.h"
#include "AliCFEventRecCuts.h"
ClassImp(AliCFEventRecCuts) 
//____________________________________________________________________
AliCFEventRecCuts::AliCFEventRecCuts() : 
  AliCFCutBase(),
  fNTracksMin(-1),
  fNTracksMax(1000000),
  fRequireVtxCuts(kFALSE),
  fVtxXMax(1.e99),
  fVtxYMax(1.e99),
  fVtxZMax(1.e99),
  fVtxXMin(-1.e99),
  fVtxYMin(-1.e99),
  fVtxZMin(-1.e99),
  fVtxXResMax(1.e99),
  fVtxYResMax(1.e99),
  fVtxZResMax(1.e99),
  fVtxNCtrbMin(0),
  fVtxNCtrbMax((Int_t)1.e9),
  fVtxTPC(0),
  fVtxSPD(0),
  fBitMap(0x0)
{
  //
  //ctor
  //
  fBitMap=new TBits(0);
  Initialise();
}

//____________________________________________________________________
AliCFEventRecCuts::AliCFEventRecCuts(Char_t* name, Char_t* title) : 
  AliCFCutBase(name,title),
  fNTracksMin(-1),
  fNTracksMax(1000000),
  fRequireVtxCuts(kFALSE),
  fVtxXMax(1.e99),
  fVtxYMax(1.e99),
  fVtxZMax(1.e99),
  fVtxXMin(-1.e99),
  fVtxYMin(-1.e99),
  fVtxZMin(-1.e99),
  fVtxXResMax(1.e99),
  fVtxYResMax(1.e99),
  fVtxZResMax(1.e99),
  fVtxNCtrbMin(0),
  fVtxNCtrbMax((Int_t)1.e9),
  fVtxTPC(0),
  fVtxSPD(0),
  fBitMap(0x0)
 {
  //
  //ctor
  //
  fBitMap=new TBits(0);
  Initialise();
 }

//____________________________________________________________________
AliCFEventRecCuts::AliCFEventRecCuts(const AliCFEventRecCuts& c) : 
  AliCFCutBase(c),
  fNTracksMin(c.fNTracksMin),
  fNTracksMax(c.fNTracksMax),
  fRequireVtxCuts(c.fRequireVtxCuts),
  fVtxXMax(c.fVtxXMax),
  fVtxYMax(c.fVtxYMax),
  fVtxZMax(c.fVtxZMax),
  fVtxXMin(c.fVtxXMin),
  fVtxYMin(c.fVtxYMin),
  fVtxZMin(c.fVtxZMin),
  fVtxXResMax(c.fVtxXResMax),
  fVtxYResMax(c.fVtxYResMax),
  fVtxZResMax(c.fVtxZResMax),
  fVtxNCtrbMin(c.fVtxNCtrbMin),
  fVtxNCtrbMax(c.fVtxNCtrbMax),
  fVtxTPC(c.fVtxTPC),
  fVtxSPD(c.fVtxSPD),
  fBitMap(c.fBitMap)
{
  //
  //copy constructor
  //
  for (Int_t i=0; i<c.kNCuts; i++){
    for (Int_t j=0; j<c.kNStepQA; j++){
      if(c.fhQA[i][j]) fhQA[i][j] = (TH1F*)c.fhQA[i][j]->Clone();
    }
  }

}

//____________________________________________________________________
AliCFEventRecCuts::~AliCFEventRecCuts() {
  //
  //dtor
  //

  for (Int_t i=0; i<kNCuts; i++){
    for (Int_t j=0; j<kNStepQA; j++){
      if(fhQA[i][j]) delete fhQA[i][j];
    }
  }

  if(fBitMap)delete fBitMap;

}
//_____________________________________________________________________________
void AliCFEventRecCuts::Initialise()
{

  //
  //initialization
  //

  //
  // sets pointers to histos to zero
  //

  for(Int_t i=0; i<kNCuts; i++){
    for(Int_t j =0; j<kNStepQA; j++){
      fhQA[i][j]=0x0;
    }
  }
}

//____________________________________________________________________
AliCFEventRecCuts& AliCFEventRecCuts::operator=(const AliCFEventRecCuts& c)
{
  //
  // Assignment operator
  //
  if (this != &c) {
    AliCFCutBase::operator=(c) ;
    fNTracksMin=c.fNTracksMin;
    fNTracksMax=c.fNTracksMax;
    fRequireVtxCuts=c.fRequireVtxCuts;
    fVtxXMax=c.fVtxXMax;
    fVtxYMax=c.fVtxYMax;
    fVtxZMax=c.fVtxZMax;
    fVtxXMin=c.fVtxXMin;
    fVtxYMin=c.fVtxYMin;
    fVtxZMin=c.fVtxZMin;
    fVtxXResMax=c.fVtxXResMax;
    fVtxYResMax=c.fVtxYResMax;
    fVtxZResMax=c.fVtxZResMax;
    fVtxNCtrbMin=c.fVtxNCtrbMin;
    fVtxNCtrbMax=c.fVtxNCtrbMax;
    fVtxTPC=c.fVtxTPC;
    fVtxSPD=c.fVtxSPD;
    fBitMap=c.fBitMap;
  }

  for (Int_t i=0; i<c.kNCuts; i++){
    for (Int_t j=0; j<c.kNStepQA; j++){
      if(c.fhQA[i][j]) fhQA[i][j] = (TH1F*)c.fhQA[i][j]->Clone();
    }
  }


  return *this ;
}

//____________________________________________________________________
Bool_t AliCFEventRecCuts::IsSelected(TObject* obj) {
  //
  //Check if the requested cuts are passed
  //


  SelectionBitMap(obj);

  if (fIsQAOn) FillHistograms(obj,0);
  Bool_t isSelected = kTRUE;

  for (UInt_t icut=0; icut<fBitMap->GetNbits();icut++)
	if(!fBitMap->TestBitNumber(icut)) isSelected = kFALSE;

  if (!isSelected) return kFALSE ;
  if (fIsQAOn) FillHistograms(obj,1);
  return kTRUE;

}
//____________________________________________________________________
void AliCFEventRecCuts::SelectionBitMap(TObject* obj) {
  //
  //cut on the number of charged tracks and on the event vertex.
  //so far specific to AliESDEvents
  //

  //Check if the requested cuts are passed and return a bitmap
  for(Int_t j=0;j<kNCuts;j++)fBitMap->SetBitNumber(j,kFALSE);
  AliESDEvent* esd = dynamic_cast<AliESDEvent *>(obj);
  if ( !esd ) return;

  //now start checking the cuts,
  //first assume the event will be accepted: 
  for(Int_t j=0;j<kNCuts;j++)fBitMap->SetBitNumber(j,kTRUE);

  //Number of charged tracks:
  Int_t nTracks = esd->GetNumberOfTracks();
  if(nTracks<fNTracksMin || nTracks>fNTracksMax)
    fBitMap->SetBitNumber(0,kFALSE); 
  
  if(fRequireVtxCuts){
    const AliESDVertex* vtxESD = 0x0;
    if      (fVtxTPC) vtxESD = esd->GetPrimaryVertexTPC() ;
    else if (fVtxSPD) vtxESD = esd->GetPrimaryVertexSPD() ;
    else              vtxESD = esd->GetPrimaryVertexTracks() ;
    if(!vtxESD){
      for(Int_t j=1;j<kNCuts;j++)fBitMap->SetBitNumber(j,kFALSE); 
      AliWarning("Cannot get vertex, skipping event");
      return;
    }
    // Require the vertex to have been reconstructed successfully
    if (strcmp(vtxESD->GetName(), "default")==0){
      AliWarning(Form(" No reconstructed vertex found, skipping event"));    
      for(Int_t j=1;j<kNCuts;j++)fBitMap->SetBitNumber(j,kFALSE); 
      return;
    }    
    // Pick up the position and uncertainties
    
    Double_t vtxPos[3];
    vtxPos[0] = vtxESD->GetXv();
    vtxPos[1] = vtxESD->GetYv();
    vtxPos[2] = vtxESD->GetZv();
    
    Double_t vtxRes[3];
    vtxRes[0] = vtxESD->GetXRes();
    vtxRes[1] = vtxESD->GetYRes();
    vtxRes[2] = vtxESD->GetZRes();
 
    Int_t nCtrb = vtxESD->GetNContributors();

    // Apply the cut
    
    if (vtxPos[0]>fVtxXMax || vtxPos[0]<fVtxXMin)
      fBitMap->SetBitNumber(1,kFALSE); 
    if (vtxPos[1]>fVtxYMax || vtxPos[1]<fVtxYMin)
      fBitMap->SetBitNumber(2,kFALSE); 
    if (vtxPos[2]>fVtxZMax || vtxPos[2]<fVtxZMin)
      fBitMap->SetBitNumber(3,kFALSE); 
    if (vtxRes[0]==0 || vtxRes[0]>fVtxXResMax)
      fBitMap->SetBitNumber(4,kFALSE); 
    if (vtxRes[1]==0 || vtxRes[1]>fVtxYResMax)
      fBitMap->SetBitNumber(5,kFALSE); 
    if (vtxRes[2]==0 || vtxRes[2]>fVtxZResMax)
      fBitMap->SetBitNumber(6,kFALSE); 
    if (nCtrb<fVtxNCtrbMin || nCtrb>fVtxNCtrbMax)
      fBitMap->SetBitNumber(7,kFALSE);

  }  
  return;
}

//_____________________________________________________________________________
void AliCFEventRecCuts::FillHistograms(TObject* obj, Bool_t b)
{
  //
  // fill the QA histograms
  //

  if(!fIsQAOn) return;
  // cast TObject into VParticle
  AliESDEvent* esd = dynamic_cast<AliESDEvent *>(obj);
  if (!esd ) return  ;

  // index = 0: fill histograms before cuts
  // index = 1: fill histograms after cuts
  Int_t index = -1;
  index = ((b) ? 1 : 0);


  //number of charged tracks:
  Int_t nTracks = esd->GetNumberOfTracks();
  fhQA[kNTracks][index]->Fill(nTracks);

  //look at vertex parameters:
  const AliESDVertex* vtxESD = 0x0;
  if      (fVtxTPC) vtxESD = esd->GetPrimaryVertexTPC() ;
  else if (fVtxSPD) vtxESD = esd->GetPrimaryVertexSPD() ;
  else              vtxESD = esd->GetPrimaryVertexTracks() ;
  if(!vtxESD)return;
  // Require the vertex to have been reconstructed successfully
  if (strcmp(vtxESD->GetName(), "default")==0)return;
  // vertex position and uncertainties
  fhQA[kVtxPosX] [index]->Fill(vtxESD->GetXv());
  fhQA[kVtxPosY] [index]->Fill(vtxESD->GetYv());
  fhQA[kVtxPosZ] [index]->Fill(vtxESD->GetZv());
  fhQA[kVtxResX] [index]->Fill(vtxESD->GetXRes());
  fhQA[kVtxResY] [index]->Fill(vtxESD->GetYRes());
  fhQA[kVtxResZ] [index]->Fill(vtxESD->GetZRes());
  fhQA[kVtxNCtrb][index]->Fill(vtxESD->GetNContributors());
  
}

//____________________________________________________________________
void AliCFEventRecCuts::SetHistogramBins(Int_t index, Int_t nbins, Double_t *bins)
{
  //
  //setting x-axis bin limits of QA histogram fhQA[index] 
  // 

  for(Int_t i=0;i<kNStepQA;i++){
    if(!fhQA[index][i]){AliWarning("non-existing histogram!");
    return;
    }
    fhQA[index][i]->GetXaxis()->Set(nbins,bins);
  }
}
//____________________________________________________________________
void AliCFEventRecCuts::SetHistogramBins(Int_t index, Int_t nbins, Double_t xmin, Double_t xmax)
{
  //
  //setting x-axis bins and range of QA histogram fhQA[index] 
  // 

  for(Int_t i=0;i<kNStepQA;i++){
    if(!fhQA[index][i]){AliWarning("non-existing histogram!");
    return;
    }
    fhQA[index][i]->GetXaxis()->Set(nbins,xmin,xmax);
  }
}

//_____________________________________________________________________________
 void AliCFEventRecCuts::DefineHistograms() {
  //
  // histograms for cut variables
  //
  Int_t color = 2;

  if(!fIsQAOn) {
    AliInfo(Form("No QA histos requested, Please first set the QA flag on!"));
    return;
  }  
  
  // book QA histograms

  Char_t str[256];
  for (Int_t i=0; i<kNStepQA; i++) {
    if (i==0) sprintf(str," ");
    else sprintf(str,"_cut");

    fhQA[kNTracks][i]	= new  TH1F(Form("%s_NTracks%s",GetName(),str),	                "",501,-0.5,500.5);
    fhQA[kVtxPosX][i]	= new  TH1F(Form("%s_Vtx_Pos_X%s",GetName(),str),		"",100,-5.,5.);
    fhQA[kVtxPosY][i]	= new  TH1F(Form("%s_Vtx_Pos_Y%s",GetName(),str),		"",100,-5.,5.);
    fhQA[kVtxPosZ][i]	= new  TH1F(Form("%s_Vtx_Pos_Z%s",GetName(),str),		"",200,-50.,50.);
    fhQA[kVtxResX][i]	= new  TH1F(Form("%s_Vtx_Res_X%s",GetName(),str),		"",100,-1.,1.);
    fhQA[kVtxResY][i]	= new  TH1F(Form("%s_Vtx_Res_Y%s",GetName(),str),		"",100,-1.,1.);
    fhQA[kVtxResZ][i]	= new  TH1F(Form("%s_Vtx_Res_Z%s",GetName(),str),		"",100,-1.,1.);
    fhQA[kVtxNCtrb][i]	= new  TH1F(Form("%s_Vtx_N_Ctrb%s",GetName(),str),		"",1000,0.,1000.);
 
    fhQA[kNTracks][i]	->SetXTitle("Number of ESD tracks");
    fhQA[kVtxPosX][i]	->SetXTitle("Vertex Position X (cm)");
    fhQA[kVtxPosY][i]	->SetXTitle("Vertex Position Y (cm)");
    fhQA[kVtxPosZ][i]	->SetXTitle("Vertex Position Z (cm)");
    fhQA[kVtxResX][i]	->SetXTitle("Vertex Resolution X (cm)");
    fhQA[kVtxResY][i]	->SetXTitle("Vertex Resolution Y (cm)");
    fhQA[kVtxResZ][i]	->SetXTitle("Vertex Resolution Z (cm)");
    fhQA[kVtxNCtrb][i]	->SetXTitle("Number of contributors");
  }

  for(Int_t i=0; i<kNCuts; i++) fhQA[i][1]->SetLineColor(color);

}

//_____________________________________________________________________________
void AliCFEventRecCuts::AddQAHistograms(TList *qaList) {
  //
  // saves the histograms in a TList
  //

  DefineHistograms();

  for (Int_t j=0; j<kNStepQA; j++) {
    for(Int_t i=0; i<kNCuts; i++)
	qaList->Add(fhQA[i][j]);
  }
}
Commit	Line	Data
563113d0	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15	// Cut on the Event at reconstructed level: for the moment
	16	// the requirements on the number of charged tracks and on
	17	// the vertex position and resolution are implemented
	18	// The argument of IsSelected member function (passed object) is cast into
	19	// an AliESDEvent. In the future may be modified to use AliVEvent interface
	20	// and include more cut variables.
	21	// The class derives from AliCFCutBase
	22	// Author:S.Arcelli Silvia.Arcelli@cern.ch
	23	//
	24	//
89dfdd61	25	#include "TH1F.h"
89dfdd61	26	#include "TList.h"
563113d0	27	#include "TBits.h"
	28	#include "AliLog.h"
	29	#include "AliESDEvent.h"
	30	#include "AliESDVertex.h"
	31	#include "AliCFEventRecCuts.h"
	32	ClassImp(AliCFEventRecCuts)
	33	//____________________________________________________________________
	34	AliCFEventRecCuts::AliCFEventRecCuts() :
	35	AliCFCutBase(),
	36	fNTracksMin(-1),
	37	fNTracksMax(1000000),
	38	fRequireVtxCuts(kFALSE),
	39	fVtxXMax(1.e99),
	40	fVtxYMax(1.e99),
	41	fVtxZMax(1.e99),
	42	fVtxXMin(-1.e99),
	43	fVtxYMin(-1.e99),
	44	fVtxZMin(-1.e99),
	45	fVtxXResMax(1.e99),
	46	fVtxYResMax(1.e99),
	47	fVtxZResMax(1.e99),
fc885b03	48	fVtxNCtrbMin(0),
	49	fVtxNCtrbMax((Int_t)1.e9),
	50	fVtxTPC(0),
7e1fc4c2	51	fVtxSPD(0),
107a3100	52	fBitMap(0x0)
563113d0	53	{
	54	//
	55	//ctor
	56	//
	57	fBitMap=new TBits(0);
89dfdd61	58	Initialise();
563113d0	59	}
89dfdd61	60
563113d0	61	//____________________________________________________________________
	62	AliCFEventRecCuts::AliCFEventRecCuts(Char_t* name, Char_t* title) :
	63	AliCFCutBase(name,title),
	64	fNTracksMin(-1),
	65	fNTracksMax(1000000),
	66	fRequireVtxCuts(kFALSE),
	67	fVtxXMax(1.e99),
	68	fVtxYMax(1.e99),
	69	fVtxZMax(1.e99),
	70	fVtxXMin(-1.e99),
	71	fVtxYMin(-1.e99),
	72	fVtxZMin(-1.e99),
	73	fVtxXResMax(1.e99),
	74	fVtxYResMax(1.e99),
	75	fVtxZResMax(1.e99),
fc885b03	76	fVtxNCtrbMin(0),
	77	fVtxNCtrbMax((Int_t)1.e9),
	78	fVtxTPC(0),
7e1fc4c2	79	fVtxSPD(0),
107a3100	80	fBitMap(0x0)
563113d0	81	{
	82	//
	83	//ctor
	84	//
	85	fBitMap=new TBits(0);
89dfdd61	86	Initialise();
563113d0	87	}
89dfdd61	88
563113d0	89	//____________________________________________________________________
	90	AliCFEventRecCuts::AliCFEventRecCuts(const AliCFEventRecCuts& c) :
	91	AliCFCutBase(c),
	92	fNTracksMin(c.fNTracksMin),
	93	fNTracksMax(c.fNTracksMax),
	94	fRequireVtxCuts(c.fRequireVtxCuts),
	95	fVtxXMax(c.fVtxXMax),
	96	fVtxYMax(c.fVtxYMax),
	97	fVtxZMax(c.fVtxZMax),
	98	fVtxXMin(c.fVtxXMin),
	99	fVtxYMin(c.fVtxYMin),
	100	fVtxZMin(c.fVtxZMin),
	101	fVtxXResMax(c.fVtxXResMax),
	102	fVtxYResMax(c.fVtxYResMax),
	103	fVtxZResMax(c.fVtxZResMax),
fc885b03	104	fVtxNCtrbMin(c.fVtxNCtrbMin),
	105	fVtxNCtrbMax(c.fVtxNCtrbMax),
	106	fVtxTPC(c.fVtxTPC),
7e1fc4c2	107	fVtxSPD(c.fVtxSPD),
107a3100	108	fBitMap(c.fBitMap)
563113d0	109	{
	110	//
	111	//copy constructor
	112	//
89dfdd61	113	for (Int_t i=0; i<c.kNCuts; i++){
	114	for (Int_t j=0; j<c.kNStepQA; j++){
	115	if(c.fhQA[i][j]) fhQA[i][j] = (TH1F*)c.fhQA[i][j]->Clone();
	116	}
	117	}
	118
563113d0	119	}
89dfdd61	120
563113d0	121	//____________________________________________________________________
	122	AliCFEventRecCuts::~AliCFEventRecCuts() {
	123	//
	124	//dtor
	125	//
	126
89dfdd61	127	for (Int_t i=0; i<kNCuts; i++){
	128	for (Int_t j=0; j<kNStepQA; j++){
	129	if(fhQA[i][j]) delete fhQA[i][j];
	130	}
	131	}
	132
563113d0	133	if(fBitMap)delete fBitMap;
89dfdd61	134
89dfdd61	135	}
89dfdd61	136	//_____________________________________________________________________________
	137	void AliCFEventRecCuts::Initialise()
	138	{
	139
	140	//
	141	//initialization
	142	//
	143
	144	//
	145	// sets pointers to histos to zero
	146	//
	147
	148	for(Int_t i=0; i<kNCuts; i++){
	149	for(Int_t j =0; j<kNStepQA; j++){
	150	fhQA[i][j]=0x0;
	151	}
	152	}
563113d0	153	}
89dfdd61	154
563113d0	155	//____________________________________________________________________
	156	AliCFEventRecCuts& AliCFEventRecCuts::operator=(const AliCFEventRecCuts& c)
	157	{
	158	//
	159	// Assignment operator
	160	//
	161	if (this != &c) {
	162	AliCFCutBase::operator=(c) ;
	163	fNTracksMin=c.fNTracksMin;
	164	fNTracksMax=c.fNTracksMax;
	165	fRequireVtxCuts=c.fRequireVtxCuts;
	166	fVtxXMax=c.fVtxXMax;
	167	fVtxYMax=c.fVtxYMax;
	168	fVtxZMax=c.fVtxZMax;
	169	fVtxXMin=c.fVtxXMin;
	170	fVtxYMin=c.fVtxYMin;
	171	fVtxZMin=c.fVtxZMin;
	172	fVtxXResMax=c.fVtxXResMax;
	173	fVtxYResMax=c.fVtxYResMax;
	174	fVtxZResMax=c.fVtxZResMax;
fc885b03	175	fVtxNCtrbMin=c.fVtxNCtrbMin;
	176	fVtxNCtrbMax=c.fVtxNCtrbMax;
	177	fVtxTPC=c.fVtxTPC;
7e1fc4c2	178	fVtxSPD=c.fVtxSPD;
563113d0	179	fBitMap=c.fBitMap;
89dfdd61	180	}
	181
	182	for (Int_t i=0; i<c.kNCuts; i++){
	183	for (Int_t j=0; j<c.kNStepQA; j++){
	184	if(c.fhQA[i][j]) fhQA[i][j] = (TH1F*)c.fhQA[i][j]->Clone();
	185	}
563113d0	186	}
89dfdd61	187
89dfdd61	188
563113d0	189	return *this ;
563113d0	190	}
89dfdd61	191
563113d0	192	//____________________________________________________________________
9eeae5d5	193	Bool_t AliCFEventRecCuts::IsSelected(TObject* obj) {
563113d0	194	//
	195	//Check if the requested cuts are passed
	196	//
	197
563113d0	198
107a3100	199	SelectionBitMap(obj);
	200
	201	if (fIsQAOn) FillHistograms(obj,0);
563113d0	202	Bool_t isSelected = kTRUE;
563113d0	203
107a3100	204	for (UInt_t icut=0; icut<fBitMap->GetNbits();icut++)
107a3100	205	if(!fBitMap->TestBitNumber(icut)) isSelected = kFALSE;
563113d0	206
107a3100	207	if (!isSelected) return kFALSE ;
	208	if (fIsQAOn) FillHistograms(obj,1);
	209	return kTRUE;
563113d0	210
563113d0	211	}
563113d0	212	//____________________________________________________________________
107a3100	213	void AliCFEventRecCuts::SelectionBitMap(TObject* obj) {
563113d0	214	//
	215	//cut on the number of charged tracks and on the event vertex.
	216	//so far specific to AliESDEvents
	217	//
	218
	219	//Check if the requested cuts are passed and return a bitmap
	220	for(Int_t j=0;j<kNCuts;j++)fBitMap->SetBitNumber(j,kFALSE);
	221	AliESDEvent* esd = dynamic_cast<AliESDEvent *>(obj);
107a3100	222	if ( !esd ) return;
563113d0	223
	224	//now start checking the cuts,
	225	//first assume the event will be accepted:
	226	for(Int_t j=0;j<kNCuts;j++)fBitMap->SetBitNumber(j,kTRUE);
	227
	228	//Number of charged tracks:
	229	Int_t nTracks = esd->GetNumberOfTracks();
	230	if(nTracks<fNTracksMin \|\| nTracks>fNTracksMax)
	231	fBitMap->SetBitNumber(0,kFALSE);
	232
	233	if(fRequireVtxCuts){
7e1fc4c2	234	const AliESDVertex* vtxESD = 0x0;
	235	if (fVtxTPC) vtxESD = esd->GetPrimaryVertexTPC() ;
	236	else if (fVtxSPD) vtxESD = esd->GetPrimaryVertexSPD() ;
	237	else vtxESD = esd->GetPrimaryVertexTracks() ;
563113d0	238	if(!vtxESD){
563113d0	239	for(Int_t j=1;j<kNCuts;j++)fBitMap->SetBitNumber(j,kFALSE);
fc885b03	240	AliWarning("Cannot get vertex, skipping event");
107a3100	241	return;
563113d0	242	}
	243	// Require the vertex to have been reconstructed successfully
	244	if (strcmp(vtxESD->GetName(), "default")==0){
fc885b03	245	AliWarning(Form(" No reconstructed vertex found, skipping event"));
563113d0	246	for(Int_t j=1;j<kNCuts;j++)fBitMap->SetBitNumber(j,kFALSE);
107a3100	247	return;
563113d0	248	}
	249	// Pick up the position and uncertainties
	250
	251	Double_t vtxPos[3];
	252	vtxPos[0] = vtxESD->GetXv();
	253	vtxPos[1] = vtxESD->GetYv();
	254	vtxPos[2] = vtxESD->GetZv();
	255
	256	Double_t vtxRes[3];
	257	vtxRes[0] = vtxESD->GetXRes();
	258	vtxRes[1] = vtxESD->GetYRes();
	259	vtxRes[2] = vtxESD->GetZRes();
	260
fc885b03	261	Int_t nCtrb = vtxESD->GetNContributors();
fc885b03	262
563113d0	263	// Apply the cut
	264
	265	if (vtxPos[0]>fVtxXMax \|\| vtxPos[0]<fVtxXMin)
	266	fBitMap->SetBitNumber(1,kFALSE);
	267	if (vtxPos[1]>fVtxYMax \|\| vtxPos[1]<fVtxYMin)
	268	fBitMap->SetBitNumber(2,kFALSE);
	269	if (vtxPos[2]>fVtxZMax \|\| vtxPos[2]<fVtxZMin)
	270	fBitMap->SetBitNumber(3,kFALSE);
	271	if (vtxRes[0]==0 \|\| vtxRes[0]>fVtxXResMax)
	272	fBitMap->SetBitNumber(4,kFALSE);
	273	if (vtxRes[1]==0 \|\| vtxRes[1]>fVtxYResMax)
	274	fBitMap->SetBitNumber(5,kFALSE);
	275	if (vtxRes[2]==0 \|\| vtxRes[2]>fVtxZResMax)
	276	fBitMap->SetBitNumber(6,kFALSE);
fc885b03	277	if (nCtrb<fVtxNCtrbMin \|\| nCtrb>fVtxNCtrbMax)
	278	fBitMap->SetBitNumber(7,kFALSE);
	279
563113d0	280	}
107a3100	281	return;
563113d0	282	}
89dfdd61	283
	284	//_____________________________________________________________________________
	285	void AliCFEventRecCuts::FillHistograms(TObject* obj, Bool_t b)
	286	{
	287	//
	288	// fill the QA histograms
	289	//
	290
	291	if(!fIsQAOn) return;
	292	// cast TObject into VParticle
	293	AliESDEvent* esd = dynamic_cast<AliESDEvent *>(obj);
	294	if (!esd ) return ;
	295
	296	// index = 0: fill histograms before cuts
	297	// index = 1: fill histograms after cuts
	298	Int_t index = -1;
	299	index = ((b) ? 1 : 0);
	300
	301
	302	//number of charged tracks:
	303	Int_t nTracks = esd->GetNumberOfTracks();
	304	fhQA[kNTracks][index]->Fill(nTracks);
	305
	306	//look at vertex parameters:
7e1fc4c2	307	const AliESDVertex* vtxESD = 0x0;
	308	if (fVtxTPC) vtxESD = esd->GetPrimaryVertexTPC() ;
	309	else if (fVtxSPD) vtxESD = esd->GetPrimaryVertexSPD() ;
	310	else vtxESD = esd->GetPrimaryVertexTracks() ;
89dfdd61	311	if(!vtxESD)return;
	312	// Require the vertex to have been reconstructed successfully
	313	if (strcmp(vtxESD->GetName(), "default")==0)return;
	314	// vertex position and uncertainties
fc885b03	315	fhQA[kVtxPosX] [index]->Fill(vtxESD->GetXv());
	316	fhQA[kVtxPosY] [index]->Fill(vtxESD->GetYv());
	317	fhQA[kVtxPosZ] [index]->Fill(vtxESD->GetZv());
	318	fhQA[kVtxResX] [index]->Fill(vtxESD->GetXRes());
	319	fhQA[kVtxResY] [index]->Fill(vtxESD->GetYRes());
	320	fhQA[kVtxResZ] [index]->Fill(vtxESD->GetZRes());
	321	fhQA[kVtxNCtrb][index]->Fill(vtxESD->GetNContributors());
89dfdd61	322
	323	}
	324
107a3100	325	//____________________________________________________________________
89dfdd61	326	void AliCFEventRecCuts::SetHistogramBins(Int_t index, Int_t nbins, Double_t *bins)
	327	{
	328	//
107a3100	329	//setting x-axis bin limits of QA histogram fhQA[index]
107a3100	330	//
89dfdd61	331
107a3100	332	for(Int_t i=0;i<kNStepQA;i++){
	333	if(!fhQA[index][i]){AliWarning("non-existing histogram!");
	334	return;
	335	}
	336	fhQA[index][i]->GetXaxis()->Set(nbins,bins);
89dfdd61	337	}
89dfdd61	338	}
107a3100	339	//____________________________________________________________________
89dfdd61	340	void AliCFEventRecCuts::SetHistogramBins(Int_t index, Int_t nbins, Double_t xmin, Double_t xmax)
	341	{
	342	//
107a3100	343	//setting x-axis bins and range of QA histogram fhQA[index]
	344	//
	345
	346	for(Int_t i=0;i<kNStepQA;i++){
	347	if(!fhQA[index][i]){AliWarning("non-existing histogram!");
	348	return;
	349	}
	350	fhQA[index][i]->GetXaxis()->Set(nbins,xmin,xmax);
89dfdd61	351	}
	352	}
	353
	354	//_____________________________________________________________________________
	355	void AliCFEventRecCuts::DefineHistograms() {
	356	//
	357	// histograms for cut variables
	358	//
	359	Int_t color = 2;
	360
	361	if(!fIsQAOn) {
	362	AliInfo(Form("No QA histos requested, Please first set the QA flag on!"));
	363	return;
	364	}
	365
	366	// book QA histograms
	367
	368	Char_t str[256];
	369	for (Int_t i=0; i<kNStepQA; i++) {
	370	if (i==0) sprintf(str," ");
	371	else sprintf(str,"_cut");
	372
107a3100	373	fhQA[kNTracks][i] = new TH1F(Form("%s_NTracks%s",GetName(),str), "",501,-0.5,500.5);
	374	fhQA[kVtxPosX][i] = new TH1F(Form("%s_Vtx_Pos_X%s",GetName(),str), "",100,-5.,5.);
	375	fhQA[kVtxPosY][i] = new TH1F(Form("%s_Vtx_Pos_Y%s",GetName(),str), "",100,-5.,5.);
	376	fhQA[kVtxPosZ][i] = new TH1F(Form("%s_Vtx_Pos_Z%s",GetName(),str), "",200,-50.,50.);
107a3100	377	fhQA[kVtxResX][i] = new TH1F(Form("%s_Vtx_Res_X%s",GetName(),str), "",100,-1.,1.);
	378	fhQA[kVtxResY][i] = new TH1F(Form("%s_Vtx_Res_Y%s",GetName(),str), "",100,-1.,1.);
	379	fhQA[kVtxResZ][i] = new TH1F(Form("%s_Vtx_Res_Z%s",GetName(),str), "",100,-1.,1.);
fc885b03	380	fhQA[kVtxNCtrb][i] = new TH1F(Form("%s_Vtx_N_Ctrb%s",GetName(),str), "",1000,0.,1000.);
89dfdd61	381
	382	fhQA[kNTracks][i] ->SetXTitle("Number of ESD tracks");
	383	fhQA[kVtxPosX][i] ->SetXTitle("Vertex Position X (cm)");
	384	fhQA[kVtxPosY][i] ->SetXTitle("Vertex Position Y (cm)");
	385	fhQA[kVtxPosZ][i] ->SetXTitle("Vertex Position Z (cm)");
	386	fhQA[kVtxResX][i] ->SetXTitle("Vertex Resolution X (cm)");
	387	fhQA[kVtxResY][i] ->SetXTitle("Vertex Resolution Y (cm)");
	388	fhQA[kVtxResZ][i] ->SetXTitle("Vertex Resolution Z (cm)");
fc885b03	389	fhQA[kVtxNCtrb][i] ->SetXTitle("Number of contributors");
89dfdd61	390	}
	391
	392	for(Int_t i=0; i<kNCuts; i++) fhQA[i][1]->SetLineColor(color);
	393
	394	}
	395
	396	//_____________________________________________________________________________
107a3100	397	void AliCFEventRecCuts::AddQAHistograms(TList *qaList) {
89dfdd61	398	//
	399	// saves the histograms in a TList
	400	//
107a3100	401
107a3100	402	DefineHistograms();
89dfdd61	403
	404	for (Int_t j=0; j<kNStepQA; j++) {
	405	for(Int_t i=0; i<kNCuts; i++)
107a3100	406	qaList->Add(fhQA[i][j]);
89dfdd61	407	}
89dfdd61	408	}