PtAssignmentEngine Class Reference

#include <PtAssignmentEngine.h>

Inheritance diagram for PtAssignmentEngine:

Public Types
typedef uint64_t	address_t

Public Member Functions
const PtAssignmentEngineAux &	aux () const
virtual address_t	calculate_address (const EMTFTrack &track) const
virtual float	calculate_pt (const address_t &address) const
virtual float	calculate_pt (const EMTFTrack &track) const
virtual float	calculate_pt_lut (const address_t &address) const
virtual float	calculate_pt_xml (const address_t &address) const
virtual float	calculate_pt_xml (const EMTFTrack &track) const
void	configure (int verbose, bool readPtLUTFile, bool fixMode15HighPt, bool bug9BitDPhi, bool bugMode7CLCT, bool bugNegPt)
void	configure_details ()
int	get_pt_lut_version () const
const std::vector< int > &	getAllowedModes (void) const
const std::array< emtf::Forest, 16 > &	getForests (void) const
void	load (int pt_lut_version, const L1TMuonEndCapForest *payload)
	PtAssignmentEngine ()
void	read (int pt_lut_version, const std::string &xml_dir)
virtual float	scale_pt (const float pt, const int mode=15) const =0
virtual float	unscale_pt (const float pt, const int mode=15) const =0
virtual	~PtAssignmentEngine ()

Protected Attributes
std::vector< int >	allowedModes_
bool	bug9BitDPhi_
bool	bugMode7CLCT_
bool	bugNegPt_
bool	fixMode15HighPt_
std::array< emtf::Forest, 16 >	forests_
PtLUTReader	ptlut_reader_
int	ptLUTVersion_
bool	readPtLUTFile_
int	verbose_

Detailed Description

Definition at line 16 of file PtAssignmentEngine.h.

Member Typedef Documentation

typedef uint64_t PtAssignmentEngine::address_t

Definition at line 21 of file PtAssignmentEngine.h.

Constructor & Destructor Documentation

PtAssignmentEngine::PtAssignmentEngine ( )

explicit

Definition at line 6 of file PtAssignmentEngine.cc.

    : allowedModes_({3, 5, 9, 6, 10, 12, 7, 11, 13, 14, 15}), forests_(), ptlut_reader_(), ptLUTVersion_(0xFFFFFFFF) {}

PtAssignmentEngine::~PtAssignmentEngine ( )

virtual

Definition at line 9 of file PtAssignmentEngine.cc.

{}

Member Function Documentation

const PtAssignmentEngineAux & PtAssignmentEngine::aux

(

)

const

Definition at line 95 of file PtAssignmentEngine.cc.

References instance.

Referenced by PtAssignment::aux(), and calculate_pt_lut().

                                                           {
  static const PtAssignmentEngineAux instance;
  return instance;
}

virtual address_t PtAssignmentEngine::calculate_address ( const EMTFTrack &  track ) const

inlinevirtual

Reimplemented in PtAssignmentEngine2016, and PtAssignmentEngine2017.

Definition at line 40 of file PtAssignmentEngine.h.

Referenced by PtAssignment::process().

{ return 0; }

float PtAssignmentEngine::calculate_pt ( const address_t &  address ) const

virtual

Definition at line 100 of file PtAssignmentEngine.cc.

References calculate_pt_lut(), calculate_pt_xml(), DiDispStaMuonMonitor_cfi::pt, and readPtLUTFile_.

Referenced by PtAssignment::process().

                                                                     {
  float pt = 0.;

  if (readPtLUTFile_) {
    pt = calculate_pt_lut(address);
  } else {
    pt = calculate_pt_xml(address);
  }

  return pt;
}

float PtAssignmentEngine::calculate_pt ( const EMTFTrack &  track ) const

virtual

Definition at line 112 of file PtAssignmentEngine.cc.

References calculate_pt_xml(), and DiDispStaMuonMonitor_cfi::pt.

                                                                   {
  float pt = 0.;

  pt = calculate_pt_xml(track);

  return pt;
}

float PtAssignmentEngine::calculate_pt_lut ( const address_t &  address ) const

virtual

Definition at line 120 of file PtAssignmentEngine.cc.

References aux(), PtAssignmentEngineAux::getPtFromGMTPt(), PtLUTReader::lookup(), DiDispStaMuonMonitor_cfi::pt, ptlut_reader_, and unscale_pt().

Referenced by calculate_pt().

                                                                         {
  // LUT outputs 'gmt_pt', so need to convert back to 'xmlpt'
  int gmt_pt = ptlut_reader_.lookup(address);
  float pt = aux().getPtFromGMTPt(gmt_pt);
  float xmlpt = pt;
  xmlpt *= unscale_pt(pt);

  return xmlpt;
}

virtual float PtAssignmentEngine::calculate_pt_xml ( const address_t &  address ) const

inlinevirtual

Reimplemented in PtAssignmentEngine2016, and PtAssignmentEngine2017.

Definition at line 46 of file PtAssignmentEngine.h.

Referenced by calculate_pt().

{ return 0.; }

virtual float PtAssignmentEngine::calculate_pt_xml ( const EMTFTrack &  track ) const

inlinevirtual

Reimplemented in PtAssignmentEngine2016, and PtAssignmentEngine2017.

Definition at line 47 of file PtAssignmentEngine.h.

{ return 0.; }

void PtAssignmentEngine::configure	(	int	verbose,
		bool	readPtLUTFile,
		bool	fixMode15HighPt,
		bool	bug9BitDPhi,
		bool	bugMode7CLCT,
		bool	bugNegPt
	)

Definition at line 72 of file PtAssignmentEngine.cc.

References bug9BitDPhi_, bugMode7CLCT_, bugNegPt_, configure_details(), fixMode15HighPt_, readPtLUTFile_, verbose, and verbose_.

Referenced by PtAssignment::configure().

                                                                                                               {
  verbose_ = verbose;

  readPtLUTFile_ = readPtLUTFile;
  fixMode15HighPt_ = fixMode15HighPt;
  bug9BitDPhi_ = bug9BitDPhi;
  bugMode7CLCT_ = bugMode7CLCT;
  bugNegPt_ = bugNegPt;

  configure_details();
}

void PtAssignmentEngine::configure_details

(

)

Definition at line 85 of file PtAssignmentEngine.cc.

References ptlut_reader_, PtLUTReader::read(), readPtLUTFile_, contentValuesCheck::ss, and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by configure().

                                           {
  if (readPtLUTFile_) {
    std::stringstream ss;
    // Hardcoded - this 2 GB LUT file does not exist in CMSSW
    ss << "/afs/cern.ch/work/a/abrinke1/public/EMTF/PtAssign2017/LUTs/2017_06_07/LUT_v07_07June17.dat";
    std::string lut_full_path = ss.str();
    ptlut_reader_.read(lut_full_path);
  }
}

int PtAssignmentEngine::get_pt_lut_version ( ) const

inline

Definition at line 28 of file PtAssignmentEngine.h.

References ptLUTVersion_.

Referenced by TrackFinder::process(), and PtAssignment::process().

{ return ptLUTVersion_; }

const std::vector<int>& PtAssignmentEngine::getAllowedModes ( void  ) const

inline

Definition at line 26 of file PtAssignmentEngine.h.

References allowedModes_.

Referenced by L1TMuonEndCapForestESProducer::produce().

{ return allowedModes_; }

const std::array<emtf::Forest, 16>& PtAssignmentEngine::getForests ( void  ) const

inline

Definition at line 25 of file PtAssignmentEngine.h.

References forests_.

Referenced by L1TMuonEndCapForestESProducer::produce().

{ return forests_; }

void PtAssignmentEngine::load	(	int	pt_lut_version,
		const L1TMuonEndCapForest *	payload
	)

Definition at line 34 of file PtAssignmentEngine.cc.

References allowedModes_, emtf_assert, L1TMuonEndCapForest::forest_coll_, L1TMuonEndCapForest::forest_map_, forests_, mps_fire::i, universalConfigTemplate::mode, and ptLUTVersion_.

                                                                                    {
  if (ptLUTVersion_ == pt_lut_version)
    return;
  ptLUTVersion_ = pt_lut_version;

  edm::LogInfo("L1T") << "EMTF using pt_lut_ver: " << pt_lut_version;

  for (unsigned i = 0; i < allowedModes_.size(); ++i) {
    int mode = allowedModes_.at(i);

    L1TMuonEndCapForest::DForestMap::const_iterator index =
        payload->forest_map_.find(mode);  // associates mode to index
    if (index == payload->forest_map_.end())
      continue;

    forests_.at(mode).loadFromCondPayload(payload->forest_coll_[index->second]);

    double boostWeight_ = payload->forest_map_.find(mode + 16)->second / 1000000.;
    // std::cout << "Loaded forest for mode " << mode << " with boostWeight_ = " << boostWeight_ << std::endl;
    // std::cout << "  * ptLUTVersion_ = " << ptLUTVersion_ << std::endl;
    forests_.at(mode).getTree(0)->setBoostWeight(boostWeight_);

    emtf_assert(boostWeight_ == 0 || ptLUTVersion_ >= 6);  // Check that XMLs and pT LUT version are consistent

    // Will catch user trying to run with Global Tag settings on 2017 data, rather than fakeEmtfParams. - AWB 08.06.17

    // // Code below can be used to save out trees in XML format
    // for (int t = 0; t < 64; t++) {
    //   emtf::Tree* tree = forests_.at(mode).getTree(t);
    //   std::stringstream ss;
    //   ss << mode << "/" << t << ".xml";
    //   tree->saveToXML( ss.str().c_str() );
    // }
  }

  return;
}

void PtAssignmentEngine::read	(	int	pt_lut_version,
		const std::string &	xml_dir
	)

Definition at line 14 of file PtAssignmentEngine.cc.

References allowedModes_, gather_cfg::cout, forests_, mps_fire::i, universalConfigTemplate::mode, contentValuesCheck::ss, and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by edmIntegrityCheck.PublishToFileSystem::get(), and L1TMuonEndCapForestESProducer::produce().

                                                                          {
  std::string xml_dir_full = "L1Trigger/L1TMuonEndCap/data/pt_xmls/" + xml_dir;
  unsigned xml_nTrees = 64;  // 2016 XMLs
  if (pt_lut_version >= 6)
    xml_nTrees = 400;  // First 2017 XMLs

  std::cout << "EMTF emulator: attempting to read " << xml_nTrees << " pT LUT XMLs from local directory" << std::endl;
  std::cout << xml_dir_full << std::endl;
  std::cout << "Non-standard operation; if it fails, now you know why" << std::endl;

  for (unsigned i = 0; i < allowedModes_.size(); ++i) {
    int mode = allowedModes_.at(i);  // For 2016, maps to "mode_inv"
    std::stringstream ss;
    ss << xml_dir_full << "/" << mode;
    forests_.at(mode).loadForestFromXML(ss.str().c_str(), xml_nTrees);
  }

  return;
}

virtual float PtAssignmentEngine::scale_pt ( const float  pt, const int  mode = 15  ) const

pure virtual

Implemented in PtAssignmentEngine2016, and PtAssignmentEngine2017.

Referenced by PtAssignment::process().

virtual float PtAssignmentEngine::unscale_pt ( const float  pt, const int  mode = 15  ) const

pure virtual

Implemented in PtAssignmentEngine2016, and PtAssignmentEngine2017.

Referenced by calculate_pt_lut().

Member Data Documentation

std::vector<int> PtAssignmentEngine::allowedModes_

protected

Definition at line 50 of file PtAssignmentEngine.h.

Referenced by PtAssignmentEngine2016::calculate_pt_xml(), PtAssignmentEngine2017::calculate_pt_xml(), getAllowedModes(), load(), and read().

bool PtAssignmentEngine::bug9BitDPhi_

protected

Definition at line 58 of file PtAssignmentEngine.h.

Referenced by PtAssignmentEngine2016::calculate_address(), and configure().

bool PtAssignmentEngine::bugMode7CLCT_

protected

Definition at line 58 of file PtAssignmentEngine.h.

Referenced by PtAssignmentEngine2016::calculate_pt_xml(), and configure().

bool PtAssignmentEngine::bugNegPt_

protected

Definition at line 58 of file PtAssignmentEngine.h.

Referenced by PtAssignmentEngine2016::calculate_pt_xml(), and configure().

bool PtAssignmentEngine::fixMode15HighPt_

protected

Definition at line 57 of file PtAssignmentEngine.h.

Referenced by PtAssignmentEngine2016::calculate_pt_xml(), and configure().

std::array<emtf::Forest, 16> PtAssignmentEngine::forests_

protected

Definition at line 51 of file PtAssignmentEngine.h.

Referenced by PtAssignmentEngine2016::calculate_pt_xml(), PtAssignmentEngine2017::calculate_pt_xml(), getForests(), load(), and read().

PtLUTReader PtAssignmentEngine::ptlut_reader_

protected

Definition at line 52 of file PtAssignmentEngine.h.

Referenced by calculate_pt_lut(), and configure_details().

int PtAssignmentEngine::ptLUTVersion_

protected

Definition at line 56 of file PtAssignmentEngine.h.

Referenced by get_pt_lut_version(), load(), PtAssignmentEngine2017::scale_pt(), and PtAssignmentEngine2017::unscale_pt().

bool PtAssignmentEngine::readPtLUTFile_

protected

Definition at line 57 of file PtAssignmentEngine.h.

Referenced by calculate_pt(), configure(), and configure_details().

int PtAssignmentEngine::verbose_

protected

Definition at line 54 of file PtAssignmentEngine.h.

Referenced by PtAssignmentEngine2016::calculate_pt_xml(), and configure().