[u/mrichter/AliRoot.git] / TRD / qaRec / AliTRDtrackingEfficiency.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.                  *
 **************************************************************************/

/* $Id: AliTRDtrackingEfficiency.cxx 27496 2008-07-22 08:35:45Z cblume $ */

////////////////////////////////////////////////////////////////////////////
//                                                                        //
//  Reconstruction QA                                                     //
//                                                                        //
//  Authors:                                                              //
//    Markus Fasel <M.Fasel@gsi.de>                                       //
//                                                                        //
////////////////////////////////////////////////////////////////////////////

#include <TClonesArray.h>
#include <TObjArray.h>
#include <TProfile.h>
#include "TTreeStream.h"

#include "AliPID.h"
#include "AliESDtrack.h"
#include "AliTrackReference.h"
#include "AliExternalTrackParam.h"
#include "AliTracker.h"
#include "AliMagFMaps.h"
#include "AliAnalysisManager.h"

#include "Cal/AliTRDCalPID.h"
#include "AliTRDtrackingEfficiency.h"
#include "AliTRDtrackInfo/AliTRDtrackInfo.h"

ClassImp(AliTRDtrackingEfficiency)

//____________________________________________________________________
AliTRDtrackingEfficiency::AliTRDtrackingEfficiency()
  :AliTRDrecoTask("TrackingEff", "Barrel Tracking Efficiency")
  ,fMissed(0x0)
{
  //
  // Default constructor
  //
}

//____________________________________________________________________
AliTRDtrackingEfficiency::~AliTRDtrackingEfficiency()
{
  if(fMissed){
    fMissed->Delete();
    delete fMissed;
  }
}

//____________________________________________________________________
void  AliTRDtrackingEfficiency::CreateOutputObjects()
{
  //
  // Create output objects
  //

  OpenFile(0, "RECREATE");
  const Int_t nbins = AliTRDCalPID::kNMom;
  Float_t xbins[nbins+1] = {.5, .7, .9, 1.3, 1.7, 2.4, 3.5, 4.5, 5.5, 7., 9., 11.};

  TH1 *h = 0x0;
  fContainer = new TObjArray();
  for(Int_t is=0; is<AliPID::kSPECIES; is++){
    fContainer->Add(h = new TProfile(Form("h%s", AliTRDCalPID::GetPartSymb(is)), "", nbins, xbins));
    h->SetLineColor(AliTRDCalPID::GetPartColor(is));
    h->SetMarkerColor(AliTRDCalPID::GetPartColor(is));
    h->SetMarkerStyle(7);
  }
  fContainer->Add(h = new TProfile("h", "", nbins, xbins));
  h->SetMarkerStyle(7);
} 

//____________________________________________________________________
void AliTRDtrackingEfficiency::Exec(Option_t *)
{
  //
  // Do it
  //

  Int_t labelsacc[10000]; 
  memset(labelsacc, 0, sizeof(Int_t) * 10000);
	
  if(!fMissed){ 
    fMissed = new TClonesArray("AliTRDtrackInfo", 10);
    fMissed->SetOwner();
  }

  Float_t mom;
  Int_t selection[10000], nselect = 0;
  ULong_t status; Int_t pidx;
  Int_t nTRD = 0, nTPC = 0, nMiss = 0;
  AliTRDtrackInfo     *track = 0x0;
  AliTrackReference     *ref = 0x0;
  AliExternalTrackParam *esd = 0x0;
  for(Int_t itrk=0; itrk<fTracks->GetEntriesFast(); itrk++){
    track = (AliTRDtrackInfo*)fTracks->UncheckedAt(itrk);

		if(!track->HasESDtrack()) continue;
    status = track->GetStatus();

    // missing TPC propagation - interesting for SA
    if(!(status&AliESDtrack::kTPCout)) continue;

    // missing MC info.
    if(HasMCdata() && track->GetNTrackRefs() <= 1) continue;
   
    nTPC++;
    selection[nselect++]=itrk;
    ref  = track->GetTrackRef(0);
    esd  = track->GetESDinfo()->GetOuterParam();
    mom  = ref ? ref->P(): esd->P();
    pidx = AliTRDCalPID::GetPartIndex(track->GetPDG());
    pidx = TMath::Max(pidx, 0);

    //Int_t n = track->GetNumberOfClusters(); 
    // where are this tracklets ???
    //if(ncls0 > ncls1) printf("%3d ESD[%3d] TRD[%3d|%3d]\n", itrk, ncls0, ncls1, n);
    if(track->GetNumberOfClustersRefit()){ 
      ((TProfile*)fContainer->At(pidx))->Fill(mom, 1.);
			labelsacc[nTRD] = track->GetLabel();
      nTRD++;
      continue;
    }


    Float_t xmed, xleng;
    Int_t iref = 1; Bool_t found = kFALSE;
    while((ref = track->GetTrackRef(iref))){
      xmed = .5*(ref->LocalX() + track->GetTrackRef(iref-1)->LocalX());
      xleng= (ref->LocalX() - track->GetTrackRef(iref-1)->LocalX());
      if(TMath::Abs(xmed - 298.5) < .5 &&
        TMath::Abs(xleng - 3.7) < .1){ 
        found = kTRUE;
        break;
      }
      iref++;
    }
    if(!found){ 
      nTPC--;
      // track missing first layer. Maybe interesting for SA.
      continue;
    }
    nselect--;
    new ((*fMissed)[nMiss]) AliTRDtrackInfo(*track);
    nMiss++;
  }
  if(fDebugLevel>=2) printf("%3d Tracks: ESD[%3d] TPC[%3d] TRD[%3d | %5.2f%%] Off[%d]\n", (Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), fTracks->GetEntriesFast(), nTPC, nTRD, nTPC ? 1.E2*nTRD/float(nTPC) : 0., fMissed->GetEntriesFast());


  // Find double tracks
  Float_t threshold = 10.;
  AliTrackReference *refMiss = 0x0;
  AliExternalTrackParam *op = 0x0;
  AliTRDtrackInfo       *tt = 0x0;
  for(Int_t imiss=0; imiss<nMiss; imiss++){
    //printf("Searching missing %d ...\n", imiss);

    // get outer param of missed
    tt = (AliTRDtrackInfo*)fMissed->UncheckedAt(imiss);
    op = tt->GetESDinfo()->GetOuterParam();
    Double_t alpha = op->GetAlpha(), cosa = TMath::Cos(alpha), sina = TMath::Sin(alpha);

    Double_t xyz[3], x0, y0, z0, x, y, z, dx, dy, dz, d;

    Bool_t FOUND = kFALSE;
    for(Int_t iselect=0; iselect<nselect; iselect++){
      track = (AliTRDtrackInfo*)fTracks->UncheckedAt(selection[iselect]);

      // check first MC ... if available
      d = 0;
      for(Int_t iref=0; iref<track->GetNTrackRefs(); iref++){
        if(!(ref = track->GetTrackRef(iref))) continue;
        if((refMiss = tt->GetTrackRef(iref))){
          dy = ref->LocalY() - refMiss->LocalY();
          dz = ref->Z() - refMiss->Z();
        } else {
          // compare missOP with refTrackRef in LTC
          x0 = ref->LocalX();
          op->GetYAt(x0, AliTracker::GetBz(), y0);
          op->GetZAt(x0, AliTracker::GetBz(), z0);
          dy = y0 - ref->LocalY();
          dz = z0 - ref->Z();
        }
        d += (dy*dy + dz*dz);
      }
      //printf("\td[%d] = %f N[%d]\n", selection[iselect], d, track->GetNTrackRefs());
      if((track->GetNTrackRefs())){ 
        d /= track->GetNTrackRefs();
        if(d < threshold){
          //printf("\t\tFound %2d in ref[%3d] : d[%f]\n", imiss, selection[iselect], d/track->GetNTrackRefs());
          FOUND = kTRUE; break;
        }
      }

      // process outer param ... always available
      // compare missOP with OP in GTC
      esd = track->GetESDinfo()->GetOuterParam();
      esd->GetXYZ(xyz);
      x0 = esd->GetX();
      op->GetYAt(x0, AliTracker::GetBz(), y0);
      op->GetZAt(x0, AliTracker::GetBz(), z0);
      x = x0*cosa - y0*sina;
      y = x0*sina + y0*cosa;
      z = z0;
      dx=xyz[0]-x;
      dy=xyz[1]-y;
      dz=xyz[2]-z;
      d = dx*dx+dy*dy+dz*dz;
      //printf("\td[%d] = %f op\n", selection[iselect], d);
      if(d < threshold){
        //printf("\t\tFound %2d in op[%3d]  : d[%f] dx[%5.2f] dy[%5.2f] dz[%5.2f]\n", imiss, selection[iselect], d, dx, dy, dz);
        FOUND = kTRUE; break;
      }
    }
    if(FOUND) nTPC--;
    else{ 
      ref = tt->GetTrackRef(0);
      mom = ref ? ref->P(): op->P();
      pidx = AliTRDCalPID::GetPartIndex(tt->GetPDG());
      pidx = TMath::Max(pidx, 0);
      ((TProfile*)fContainer->At(pidx))->Fill(mom, 0.);
      if(fDebugLevel>=2) printf("\tNOT FOUND Id[%d] Mom[%f]\n", tt->GetTrackId(), mom);
    }
  }

  if(fDebugLevel>=2) printf("%3d Tracks: ESD[%3d] TPC[%3d] TRD[%3d | %5.2f%%] Off[%d]\n", (Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), fTracks->GetEntriesFast(), nTPC, nTRD, nTPC ? 1.E2*nTRD/float(nTPC) : 0., fMissed->GetEntriesFast());

  //fMissed->Delete();
	// check for double countings
	Int_t indices[10000]; memset(indices, 0, sizeof(Int_t) * 10000);
	TMath::Sort(nTRD, labelsacc, indices);
	if(fDebugLevel > 2){
	for(Int_t itk = 0; itk < nTRD - 1; itk++)
		if(labelsacc[indices[itk]] ==labelsacc[indices[itk + 1]]) printf("Double counted MC track: %d\n", labelsacc[indices[itk]]);
	}
  PostData(0, fContainer);
}

//____________________________________________________________________
void AliTRDtrackingEfficiency::Terminate(Option_t *)
{
  //
  // Terminate
  //

  if(fDebugStream){ 
    delete fDebugStream;
    fDebugStream = 0x0;
    fDebugLevel = 0;
  }

  fContainer = dynamic_cast<TObjArray*>(GetOutputData(0));
  if (!fContainer) {
    Printf("ERROR: list not available");
    return;
  }

}


//____________________________________________________________________
Bool_t AliTRDtrackingEfficiency::GetRefFigure(Int_t ifig)
{
  Bool_t FIRST = kTRUE;
  TProfile *h = 0x0;
  switch(ifig){
  case 0:
    h = (TProfile*)fContainer->At(AliPID::kSPECIES);
    for(Int_t is=0; is<AliPID::kSPECIES; is++){
      h->Add((TProfile*)fContainer->At(is));
    }
    h->SetMarkerColor(kBlack);
    h->SetLineColor(kBlack);
    h->GetXaxis()->SetTitle("p [GeV/c]");
    h->Draw("e1");
    break;
  case 1:
    FIRST = kTRUE;
    for(Int_t is=0; is<AliPID::kSPECIES; is++){
      if(!(h = (TProfile*)fContainer->At(is))) continue;
      if(FIRST){
        h->Draw("e1");
        h->GetXaxis()->SetTitle("p [GeV/c]");
      } else h->Draw("same e1");
      FIRST = kFALSE;
    }
    break;
  }
  return kTRUE;
}


//____________________________________________________________________
Bool_t AliTRDtrackingEfficiency::PostProcess()
{
  fNRefFigures = HasMCdata() ? 2 : 1; 
  return kTRUE;
}
Commit	Line	Data
814ecea4	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
	16	/* $Id: AliTRDtrackingEfficiency.cxx 27496 2008-07-22 08:35:45Z cblume $ */
	17
	18	////////////////////////////////////////////////////////////////////////////
	19	// //
	20	// Reconstruction QA //
	21	// //
	22	// Authors: //
	23	// Markus Fasel <M.Fasel@gsi.de> //
	24	// //
	25	////////////////////////////////////////////////////////////////////////////
	26
	27	#include <TClonesArray.h>
	28	#include <TObjArray.h>
814ecea4	29	#include <TProfile.h>
ed383798	30	#include "TTreeStream.h"
814ecea4	31
5d6dc395	32	#include "AliPID.h"
814ecea4	33	#include "AliESDtrack.h"
	34	#include "AliTrackReference.h"
	35	#include "AliExternalTrackParam.h"
	36	#include "AliTracker.h"
	37	#include "AliMagFMaps.h"
	38	#include "AliAnalysisManager.h"
	39
5d6dc395	40	#include "Cal/AliTRDCalPID.h"
814ecea4	41	#include "AliTRDtrackingEfficiency.h"
	42	#include "AliTRDtrackInfo/AliTRDtrackInfo.h"
	43
	44	ClassImp(AliTRDtrackingEfficiency)
	45
	46	//____________________________________________________________________
3d86166d	47	AliTRDtrackingEfficiency::AliTRDtrackingEfficiency()
b38d0daf	48	:AliTRDrecoTask("TrackingEff", "Barrel Tracking Efficiency")
814ecea4	49	,fMissed(0x0)
814ecea4	50	{
	51	//
	52	// Default constructor
	53	//
814ecea4	54	}
814ecea4	55
ed383798	56	//____________________________________________________________________
	57	AliTRDtrackingEfficiency::~AliTRDtrackingEfficiency()
	58	{
ed383798	59	if(fMissed){
	60	fMissed->Delete();
	61	delete fMissed;
	62	}
	63	}
	64
814ecea4	65	//____________________________________________________________________
	66	void AliTRDtrackingEfficiency::CreateOutputObjects()
	67	{
	68	//
	69	// Create output objects
	70	//
	71
	72	OpenFile(0, "RECREATE");
5d6dc395	73	const Int_t nbins = AliTRDCalPID::kNMom;
814ecea4	74	Float_t xbins[nbins+1] = {.5, .7, .9, 1.3, 1.7, 2.4, 3.5, 4.5, 5.5, 7., 9., 11.};
5d6dc395	75
5d6dc395	76	TH1 *h = 0x0;
3d86166d	77	fContainer = new TObjArray();
5d6dc395	78	for(Int_t is=0; is<AliPID::kSPECIES; is++){
	79	fContainer->Add(h = new TProfile(Form("h%s", AliTRDCalPID::GetPartSymb(is)), "", nbins, xbins));
	80	h->SetLineColor(AliTRDCalPID::GetPartColor(is));
	81	h->SetMarkerColor(AliTRDCalPID::GetPartColor(is));
	82	h->SetMarkerStyle(7);
	83	}
	84	fContainer->Add(h = new TProfile("h", "", nbins, xbins));
	85	h->SetMarkerStyle(7);
814ecea4	86	}
	87
	88	//____________________________________________________________________
	89	void AliTRDtrackingEfficiency::Exec(Option_t *)
	90	{
	91	//
	92	// Do it
	93	//
	94
5d6dc395	95	Int_t labelsacc[10000];
5d6dc395	96	memset(labelsacc, 0, sizeof(Int_t) * 10000);
814ecea4	97
	98	if(!fMissed){
	99	fMissed = new TClonesArray("AliTRDtrackInfo", 10);
	100	fMissed->SetOwner();
	101	}
	102
	103	Float_t mom;
	104	Int_t selection[10000], nselect = 0;
5d6dc395	105	ULong_t status; Int_t pidx;
814ecea4	106	Int_t nTRD = 0, nTPC = 0, nMiss = 0;
	107	AliTRDtrackInfo *track = 0x0;
	108	AliTrackReference *ref = 0x0;
	109	AliExternalTrackParam *esd = 0x0;
	110	for(Int_t itrk=0; itrk<fTracks->GetEntriesFast(); itrk++){
	111	track = (AliTRDtrackInfo*)fTracks->UncheckedAt(itrk);
5d6dc395	112
814ecea4	113	if(!track->HasESDtrack()) continue;
814ecea4	114	status = track->GetStatus();
814ecea4	115
5d6dc395	116	// missing TPC propagation - interesting for SA
	117	if(!(status&AliESDtrack::kTPCout)) continue;
	118
	119	// missing MC info.
	120	if(HasMCdata() && track->GetNTrackRefs() <= 1) continue;
	121
814ecea4	122	nTPC++;
814ecea4	123	selection[nselect++]=itrk;
5d6dc395	124	ref = track->GetTrackRef(0);
b38d0daf	125	esd = track->GetESDinfo()->GetOuterParam();
5d6dc395	126	mom = ref ? ref->P(): esd->P();
	127	pidx = AliTRDCalPID::GetPartIndex(track->GetPDG());
	128	pidx = TMath::Max(pidx, 0);
814ecea4	129
	130	//Int_t n = track->GetNumberOfClusters();
	131	// where are this tracklets ???
	132	//if(ncls0 > ncls1) printf("%3d ESD[%3d] TRD[%3d\|%3d]\n", itrk, ncls0, ncls1, n);
	133	if(track->GetNumberOfClustersRefit()){
5d6dc395	134	((TProfile*)fContainer->At(pidx))->Fill(mom, 1.);
814ecea4	135	labelsacc[nTRD] = track->GetLabel();
	136	nTRD++;
	137	continue;
	138	}
	139
	140
814ecea4	141
	142	Float_t xmed, xleng;
	143	Int_t iref = 1; Bool_t found = kFALSE;
	144	while((ref = track->GetTrackRef(iref))){
	145	xmed = .5*(ref->LocalX() + track->GetTrackRef(iref-1)->LocalX());
	146	xleng= (ref->LocalX() - track->GetTrackRef(iref-1)->LocalX());
	147	if(TMath::Abs(xmed - 298.5) < .5 &&
	148	TMath::Abs(xleng - 3.7) < .1){
	149	found = kTRUE;
	150	break;
	151	}
	152	iref++;
	153	}
	154	if(!found){
	155	nTPC--;
	156	// track missing first layer. Maybe interesting for SA.
	157	continue;
	158	}
	159	nselect--;
	160	new ((fMissed)[nMiss]) AliTRDtrackInfo(track);
	161	nMiss++;
	162	}
5d6dc395	163	if(fDebugLevel>=2) printf("%3d Tracks: ESD[%3d] TPC[%3d] TRD[%3d \| %5.2f%%] Off[%d]\n", (Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), fTracks->GetEntriesFast(), nTPC, nTRD, nTPC ? 1.E2*nTRD/float(nTPC) : 0., fMissed->GetEntriesFast());
814ecea4	164
	165
	166	// Find double tracks
	167	Float_t threshold = 10.;
	168	AliTrackReference *refMiss = 0x0;
	169	AliExternalTrackParam *op = 0x0;
	170	AliTRDtrackInfo *tt = 0x0;
	171	for(Int_t imiss=0; imiss<nMiss; imiss++){
	172	//printf("Searching missing %d ...\n", imiss);
	173
	174	// get outer param of missed
	175	tt = (AliTRDtrackInfo*)fMissed->UncheckedAt(imiss);
b38d0daf	176	op = tt->GetESDinfo()->GetOuterParam();
814ecea4	177	Double_t alpha = op->GetAlpha(), cosa = TMath::Cos(alpha), sina = TMath::Sin(alpha);
	178
	179	Double_t xyz[3], x0, y0, z0, x, y, z, dx, dy, dz, d;
	180
	181	Bool_t FOUND = kFALSE;
	182	for(Int_t iselect=0; iselect<nselect; iselect++){
	183	track = (AliTRDtrackInfo*)fTracks->UncheckedAt(selection[iselect]);
	184
	185	// check first MC ... if available
	186	d = 0;
	187	for(Int_t iref=0; iref<track->GetNTrackRefs(); iref++){
	188	if(!(ref = track->GetTrackRef(iref))) continue;
	189	if((refMiss = tt->GetTrackRef(iref))){
	190	dy = ref->LocalY() - refMiss->LocalY();
	191	dz = ref->Z() - refMiss->Z();
	192	} else {
	193	// compare missOP with refTrackRef in LTC
	194	x0 = ref->LocalX();
	195	op->GetYAt(x0, AliTracker::GetBz(), y0);
	196	op->GetZAt(x0, AliTracker::GetBz(), z0);
	197	dy = y0 - ref->LocalY();
	198	dz = z0 - ref->Z();
	199	}
	200	d += (dydy + dzdz);
	201	}
	202	//printf("\td[%d] = %f N[%d]\n", selection[iselect], d, track->GetNTrackRefs());
	203	if((track->GetNTrackRefs())){
	204	d /= track->GetNTrackRefs();
	205	if(d < threshold){
	206	//printf("\t\tFound %2d in ref[%3d] : d[%f]\n", imiss, selection[iselect], d/track->GetNTrackRefs());
	207	FOUND = kTRUE; break;
	208	}
	209	}
	210
	211	// process outer param ... always available
	212	// compare missOP with OP in GTC
b38d0daf	213	esd = track->GetESDinfo()->GetOuterParam();
814ecea4	214	esd->GetXYZ(xyz);
	215	x0 = esd->GetX();
	216	op->GetYAt(x0, AliTracker::GetBz(), y0);
	217	op->GetZAt(x0, AliTracker::GetBz(), z0);
	218	x = x0cosa - y0sina;
	219	y = x0sina + y0cosa;
	220	z = z0;
	221	dx=xyz[0]-x;
	222	dy=xyz[1]-y;
	223	dz=xyz[2]-z;
	224	d = dxdx+dydy+dz*dz;
	225	//printf("\td[%d] = %f op\n", selection[iselect], d);
	226	if(d < threshold){
	227	//printf("\t\tFound %2d in op[%3d] : d[%f] dx[%5.2f] dy[%5.2f] dz[%5.2f]\n", imiss, selection[iselect], d, dx, dy, dz);
	228	FOUND = kTRUE; break;
	229	}
	230	}
	231	if(FOUND) nTPC--;
	232	else{
	233	ref = tt->GetTrackRef(0);
	234	mom = ref ? ref->P(): op->P();
5d6dc395	235	pidx = AliTRDCalPID::GetPartIndex(tt->GetPDG());
	236	pidx = TMath::Max(pidx, 0);
	237	((TProfile*)fContainer->At(pidx))->Fill(mom, 0.);
	238	if(fDebugLevel>=2) printf("\tNOT FOUND Id[%d] Mom[%f]\n", tt->GetTrackId(), mom);
814ecea4	239	}
	240	}
	241
5d6dc395	242	if(fDebugLevel>=2) printf("%3d Tracks: ESD[%3d] TPC[%3d] TRD[%3d \| %5.2f%%] Off[%d]\n", (Int_t)AliAnalysisManager::GetAnalysisManager()->GetCurrentEntry(), fTracks->GetEntriesFast(), nTPC, nTRD, nTPC ? 1.E2*nTRD/float(nTPC) : 0., fMissed->GetEntriesFast());
814ecea4	243
	244	//fMissed->Delete();
	245	// check for double countings
	246	Int_t indices[10000]; memset(indices, 0, sizeof(Int_t) * 10000);
	247	TMath::Sort(nTRD, labelsacc, indices);
	248	if(fDebugLevel > 2){
	249	for(Int_t itk = 0; itk < nTRD - 1; itk++)
	250	if(labelsacc[indices[itk]] ==labelsacc[indices[itk + 1]]) printf("Double counted MC track: %d\n", labelsacc[indices[itk]]);
	251	}
3d86166d	252	PostData(0, fContainer);
814ecea4	253	}
	254
	255	//____________________________________________________________________
	256	void AliTRDtrackingEfficiency::Terminate(Option_t *)
	257	{
	258	//
	259	// Terminate
	260	//
	261
3d86166d	262	if(fDebugStream){
	263	delete fDebugStream;
	264	fDebugStream = 0x0;
	265	fDebugLevel = 0;
	266	}
ed383798	267
3d86166d	268	fContainer = dynamic_cast<TObjArray*>(GetOutputData(0));
3d86166d	269	if (!fContainer) {
814ecea4	270	Printf("ERROR: list not available");
	271	return;
	272	}
	273
	274	}
	275
5d6dc395	276
	277
	278	//____________________________________________________________________
e15179be	279	Bool_t AliTRDtrackingEfficiency::GetRefFigure(Int_t ifig)
5d6dc395	280	{
a391a274	281	Bool_t FIRST = kTRUE;
a391a274	282	TProfile *h = 0x0;
5d6dc395	283	switch(ifig){
5d6dc395	284	case 0:
a391a274	285	h = (TProfile*)fContainer->At(AliPID::kSPECIES);
	286	for(Int_t is=0; is<AliPID::kSPECIES; is++){
	287	h->Add((TProfile*)fContainer->At(is));
	288	}
	289	h->SetMarkerColor(kBlack);
	290	h->SetLineColor(kBlack);
	291	h->GetXaxis()->SetTitle("p [GeV/c]");
	292	h->Draw("e1");
5d6dc395	293	break;
5d6dc395	294	case 1:
a391a274	295	FIRST = kTRUE;
	296	for(Int_t is=0; is<AliPID::kSPECIES; is++){
	297	if(!(h = (TProfile*)fContainer->At(is))) continue;
	298	if(FIRST){
	299	h->Draw("e1");
	300	h->GetXaxis()->SetTitle("p [GeV/c]");
	301	} else h->Draw("same e1");
	302	FIRST = kFALSE;
	303	}
5d6dc395	304	break;
5d6dc395	305	}
e15179be	306	return kTRUE;
5d6dc395	307	}
	308
	309
	310	//____________________________________________________________________
	311	Bool_t AliTRDtrackingEfficiency::PostProcess()
	312	{
5d6dc395	313	fNRefFigures = HasMCdata() ? 2 : 1;
	314	return kTRUE;
	315	}