28 #include <nlohmann/json.hpp>
34 nearZero(
const double &
v){
44 static inline std::string
to_string(
const std::vector<T>&
v)
46 std::ostringstream oss;
50 std::copy(
v.begin(),
v.end() - 1, std::ostream_iterator<T>(oss,
", "));
64 declareInterface<IFPGATrackSimRoadFinderTool>(
this);
71 const std::vector<Gaudi::Details::PropertyBase*> props = this->getProperties();
72 for( Gaudi::Details::PropertyBase* prop : props ) {
73 if (prop->ownerTypeName()==this->type()) {
74 ATH_MSG_DEBUG(
"Property:\t" << prop->name() <<
"\t : \t" << prop->toString());
88 if (nLogicalLayers == 0){
89 ATH_MSG_ERROR(
"Number of logical layers is zero in FPGATrackSimGenScanTool::initialize");
90 return StatusCode::FAILURE;
103 const int signedSize =
static_cast<int>(
m_binnedhits->getNLayers()) - 1;
119 return StatusCode::FAILURE;
128 return StatusCode::SUCCESS;
135 std::vector<std::shared_ptr<const FPGATrackSimRoad>> &roads)
164 std::vector<HitPairSet> pairsets;
182 if (pairsets.size() >1) {
186 s +=
"(" + std::to_string(hit->
layer) +
"," + std::to_string(hit->
hitptr->getR()) +
"), ";
197 for (std::unique_ptr<FPGATrackSimRoad>&
r :
m_roads) {
198 const std::vector<std::vector<std::shared_ptr<const FPGATrackSimHit>>>& theseHits =
r->getAllHits();
202 std::vector<std::vector<std::shared_ptr<const FPGATrackSimHit>>>
vec(5);
203 for (
size_t ihit = 0; ihit < toUse.size(); ++ihit) {
204 unsigned int layer = toUse[ihit];
205 if (
layer >= theseHits.size() || theseHits[
layer].empty()) {
206 ATH_MSG_ERROR(
"Hit index out of range in keepHitsStrategy: layer=" <<
layer <<
", hits.size()=" << theseHits.size());
207 return StatusCode::FAILURE;
209 vec[ihit].push_back(theseHits[
layer][0]);
211 r->setHits(std::move(
vec));
216 for (
auto &
r :
m_roads) roads.push_back(std::move(
r));
224 return StatusCode::SUCCESS;
229 std::vector<HitPairSet> &output_pairsets)
234 std::vector<std::vector<const StoredHit *>> hitsByLayer(
m_binnedhits->getNLayers());
254 if (passedPairFilter)
257 std::vector<HitPairSet> pairsets;
267 output_pairsets.push_back(pairset);
276 output_pairsets.push_back(std::move(
pairs));
281 output_pairsets.push_back(std::move(filteredpairs));
284 return StatusCode::SUCCESS;
291 const std::vector<const StoredHit *>& newhits)
295 std::vector<bool> pairset_used(inputstate.
pairsets.size(),
false);
297 for (
auto &newhit : newhits) {
300 std::set<const StoredHit *> vetoList;
303 for (
unsigned ps_idx = 0; ps_idx < inputstate.
pairsets.size(); ++ps_idx) {
304 auto &pairset = inputstate.
pairsets[ps_idx];
309 pairset_used[ps_idx]=
true;
310 outputstate.
pairsets.push_back(std::move(newset));
312 for (
auto vetohit : pairset.hitlist) {
313 vetoList.insert(vetohit);
320 if (vetoList.count(prevhit) == 0) {
325 outputstate.
pairsets.push_back(std::move(newset));
332 if (lyridx <= allowed_missed_hits) {
341 for (
unsigned ps_idx = 0; ps_idx < inputstate.
pairsets.size(); ++ps_idx) {
342 auto &pairset = inputstate.
pairsets[ps_idx];
343 if ((!pairset_used[ps_idx])&&(lyridx < (pairset.hitlist.size() + allowed_missed_hits))) {
344 outputstate.
pairsets.push_back(pairset);
360 if (lyridx > allowed_missed_hits) {
373 std::vector<HitPairSet> &output_pairsets)
378 std::vector<std::vector<const StoredHit *>> hitsByLayer(
m_binnedhits->getNLayers());
386 for (
unsigned lyridx = 0; lyridx <
m_binnedhits->getNLayers(); lyridx++) {
396 return StatusCode::SUCCESS;
402 std::vector<std::vector<const StoredHit *>>& hitsByLayer)
407 hitsByLayer.at(hit.
layer).push_back(&hit);
410 return StatusCode::SUCCESS;
423 std::vector<const StoredHit *>
const * lastlyr = 0;
424 std::vector<const StoredHit *>
const * lastlastlyr = 0;
431 for (
const StoredHit *
const &ptr2 : *lastlyr) {
436 for (
const StoredHit *
const &ptr2 : *lastlastlyr) {
442 lastlastlyr = lastlyr;
443 lastlyr = &hitsByLayer[lyr];
447 return StatusCode::SUCCESS;
455 int lyr =
std::min(pair.first->layer,pair.second->layer);
473 return StatusCode::SUCCESS;
479 std::vector<HitPairSet> &pairsets,
489 if ((std::abs(
int(pair.second->layer) -
int(pair.first->layer)) > 1)
490 && (pairset.hasLayer(
std::min(pair.first->layer,pair.second->layer) + 1)))
504 int size = pairset.addPair(pair);
506 ATH_MSG_VERBOSE(
"addPair " << pairsets.size() <<
" " << pairset.pairList.size() <<
" " <<
size);
517 pairsets.push_back(std::move(newpairset));
521 return StatusCode::SUCCESS;
536 cutvar(std::string
name,
double val,
double cut, std::vector<TH1D *>& histset) :
538 bool passed() {
return std::abs(m_val) < m_cut; }
539 void fill(
unsigned cat) { m_histset[cat]->Fill(m_val); }
543 std::vector<TH1D *> &m_histset;
547 std::vector<cutvar> allcuts;
548 allcuts.push_back(cutvar(
"MatchPhi", pairset.
MatchPhi(pair),
551 allcuts.push_back(cutvar(
"MatchEta", pairset.
MatchEta(pair),
554 allcuts.push_back(cutvar(
"DeltaDeltaPhi", pairset.
DeltaDeltaPhi(pair),
557 allcuts.push_back(cutvar(
"DeltaDeltaEta", pairset.
DeltaDeltaEta(pair),
560 allcuts.push_back(cutvar(
"PhiCurvature", pairset.
PhiCurvature(pair),
563 allcuts.push_back(cutvar(
"EtaCurvature", pairset.
EtaCurvature(pair),
567 allcuts.push_back(cutvar(
570 allcuts.push_back(cutvar(
574 allcuts.push_back(cutvar(
577 allcuts.push_back(cutvar(
582 unsigned passedCuts = std::count_if(allcuts.begin(), allcuts.end(),
583 [](cutvar&
cut) { return cut.passed(); });
584 bool passedAll = (passedCuts == allcuts.size());
587 bool passedAllButOne = (passedCuts == allcuts.size() - 1);
588 for (cutvar&
cut: allcuts) {
592 (passedAll || (passedAllButOne && !
cut.passed())));
598 for (cutvar &
cut : allcuts)
600 s +=
cut.m_name +
" : (" +
cut.passed() +
", " +
cut.m_val +
"), ";
604 << *pairset.
lastpair().second <<
"\n "
605 << *pair.first <<
"\n "
616 std::vector<std::vector<std::shared_ptr<const FPGATrackSimHit>>>
617 sorted_hits(
m_binnedhits->getNLayers(),std::vector<std::shared_ptr<const FPGATrackSimHit>>());
620 hitLayers |= 1 << hit->layer;
621 sorted_hits[hit->layer].push_back(hit->hitptr);
630 m_roads.emplace_back(std::make_unique<FPGATrackSimRoad>());
635 r->setHits(std::move(sorted_hits));
645 r->setHitLayers(hitLayers);
649 r->setFitParams(fittedpars);
668 if (pairList.size() > 0)
671 LastPhiCurvature = PhiCurvature(pair);
672 LastEtaCurvature = EtaCurvature(pair);
675 pairList.push_back(pair);
677 hitLayers |= (0x1 << pair.first->layer);
678 hitLayers |= (0x1 << pair.second->layer);
679 if (!hasHit(pair.first))
680 hitlist.push_back(pair.first);
681 if (!hasHit(pair.second))
682 hitlist.push_back(pair.second);
683 return this->pairList.size();
689 if ((lastpair().
first->hitptr==pair.first->hitptr)||
690 (lastpair().
first->hitptr==pair.second->hitptr)||
691 (lastpair().
second->hitptr==pair.first->hitptr)||
692 (lastpair().
second->hitptr==pair.second->hitptr))
695 double dr = (lastpair().second->hitptr->getR() - pair.first->hitptr->getR()) / 2.0;
696 double lastpairextrap = lastpair().second->phiShift - lastpair().dPhi() / lastpair().dR() *
dr;
697 double newpairextrap = pair.first->phiShift + pair.
dPhi() / pair.
dR() *
dr;
698 return lastpairextrap - newpairextrap;
703 if ((lastpair().
first->hitptr==pair.first->hitptr)||
704 (lastpair().
first->hitptr==pair.second->hitptr)||
705 (lastpair().
second->hitptr==pair.first->hitptr)||
706 (lastpair().
second->hitptr==pair.second->hitptr))
709 double dr = (lastpair().second->hitptr->getR() - pair.first->hitptr->getR()) / 2.0;
710 double lastpairextrap = lastpair().second->etaShift - lastpair().dEta() / lastpair().dR() *
dr;
711 double newpairextrap = pair.first->etaShift + pair.
dEta() / pair.
dR() *
dr;
712 return lastpairextrap - newpairextrap;
717 return (pair.
dPhi() * lastpair().dR() - lastpair().
dPhi() * pair.
dR()) / (lastpair().dR() * pair.
dR());
722 return (pair.
dEta() * lastpair().dR() - lastpair().
dEta() * pair.
dR()) / (lastpair().dR() * pair.
dR());
727 return 2 * DeltaDeltaPhi(pair) / (lastpair().dR() + pair.
dR());
731 return 2 * DeltaDeltaEta(pair) / (lastpair().dR() + pair.
dR());
735 return PhiCurvature(pair) - LastPhiCurvature;
739 return EtaCurvature(pair) - LastEtaCurvature;
744 return lastpair().PhiInExtrap(r_in) + 0.5 * PhiCurvature(pair) * (r_in -
r) * (r_in -
r);
749 return pair.
PhiOutExtrap(r_out) + 0.5 * PhiCurvature(pair) * (r_out -
r) * (r_out -
r);
757 double N =
hits.size();
761 double sum_PhiR2 = 0;
774 double r = hit->hitptr->getR();
778 double dphi = hit->phiShift;
779 double dphi2 = dphi*dphi;
780 double deta = hit->etaShift;
781 double deta2 = deta*deta;
786 sum_PhiR2 += dphi*r2;
802 double r6_t0 = (-sum_R2 * sum_R4 + sum_R3*sum_R3);
803 double r5_t0 = (sum_R*sum_R4 - sum_R2*sum_R3);
804 double r4_t0 = (-sum_R * sum_R3 + sum_R2*sum_R2);
805 double r4_t1 = (-
N*sum_R4 + sum_R2*sum_R2);
806 double r3_t0 = (
N*sum_R3 - sum_R*sum_R2);
807 double r2_t0 = (-
N*sum_R2 + sum_R*sum_R);
810 const double denom_phi =
N * r6_t0 + sum_R * r5_t0 + sum_R2 * r4_t0;
811 if (nearZero(denom_phi)){
812 ATH_MSG_ERROR(
"Divide by zero (phi) trapped in FPGATrackSimGenScanTool::fitRoad");
817 std::vector<double> phivars(3);
818 phivars[0] = (sum_Phi*r6_t0 + sum_PhiR*r5_t0 + sum_PhiR2*r4_t0)/denom_phi;
819 phivars[1] = (sum_Phi*r5_t0 + sum_PhiR*r4_t1 + sum_PhiR2*r3_t0)/denom_phi;
820 phivars[2] = (sum_Phi*r4_t0 + sum_PhiR*r3_t0 + sum_PhiR2*r2_t0)/denom_phi;
824 const double denom_eta =
N*sum_R2 - sum_R*sum_R;
825 if (nearZero(denom_eta)){
826 ATH_MSG_ERROR(
"Divide by zero (eta) trapped in FPGATrackSimGenScanTool::fitRoad");
830 std::vector<double> etavars(2);
831 etavars[0] = (-sum_R*sum_EtaR + sum_R2*sum_Eta)/denom_eta;
832 etavars[1] = (
N*sum_EtaR - sum_R*sum_Eta)/denom_eta;
837 parshift[0] = etavars[0] + etavars[1]*
m_rin;
838 parshift[1]= etavars[0] + etavars[1]*
m_rout;
839 parshift[2]= -(phivars[0]/
m_rin + phivars[1] + phivars[2]*
m_rin) ;
840 parshift[3]= -(phivars[0]/
m_rout + phivars[1] + phivars[2]*
m_rout) ;
842 parshift[4]= -phivars[2]/4.0*
y*
y ;
845 const auto& lastStep =
m_binnedhits->getBinTool().lastStep();
847 parset[
par]+=parshift[
par];
848 inBin = inBin && (std::abs(parshift[
par]) < lastStep->binWidth(
par)/2.0);
851 double ec0 = etavars[0];
852 double ec1 = etavars[1];
853 double eta_chi2 = sum_Eta2 - 2*ec0*sum_Eta - 2*ec1*sum_EtaR +
N*ec0*ec0 + 2*ec0*ec1*sum_R + ec1*ec1*sum_R2;
855 double pc0 = phivars[0];
856 double pc1 = phivars[1];
857 double pc2 = phivars[2];
859 double phi_chi2 = sum_Phi2 - 2*pc0*sum_Phi - 2*pc1*sum_PhiR - 2*pc2*sum_PhiR2 +
N*pc0*pc0 + 2*pc0*pc1*sum_R + 2*pc0*pc2*sum_R2 + 2*pc1*pc2*sum_R3 + pc1*pc1*sum_R2 + pc2*pc2*sum_R4;
863 double r = hit->hitptr->getR();
864 ATH_MSG_VERBOSE(
"Fitted r= " <<
r <<
" phishift " << hit->phiShift <<
" =?= " << pc0+pc1*
r+pc2*
r*
r <<
" etashift " << hit->etaShift <<
" =?= " << ec0+ec1*
r);
868 ATH_MSG_DEBUG(
"Fitted parset inBin="<< inBin <<
" nhits=" <<
hits.size() <<
" " << parset <<
" chi2 " << eta_chi2 <<
"," << phi_chi2);
871 trackpars =
m_binnedhits->getBinTool().binDesc()->parSetToTrackPars(parset);
887 std::vector<unsigned> toUse;
891 if (hitmask == 0x1f) {
894 else if (hitmask == 0x1e) {
897 else if (hitmask == 0x1d) {
900 else if (hitmask == 0x1b) {
903 else if (hitmask == 0x17) {
906 else if (hitmask == 0x0f) {
913 if (hitmask == 0x1f) {
916 else if (hitmask == 0x1e) {
919 else if (hitmask == 0x1d) {
922 else if (hitmask == 0x1b) {
925 else if (hitmask == 0x17) {
928 else if (hitmask == 0x0f) {
935 if (hitmask == 0x1f) {
938 else if (hitmask == 0x1e) {
941 else if (hitmask == 0x1d) {
944 else if (hitmask == 0x1b) {
947 else if (hitmask == 0x17) {
950 else if (hitmask == 0x0f) {
957 if (hitmask == 0x1f) {
960 else if (hitmask == 0x1e) {
963 else if (hitmask == 0x1d) {
966 else if (hitmask == 0x1b) {
969 else if (hitmask == 0x17) {
972 else if (hitmask == 0x0f) {