OMTFReconstruction.cc

Go to the documentation of this file.

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "DataFormats/L1TMuon/interface/RegionalMuonCand.h"
#include "DataFormats/L1TMuon/interface/RegionalMuonCandFwd.h"

#include "CondFormats/DataRecord/interface/L1TMuonOverlapParamsRcd.h"
#include "CondFormats/L1TObjects/interface/L1TMuonOverlapParams.h"

#include "L1Trigger/L1TMuonOverlap/plugins/L1TMuonOverlapTrackProducer.h"
#include "L1Trigger/L1TMuonOverlap/interface/OMTFProcessor.h"
#include "L1Trigger/L1TMuonOverlap/interface/OMTFinput.h"
#include "L1Trigger/L1TMuonOverlap/interface/OMTFReconstruction.h"
#include "L1Trigger/L1TMuonOverlap/interface/OMTFConfiguration.h"
#include "L1Trigger/L1TMuonOverlap/interface/XMLConfigWriter.h"
#include "L1Trigger/L1TMuonOverlap/interface/OmtfName.h"
#include "L1Trigger/L1TMuonOverlap/interface/GhostBusterPreferRefDt.h"

#include "L1Trigger/RPCTrigger/interface/RPCConst.h"

OMTFReconstruction::OMTFReconstruction() :
  m_OMTFConfig(nullptr), m_OMTF(nullptr), aTopElement(nullptr), m_OMTFConfigMaker(nullptr), m_Writer(nullptr){}
OMTFReconstruction::OMTFReconstruction(const edm::ParameterSet& theConfig) :
  m_Config(theConfig), m_OMTFConfig(nullptr), m_OMTF(nullptr), aTopElement(nullptr), m_OMTFConfigMaker(nullptr), m_Writer(nullptr) {

  dumpResultToXML = m_Config.getParameter<bool>("dumpResultToXML");
  dumpDetailedResultToXML = m_Config.getParameter<bool>("dumpDetailedResultToXML");
  m_Config.getParameter<std::string>("XMLDumpFileName");  
}
OMTFReconstruction::~OMTFReconstruction(){
  
  delete m_OMTFConfig;
  delete m_OMTF;  

  if (m_Writer) delete m_Writer;
}

void OMTFReconstruction::beginJob() {
  
    m_OMTFConfig = new OMTFConfiguration();
    m_OMTF = new OMTFProcessor();

}
void OMTFReconstruction::endJob(){

  if(dumpResultToXML){
    std::string fName = m_Config.getParameter<std::string>("XMLDumpFileName");
    m_Writer->finaliseXMLDocument(fName);
  } 
}
void OMTFReconstruction::beginRun(edm::Run const& run, edm::EventSetup const& iSetup) {

  const L1TMuonOverlapParamsRcd& omtfRcd = iSetup.get<L1TMuonOverlapParamsRcd>();
  
  edm::ESHandle<L1TMuonOverlapParams> omtfParamsHandle;
  omtfRcd.get(omtfParamsHandle);

  const L1TMuonOverlapParams* omtfParams = omtfParamsHandle.product();

  if (!omtfParams) {
    edm::LogError("L1TMuonOverlapTrackProducer") << "Could not retrieve parameters from Event Setup" << std::endl;
  }

  m_OMTFConfig->configure(omtfParams);
  m_OMTF->configure(m_OMTFConfig, omtfParams);
  //m_GhostBuster.setNphiBins(m_OMTFConfig->nPhiBins());

  if(m_Config.exists("ghostBusterType") ) {
    if(m_Config.getParameter<std::string>("ghostBusterType") == "GhostBusterPreferRefDt")
        m_GhostBuster.reset(new GhostBusterPreferRefDt(m_OMTFConfig) );
  }
  else
      m_GhostBuster.reset(new GhostBuster() );

  m_Sorter.setNphiBins(m_OMTFConfig->nPhiBins());

  m_InputMaker.initialize(iSetup, m_OMTFConfig);

  if(dumpResultToXML){
    m_Writer = new XMLConfigWriter(m_OMTFConfig);
    std::string fName = "OMTF";
    m_Writer->initialiseXMLDocument(fName);
  }
}
std::unique_ptr<l1t::RegionalMuonCandBxCollection> OMTFReconstruction::reconstruct(const edm::Event& iEvent, const edm::EventSetup& evSetup) {

  loadAndFilterDigis(iEvent);

  if(dumpResultToXML) aTopElement = m_Writer->writeEventHeader(iEvent.id().event());

  // NOTE: assuming all is for bx 0
  int bx = 0;
  std::unique_ptr<l1t::RegionalMuonCandBxCollection> candidates(new l1t::RegionalMuonCandBxCollection);

  for(unsigned int iProcessor=0; iProcessor<m_OMTFConfig->nProcessors(); ++iProcessor)
    getProcessorCandidates(iProcessor, l1t::tftype::omtf_pos, bx, *candidates);

  for(unsigned int iProcessor=0; iProcessor<m_OMTFConfig->nProcessors(); ++iProcessor)
    getProcessorCandidates(iProcessor, l1t::tftype::omtf_neg, bx, *candidates);
    
  return candidates;
}

void OMTFReconstruction::loadAndFilterDigis(const edm::Event& iEvent){

  // Filter digis by dropping digis from selected (by cfg.py) subsystems
  if(!m_Config.getParameter<bool>("dropDTPrimitives")){
    iEvent.getByLabel(m_Config.getParameter<edm::InputTag>("srcDTPh"),dtPhDigis);
    iEvent.getByLabel(m_Config.getParameter<edm::InputTag>("srcDTTh"),dtThDigis);
  }
  if(!m_Config.getParameter<bool>("dropRPCPrimitives")) iEvent.getByLabel(m_Config.getParameter<edm::InputTag>("srcRPC"),rpcDigis);
  if(!m_Config.getParameter<bool>("dropCSCPrimitives")) iEvent.getByLabel(m_Config.getParameter<edm::InputTag>("srcCSC"),cscDigis);
  
}

void OMTFReconstruction::getProcessorCandidates(unsigned int iProcessor, l1t::tftype mtfType, int bx,
               l1t::RegionalMuonCandBxCollection & omtfCandidates){

  
  m_InputMaker.setFlag(0);
  OMTFinput input = m_InputMaker.buildInputForProcessor(dtPhDigis.product(),
                dtThDigis.product(),
                cscDigis.product(),
                rpcDigis.product(),
                iProcessor, mtfType);
  int flag = m_InputMaker.getFlag();
  
  const std::vector<OMTFProcessor::resultsMap> & results = m_OMTF->processInput(iProcessor,input);

  std::vector<AlgoMuon> algoCandidates;

  m_Sorter.sortRefHitResults(results, algoCandidates);  

  // perform GB 
  std::vector<AlgoMuon> gbCandidates =  m_GhostBuster->select(algoCandidates);
  
  // fill RegionalMuonCand colleciton
  std::vector<l1t::RegionalMuonCand> candMuons = m_Sorter.candidates(iProcessor, mtfType, gbCandidates);

  //fill outgoing collection
  for (auto & candMuon :  candMuons) {
     candMuon.setHwQual( candMuon.hwQual() | flag);         //FIXME temporary debug fix
     omtfCandidates.push_back(bx, candMuon);
  }
  
  //dump to XML
  writeResultToXML(iProcessor, mtfType,  input, results, candMuons);
}
void OMTFReconstruction::writeResultToXML(unsigned int iProcessor,  l1t::tftype mtfType,  const OMTFinput &input, 
               const std::vector<OMTFProcessor::resultsMap> & results,
               const std::vector<l1t::RegionalMuonCand> & candMuons ){

  int endcap =  (mtfType == l1t::omtf_neg) ? -1 : ( ( mtfType == l1t::omtf_pos) ? +1 : 0 );
  OmtfName board(iProcessor, endcap); 

  //Write data to XML file
  if(dumpResultToXML){
    xercesc::DOMElement * aProcElement = m_Writer->writeEventData(aTopElement, board, input);
    for(unsigned int iRefHit=0;iRefHit<m_OMTFConfig->nTestRefHits();++iRefHit){
      AlgoMuon algoMuon = m_Sorter.sortRefHitResults(results[iRefHit],0);//charge=0 means ignore charge
      if(algoMuon.getPt()) {
        m_Writer->writeAlgoMuon(aProcElement,iRefHit,algoMuon);
        if(dumpDetailedResultToXML){
          for(auto & itKey: results[iRefHit])
            m_Writer->writeResultsData(aProcElement, iRefHit, itKey.first,itKey.second);
        }
      }
    }
    for (auto & candMuon :  candMuons) m_Writer->writeCandMuon(aProcElement, candMuon);
  }
}