l1t::ForestHelper Class Reference

#include <ForestHelper.h>

Public Types
enum	{ VERSION = 1 }
typedef L1TMuonEndCapForest::DForest	DForest
typedef L1TMuonEndCapForest::DForestColl	DForestColl
typedef L1TMuonEndCapForest::DForestMap	DForestMap
typedef L1TMuonEndCapForest::DTree	DTree
typedef L1TMuonEndCapForest::DTreeNode	DTreeNode

Public Member Functions
double	evaluate (int mode, const std::vector< double > &data) const
	ForestHelper (L1TMuonEndCapForest *w)
const L1TMuonEndCapForest *	getReadInstance () const
L1TMuonEndCapForest *	getWriteInstance ()
void	initializeFromXML (const char *dirname, const std::vector< int > &modes, int ntrees)
void	print (std::ostream &) const
	~ForestHelper ()

Static Public Member Functions
static ForestHelper *	readAndWriteFromEventSetup (const L1TMuonEndCapForest *es)
static const ForestHelper *	readFromEventSetup (const L1TMuonEndCapForest *es)

Private Member Functions
void	check_write ()
double	evalTreeRecursive (const std::vector< double > &data, const DTree &tree, int index) const
	ForestHelper (const L1TMuonEndCapForest *es)
void	loadTreeFromXMLRecursive (TXMLEngine *xml, XMLNodePointer_t xnode, DTree &tree, unsigned index)
void	useCopy ()

Private Attributes
const L1TMuonEndCapForest *	read_
bool	we_own_write_
L1TMuonEndCapForest *	write_

Detailed Description

Definition at line 39 of file ForestHelper.h.

Member Typedef Documentation

typedef L1TMuonEndCapForest::DForest l1t::ForestHelper::DForest

Definition at line 45 of file ForestHelper.h.

typedef L1TMuonEndCapForest::DForestColl l1t::ForestHelper::DForestColl

Definition at line 46 of file ForestHelper.h.

typedef L1TMuonEndCapForest::DForestMap l1t::ForestHelper::DForestMap

Definition at line 47 of file ForestHelper.h.

typedef L1TMuonEndCapForest::DTree l1t::ForestHelper::DTree

Definition at line 44 of file ForestHelper.h.

typedef L1TMuonEndCapForest::DTreeNode l1t::ForestHelper::DTreeNode

Definition at line 43 of file ForestHelper.h.

Member Enumeration Documentation

anonymous enum

Enumerator
VERSION

Definition at line 41 of file ForestHelper.h.

{VERSION = 1};

Constructor & Destructor Documentation

ForestHelper::~ForestHelper

(

)

Definition at line 36 of file ForestHelper.cc.

                            {
  if (we_own_write_ && write_) delete write_;
}

ForestHelper::ForestHelper

(

L1TMuonEndCapForest *

w

)

Definition at line 20 of file ForestHelper.cc.

References w.

Referenced by getWriteInstance().

                                                  {
  write_ = w; 
  check_write(); 
  we_own_write_ = false;
  //write_->m_version = VERSION; 
  read_ = write_; 
}

ForestHelper::ForestHelper ( const L1TMuonEndCapForest *  es )

private

Definition at line 28 of file ForestHelper.cc.

References NULL.

{read_ = es; write_=NULL;}

Member Function Documentation

void l1t::ForestHelper::check_write ( )

inlineprivate

Definition at line 72 of file ForestHelper.h.

References write_.

{ assert(write_); }

double ForestHelper::evalTreeRecursive ( const std::vector< double > &  data, const DTree &  tree, int  index  ) const

private

Definition at line 210 of file ForestHelper.cc.

References L1TMuonEndCapForest::DTreeNode::fitVal, L1TMuonEndCapForest::DTreeNode::ileft, diffTreeTool::index, L1TMuonEndCapForest::DTreeNode::iright, L1TMuonEndCapForest::DTreeNode::splitVal, and L1TMuonEndCapForest::DTreeNode::splitVar.

                                                                                                         {
  const DTreeNode & node = tree[index];
  if ((node.ileft == 0) && (node.iright==0)){
    //cout << "terminal node:  fitVal:  " << node.fitVal << "\n";
    return node.fitVal;
  }
  assert(data.size() > (unsigned) node.splitVar);
  // by convention, nodes are either not terminal or fully terminal
  assert(node.ileft > 0);
  assert(node.iright > 0);

  //cout << "NODE:  svar: " << node.splitVar << " sval: " << node.splitVal << " data: " << data[node.splitVar] << "fit:  " << node.fitVal << "\n";
  
  if (data[node.splitVar] < node.splitVal){
    //cout << "going left to " << node.ileft << "\n";
    return evalTreeRecursive(data, tree, node.ileft);
  } else {
    //cout << "going right to " << node.iright << "\n";
    return evalTreeRecursive(data, tree, node.iright);
  }
}

double ForestHelper::evaluate	(	int	mode,
		const std::vector< double > &	data
	)		const

Definition at line 128 of file ForestHelper.cc.

References x().

                                                                            {
  auto it = read_->forest_map_.find(mode);
  if (it == read_->forest_map_.end())
    return 0;

  const DForest & dforest = read_->forest_coll_[it->second];
  
  double sum = 0;
  for (auto itree = dforest.begin(); itree != dforest.end(); itree++){
    double x = evalTreeRecursive(data, *itree, 0);
    //cout << "forest eval to " << x << "\n";
    sum += x;
  }
  return sum;  
}

const L1TMuonEndCapForest* l1t::ForestHelper::getReadInstance ( ) const

inline

Definition at line 66 of file ForestHelper.h.

References read_.

{return read_;}

L1TMuonEndCapForest* l1t::ForestHelper::getWriteInstance ( )

inline

Definition at line 67 of file ForestHelper.h.

References ForestHelper(), useCopy(), and write_.

Referenced by L1TMuonEndCapForestESProducer::produce().

{return write_; }

void ForestHelper::initializeFromXML	(	const char *	dirname,
		const std::vector< int > &	modes,
		int	ntrees
	)

Definition at line 72 of file ForestHelper.cc.

References createBeamHaloJobs::directory, corrVsCorr::filename, i, createfilelist::int, j, ALCARECOPromptCalibProdSiPixelAli0T_cff::mode, AlCaHLTBitMon_QueryRunRegistry::string, and cmsPerfSuiteHarvest::xmldoc.

Referenced by L1TMuonEndCapForestESProducer::L1TMuonEndCapForestESProducer().

                                                                                                  {

  //cout << "DEBUG: starting initializeFromXML...\n";

  assert(write_->forest_coll_.size() == 0);

  for(int i =0; i < (int) modes.size(); i++){
    int mode = modes[i];
    //cout << "DEBUG: initializing Decision Forest for mode=" << mode << "\n";

    //DForest tmp;
    write_->forest_coll_.push_back(DForest());
    DForest & dforest = write_->forest_coll_[i];
    write_->forest_map_[mode]=i;

    std::stringstream ss;
    ss << dirname << "/" << mode;
    std::string directory;
    ss >> directory;

    for(int j=0; j < ntrees; j++){   
      std::stringstream ss;
      ss << directory << "/" << j << ".xml";
      std::string filename;
      ss >> filename;
      //cout << "DEBUG: loading tree " << filename << "\n";  

      
      dforest.push_back(DTree());
      DTree & dtree = dforest[j];
      dtree.push_back(DTreeNode());


      // First create the engine.
      TXMLEngine* xml = new TXMLEngine();

      // Now try to parse xml file.
      XMLDocPointer_t xmldoc = xml->ParseFile(edm::FileInPath(filename.c_str()).fullPath().c_str());
      if (xmldoc==0){
    delete xml;
    continue;
      }
      // Get access to main node of the xml file.
      XMLNodePointer_t mainnode = xml->DocGetRootElement(xmldoc);
      
      loadTreeFromXMLRecursive(xml, mainnode, dtree, 0);

      //cout << "DEBUG: parsed tree of size " << dtree.size() << " for mode " << mode << "\n";

      xml->FreeDoc(xmldoc);
      delete xml;

     }
  }
}

void ForestHelper::loadTreeFromXMLRecursive ( TXMLEngine *  xml, XMLNodePointer_t  xnode, DTree &  tree, unsigned  index  )

private

Definition at line 147 of file ForestHelper.cc.

References i, and diffTreeTool::index.

{
  assert(tree.size() > index); 
  //cout << "DEBUG:  recursive call at index " << index <<"\n";

  // Get the split information from xml.
  XMLAttrPointer_t attr = xml->GetFirstAttr(xnode);
  std::vector<std::string> splitInfo(3);
  for(unsigned int i=0; i<3; i++)
    {
      splitInfo[i] = xml->GetAttrValue(attr); 
      attr = xml->GetNextAttr(attr);  
    }
  
  // Convert strings into numbers.
  std::stringstream converter;
  Int_t splitVar;
  Double_t splitVal;
  Double_t fitVal;  

  converter << splitInfo[0];
  converter >> splitVar;
  converter.str("");
  converter.clear();
  
  converter << splitInfo[1];
  converter >> splitVal;
  converter.str("");
  converter.clear();
  
  converter << splitInfo[2];
  converter >> fitVal;
  converter.str("");
  converter.clear();
  
  //cout << "fitval:  " << fitVal << "\n";

  // Store gathered splitInfo into the node object.
  tree[index].splitVar = splitVar;
  tree[index].splitVal = splitVal;
  tree[index].fitVal   = fitVal;

  // Get the xml daughters of the current xml node. 
  XMLNodePointer_t xleft = xml->GetChild(xnode);
  XMLNodePointer_t xright = xml->GetNext(xleft);

  assert( ((xleft!=0)&&(xright!=0)) || ((xleft==0)&&(xright==0)) );
  
  // This seems potentially problematic, but leaving for now until
  // bitwise equivalence is demonstrated...
  if(xleft == 0 || xright == 0) return;

  // append two more nodes at end of tree and update this indices in this node:
  tree[index].ileft  = tree.size();
  tree[index].iright = tree[index].ileft + 1;
  tree.push_back(DTreeNode());
  tree.push_back(DTreeNode());

  // recursively handle the next two nodes:
  loadTreeFromXMLRecursive(xml, xleft, tree, tree[index].ileft);
  loadTreeFromXMLRecursive(xml, xright, tree, tree[index].iright);
}

void ForestHelper::print

(

std::ostream &

myStr

)

const

Definition at line 42 of file ForestHelper.cc.

References KineDebug3::count(), gather_cfg::cout, L1TMuonEndCapForest::DTreeNode::fitVal, L1TMuonEndCapForest::DTreeNode::ileft, diffTreeTool::index, L1TMuonEndCapForest::DTreeNode::iright, ALCARECOPromptCalibProdSiPixelAli0T_cff::mode, L1TMuonEndCapForest::DTreeNode::splitVal, and L1TMuonEndCapForest::DTreeNode::splitVar.

                                                {
    myStr << "\nL1T EndCap  Parameters \n" << std::endl;


    for (int mode=0; mode<16; mode++){
      if (mode != 15) continue;

      auto it = read_->forest_map_.find(mode);
      if (it == read_->forest_map_.end())
    continue;

      const DForest & dforest = read_->forest_coll_[it->second];
      
      int count = 0;
      for (auto itree = dforest.begin(); itree != dforest.end(); itree++){
    const DTree & tree = *itree;
    cout << "DUMP: ***** Tree " << count << " with size " << tree.size() << "\n";
    for (unsigned index=0; index<tree.size(); index++){
      const DTreeNode & node = tree[index];
      cout << "node " << index << " l: " << node.ileft << " r: " << node.iright << "svar: " << node.splitVar << " sval: " << node.splitVal << " fit: " << node.fitVal << "\n";
    }


    count++;
      }
    }
}

ForestHelper * ForestHelper::readAndWriteFromEventSetup ( const L1TMuonEndCapForest *  es )

static

Definition at line 14 of file ForestHelper.cc.

References useCopy(), and x().

                                                                                      {
  ForestHelper * x = new ForestHelper(es);
  x->useCopy();
  return x;
}

const ForestHelper * ForestHelper::readFromEventSetup ( const L1TMuonEndCapForest *  es )

static

Definition at line 10 of file ForestHelper.cc.

                                                                                    {
  return new ForestHelper(es);
}

void ForestHelper::useCopy ( )

private

Definition at line 30 of file ForestHelper.cc.

Referenced by getWriteInstance(), and readAndWriteFromEventSetup().

                          {
  write_ = new L1TMuonEndCapForest(*read_);
  we_own_write_ = true;
  read_  = write_;
}

Member Data Documentation

const L1TMuonEndCapForest* l1t::ForestHelper::read_

private

Definition at line 74 of file ForestHelper.h.

Referenced by getReadInstance().

bool l1t::ForestHelper::we_own_write_

private

Definition at line 76 of file ForestHelper.h.

L1TMuonEndCapForest* l1t::ForestHelper::write_

private

Definition at line 75 of file ForestHelper.h.

Referenced by check_write(), and getWriteInstance().