PWG3/hfe/AliHFEpidQAmanager.cxx

   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$ */
  17
  18 //
  19 // Class AliHFEpidQAmanager
  20 // Does steering of the PID QA. The PID QA manager is initialized according
  21 // to the configuration used for PID. It contains PID objects inheriting
  22 // from AliHFEdetPIDqa, which are
  23 //     AliHFEtpcPIDqa
  24 //     AliHFEtrdPIDqaV1
  25 //     AliHFEtofPIDqa
  26 // PID QA objects are filled for every detector before PID decision and
  27 // after PID decision for tracks which are selected by the given detector
  28 // as electron candidates.
  29 //
  30 // Author
  31 //   Markus Fasel <M.Fasel@gsi.de>
  32 //
  33 #include <TList.h>
  34
  35 #include "AliAODpidUtil.h"
  36 #include "AliESDpid.h"
  37 #include "AliVParticle.h"
  38
  39 #include "AliHFEtpcPIDqa.h"
  40 #include "AliHFEtrdPIDqaV1.h"
  41 #include "AliHFEtofPIDqa.h"
  42 #include "AliHFEpidQAmanager.h"
  43
  44 ClassImp(AliHFEpidQAmanager)
  45
  46 //____________________________________________________________
  47 AliHFEpidQAmanager::AliHFEpidQAmanager():
  48   TObject()
  49 {
  50   //
  51   // Dummy constructor
  52   //
  53   memset(fDetPIDqa, 0, sizeof(AliHFEdetPIDqa *) * AliHFEpid::kNdetectorPID);
  54   memset(fDetPID, 0, sizeof(AliHFEdetPIDqa *) * AliHFEpid::kNdetectorPID);
  55   SetOwner();
  56 }
  57
  58 //____________________________________________________________
  59 AliHFEpidQAmanager::AliHFEpidQAmanager(const AliHFEpidQAmanager &ref):
  60   TObject(ref)
  61 {
  62   //
  63   // Copy constructor
  64   //
  65   ref.Copy(*this);
  66   SetOwner();
  67 }
  68
  69 //____________________________________________________________
  70 AliHFEpidQAmanager &AliHFEpidQAmanager::operator=(const AliHFEpidQAmanager &ref){
  71   //
  72   // Assignment operator
  73   //
  74   if(this != &ref) ref.Copy(*this);
  75   SetOwner();
  76   return *this;
  77 }
  78
  79 //____________________________________________________________
  80 void AliHFEpidQAmanager::Copy(TObject &o) const{
  81   //
  82   // Make copy
  83   //
  84   TObject::Copy(o);
  85   AliHFEpidQAmanager &target = dynamic_cast<AliHFEpidQAmanager &>(o);
  86
  87   for(Int_t idet = 0; idet < AliHFEpid::kNdetectorPID; idet++){
  88     target.fDetPID[idet] = fDetPID[idet];
  89     if(target.fDetPIDqa[idet]) delete target.fDetPIDqa[idet];
  90     if(fDetPIDqa[idet]) target.CreateDetPIDqa(static_cast<AliHFEpid::EDETtype_t>(idet));
  91   }
  92 }
  93
  94 //____________________________________________________________
  95 AliHFEpidQAmanager::~AliHFEpidQAmanager(){
  96   //
  97   // Destructor
  98   //
  99   if(IsOwner())
 100     for(Int_t idet = 0; idet < AliHFEpid::kNdetectorPID; idet++){
 101       if(fDetPIDqa[idet]) delete fDetPIDqa[idet];
 102     }
 103 }
 104
 105 //____________________________________________________________
 106 void AliHFEpidQAmanager::Initialize(AliHFEpid *pid){
 107   //
 108   // Initialize PID QA manager according to the detector
 109   // configuration used in the PID
 110   //
 111   for(Int_t idet = 0; idet < AliHFEpid::kNdetectorPID; idet++){
 112     // Initialize Array for detector PID for all detectors
 113     fDetPID[idet] = pid->GetDetPID(static_cast<AliHFEpid::EDETtype_t>(idet));
 114     if(pid->HasDetector(static_cast<AliHFEpid::EDETtype_t>(idet))){
 115       CreateDetPIDqa(static_cast<AliHFEpid::EDETtype_t>(idet));
 116       if(fDetPIDqa[idet]){
 117         fDetPIDqa[idet]->SetPIDqaManager(this);
 118         fDetPIDqa[idet]->Initialize();
 119       }
 120     }
 121   }
 122 }
 123
 124 //____________________________________________________________
 125 void AliHFEpidQAmanager::CreateDetPIDqa(AliHFEpid::EDETtype_t idet){
 126   //
 127   // Create new PID QA object
 128   //
 129   switch(idet){
 130     case AliHFEpid::kTPCpid:
 131           fDetPIDqa[idet] = new AliHFEtpcPIDqa("TPCQA");
 132           break;
 133     case AliHFEpid::kTRDpid:
 134           fDetPIDqa[idet] = new AliHFEtrdPIDqaV1("TRDQA");
 135           break;
 136     case AliHFEpid::kTOFpid:
 137           fDetPIDqa[idet] = new AliHFEtofPIDqa("TOFQA");
 138           break;
 139     default:
 140           break;
 141   };
 142 }
 143
 144 //____________________________________________________________
 145 void AliHFEpidQAmanager::ProcessTrack(const AliHFEpidObject *track, AliHFEpid::EDETtype_t det, AliHFEdetPIDqa::EStep_t step){
 146   //
 147   // Process single Track
 148   //
 149   if(!fDetPIDqa[det]){
 150     AliDebug(1, Form("QA for detector %d not available", det));
 151     return;
 152   }
 153   AliDebug(1, Form("Doing QA for detector %d\n", det));
 154   fDetPIDqa[det]->ProcessTrack(track, step);
 155 }
 156
 157 //____________________________________________________________
 158 TList *AliHFEpidQAmanager::MakeList(const Char_t *name){
 159   //
 160   // Make List of PID QA objects
 161   //
 162   TList *list = new TList;
 163   list->SetName(name);
 164   list->SetOwner();
 165   for(Int_t idet = 0; idet < AliHFEpid::kNdetectorPID; idet++){
 166     if(fDetPIDqa[idet]) list->Add(fDetPIDqa[idet]);
 167   }
 168   ReleaseOwnerShip();
 169   return list;
 170 }
 171