EvtParticleDecayList.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: EvtDecayParm.cc
//
// Description: Store decay parameters for one decay.
//
// Modification history:
//
//    RYD     April 5, 1997         Module created
//
//------------------------------------------------------------------------
//
#include "EvtGenBase/EvtPatches.hh"
#include "EvtGenBase/EvtPatches.hh"
#include <iostream>
#include <fstream>
#include <stdlib.h>
#include <ctype.h>
#include "EvtGenBase/EvtParticleDecayList.hh"
#include "EvtGenBase/EvtParticle.hh"
#include "EvtGenBase/EvtRandom.hh"
#include "EvtGenBase/EvtReport.hh"
#include "EvtGenBase/EvtPDL.hh"
#include "EvtGenBase/EvtStatus.hh"
using std::endl;
using std::fstream;
 
EvtParticleDecayList::EvtParticleDecayList(const EvtParticleDecayList &o) {
  _nmode=o._nmode;
  _rawbrfrsum=o._rawbrfrsum;
  _decaylist=new EvtParticleDecayPtr[_nmode];
 
  int i;
  for(i=0;i<_nmode;i++){
    _decaylist[i]=new EvtParticleDecay;  
 
    EvtDecayBase *tModel=o._decaylist[i]->getDecayModel();
    
    EvtDecayBase *tModelNew=tModel->clone();
    if (tModel->getPHOTOS()){
      tModelNew->setPHOTOS();
    }
    if (tModel->verbose()){
      tModelNew->setVerbose();
    }
    if (tModel->summary()){
      tModelNew->setSummary();
    }
    std::vector<std::string> args;
    int j;
    for(j=0;j<tModel->getNArg();j++){
      args.push_back(tModel->getArgStr(j));
    }
    tModelNew->saveDecayInfo(tModel->getParentId(),tModel->getNDaug(),
                 tModel->getDaugs(),
                 tModel->getNArg(),
                 args,
                 tModel->getModelName(),
                 tModel->getBranchingFraction());
    _decaylist[i]->setDecayModel(tModelNew);
    
    //_decaylist[i]->setDecayModel(tModel);
    _decaylist[i]->setBrfrSum(o._decaylist[i]->getBrfrSum());
    _decaylist[i]->setMassMin(o._decaylist[i]->getMassMin());
  }
 
 
}
 
 
EvtParticleDecayList::~EvtParticleDecayList(){
 
  int i;
  for(i=0;i<_nmode;i++){
    delete _decaylist[i];
  }
  
  if (_decaylist!=0) delete [] _decaylist;
}
 
void EvtParticleDecayList::printSummary(){
 
  int i;
  for(i=0;i<_nmode;i++){
    _decaylist[i]->printSummary();
  }
  
}
 
void EvtParticleDecayList::removeDecay(){
  
  int i;
  for(i=0;i<_nmode;i++){
    delete _decaylist[i];
  }
  
  delete [] _decaylist; 
  _decaylist=0; 
  _nmode=0; 
  _rawbrfrsum=0.0;
  
}
 
 
EvtDecayBase* EvtParticleDecayList::getDecayModel(EvtParticle *p){
 
  if (p->getNDaug()!=0) {
    assert(p->getChannel()>=0);
    return getDecay(p->getChannel()).getDecayModel();
  }
  if (p->getChannel() >(-1) ) {
    return getDecay(p->getChannel()).getDecayModel();
  }
 
 
  if (getNMode()==0 ) {
    return 0;
  }
  if (getRawBrfrSum()<0.00000001 ) {
    return 0;
  }
 
  if (getNMode()==1) {
    p->setChannel(0);
    return getDecay(0).getDecayModel();
  }
  
  if (p->getChannel() >(-1)) {
    report(ERROR,"EvtGen") << "Internal error!!!"<<endl;
    ::abort();
  }
 
  int j;
 
  for (j=0;j<1000;j++){
 
    double u=EvtRandom::Flat();
 
    int i;
    bool breakL=false;
    for (i=0;i<getNMode();i++) {
 
      if ( breakL ) continue;
      if (u<getDecay(i).getBrfrSum()) {
    breakL=true;
    //special case for decay of on particel to another
    // e.g. K0->K0S
 
    if (getDecay(i).getDecayModel()->getNDaug()==1 ) {
      p->setChannel(i);
      return getDecay(i).getDecayModel(); 
    } 
 
    if ( p->hasValidP4() ) {
      //report(INFO,"EvtGen") << "amazing " << EvtPDL::name(p->getId()) << " " << getDecay(i).getMassMin() << " "<<p->mass() << " " << i << endl;
      if (getDecay(i).getMassMin() < p->mass() ) {
        p->setChannel(i);
        return getDecay(i).getDecayModel(); 
      }
    }
    else{
    //Lange apr29-2002 - dont know the mass yet
        p->setChannel(i);
        return getDecay(i).getDecayModel(); 
    }
      }
    }
  }
 
  //Ok, we tried 1000 times above to pick a decay channel that is
  //kinematically allowed! Now we give up and search all channels!
  //if that fails, the particle will not be decayed!
  
  int i;
 
  for (i=0;i<getNMode();i++) {
 
    if ( getDecay(i).getMassMin() < p->mass() ) {
      return getDecay(i).getDecayModel(); 
    }
  }
 
  report(ERROR,"EvtGen") << "Could not decay:"
             <<EvtPDL::name(p->getId()).c_str()
             <<" with mass:"<<p->mass()
             <<" will throw event away! "<<endl;
  
  //  report(ERROR,"EvtGen") << "Will terminate execution."<<endl;
 
  //  ::abort();  
  EvtStatus::setRejectFlag();
  return 0;
 
}
 
 
void EvtParticleDecayList::setNMode(int nmode){
 
  EvtParticleDecayPtr* _decaylist_new= new EvtParticleDecayPtr[nmode];
 
  if (_nmode!=0){
    report(ERROR,"EvtGen") << "Error _nmode not equal to zero!!!"<<endl;
    ::abort();
    delete [] _decaylist;
  }
  _decaylist=_decaylist_new;
  _nmode=nmode;
 
}
 
 
EvtParticleDecay& EvtParticleDecayList::getDecay(int nchannel) {
  if (nchannel>=_nmode) {
    report(ERROR,"EvtGen") <<"Error getting channel:"
               <<nchannel<<" with only "<<_nmode
               <<" stored!"<<endl;
    ::abort();
  }
  return *(_decaylist[nchannel]);
}
 
void EvtParticleDecayList::makeChargeConj(EvtParticleDecayList* conjDecayList){
 
  _rawbrfrsum=conjDecayList->_rawbrfrsum;
 
  setNMode(conjDecayList->_nmode);
  
  int i;
 
  for(i=0;i<_nmode;i++){
    _decaylist[i]=new EvtParticleDecay;
    _decaylist[i]->chargeConj(conjDecayList->_decaylist[i]);
  }
 
}
 
void EvtParticleDecayList::addMode(EvtDecayBase* decay, double brfrsum,
                   double massmin){
 
  EvtParticleDecayPtr* newlist=new EvtParticleDecayPtr[_nmode+1];
 
  int i;
  for(i=0;i<_nmode;i++){
    newlist[i]=_decaylist[i];
  }
 
  _rawbrfrsum=brfrsum;
 
  newlist[_nmode]=new EvtParticleDecay;  
 
  newlist[_nmode]->setDecayModel(decay);
  newlist[_nmode]->setBrfrSum(brfrsum);
  newlist[_nmode]->setMassMin(massmin);
 
  EvtDecayBase *newDec=newlist[_nmode]->getDecayModel();
  for(i=0;i<_nmode;i++){
    if ( newDec->matchingDecay(*(newlist[i]->getDecayModel())) ) {
 
      //sometimes its ok..
      if ( newDec->getModelName() == "JETSET" || newDec->getModelName() == "PYTHIA" ) continue;
      if ( newDec->getModelName() == "JSCONT" || newDec->getModelName() == "PYCONT" ) continue;
      if ( newDec->getModelName() == "PYGAGA"  ) continue;
      report(ERROR,"EvtGen") << "Two matching decays with same parent in decay table\n";
      report(ERROR,"EvtGen") << "Please fix that\n";
      report(ERROR,"EvtGen") << "Parent " << EvtPDL::name(newDec->getParentId()).c_str() << endl;
      for (int j=0; j<newDec->getNDaug(); j++)
    report(ERROR,"EvtGen") << "Daughter " << EvtPDL::name(newDec->getDaug(j)).c_str() << endl;
      assert(0);
    }
  }
 
  if (_nmode!=0){
    delete [] _decaylist;
  }
 
  _nmode++;
 
  _decaylist=newlist;
 
}
 
 
void EvtParticleDecayList::finalize(){
 
  if (_nmode>0) {
    if ( _rawbrfrsum< 0.000001 ) {
      report(ERROR,"EvtGen") << "Please give me a "
                 <<    "branching fraction sum greater than 0\n";
      assert(0);
    }
    if (fabs(_rawbrfrsum-1.0)>0.0001) {
      report(INFO,"EvtGen") <<"Warning, sum of branching fractions for "
                    <<EvtPDL::name(_decaylist[0]->getDecayModel()->getParentId()).c_str()
                    <<" is "<<_rawbrfrsum<<endl;
      report(INFO,"EvtGen") << "rescaled to one! "<<endl;
      
    }
 
    int i;
 
    for(i=0;i<_nmode;i++){
      double brfrsum=_decaylist[i]->getBrfrSum()/_rawbrfrsum;
      _decaylist[i]->setBrfrSum(brfrsum);      
    }
 
  }
 
}
void EvtParticleDecayList::removeMode(EvtDecayBase* decay) {
   // here we will delete a decay with the same final state particles
   // and recalculate the branching fractions for the remaining modes
   int match = -1;
   int i;
   double match_bf;
 
   for(i=0;i<_nmode;i++){
      if ( decay->matchingDecay(*(_decaylist[i]->getDecayModel())) ) {
       match = i;
      }
   }
 
   if (match < 0) {
      report(ERROR,"EvtGen") << " Attempt to remove undefined mode for" << endl
                           << "Parent " << EvtPDL::name(decay->getParentId()).c_str() << endl
                           << "Daughters: ";
      for (int j=0; j<decay->getNDaug(); j++)
      report(ERROR,"") << EvtPDL::name(decay->getDaug(j)).c_str() << " ";
      report(ERROR,"") << endl;
      ::abort();
   }
 
   if (match == 0) {
      match_bf = _decaylist[match]->getBrfrSum();
   } else {
      match_bf = (_decaylist[match]->getBrfrSum()
                -_decaylist[match-1]->getBrfrSum());
   }
 
   double divisor = 1-match_bf;
   if (divisor < 0.000001 && _nmode > 1) {
      report(ERROR,"EvtGen") << "Removing requested mode leaves "
                           <<  EvtPDL::name(decay->getParentId()).c_str()
                           << " with zero sum branching fraction," << endl
                           << "but more than one decay mode remains. Aborting."
                           << endl;
      ::abort();
   }
 
   EvtParticleDecayPtr* newlist=new EvtParticleDecayPtr[_nmode-1];
 
   for(i=0;i<match;i++){
      newlist[i]=_decaylist[i];
      newlist[i]->setBrfrSum(newlist[i]->getBrfrSum()/divisor);
   }
   for(i=match+1; i<_nmode; i++) {
      newlist[i-1]=_decaylist[i];
      newlist[i-1]->setBrfrSum((newlist[i-1]->getBrfrSum()-match_bf)/divisor);
   }
 
 
   delete [] _decaylist;
 
  _nmode--;
 
  _decaylist=newlist;
 
  if (_nmode == 0) {
     delete [] _decaylist;
  }
 
}