#include "TObject.h"
#include "TObjArray.h"
#include "TClass.h"
+#include "TMath.h"
#include "TH1F.h"
#include "AliESDTrdTrack.h"
#include "AliESDTrdTrack.h"
ClassImp(AliTRDtrackGTU)
-
+
AliTRDtrackGTU::AliTRDtrackGTU() :
TObject(),
fStack(-1),
fSector(-1),
- fPt(0),
fPID(0),
fTracklets(0x0),
fTrackletMask(0),
// fTracklets->BypassStreamer(kFALSE);
}
+AliTRDtrackGTU::AliTRDtrackGTU(const AliTRDtrackGTU &rhs) :
+ TObject(),
+ fStack(rhs.fStack),
+ fSector(rhs.fSector),
+ fPID(rhs.fPID),
+ fTracklets(0x0),
+ fTrackletMask(rhs.fTrackletMask),
+ fNTracklets(rhs.fNTracklets),
+ fRefLayerIdx(rhs.fRefLayerIdx),
+ fZChannel(rhs.fZChannel),
+ fZSubChannel(rhs.fZSubChannel),
+ fA(rhs.fA),
+ fB(rhs.fB),
+ fC(rhs.fC),
+ fLabel(rhs.fLabel)
+{
+ fTracklets = new TClonesArray("AliTRDtrackletGTU", 6);
+ for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++)
+ new ((*fTracklets)[iTracklet]) AliTRDtrackletGTU(*((AliTRDtrackletGTU*)(*(rhs.fTracklets))[iTracklet]));
+}
+
+AliTRDtrackGTU& AliTRDtrackGTU::operator=(const AliTRDtrackGTU &rhs)
+{
+ if (&rhs != this) {
+ TObject::operator=(rhs);
+ fStack = rhs.fStack;
+ fSector = rhs.fSector;
+ fPID = rhs.fPID;
+ fTrackletMask = rhs.fTrackletMask;
+ fNTracklets = rhs.fNTracklets;
+ fRefLayerIdx = rhs.fRefLayerIdx;
+ fZChannel = rhs.fZChannel;
+ fZSubChannel = rhs.fZSubChannel;
+ fA = rhs.fA;
+ fB = rhs.fB;
+ fC = rhs.fC;
+ fLabel = rhs.fLabel;
+ for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++)
+ new ((*fTracklets)[iTracklet]) AliTRDtrackletGTU(*((AliTRDtrackletGTU*)(*(rhs.fTracklets))[iTracklet]));
+ }
+
+ return *this;
+}
+
+AliTRDtrackGTU& AliTRDtrackGTU::operator=(const AliESDTrdTrack &rhs)
+{
+ if ((void*) &rhs != (void*) this) {
+ TObject::operator=(rhs);
+ fStack = rhs.GetStack();
+ fSector = rhs.GetSector();
+ fPID = rhs.GetPID();
+ fTrackletMask = rhs.GetLayerMask();
+ fNTracklets = 0;
+ fRefLayerIdx = -1;
+ fZChannel = -1;
+ fZSubChannel = -1;
+ fA = rhs.GetA();
+ fB = rhs.GetB();
+ fC = rhs.GetC();
+ fLabel = rhs.GetLabel();
+ for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++) {
+ AliTRDtrackletGTU *trkl = new ((*fTracklets)[iTracklet]) AliTRDtrackletGTU();
+ if (fTrackletMask & (1 << iTracklet)) {
+ ++fNTracklets;
+ trkl->SetIndex(rhs.GetTrackletIndex(iTracklet));
+ }
+ else
+ trkl->SetIndex(-1);
+ }
+ }
+
+ return *this;
+}
+
AliTRDtrackGTU::~AliTRDtrackGTU()
{
// dtor
delete fTracklets;
}
-void AliTRDtrackGTU::AddTracklet(const AliTRDtrackletGTU * const tracklet, Int_t layer)
+void AliTRDtrackGTU::AddTracklet(const AliTRDtrackletGTU * const tracklet, Int_t layer)
{
// add a tracklet to this track
fTrackletMask |= (1 << layer);
}
-AliTRDtrackletGTU* AliTRDtrackGTU::GetTracklet(Int_t layer)
+AliTRDtrackletGTU* AliTRDtrackGTU::GetTracklet(Int_t layer) const
{
// get a pointer to the tracklet in the layer specified
if (IsTrackletInLayer(layer))
return ((AliTRDtrackletGTU*) (*fTracklets)[layer]);
- else
+ else
return 0x0;
}
return fNTracklets;
}
-Bool_t AliTRDtrackGTU::IsTrackletInLayer(Int_t layer) const
+Bool_t AliTRDtrackGTU::IsTrackletInLayer(Int_t layer) const
{
// checks for a tracklet in the given layer
if ( (GetTrackletMask() & (1 << layer)) != 0)
return kTRUE;
- else
+ else
return kFALSE;
}
-void AliTRDtrackGTU::SetFitParams(Float_t a, Float_t b, Float_t c)
+void AliTRDtrackGTU::SetFitParams(Float_t a, Float_t b, Float_t c)
{
// set the fit parameters
- fA = a;
+ fA = a;
fB = b;
fC = c;
}
-Int_t AliTRDtrackGTU::GetZSubChannel()
+Int_t AliTRDtrackGTU::GetZSubChannel()
{
// returns the z-subchannel
if (fZSubChannel < 0) {
- for (Int_t layer = 0; layer < AliTRDgtuParam::GetNLayers(); layer++)
- {
- if (IsTrackletInLayer(layer))
- fZSubChannel = ((AliTRDtrackletGTU*) (*fTracklets)[layer])->GetSubChannel(GetZChannel());
+ for (Int_t layer = 0; layer < AliTRDgtuParam::GetNLayers(); layer++) {
+ if (IsTrackletInLayer(layer)) {
+ AliTRDtrackletGTU *trkl = (AliTRDtrackletGTU*) (*fTracklets)[layer];
+ if (trkl) {
+ if ((fZSubChannel > -1) &&
+ (fZSubChannel != trkl->GetSubChannel(GetZChannel())))
+ AliError(Form("found inconsistent z-subchannels: track = %i/%i, trkl = %i",
+ GetZChannel(), fZSubChannel, trkl->GetSubChannel(GetZChannel())));
+ fZSubChannel = trkl->GetSubChannel(GetZChannel());
+ }
+ else {
+ AliError("no tracklet where one should be according to layer mask");
+ }
+ }
}
}
return fZSubChannel;
}
-Int_t AliTRDtrackGTU::GetYapprox()
+Int_t AliTRDtrackGTU::GetYapprox()
{
-// returns an approximated y-position for the track
+ // returns an approximated y-position for the track
+ // taken from the projected y-position of the tracklet in the reference layer
+ // in which the track was found
- for (Int_t layer = 0; layer < AliTRDgtuParam::GetNLayers(); layer++)
- {
- if (IsTrackletInLayer(layer))
- return ((AliTRDtrackletGTU*) (*fTracklets)[layer])->GetYProj();
- }
- return 0;
+ if ((fRefLayerIdx > -1) && (fRefLayerIdx < AliTRDgtuParam::GetNRefLayers()))
+ return ((AliTRDtrackletGTU*) (*fTracklets)[AliTRDgtuParam::GetRefLayer(fRefLayerIdx)])->GetYProj();
+ else
+ return 0;
}
AliESDTrdTrack* AliTRDtrackGTU::CreateTrdTrack() const
{
-// creates an AliESDTrdTrack to be added to the ESD
-
- AliESDTrdTrack *trk = new AliESDTrdTrack();
- trk->SetPt(1./128. * fPt);
- trk->SetPID(fPID);
- trk->SetDetector((Char_t) (fSector * 5 + fStack));
- if (fLabel >= 0)
- trk->SetLabel(fLabel);
- AliInfo(Form("setting detector to: %i (sector: %i, stack: %i), readback: %i", fSector * 30 + fStack * 6, fSector, fStack, trk->GetDetector()));
- return trk;
+ // creates an AliESDTrdTrack to be added to the ESD
+
+ AliESDTrdTrack *trk = new AliESDTrdTrack();
+ trk->SetA((Int_t) fA);
+ trk->SetB(TMath::Nint(128. * fB));
+ trk->SetC(TMath::Nint(256. * fC));
+ trk->SetLayerMask(fTrackletMask);
+ trk->SetPID(fPID);
+ trk->SetStack(fStack);
+ trk->SetSector(fSector);
+ trk->SetLabel(fLabel);
+
+ for (Int_t iLayer = 0; iLayer < AliTRDgtuParam::GetNLayers(); iLayer++) {
+ AliTRDtrackletGTU *trklGTU = GetTracklet(iLayer);
+ if (trklGTU) {
+ trk->SetTrackletIndex(trklGTU->GetIndex(), iLayer);
+ AliESDTrdTracklet *trkl = trklGTU->GetTrackletESD();
+ if (trkl)
+ trk->AddTrackletReference(trkl, iLayer);
+ }
+ }
+
+ return trk;
}
-Bool_t AliTRDtrackGTU::CookLabel()
+Bool_t AliTRDtrackGTU::CookLabel()
{
- TH1F *h = new TH1F("trkref", "trkref", 100000, 0, 100000);
- for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++) {
- if (!IsTrackletInLayer(iTracklet))
- continue;
- h->Fill( ((AliTRDtrackletGTU*) (*fTracklets)[iTracklet])->GetLabel());
+ // assign label from tracklets according to frequency
+
+ Int_t nLabels = 0;
+ Int_t label[6] = { 0 };
+ Int_t count[6] = { 0 };
+
+ for (Int_t iTracklet = 0; iTracklet < 6; iTracklet++) {
+ if (!IsTrackletInLayer(iTracklet))
+ continue;
+
+ Int_t currLabel = GetTracklet(iTracklet)->GetLabel();
+
+ Bool_t assigned = kFALSE;
+ for (Int_t iLabel = 0; iLabel < nLabels; iLabel++) {
+ if (currLabel == label[iLabel]) {
+ count[iLabel]++;
+ assigned = kTRUE;
+ break;
+ }
}
- if (h->GetEntries() > 0)
- fLabel = h->GetMaximumBin() - 1;
- else
- fLabel = -1;
- delete h;
- return (fLabel >= 0);
+
+ if (!assigned) {
+ label[nLabels] = currLabel;
+ count[nLabels] = 1;
+ nLabels++;
+ }
+ }
+
+ Int_t index[32];
+ TMath::Sort(6, count, index);
+ fLabel = label[index[0]];
+
+ return kTRUE;
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff