Pythia6EventBuilder.cxx

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Pythia6EventBuilder.cxx

#include "eicsmear/erhic/Pythia6ParticleBuilder.h"

#include <cmath>
#include <iostream>
#include <memory>
#include <string>

#include <TBranch.h>
#include <TClass.h>
#include <TMCParticle.h>
#include <TObjArray.h>
#include <TProcessID.h>
#include <TPythia6.h>
#include <TTree.h>

#include "eicsmear/erhic/EventMCFilterABC.h"
#include "eicsmear/erhic/EventPythia.h"
#include "eicsmear/erhic/Kinematics.h"
#include "eicsmear/erhic/ParticleMC.h"
#include "eicsmear/erhic/Pythia6EventBuilder.h"

namespace erhic {

Pythia6EventBuilder::Pythia6EventBuilder(EventMCFilterABC* filter)
: mNGenerated(0)
, mNTrials(0)
, mFilter(filter)
, mEvent(NULL) {
}

Pythia6EventBuilder::~Pythia6EventBuilder() {
  delete mFilter;
  mFilter = NULL;
}

EventPythia* Pythia6EventBuilder::Create() {
  std::unique_ptr<EventPythia> event(BuildEvent());
  if (mFilter) {
    while (!mFilter->Accept(*event)) {
      event.reset(BuildEvent());
    }  // while
  }  // if
  return event.release();
}

EventPythia* Pythia6EventBuilder::BuildEvent() {
  // Save current object count
  int objectNumber = TProcessID::GetObjectCount();
  // Generate a new PYTHIA event
  TPythia6* pythia = TPythia6::Instance();
  pythia->GenerateEvent();
  // Import all particles (not just final-state)
  TObjArray* particles = pythia->ImportParticles("All");
  // Construct the EventPythia object from the current
  // state of TPythia6.
  std::unique_ptr<EventPythia> event(new EventPythia);
  // Extract the event-wise quantities:
  event->SetNucleon(pythia->GetMSTI(12));
  event->SetTargetParton(pythia->GetMSTI(16));
  event->SetBeamParton(pythia->GetMSTI(15));
  event->SetGenEvent(1);
  event->SetTargetPartonX(pythia->GetPARI(34));
  event->SetBeamPartonX(pythia->GetPARI(33));
  event->SetBeamPartonTheta(pythia->GetPARI(53));
  event->SetLeptonPhi(pythia->GetVINT(313));
  event->SetHardS(pythia->GetPARI(14));
  event->SetHardT(pythia->GetPARI(15));
  event->SetHardU(pythia->GetPARI(16));
  event->SetHardQ2(pythia->GetPARI(22));
  event->SetHardPt2(pythia->GetPARI(18));
  event->SetPhotonFlux(pythia->GetVINT(319));
  event->SetProcess(pythia->GetMSTI(1));
  // We need the beam energies to compute the true x, W2 and nu.
  // The beam lepton should be the first particle
  // and the beam hadron the second particle.
  const double eLepton =
  static_cast<TMCParticle*>(particles->At(0))->GetEnergy();
  const double eHadron =
  static_cast<TMCParticle*>(particles->At(1))->GetEnergy();
  const double mHadron =
  static_cast<TMCParticle*>(particles->At(1))->GetMass();
  // x, W2 and nu are calculated from y, Q2 and the beam energies.
  // y and Q2 come from PYTHIA.
  // Use (approximate expression) Q2 = sxy, where s = 4.E_e.E_h
  double y = pythia->GetVINT(309);
  double Q2 = pythia->GetVINT(307);
  double x = Q2 / y / 4. / eLepton / eHadron;
  double W2 = pow(mHadron, 2.) + Q2 * (1. / x - 1.);
  double nu = (W2 + Q2 - pow(mHadron, 2.)) / 2. / mHadron;
  event->SetTrueY(y);
  event->SetTrueQ2(Q2);
  event->SetTrueX(x);
  event->SetTrueW2(W2);
  event->SetTrueNu(nu);
  // Set the number of event generation trials taken since the last call
  event->SetN(pythia->GetMSTI(5));
  event->SetGenEvent(pythia->GetPyint5()->NGEN[2][0] - mNTrials);
  mNTrials = pythia->GetPyint5()->NGEN[2][0];
  // Now populate the particle list.
  Pythia6ParticleBuilder builder;
  for (int i(0); i < particles->GetEntries(); ++i) {
    TMCParticle* p =
    static_cast<TMCParticle*>(particles->At(i));
    std::unique_ptr<ParticleMC> particle = builder.Create(*p);
    particle->SetIndex(i + 1);
    particle->SetEvent(event.get());
    event->AddLast(particle.get());
  }  // for
  // Compute derived event kinematics
  DisKinematics* nm = LeptonKinematicsComputer(*event).Calculate();
  DisKinematics* jb = JacquetBlondelComputer(*event).Calculate();
  DisKinematics* da = DoubleAngleComputer(*event).Calculate();
  if (nm) {
    event->SetLeptonKinematics(*nm);
  }  // if
  if (jb) {
    event->SetJacquetBlondelKinematics(*jb);
  }  // if
  if (da) {
    event->SetDoubleAngleKinematics(*da);
  }  // if
  // We also have to set the remaining variables not taken care of
  // by the general DIS event kinematic computations.
  // Find the beams, exchange boson, scattered lepton.
  BeamParticles beams;
  if (ParticleIdentifier::IdentifyBeams(*event, beams)) {
    const TLorentzVector h = beams.BeamHadron();
    TLorentzVector l = beams.BeamLepton();
    TLorentzVector s = beams.ScatteredLepton();
    TVector3 boost = -h.BoostVector();
    l.Boost(boost);
    s.Boost(boost);
    event->SetELeptonInNuclearFrame(l.E());
    event->SetEScatteredInNuclearFrame(s.E());
  }  // if
  for (unsigned i(0); i < event->GetNTracks(); ++i) {
    event->GetTrack(i)->ComputeEventDependentQuantities(*event);
  }  // for
  // Restore Object count
  // See example in $ROOTSYS/test/Event.cxx
  // To save space in the table keeping track of all referenced objects
  // we assume that our events do not address each other. We reset the
  // object count to what it was at the beginning of the event.
  TProcessID::SetObjectCount(objectNumber);
  return event.release();
}

std::string Pythia6EventBuilder::EventName() const {
  return EventPythia::Class()->GetName();
}

TBranch* Pythia6EventBuilder::Branch(TTree& tree, const std::string& name) {
  EventPythia* event(NULL);
  TBranch* branch =
  tree.Branch(name.c_str(), EventName().c_str(), &event, 32000, 99);
  tree.ResetBranchAddress(branch);
  return branch;
}

void Pythia6EventBuilder::Fill(TBranch& branch) {
  if (mEvent) {
    branch.ResetAddress();
    delete mEvent;
    mEvent = NULL;
  }  // if
  mEvent = Create();
  branch.SetAddress(&mEvent);
}

}  // namespace erhic