FSimDisplacedVertex.h

Go to the documentation of this file.

#ifndef FastSimDataFormats_NuclearInteractions_FSimDisplacedVertex_h
#define FastSimDataFormats_NuclearInteractions_FSimDisplacedVertex_h

#include "SimDataFormats/Vertex/interface/SimVertex.h"

#include "FastSimDataFormats/NuclearInteractions/interface/FSimVertexType.h"

#include <iostream>
#include <vector>

class FSimDisplacedVertex {
public:
  FSimDisplacedVertex();
  FSimDisplacedVertex(const SimVertex& vertex,
                      unsigned id,
                      int motherId,
                      unsigned nCharged,
                      const std::vector<int>& daughterIds,
                      const FSimVertexType::VertexType vertexType);

  FSimDisplacedVertex(const FSimDisplacedVertex& other);

  virtual ~FSimDisplacedVertex() {}

  const SimVertex vertex() const { return vertex_; }

  int id() const { return id_; }

  int motherId() const { return motherId_; }

  unsigned int nDaughters() const { return daughterIds_.size(); }

  unsigned int nChargedDaughters() const { return nCharged_; }

  const std::vector<int>& daughterIds() const { return daughterIds_; }

  const FSimVertexType::VertexType vertexType() const { return vertexType_; }

  const bool isRecoVertex() const { return isRecoVertex_; }

  const int recoVertexId() const { return recoVertexId_; }

  void setRecoVertex(int recoVertexId) {
    isRecoVertex_ = true;
    recoVertexId_ = recoVertexId;
  }

  void removeRecoVertex() {
    isRecoVertex_ = false;
    recoVertexId_ = -1;
  }

private:
  SimVertex vertex_;

  int id_;

  int motherId_;

  unsigned int nCharged_;

  std::vector<int> daughterIds_;

  FSimVertexType::VertexType vertexType_;

  bool isRecoVertex_;

  int recoVertexId_;

  friend std::ostream& operator<<(std::ostream& out, const FSimDisplacedVertex& co);
};

#endif