GFRecoHitFactory.cxx

Go to the documentation of this file.

/* Copyright 2008-2010, Technische Universitaet Muenchen,
   Authors: Christian Hoeppner & Sebastian Neubert

   This file is part of GENFIT.

   GENFIT is free software: you can redistribute it and/or modify
   it under the terms of the GNU Lesser General Public License as published
   by the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   GENFIT is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public License
   along with GENFIT.  If not, see <http://www.gnu.org/licenses/>.
*/
#include "larreco/Genfit/GFRecoHitFactory.h"
#include "larreco/Genfit/GFException.h"
#include "larreco/Genfit/GFRecoHitProducer.h"
#include "larreco/Genfit/GFTrackCand.h"

genf::GFRecoHitFactory::GFRecoHitFactory() {}

genf::GFRecoHitFactory::~GFRecoHitFactory()
{
  clear();
}

void genf::GFRecoHitFactory::addProducer(int detID, GFAbsRecoHitProducer* hitProd)
{
  if (fHitProdMap[detID] != NULL) {
    GFException exc("GFRecoHitFactory: detID already in use", __LINE__, __FILE__);
    exc.setFatal();
    std::vector<double> numbers;
    numbers.push_back(detID);
    exc.setNumbers("detID", numbers);
    throw exc;
  }
  else {
    fHitProdMap[detID] = hitProd;
  }
}

void genf::GFRecoHitFactory::clear()
{
  std::map<int, GFAbsRecoHitProducer*>::iterator it = fHitProdMap.begin();
  while (it != fHitProdMap.end()) {
    delete it->second;
    ++it;
  }
  fHitProdMap.clear();
}

genf::GFAbsRecoHit* genf::GFRecoHitFactory::createOne(int detID, int index)
{
  if (fHitProdMap[detID] != NULL) {
    return (genf::GFAbsRecoHit*)(fHitProdMap[detID]->produce(index));
  }

  else {
    GFException exc(
      "GFRecoHitFactory: no hitProducer for this detID available", __LINE__, __FILE__);
    exc.setFatal();
    std::vector<double> numbers;
    numbers.push_back(detID);
    exc.setNumbers("detID", numbers);
    throw exc;
  }
}

std::vector<genf::GFAbsRecoHit*> genf::GFRecoHitFactory::createMany(const GFTrackCand& cand)
{
  std::vector<genf::GFAbsRecoHit*> hitVec;
  unsigned int nHits = cand.getNHits();
  for (unsigned int i = 0; i < nHits; i++) {
    unsigned int detID;
    unsigned int index;
    cand.getHit(i, detID, index);
    hitVec.push_back(createOne(detID, index));
  }
  return hitVec;
}