456 if (
hit.towerIEta() == -1016 &&
hit.towerIPhi() == -962) {
461 SimpleTowerHit l1Hit;
462 l1Hit.isBarrel = 0x1;
467 l1Hit.towerIeta =
hit.towerIEta();
468 l1Hit.towerIphi =
hit.towerIPhi();
474 <<
" TOWERS WITH towerIeta=-1016 AND towerIphi=-962" << std::endl;
479 SimpleTowerHit l1Hit;
480 l1Hit.isBarrel = 0x0;
481 l1Hit.l1egTowerEt = 0.0;
493 return a.towerEt >
b.towerEt;
497 std::vector<SimpleHGCluster> AllHGClusters;
501 auto& cl3d = *cl3dIt;
506 l1t::PFCluster cluster(cl3d.pt(), cl3d.eta(), cl3d.phi(), cl3d.hOverE());
510 float pt_new = cl3d.iPt(l1t::HGCalMulticluster::EnergyInterpretation::EM);
512 cluster =
l1t::PFCluster(pt_new, cl3d.eta(), cl3d.phi(), hoe_new, isEM);
516 float had_old = cl3d.
pt() - cluster.emEt();
517 float em_new = cl3d.iPt(l1t::HGCalMulticluster::EnergyInterpretation::EM);
518 float pt_new = had_old + em_new;
519 float hoe_new = em_new > 0 ? (had_old / em_new) : -1;
520 cluster =
l1t::PFCluster(pt_new, cl3d.eta(), cl3d.phi(), hoe_new, isEM);
523 float em_new = cl3d.iPt(l1t::HGCalMulticluster::EnergyInterpretation::EM);
524 float hoe_new = em_new > 0 ? (cl3d.pt() / em_new - 1) : -1;
525 cluster =
l1t::PFCluster(cl3d.pt(), cl3d.eta(), cl3d.phi(), hoe_new, isEM);
531 idScore = cluster.egVsPUMVAOut();
537 SimpleHGCluster HGCluster;
541 HGCluster.showerlength = cl3d.showerLength();
542 HGCluster.coreshowerlength = cl3d.coreShowerLength();
549 AllHGClusters.push_back(HGCluster);
553 std::sort(
begin(AllHGClusters),
end(AllHGClusters), [](
const SimpleHGCluster&
a, SimpleHGCluster&
b) {
565 std::vector<InputTowerCluster> l1TowerClustersNxM_CB;
566 std::vector<InputTowerCluster_pstn> l1TowerClustersNxM_CB_pstn;
567 std::vector<InputTowerCluster> l1TowerClustersNxM_CE;
568 std::vector<InputTowerCluster_pstn> l1TowerClustersNxM_CE_pstn;
569 std::vector<InputHGCluster> HGClusters;
572 std::vector<InputTowerCluster> AllL1TowerClustersNxM_CE;
573 std::vector<InputTowerCluster_pstn> AllL1TowerClustersNxM_CE_pstn;
575 int Nclusters_CB = 0;
576 int AllNclusters_CE = 0;
577 bool caloTauSeedingFinished =
false;
579 while (!caloTauSeedingFinished) {
580 InputTowerCluster clNxM;
582 InputTowerCluster_pstn clNxM_pstn;
589 l1CaloTower.stale4seed) {
597 caloTauSeedingFinished =
true;
601 clNxM.fill(
seedIdx, l1CaloTower);
602 clNxM_pstn.fill(l1CaloTower);
603 if (l1CaloTower.isBarrel) {
604 clNxM.barrelSeeded = 0x1;
607 l1CaloTower.stale4seed = 0x1;
608 l1CaloTower.stale = 0x1;
615 dIEtaPhi_t d_iPhi = dPhi<dIEtaPhi_t, IPhi_t>(l1CaloTower.towerIphi, clNxM_pstn.seedIphi);
619 l1CaloTower.stale4seed = 0x1;
627 l1TowerClustersNxM_CB.push_back(clNxM);
628 l1TowerClustersNxM_CB_pstn.push_back(clNxM_pstn);
631 AllL1TowerClustersNxM_CE.push_back(clNxM);
632 AllL1TowerClustersNxM_CE_pstn.push_back(clNxM_pstn);
640 for (
int clNxMIdx = 0; clNxMIdx < Nclusters_CB; clNxMIdx++) {
643 if (l1CaloTower.stale) {
647 dIEtaPhi_t d_iEta =
tower_dIEta(l1CaloTower.towerIeta, l1TowerClustersNxM_CB_pstn[clNxMIdx].seedIeta);
649 dPhi<dIEtaPhi_t, IPhi_t>(l1CaloTower.towerIphi, l1TowerClustersNxM_CB_pstn[clNxMIdx].seedIphi);
650 int hitIdx = d_iEta * 9 + d_iPhi +
seedIdx;
654 l1TowerClustersNxM_CB[clNxMIdx].fill(hitIdx, l1CaloTower);
655 l1CaloTower.stale = 0x1;
664 int Nclusters_CE = 0;
665 for (
int clNxMIdx = 0; clNxMIdx < AllNclusters_CE; clNxMIdx++) {
667 for (
auto& HGCluster : AllHGClusters) {
674 dEtaPhi_t d_iEta =
tw2cl_dEta(HGCluster.eta, AllL1TowerClustersNxM_CE_pstn[clNxMIdx].seedIeta);
675 dEtaPhi_t d_iPhi =
tw2cl_dPhi(HGCluster.phi, AllL1TowerClustersNxM_CE_pstn[clNxMIdx].seedIphi);
679 HGCluster.stale = 0x1;
681 cl3d.fill(HGCluster);
682 HGClusters.push_back(cl3d);
683 l1TowerClustersNxM_CE.push_back(AllL1TowerClustersNxM_CE[clNxMIdx]);
684 l1TowerClustersNxM_CE_pstn.push_back(AllL1TowerClustersNxM_CE_pstn[clNxMIdx]);
693 for (
int clNxMIdx = 0; clNxMIdx < Nclusters_CE; clNxMIdx++) {
696 if (l1CaloTower.stale) {
700 dIEtaPhi_t d_iEta =
tower_dIEta(l1CaloTower.towerIeta, l1TowerClustersNxM_CE_pstn[clNxMIdx].seedIeta);
702 dPhi<dIEtaPhi_t, IPhi_t>(l1CaloTower.towerIphi, l1TowerClustersNxM_CE_pstn[clNxMIdx].seedIphi);
703 int hitIdx = d_iEta * 9 + d_iPhi +
seedIdx;
707 l1TowerClustersNxM_CE[clNxMIdx].fill(hitIdx, l1CaloTower);
708 l1CaloTower.stale = 0x1;
717 int batchSize_CB = (
int)(Nclusters_CB);
718 tensorflow::TensorShape imageShape_CB({batchSize_CB,
IEta_dim,
IPhi_dim, 2});
719 tensorflow::TensorShape positionShape_CB({batchSize_CB, 2});
720 tensorflow::Tensor TowerClusterImage_CB(tensorflow::DT_FLOAT, imageShape_CB);
721 tensorflow::Tensor TowerClusterPosition_CB(tensorflow::DT_FLOAT, positionShape_CB);
723 for (
int clNxMIdx = 0; clNxMIdx < Nclusters_CB; clNxMIdx++) {
728 TowerClusterImage_CB.tensor<
float, 4>()(clNxMIdx,
eta,
phi, 0) =
729 (l1TowerClustersNxM_CB[clNxMIdx].towerHits[towerIdx].l1egTowerEt.to_float() +
730 l1TowerClustersNxM_CB[clNxMIdx].towerHits[towerIdx].towerEm.to_float());
731 TowerClusterImage_CB.tensor<
float, 4>()(clNxMIdx,
eta,
phi, 1) =
732 (l1TowerClustersNxM_CB[clNxMIdx].towerHits[towerIdx].towerHad.to_float());
736 TowerClusterPosition_CB.tensor<
float, 2>()(clNxMIdx, 0) =
floatIEta(l1TowerClustersNxM_CB_pstn[clNxMIdx].seedIeta);
737 TowerClusterPosition_CB.tensor<
float, 2>()(clNxMIdx, 1) =
floatIPhi(l1TowerClustersNxM_CB_pstn[clNxMIdx].seedIphi);
745 {
"TowerClusterPosition", TowerClusterPosition_CB}};
746 std::vector<tensorflow::Tensor> CNNmodel_CBoutputs;
748 CNNmodel_CBinputList,
749 {
"TauMinator_CB_conv/middleMan/concat"},
750 &CNNmodel_CBoutputs);
754 std::vector<tensorflow::Tensor> DNN_CBoutputsIdent;
757 {
"TauMinator_CB_ident/sigmoid_IDout/Sigmoid"},
758 &DNN_CBoutputsIdent);
761 std::vector<tensorflow::Tensor> DNN_CBoutputsCalib;
764 {
"TauMinator_CB_calib/DNNout/MatMul"},
765 &DNN_CBoutputsCalib);
768 for (
int clNxMIdx = 0; clNxMIdx < Nclusters_CB; clNxMIdx++) {
770 MakeTauCandidate(
true, clNxMIdx, DNN_CBoutputsIdent, DNN_CBoutputsCalib, l1TowerClustersNxM_CB_pstn);
771 if (
l1Tau.pt() < 0) {
774 L1NNCaloTauCollectionBXV->push_back(0,
l1Tau);
779 int batchSize_CE = (
int)(Nclusters_CE);
780 tensorflow::TensorShape imageShape_CE({batchSize_CE,
IEta_dim,
IPhi_dim, 2});
781 tensorflow::TensorShape positionShape_CE({batchSize_CE, 2});
782 tensorflow::TensorShape cl3dfeatShape_CE({batchSize_CE, 8});
783 tensorflow::Tensor TowerClusterImage_CE(tensorflow::DT_FLOAT, imageShape_CE);
784 tensorflow::Tensor TowerClusterPosition_CE(tensorflow::DT_FLOAT, positionShape_CE);
785 tensorflow::Tensor Cl3dShapeFeatures_CE(tensorflow::DT_FLOAT, cl3dfeatShape_CE);
787 for (
int clNxMIdx = 0; clNxMIdx < Nclusters_CE; clNxMIdx++) {
789 InputHGCluster HGClu = HGClusters[clNxMIdx];
795 TowerClusterImage_CE.tensor<
float, 4>()(clNxMIdx,
eta,
phi, 0) =
796 (l1TowerClustersNxM_CE[clNxMIdx].towerHits[towerIdx].l1egTowerEt.to_float() +
797 l1TowerClustersNxM_CE[clNxMIdx].towerHits[towerIdx].towerEm.to_float());
798 TowerClusterImage_CE.tensor<
float, 4>()(clNxMIdx,
eta,
phi, 1) =
799 (l1TowerClustersNxM_CE[clNxMIdx].towerHits[towerIdx].towerHad.to_float());
803 TowerClusterPosition_CE.tensor<
float, 2>()(clNxMIdx, 0) =
floatIEta(l1TowerClustersNxM_CE_pstn[clNxMIdx].seedIeta);
804 TowerClusterPosition_CE.tensor<
float, 2>()(clNxMIdx, 1) =
floatIPhi(l1TowerClustersNxM_CE_pstn[clNxMIdx].seedIphi);
806 Cl3dShapeFeatures_CE.tensor<
float, 2>()(clNxMIdx, 0) =
inputScaler(HGClu.pt.to_float(),
"pt");
808 Cl3dShapeFeatures_CE.tensor<
float, 2>()(clNxMIdx, 2) =
inputScaler(HGClu.showerlength.to_float(),
"showerlength");
809 Cl3dShapeFeatures_CE.tensor<
float, 2>()(clNxMIdx, 3) =
810 inputScaler(HGClu.coreshowerlength.to_float(),
"coreshowerlength");
822 {
"TowerClusterPosition", TowerClusterPosition_CE},
823 {
"AssociatedCl3dFeatures", Cl3dShapeFeatures_CE}};
824 std::vector<tensorflow::Tensor> CNNmodel_CEoutputs;
826 CNNmodel_CEinputList,
827 {
"TauMinator_CE_conv/middleMan/concat"},
828 &CNNmodel_CEoutputs);
832 std::vector<tensorflow::Tensor> DNN_CEoutputsIdent;
835 {
"TauMinator_CE_ident/sigmoid_IDout/Sigmoid"},
836 &DNN_CEoutputsIdent);
839 std::vector<tensorflow::Tensor> DNN_CEoutputsCalib;
842 {
"TauMinator_CE_calib/LIN_DNNout/Relu"},
843 &DNN_CEoutputsCalib);
846 for (
int clNxMIdx = 0; clNxMIdx < Nclusters_CE; clNxMIdx++) {
848 MakeTauCandidate(
false, clNxMIdx, DNN_CEoutputsIdent, DNN_CEoutputsCalib, l1TowerClustersNxM_CE_pstn);
849 if (
l1Tau.pt() < 0) {
852 L1NNCaloTauCollectionBXV->push_back(0,
l1Tau);
857 iEvent.put(
std::move(L1NNCaloTauCollectionBXV),
"L1NNCaloTauCollectionBXV");
edm::EDGetToken hgcalTowersToken
std::vector< NamedTensor > NamedTensorList
static constexpr int PUID_W
int apintQuantizer(float inputF, float LSB, int nbits)
static constexpr int ET_W
StringCutObjectSelector< l1t::HGCalMulticluster > preEmId
double pt() const final
transverse momentum
static constexpr float MEANZ_LSB
static constexpr int PT_W
edm::Handle< l1tp2::CaloTowerCollection > l1CaloTowerHandle
T const * product() const
dEtaPhi_t tw2cl_dPhi(EtaPhi_t iPhi_1, IPhi_t iPhi_2)
static constexpr int MEANZ_W
float floatEta(EtaPhi_t eta)
float correctInputMeanzCl3d(float meanz)
static constexpr int SHAPEFEAT_W
ap_int< DIETAPHI_W > dIEtaPhi_t
ap_int< DETAPHI_W > dEtaPhi_t
ap_ufixed< PT_W, PT_I, AP_TRN, AP_SAT > Pt_t
float floatSzz(ShapeFeat_t szz)
void run(Session *session, const NamedTensorList &inputs, const std::vector< std::string > &outputNames, std::vector< Tensor > *outputs, const thread::ThreadPoolOptions &threadPoolOptions)
Abs< T >::type abs(const T &t)
static constexpr float PTET_LSB
float inputScaler(float inputF, std::string feature)
IEta_t makeEndcapHwIEta(float eta)
IPhi_t makeEndcapHwIPhi(float phi)
ap_int< W > ap_abs(ap_int< W > x)
dIEtaPhi_t tower_dIEta(IEta_t iEta_1, IEta_t iEta_2)
l1tpf::HGC3DClusterEgID VsPuId
float apfixedQuantizer(float inputF, float LSB, int nbits)
edm::Handle< l1t::HGCalTowerBxCollection > hgcalTowersHandle
static constexpr float PUID_LSB
void setLogging(const std::string &level="3")
float correctInputEtaCl3d(float eta)
static constexpr float SZZ_LSB
static constexpr int R2cone
float floatIPhi(IPhi_t phi)
edm::EDGetTokenT< l1t::HGCalMulticlusterBxCollection > HGClusterToken
float floatIEta(IEta_t eta)
l1t::Tau MakeTauCandidate(bool isBarrel, int clNxMIdx, std::vector< tensorflow::Tensor > outputsIdent, std::vector< tensorflow::Tensor > outputsCalib, std::vector< InputTowerCluster_pstn > clustersNxM_pstn)
float floatShape(ShapeFeat_t shape)
static constexpr float SHAPEFEAT_LSB
edm::Handle< l1t::HGCalMulticlusterBxCollection > HGClusterHandle
Log< level::Warning, false > LogWarning
static constexpr float ETAPHI_LSB
float passID(l1t::HGCalMulticluster c, l1t::PFCluster &cpf)
static constexpr int ETAPHI_W
dEtaPhi_t tw2cl_dEta(EtaPhi_t iEta_1, IEta_t iEta_2)
float floatMeanZHgcalCoord(Meanz_t meanz)
edm::EDGetTokenT< l1tp2::CaloTowerCollection > l1TowersToken
Public Member Functions inherited from edm::stream::EDProducer< edm::GlobalCache< NNmodels_GlobalCache > >
1.8.14