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;
}