edm::HepMC3Product Class Reference

#include <HepMC3Product.h>

Public Member Functions
void	addHepMCData (HepMC3::GenEvent *evt)
void	applyVtxGen (HepMC3::FourVector const *vtxShift)
void	applyVtxGen (HepMC3::FourVector const &vtxShift)
void	boostToLab (TMatrixD const *lorentz, std::string const &type)
const HepMC3::GenEvent *	GetEvent () const
const HepMC3::GenEvent &	getHepMCData () const
	HepMC3Product ()
	HepMC3Product (HepMC3::GenEvent *evt)
	HepMC3Product (HepMC3Product const &orig)
	HepMC3Product (HepMC3Product &&orig)
bool	isPBoostApplied () const
bool	isVtxBoostApplied () const
bool	isVtxGenApplied () const
HepMC3Product &	operator= (HepMC3Product const &other)
HepMC3Product &	operator= (HepMC3Product &&other)
void	swap (HepMC3Product &other)
virtual	~HepMC3Product ()

Private Attributes
HepMC3::GenEvent *	evt_
bool	isPBoostApplied_
bool	isVtxBoostApplied_
bool	isVtxGenApplied_

Detailed Description

Definition at line 21 of file HepMC3Product.h.

Constructor & Destructor Documentation

HepMC3Product() [1/4]

edm::HepMC3Product::HepMC3Product ( )

inline

Definition at line 23 of file HepMC3Product.h.

: evt_(nullptr), isVtxGenApplied_(false), isVtxBoostApplied_(false), isPBoostApplied_(false) {}

HepMC3Product() [2/4]

HepMC3Product::HepMC3Product ( HepMC3::GenEvent *  evt )

explicit

Definition at line 19 of file HepMC3Product.cc.

    : evt_(evt), isVtxGenApplied_(false), isVtxBoostApplied_(false), isPBoostApplied_(false) {}

~HepMC3Product()

HepMC3Product::~HepMC3Product ( )

virtual

Definition at line 22 of file HepMC3Product.cc.

References evt_, isPBoostApplied_, isVtxBoostApplied_, and isVtxGenApplied_.

                              {
  delete evt_;
  evt_ = nullptr;
  isVtxGenApplied_ = false;
  isVtxBoostApplied_ = false;
  isPBoostApplied_ = false;
}

HepMC3Product() [3/4]

HepMC3Product::HepMC3Product

(

HepMC3Product const &

orig

)

Definition at line 116 of file HepMC3Product.cc.

References evt_, isPBoostApplied_, isVtxBoostApplied_, isVtxGenApplied_, and trackingPlots::other.

                                                       : evt_(nullptr) {
  if (other.evt_)
    evt_ = new HepMC3::GenEvent(*other.evt_);
  isVtxGenApplied_ = other.isVtxGenApplied_;
  isVtxBoostApplied_ = other.isVtxBoostApplied_;
  isPBoostApplied_ = other.isPBoostApplied_;
  //fTimeOffset = other.fTimeOffset;
}

HepMC3Product() [4/4]

HepMC3Product::HepMC3Product

(

HepMC3Product &&

orig

)

Definition at line 142 of file HepMC3Product.cc.

References trackingPlots::other, and swap().

: evt_(nullptr) { swap(other); }

Member Function Documentation

addHepMCData()

void HepMC3Product::addHepMCData

(

HepMC3::GenEvent *

evt

)

Definition at line 30 of file HepMC3Product.cc.

References evt_.

                                                    {
  //  evt->print();
  // cout <<sizeof (evt)  <<"  " << sizeof ( HepMC::GenEvent)   << endl;
  evt_ = evt;

  // same story about vertex smearing - GenEvent won't know it...
  // in fact, would be better to implement CmsGenEvent...
}

applyVtxGen() [1/2]

void edm::HepMC3Product::applyVtxGen ( HepMC3::FourVector const *  vtxShift )

inline

Definition at line 30 of file HepMC3Product.h.

References applyVtxGen().

Referenced by applyVtxGen().

{ applyVtxGen(*vtxShift); }

applyVtxGen() [2/2]

void HepMC3Product::applyVtxGen

(

HepMC3::FourVector const &

vtxShift

)

Definition at line 39 of file HepMC3Product.cc.

References evt_, isVtxGenApplied(), and isVtxGenApplied_.

                                                                {
  //std::cout<< " applyVtxGen called " << isVtxGenApplied_ << endl;
  //fTimeOffset = 0;
  if (isVtxGenApplied())
    return;
  evt_->shift_position_by(vtxShift);
  isVtxGenApplied_ = true;
  return;
}

boostToLab()

void HepMC3Product::boostToLab	(	TMatrixD const *	lorentz,
		std::string const &	type
	)

Definition at line 49 of file HepMC3Product.cc.

References evt_, isPBoostApplied(), isPBoostApplied_, isVtxBoostApplied(), and isVtxBoostApplied_.

                                                                             {
  //std::cout << "from boostToLab:" << std::endl;

  if (lorentz == nullptr) {
    //std::cout << " lorentz = 0 " << std::endl;
    return;
  }

  //lorentz->Print();

  TMatrixD tmplorentz(*lorentz);
  //tmplorentz.Print();

  if (type == "vertex") {
    if (isVtxBoostApplied()) {
      //std::cout << " isVtxBoostApplied true " << std::endl;
      return;
    }

    for (const HepMC3::GenVertexPtr& vt : evt_->vertices()) {
      // change basis to lorentz boost definition: (t,x,z,y)
      TMatrixD p4(4, 1);
      p4(0, 0) = vt->position().t();
      p4(1, 0) = vt->position().x();
      p4(2, 0) = vt->position().z();
      p4(3, 0) = vt->position().y();

      TMatrixD p4lab(4, 1);
      p4lab = tmplorentz * p4;
      //std::cout << " vertex lorentz: " << p4lab(1,0) << " " << p4lab(3,0) << " " << p4lab(2,0) << std::endl;
      vt->set_position(HepMC3::FourVector(p4lab(1, 0), p4lab(3, 0), p4lab(2, 0), p4lab(0, 0)));
    }

    isVtxBoostApplied_ = true;
  } else if (type == "momentum") {
    if (isPBoostApplied()) {
      //std::cout << " isPBoostApplied true " << std::endl;
      return;
    }

    for (const HepMC3::GenParticlePtr& part : evt_->particles()) {
      // change basis to lorentz boost definition: (E,Px,Pz,Py)
      TMatrixD p4(4, 1);
      p4(0, 0) = part->momentum().e();
      p4(1, 0) = part->momentum().x();
      p4(2, 0) = part->momentum().z();
      p4(3, 0) = part->momentum().y();

      TMatrixD p4lab(4, 1);
      p4lab = tmplorentz * p4;
      //std::cout << " momentum lorentz: " << p4lab(1,0) << " " << p4lab(3,0) << " " << p4lab(2,0) << std::endl;
      part->set_momentum(HepMC3::FourVector(p4lab(1, 0), p4lab(3, 0), p4lab(2, 0), p4lab(0, 0)));
    }

    isPBoostApplied_ = true;
  } else {
    edm::LogWarning("GeneratorProducts") << "no type found for boostToLab(std::string), options are vertex or momentum";
    //throw edm::Exception(edm::errors::Configuration, "GeneratorProducts")
    //  << "no type found for boostToLab(std::string), options are vertex or momentum \n";
  }

  return;
}

GetEvent()

const HepMC3::GenEvent* edm::HepMC3Product::GetEvent ( ) const

inline

Definition at line 37 of file HepMC3Product.h.

References evt_.

{ return evt_; }

getHepMCData()

const HepMC3::GenEvent & HepMC3Product::getHepMCData

(

)

const

Definition at line 113 of file HepMC3Product.cc.

References evt_.

{ return *evt_; }

isPBoostApplied()

bool edm::HepMC3Product::isPBoostApplied ( ) const

inline

Definition at line 41 of file HepMC3Product.h.

References isPBoostApplied_.

Referenced by boostToLab().

{ return isPBoostApplied_; }

isVtxBoostApplied()

bool edm::HepMC3Product::isVtxBoostApplied ( ) const

inline

Definition at line 40 of file HepMC3Product.h.

References isVtxBoostApplied_.

Referenced by boostToLab().

{ return isVtxBoostApplied_; }

isVtxGenApplied()

bool edm::HepMC3Product::isVtxGenApplied ( ) const

inline

Definition at line 39 of file HepMC3Product.h.

References isVtxGenApplied_.

Referenced by applyVtxGen().

{ return isVtxGenApplied_; }

operator=() [1/2]

HepMC3Product & HepMC3Product::operator=

(

HepMC3Product const &

other

)

Definition at line 135 of file HepMC3Product.cc.

References trackingPlots::other, swap(), and groupFilesInBlocks::temp.

                                                                  {
  HepMC3Product temp(other);
  swap(temp);
  return *this;
}

operator=() [2/2]

HepMC3Product & HepMC3Product::operator=

(

HepMC3Product &&

other

)

Definition at line 143 of file HepMC3Product.cc.

References trackingPlots::other, and swap().

                                                             {
  swap(other);
  return *this;
}

swap()

void HepMC3Product::swap

(

HepMC3Product &

other

)

Definition at line 126 of file HepMC3Product.cc.

References evt_, isPBoostApplied_, isVtxBoostApplied_, isVtxGenApplied_, trackingPlots::other, and edm::swap().

Referenced by HepMC3Product(), and operator=().

                                             {
  std::swap(evt_, other.evt_);
  std::swap(isVtxGenApplied_, other.isVtxGenApplied_);
  std::swap(isVtxBoostApplied_, other.isVtxBoostApplied_);
  std::swap(isPBoostApplied_, other.isPBoostApplied_);
  //std::swap(fTimeOffset, other.fTimeOffset);
}

Member Data Documentation

evt_

HepMC3::GenEvent* edm::HepMC3Product::evt_

private

Definition at line 50 of file HepMC3Product.h.

Referenced by addHepMCData(), applyVtxGen(), boostToLab(), GetEvent(), getHepMCData(), HepMC3Product(), swap(), and ~HepMC3Product().

isPBoostApplied_

bool edm::HepMC3Product::isPBoostApplied_

private

Definition at line 54 of file HepMC3Product.h.

Referenced by boostToLab(), HepMC3Product(), isPBoostApplied(), swap(), and ~HepMC3Product().

isVtxBoostApplied_

bool edm::HepMC3Product::isVtxBoostApplied_

private

Definition at line 53 of file HepMC3Product.h.

Referenced by boostToLab(), HepMC3Product(), isVtxBoostApplied(), swap(), and ~HepMC3Product().

isVtxGenApplied_

bool edm::HepMC3Product::isVtxGenApplied_

private

Definition at line 52 of file HepMC3Product.h.

Referenced by applyVtxGen(), HepMC3Product(), isVtxGenApplied(), swap(), and ~HepMC3Product().