GlobalSim::Egamma1BDTAlgTool Class Reference

#include <Egamma1BDTAlgTool.h>

Inheritance diagram for GlobalSim::Egamma1BDTAlgTool:

Collaboration diagram for GlobalSim::Egamma1BDTAlgTool:

Public Member Functions
	Egamma1BDTAlgTool (const std::string &type, const std::string &name, const IInterface *parent)
virtual	~Egamma1BDTAlgTool ()=default
StatusCode	initialize () override
virtual StatusCode	run (const std::unique_ptr< IDataCollector > &, const EventContext &ctx) const override
virtual std::string	toString () const override

Static Public Attributes
static constexpr std::size_t	BDT_ouput_width = 10

Private Types
using	eEmEg1BDTTOBContainer = GlobalSim::IOBitwise::eEmEg1BDTTOBContainer
using	eEmNbhoodTOBContainer = GlobalSim::IOBitwise::eEmNbhoodTOBContainer

Private Member Functions
int	bitSetToInt (std::bitset< BDT_ouput_width > bitSet) const
std::vector< double >	combine_phi (const IOBitwise::eEmNbhoodTOB *) const

Private Attributes
Gaudi::Property< bool >	m_enableDump
SG::WriteHandleKey< std::vector< float > >	m_BDTScoreKey
SG::ReadHandleKey< eEmNbhoodTOBContainer >	m_nbhdTOBContainerReadKey
SG::WriteHandleKey< eEmEg1BDTTOBContainer >	m_eEmEg1BDTTOBContainerKey

Static Private Attributes
static constexpr int	s_required_phi_len = 17
static constexpr int	s_combination_len = 18

Detailed Description

Definition at line 32 of file Egamma1BDTAlgTool.h.

Member Typedef Documentation

eEmEg1BDTTOBContainer

using GlobalSim::Egamma1BDTAlgTool::eEmEg1BDTTOBContainer = GlobalSim::IOBitwise::eEmEg1BDTTOBContainer

private

Definition at line 34 of file Egamma1BDTAlgTool.h.

eEmNbhoodTOBContainer

using GlobalSim::Egamma1BDTAlgTool::eEmNbhoodTOBContainer = GlobalSim::IOBitwise::eEmNbhoodTOBContainer

private

Definition at line 35 of file Egamma1BDTAlgTool.h.

Constructor & Destructor Documentation

Egamma1BDTAlgTool()

GlobalSim::Egamma1BDTAlgTool::Egamma1BDTAlgTool	(	const std::string &	type,
		const std::string &	name,
		const IInterface *	parent )

Definition at line 19 of file Egamma1BDTAlgTool.cxx.

                                                 :
    base_class(type, name, parent){
  }

~Egamma1BDTAlgTool()

virtual GlobalSim::Egamma1BDTAlgTool::~Egamma1BDTAlgTool ( )

virtualdefault

Member Function Documentation

bitSetToInt()

int GlobalSim::Egamma1BDTAlgTool::bitSetToInt ( std::bitset< BDT_ouput_width > bitSet ) const

private

Definition at line 128 of file Egamma1BDTAlgTool.cxx.

                                                                            {
    if (!bitSet[BDT_ouput_width - 1]) return bitSet.to_ulong();
    bitSet.flip();
    return -(bitSet.to_ulong() + 1);
  }

combine_phi()

std::vector< double > GlobalSim::Egamma1BDTAlgTool::combine_phi ( const IOBitwise::eEmNbhoodTOB * nbhdTOB ) const

private

Definition at line 135 of file Egamma1BDTAlgTool.cxx.

                                                                            {
    auto result = std::vector<double>();
 
    const auto& phi_low = nbhdTOB->Neighbourhood().phi_low();
    if (phi_low.size() != s_required_phi_len) {return result;}
 
    const auto& phi_center = nbhdTOB->Neighbourhood().phi_center();
    if (phi_center.size() != s_required_phi_len) {return result;}
 
    
    const auto& phi_high = nbhdTOB->Neighbourhood().phi_high();
    if (phi_high.size() != s_required_phi_len) {return result;}
 
    result.resize(s_combination_len);
 
    constexpr int c{8};
 
    result.at(0) = phi_center.at(c).m_e;
 
    result.at(1) = std::max(phi_low.at(c).m_e, phi_high.at(c).m_e);
 
    int ri{2};
    for (int diff = 1; diff != 9; ++diff) { 
      result.at(ri) =
    std::max({phi_center.at(c-diff).m_e,
        phi_center.at(c+diff).m_e});
 
      result.at(ri+1) =
    std::max({phi_low.at(c-diff).m_e,
      phi_low.at(c+diff).m_e,
      phi_high.at(c-diff).m_e,
      phi_high.at(c+diff).m_e});
      
      ri += 2;
    }
    
    return result;
  }

initialize()

StatusCode GlobalSim::Egamma1BDTAlgTool::initialize ( )

override

Definition at line 25 of file Egamma1BDTAlgTool.cxx.

                                           {
 
    //Input keys
    CHECK(m_nbhdTOBContainerReadKey.initialize());
    CHECK(m_BDTScoreKey.initialize());
    //Output keys
    CHECK(m_eEmEg1BDTTOBContainerKey.initialize());
    
    return StatusCode::SUCCESS;
  }

run()

StatusCode GlobalSim::Egamma1BDTAlgTool::run ( const std::unique_ptr< IDataCollector > & dc, const EventContext & ctx ) const

overridevirtual

Definition at line 37 of file Egamma1BDTAlgTool.cxx.

                                            {
    ATH_MSG_DEBUG("run()");
    if (dc){dc->collect(*this, "start");}
 
  
    // read in LArStrip neighborhood TOBs from the event store
    auto in =
      SG::ReadHandle<IOBitwise::eEmNbhoodTOBContainer>(m_nbhdTOBContainerReadKey,
                               ctx);
    CHECK(in.isValid());
 
    ATH_MSG_DEBUG("read in " << (*in).size() << " neighborhoods");
 
    SG::WriteHandle<std::vector<float> > h_BDTScore(m_BDTScoreKey, ctx);
    CHECK(h_BDTScore.record(std::make_unique<std::vector<float> >()));
 
    //Setup a container of TOBs with associated GlobalLArCell windows
    auto eEmEg1BDTTOBs = std::make_unique<IOBitwise::eEmEg1BDTTOBContainer>();
    
    for (const auto nbhdTOB : *in) {
      auto c_phi = combine_phi(nbhdTOB);
      if (c_phi.empty()) {continue;}  // corner case: not all phi have len 17
      auto input = digitizer::digitize10(c_phi);
 
      assert(input.size() == n_features);
      ap_int<BDT_ouput_width>* c_input = &input[0];  // vector->array
 
      score_t scores[GlobalSim::BDT::fn_classes(n_classes)];
 
      // the bdt variable is already set up 
      bdt.decision_function(c_input, scores);
      if (msgLevel() <= MSG::DEBUG) {
    std::stringstream ss;
    ss << "BDT input: ";
    for (const auto& i : input) {ss << i << ' ';}
    ATH_MSG_DEBUG(ss.str());
      }
 
      if (msgLevel() <= MSG::DEBUG) {
    std::stringstream ss;
    ss << "C BDT output: ";
    for (const auto& i : scores) {ss << i << ' ';}
    ATH_MSG_DEBUG(ss.str());
      }
 
      //Extract the bits from the ap_fixed<10,5> object.
      std::bitset<BDT_ouput_width> BDT_bits;
      for (int i=0;i<scores[0].length();i++){
    BDT_bits[i] = scores[0][i];
      }
      ATH_MSG_DEBUG("BDT bits " << BDT_bits);
 
      //Shift to an unsigned range, stored in a Bitset<8> as the hardware will.
      std::bitset<eEmEg1BDTTOB::s_eGamma1BDT_width> result;
      //First, interpret the ap_fixed<10,5> as an integer.
      int BDT_int = bitSetToInt(BDT_bits);
      ATH_MSG_DEBUG("BDT int " << BDT_int);
      //We are going to restrict the range, but keep the resolution.
      if(BDT_int >= 128) {
    //So, cut off at 2^8/2-1 as the maximum possible value.
    result = std::bitset<eEmEg1BDTTOB::s_eGamma1BDT_width>{0xFF};
      } else if (BDT_int < -128) {
    //So, cut off at and -2^8/2 as the minimum possible value.
    result = std::bitset<eEmEg1BDTTOB::s_eGamma1BDT_width>{0x00};
      } else {
    //Then we take the remaining bits, flipping the maximum remaining bit.
    //Due to the above logic, this is only 1 if -ive.
    result[eEmEg1BDTTOB::s_eGamma1BDT_width-1] = ~BDT_bits[eEmEg1BDTTOB::s_eGamma1BDT_width-1];
    for(uint i = 0; i <= eEmEg1BDTTOB::s_eGamma1BDT_width-2;i++){
      result[i] = BDT_bits[i];
    }
      }
 
      ATH_MSG_DEBUG("Result bits " << result);
      ATH_MSG_DEBUG("Result int " << result.to_ulong());
      
      //Outpout the ulong for debug, and store the result in the output TOB
      h_BDTScore->push_back(result.to_ulong());
      eEmEg1BDTTOBs->push_back(std::make_unique<IOBitwise::eEmEg1BDTTOB>(*nbhdTOB, result));
    }
 
    //Setup the write out of the resultant TOBs
    SG::WriteHandle<GlobalSim::IOBitwise::eEmEg1BDTTOBContainer> h_eEmEg1BDTTOBs(m_eEmEg1BDTTOBContainerKey, ctx);
    CHECK(h_eEmEg1BDTTOBs.record(std::move(eEmEg1BDTTOBs)));
 
    if (dc){dc->collect(*this, "end");}
 
    return StatusCode::SUCCESS;
  }

toString()

std::string GlobalSim::Egamma1BDTAlgTool::toString ( ) const

overridevirtual

Definition at line 174 of file Egamma1BDTAlgTool.cxx.

                                              {
 
    std::stringstream ss;
    ss << "Egamma1BDTAlgTool. name: " << name() << '\n'
       << m_nbhdTOBContainerReadKey << '\n'
       << '\n';
    return ss.str();
  }

Member Data Documentation

BDT_ouput_width

std::size_t GlobalSim::Egamma1BDTAlgTool::BDT_ouput_width = 10

staticconstexpr

Definition at line 51 of file Egamma1BDTAlgTool.h.

m_BDTScoreKey

SG::WriteHandleKey<std::vector<float> > GlobalSim::Egamma1BDTAlgTool::m_BDTScoreKey

private

Initial value:

{
      this,
      "BDTScoreKey",
      "eGamma1BDT"}

Definition at line 63 of file Egamma1BDTAlgTool.h.

                  {
      this,
      "BDTScoreKey",
      "eGamma1BDT"};    

m_eEmEg1BDTTOBContainerKey

SG::WriteHandleKey<eEmEg1BDTTOBContainer> GlobalSim::Egamma1BDTAlgTool::m_eEmEg1BDTTOBContainerKey

private

Initial value:

{
      this,
      "eEmEg1BDTTOBContainerKey",
      "eEmEg1BDTTOBContainer"}

Definition at line 77 of file Egamma1BDTAlgTool.h.

                               {
      this,
      "eEmEg1BDTTOBContainerKey",
      "eEmEg1BDTTOBContainer"};

m_enableDump

Gaudi::Property<bool> GlobalSim::Egamma1BDTAlgTool::m_enableDump

private

Initial value:

{this,
         "enableDump",
         {false},
         "flag to enable dumps"}

Definition at line 57 of file Egamma1BDTAlgTool.h.

                {this,
         "enableDump",
         {false},
         "flag to enable dumps"};

m_nbhdTOBContainerReadKey

SG::ReadHandleKey<eEmNbhoodTOBContainer> GlobalSim::Egamma1BDTAlgTool::m_nbhdTOBContainerReadKey

private

Initial value:

{
      this,
      "LArNeighborhoodTOBContainerReadKey",
      "stripNeighborhoodTOBContainer",
      "key to read inLArNeighborhoodTOBsReadKeys"}

Definition at line 70 of file Egamma1BDTAlgTool.h.

                              {
      this,
      "LArNeighborhoodTOBContainerReadKey",
      "stripNeighborhoodTOBContainer",
      "key to read inLArNeighborhoodTOBsReadKeys"};

s_combination_len

int GlobalSim::Egamma1BDTAlgTool::s_combination_len = 18

inlinestaticconstexprprivate

Definition at line 92 of file Egamma1BDTAlgTool.h.

s_required_phi_len

int GlobalSim::Egamma1BDTAlgTool::s_required_phi_len = 17

inlinestaticconstexprprivate

Definition at line 88 of file Egamma1BDTAlgTool.h.

---

The documentation for this class was generated from the following files:

• Egamma1BDTAlgTool.h
• Egamma1BDTAlgTool.cxx