trigger names changed (Markus)
[u/mrichter/AliRoot.git] / PWG1 / TRD / AliTRDalignmentTask.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-commercialf 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 ///////////////////////////////////////////////
17 // TRD alignment task
18 //
19 // Some documentation
20 //
21 // Authors :
22 //
23 ////////////////////////////////////////////////
24
25 #include "TTreeStream.h"
26 #include "TROOT.h"
27 #include "TH1.h"
28 #include "TH2.h"
29
30 #include "AliTrackPointArray.h"
31 #include "AliLog.h"
32
33 #include "AliTRDgeometry.h"
34 #include "AliTRDcluster.h"
35 #include "AliTRDseedV1.h"
36 #include "AliTRDtrackV1.h"
37
38 #include "AliTRDalignmentTask.h"
39
40 ClassImp(AliTRDalignmentTask)
41
42 //________________________________________________________
43 AliTRDalignmentTask::AliTRDalignmentTask()
44   :AliTRDrecoTask()
45   ,fTree(0x0)
46   ,fArray(0x0)
47 {
48   InitFunctorList();
49 }
50
51 AliTRDalignmentTask::AliTRDalignmentTask(char* name)
52   :AliTRDrecoTask(name, "TRD alignment")
53   ,fTree(0x0)
54   ,fArray(0x0)
55 {
56   InitFunctorList();
57   DefineOutput(2, TTree::Class());
58 }
59
60 //________________________________________________________
61 AliTRDalignmentTask::~AliTRDalignmentTask()
62 {
63   if (fArray) delete fArray;
64 }
65
66
67 //________________________________________________________
68 void AliTRDalignmentTask::UserCreateOutputObjects()
69 {
70   // spatial resolution
71
72   if(!HasFunctorList()) InitFunctorList();
73   OpenFile(2, "RECREATE");
74
75   fTree = new TTree("spTree", "Tree with track space point arrays");
76   fTree->Branch("SP","AliTrackPointArray", &fArray);
77 }
78
79
80 //________________________________________________________
81 void AliTRDalignmentTask::UserExec(Option_t *opt)
82 {
83 // Documentation to come
84
85   AliTRDrecoTask::UserExec(opt);
86   PostData(2, fTree);
87 }
88
89
90 //________________________________________________________
91 TH1* AliTRDalignmentTask::PlotTrackPoints(const AliTRDtrackV1 *track)
92 {
93 // Documentation to come 
94
95   if(track) fkTrack = track;
96   if(!fkTrack){
97     AliWarning("No Track defined.");
98     return 0x0;
99   }
100
101   if (fArray) delete fArray;
102   fArray = new AliTrackPointArray(fkTrack->GetNumberOfTracklets());
103
104   // Filling the track points array
105   Float_t x, y, z;
106   AliTrackPoint p; Int_t ip = 0;
107   AliTRDseedV1 *tracklet = 0x0;
108   for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++){
109     if(!(tracklet = fkTrack->GetTracklet(il))) continue;
110     if(!tracklet->IsOK()) continue;
111     tracklet->Fit(kTRUE);
112
113     x = tracklet->GetX0();
114     y = tracklet->GetYfit(0)-tracklet->GetYfit(1)*(tracklet->GetX0()-x);
115     z = tracklet->GetZfit(0);
116     p.SetXYZ(x,y,z);
117     fArray->AddPoint(ip++, &p);
118   }
119   fTree->Fill();
120
121 /*  if(fDebugLevel>=1){
122     Float_t yt = fRim.GetYat(x[il]);
123     (*fDebugStream) << "TrkltResiduals"
124       << "layer="  << il
125       << "x="      <<x[il]
126       << "y="      <<y[il]
127       << "yt="     <<yt
128       << "dydx="   << dydx[il]
129       << "dy="     << dy
130       << "\n";
131   }*/
132   return 0x0;
133 }
134
135
136
137 //________________________________________________________
138 Bool_t AliTRDalignmentTask::IsIdenticalWithOneOf(AliTrackPoint * const p, AliTrackPointArray *parray, int nmax) {
139
140   // Is the point p identical with one of the points on the list parray?
141   // This is a fix for aliroot 4-16-Rev-01 (and before) writing some 
142   // spurious unitialized points. 
143  
144   for (int i=0; i<parray->GetNPoints() && i<nmax; i++) {
145     AliTrackPoint pa;
146     parray->GetPoint(pa,i);
147     //printf("comparing %7.3f with %7.3f\n",p->GetY(),pa.GetY());
148     if (p->GetResidual(pa,0)<1e-8) return kTRUE;
149     //printf("different\n");
150   }
151   return kFALSE;
152 }