edm::HepMCProduct Class Reference

#include <HepMCProduct.h>

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

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

Detailed Description

Definition at line 21 of file HepMCProduct.h.

Constructor & Destructor Documentation

edm::HepMCProduct::HepMCProduct ( )

inline

Definition at line 23 of file HepMCProduct.h.

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

HepMCProduct::HepMCProduct ( HepMC::GenEvent *  evt )

explicit

Definition at line 17 of file HepMCProduct.cc.

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

HepMCProduct::~HepMCProduct ( )

virtual

Definition at line 20 of file HepMCProduct.cc.

References evt_, isPBoostApplied_, isVtxBoostApplied_, and isVtxGenApplied_.

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

HepMCProduct::HepMCProduct

(

HepMCProduct const &

orig

)

Definition at line 123 of file HepMCProduct.cc.

References evt_, isPBoostApplied_, isVtxBoostApplied_, and isVtxGenApplied_.

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

HepMCProduct::HepMCProduct

(

HepMCProduct &&

orig

)

Definition at line 149 of file HepMCProduct.cc.

References swap().

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

Member Function Documentation

void HepMCProduct::addHepMCData

(

HepMC::GenEvent *

evt

)

Definition at line 28 of file HepMCProduct.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...
}

void edm::HepMCProduct::applyVtxGen ( HepMC::FourVector const *  vtxShift )

inline

Definition at line 30 of file HepMCProduct.h.

References applyVtxGen().

Referenced by applyVtxGen().

{ applyVtxGen(*vtxShift); }

void HepMCProduct::applyVtxGen

(

HepMC::FourVector const &

vtxShift

)

Definition at line 37 of file HepMCProduct.cc.

References evt_, isVtxGenApplied(), isVtxGenApplied_, and submitPVValidationJobs::t.

                                                              {
  //std::cout<< " applyVtxGen called " << isVtxGenApplied_ << endl;
  //fTimeOffset = 0;

  if (isVtxGenApplied())
    return;

  for (HepMC::GenEvent::vertex_iterator vt = evt_->vertices_begin(); vt != evt_->vertices_end(); ++vt) {
    double x = (*vt)->position().x() + vtxShift.x();
    double y = (*vt)->position().y() + vtxShift.y();
    double z = (*vt)->position().z() + vtxShift.z();
    double t = (*vt)->position().t() + vtxShift.t();
    //std::cout << " vertex (x,y,z)= " << x <<" " << y << " " << z << std::endl;
    (*vt)->set_position(HepMC::FourVector(x, y, z, t));
  }

  isVtxGenApplied_ = true;

  return;
}

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

Definition at line 58 of file HepMCProduct.cc.

References gather_cfg::cout, 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 (HepMC::GenEvent::vertex_iterator vt = evt_->vertices_begin(); vt != evt_->vertices_end(); ++vt) {
      // 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(HepMC::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 (HepMC::GenEvent::particle_iterator part = evt_->particles_begin(); part != evt_->particles_end(); ++part) {
      // 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(HepMC::FourVector(p4lab(1, 0), p4lab(3, 0), p4lab(2, 0), p4lab(0, 0)));
    }

    isPBoostApplied_ = true;
  } else {
    std::cout << " no type found for boostToLab(std::string), options are vertex or momentum" << std::endl;
  }

  return;
}

const HepMC::GenEvent* edm::HepMCProduct::GetEvent ( ) const

inline

Definition at line 37 of file HepMCProduct.h.

References evt_.

Referenced by edm::TestMix::analyze(), MuScleFitPlotter::fillGen(), MuScleFitUtils::findGenMuFromRes(), MuScleFitMuonSelector::findGenMuFromRes(), gen::PyquenHadronizer::generatePartonsAndHadronize(), gen::HydjetHadronizer::generatePartonsAndHadronize(), MixEvtVtxGenerator::getVertex(), and GenParticleProducer::produce().

{ return evt_; }

const HepMC::GenEvent & HepMCProduct::getHepMCData

(

)

const

Definition at line 120 of file HepMCProduct.cc.

References evt_.

{ return *evt_; }

bool edm::HepMCProduct::isPBoostApplied ( ) const

inline

Definition at line 41 of file HepMCProduct.h.

References isPBoostApplied_.

Referenced by boostToLab().

{ return isPBoostApplied_; }

bool edm::HepMCProduct::isVtxBoostApplied ( ) const

inline

Definition at line 40 of file HepMCProduct.h.

References isVtxBoostApplied_.

Referenced by boostToLab().

{ return isVtxBoostApplied_; }

bool edm::HepMCProduct::isVtxGenApplied ( ) const

inline

Definition at line 39 of file HepMCProduct.h.

References isVtxGenApplied_.

Referenced by applyVtxGen(), gen::PyquenHadronizer::generatePartonsAndHadronize(), gen::HydjetHadronizer::generatePartonsAndHadronize(), and MixEvtVtxGenerator::getVertex().

{ return isVtxGenApplied_; }

HepMCProduct & HepMCProduct::operator=

(

HepMCProduct const &

other

)

Definition at line 142 of file HepMCProduct.cc.

References swap(), and groupFilesInBlocks::temp.

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

HepMCProduct & HepMCProduct::operator=

(

HepMCProduct &&

other

)

Definition at line 150 of file HepMCProduct.cc.

References swap().

                                                          {
  swap(other);
  return *this;
}

void HepMCProduct::swap

(

HepMCProduct &

other

)

Definition at line 133 of file HepMCProduct.cc.

References evt_, isPBoostApplied_, isVtxBoostApplied_, isVtxGenApplied_, and std::swap().

Referenced by HepMCProduct(), 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

HepMC::GenEvent* edm::HepMCProduct::evt_

private

Definition at line 50 of file HepMCProduct.h.

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

bool edm::HepMCProduct::isPBoostApplied_

private

Definition at line 54 of file HepMCProduct.h.

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

bool edm::HepMCProduct::isVtxBoostApplied_

private

Definition at line 53 of file HepMCProduct.h.

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

bool edm::HepMCProduct::isVtxGenApplied_

private

Definition at line 52 of file HepMCProduct.h.

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