PHG4CellContainer.cc

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHG4CellContainer.cc

#include "PHG4CellContainer.h"

#include "PHG4Cell.h"  // for PHG4Cell
#include "PHG4CellDefs.h"
#include "PHG4Cellv1.h"

#include <cstdlib>

using namespace std;

PHG4CellContainer::PHG4CellContainer() = default;

void PHG4CellContainer::Reset()
{
  while (cellmap.begin() != cellmap.end())
  {
    delete cellmap.begin()->second;
    cellmap.erase(cellmap.begin());
  }
  return;
}

void PHG4CellContainer::identify(ostream& os) const
{
  ConstIterator iter;
  os << "Number of cells: " << size() << endl;
  for (iter = cellmap.begin(); iter != cellmap.end(); ++iter)
  {
    os << "cell key 0x" << hex << iter->first << dec << endl;
    (iter->second)->identify();
  }
  return;
}

PHG4CellContainer::ConstIterator
PHG4CellContainer::AddCell(PHG4Cell* newcell)
{
  PHG4CellDefs::keytype key = newcell->get_cellid();
  if (cellmap.find(key) != cellmap.end())
  {
    cout << "overwriting cell 0x" << hex << key << dec << endl;
    cout << "layer: " << PHG4CellDefs::get_detid(key) << endl;
  }
  cellmap[key] = newcell;
  return cellmap.find(key);
}

PHG4CellContainer::ConstIterator
PHG4CellContainer::AddCellSpecifyKey(const PHG4CellDefs::keytype key, PHG4Cell* newcell)
{
  if (cellmap.find(key) != cellmap.end())
  {
    cout << "PHG4CellContainer::AddCellSpecifyKey: duplicate key: " << key << " exiting now" << endl;
    exit(1);
  }
  newcell->set_cellid(key);
  cellmap[key] = newcell;
  return cellmap.find(key);
}

PHG4CellContainer::ConstRange
PHG4CellContainer::getCells(const unsigned short int detid) const
{
  PHG4CellDefs::keytype tmp = detid;
  PHG4CellDefs::keytype keylow = tmp << PHG4CellDefs::bitshift_layer;
  PHG4CellDefs::keytype keyup = ((tmp + 1) << PHG4CellDefs::bitshift_layer) - 1;
  //   cout << "keylow: 0x" << hex << keylow << dec << endl;
  //   cout << "keyup: 0x" << hex << keyup << dec << endl;
  ConstRange retpair;
  retpair.first = cellmap.lower_bound(keylow);
  retpair.second = cellmap.upper_bound(keyup);
  return retpair;
}

PHG4CellContainer::ConstRange
PHG4CellContainer::getCells() const
{
  return std::make_pair(cellmap.begin(), cellmap.end());
}

PHG4CellContainer::Iterator
PHG4CellContainer::findOrAddCell(PHG4CellDefs::keytype key)
{
  PHG4CellContainer::Iterator it = cellmap.find(key);
  if (it == cellmap.end())
  {
    cellmap[key] = new PHG4Cellv1();
    it = cellmap.find(key);
    PHG4Cell* mcell = it->second;
    mcell->set_cellid(key);
  }
  return it;
}

PHG4Cell*
PHG4CellContainer::findCell(PHG4CellDefs::keytype key)
{
  PHG4CellContainer::ConstIterator it = cellmap.find(key);

  if (it != cellmap.end())
  {
    return it->second;
  }

  return nullptr;
}

double
PHG4CellContainer::getTotalEdep() const
{
  ConstIterator iter;
  double totalenergy = 0;
  for (iter = cellmap.begin(); iter != cellmap.end(); ++iter)
  {
    totalenergy += iter->second->get_edep();
  }
  return totalenergy;
}