FSimVertex.h

Go to the documentation of this file.

#ifndef FastSimulation_Event_FSimVertex_H
#define FastSimulation_Event_FSimVertex_H

// CMSSW Headers
#include "SimDataFormats/Vertex/interface/SimVertex.h"
#include "DataFormats/Math/interface/LorentzVector.h"

#include <vector>

class FBaseSimEvent;
class FSimTrack;

class FSimVertex : public SimVertex {
public:
  FSimVertex();

  FSimVertex(const math::XYZTLorentzVector& v, int im, int id, FBaseSimEvent* mom);

  inline const FSimTrack& parent() const;

  inline const std::vector<int>& daughters() const { return daugh_; }

  inline int nDaughters() const { return daugh_.size(); }

  inline const FSimTrack& daughter(int i) const;

  inline bool noDaughter() const { return !nDaughters(); }

  inline int id() const { return id_; }

  inline void addDaughter(int i) { daugh_.push_back(i); }

  inline const math::XYZTLorentzVector& position() const { return position_; }

  inline void setPosition(const math::XYZTLorentzVector& newPosition) { position_ = newPosition; }

  inline const SimVertex& simVertex() const { return *this; }

private:
  const FBaseSimEvent* mom_;
  int id_;                  // The index in the FSimVertex vector
  std::vector<int> daugh_;  // The indices of the daughters in FSimTrack

  math::XYZTLorentzVector position_;
};

#include <iosfwd>
std::ostream& operator<<(std::ostream& o, const FSimVertex& t);

#include "FastSimulation/Event/interface/FSimVertex.icc"

#endif  // FSimVertex_H