217 CADEBUG(int32_t ii; printf(
"\nRefitting track\n"));
218 typename internal::refitTrackTypes<S>::propagator prop;
220 float TrackParCovChi2 = 0.f;
221 convertTrack<S, T, typename internal::refitTrackTypes<S>::propagator>(trk,
trkX, prop, &TrackParCovChi2);
224 if constexpr (std::is_same_v<T, GPUTPCGMMergedTrack>) {
226 tOffset =
trkX.GetParam().GetTOffset();
227 }
else if constexpr (std::is_same_v<T, TrackTPC>) {
229 tOffset =
trkX.getTime0();
230 }
else if constexpr (std::is_same_v<T, TrackParCovWithArgs>) {
232 tOffset =
trkX.time0;
234 static_assert(
"Invalid template");
236 if constexpr (std::is_same_v<S, GPUTPCGMTrackParam>) {
237 CADEBUG(printf(
"\t%21sInit Alpha %8.3f , X %8.3f - Y %8.3f, Z %8.3f - QPt %7.2f (%7.2f), SP %5.2f (%5.2f) --- Cov sY %8.3f sZ %8.3f sSP %8.3f sPt %8.3f\n",
"", prop.GetAlpha(), trk.GetX(), trk.Par()[0], trk.Par()[1], trk.Par()[4], prop.GetQPt0(), trk.Par()[2], prop.GetSinPhi0(), sqrtf(trk.Cov()[0]), sqrtf(trk.Cov()[2]), sqrtf(trk.Cov()[5]), sqrtf(trk.Cov()[14])));
240 int32_t direction = outward ? -1 : 1;
244 uint8_t sector = 255,
row = 255;
245 int32_t lastSector = -1, currentSector = -1, currentRow = -1;
246 int16_t clusterState = 0, nextState = 0;
248 float sumInvSqrtCharge = 0.f;
249 int32_t nAvgCharge = 0;
251 for (int32_t
i =
start;
i != stop;
i += cl ? 0 : direction) {
253 float time = 0.f, invCharge = 0.f, invSqrtCharge = 0.f;
258 if constexpr (std::is_same_v<T, GPUTPCGMMergedTrack>) {
259 const auto& hit = mPtrackHits[
trkX.FirstClusterRef() +
i];
260 cl = &mPclusterNative->clustersLinear[hit.num];
263 if (
i + direction != stop) {
271 nextState = mPclusterState[hit.num];
272 }
else if constexpr (std::is_same_v<T, TrackTPC>) {
273 cl = &
trkX.getCluster(mPtrackHitReferences,
i, *mPclusterNative, sector,
row);
274 nextState = mPclusterState[cl - mPclusterNative->clustersLinear];
275 }
else if constexpr (std::is_same_v<T, TrackParCovWithArgs>) {
276 cl = &TrackTPC::getCluster(mPtrackHitReferences,
i, *mPclusterNative, sector,
row,
trkX.clusRef);
277 nextState = mPclusterState[cl - mPclusterNative->clustersLinear];
279 static_assert(
"Invalid template");
282 if (
clusters == 0 || (
row == currentRow && sector == currentSector)) {
289 mPfastTransformHelper->Transform(sector,
row, cl->getPad(), cl->getTime(),
x,
y,
z, tOffset);
290 CADEBUG(printf(
"\tHit %3d/%3d Row %3d: Cluster Alpha %8.3f %3d, X %8.3f - Y %8.3f, Z %8.3f - State %d\n", ii,
count,
row, mPparam->Alpha(sector), (int32_t)sector,
x,
y,
z, (int32_t)nextState));
292 currentSector = sector;
294 clusterState = nextState;
295 time = cl->getTime();
296 invSqrtCharge = CAMath::InvSqrt(cl->
qMax);
297 invCharge = (1.f / cl->
qMax);
300 mPfastTransformHelper->Transform(sector,
row, cl->getPad(), cl->getTime(), xx, yy, zz, tOffset);
301 CADEBUG(printf(
"\tHit %3d/%3d Row %3d: Cluster Alpha %8.3f %3d, X %8.3f - Y %8.3f, Z %8.3f - State %d\n", ii,
count,
row, mPparam->Alpha(sector), (int32_t)sector, xx, yy, zz, (int32_t)nextState));
306 clusterState |= nextState;
310 if (
i + direction != stop) {
323 CADEBUG(printf(
"\tMerged Hit Row %3d: Cluster Alpha %8.3f %3d, X %8.3f - Y %8.3f, Z %8.3f - State %d\n",
row, mPparam->Alpha(sector), (int32_t)sector,
x,
y,
z, (int32_t)clusterState));
326 float invAvgCharge = (sumInvSqrtCharge += invSqrtCharge) / ++nAvgCharge;
327 invAvgCharge *= invAvgCharge;
329 if constexpr (std::is_same_v<S, GPUTPCGMTrackParam>) {
330 if (prop.PropagateToXAlpha(
x, mPparam->Alpha(currentSector), !outward)) {
335 trk.ResetCovariance();
336 }
else if (lastSector != -1 && (lastSector < 18) != (sector < 18)) {
337 if (mPparam->rec.tpc.addErrorsCECrossing) {
338 if (mPparam->rec.tpc.addErrorsCECrossing >= 2) {
339 trk.AddCovDiagErrorsWithCorrelations(mPparam->rec.tpc.errorsCECrossing);
341 trk.AddCovDiagErrors(mPparam->rec.tpc.errorsCECrossing);
343 }
else if (trk.Cov()[2] < 0.5f) {
347 CADEBUG(printf(
"\t%21sPropaga Alpha %8.3f , X %8.3f - Y %8.3f, Z %8.3f - QPt %7.2f (%7.2f), SP %5.2f (%5.2f) --- Res %8.3f %8.3f --- Cov sY %8.3f sZ %8.3f sSP %8.3f sPt %8.3f - YPt %8.3f\n",
"", prop.GetAlpha(),
x, trk.Par()[0], trk.Par()[1], trk.Par()[4], prop.GetQPt0(), trk.Par()[2], prop.GetSinPhi0(), trk.Par()[0] -
y, trk.Par()[1] -
z, sqrtf(trk.Cov()[0]), sqrtf(trk.Cov()[2]), sqrtf(trk.Cov()[5]), sqrtf(trk.Cov()[14]), trk.Cov()[10]));
349 if (prop.Update(
y,
z,
row, *mPparam, clusterState, 0,
true, sector,
time, invAvgCharge, invCharge)) {
353 trk.ConstrainSinPhi();
354 CADEBUG(printf(
"\t%21sFit Alpha %8.3f , X %8.3f - Y %8.3f, Z %8.3f - QPt %7.2f (%7.2f), SP %5.2f (%5.2f), DzDs %5.2f %16s --- Cov sY %8.3f sZ %8.3f sSP %8.3f sPt %8.3f - YPt %8.3f\n",
"", prop.GetAlpha(),
x, trk.Par()[0], trk.Par()[1], trk.Par()[4], prop.GetQPt0(), trk.Par()[2], prop.GetSinPhi0(), trk.Par()[3],
"", sqrtf(trk.Cov()[0]), sqrtf(trk.Cov()[2]), sqrtf(trk.Cov()[5]), sqrtf(trk.Cov()[14]), trk.Cov()[10]));
355 }
else if constexpr (std::is_same_v<S, TrackParCov>) {
356 if (!trk.rotate(mPparam->Alpha(currentSector))) {
364 if (lastSector != -1 && (lastSector < 18) != (sector < 18)) {
365 if (mPparam->rec.tpc.addErrorsCECrossing) {
366 trk.updateCov(mPparam->rec.tpc.errorsCECrossing, mPparam->rec.tpc.addErrorsCECrossing >= 2);
367 }
else if (trk.getCov()[2] < 0.5f) {
372 trk.resetCovariance();
373 float bzkG = prop->getNominalBz(), qptB5Scale = CAMath::Abs(bzkG) > 0.1f ? CAMath::Abs(bzkG) / 5.006680f : 1.f;
374 float q2pt2 = trk.getQ2Pt() * trk.getQ2Pt(), q2pt2Wgh = q2pt2 * qptB5Scale * qptB5Scale;
375 float err2 = (100.f + q2pt2Wgh) / (1.f + q2pt2Wgh) * q2pt2;
376 trk.setCov(err2, 14);
377 TrackParCovChi2 = 0.f;
379 CADEBUG(printf(
"\t%21sPropaga Alpha %8.3f , X %8.3f - Y %8.3f, Z %8.3f - QPt %7.2f (%7.2f), SP %5.2f (%5.2f) --- Res %8.3f %8.3f --- Cov sY %8.3f sZ %8.3f sSP %8.3f sPt %8.3f - YPt %8.3f\n",
"", trk.getAlpha(),
x, trk.getParams()[0], trk.getParams()[1], trk.getParams()[4], trk.getParams()[4], trk.getParams()[2], trk.getParams()[2], trk.getParams()[0] -
y, trk.getParams()[1] -
z, sqrtf(trk.getCov()[0]), sqrtf(trk.getCov()[2]), sqrtf(trk.getCov()[5]), sqrtf(trk.getCov()[14]), trk.getCov()[10]));
380 std::array<float, 2> p = {
y,
z};
381 std::array<float, 3>
c = {0, 0, 0};
382 GPUTPCGMPropagator::GetErr2(
c[0],
c[2], *mPparam, getPar(trk)[2], getPar(trk)[3],
z,
x,
y, currentRow, clusterState, sector,
time, invAvgCharge, invCharge,
false);
383 TrackParCovChi2 += trk.getPredictedChi2(p,
c);
384 if (!trk.update(p,
c)) {
388 CADEBUG(printf(
"\t%21sFit Alpha %8.3f , X %8.3f - Y %8.3f, Z %8.3f - QPt %7.2f (%7.2f), SP %5.2f (%5.2f), DzDs %5.2f %16s --- Cov sY %8.3f sZ %8.3f sSP %8.3f sPt %8.3f - YPt %8.3f\n",
"", trk.getAlpha(),
x, trk.getParams()[0], trk.getParams()[1], trk.getParams()[4], trk.getParams()[4], trk.getParams()[2], trk.getParams()[2], trk.getParams()[3],
"", sqrtf(trk.getCov()[0]), sqrtf(trk.getCov()[2]), sqrtf(trk.getCov()[5]), sqrtf(trk.getCov()[14]), trk.getCov()[10]));
390 static_assert(
"Invalid template");
395 if constexpr (std::is_same_v<S, GPUTPCGMTrackParam>) {
396 float alpha = prop.GetAlpha();
397 trk.MoveToReference(prop, *mPparam,
alpha);
398 trk.NormalizeAlpha(
alpha);
399 prop.SetAlpha(
alpha);
400 }
else if constexpr (std::is_same_v<S, TrackParCov>) {
401 static constexpr float kDeg2Rad = M_PI / 180.f;
402 static constexpr float kSectAngle = 2 * M_PI / 18.f;
403 if (mPparam->rec.tpc.trackReferenceX <= 500) {
404 if (prop->PropagateToXBxByBz(trk, mPparam->rec.tpc.trackReferenceX)) {
405 if (CAMath::Abs(trk.getY()) > trk.getX() * CAMath::Tan(kSectAngle / 2.f)) {
406 float newAlpha = trk.getAlpha() + CAMath::Round(CAMath::ATan2(trk.getY(), trk.getX()) / kDeg2Rad / 20.f) * kSectAngle;
407 GPUTPCGMTrackParam::NormalizeAlpha(newAlpha);
408 trk.rotate(newAlpha) && prop->PropagateToXBxByBz(trk, mPparam->rec.tpc.trackReferenceX);
413 static_assert(
"Invalid template");
416 convertTrack<T, S, typename internal::refitTrackTypes<S>::propagator>(
trkX, trk, prop, &TrackParCovChi2);