EvtJetSet.cc

Go to the documentation of this file.

//--------------------------------------------------------------------------
//
// Environment:
//      This software is part of the EvtGen package developed jointly
//      for the BaBar and CLEO collaborations.  If you use all or part
//      of it, please give an appropriate acknowledgement.
//
// Copyright Information: See EvtGen/COPYRIGHT
//      Copyright (C) 1998      Caltech, UCSB
//
// Module: EvtJetSet.cc
//
// Description: Routine to use JetSet for decaying particles.
//
// Modification history:
//
//    RYD     July 24, 1997        Module created
//
//------------------------------------------------------------------------
// 
#include "EvtGenBase/EvtPatches.hh"
#include "EvtGenBase/EvtPatches.hh"
#include "EvtGenBase/EvtParticle.hh"
#include "EvtGenBase/EvtStringParticle.hh"
#include "EvtGenBase/EvtDecayTable.hh"
#include "EvtGenBase/EvtPDL.hh"
#include "EvtGenModels/EvtJetSet.hh"
#include "EvtGenBase/EvtReport.hh"
#include <string>
#include "EvtGenBase/EvtId.hh"
#include <iostream>
#include <iomanip>
#include <fstream>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdio.h>
using std::endl;
using std::fstream;
using std::ios;
using std::ofstream;
using std::resetiosflags;
using std::setiosflags;
using std::setw;
 
 
int EvtJetSet::njetsetdecays=0;
  EvtDecayBasePtr* EvtJetSet::jetsetdecays=0; 
int EvtJetSet::ntable=0;
 
int EvtJetSet::ncommand=0;
int EvtJetSet::lcommand=0;
std::string* EvtJetSet::commands=0;
 
extern "C" {
  extern void evtjetsetinit_(char* fname, int len);
}
 
 
extern "C" {
  extern void jetset1_(int *,double *,int *,int *,int *,
               double *,double *,double *,double *);
}
 
extern "C" {
  extern void lugive_(const char *cnfgstr,int length);
}
 
extern "C" {
  extern int lucomp_(int* kf);
}
 
 
EvtJetSet::EvtJetSet(){}
 
EvtJetSet::~EvtJetSet(){
 
 
  int i;
 
 
  //the deletion of commands is really uggly!
 
  if (njetsetdecays==0) {
    delete [] commands;
    commands=0;
    return;
  }
 
  for(i=0;i<njetsetdecays;i++){
    if (jetsetdecays[i]==this){
      jetsetdecays[i]=jetsetdecays[njetsetdecays-1];
      njetsetdecays--;
      if (njetsetdecays==0) {
    delete [] commands;
    commands=0;
      }
      return;
    }
  }
 
  report(ERROR,"EvtGen") << "Error in destroying JetSet model!"<<endl;
 
}
 
 
void EvtJetSet::getName(std::string& model_name){
 
  model_name="JETSET";     
 
}
 
EvtDecayBase* EvtJetSet::clone(){
 
  return new EvtJetSet;
 
}
 
 
void EvtJetSet::initProbMax(){
 
  noProbMax();
 
}
 
 
void EvtJetSet::init(){
 
  checkNArg(1);
 
 
  if (getParentId().isAlias()){
 
    report(ERROR,"EvtGen") << "EvtJetSet finds that you are decaying the"<<endl
                           << " aliased particle "
               << EvtPDL::name(getParentId()).c_str()
               << " with the JetSet model"<<endl
               << " this does not work, please modify decay table."
               << endl;
    report(ERROR,"EvtGen") << "Will terminate execution!"<<endl;
    ::abort();
 
  }
 
  store(this);
 
}
 
 
std::string EvtJetSet::commandName(){
 
  return std::string("JetSetPar");
  
}
 
 
void EvtJetSet::command(std::string cmd){
 
  if (ncommand==lcommand){
 
    lcommand=10+2*lcommand;
 
    std::string* newcommands=new std::string[lcommand];
    
    int i;
 
    for(i=0;i<ncommand;i++){
      newcommands[i]=commands[i];
    }
    
    delete [] commands;
 
    commands=newcommands;
 
  }
 
  commands[ncommand]=cmd;
 
  ncommand++;
 
}
 
 
 
void EvtJetSet::decay( EvtParticle *p){
 
 
  //added by Lange Jan4,2000
  static EvtId STRNG=EvtPDL::getId("string");
 
  int istdheppar=EvtPDL::getStdHep(p->getId());
 
  if (lucomp_(&istdheppar)==0){
    report(ERROR,"EvtGen") << "Jetset can not decay:"
      <<EvtPDL::name(p->getId()).c_str()<<endl;
    return;
  }
 
  double mp=p->mass();
 
  EvtVector4R p4[20];
  
  int i,more;
  int ip=EvtPDL::getStdHep(p->getId());
  int ndaugjs;
  int kf[100];
  EvtId evtnumstable[100],evtnumparton[100];
  int stableindex[100],partonindex[100];
  int numstable;
  int numparton;
  int km[100];
  EvtId type[MAX_DAUG];
 
  jetSetInit();
 
  double px[100],py[100],pz[100],e[100];
 
  if ( p->getNDaug() != 0 ) { p->deleteDaughters(true);}
 
  int count=0;
 
  do{
    //report(INFO,"EvtGen") << "calling jetset " << ip<< " " << mp <<endl;
    jetset1_(&ip,&mp,&ndaugjs,kf,km,px,py,pz,e);
    
 
    numstable=0;
    numparton=0;
    //report(INFO,"EvtGen") << "found some daughters " << ndaugjs << endl;
    for(i=0;i<ndaugjs;i++){
 
      if (EvtPDL::evtIdFromStdHep(kf[i])==EvtId(-1,-1)) {
    report(ERROR,"EvtGen") << "JetSet returned particle:"<<kf[i]<<endl;
    report(ERROR,"EvtGen") << "This can not be translated to evt number"<<endl;
    report(ERROR,"EvtGen") << "and the decay will be rejected!"<<endl;
    report(ERROR,"EvtGen") << "The decay was of particle:"<<ip<<endl;
 
      }
 
      //sort out the partons
      if (abs(kf[i])<=6||kf[i]==21){
    partonindex[numparton]=i;
    evtnumparton[numparton]=EvtPDL::evtIdFromStdHep(kf[i]);
    numparton++;
      }
      else{
    stableindex[numstable]=i;
    evtnumstable[numstable]=EvtPDL::evtIdFromStdHep(kf[i]); 
    numstable++;
      }
 
 
      // have to protect against negative mass^2 for massless particles
      // i.e. neutrinos and photons.
      // this is uggly but I need to fix it right now....
 
      if (px[i]*px[i]+py[i]*py[i]+pz[i]*pz[i]>=e[i]*e[i]){
 
        e[i]=sqrt(px[i]*px[i]+py[i]*py[i]+pz[i]*pz[i])+0.0000000000001;
 
      }
 
      p4[i].set(e[i],px[i],py[i],pz[i]);
 
 
    }
 
    int channel=EvtDecayTable::inChannelList(p->getId(),numstable,evtnumstable);
 
 
   more=(channel!=-1);
 
   
 
 
  count++;
 
  }while( more && (count<10000) );
 
  if (count>9999) {
    report(INFO,"EvtGen") << "Too many loops in EvtJetSet!!!"<<endl;
    report(INFO,"EvtGen") << "Parent:"<<EvtPDL::name(getParentId()).c_str()<<endl;
    for(i=0;i<numstable;i++){
      report(INFO,"EvtGen") << "Daug("<<i<<")"<<EvtPDL::name(evtnumstable[i]).c_str()<<endl;
    }
 
  }
 
 
 
  if (numparton==0){
 
    p->makeDaughters(numstable,evtnumstable);
    int ndaugFound=0;
    for(i=0;i<numstable;i++){
      p->getDaug(i)->init(evtnumstable[i],p4[stableindex[i]]);
      ndaugFound++;
    }
    if ( ndaugFound == 0 ) {
      report(ERROR,"EvtGen") << "Jetset has failed to do a decay ";
      report(ERROR,"EvtGen") << EvtPDL::name(p->getId()).c_str() << " " << p->mass()<<endl;
      assert(0);
    }
 
    fixPolarizations(p);
 
    return ;
   
  }
  else{
 
    //have partons in JETSET
 
    EvtVector4R p4string(0.0,0.0,0.0,0.0);
 
    for(i=0;i<numparton;i++){
      p4string+=p4[partonindex[i]];
    }
 
    int nprimary=1;
    type[0]=STRNG;
    for(i=0;i<numstable;i++){
      if (km[stableindex[i]]==0){
    type[nprimary++]=evtnumstable[i];
      }
    }
 
    p->makeDaughters(nprimary,type);
 
    p->getDaug(0)->init(STRNG,p4string);
 
    EvtVector4R p4partons[10];
 
    for(i=0;i<numparton;i++){
      p4partons[i]=p4[partonindex[i]];
    }
 
    ((EvtStringParticle*)p->getDaug(0))->initPartons(numparton,p4partons,evtnumparton);
 
 
 
    nprimary=1;
 
    for(i=0;i<numstable;i++){
 
      if (km[stableindex[i]]==0){
    p->getDaug(nprimary++)->init(evtnumstable[i],p4[stableindex[i]]);
      }
    }
 
 
    int nsecond=0;
    for(i=0;i<numstable;i++){
      if (km[stableindex[i]]!=0){
    type[nsecond++]=evtnumstable[i];
      }
    }
 
 
    p->getDaug(0)->makeDaughters(nsecond,type);
 
    EvtVector4R p4stringboost(p4string.get(0),-p4string.get(1),
                  -p4string.get(2),-p4string.get(3));
 
    nsecond=0;
    for(i=0;i<numstable;i++){
      if (km[stableindex[i]]!=0){
    p4[stableindex[i]]=boostTo(p4[stableindex[i]],p4stringboost);
    p->getDaug(0)->getDaug(nsecond)->init(evtnumstable[i],p4[stableindex[i]]);
    p->getDaug(0)->getDaug(nsecond)->setDiagonalSpinDensity();
    p->getDaug(0)->getDaug(nsecond)->decay();
    nsecond++;
      }
    }
 
    if ( nsecond == 0 ) {
      report(ERROR,"EvtGen") << "Jetset has failed to do a decay ";
      report(ERROR,"EvtGen") << EvtPDL::name(p->getId()).c_str() << " " << p->mass() <<endl;
      assert(0);
    }
 
    fixPolarizations(p);
 
    return ;
 
  }
 
}
 
void EvtJetSet::fixPolarizations(EvtParticle *p){
 
  //special case for now to handle the J/psi polarization
 
  int ndaug=p->getNDaug();
  
  int i;
 
  static EvtId Jpsi=EvtPDL::getId("J/psi");
 
  for(i=0;i<ndaug;i++){
    if(p->getDaug(i)->getId()==Jpsi){
  
      EvtSpinDensity rho;
      
      rho.SetDim(3);
      rho.Set(0,0,0.5);
      rho.Set(0,1,0.0);
      rho.Set(0,2,0.0);
 
      rho.Set(1,0,0.0);
      rho.Set(1,1,1.0);
      rho.Set(1,2,0.0);
 
      rho.Set(2,0,0.0);
      rho.Set(2,1,0.0);
      rho.Set(2,2,0.5);
 
      EvtVector4R p4Psi=p->getDaug(i)->getP4();
 
      double alpha=atan2(p4Psi.get(2),p4Psi.get(1));
      double beta=acos(p4Psi.get(3)/p4Psi.d3mag());
 
 
      p->getDaug(i)->setSpinDensityForwardHelicityBasis(rho,alpha,beta,0.0);
      setDaughterSpinDensity(i);
 
    }
  }
 
}
 
void EvtJetSet::store(EvtDecayBase* jsdecay){
 
  if (njetsetdecays==ntable){
 
    EvtDecayBasePtr* newjetsetdecays=new EvtDecayBasePtr[2*ntable+10];
    int i;
    for(i=0;i<ntable;i++){
      newjetsetdecays[i]=jetsetdecays[i];
    }
    ntable=2*ntable+10;
    delete [] jetsetdecays;
    jetsetdecays=newjetsetdecays;
  }
 
  jetsetdecays[njetsetdecays++]=jsdecay;
 
 
 
}
 
 
void EvtJetSet::WriteJetSetEntryHeader(ofstream &outdec, int lundkc,
                   EvtId evtnum,std::string name,
                   int chg, int cchg, int spin2,double mass,
                   double width, double maxwidth,double ctau,
                   int stable,double rawbrfrsum){
 
 
  char sname[100];
 
  int namelength=8;
 
  int i,j;
  int temp;
  temp = spin2;
 
  if (ctau>1000000.0) ctau=0.0;
 
  strcpy(sname,name.c_str());
 
  i=0;
 
  while (sname[i]!=0){
    i++;
  }
 
  // strip up to two + or -
 
  if(evtnum.getId()>=0) {
    if (sname[i-1]=='+'||sname[i-1]=='-'){ 
      sname[i-1]=0;
      i--;
    }
    if (sname[i-1]=='+'||sname[i-1]=='-'){ 
      sname[i-1]=0;
      i--;
    }
    // strip 0 except for _0 and chi...0
    if (sname[i-1]=='0' && sname[i-2]!='_' && !(sname[0]=='c' && sname[1]=='h')){
      sname[i-1]=0;
      i--;
    }
  }
 
  if (i>namelength) {
    for(j=1;j<namelength;j++){
      sname[j]=sname[j+i-namelength];
    }
    sname[namelength]=0;
  }
 
  cchg=0;
 
  if(evtnum.getId()>=0) {
    if (abs(EvtPDL::getStdHep(evtnum))==21) cchg=2;
    if (abs(EvtPDL::getStdHep(evtnum))==90) cchg=-1;
    if ((abs(EvtPDL::getStdHep(evtnum))<=8)&&
    (abs(EvtPDL::getStdHep(evtnum))!=0)) cchg=1;
 
  }
 
  outdec << setw(5) << lundkc << "  ";
  outdec.width(namelength);
  outdec << setiosflags(ios::left) << sname << resetiosflags(ios::left);
  outdec << setw(3) << chg;
  outdec << setw(3) << cchg;
  outdec.width(3);
  if (evtnum.getId()>=0) {
    if (EvtPDL::chargeConj(evtnum)==evtnum) {
      outdec << 0;
    }
    else{
      outdec << 1;
    }
  }
  else{
    outdec << 0;
  }
  outdec.setf(ios::fixed);
  outdec.precision(5);
  outdec << setw(12) << mass;
  outdec << setw(12) << width;
  outdec.width(12);
  if (fabs(width)<0.0000000001) {
    outdec << 0.0 ;
  }
  else{
    outdec << maxwidth;
  }
  outdec << setw(14) << ctau;
  outdec.width(3);
  if (evtnum.getId()>=0) {
    if (ctau>1.0 || rawbrfrsum<0.000001) {  
      stable=0;
    }
  }
  outdec << stable;
  outdec << endl;
  outdec.width(0);
 
}
 
void EvtJetSet::WriteJetSetParticle(ofstream &outdec,EvtId ipar,
                    EvtId iparname,int &first){
 
  int ijetset;
 
  double br_sum=0.0;
 
  for(ijetset=0;ijetset<njetsetdecays;ijetset++){
   
    if (jetsetdecays[ijetset]->getParentId()==ipar){
      br_sum+=jetsetdecays[ijetset]->getBranchingFraction();
    }
    if (jetsetdecays[ijetset]->getParentId()!=
    EvtPDL::chargeConj(jetsetdecays[ijetset]->getParentId())&&
    EvtPDL::chargeConj(jetsetdecays[ijetset]->getParentId())==ipar){
      br_sum+=jetsetdecays[ijetset]->getBranchingFraction();
    }
 
 
  }
 
  double br_sum_true=br_sum;
 
  if (br_sum<0.000001) br_sum=1.0;
 
  for(ijetset=0;ijetset<njetsetdecays;ijetset++){
    if (jetsetdecays[ijetset]->getParentId()==ipar){
 
      double br=jetsetdecays[ijetset]->getBranchingFraction();
    
      int i,daugs[5];
      EvtId cdaugs[5];
    
      for(i=0;i<5;i++){
      
    if(i<jetsetdecays[ijetset]->getNDaug()){
      daugs[i]=EvtPDL::getStdHep(
             jetsetdecays[ijetset]->getDaugs()[i]);
      cdaugs[i]=EvtPDL::chargeConj(jetsetdecays[ijetset]->getDaugs()[i]);
    }
    else{
      daugs[i]=0;
    }
      }
 
      int channel;
 
      channel=EvtDecayTable::findChannel(EvtPDL::chargeConj(ipar),
                 jetsetdecays[ijetset]->getModelName(),
                 jetsetdecays[ijetset]->getNDaug(),
                 cdaugs,
                 jetsetdecays[ijetset]->getNArg(),
                 jetsetdecays[ijetset]->getArgsStr());     
 
      if (jetsetdecays[ijetset]->getModelName()=="JETSET"){
    
    if (first) {
      first=0;      
      WriteJetSetEntryHeader(outdec,
                 EvtPDL::getLundKC(iparname),
                 iparname,
                 EvtPDL::name(iparname), 
                 EvtPDL::chg3(iparname),
                 0,0,EvtPDL::getMeanMass(ipar),
                 EvtPDL::getWidth(ipar),
                 EvtPDL::getMeanMass(ipar)-EvtPDL::getMinMass(ipar),
                 EvtPDL::getctau(ipar),1,br_sum_true);
    }
    
    int dflag=2;
 
    if (EvtPDL::getStdHep(ipar)<0) {
      dflag=3;
      for(i=0;i<jetsetdecays[ijetset]->getNDaug();i++){
        daugs[i]=EvtPDL::getStdHep(cdaugs[i]);
      }
 
    }
 
    //now lets check to make sure that jetset, lucomp, knows
        //about all particles!
    int unknown=0;
    for(i=0;i<jetsetdecays[ijetset]->getNDaug();i++){
      if (lucomp_(&daugs[i])==0) {
        unknown=1;
        report(ERROR,"EvtGen") << "JetSet (lucomp) does not "
                  << "know the particle:"<<
          EvtPDL::name(jetsetdecays[ijetset]->getDaugs()[i]).c_str()<<endl;
      }
    }
 
    int istdheppar=EvtPDL::getStdHep(ipar);
 
    if (lucomp_(&istdheppar)==0){
      unknown=1;
      report(ERROR,"EvtGen") << "JetSet (lucomp) does not "
              << "know the particle:"<<
          EvtPDL::name(ipar).c_str()<<endl;
    }
 
 
 
    if (unknown){
      report(ERROR,"EvtGen") << "Therfore the decay:"<<endl;
      report(ERROR,"EvtGen") << EvtPDL::name(jetsetdecays[ijetset]->getParentId()).c_str()<<" -> ";
      for(i=0;i<jetsetdecays[ijetset]->getNDaug();i++){
        report(ERROR,"") << EvtPDL::name(jetsetdecays[ijetset]->getDaugs()[i]).c_str()<<" ";
      }
      report(ERROR,"")<<endl;
      report(ERROR,"EvtGen")<<"Will not be generated."<<endl;
      return;
    }
 
 
    if (EvtPDL::chargeConj(ipar)==ipar) {
      dflag=1;
      //report(INFO,"EvtGen") << EvtPDL::name(iparname) << " dflag=1 because C(ipar)=ipar!"<<endl;
    }
 
 
    //if (channel>=0) {
    //  dflag=1;
      //report(INFO,"EvtGen") << EvtPDL::name(iparname) << " dflag=1 because channel>=0"<<endl;
    //}
 
    //  if (!(EvtPDL::getStdHep(ipar)<0&&channel>=0)){
    if (1){
 
      outdec.width(10);
      outdec <<dflag;
      outdec.width(5);
      outdec <<(int)jetsetdecays[ijetset]->getArgs()[0];
      outdec.width(12);
      if (fabs(br)<0.000000001) {
        outdec <<"0.00000";
      }
      else{
        outdec <<br/br_sum;
      }
      outdec.width(8);
      outdec <<daugs[0];
      outdec.width(8);
      outdec <<daugs[1];
      outdec.width(8);
      outdec <<daugs[2];
      outdec.width(8);
      outdec <<daugs[3];
      outdec.width(8);
      outdec <<daugs[4];
      outdec<<endl;
      outdec.width(0);
    }
      }
    }
  }
}
 
 
void EvtJetSet::MakeJetSetFile(char* fname){
  
  EvtId ipar;
  int lundkc;
  
  //int part_list[MAX_PART];
  
  ofstream outdec;
 
  outdec.open(fname);
  
  //outdec << ";"<<endl;
  //outdec << ";This decayfile has been automatically created by"<<endl;
  //outdec << ";EvtGen from the DECAY.DEC file"<<endl;
  //outdec << ";"<<endl;
 
  int nokcentry;
 
  for(lundkc=1;lundkc<=500;lundkc++){
 
    nokcentry=1;
 
    int iipar;
 
    for(iipar=0;iipar<EvtPDL::entries();iipar++){
 
      ipar=EvtId(iipar,iipar);
      //no aliased particles!
      std::string tempStr = EvtPDL::name(ipar);
      EvtId realId = EvtPDL::getId(tempStr);
      if ( realId.isAlias() != 0 ) continue;
      if (lundkc==EvtPDL::getLundKC(ipar)){
        
        nokcentry=0;
 
    int first=1;
    
    WriteJetSetParticle(outdec,ipar,ipar,first);
 
    
    EvtId ipar2=EvtPDL::chargeConj(ipar);
 
 
        if (ipar2!=ipar){
      WriteJetSetParticle(outdec,ipar2,ipar,first);
    }
 
        if (first){
      WriteJetSetEntryHeader(outdec, 
                    EvtPDL::getLundKC(ipar),
                    ipar,
                    EvtPDL::name(ipar),
                    EvtPDL::chg3(ipar),
                    0,0,EvtPDL::getMeanMass(ipar),
                    EvtPDL::getWidth(ipar),
                    EvtPDL::getMeanMass(ipar)-EvtPDL::getMinMass(ipar),
                    EvtPDL::getctau(ipar),0,0.0);
 
    }
      }
    }
    if (nokcentry){
 
      WriteJetSetEntryHeader(outdec, 
                lundkc,EvtId(-1,-1),"  ",
                0,0,0,EvtPDL::getMeanMass(ipar),0.0,0.0,
                EvtPDL::getctau(ipar),0,0.0);
 
    }
  }
    outdec.close();
}
 
void EvtJetSet::jetSetInit(){
 
  static int first=1;
 
  if (first){
 
    first=0;
 
    report(INFO,"EvtGen") << "Will initialize JetSet."<<endl;
 
    char fname[200];
 
    char hostBuffer[100];
    
    if ( gethostname( hostBuffer, 100 ) != 0 ){
      report(ERROR,"EvtGen") << " couldn't get hostname." << endl;
      strncpy( hostBuffer, "hostnameNotFound", 100 );
    }
 
    char pid[100];
 
    int thePid=getpid();
 
    if ( sprintf( pid, "%d", thePid ) == 0 ){
      report(ERROR,"EvtGen") << " couldn't get process ID." << endl;
      strncpy( pid, "666", 100 );
    }
 
    strcpy(fname,"jet.d-");
    strcat(fname,hostBuffer);
    strcat(fname,"-");
    strcat(fname,pid);
    
    MakeJetSetFile(fname);
    evtjetsetinit_(fname,strlen(fname));
 
    if (0==getenv("EVTSAVEJETD")){
      char delcmd[300];
      strcpy(delcmd,"rm -f ");
      strcat(delcmd,fname);
      system(delcmd);
    }
 
    int i;
 
    for(i=0;i<ncommand;i++){
      lugive_(commands[i].c_str(),strlen(commands[i].c_str()));
 
    }
 
    report(INFO,"EvtGen") << "Done initializing JetSet."<<endl;
 
 
  }
 
}