TPC/Rec/AliTPCReconstructor.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 //          Options for the TPC Reconstruction in rec.C
  20 //
  21 //  4 options can be set to change the input for TPC reconstruction
  22 //  which overwrites the usage of fUseHLTClusters of the AliTPCRecoParam
  23 //
  24 //  1) useRAW        - use RAW, if not present -> do nothing
  25 //  2) useRAWorHLT   - use RAW, if not present -> use HLT clusters
  26 //  3) useHLT        - use HLT clusters, if not present -> do nothing
  27 //  4) useHLTorRAW   - use HLT clusters, if not present -> use RAW
  28 //
  29 //  -> The current default is useHLTorRAW
  30 //--------------------------------------------------------------------
  31
  32 ///////////////////////////////////////////////////////////////////////////////
  33 //                                                                           //
  34 // class for TPC reconstruction                                              //
  35 //                                                                           //
  36 ///////////////////////////////////////////////////////////////////////////////
  37
  38 #include <TObject.h>
  39 #include <TString.h>
  40 #include <TObjString.h>
  41 #include <TObjArray.h>
  42 #include <TFile.h>
  43
  44 #include <AliLog.h>
  45 #include <AliPID.h>
  46 #include <AliESDpid.h>
  47 #include <AliTPCPIDResponse.h>
  48 #include "AliTPCReconstructor.h"
  49 #include "AliRunLoader.h"
  50 #include "AliRun.h"
  51 #include "AliRawReader.h"
  52 #include "AliTPCclusterer.h"
  53 #include "AliTPCtracker.h"
  54 #include "AliTPCParam.h"
  55 #include "AliTPCParamSR.h"
  56 #include "AliTPCcalibDB.h"
  57 #include "AliTracker.h"
  58 #include "AliMagF.h"
  59
  60 ClassImp(AliTPCReconstructor)
  61
  62
  63 Int_t    AliTPCReconstructor::fgStreamLevel     = 0;        // stream (debug) level
  64 AliTPCAltroEmulator *  AliTPCReconstructor::fAltroEmulator=0;    // ALTRO emulator
  65
  66 AliTPCReconstructor::AliTPCReconstructor():
  67 AliReconstructor(),
  68 fClusterer(NULL),
  69 fArrSplines(NULL)
  70 {
  71   //
  72   // default constructor
  73   //
  74   //
  75   //
  76   AliTPCcalibDB * calib = AliTPCcalibDB::Instance();
  77   const AliMagF * field = (AliMagF*)TGeoGlobalMagField::Instance()->GetField();
  78   calib->SetExBField(field);
  79   AliTPCParam* param = GetTPCParam();
  80   if (!param) {
  81     AliWarning("Loading default TPC parameters !");
  82     param = new AliTPCParamSR;
  83   }
  84   fClusterer = new AliTPCclusterer(param);
  85 }
  86
  87 AliTPCReconstructor::AliTPCReconstructor(const AliTPCReconstructor& /*rec*/):
  88 AliReconstructor(),
  89 fClusterer(NULL),
  90 fArrSplines(NULL)
  91 {
  92   //
  93   // Dummy copu constructor
  94   //
  95 }
  96
  97 AliTPCReconstructor& AliTPCReconstructor::operator=(const AliTPCReconstructor&){
  98   //
  99   // dummy operator
 100   //
 101   return *this;
 102 }
 103
 104 //_____________________________________________________________________________
 105 AliTPCReconstructor::~AliTPCReconstructor()
 106 {
 107   if (fClusterer)   delete fClusterer;
 108   delete fArrSplines;
 109   delete AliTPCcalibDB::Instance();
 110 }
 111
 112 //_____________________________________________________________________________
 113 void AliTPCReconstructor::Reconstruct(TTree* digitsTree, TTree* clustersTree) const {
 114   // single event local reconstruction
 115   // of TPC data
 116   fClusterer->SetInput(digitsTree);
 117   fClusterer->SetOutput(clustersTree);
 118   fClusterer->Digits2Clusters();
 119 }
 120
 121 //_____________________________________________________________________________
 122 void AliTPCReconstructor::Reconstruct(AliRawReader* rawReader, TTree* clustersTree) const {
 123   // single event local reconstruction
 124   // of TPC data starting from raw data
 125
 126   fClusterer->SetOutput(clustersTree);
 127   fClusterer->Digits2Clusters(rawReader);
 128 }
 129
 130 //_____________________________________________________________________________
 131 AliTracker* AliTPCReconstructor::CreateTracker() const
 132 {
 133 // create a TPC tracker
 134
 135   AliTPCParam* param = GetTPCParam();
 136   if (!param) {
 137     AliWarning("Loading default TPC parameters !");
 138     param = new AliTPCParamSR;
 139   }
 140   param->ReadGeoMatrices();
 141
 142   AliTPCtracker* tracker = new AliTPCtracker(param);
 143
 144   ParseOptions(tracker);
 145
 146   return tracker;
 147 }
 148
 149 //_____________________________________________________________________________
 150 void AliTPCReconstructor::FillESD(TTree */*digitsTree*/, TTree */*clustersTree*/,
 151                                   AliESDEvent* /*esd*/) const
 152 {
 153 // make PID
 154 /*  Now done in AliESDpid
 155   Double_t parTPC[] = {50., 0.07, 5.};  // MIP nnormalized to channel 50 -MI
 156   AliTPCpidESD tpcPID(parTPC);
 157   tpcPID.MakePID(esd);
 158 */
 159 }
 160
 161
 162 //_____________________________________________________________________________
 163 AliTPCParam* AliTPCReconstructor::GetTPCParam() const
 164 {
 165 // get the TPC parameters
 166
 167   AliTPCParam* param = AliTPCcalibDB::Instance()->GetParameters();
 168
 169   return param;
 170 }
 171
 172 //_____________________________________________________________________________
 173 void AliTPCReconstructor::SetSplinesFromOADB(const char* tmplt, AliESDpid *esdPID)
 174 {
 175   //
 176   //  load splines from the OADB using 'template'
 177   //
 178
 179   // only load splines if not already set
 180   if (!fArrSplines) {
 181     fArrSplines=new TObjArray(Int_t(AliPID::kSPECIES));
 182     fArrSplines->SetOwner();
 183     TString stemplate(tmplt);
 184
 185     TString fileNamePIDresponse("$ALICE_ROOT/OADB/COMMON/PID/data/TPCPIDResponse.root");
 186     TFile f(fileNamePIDresponse.Data());
 187
 188     TObjArray *arrPidResponseMaster=0x0;
 189
 190     if (f.IsOpen() && !f.IsZombie()){
 191       arrPidResponseMaster=dynamic_cast<TObjArray*>(f.Get("TPCPIDResponse"));
 192     }
 193     f.Close();
 194
 195     if (!arrPidResponseMaster){
 196       AliError("PID response array not found, cannot assign proper splines");
 197       return;
 198     }
 199
 200     for (Int_t ispec=0; ispec<AliPID::kSPECIES; ++ispec)
 201     {
 202       Int_t ispec2=ispec;
 203       if (ispec==Int_t(AliPID::kMuon)) ispec2=Int_t(AliPID::kPion);
 204
 205       TString particle=AliPID::ParticleName(ispec2);
 206       particle.ToUpper();
 207
 208       TString splineName;
 209       splineName.Form(stemplate.Data(),particle.Data());
 210       TObject *spline=arrPidResponseMaster->FindObject(splineName.Data());
 211       if (!spline) {
 212         AliError(Form("No spline found for '%s'", splineName.Data()));
 213         continue;
 214       };
 215       AliInfo(Form("Adding Response function %d:%s",ispec,splineName.Data()));
 216       fArrSplines->AddAt(spline->Clone(), ispec);
 217     }
 218     arrPidResponseMaster->Delete();
 219     delete arrPidResponseMaster;
 220     if (fArrSplines->GetEntries()!=Int_t(AliPID::kSPECIES)) {
 221       AliError("Splines not found for all species, cannot use proper PID");
 222       delete fArrSplines;
 223       fArrSplines=NULL;
 224       return;
 225     }
 226   }
 227
 228   for (Int_t ispec=0; ispec<AliPID::kSPECIES; ++ispec)
 229   {
 230     esdPID->GetTPCResponse().SetResponseFunction( (AliPID::EParticleType)ispec, fArrSplines->UncheckedAt(ispec) );
 231   }
 232
 233   esdPID->GetTPCResponse().SetUseDatabase(kTRUE);
 234 }
 235
 236 //_____________________________________________________________________________
 237 void AliTPCReconstructor::GetPidSettings(AliESDpid *esdPID)
 238 {
 239   //
 240   // Get TPC pid splines. They should be written to the OCDB during the CPass
 241   // the splines themselves are owned by the OCDB object
 242   //
 243
 244   // parse options
 245   TString allopt(GetOption());
 246   TObjArray *optArray=allopt.Tokenize(";");
 247
 248   // defines whether the pid was set via a specific option in the rec.C
 249   Bool_t pidSetInOptions = kFALSE;
 250
 251   for (Int_t iopt=0; iopt<optArray->GetEntriesFast(); ++iopt){
 252     if (!optArray->At(iopt)) continue;
 253     TString option(static_cast<TObjString*>(optArray->At(iopt))->GetString().Strip(TString::kBoth,' '));
 254
 255     if (!option.BeginsWith("PID.")) continue;
 256
 257     // remove 'PID.' identifyer
 258     option.Remove(0,4);
 259
 260     // parse PID type
 261     if (option.BeginsWith("Static=")){
 262       option.Remove(0,option.First('=')+1);
 263       if (option.Contains("LHC13b2_fix_PID")) {
 264         esdPID->GetTPCResponse().SetBetheBlochParameters(0.0320981, 19.9768, 2.52666e-16, 2.72123, 6.08092);
 265         esdPID->GetTPCResponse().SetMip(53.4968);
 266         pidSetInOptions=kTRUE;
 267       }
 268
 269     } else if (option.BeginsWith("OADB=")) {
 270       option.Remove(0,option.First('=')+1);
 271       AliInfo(Form("Setting splines From OADB using template: '%s'",option.Data()));
 272       SetSplinesFromOADB(option, esdPID);
 273       pidSetInOptions=kTRUE;
 274     } else if (option.BeginsWith("OCDB=")){
 275       option.Remove(0,option.First('=')+1);
 276       // not yet implemented
 277     }
 278
 279   }
 280
 281   delete optArray;
 282
 283   //
 284   // Initialisation of BB parameters from the OCDB.
 285   // They are stored in the AliTPCParam
 286   //
 287   if (!pidSetInOptions) {
 288     AliTPCParam* param = AliTPCcalibDB::Instance()->GetParameters();
 289     if (param) {
 290       TVectorD *paramBB=param->GetBetheBlochParameters();
 291       if (paramBB){
 292         esdPID->GetTPCResponse().SetBetheBlochParameters((*paramBB)(0),(*paramBB)(1),(*paramBB)(2),(*paramBB)(3),(*paramBB)(4));
 293         AliInfo(Form("Setting BB parameters from OCDB (AliTPCParam): %.2g, %.2g, %.2g, %.2g, %.2g",
 294                      (*paramBB)(0),(*paramBB)(1),(*paramBB)(2),(*paramBB)(3),(*paramBB)(4)));
 295       } else {
 296         AliError("Couldn't get BB parameters from OCDB, the old default values will be used instead");
 297       }
 298     } else {
 299       AliError("Couldn't get TPC parameters");
 300     }
 301   }
 302
 303 /*
 304   AliTPCcalibDB * calib = AliTPCcalibDB::Instance();
 305
 306   //Get pid splines array
 307   TObjArray *arrSplines=calib->GetPidResponse();
 308   if (!arrSplines) return;
 309   AliTPCPIDResponse &tpcPID=esdPID->GetTPCResponse();
 310   tpcPID.SetUseDatabase(kTRUE);
 311
 312   // check if parametrisations are already set.
 313   // since this is uniq for one run, we don't have to reload them
 314   if (tpcPID.GetResponseFunction(AliPID::kPion)) return;
 315
 316   // get the default object
 317   TObject *defaultPID=arrSplines->At(AliPID::kUnknown);
 318
 319   // loop over all particle species and set the response functions
 320   for (Int_t ispec=0; ispec<AliPID::kUnknown; ++ispec){
 321     TObject *pidSpline=arrSplines->At(ispec);
 322     if (!pidSpline) pidSpline=defaultPID;
 323     tpcPID.SetResponseFunction((AliPID::EParticleType)ispec,pidSpline);
 324   }
 325  */
 326 }
 327
 328 //_____________________________________________________________________________
 329 void AliTPCReconstructor::ParseOptions( AliTPCtracker* tracker ) const
 330 {
 331 // parse options from rec.C and set in clusterer and tracker
 332
 333   TString option = GetOption();
 334
 335   Int_t useHLTClusters = 3;
 336
 337   if (option.Contains("use")) {
 338
 339     AliInfo(Form("Overide TPC RecoParam with option %s",option.Data()));
 340
 341     if (option.Contains("useRAW")) {
 342       useHLTClusters = 1;
 343       if (option.Contains("useRAWorHLT"))
 344         useHLTClusters = 2;
 345     }
 346     else if (option.Contains("useHLT")) {
 347       useHLTClusters = 3;
 348       if (option.Contains("useHLTorRAW"))
 349         useHLTClusters = 4;
 350     }
 351   }
 352   else {
 353     const AliTPCRecoParam* param = GetRecoParam();
 354     useHLTClusters = param->GetUseHLTClusters();
 355   }
 356
 357   AliInfo(Form("Usage of HLT clusters in TPC reconstruction : %d", useHLTClusters));
 358
 359   fClusterer->SetUseHLTClusters(useHLTClusters);
 360   tracker->SetUseHLTClusters(useHLTClusters);
 361
 362   return;
 363 }