TrackPoint.h

Go to the documentation of this file.

/* 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>

#ifndef __CINT__
#include <boost/scoped_ptr.hpp>
#endif

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::vector<AbsFitterInfo *> vFitterInfos_; 

#ifndef __CINT__
  boost::scoped_ptr<ThinScatterer> thinScatterer_; // Ownership
#else
  class ThinScatterer *thinScatterer_;
#endif

 public:
  ClassDef(TrackPoint, 1)
};

} /* End of namespace genfit */
#endif // genfit_TrackPoint_h