* full copyright notice. *
**************************************************************************/
+#include "TROOT.h"
+#include "TStyle.h"
#include "TVector.h"
#include "TLinearFitter.h"
+#include "TCanvas.h"
+#include "TGeoMatrix.h"
+#include "TGeoManager.h"
+
#include "TEveTrans.h"
+#include "TEveManager.h"
+
+#include "EveBase/AliEveEventManager.h"
#include "AliEveTRDData.h"
#include "AliEveTRDModuleImp.h"
+#include "AliEveTRDLoader.h"
+#include "AliEveTRDLoaderImp.h"
+#include "AliGeomManager.h"
+#include "AliESDtrack.h"
#include "AliLog.h"
#include "AliPID.h"
#include "AliTrackPointArray.h"
+#include "AliRieman.h"
#include "AliTRDhit.h"
#include "AliTRDcluster.h"
#include "AliTRDseedV1.h"
+#include "AliTRDtrackletMCM.h"
+#include "AliTRDtrackletWord.h"
+#include "AliTRDmcmSim.h"
#include "AliTRDtrackV1.h"
#include "AliTRDtrackerV1.h"
#include "AliTRDpadPlane.h"
#include "AliTRDdigitsManager.h"
-#include "AliTRDdataArrayDigits.h"
+#include "AliTRDmcmSim.h"
+#include "AliTRDarrayADC.h"
#include "AliTRDSignalIndex.h"
#include "AliTRDgeometry.h"
#include "AliTRDtransform.h"
ClassImp(AliEveTRDClusters)
ClassImp(AliEveTRDTracklet)
ClassImp(AliEveTRDTrack)
+ClassImp(AliEveTRDTrackletOnline)
+ClassImp(AliEveTRDmcm)
///////////////////////////////////////////////////////////
///////////// AliEveTRDDigits /////////////////////
///////////////////////////////////////////////////////////
//______________________________________________________________________________
-AliEveTRDDigits::AliEveTRDDigits(AliEveTRDChamber *p) :
- TEveQuadSet("digits", ""), fParent(p), fBoxes(), fData()
+AliEveTRDDigits::AliEveTRDDigits(AliEveTRDChamber *p)
+ :TEveQuadSet("digits", "")
+ ,fParent(p)
{
// Constructor.
+ SetOwnIds(kTRUE);
+ gStyle->SetPalette(1, 0);
+ SetPalette(new TEveRGBAPalette(0, 512));
+ Reset(TEveQuadSet::kQT_RectangleYZ, kFALSE, 32);
}
//______________________________________________________________________________
// AliInfo(GetTitle());
}
-//______________________________________________________________________________
-void AliEveTRDDigits::ComputeRepresentation()
-{
- // Calculate digits representation according to user settings. The
- // user can set the following parameters:
- // - digits scale (log/lin)
- // - digits threshold
- // - digits apparence (quads/boxes)
-
- if(!fData.HasData()){
- return;
- }
-
- TEveQuadSet::Reset(TEveQuadSet::kQT_RectangleYZ, kTRUE, 64);
-
- Double_t scale, dy, dz;
- Int_t q, color;
- Int_t nrows = fData.GetNrow(),
- ncols = fData.GetNcol(),
- ntbs = fData.GetNtime(),
- det = fParent->GetID();
- Float_t threshold = fParent->GetDigitsThreshold();
-
- AliTRDtransform transform(det);
- AliTRDgeometry *geo = fParent->fGeo;
- AliTRDpadPlane *pp = geo->GetPadPlane(geo->GetLayer(det), geo->GetStack(det));
-
- // express position in tracking coordinates
- fData.Expand();
- for (Int_t ir = 0; ir < nrows; ir++) {
- dz = pp->GetRowSize(ir);
- for (Int_t ic = 0; ic < ncols; ic++) {
- dy = pp->GetColSize(ic);
- for (Int_t it = 0; it < ntbs; it++) {
- q = fData.GetDataUnchecked(ir, ic, it);
- if (q < threshold) continue;
-
- Double_t x[6] = {0., 0., Double_t(q), 0., 0., 0.};
- Int_t roc[3] = {ir, ic, 0};
- Bool_t out = kTRUE;
- transform.Transform(&x[0], &roc[0], UInt_t(it), out, 0);
-
- scale = q < 512 ? q/512. : 1.;
- color = 50+int(scale*50.);
-
- AddQuad(x[1]-.45*dy, x[2]-.5*dz*scale, x[0], .9*dy, dz*scale);
- QuadValue(q);
- QuadColor(color);
- QuadId(new TNamed(Form("Charge%d", q), "dummy title"));
- } // end time loop
- } // end col loop
- } // end row loop
- fData.Compress(1);
-
- // rotate to global coordinates
- //RefitPlex();
- TEveTrans& t = RefMainTrans();
- t.SetRotByAngles((geo->GetSector(det)+.5)*AliTRDgeometry::GetAlpha(), 0.,0.);
-}
-
//______________________________________________________________________________
void AliEveTRDDigits::SetData(AliTRDdigitsManager *digits)
{
// Set data source.
- Int_t det = fParent->GetID();
- AliTRDdataArrayDigits *data = digits->GetDigits(det);
- if(!data->HasData()) return;
+ Int_t det(fParent->GetID());
+ AliTRDarrayADC *data = digits->GetDigits(det);
+ if(!data->GetDim()) return;
data->Expand();
AliTRDSignalIndex *indexes = digits->GetIndexes(det);
if(!indexes->IsAllocated()) digits->BuildIndexes(det);
- if(!fData.HasData()) fData.Allocate(data->GetNrow(), data->GetNcol(), data->GetNtime());
- fData.Expand();
+ Double_t scale, dy, dz;
+ Int_t ly = AliTRDgeometry::GetLayer(det),
+ stk = AliTRDgeometry::GetStack(det),
+ sec = AliTRDgeometry::GetSector(det),
+ vid = AliGeomManager::LayerToVolUID(AliGeomManager::kTRD1 + ly, stk + AliTRDgeometry::Nstack() * sec);
+ SetNameTitle(Form("digits%03d", det), Form("D-%03d [%02d_%d_%d]", det, sec, stk, ly));
+ Short_t sig[7]={0,0,0,10,0,0,0};
- Int_t row, col, time, adc;
+ AliTRDtransform transform(det);
+ AliTRDpadPlane *pp(fParent->fGeo->GetPadPlane(ly, stk));
+
+ Int_t row, col;
+ AliTRDcluster c;
indexes->ResetCounters();
while (indexes->NextRCIndex(row, col)){
- indexes->ResetTbinCounter();
- while (indexes->NextTbinIndex(time)){
- if(data->IsPadCorrupted(row, col, time)){
- // we should mark this position
- break;
- }
- adc = data->GetData(row, col, time);
- if(adc <= 1) continue;
- fData.SetDataUnchecked(row, col, time, adc);
- //fIndex->AddIndexTBin(row,col,time);
- //printf("\tr[%d] c[%d] t[%d] ADC[%d]\n", row, col, time, adc);
+ dz = pp->GetRowSize(row);
+ dy = pp->GetColSize(col);
+ Short_t *const adc = data->GetDataAddress(row,col);
+ for (Int_t time(0); time<30; time++){
+ if(data->IsPadCorrupted(row, col, time)) break;
+ if(adc[time] <= 1) continue;
+ new (&c) AliTRDcluster(det, col, row, time, sig, vid);
+ transform.Transform(&c);
+
+ scale = adc[time] < 512 ? adc[time]/512. : 1.;
+ AddQuad(c.GetY()-0.5*dy, c.GetZ()-0.5*dz*scale, c.GetX(), dy*0.95, dz*scale);
+ QuadValue(Int_t(adc[time]));
+ QuadId(new TNamed(Form("ADC %d", adc[time]), Form("det[%3d(%02d_%d_%d)] col[%3d] row[%2d] tb[%2d]", det, sec, stk, ly, col, row, time)));
}
}
- fData.Compress(1);
-}
-
-//______________________________________________________________________________
-void AliEveTRDDigits::Paint(Option_t *option)
-{
- // Paint the object.
-
- if(fParent->GetDigitsBox()) fBoxes.Paint(option);
- else TEveQuadSet::Paint(option);
+ // rotate to global coordinates
+ RefitPlex();
+ TEveTrans& t = RefMainTrans();
+ t.SetRotByAngles((sec+.5)*AliTRDgeometry::GetAlpha(), 0.,0.);
}
-//______________________________________________________________________________
-void AliEveTRDDigits::Reset()
-{
- // Reset raw and visual data.
- TEveQuadSet::Reset(TEveQuadSet::kQT_RectangleYZ, kTRUE, 64);
- // MT fBoxes.fBoxes.clear();
- fData.Reset();
-}
+// //______________________________________________________________________________
+// void AliEveTRDDigits::Paint(Option_t *option)
+// {
+// // Paint the object.
+//
+// if(fParent->GetDigitsBox()) fBoxes.Paint(option);
+// else TEveQuadSet::Paint(option);
+// }
+
+// //______________________________________________________________________________
+// void AliEveTRDDigits::Reset()
+// {
+// // Reset raw and visual data.
+//
+// TEveQuadSet::Reset(TEveQuadSet::kQT_RectangleYZ, kTRUE, 64);
+// // MT fBoxes.fBoxes.clear();
+// fData.Reset();
+// }
///////////////////////////////////////////////////////////
///////////// AliEveTRDHits /////////////////////
{
// Constructor.
SetMarkerSize(.1);
- SetMarkerColor(2);
+ SetMarkerColor(kGreen);
SetOwnIds(kTRUE);
}
{
// Handle an individual point selection from GL.
- AliTRDhit *h = dynamic_cast<AliTRDhit*>(GetPointId(n));
- printf("\nDetector : %d\n", h->GetDetector());
- printf("Region of production : %c\n", h->FromAmplification() ? 'A' : 'D');
- printf("TR photon : %s\n", h->FromTRphoton() ? "Yes" : "No");
- printf("Charge : %d\n", h->GetCharge());
- printf("MC track label : %d\n", h->GetTrack());
- printf("Time from collision : %f\n", h->GetTime());
+ AliTRDhit *h = NULL;
+ if(!(h = dynamic_cast<AliTRDhit*>(GetPointId(n)))) return;
+ printf("Id[%3d] Det[%3d] Reg[%c] TR[%c] Q[%3d] MC[%d] t[%f]\n",
+ n, h->GetDetector(),
+ h->FromAmplification() ? 'A' : 'D',
+ h->FromTRphoton() ? 'y' : 'n',
+ h->GetCharge(), h->GetTrack(), h->GetTime());
}
// Handle an individual point selection from GL.
AliTRDcluster *c = dynamic_cast<AliTRDcluster*>(GetPointId(n));
- printf("\nDetector : %d\n", c->GetDetector());
- printf("Charge : %f\n", c->GetQ());
- printf("Sum S : %4.0f\n", c->GetSumS());
- printf("Time bin : %d\n", c->GetLocalTimeBin());
- printf("Signals : ");
- Short_t *cSignals = c->GetSignals();
- for(Int_t ipad=0; ipad<7; ipad++) printf("%d ", cSignals[ipad]); printf("\n");
- printf("Central pad : %d\n", c->GetPadCol());
- printf("MC track labels : ");
- for(Int_t itrk=0; itrk<3; itrk++) printf("%d ", c->GetLabel(itrk)); printf("\n");
+ if(!c) return;
+ c->Print();
+ Emit("PointSelected(Int_t)", n);
// Bool_t AliCluster::GetGlobalCov(Float_t* cov) const
// Bool_t AliCluster::GetGlobalXYZ(Float_t* xyz) const
// Float_t AliCluster::GetSigmaY2() const
// Float_t AliCluster::GetSigmaZ2() const
}
+//______________________________________________________________________________
+void AliEveTRDClusters::Print(Option_t *o) const
+{
+ AliTRDcluster *c = NULL;
+
+ for(Int_t n = GetN(); n--;){
+ if(!(c = dynamic_cast<AliTRDcluster*>(GetPointId(n)))) continue;
+ c->Print(o);
+ }
+}
+
+//______________________________________________________________________________
+void AliEveTRDClusters::Load(const Char_t *w) const
+{
+ Int_t typ = -1;
+ if(strcmp(w, "hit")==0) typ = 0;
+ else if(strcmp(w, "dig")==0) typ = 1;
+ else if(strcmp(w, "cls")==0) typ = 2;
+ else if(strcmp(w, "tlt")==0) typ = 3;
+ else if(strcmp(w, "all")==0) typ = 4;
+ else{
+ AliInfo("The following arguments are accepted:");
+ AliInfo(" \"hit\" : loading of MC hits");
+ AliInfo(" \"dig\" : loading of digits");
+ AliInfo(" \"cls\" : loading of reconstructed clusters");
+ AliInfo(" \"tlt\" : loading of online tracklets");
+ AliInfo(" \"all\" : loading everything available");
+ return;
+ }
+
+ AliTRDcluster *c = NULL;
+ Int_t n = 0;
+ while((n = GetN() && !(c = dynamic_cast<AliTRDcluster*>(GetPointId(n))))) n++;
+ if(!c) return;
+
+ Int_t det = c->GetDetector();
+ AliEveTRDLoader *loader(NULL);
+ switch(typ){
+ case 4:
+ loader = new AliEveTRDLoaderSim("MC");
+ if(loader){
+ if(!loader->Open("galice.root")){delete loader; loader=NULL;}
+ else{
+ loader->SetDataType(AliEveTRDLoader::kTRDHits | AliEveTRDLoader::kTRDDigits | AliEveTRDLoader::kTRDClusters);
+ break;
+ }
+ }
+ case 0:
+ loader = new AliEveTRDLoader("Hits");
+ if(loader){
+ if(!loader->Open("TRD.Hits.root")){delete loader; loader=NULL;}
+ else{
+ if(typ!=4) break;
+ }
+ }
+ case 1:
+ if(!loader) loader = new AliEveTRDLoader("Digits");
+ if(loader){
+ if(!loader->Open("TRD.Digits.root")){
+ if(typ==1){delete loader; loader=NULL;}
+ } else {
+ if(typ!=4) break;
+ }
+ }
+ case 2:
+ if(!loader) loader = new AliEveTRDLoader("Clusters");
+ if(loader){
+ if(!loader->Open("TRD.RecPoints.root")){
+ if(typ ==2){delete loader; loader=NULL;}
+ } else {
+ if(typ!=4) break;
+ }
+ }
+ case 3:
+ if(!loader) loader = new AliEveTRDLoader("Tracklets");
+ if(loader){
+ if(!loader->Open("TRD.Tracklets.root")){
+ if(typ ==3) {delete loader; loader=NULL;}
+ } else break;
+ }
+ default: return;
+ }
+ if(!loader) return;
+
+ loader->AddChambers(AliTRDgeometry::GetSector(det),AliTRDgeometry::GetStack(det), AliTRDgeometry::GetLayer(det));
+ // load first event
+ if(!loader->GoToEvent(AliEveEventManager::GetCurrent()->GetEventId())) AliWarning(Form("No data loaded for event %d", AliEveEventManager::GetCurrent()->GetEventId()));
+ // register loader with alieve
+ gEve->AddElement(loader->GetChamber(det), *(BeginParents()));
+ //loader->SpawnEditor();
+ gEve->Redraw3D();
+}
+
///////////////////////////////////////////////////////////
///////////// AliEveTRDTracklet /////////////////////
///////////////////////////////////////////////////////////
//______________________________________________________________________________
AliEveTRDTracklet::AliEveTRDTracklet(AliTRDseedV1 *trklt):TEveLine()
- ,fClusters(0x0)
+ ,fClusters(NULL)
{
// Constructor.
SetName("tracklet");
+ if(!gGeoManager) AliEveEventManager::AssertGeometry();
SetUserData(trklt);
- Int_t det = -1, sec;
+ // init tracklet line
+ Int_t sec = AliTRDgeometry::GetSector(trklt->GetDetector());
+ Float_t alpha((0.5+sec)*AliTRDgeometry::GetAlpha()),
+ cphi(TMath::Cos(alpha)),
+ sphi(TMath::Sin(alpha));
+ Float_t dx;
+ Float_t x0 = trklt->GetX0();
+ Float_t y0 = trklt->GetYref(0);
+ Float_t z0 = trklt->GetZref(0);
+ Float_t dydx = trklt->GetYref(1);
+ Float_t dzdx = trklt->GetZref(1);
+ Float_t tilt = trklt->GetTilt();
Float_t g[3];
- AliTRDcluster *c = 0x0;
- AddElement(fClusters = new AliEveTRDClusters());
- for(Int_t ic=0; ic<35; ic++){
+ AliTRDcluster *c = NULL;
+ for(Int_t ic=0; ic<AliTRDseedV1::kNclusters; ic++){
if(!(c = trklt->GetClusters(ic))) continue;
- det = c->GetDetector();
- c->GetGlobalXYZ(g);
- Int_t id = fClusters->SetNextPoint(g[0], g[1], g[2]);
+ if(!fClusters) AddElement(fClusters = new AliEveTRDClusters());
+ dx = x0 - c->GetX();
+ //Float_t yt = y0 - dx*dydx;
+ Float_t zt = z0 - dx*dzdx;
+ // backup yc - for testing purposes
+ Float_t yc = c->GetY();
+ c->SetY(yc-tilt*(c->GetZ()-zt));
+
+ // Ben: Temporary(?) fix -> Issue with wrongly set flag "fIsMisaligned" of AliCluster
+ // Changes can be undone ofter the issue with the flag has been solved
+ //c->GetGlobalXYZ(g);
+
+ // The following code for the trasformation local-to-global is just adapted from
+ // AliTRDgeometry::RotateBack(...)
+
+ // tracking to global coordinates transformation
+ Float_t loc[3] = { c->GetX(), c->GetY(), c->GetZ() };
+ g[0] = loc[0] * cphi - loc[1] * sphi;
+ g[1] = loc[0] * sphi + loc[1] * cphi;
+ g[2] = loc[2];
+ // Ben: End of fix
+ Int_t id = fClusters->SetNextPoint(g[0], g[1], g[2]);
+ c->SetY(yc);
fClusters->SetPointId(id, new AliTRDcluster(*c));
}
+ if(fClusters){
+ fClusters->SetName("TC clusters");
+ fClusters->SetTitle(Form("N[%d]", trklt->GetN()));
+ fClusters->SetMarkerColor(kMagenta);
+ }
- SetTitle(Form("Det[%d] Plane[%d] P[%7.3f]", det, trklt->GetPlane(), trklt->GetMomentum()));
+ SetTitle(Form("Det[%d] RC[%c] Layer[%d] P[%7.3f]", trklt->GetDetector(), trklt->IsRowCross()?'y':'n', trklt->GetPlane(), trklt->GetMomentum()));
SetLineColor(kRed);
//SetOwnIds(kTRUE);
- sec = det/30;
- Double_t alpha = AliTRDgeometry::GetAlpha() * (sec<9 ? sec + .5 : sec - 17.5);
- Double_t x0 = trklt->GetX0(),
- y0f = trklt->GetYfit(0),
- ysf = trklt->GetYfit(1),
- z0r = trklt->GetZref(0),
- zsr = trklt->GetZref(1);
- Double_t xg = x0 * TMath::Cos(alpha) - y0f * TMath::Sin(alpha);
- Double_t yg = x0 * TMath::Sin(alpha) + y0f * TMath::Cos(alpha);
- SetPoint(0, xg, yg, z0r);
- //SetPointId(0, new AliTRDseedV1(*trackletObj));
- Double_t x1 = x0-3.5,
- y1f = y0f - ysf*3.5,
- z1r = z0r - zsr*3.5;
- xg = x1 * TMath::Cos(alpha) - y1f * TMath::Sin(alpha);
- yg = x1 * TMath::Sin(alpha) + y1f * TMath::Cos(alpha);
- SetPoint(1, xg, yg, z1r);
+ //trklt->Fit(kTRUE);
+ y0 = trklt->GetYfit(0);
+ dydx = trklt->GetYfit(1);
+ Double_t xg(x0 * cphi - y0 * sphi),
+ yg(x0 * sphi + y0 * cphi);
+ SetPoint(0, xg, yg, z0);
+ //SetPoint(0, x0, y0, z0);
+
+
+ dx = .5*AliTRDgeometry::CamHght()+AliTRDgeometry::CdrHght();
+ x0 -= dx;
+ y0 -= dydx*dx,
+ z0 -= dzdx*dx;
+ xg = x0 * cphi - y0 * sphi;
+ yg = x0 * sphi + y0 * cphi;
+ SetPoint(1, xg, yg, z0);
+ //SetPoint(1, x0, y0, z0);
+}
+
+
+AliEveTRDTracklet::~AliEveTRDTracklet()
+{
+
}
//______________________________________________________________________________
-void AliEveTRDTracklet::ProcessData()
+void AliEveTRDTracklet::Print(Option_t *o) const
{
AliTRDseedV1 *tracklet = (AliTRDseedV1*)GetUserData();
- tracklet->Print();
+ if(!tracklet) return;
+ tracklet->Print(o);
}
///////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////
//______________________________________________________________________________
-AliEveTRDTrack::AliEveTRDTrack(AliTRDtrackV1 *trk) : TEveLine()
+AliEveTRDTrack::AliEveTRDTrack(AliTRDtrackV1 *trk)
+ :TEveLine()
+ ,fTrackState(0)
+ ,fESDStatus(0)
+ ,fAlpha(0.)
+ ,fPoints(NULL)
+ ,fRim(NULL)
{
// Constructor.
- SetName("track");
-
SetUserData(trk);
-
- AliTRDtrackerV1::SetNTimeBins(24);
- AliTrackPoint points[AliTRDtrackV1::kMAXCLUSTERSPERTRACK];
- Int_t nc = 0, sec = -1; Float_t alpha = 0.;
- AliTRDcluster *c = 0x0;
- AliTRDseedV1 *tracklet = 0x0;
+ SetName("");
+
+// AliTRDtrackerV1::SetNTimeBins(24);
+
+ fRim = new AliRieman(trk->GetNumberOfClusters());
+ AliTRDseedV1 *tracklet = NULL;
for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++){
if(!(tracklet = trk->GetTracklet(il))) continue;
if(!tracklet->IsOK()) continue;
- AddElement(fTracklet[il] = new AliEveTRDTracklet(tracklet));
+ AddElement(new AliEveTRDTracklet(tracklet));
+
+// tracklet->ResetClusterIter(kFALSE);
+// while((c = tracklet->PrevCluster())){
+// AliTRDcluster *c(NULL);
+/* for(Int_t ic=AliTRDseedV1::kNtb; ic--;){
+ if(!(c=tracklet->GetClusters(ic))) continue;
+ Float_t xc = c->GetX();
+ Float_t yc = c->GetY();
+ Float_t zc = c->GetZ();
+ Float_t zt = tracklet->GetZref(0) - (tracklet->GetX0()-xc)*tracklet->GetZref(1);
+ yc -= tracklet->GetTilt()*(zc-zt);
+ fRim->AddPoint(xc, yc, zc, .05, 2.3);
+ }*/
+ }
+ if(trk->GetNumberOfTracklets()>1) fRim->Update();
+ SetStatus(fTrackState);
+}
+
+//______________________________________________________________________________
+AliEveTRDTrack::~AliEveTRDTrack()
+{
+ if(fPoints) delete [] fPoints; fPoints = NULL;
+ //delete dynamic_cast<AliTRDtrackV1*>(GetUserData());
+}
- for(Int_t ic=34; ic>=0; ic--){
- if(!(c = tracklet->GetClusters(ic))) continue;
- if(sec<0){
- sec = c->GetDetector()/30;
- alpha = AliTRDgeometry::GetAlpha() * (sec<9 ? sec + .5 : sec - 17.5);
+//______________________________________________________________________________
+void AliEveTRDTrack::Print(Option_t *o) const
+{
+ AliTRDtrackV1 *track = (AliTRDtrackV1*)GetUserData();
+ if(!track) return;
+ track->Print(o);
+}
+
+//______________________________________________________________________________
+void AliEveTRDTrack::SetStatus(UChar_t s)
+{
+ // nothing to be done
+ if(fPoints && fTrackState == s) return;
+ //return;
+ const Int_t nc = AliTRDtrackV1::kMAXCLUSTERSPERTRACK;
+ AliTRDtrackV1 *trk(NULL);
+ if(!(trk=static_cast<AliTRDtrackV1*>(GetUserData()))) {
+ AliError("Failed casting data to TRD track.");
+ return;
+ }
+
+ Bool_t BUILD = kFALSE;
+ if(!fPoints){
+ fPoints = new AliTrackPoint[nc];
+
+ // define the radial span of the track in the TRD
+ Double_t xmin = -1., xmax = -1.;
+ Int_t det = 0;
+ AliTRDseedV1 *trklt = NULL;
+ for(Int_t ily=0; ily<AliTRDgeometry::kNlayer; ily++){
+ if(!(trklt = trk->GetTracklet(ily))) continue;
+ if(xmin<0.) xmin = trklt->GetX0() - AliTRDgeometry::CamHght() - AliTRDgeometry::CdrHght();
+ if(trklt->GetX0()>xmax) xmax = trklt->GetX0();
+ det = trklt->GetDetector();
+ }
+ Int_t sec = det/AliTRDgeometry::kNdets;
+ fAlpha = AliTRDgeometry::GetAlpha() * (sec<9 ? sec + .5 : sec - 17.5); //trk->GetAlpha()
+
+ Double_t dx =(xmax - xmin)/nc;
+ for(Int_t ip=0; ip<nc; ip++){
+ fPoints[ip].SetXYZ(xmin, 0., 0.);
+ xmin+=dx;
+ }
+ BUILD = kTRUE;
+ }
+
+ // select track model
+ if(BUILD || ((s&12) != (fTrackState&12))){
+ if(TESTBIT(s, kTrackCosmics)){
+ //printf("Straight track\n");
+ AliTRDtrackerV1::FitLine(trk, NULL, kFALSE, nc, fPoints);
+ } else {
+ if(TESTBIT(s, kTrackModel)){
+ //printf("Kalman track\n");
+ if(trk->GetNumberOfTracklets() >=4) AliTRDtrackerV1::FitKalman(trk, NULL, kFALSE, nc, fPoints);
+ } else {
+ //printf("Rieman track rim[%p] nc[%d]\n", (void*)fRim, nc);
+ //if(trk->GetNumberOfTracklets() >=4) AliTRDtrackerV1::FitRiemanTilt(trk, NULL, kTRUE, nc, fPoints);
+ Float_t x = 0.;
+ for(Int_t ip = nc; ip--;){
+ x = fPoints[ip].GetX();
+ //printf("%2d x[%f] y[%f] z[%f]\n", ip, x, fRim->GetYat(x), fRim->GetZat(x));
+ fPoints[ip].SetXYZ(x, fRim->GetYat(x), fRim->GetZat(x));
+ }
}
- points[nc].SetXYZ(c->GetX(),0.,0.);
- nc++;
}
+
+ Float_t global[3];
+ for(Int_t ip=0; ip<nc; ip++){
+ fPoints[ip].Rotate(-fAlpha).GetXYZ(global);
+ SetPoint(ip, global[0], global[1], global[2]);
+ //printf("*** %2d x[%f] y[%f] z[%f]\n", ip, global[0], global[1], global[2]);
+
+ }
+ SetSmooth(kTRUE);
}
- //AliTRDtrackerV1::FitRiemanTilt(trk, 0x0, kTRUE, nc, points);
- //AliTRDtrackerV1::FitKalman(trk, 0x0, kFALSE, nc, points);
- AliTRDtrackerV1::FitLine(trk, 0x0, kFALSE, nc, points);
+ // set color
+ if(BUILD || ((s&3) != (fTrackState&3))){
+ if(TESTBIT(s, kSource)){
+ //printf("Source color\n");
+ if(fESDStatus&AliESDtrack::kTRDin){
+ SetMarkerColor(kGreen);
+ SetLineColor(kGreen);
+ } else {
+ SetMarkerColor(kMagenta);
+ SetLineColor(kMagenta);
+ }
+ } else {
+ if(TESTBIT(s, kPID) == AliTRDpidUtil::kLQ){
+ //printf("PID color kLQPID\n");
+ //trk->GetReconstructor()->SetOption("!nn");
+ } else {
+ //printf("PID color kNNPID\n");
+ //trk->GetReconstructor()->SetOption("nn");
+ }
+ //trk->CookPID();
+
+ Int_t species = 0; Float_t pid = 0.;
+ for(Int_t is=0; is<AliPID::kSPECIES; ++is)
+ if(trk->GetPID(is) > pid){
+ pid = trk->GetPID(is);
+ species = (AliPID::EParticleType) is;
+ }
+ switch(species){
+ case AliPID::kElectron:
+ SetMarkerColor(kRed);
+ SetLineColor(kRed);
+ break;
+ default:
+ SetMarkerColor(kBlue);
+ SetLineColor(kBlue);
+ break;
+ }
+ }
+ SetLineWidth(2);
+ }
+
+ const Char_t *model = "line";
+ if(!TESTBIT(s, kTrackCosmics)){
+ if(TESTBIT(s, kTrackModel)) model = "kalman";
+ else model = "rieman";
+ }
+ Int_t species = 0; Float_t pid = 0.;
+ for(Int_t is=0; is<AliPID::kSPECIES; is++)
+ if(trk->GetPID(is) > pid){
+ pid = trk->GetPID(is);
+ species = is;
+ }
- Float_t global[3];
- for(Int_t ip=0; ip<nc; ip++){
- points[ip].Rotate(-alpha).GetXYZ(global);
- SetNextPoint(global[0], global[1], global[2]);
+ SetTitle(Form(
+ "Tracklets[%d] Clusters[%d]\n"
+ "Reconstruction Source[%s]\n"
+ "PID[%4.1f %4.1f %4.1f %4.1f %4.1f]\n"
+ "MC[%d]", trk->GetNumberOfTracklets(), trk->GetNumberOfClusters(), fESDStatus&AliESDtrack::kTRDin ? "barrel" : "sa",
+ 1.E2*trk->GetPID(0), 1.E2*trk->GetPID(1),
+ 1.E2*trk->GetPID(2), 1.E2*trk->GetPID(3), 1.E2*trk->GetPID(4), trk->GetLabel()));
+
+ if(GetName()){
+ char id[6]; snprintf(id, 6, "%s", GetName());
+ SetName(Form("%s %s", id, AliPID::ParticleName(species)));
}
- SetMarkerColor(kBlack);
- SetLineColor(kBlack);
- SetSmooth(kTRUE);
+ // save track status
+ fTrackState = s;
}
//______________________________________________________________________________
-void AliEveTRDTrack::ProcessData()
+void AliEveTRDTrack::Load(const Char_t *what) const
{
- AliTRDtrackV1 *track = (AliTRDtrackV1*)GetUserData();
- AliInfo(Form("Clusters[%d]", track->GetNumberOfClusters()));
+// Spread downwards to tracklets the command "what"
+
+ const AliEveTRDTracklet* trklt(NULL);
+ TEveElement::List_ci itrklt=BeginChildren();
+ while(itrklt!=EndChildren()){
+ if((trklt = dynamic_cast<const AliEveTRDTracklet*>(*itrklt))) trklt->Load(what);
+ itrklt++;
+ }
+}
+
+//______________________________________________________________________________
+AliEveTRDTrackletOnline::AliEveTRDTrackletOnline(AliTRDtrackletMCM *tracklet) :
+ TEveLine(),
+ fDetector(-1),
+ fROB(-1),
+ fMCM(-1)
+{
+ AliTRDtrackletMCM *trkl = new AliTRDtrackletMCM(*tracklet);
+ SetUserData(trkl);
+
+ fDetector = trkl->GetDetector();
+ fROB = trkl->GetROB();
+ fMCM = trkl->GetMCM();
+
+ SetName("sim. tracklet");
+ SetTitle(Form("Det: %i, ROB: %i, MCM: %i, Label: %i\n0x%08x",
+ trkl->GetDetector(), trkl->GetROB(), trkl->GetMCM(), trkl->GetLabel(),
+ trkl->GetTrackletWord()));
+ SetLineColor(kGreen);
+ SetLineWidth(3);
+
+ AliTRDgeometry geo;
+ TGeoHMatrix *matrix = geo.GetClusterMatrix(trkl->GetDetector());
+
+ fDetector = trkl->GetDetector();
+ fROB = trkl->GetROB();
+ fMCM = trkl->GetMCM();
+
+ Float_t length = 3.;
+ Double_t x[3];
+ Double_t p[3];
+ Double_t p2[3];
+ x[0] = AliTRDgeometry::AnodePos();
+ x[1] = trkl->GetY();
+ x[2] = trkl->GetLocalZ();
+
+ matrix->LocalToMaster(x, p);
+ geo.RotateBack(trkl->GetDetector(), p, p2);
+ SetPoint(0, p2[0], p2[1], p2[2]);
+
+ x[0] -= length;
+ x[1] -= length * trkl->GetdYdX();
+ matrix->LocalToMaster(x, p);
+ p[2] *= p[0] / (p[0] + length);
+ geo.RotateBack(trkl->GetDetector(), p, p2);
+ SetPoint(1, p2[0], p2[1], p2[2]);
+}
+
+AliEveTRDTrackletOnline::AliEveTRDTrackletOnline(AliTRDtrackletWord *tracklet) :
+ TEveLine(),
+ fDetector(-1),
+ fROB(-1),
+ fMCM(-1)
+{
+ AliTRDtrackletWord *trkl = new AliTRDtrackletWord(*tracklet);
+ SetUserData(trkl);
+
+ fDetector = trkl->GetDetector();
+ fROB = trkl->GetROB();
+ fMCM = trkl->GetMCM();
+
+ SetName("raw tracklet");
+ SetTitle(Form("Det: %i, ROB: %i, MCM: %i, Label: %i\n0x%08x",
+ trkl->GetDetector(), fROB, fMCM, -1,
+ trkl->GetTrackletWord()));
+ SetLineColor(kRed);
+ SetLineWidth(3);
+
+ AliTRDgeometry geo;
+ TGeoHMatrix *matrix = geo.GetClusterMatrix(trkl->GetDetector());
+
+ Float_t length = 3.;
+ Double_t x[3];
+ Double_t p[3];
+ Double_t p2[3];
+ x[0] = AliTRDgeometry::AnodePos();
+ x[1] = trkl->GetY();
+ x[2] = trkl->GetLocalZ();
- track->CookPID();
- printf("PIDLQ : "); for(int is=0; is<AliPID::kSPECIES; is++) printf("%s[%5.2f] ", AliPID::ParticleName(is), 1.E2*track->GetPID(is)); printf("\n");
+ matrix->LocalToMaster(x, p);
+ geo.RotateBack(trkl->GetDetector(), p, p2);
+ SetPoint(0, p2[0], p2[1], p2[2]);
+
+ x[0] -= length;
+ x[1] -= length * trkl->GetdYdX();
+ matrix->LocalToMaster(x, p);
+ p[2] *= p[0] / (p[0] + length);
+ geo.RotateBack(trkl->GetDetector(), p, p2);
+ SetPoint(1, p2[0], p2[1], p2[2]);
+}
+
+AliEveTRDTrackletOnline::~AliEveTRDTrackletOnline()
+{
+ delete ((AliTRDtrackletBase*) GetUserData());
+ SetUserData(0x0);
+}
+
+void AliEveTRDTrackletOnline::ShowMCM(Option_t *opt) const
+{
+ if (fDetector < 0 || fROB < 0 || fMCM < 0)
+ return;
+
+ AliEveTRDmcm *evemcm = new AliEveTRDmcm();
+ evemcm->Init(fDetector, fROB, fMCM);
+ evemcm->LoadDigits();
+ evemcm->Draw(opt);
+
+ TEveElementList *mcmlist = NULL;
+ if (gEve->GetCurrentEvent())
+ mcmlist = (TEveElementList*) gEve->GetCurrentEvent()->FindChild("TRD MCMs");
+ if (!mcmlist) {
+ mcmlist = new TEveElementList("TRD MCMs");
+ gEve->AddElement(mcmlist);
+ }
+ gEve->AddElement(evemcm, mcmlist);
+}
+
+
+AliEveTRDmcm::AliEveTRDmcm() :
+ TEveElement(),
+ TNamed(),
+ fMCM(new AliTRDmcmSim())
+{
+ SetName("MCM");
+ SetTitle("Unknown MCM");
+}
+
+AliEveTRDmcm::~AliEveTRDmcm()
+{
+ delete fMCM;
+}
+
+Bool_t AliEveTRDmcm::Init(Int_t det, Int_t rob, Int_t mcm)
+{
+ SetName(Form("MCM: Det. %i, ROB %i, MCM %i", det, rob, mcm));
+ SetTitle(Form("MCM: Det. %i, ROB %i, MCM %i", det, rob, mcm));
+ fMCM->Init(det, rob, mcm);
+ fMCM->Reset();
+ return kTRUE;
}
+Bool_t AliEveTRDmcm::LoadDigits()
+{
+ AliRunLoader *rl = AliEveEventManager::AssertRunLoader();
+ return fMCM->LoadMCM(rl, fMCM->GetDetector(),
+ fMCM->GetRobPos(), fMCM->GetMcmPos());
+}
+
+Bool_t AliEveTRDmcm::Filter()
+{
+ fMCM->Filter();
+ return kTRUE;
+}
+
+Bool_t AliEveTRDmcm::Tracklet()
+{
+ fMCM->Tracklet();
+ return kTRUE;
+}
+
+void AliEveTRDmcm::Draw(Option_t* option)
+{
+ const char *mcmname = Form("mcm_%i_%i_%i", fMCM->GetDetector(),
+ fMCM->GetRobPos(), fMCM->GetMcmPos());
+
+ TCanvas *c = dynamic_cast<TCanvas*> (gROOT->FindObject(mcmname));
+ if (!c)
+ c = gEve->AddCanvasTab("TRD MCM");
+ c->SetTitle(Form("MCM %i on ROB %i of det. %i",
+ fMCM->GetMcmPos(), fMCM->GetRobPos(), fMCM->GetDetector()));
+ c->SetName(mcmname);
+ c->cd();
+ fMCM->Draw(option);
+}
+
+Bool_t AliEveTRDmcm::AssignPointer(const char* ptrname)
+{
+ gROOT->ProcessLine(Form("AliTRDmcmSim* %s = (AliTRDmcmSim *)%p", ptrname, (void*)fMCM));
+ return kTRUE;
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff