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::GenEventData *	GetEvent () const
	HepMC3Product ()
	HepMC3Product (HepMC3::GenEvent *evt)
bool	isPBoostApplied () const
bool	isVtxBoostApplied () const
bool	isVtxGenApplied () const
	~HepMC3Product ()

Private Attributes
HepMC3::GenEventData	evt_
bool	isPBoostApplied_
bool	isVtxBoostApplied_
bool	isVtxGenApplied_

Detailed Description

Definition at line 22 of file HepMC3Product.h.

Constructor & Destructor Documentation

HepMC3Product() [1/2]

edm::HepMC3Product::HepMC3Product ( )

inline

Definition at line 24 of file HepMC3Product.h.

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

HepMC3Product() [2/2]

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

explicit

Definition at line 19 of file HepMC3Product.cc.

References addHepMCData().

    : isVtxGenApplied_(false), isVtxBoostApplied_(false), isPBoostApplied_(false) {
  addHepMCData(evt);
}

~HepMC3Product()

HepMC3Product::~HepMC3Product ( )

default

Member Function Documentation

addHepMCData()

void HepMC3Product::addHepMCData

(

HepMC3::GenEvent *

evt

)

Definition at line 26 of file HepMC3Product.cc.

References evt_.

Referenced by HepMC3Product().

{ evt->write_data(evt_); }

applyVtxGen() [1/2]

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

inline

Definition at line 31 of file HepMC3Product.h.

References applyVtxGen().

Referenced by applyVtxGen().

{ applyVtxGen(*vtxShift); }

applyVtxGen() [2/2]

void HepMC3Product::applyVtxGen

(

HepMC3::FourVector const &

vtxShift

)

Definition at line 28 of file HepMC3Product.cc.

References evt_, isVtxGenApplied(), and isVtxGenApplied_.

                                                                {
  //std::cout<< " applyVtxGen called " << isVtxGenApplied_ << endl;
  //fTimeOffset = 0;
  if (isVtxGenApplied())
    return;
  HepMC3::GenEvent evt;
  evt.read_data(evt_);
  evt.shift_position_by(vtxShift);
  evt.write_data(evt_);
  isVtxGenApplied_ = true;
  return;
}

boostToLab()

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

Definition at line 41 of file HepMC3Product.cc.

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

                                                                             {
  //std::cout << "from boostToLab:" << std::endl;
  HepMC3::GenEvent evt;
  evt.read_data(evt_);

  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";
  }

  evt.write_data(evt_);
  return;
}

GetEvent()

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

inline

Definition at line 36 of file HepMC3Product.h.

References evt_.

Referenced by RivetAnalyzer::analyze(), and HTXSRivetProducer::produce().

{ return &evt_; }

isPBoostApplied()

bool edm::HepMC3Product::isPBoostApplied ( ) const

inline

Definition at line 40 of file HepMC3Product.h.

References isPBoostApplied_.

Referenced by boostToLab().

{ return isPBoostApplied_; }

isVtxBoostApplied()

bool edm::HepMC3Product::isVtxBoostApplied ( ) const

inline

Definition at line 39 of file HepMC3Product.h.

References isVtxBoostApplied_.

Referenced by boostToLab().

{ return isVtxBoostApplied_; }

isVtxGenApplied()

bool edm::HepMC3Product::isVtxGenApplied ( ) const

inline

Definition at line 38 of file HepMC3Product.h.

References isVtxGenApplied_.

Referenced by applyVtxGen().

{ return isVtxGenApplied_; }

Member Data Documentation

evt_

HepMC3::GenEventData edm::HepMC3Product::evt_

private

Definition at line 43 of file HepMC3Product.h.

Referenced by addHepMCData(), applyVtxGen(), boostToLab(), and GetEvent().

isPBoostApplied_

bool edm::HepMC3Product::isPBoostApplied_

private

Definition at line 47 of file HepMC3Product.h.

Referenced by boostToLab(), and isPBoostApplied().

isVtxBoostApplied_

bool edm::HepMC3Product::isVtxBoostApplied_

private

Definition at line 46 of file HepMC3Product.h.

Referenced by boostToLab(), and isVtxBoostApplied().

isVtxGenApplied_

bool edm::HepMC3Product::isVtxGenApplied_

private

Definition at line 45 of file HepMC3Product.h.

Referenced by applyVtxGen(), and isVtxGenApplied().