1 // @(#) $Id: AliHLTTPCCATrackletSelector.cxx 27042 2008-07-02 12:06:02Z richterm $
2 // **************************************************************************
3 // This file is property of and copyright by the ALICE HLT Project *
4 // ALICE Experiment at CERN, All rights reserved. *
6 // Primary Authors: Sergey Gorbunov <sergey.gorbunov@kip.uni-heidelberg.de> *
7 // Ivan Kisel <kisel@kip.uni-heidelberg.de> *
8 // for The ALICE HLT Project. *
10 // Permission to use, copy, modify and distribute this software and its *
11 // documentation strictly for non-commercial purposes is hereby granted *
12 // without fee, provided that the above copyright notice appears in all *
13 // copies and that both the copyright notice and this permission notice *
14 // appear in the supporting documentation. The authors make no claims *
15 // about the suitability of this software for any purpose. It is *
16 // provided "as is" without express or implied warranty. *
18 //***************************************************************************
21 #include "AliHLTTPCCATrackletSelector.h"
22 #include "AliHLTTPCCATrack.h"
23 #include "AliHLTTPCCATracker.h"
24 #include "AliHLTTPCCATrackParam.h"
25 #include "AliHLTTPCCATracklet.h"
26 #include "AliHLTTPCCAMath.h"
28 GPUdi() void AliHLTTPCCATrackletSelector::Thread
29 ( int nBlocks, int nThreads, int iBlock, int iThread, int iSync,
30 GPUsharedref() MEM_LOCAL(AliHLTTPCCASharedMemory) &s, GPUconstant() MEM_CONSTANT(AliHLTTPCCATracker) &tracker )
32 // select best tracklets and kill clones
36 s.fNTracklets = *tracker.NTracklets();
37 s.fNThreadsTotal = nThreads * nBlocks;
38 s.fItr0 = nThreads * iBlock;
40 } else if ( iSync == 1 ) {
41 int nHits, nFirstTrackHit;
42 AliHLTTPCCAHitId trackHits[160 - HLTCA_GPU_TRACKLET_SELECTOR_HITS_REG_SIZE];
44 for ( int itr = s.fItr0 + iThread; itr < s.fNTracklets; itr += s.fNThreadsTotal ) {
46 #ifdef HLTCA_GPU_EMULATION_DEBUG_TRACKLET
47 if (itr == HLTCA_GPU_EMULATION_DEBUG_TRACKLET)
49 tracker.GPUParameters()->fGPUError = 1;
51 #endif //HLTCA_GPU_EMULATION_DEBUG_TRACKLET
53 while (tracker.Tracklets()[itr].NHits() == 0)
55 itr += s.fNThreadsTotal;
56 if (itr >= s.fNTracklets) return;
59 GPUglobalref() MEM_GLOBAL(AliHLTTPCCATracklet) &tracklet = tracker.Tracklets()[itr];
60 const int kMaxRowGap = 4;
61 const float kMaxShared = .1;
63 int firstRow = tracklet.FirstRow();
64 int lastRow = tracklet.LastRow();
65 if (firstRow < 0 || lastRow > tracker.Param().NRows() || tracklet.NHits() < 0)
67 #ifdef HLTCA_GPU_ALTERNATIVE_SCHEDULER
68 //tracker.GPUParameters()->fGPUError = HLTCA_GPU_ERROR_WRONG_ROW;
75 const int w = tracklet.HitWeight();
77 //int w = (tNHits<<16)+itr;
78 //int nRows = tracker.Param().NRows();
79 //std::cout<<" store tracklet: "<<firstRow<<" "<<lastRow<<std::endl;
87 for (irow = firstRow; irow <= lastRow && lastRow - irow + nHits >= TRACKLET_SELECTOR_MIN_HITS; irow++ )
90 #ifdef EXTERN_ROW_HITS
91 int ih = tracker.TrackletRowHits()[irow * s.fNTracklets + itr];
93 int ih = tracklet.RowHit( irow );
94 #endif //EXTERN_ROW_HITS
96 GPUglobalref() const MEM_GLOBAL(AliHLTTPCCARow) &row = tracker.Row( irow );
97 #ifdef GLOBAL_TRACKING_ONLY_UNASSIGNED_HITS
98 bool own = ( abs(tracker.HitWeight( row, ih )) <= w );
100 bool own = ( tracker.HitWeight( row, ih ) <= w );
102 bool sharedOK = ( ( nShared < nHits * kMaxShared ) );
103 if ( own || sharedOK ) {//SG!!!
105 #if HLTCA_GPU_TRACKLET_SELECTOR_HITS_REG_SIZE != 0
106 if (nHits < HLTCA_GPU_TRACKLET_SELECTOR_HITS_REG_SIZE)
107 s.fHits[iThread][nHits].Set( irow, ih );
109 #endif //HLTCA_GPU_TRACKLET_SELECTOR_HITS_REG_SIZE != 0
110 trackHits[nHits - HLTCA_GPU_TRACKLET_SELECTOR_HITS_REG_SIZE].Set( irow, ih );
112 if ( !own ) nShared++;
116 if ( gap > kMaxRowGap || irow == lastRow ) { // store
117 if ( nHits >= TRACKLET_SELECTOR_MIN_HITS ) { //SG!!!
118 int itrout = CAMath::AtomicAdd( tracker.NTracks(), 1 );
120 if (itrout >= HLTCA_GPU_MAX_TRACKS)
122 tracker.GPUParameters()->fGPUError = HLTCA_GPU_ERROR_TRACK_OVERFLOW;
123 CAMath::AtomicExch( tracker.NTracks(), 0 );
126 #endif //HLTCA_GPUCODE
127 nFirstTrackHit = CAMath::AtomicAdd( tracker.NTrackHits(), nHits );
128 tracker.Tracks()[itrout].SetAlive(1);
129 tracker.Tracks()[itrout].SetLocalTrackId(itrout);
130 tracker.Tracks()[itrout].SetParam(tracklet.Param());
131 tracker.Tracks()[itrout].SetFirstHitID(nFirstTrackHit);
132 tracker.Tracks()[itrout].SetNHits(nHits);
133 for ( int jh = 0; jh < nHits; jh++ ) {
134 #if HLTCA_GPU_TRACKLET_SELECTOR_HITS_REG_SIZE != 0
135 if (jh < HLTCA_GPU_TRACKLET_SELECTOR_HITS_REG_SIZE)
137 tracker.TrackHits()[nFirstTrackHit + jh] = s.fHits[iThread][jh];
138 #ifdef GLOBAL_TRACKING_ONLY_UNASSIGNED_HITS
139 tracker.SetHitWeight( tracker.Row( s.fHits[iThread][jh].RowIndex() ), s.fHits[iThread][jh].HitIndex(), -w );
143 #endif //HLTCA_GPU_TRACKLET_SELECTOR_HITS_REG_SIZE != 0
145 tracker.TrackHits()[nFirstTrackHit + jh] = trackHits[jh - HLTCA_GPU_TRACKLET_SELECTOR_HITS_REG_SIZE];
146 #ifdef GLOBAL_TRACKING_ONLY_UNASSIGNED_HITS
147 tracker.SetHitWeight( tracker.Row( trackHits[jh - HLTCA_GPU_TRACKLET_SELECTOR_HITS_REG_SIZE].RowIndex() ), trackHits[jh - HLTCA_GPU_TRACKLET_SELECTOR_HITS_REG_SIZE].HitIndex(), -w );