* provided "as is" without express or implied warranty. *
**************************************************************************/
-/*
-$Log$
-Revision 1.9 2001/10/21 18:30:02 hristov
-Several pointers were set to zero in the default constructors to avoid memory management problems
-
-Revision 1.8 2001/05/07 08:06:44 cblume
-Speedup of the code. Create only AliTRDcluster
-
-Revision 1.7 2001/03/30 14:40:14 cblume
-Update of the digitization parameter
-
-Revision 1.6 2000/11/01 14:53:20 cblume
-Merge with TRD-develop
-
-
-Revision 1.1.4.5 2000/10/15 23:40:01 cblume
-Remove AliTRDconst
-
-Revision 1.1.4.4 2000/10/06 16:49:46 cblume
-Made Getters const
-
-Revision 1.1.4.3 2000/10/04 16:34:58 cblume
-Replace include files by forward declarations
-
-Revision 1.1.4.2 2000/09/22 14:49:49 cblume
-Adapted to tracking code
-
-Revision 1.5 2000/10/02 21:28:19 fca
-Removal of useless dependecies via forward declarations
-
-Revision 1.4 2000/06/09 11:10:07 cblume
-Compiler warnings and coding conventions, next round
-
-Revision 1.3 2000/06/08 18:32:58 cblume
-Make code compliant to coding conventions
-
-Revision 1.2 2000/05/08 16:17:27 cblume
-Merge TRD-develop
-
-Revision 1.1.4.1 2000/05/08 15:08:03 cblume
-Remove the class AliTRDcluster
-
-Revision 1.4 2000/06/09 11:10:07 cblume
-Compiler warnings and coding conventions, next round
-
-Revision 1.3 2000/06/08 18:32:58 cblume
-Make code compliant to coding conventions
-
-Revision 1.2 2000/05/08 16:17:27 cblume
-Merge TRD-develop
-
-Revision 1.1.4.1 2000/05/08 15:08:03 cblume
-Remove the class AliTRDcluster
-
-Revision 1.1 2000/02/28 18:57:58 cblume
-Add new TRD classes
-
-*/
+/* $Id$ */
///////////////////////////////////////////////////////////////////////////////
// //
#include <TFile.h>
#include "AliRun.h"
-#include "AliTRD.h"
+#include "AliRunLoader.h"
+#include "AliLoader.h"
+
#include "AliTRDclusterizer.h"
#include "AliTRDcluster.h"
#include "AliTRDrecPoint.h"
#include "AliTRDgeometry.h"
+#include "AliTRDcalibDB.h"
ClassImp(AliTRDclusterizer)
// AliTRDclusterizer default constructor
//
- fInputFile = NULL;
- fOutputFile = NULL;
fClusterTree = NULL;
- fTRD = 0;
- fEvent = 0;
+ fRecPoints = 0;
fVerbose = 0;
}
// AliTRDclusterizer default constructor
//
- fInputFile = NULL;
- fOutputFile = NULL;
fClusterTree = NULL;
- fEvent = 0;
+ fRecPoints = 0;
fVerbose = 0;
}
//_____________________________________________________________________________
-AliTRDclusterizer::AliTRDclusterizer(const AliTRDclusterizer &c)
+AliTRDclusterizer::AliTRDclusterizer(const AliTRDclusterizer &c):TNamed(c)
{
//
// AliTRDclusterizer copy constructor
// AliTRDclusterizer destructor
//
- if (fInputFile) {
- fInputFile->Close();
- delete fInputFile;
+ if (fRecPoints) {
+ fRecPoints->Delete();
+ delete fRecPoints;
}
-
- if (fOutputFile) {
- fOutputFile->Close();
- delete fOutputFile;
- }
-
- if (fClusterTree) {
- delete fClusterTree;
- }
-
}
//_____________________________________________________________________________
}
//_____________________________________________________________________________
-void AliTRDclusterizer::Copy(TObject &c)
+void AliTRDclusterizer::Copy(TObject &c) const
{
//
// Copy function
//
- ((AliTRDclusterizer &) c).fInputFile = NULL;
- ((AliTRDclusterizer &) c).fOutputFile = NULL;
((AliTRDclusterizer &) c).fClusterTree = NULL;
- ((AliTRDclusterizer &) c).fEvent = 0;
+ ((AliTRDclusterizer &) c).fRecPoints = NULL;
((AliTRDclusterizer &) c).fVerbose = fVerbose;
}
//
// Opens the AliROOT file. Output and input are in the same file
//
-
- OpenInput(name,nEvent);
- OpenOutput(name);
-
+ TString evfoldname = AliConfig::GetDefaultEventFolderName();
+ fRunLoader = AliRunLoader::GetRunLoader(evfoldname);
+ if (!fRunLoader)
+ fRunLoader = AliRunLoader::Open(name);
+ if (!fRunLoader)
+ {
+ Error("Open","Can not open session for file %s.",name);
+ return kFALSE;
+ }
+
+ OpenInput(nEvent);
+ OpenOutput();
return kTRUE;
-
}
-//_____________________________________________________________________________
-Bool_t AliTRDclusterizer::Open(const Char_t *inname, const Char_t *outname
- , Int_t nEvent)
-{
- //
- // Opens the AliROOT file. Output and input are in different files
- //
-
- OpenInput(inname,nEvent);
- OpenOutput(outname);
-
- return kTRUE;
-
-}
//_____________________________________________________________________________
-Bool_t AliTRDclusterizer::OpenOutput(const Char_t *name)
+Bool_t AliTRDclusterizer::OpenOutput()
{
//
// Open the output file
//
- TDirectory *savedir = NULL;
-
- if (!fInputFile) return kFALSE;
-
- if (strcmp(name,fInputFile->GetName()) != 0) {
- savedir = gDirectory;
- printf("AliTRDclusterizer::OpenOutput -- ");
- printf("Open the output file %s.\n",name);
- fOutputFile = new TFile(name,"RECREATE");
- }
-
- // Create a tree for the cluster
- Char_t treeName[12];
- sprintf(treeName,"TreeR%d_TRD",fEvent);
- fClusterTree = new TTree(treeName,"TRD cluster");
TObjArray *ioArray = 0;
- fClusterTree->BranchOld("TRDcluster","TObjArray",&ioArray,32000,0);
- if (savedir) {
- savedir->cd();
- }
+ AliLoader* loader = fRunLoader->GetLoader("TRDLoader");
+ loader->MakeTree("R");
+ fClusterTree = loader->TreeR();
+ fClusterTree->Branch("TRDcluster","TObjArray",&ioArray,32000,0);
+
return kTRUE;
}
//_____________________________________________________________________________
-Bool_t AliTRDclusterizer::OpenInput(const Char_t *name, Int_t nEvent)
+Bool_t AliTRDclusterizer::OpenInput(Int_t nEvent)
{
//
// Opens a ROOT-file with TRD-hits and reads in the digits-tree
//
// Connect the AliRoot file containing Geometry, Kine, and Hits
- fInputFile = (TFile*) gROOT->GetListOfFiles()->FindObject(name);
- if (!fInputFile) {
- printf("AliTRDclusterizer::OpenInput -- ");
- printf("Open the ALIROOT-file %s.\n",name);
- fInputFile = new TFile(name,"UPDATE");
- }
- else {
- printf("AliTRDclusterizer::OpenInput -- ");
- printf("%s is already open.\n",name);
- }
+ if (fRunLoader->GetAliRun() == 0x0) fRunLoader->LoadgAlice();
+ gAlice = fRunLoader->GetAliRun();
- // Get AliRun object from file
- gAlice = (AliRun *) fInputFile->Get("gAlice");
if (!(gAlice)) {
- printf("AliTRDclusterizer::OpenInput -- ");
- printf("Could not find AliRun object.\n");
- return kFALSE;
+ fRunLoader->LoadgAlice();
+ gAlice = fRunLoader->GetAliRun();
+ if (!(gAlice)) {
+ printf("AliTRDclusterizer::OpenInput -- ");
+ printf("Could not find AliRun object.\n");
+ return kFALSE;
+ }
}
- fEvent = nEvent;
-
// Import the Trees for the event nEvent in the file
- Int_t nparticles = gAlice->GetEvent(fEvent);
- if (nparticles <= 0) {
- printf("AliTRDclusterizer::OpenInput -- ");
- printf("No entries in the trees for event %d.\n",fEvent);
- return kFALSE;
- }
-
- // Get the TRD object
- fTRD = (AliTRD*) gAlice->GetDetector("TRD");
- if (!fTRD) {
- printf("AliTRDclusterizer::OpenInput -- ");
- printf("No TRD detector object found\n");
- return kFALSE;
- }
-
+ fRunLoader->GetEvent(nEvent);
+
return kTRUE;
}
return kFALSE;
}
- TDirectory *savedir = gDirectory;
-
- if (fOutputFile) {
- fOutputFile->cd();
- }
TBranch *branch = fClusterTree->GetBranch("TRDcluster");
if (!branch) {
if ((det >= 0) && (det < AliTRDgeometry::Ndet())) {
- Int_t nRecPoints = fTRD->RecPoints()->GetEntriesFast();
+ Int_t nRecPoints = RecPoints()->GetEntriesFast();
TObjArray *detRecPoints = new TObjArray(400);
for (Int_t i = 0; i < nRecPoints; i++) {
- AliTRDcluster *c = (AliTRDcluster *) fTRD->RecPoints()->UncheckedAt(i);
+ AliTRDcluster *c = (AliTRDcluster *) RecPoints()->UncheckedAt(i);
if (det == c->GetDetector()) {
detRecPoints->AddLast(c);
}
branch->SetAddress(&detRecPoints);
fClusterTree->Fill();
+ delete detRecPoints;
+
return kTRUE;
}
if (det == -1) {
- printf("AliTRDclusterizer::WriteClusters -- ");
- printf("Writing the cluster tree %-18s for event %d.\n"
- ,fClusterTree->GetName(),fEvent);
-
+ Info("WriteClusters","Writing the cluster tree %s for event %d."
+ ,fClusterTree->GetName(),fRunLoader->GetEventNumber());
+ /*
fClusterTree->Write();
-
+ AliTRDgeometry *geo = fTRD->GetGeometry();
+ geo->SetName("TRDgeometry");
+ geo->Write();
+ */
+ AliLoader* loader = fRunLoader->GetLoader("TRDLoader");
+ loader->WriteRecPoints("OVERWRITE");
+
return kTRUE;
}
-
- savedir->cd();
-
+ /*
+ AliLoader* loader = fRunLoader->GetLoader("TRDLoader");
+ loader->WriteDigits("OVERWRITE");
+ */
printf("AliTRDclusterizer::WriteClusters -- ");
printf("Unexpected detector index %d.\n",det);
return kFALSE;
+
+}
+
+//_____________________________________________________________________________
+AliTRDcluster* AliTRDclusterizer::AddCluster(Double_t *pos, Int_t timebin, Int_t det, Double_t amp
+ , Int_t *tracks, Double_t *sig, Int_t iType, Float_t center)
+{
+ //
+ // Add a cluster for the TRD
+ //
+
+ AliTRDcluster *c = new AliTRDcluster();
+
+ c->SetDetector(det);
+ c->AddTrackIndex(tracks);
+ c->SetQ(amp);
+ c->SetX(pos[2]);
+ c->SetY(pos[0]);
+ c->SetZ(pos[1]);
+ c->SetSigmaY2(sig[0]);
+ c->SetSigmaZ2(sig[1]);
+ c->SetLocalTimeBin(timebin);
+ c->SetCenter(center);
+ switch (iType) {
+ case 0:
+ c->Set2pad();
+ break;
+ case 1:
+ c->Set3pad();
+ break;
+ case 2:
+ c->Set4pad();
+ break;
+ case 3:
+ c->Set5pad();
+ break;
+ case 4:
+ c->SetLarge();
+ break;
+ };
+
+ RecPoints()->Add(c);
+ return c;
}
+//_____________________________________________________________________________
+Double_t AliTRDclusterizer::CalcXposFromTimebin(Float_t timebin, Int_t idet, Int_t col, Int_t row)
+{
+ //
+ // Calculates the local x position in the detector from the timebin, depends on the drift velocity
+ // and t0
+ //
+
+ AliTRDcalibDB* calibration = AliTRDcalibDB::Instance();
+ if (!calibration)
+ return -1;
+ Float_t vdrift = calibration->GetVdrift(idet, col, row);
+ Float_t t0 = calibration->GetT0(idet, col, row);
+ Float_t samplingFrequency = calibration->GetSamplingFrequency();
+ timebin -= t0;
+
+ return timebin / samplingFrequency * vdrift;
+}