@@ -317,6 +317,21 @@ struct HfTaskCharmHadronsFemtoDream {
317317 continue ;
318318 }
319319
320+ float kstar = FemtoDreamMath::getkstar (p1, massOne, p2, massTwo);
321+ if (kstar > highkstarCut) {
322+ continue ;
323+ }
324+
325+ float invMass = getCharmHadronMass (p2);
326+
327+ if (invMass < charmHadMinInvMass || invMass > charmHadMaxInvMass) {
328+ continue ;
329+ }
330+
331+ if (p2.pt () < charmHadMinPt || p2.pt () > charmHadMaxPt) {
332+ continue ;
333+ }
334+
320335 constexpr int CutBitChargePositive = 2 ;
321336 // proton track charge
322337 float chargeTrack = 0 .;
@@ -332,20 +347,6 @@ struct HfTaskCharmHadronsFemtoDream {
332347 pairSign = UnLikeSignPair;
333348 }
334349
335- float kstar = FemtoDreamMath::getkstar (p1, massOne, p2, massTwo);
336- if (kstar > highkstarCut) {
337- continue ;
338- }
339-
340- float invMass = getCharmHadronMass (p2);
341-
342- if (invMass < charmHadMinInvMass || invMass > charmHadMaxInvMass) {
343- continue ;
344- }
345-
346- if (p2.pt () < charmHadMinPt || p2.pt () > charmHadMaxPt) {
347- continue ;
348- }
349350 // / Filling QA histograms of the selected tracks
350351 selectedTrackHisto.fillQA <isMc, true >(p1, static_cast <aod::femtodreamparticle::MomentumType>(confTempFitVarMomentum.value ), col.multNtr (), col.multV0M ());
351352
@@ -425,13 +426,14 @@ struct HfTaskCharmHadronsFemtoDream {
425426 if (p2.pt () < charmHadMinPt || p2.pt () > charmHadMaxPt) {
426427 continue ;
427428 }
429+ constexpr int CutBitChargePositive = 2 ;
430+ // proton track charge
428431 float chargeTrack = 0 .;
429- if ((p1.cut () & 2 ) == 2 ) {
432+ if ((p1.cut () & CutBitChargePositive ) == CutBitChargePositive ) {
430433 chargeTrack = PositiveCharge;
431434 } else {
432435 chargeTrack = NegativeCharge;
433436 }
434-
435437 int pairSign = 0 ;
436438 if (chargeTrack == p2.charge ()) {
437439 pairSign = LikeSignPair;
@@ -445,6 +447,7 @@ struct HfTaskCharmHadronsFemtoDream {
445447 charmHadMc = p2.flagMc ();
446448 originType = p2.originMcRec ();
447449 }
450+
448451 rowFemtoResultPairs (
449452 invMass,
450453 p2.pt (),
0 commit comments