PHG4CylinderCellGeomContainer.cc

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

#include "PHG4CylinderCellGeomContainer.h"

#include "PHG4CylinderCellGeom.h"

using namespace std;

PHG4CylinderCellGeomContainer::~PHG4CylinderCellGeomContainer()
{
  while (layergeoms.begin() != layergeoms.end())
  {
    delete layergeoms.begin()->second;
    layergeoms.erase(layergeoms.begin());
  }
  return;
}

void PHG4CylinderCellGeomContainer::identify(std::ostream &os) const
{
  map<int, PHG4CylinderCellGeom *>::const_iterator iter;
  for (iter = layergeoms.begin(); iter != layergeoms.end(); ++iter)
  {
    cout << "layer " << iter->first << endl;
    (iter->second)->identify(os);
  }
  return;
}

int PHG4CylinderCellGeomContainer::AddLayerCellGeom(const int i, PHG4CylinderCellGeom *mygeom)
{
  if (layergeoms.find(i) != layergeoms.end())
  {
    cout << "layer " << i << " already added to PHCylinderCellGeomContainer" << endl;
    return -1;
  }
  mygeom->set_layer(i);
  layergeoms[i] = mygeom;
  return 0;
}

int PHG4CylinderCellGeomContainer::AddLayerCellGeom(PHG4CylinderCellGeom *mygeom)
{
  int layer = mygeom->get_layer();
  if (layergeoms.find(layer) != layergeoms.end())
  {
    cout << "layer " << layer << " already added to PHCylinderCellGeomContainer" << endl;
    return -1;
  }
  layergeoms[layer] = mygeom;
  return 0;
}

PHG4CylinderCellGeom *
PHG4CylinderCellGeomContainer::GetLayerCellGeom(const int i)
{
  map<int, PHG4CylinderCellGeom *>::const_iterator iter = layergeoms.find(i);
  if (iter != layergeoms.end())
  {
    return iter->second;
  }
  cout << "Could not locate layer " << i << " in PHG4CylinderCellGeomContainer" << endl;
  return nullptr;
}

PHG4CylinderCellGeom *
PHG4CylinderCellGeomContainer::GetFirstLayerCellGeom()
{
  if (layergeoms.empty())
  {
    return nullptr;
  }
  return layergeoms.begin()->second;
}