GFRecoHitProducer.h

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/>.
*/
#ifndef GFRECOHITPRODUCER_H
#define GFRECOHITPRODUCER_H

#include<vector>
#include<map>
#include<assert.h>
#include<iostream>

#include "TClonesArray.h"

#include "GFException.h"

class GFAbsRecoHit;

class GFAbsRecoHitProducer {
public:
  virtual GFAbsRecoHit* produce(int index)=0;
  virtual ~GFAbsRecoHitProducer();
};


template <class hit_T,class recoHit_T>
class GFRecoHitProducer : public GFAbsRecoHitProducer {
 private:
  TClonesArray* hitArrayTClones;
  //std::vector<GFAbsRecoHit*>* hitArrayVector;
 public:

  GFRecoHitProducer(TClonesArray*);
  //GFRecoHitProducer(std::vector<GFAbsRecoHit*>*);
  virtual ~GFRecoHitProducer();

  virtual GFAbsRecoHit* produce(int index);     
};
template <class hit_T,class recoHit_T>
GFRecoHitProducer<hit_T,recoHit_T>::GFRecoHitProducer(TClonesArray* theArr) {
  hitArrayTClones = theArr;
  //hitArrayVector = NULL;
}
/*
template <class hit_T,class recoHit_T>
  GFRecoHitProducer<hit_T,recoHit_T>::GFRecoHitProducer(std::vector<GFAbsRecoHit*>* theArr) {
  hitArrayTClones = NULL;
  hitArrayVector = theArr;
}
*/

template <class hit_T,class recoHit_T>
GFRecoHitProducer<hit_T,recoHit_T>::~GFRecoHitProducer() {
  //we dont assume ownership over the hit arrays
}


template <class hit_T,class recoHit_T>
GFAbsRecoHit* GFRecoHitProducer<hit_T,recoHit_T>::produce(int index) {
  assert(hitArrayTClones!=NULL);
  //assert(hitArrayTClones!=NULL || hitArrayVector!=NULL);//at least one exists
  //assert(!(hitArrayTClones!=NULL && hitArrayVector!=NULL));//but not both
  //if(hitArrayTClones!=NULL){
    //the ROOT guys really use 0 and not NULL grrr...
  if(hitArrayTClones->At(index) == 0) {
    GFException e("In GFRecoHitProducer: index for hit in TClonesArray out of bounds",__LINE__,__FILE__);
    e.setFatal();
    throw e;
  }
  return ( new recoHit_T( (hit_T*) hitArrayTClones->At(index) ) );
  //}
  //else{//after assertions this is save: the hitArrayVector is good
  //  if(index >= hitArrayVector->size()) {
  //    GFException e("In GFRecoHitProducer: index for hit in std::vector out of bounds",__LINE__,__FILE__);
  //    e.setFatal();
  //    throw e;
  //  }
  //  return ( new recoHit_T( (hit_T*) hitArrayVector->at(index) ) );
  //}
}


#endif