TrackPoint.h

/* Copyright 2008-2010, Technische Universitaet Muenchen,
   Authors: Christian Hoeppner & Sebastian Neubert & Johannes Rauch
 
   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 genfit_TrackPoint_h
#define genfit_TrackPoint_h
 
#include "AbsMeasurement.h"
#include "AbsFitterInfo.h"
#include "ThinScatterer.h"
 
#include <TObject.h>
 
#include <map>
#include <vector>
#include <memory>
 
 
namespace genfit {
 
class Track;
class KalmanFitterInfo;
 
class TrackPoint : public TObject {
 
 public:
 
  TrackPoint();
  TrackPoint(Track* track);
 
  TrackPoint(const std::vector< genfit::AbsMeasurement* >& rawMeasurements, Track* track);
 
  TrackPoint(genfit::AbsMeasurement* rawMeasurement, Track* track);
 
  TrackPoint(const TrackPoint&); // copy constructor
  TrackPoint& operator=(TrackPoint); // assignment operator
  void swap(TrackPoint& other);
 
  TrackPoint(const TrackPoint& rhs,
      const std::map<const genfit::AbsTrackRep*, genfit::AbsTrackRep*>& map,
      const std::vector<const genfit::AbsTrackRep*> * repsToIgnore = nullptr);
 
  virtual ~TrackPoint();
 
 
  double getSortingParameter() const {return sortingParameter_;}
 
  Track* getTrack() const {return track_;}
  void setTrack(Track* track) {track_ = track;}
 
  const std::vector< genfit::AbsMeasurement* >& getRawMeasurements() const {return rawMeasurements_;}
  AbsMeasurement* getRawMeasurement(int i = 0) const;
  unsigned int getNumRawMeasurements() const {return rawMeasurements_.size();}
  bool hasRawMeasurements() const {return (rawMeasurements_.size() != 0);}
  std::vector< genfit::AbsFitterInfo* > getFitterInfos() const;
  AbsFitterInfo* getFitterInfo(const AbsTrackRep* rep = nullptr) const;
  KalmanFitterInfo* getKalmanFitterInfo(const AbsTrackRep* rep = nullptr) const;
  bool hasFitterInfo(const AbsTrackRep* rep) const {
    return (fitterInfos_.find(rep) != fitterInfos_.end());
  }
 
  ThinScatterer* getMaterialInfo() const {return thinScatterer_.get();}
  bool hasThinScatterer() const {return thinScatterer_.get() != nullptr;}
 
 
  void setSortingParameter(double sortingParameter) {sortingParameter_ = sortingParameter;}
  void addRawMeasurement(genfit::AbsMeasurement* rawMeasurement) {assert(rawMeasurement!=nullptr); rawMeasurement->setTrackPoint(this); rawMeasurements_.push_back(rawMeasurement);}
  void deleteRawMeasurements();
  void setFitterInfo(genfit::AbsFitterInfo* fitterInfo);
  void deleteFitterInfo(const AbsTrackRep* rep) {delete fitterInfos_[rep]; fitterInfos_.erase(rep);}
 
  void setScatterer(ThinScatterer* scatterer) {thinScatterer_.reset(scatterer);}
 
  void Print(const Option_t* = "") const;
 
  void fixupRepsForReading();
 
 private:
  double sortingParameter_;
 
  Track* track_; 
 
  std::vector<AbsMeasurement*> rawMeasurements_; // Ownership
 
  std::map< const AbsTrackRep*, AbsFitterInfo* > fitterInfos_; 
 
  std::map<unsigned int, AbsFitterInfo*> vFitterInfos_; 
 
  std::unique_ptr<ThinScatterer> thinScatterer_; // Ownership
 
 public:
 
  ClassDef(TrackPoint,1)
 
};
 
} /* End of namespace genfit */
#endif // genfit_TrackPoint_h