28#include <nlohmann/json.hpp>
34 nearZero(
const double & v){
35 return std::abs(v)<=std::numeric_limits<double>::min();
44static 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,
", "));
62 base_class(algname, name, ifc)
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)
145 m_monitoring->parseTruthInfo(getTruthTracks(),(hits.size() < 100));
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());
251 m_monitoring->pairFilterCheck(pairs, filteredpairs, passedPairFilter);
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());
385 std::vector<IntermediateState> states{
m_binnedhits->getNLayers()+1};
386 for (
unsigned lyridx = 0; lyridx <
m_binnedhits->getNLayers(); lyridx++) {
388 updateState(states[lyridx] , states[lyridx+1], lyridx, hitsByLayer[lyr]);
392 output_pairsets=states[
m_binnedhits->getNLayers()].pairsets;
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) :
537 m_name(std::move(name)), m_val(val), m_cut(cut), m_histset(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),
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);
626 double chi2,chi2_phi,chi2_eta;
627 bool inBin =
fitRoad(hits, idx, fittedpars,
chi2, chi2_phi,chi2_eta);
630 m_roads.emplace_back(std::make_unique<FPGATrackSimRoad>());
635 r->setHits(std::move(sorted_hits));
645 r->setHitLayers(hitLayers);
649 r->setFitParams(fittedpars);
651 r->setFitChi2_2d(chi2_phi,chi2_eta);
696 double dr = (
lastpair().second->hitptr->getR() -
pair.first->hitptr->getR()) / 2.0;
698 double newpairextrap =
pair.first->phiShift +
pair.dPhi() /
pair.dR() * dr;
699 return lastpairextrap - newpairextrap;
710 double dr = (
lastpair().second->hitptr->getR() -
pair.first->hitptr->getR()) / 2.0;
712 double newpairextrap =
pair.first->etaShift +
pair.dEta() /
pair.dR() * dr;
713 return lastpairextrap - newpairextrap;
744 double r = std::min(
lastpair().first->hitptr->getR(),
lastpair().second->hitptr->getR());
749 double r = std::max(
lastpair().first->hitptr->getR(),
lastpair().second->hitptr->getR());
758 double N =hits.size();
762 double sum_PhiR2 = 0;
775 double r = hit->hitptr->getR();
779 double dphi = hit->phiShift;
780 double dphi2 = dphi*dphi;
781 double deta = hit->etaShift;
782 double deta2 = deta*deta;
787 sum_PhiR2 += dphi*r2;
803 double r6_t0 = (-sum_R2 * sum_R4 + sum_R3*sum_R3);
804 double r5_t0 = (sum_R*sum_R4 - sum_R2*sum_R3);
805 double r4_t0 = (-sum_R * sum_R3 + sum_R2*sum_R2);
806 double r4_t1 = (-N*sum_R4 + sum_R2*sum_R2);
807 double r3_t0 = (N*sum_R3 - sum_R*sum_R2);
808 double r2_t0 = (-N*sum_R2 + sum_R*sum_R);
811 const double denom_phi = N * r6_t0 + sum_R * r5_t0 + sum_R2 * r4_t0;
812 if (nearZero(denom_phi)){
813 ATH_MSG_ERROR(
"Divide by zero (phi) trapped in FPGATrackSimGenScanTool::fitRoad");
818 std::vector<double> phivars(3);
819 phivars[0] = (sum_Phi*r6_t0 + sum_PhiR*r5_t0 + sum_PhiR2*r4_t0)/denom_phi;
820 phivars[1] = (sum_Phi*r5_t0 + sum_PhiR*r4_t1 + sum_PhiR2*r3_t0)/denom_phi;
821 phivars[2] = (sum_Phi*r4_t0 + sum_PhiR*r3_t0 + sum_PhiR2*r2_t0)/denom_phi;
825 const double denom_eta = N*sum_R2 - sum_R*sum_R;
826 if (nearZero(denom_eta)){
827 ATH_MSG_ERROR(
"Divide by zero (eta) trapped in FPGATrackSimGenScanTool::fitRoad");
831 std::vector<double> etavars(2);
832 etavars[0] = (-sum_R*sum_EtaR + sum_R2*sum_Eta)/denom_eta;
833 etavars[1] = (N*sum_EtaR - sum_R*sum_Eta)/denom_eta;
838 parshift[0] = etavars[0] + etavars[1]*
m_rin;
839 parshift[1]= etavars[0] + etavars[1]*
m_rout;
840 parshift[2]= -(phivars[0]/
m_rin + phivars[1] + phivars[2]*
m_rin) ;
841 parshift[3]= -(phivars[0]/
m_rout + phivars[1] + phivars[2]*
m_rout) ;
843 parshift[4]= -phivars[2]/4.0*
y*
y ;
846 const auto& lastStep =
m_binnedhits->getBinTool().lastStep();
848 parset[par]+=parshift[par];
849 inBin = inBin && (std::abs(parshift[par]) < lastStep->binWidth(par)/2.0);
852 double ec0 = etavars[0];
853 double ec1 = etavars[1];
854 eta_chi2 = sum_Eta2 - 2*ec0*sum_Eta - 2*ec1*sum_EtaR + N*ec0*ec0 + 2*ec0*ec1*sum_R + ec1*ec1*sum_R2;
856 double pc0 = phivars[0];
857 double pc1 = phivars[1];
858 double pc2 = phivars[2];
860 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;
864 double r = hit->hitptr->getR();
865 ATH_MSG_VERBOSE(
"Fitted r= " <<
r <<
" phishift " << hit->phiShift <<
" =?= " << pc0+pc1*
r+pc2*
r*
r <<
" etashift " << hit->etaShift <<
" =?= " << ec0+ec1*
r);
869 ATH_MSG_DEBUG(
"Fitted parset inBin="<< inBin <<
" nhits=" << hits.size() <<
" " << parset <<
" chi2 " << eta_chi2 <<
"," << phi_chi2);
872 trackpars =
m_binnedhits->getBinTool().binDesc()->parSetToTrackPars(parset);
878 chi2 = (hits.size() == 5) ?
888 std::vector<unsigned> toUse;
892 if (hitmask == 0x1f) {
895 else if (hitmask == 0x1e) {
898 else if (hitmask == 0x1d) {
901 else if (hitmask == 0x1b) {
904 else if (hitmask == 0x17) {
907 else if (hitmask == 0x0f) {
914 if (hitmask == 0x1f) {
917 else if (hitmask == 0x1e) {
920 else if (hitmask == 0x1d) {
923 else if (hitmask == 0x1b) {
926 else if (hitmask == 0x17) {
929 else if (hitmask == 0x0f) {
936 if (hitmask == 0x1f) {
939 else if (hitmask == 0x1e) {
942 else if (hitmask == 0x1d) {
945 else if (hitmask == 0x1b) {
948 else if (hitmask == 0x17) {
951 else if (hitmask == 0x0f) {
958 if (hitmask == 0x1f) {
961 else if (hitmask == 0x1e) {
964 else if (hitmask == 0x1d) {
967 else if (hitmask == 0x1b) {
970 else if (hitmask == 0x17) {
973 else if (hitmask == 0x0f) {
#define ATH_CHECK
Evaluate an expression and check for errors.
#define ATH_MSG_VERBOSE(x)
std::vector< size_t > vec
Binning Utilities for GenScanTool.
This is the monitoring for the FPGATrackSimGenScanTool.
: FPGATrackSim-specific class to represent an hit in the detector.
Maps physical layers to logical layers.
Maps ITK module indices to FPGATrackSim regions.
Structs that store the 5 track parameters.
double chi2(TH1 *h0, TH1 *h1)
void reverse(typename DataModel_detail::iterator< DVL > beg, typename DataModel_detail::iterator< DVL > end)
Specialization of reverse for DataVector/List.
std::shared_ptr< const FPGATrackSimHit > hitptr
std::vector< FPGATrackSimBinUtil::StoredHit > hits
void fill(H5::Group &out_file, size_t iterations)