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, int ptLUTVersion, bool readPtLUTFile, bool fixMode15HighPt, bool bug9BitDPhi, bool bugMode7CLCT, bool bugNegPt)
void	configure_details ()
const std::vector< int > &	getAllowedModes (void) const
const std::array< emtf::Forest, 16 > &	getForests (void) const
void	load (const L1TMuonEndCapForest *payload)
	PtAssignmentEngine ()
void	read (const std::string &xml_dir)
int	readPtLUTFile ()
virtual float	scale_pt (const float pt, const int mode=15) const
void	set_ptLUTVersion (int ptLUTVersion)
virtual float	unscale_pt (const float pt, const int mode=15) const
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 17 of file PtAssignmentEngine.h.

Member Typedef Documentation

typedef uint64_t PtAssignmentEngine::address_t

Definition at line 22 of file PtAssignmentEngine.h.

Constructor & Destructor Documentation

PtAssignmentEngine::PtAssignmentEngine ( )

explicit

Definition at line 8 of file PtAssignmentEngine.cc.

References forests_, and ptlut_reader_.

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

}

PtAssignmentEngine::~PtAssignmentEngine ( )

virtual

Definition at line 16 of file PtAssignmentEngine.cc.

                                        {

}

Member Function Documentation

const PtAssignmentEngineAux & PtAssignmentEngine::aux

(

)

const

Definition at line 107 of file PtAssignmentEngine.cc.

References instance.

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

                                                           {
  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 42 of file PtAssignmentEngine.h.

References calculate_pt(), calculate_pt_lut(), and HiIsolationCommonParameters_cff::track.

Referenced by PtAssignment::process().

{ return 0; }

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

virtual

Definition at line 120 of file PtAssignmentEngine.cc.

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

Referenced by calculate_address(), and 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 132 of file PtAssignmentEngine.cc.

References calculate_pt_xml(), and EnergyCorrector::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 140 of file PtAssignmentEngine.cc.

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

Referenced by calculate_address(), and 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 48 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 49 of file PtAssignmentEngine.h.

{ return 0.; }

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

Definition at line 80 of file PtAssignmentEngine.cc.

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

Referenced by PtAssignment::configure(), getAllowedModes(), and L1TMuonEndCapForestESProducer::produce().

  {
  verbose_ = verbose;

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

  configure_details();
}

void PtAssignmentEngine::configure_details

(

)

Definition at line 97 of file PtAssignmentEngine.cc.

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

Referenced by configure(), and getAllowedModes().

                                           {
  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);
  }
}

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

inline

Definition at line 27 of file PtAssignmentEngine.h.

References allowedModes_, aux(), configure(), configure_details(), ALCARECOPromptCalibProdSiPixelAli0T_cff::mode, EnergyCorrector::pt, readPtLUTFile(), scale_pt(), and unscale_pt().

Referenced by L1TMuonEndCapForestESProducer::produce().

{ return allowedModes_; }

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

inline

Definition at line 26 of file PtAssignmentEngine.h.

References forests_.

Referenced by L1TMuonEndCapForestESProducer::produce().

{ return forests_; }

void PtAssignmentEngine::load

(

const L1TMuonEndCapForest *

payload

)

Definition at line 44 of file PtAssignmentEngine.cc.

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

Referenced by TrackFinder::process().

                                                                {
  // unsigned pt_lut_version = payload->version_;  // Why is payload->version_ always 0? - AWB 02.06.17
  // std::cout << "ptLUTVersion_ from configuration = " << ptLUTVersion_ << ", payload->version_ = " << payload->version_ << std::endl;
  // if (not(pt_lut_version == unsigned(ptLUTVersion_)))
  //   { edm::LogError("L1T") << "pt_lut_version = " << pt_lut_version << ", ptLUTVersion_ = " << ptLUTVersion_; return; }
    
  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_ );

    if (not(boostWeight_ == 0 || ptLUTVersion_ >= 6))  // Check that XMLs and pT LUT version are consistent
      { edm::LogError("L1T") << "boostWeight_ = " << boostWeight_ << ", ptLUTVersion_ = " << ptLUTVersion_; return; }
    // 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

(

const std::string &

xml_dir

)

Definition at line 23 of file PtAssignmentEngine.cc.

References allowedModes_, gather_cfg::cout, forests_, mps_fire::i, ALCARECOPromptCalibProdSiPixelAli0T_cff::mode, ptLUTVersion_, 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 (ptLUTVersion_ >= 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;
}

int PtAssignmentEngine::readPtLUTFile ( )

inline

Definition at line 51 of file PtAssignmentEngine.h.

References readPtLUTFile_.

Referenced by configure(), and getAllowedModes().

{ return readPtLUTFile_; }

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

virtual

Reimplemented in PtAssignmentEngine2016, and PtAssignmentEngine2017.

Definition at line 112 of file PtAssignmentEngine.cc.

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

                                                                       {
  return scale_pt(pt, mode);
}

void PtAssignmentEngine::set_ptLUTVersion ( int  ptLUTVersion )

inline

Definition at line 52 of file PtAssignmentEngine.h.

References ptLUTVersion_.

Referenced by TrackFinder::process().

{ ptLUTVersion_ = ptLUTVersion; }

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

virtual

Reimplemented in PtAssignmentEngine2016, and PtAssignmentEngine2017.

Definition at line 116 of file PtAssignmentEngine.cc.

Referenced by calculate_pt_lut(), and getAllowedModes().

                                                                         {
  return unscale_pt(pt, mode);
}

Member Data Documentation

std::vector<int> PtAssignmentEngine::allowedModes_

protected

Definition at line 55 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 63 of file PtAssignmentEngine.h.

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

bool PtAssignmentEngine::bugMode7CLCT_

protected

Definition at line 63 of file PtAssignmentEngine.h.

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

bool PtAssignmentEngine::bugNegPt_

protected

Definition at line 63 of file PtAssignmentEngine.h.

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

bool PtAssignmentEngine::fixMode15HighPt_

protected

Definition at line 62 of file PtAssignmentEngine.h.

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

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

protected

Definition at line 56 of file PtAssignmentEngine.h.

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

PtLUTReader PtAssignmentEngine::ptlut_reader_

protected

Definition at line 57 of file PtAssignmentEngine.h.

Referenced by calculate_pt_lut(), configure_details(), and PtAssignmentEngine().

int PtAssignmentEngine::ptLUTVersion_

protected

Definition at line 61 of file PtAssignmentEngine.h.

Referenced by configure(), load(), read(), PtAssignmentEngine2017::scale_pt(), set_ptLUTVersion(), and PtAssignmentEngine2017::unscale_pt().

bool PtAssignmentEngine::readPtLUTFile_

protected

Definition at line 62 of file PtAssignmentEngine.h.

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

int PtAssignmentEngine::verbose_

protected

Definition at line 59 of file PtAssignmentEngine.h.

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