PtAssignmentNNRegression Class Reference

#include <PtAssignmentNNRegression.h>

Inheritance diagram for PtAssignmentNNRegression:

Public Member Functions
std::vector< float >	getPts (AlgoMuons::value_type &algoMuon, std::vector< std::unique_ptr< IOMTFEmulationObserver > > &observers) override
	PtAssignmentNNRegression (const edm::ParameterSet &edmCfg, const OMTFConfiguration *omtfConfig, std::string networkFile)
	~PtAssignmentNNRegression () override=default
Public Member Functions inherited from PtAssignmentBase
	PtAssignmentBase (const OMTFConfiguration *omtfConfig)
virtual	~PtAssignmentBase ()

Private Attributes
unique_ptr< lutNN::LutNetworkFixedPointRegressionBase >	lutNetworkFP

Additional Inherited Members
Protected Attributes inherited from PtAssignmentBase
const OMTFConfiguration *	omtfConfig = nullptr

Detailed Description

Definition at line 14 of file PtAssignmentNNRegression.h.

Constructor & Destructor Documentation

PtAssignmentNNRegression()

PtAssignmentNNRegression::PtAssignmentNNRegression	(	const edm::ParameterSet &	edmCfg,
		const OMTFConfiguration *	omtfConfig,
		std::string	networkFile
	)

Definition at line 79 of file PtAssignmentNNRegression.cc.

References lutNetworkFP.

    : PtAssignmentBase(omtfConfig), lutNetworkFP(make_unique<lutNN::LutNetworkFP>()) {
  std::ifstream ifs(networkFile);

  edm::LogImportant("OMTFReconstruction")
      << " " << __FUNCTION__ << ":" << __LINE__ << " networkFile " << networkFile << std::endl;

  lutNetworkFP->load(networkFile);

  edm::LogImportant("OMTFReconstruction") << " " << __FUNCTION__ << ":" << __LINE__ << std::endl;
}

~PtAssignmentNNRegression()

PtAssignmentNNRegression::~PtAssignmentNNRegression ( )

overridedefault

Member Function Documentation

getPts()

std::vector< float > PtAssignmentNNRegression::getPts ( AlgoMuons::value_type &  algoMuon, std::vector< std::unique_ptr< IOMTFEmulationObserver > > &  observers  )

overridevirtual

Implements PtAssignmentBase.

Definition at line 169 of file PtAssignmentNNRegression.cc.

References CSCDetId::chamber(), MuonSubdetId::CSC, hcalRecHitTable_cff::detId, alignBH_cfg::fixed, OMTFConfiguration::getRefToLogicNumber(), mps_fire::i, PixelMapPlotter::inputs, createfilelist::int, OMTFConfiguration::isBendingLayer(), LogTrace, lutNetworkFP, PtAssignmentBase::omtfConfig, omtfHitToEventInput(), DiDispStaMuonMonitor_cfi::pt, RPCpg::pts, and hit::z.

                                                                                                                 {
  LogTrace("l1tOmtfEventPrint") << " " << __FUNCTION__ << ":" << __LINE__ << std::endl;
  auto& gpResult = algoMuon->getGpResultConstr();
  //int pdfMiddle = 1<<(omtfConfig->nPdfAddrBits()-1);

  LogTrace("l1tOmtfEventPrint") << " " << __FUNCTION__ << ":" << __LINE__ << std::endl;
  /*
  edm::LogVerbatim("l1tOmtfEventPrint")<<"DataROOTDumper2:;observeEventEnd muonPt "<<event.muonPt<<" muonCharge "<<event.muonCharge
      <<" omtfPt "<<event.omtfPt<<" RefLayer "<<event.omtfRefLayer<<" omtfPtCont "<<event.omtfPtCont
      <<std::endl;
*/

  unsigned int inputCnt = 18;  //TDOO!!!!!
  unsigned int outputCnt = 2;
  const float noHitVal = 1023.;

  //edm::LogImportant("OMTFReconstruction") <<"\n----------------------"<<endl;
  //edm::LogImportant("OMTFReconstruction") <<(*algoMuon)<<std::endl;

  std::vector<float> inputs(inputCnt, noHitVal);

  for (unsigned int iLogicLayer = 0; iLogicLayer < gpResult.getStubResults().size(); ++iLogicLayer) {
    auto& stubResult = gpResult.getStubResults()[iLogicLayer];
    if (stubResult.getMuonStub()) {  //&& stubResult.getValid() //TODO!!!!!!!!!!!!!!!!1
      int hitPhi = stubResult.getMuonStub()->phiHw;
      unsigned int refLayerLogicNum = omtfConfig->getRefToLogicNumber()[algoMuon->getRefLayer()];
      int phiRefHit = gpResult.getStubResults()[refLayerLogicNum].getMuonStub()->phiHw;

      if (omtfConfig->isBendingLayer(iLogicLayer)) {
        hitPhi = stubResult.getMuonStub()->phiBHw;
        phiRefHit = 0;  //phi ref hit for the banding layer set to 0, since it should not be included in the phiDist
      }

      OmtfHit hit(0);

      hit.layer = iLogicLayer;
      hit.quality = stubResult.getMuonStub()->qualityHw;
      hit.eta = stubResult.getMuonStub()->etaHw;  //in which scale?
      hit.valid = stubResult.getValid();

      //phiDist = hitPhi - phiRefHit;
      hit.phiDist = hitPhi - phiRefHit;

      /*
      LogTrace("l1tOmtfEventPrint") <<" muonPt "<<event.muonPt<<" omtfPt "<<event.omtfPt<<" RefLayer "<<event.omtfRefLayer
          <<" layer "<<int(hit.layer)<<" PdfBin "<<stubResult.getPdfBin()<<" hit.phiDist "<<hit.phiDist<<" valid "<<stubResult.getValid()<<" " //<<" phiDist "<<phiDist
          <<" getDistPhiBitShift "<<omtfCand->getGoldenPatern()->getDistPhiBitShift(iLogicLayer, omtfCand->getRefLayer())
          <<" meanDistPhiValue   "<<omtfCand->getGoldenPatern()->meanDistPhiValue(iLogicLayer, omtfCand->getRefLayer())//<<(phiDist != hit.phiDist? "!!!!!!!<<<<<" : "")
          <<endl;*/

      if (hit.phiDist > 504 || hit.phiDist < -512) {
        edm::LogVerbatim("l1tOmtfEventPrint")
            //<<" muonPt "<<event.muonPt<<" omtfPt "<<event.omtfPt<<" RefLayer "<<event.omtfRefLayer
            << " layer " << int(hit.layer) << " hit.phiDist " << hit.phiDist << " valid " << stubResult.getValid()
            << " !!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
      }

      DetId detId(stubResult.getMuonStub()->detId);
      if (detId.subdetId() == MuonSubdetId::CSC) {
        CSCDetId cscId(detId);
        hit.z = cscId.chamber() % 2;
      }

      LogTrace("l1tOmtfEventPrint") << "hit: layer " << (int)hit.layer << " quality " << (int)hit.quality << " eta "
                                    << (int)hit.eta << " valid " << (int)hit.valid << " phiDist " << (int)hit.phiDist
                                    << " z " << (int)hit.z << std::endl;

      omtfHitToEventInput(hit, inputs, algoMuon->getRefLayer(), false);
    }
  }

  LogTrace("l1tOmtfEventPrint") << " " << __FUNCTION__ << ":" << __LINE__ << std::endl;

  std::vector<double> nnResult(outputCnt);
  lutNetworkFP->run(inputs, noHitVal, nnResult);

  LogTrace("l1tOmtfEventPrint") << " " << __FUNCTION__ << ":" << __LINE__ << std::endl;

  double pt = std::copysign(nnResult.at(0), nnResult.at(1));

  LogTrace("l1tOmtfEventPrint") << " " << __FUNCTION__ << ":" << __LINE__ << " nnResult.at(0) " << nnResult.at(0)
                                << " nnResult.at(1) " << nnResult.at(1) << " pt " << pt << std::endl;

  std::vector<float> pts;
  pts.emplace_back(pt);

  //algoMuon->setPtNN(omtfConfig->ptGevToHw(nnResult.at(0)));
  auto calibratedHwPt = lutNetworkFP->getCalibratedHwPt();
  algoMuon->setPtNNConstr(calibratedHwPt);

  algoMuon->setChargeNNConstr(nnResult[1] >= 0 ? 1 : -1);

  //TODO add some if here, such that the property_tree is filled only when needed
  boost::property_tree::ptree procDataTree;
  for (unsigned int i = 0; i < inputs.size(); i++) {
    auto& inputTree = procDataTree.add("input", "");
    inputTree.add("<xmlattr>.num", i);
    inputTree.add("<xmlattr>.val", inputs[i]);
  }

  std::ostringstream ostr;
  ostr << std::fixed << std::setprecision(19) << nnResult.at(0);
  procDataTree.add("output0.<xmlattr>.val", ostr.str());

  ostr.str("");
  ostr << std::fixed << std::setprecision(19) << nnResult.at(1);
  procDataTree.add("output1.<xmlattr>.val", ostr.str());

  procDataTree.add("calibratedHwPt.<xmlattr>.val", calibratedHwPt);

  procDataTree.add("hwSign.<xmlattr>.val", algoMuon->getChargeNNConstr() < 0 ? 1 : 0);

  for (auto& obs : observers)
    obs->addProcesorData("regressionNN", procDataTree);

  return pts;

  //event.print();
  /*
  std::vector<float> pts(classifierToRegressions.size(), 0);

  unsigned int i =0;
  for(auto& classifierToRegression : classifierToRegressions) {
    auto orgValue = classifierToRegression->getValue(&event);
    auto absOrgValue = std::abs(orgValue);
    pts.at(i) = classifierToRegression->getCalibratedValue(absOrgValue);
    pts.at(i) = std::copysign(pts.at(i), orgValue);

    LogTrace("OMTFReconstruction") <<" "<<__FUNCTION__<<":"<<__LINE__<<" orgValue "<<orgValue<<" pts["<<i<<"] "<<pts[i]<<std::endl;
    //std::cout<<"nn pts["<<i<<"] "<<pts[i]<< std::endl;
    i++;
  }

  return pts;*/
}

Member Data Documentation

lutNetworkFP

unique_ptr<lutNN::LutNetworkFixedPointRegressionBase> PtAssignmentNNRegression::lutNetworkFP

private

Definition at line 25 of file PtAssignmentNNRegression.h.

Referenced by getPts(), and PtAssignmentNNRegression().