DataROOTDumper2 Class Reference

#include <DataROOTDumper2.h>

Inheritance diagram for DataROOTDumper2:

Public Member Functions
	DataROOTDumper2 (const edm::ParameterSet &edmCfg, const OMTFConfiguration *omtfConfig, std::string rootFileName)
void	endJob () override
void	observeEventEnd (const edm::Event &iEvent, std::unique_ptr< l1t::RegionalMuonCandBxCollection > &finalCandidates) override
	~DataROOTDumper2 () override
Public Member Functions inherited from EmulationObserverBase
	EmulationObserverBase (const edm::ParameterSet &edmCfg, const OMTFConfiguration *omtfConfig)
const SimTrack *	findSimMuon (const edm::Event &event, const SimTrack *previous=nullptr)
void	observeEventBegin (const edm::Event &iEvent) override
void	observeProcesorEmulation (unsigned int iProcessor, l1t::tftype mtfType, const std::shared_ptr< OMTFinput > &input, const AlgoMuons &algoCandidates, const AlgoMuons &gbCandidates, const std::vector< l1t::RegionalMuonCand > &candMuons) override
	~EmulationObserverBase () override
Public Member Functions inherited from IOMTFEmulationObserver
virtual void	beginRun (edm::EventSetup const &eventSetup)
	IOMTFEmulationObserver ()
virtual	~IOMTFEmulationObserver ()

Private Member Functions
void	initializeTTree (std::string rootFileName)
void	saveTTree ()

Private Attributes
unsigned int	evntCnt = 0
std::vector< TH2 * >	hitVsPt
OmtfEvent	omtfEvent
TH1I *	ptGenNeg = nullptr
TH1I *	ptGenPos = nullptr
TFile *	rootFile = nullptr
TTree *	rootTree = nullptr

Additional Inherited Members
Protected Attributes inherited from EmulationObserverBase
unsigned int	candProcIndx = 0
edm::ParameterSet	edmCfg
AlgoMuons::value_type	omtfCand
const OMTFConfiguration *	omtfConfig
l1t::RegionalMuonCand	regionalMuonCand
const SimTrack *	simMuon = nullptr

Detailed Description

Definition at line 59 of file DataROOTDumper2.h.

Constructor & Destructor Documentation

DataROOTDumper2::DataROOTDumper2	(	const edm::ParameterSet &	edmCfg,
		const OMTFConfiguration *	omtfConfig,
		std::string	rootFileName
	)

Definition at line 28 of file DataROOTDumper2.cc.

References initializeTTree(), and OMTFConfiguration::nTestRefHits().

    : EmulationObserverBase(edmCfg, omtfConfig) {
  edm::LogVerbatim("l1tOmtfEventPrint") << " omtfConfig->nTestRefHits() " << omtfConfig->nTestRefHits()
                                        << " event.omtfGpResultsPdfSum.num_elements() " << endl;
  initializeTTree(rootFileName);

  edm::LogVerbatim("l1tOmtfEventPrint") << " DataROOTDumper2 created" << std::endl;
}

DataROOTDumper2::~DataROOTDumper2 ( )

override

Definition at line 39 of file DataROOTDumper2.cc.

References saveTTree().

{ saveTTree(); }

Member Function Documentation

void DataROOTDumper2::endJob ( void  )

overridevirtual

Implements IOMTFEmulationObserver.

Definition at line 178 of file DataROOTDumper2.cc.

References evntCnt.

{ edm::LogVerbatim("l1tOmtfEventPrint") << " evntCnt " << evntCnt << endl; }

void DataROOTDumper2::initializeTTree ( std::string  rootFileName )

private

Definition at line 41 of file DataROOTDumper2.cc.

References OmtfEvent::muonCharge, OmtfEvent::muonEta, OmtfEvent::muonPhi, OmtfEvent::muonPt, OmtfEvent::omtfCharge, OmtfEvent::omtfEta, omtfEvent, OmtfEvent::omtfFiredLayers, OmtfEvent::omtfPhi, OmtfEvent::omtfProcessor, OmtfEvent::omtfPt, OmtfEvent::omtfQuality, OmtfEvent::omtfRefLayer, OmtfEvent::omtfScore, ptGenNeg, ptGenPos, rootFile, and rootTree.

Referenced by DataROOTDumper2().

                                                            {
  rootFile = new TFile(rootFileName.c_str(), "RECREATE");
  rootTree = new TTree("OMTFHitsTree", "");

  rootTree->Branch("muonPt", &omtfEvent.muonPt);
  rootTree->Branch("muonEta", &omtfEvent.muonEta);
  rootTree->Branch("muonPhi", &omtfEvent.muonPhi);
  rootTree->Branch("muonCharge", &omtfEvent.muonCharge);

  rootTree->Branch("omtfPt", &omtfEvent.omtfPt);
  rootTree->Branch("omtfEta", &omtfEvent.omtfEta);
  rootTree->Branch("omtfPhi", &omtfEvent.omtfPhi);
  rootTree->Branch("omtfCharge", &omtfEvent.omtfCharge);

  rootTree->Branch("omtfScore", &omtfEvent.omtfScore);
  rootTree->Branch("omtfQuality", &omtfEvent.omtfQuality);
  rootTree->Branch("omtfRefLayer", &omtfEvent.omtfRefLayer);
  rootTree->Branch("omtfProcessor", &omtfEvent.omtfProcessor);

  rootTree->Branch("omtfFiredLayers", &omtfEvent.omtfFiredLayers);  //<<<<<<<<<<<<<<<<<<<<<<!!!!TODOO

  ptGenPos = new TH1I("ptGenPos", "ptGenPos", 400, 0, 200);
  ptGenNeg = new TH1I("ptGenNeg", "ptGenNeg", 400, 0, 200);
}

void DataROOTDumper2::observeEventEnd ( const edm::Event &  iEvent, std::unique_ptr< l1t::RegionalMuonCandBxCollection > &  finalCandidates  )

overridevirtual

Reimplemented from IOMTFEmulationObserver.

Definition at line 75 of file DataROOTDumper2.cc.

References funct::abs(), EmulationObserverBase::candProcIndx, CSCDetId::chamber(), MuonSubdetId::CSC, OmtfEvent::Hit::eta, evntCnt, OMTFConfiguration::getRefToLogicNumber(), OmtfEvent::hits, OMTFConfiguration::hwEtaToEta(), OMTFConfiguration::hwPtToGev(), l1t::RegionalMuonCand::hwQual(), l1t::RegionalMuonCand::hwSign(), OMTFConfiguration::isBendingLayer(), OmtfEvent::Hit::layer, CoreSimTrack::momentum(), OmtfEvent::muonCharge, OmtfEvent::muonEta, OmtfEvent::muonPhi, OmtfEvent::muonPt, EmulationObserverBase::omtfCand, OmtfEvent::omtfCharge, EmulationObserverBase::omtfConfig, OmtfEvent::omtfEta, omtfEvent, OmtfEvent::omtfFiredLayers, OmtfEvent::omtfPhi, OmtfEvent::omtfProcessor, OmtfEvent::omtfPt, OmtfEvent::omtfQuality, OmtfEvent::omtfRefLayer, OmtfEvent::omtfScore, OmtfEvent::Hit::phiDist, funct::pow(), ptGenNeg, ptGenPos, OmtfEvent::Hit::quality, OmtfEvent::Hit::rawData, EmulationObserverBase::regionalMuonCand, rootTree, EmulationObserverBase::simMuon, CoreSimTrack::type(), OmtfEvent::Hit::valid, and OmtfEvent::Hit::z.

                                                                                                       {
  int muonCharge = 0;
  if (simMuon) {
    if (abs(simMuon->momentum().eta()) < 0.8 || abs(simMuon->momentum().eta()) > 1.24)
      return;

    muonCharge = (abs(simMuon->type()) == 13) ? simMuon->type() / -13 : 0;
    if (muonCharge > 0)
      ptGenPos->Fill(simMuon->momentum().pt());
    else
      ptGenNeg->Fill(simMuon->momentum().pt());
  }

  if (simMuon == nullptr || !omtfCand->isValid())  //no sim muon or empty candidate
    return;

  omtfEvent.muonPt = simMuon->momentum().pt();
  omtfEvent.muonEta = simMuon->momentum().eta();

  //TODO add cut on ete if needed
  /*  if(abs(event.muonEta) < 0.8 || abs(event.muonEta) > 1.24)
    return;*/

  omtfEvent.muonPhi = simMuon->momentum().phi();
  omtfEvent.muonCharge = muonCharge;  //TODO

  if (omtfCand->getPt() > 0) {  //&& omtfCand->getFiredLayerCnt() > 3 TODO add to the if needed
    omtfEvent.omtfPt = omtfConfig->hwPtToGev(omtfCand->getPt());
    omtfEvent.omtfEta = omtfConfig->hwEtaToEta(omtfCand->getEtaHw());
    omtfEvent.omtfPhi = omtfCand->getPhi();
    omtfEvent.omtfCharge = std::pow(-1, regionalMuonCand.hwSign());
    omtfEvent.omtfScore = omtfCand->getPdfSum();
    omtfEvent.omtfQuality = regionalMuonCand.hwQual();  //omtfCand->getQ();
    omtfEvent.omtfFiredLayers = omtfCand->getFiredLayerBits();
    omtfEvent.omtfRefLayer = omtfCand->getRefLayer();
    omtfEvent.omtfProcessor = candProcIndx;

    omtfEvent.hits.clear();

    auto& gpResult = omtfCand->getGpResult();

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

    for (unsigned int iLogicLayer = 0; iLogicLayer < gpResult.getStubResults().size(); ++iLogicLayer) {
      auto& stubResult = gpResult.getStubResults()[iLogicLayer];
      if (stubResult.getMuonStub()) {  //&& stubResult.getValid() //TODO!!!!!!!!!!!!!!!!1
        OmtfEvent::Hit hit;
        hit.layer = iLogicLayer;
        hit.quality = stubResult.getMuonStub()->qualityHw;
        hit.eta = stubResult.getMuonStub()->etaHw;  //in which scale?
        hit.valid = stubResult.getValid();

        int hitPhi = stubResult.getMuonStub()->phiHw;
        unsigned int refLayerLogicNum = omtfConfig->getRefToLogicNumber()[omtfCand->getRefLayer()];
        int phiRefHit = gpResult.getStubResults()[refLayerLogicNum].getMuonStub()->phiHw;

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

        //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 " << omtfEvent.muonPt << " omtfPt " << omtfEvent.omtfPt << " RefLayer "
              << omtfEvent.omtfRefLayer << " layer " << int(hit.layer) << " hit.phiDist " << hit.phiDist << " valid "
              << stubResult.getValid() << " !!!!!!!!!!!!!!!!!!!!!!!!" << endl;
        }

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

        omtfEvent.hits.push_back(hit.rawData);
      }
    }

    //debug
    /*if( (int)event.hits.size() != omtfCand->getQ()) {
      LogTrace("l1tOmtfEventPrint")<<" muonPt "<<event.muonPt<<" omtfPt "<<event.omtfPt<<" RefLayer "<<event.omtfRefLayer
                    <<" hits.size "<<event.hits.size()<<" omtfCand->getQ "<<omtfCand->getQ()<<" !!!!!!!!!!!!!!!!!!aaa!!!!!!"<<endl;
    }*/

    rootTree->Fill();
    evntCnt++;
  }
}

void DataROOTDumper2::saveTTree ( )

private

Definition at line 66 of file DataROOTDumper2.cc.

References ptGenNeg, ptGenPos, rootFile, and rootTree.

Referenced by ~DataROOTDumper2().

                                {
  rootFile->Write();
  ptGenPos->Write();
  ptGenNeg->Write();

  delete rootFile;
  delete rootTree;
}

Member Data Documentation

unsigned int DataROOTDumper2::evntCnt = 0

private

Definition at line 79 of file DataROOTDumper2.h.

Referenced by endJob(), and observeEventEnd().

std::vector<TH2*> DataROOTDumper2::hitVsPt

private

Definition at line 84 of file DataROOTDumper2.h.

OmtfEvent DataROOTDumper2::omtfEvent

private

Definition at line 77 of file DataROOTDumper2.h.

Referenced by initializeTTree(), and observeEventEnd().

TH1I* DataROOTDumper2::ptGenNeg = nullptr

private

Definition at line 82 of file DataROOTDumper2.h.

Referenced by initializeTTree(), observeEventEnd(), and saveTTree().

TH1I* DataROOTDumper2::ptGenPos = nullptr

private

Definition at line 81 of file DataROOTDumper2.h.

Referenced by initializeTTree(), observeEventEnd(), and saveTTree().

TFile* DataROOTDumper2::rootFile = nullptr

private

Definition at line 74 of file DataROOTDumper2.h.

Referenced by initializeTTree(), and saveTTree().

TTree* DataROOTDumper2::rootTree = nullptr

private

Definition at line 75 of file DataROOTDumper2.h.

Referenced by initializeTTree(), observeEventEnd(), and saveTTree().