EvtFSPick.cc

Go to the documentation of this file.

//--------------------------------------------------------------------------
//
// Environment:
//      This software is part of models developed at BES collaboration
//      based on the EvtGen framework.  If you use all or part
//      of it, please give an appropriate acknowledgement.
//
// Copyright Information: See EvtGen/BesCopyright
//      Copyright (A) 2006      Ping Rong-Gang @IHEP
//
// Module:  EvtDecayTag.hh
//
// Description:  Class to pick out the final state
//
// Modification history:
//
//    Ping R.-G.       December, 2011-Mar.       Module created
//
//------------------------------------------------------------------------
//
 
#include "EvtFSPick.hh"
 
void EvtFSPick::FSPick( EvtParticle* par ) {
 
  int   ndaug = par->getNDaug();
  EvtId eid   = par->getId();
  if ( ndaug < 2 && EvtPDL::getStdHep( eid ) != -22 )
  {
    _EvtFS.push_back( par );
    _fsid.push_back( eid );
    // std::cout<<"I get FS "<<EvtPDL::getStdHep(eid)<<std::endl;
  }
  else
  {
 
    int          ND = 0;
    EvtParticle* theD;
    for ( int i = 0; i < ndaug; i++ )
    {
      EvtParticle* theDaug = par->getDaug( i );
      FSPick( theDaug );
    }
  }
}
 
void EvtFSPick::setFinalStatePar( std::vector<std::string> vnames ) {
  if ( vnames.size() != _EvtFS.size() )
  {
    std::cout << "EvtFSPick::The number in final state is inconsistent" << std::endl;
    abort();
  }
  for ( int i = 0; i < vnames.size(); i++ )
  {
 
    _vnames.push_back( vnames[i] );
    EvtId eid = EvtPDL::getId( std::string( vnames[i] ) );
    _vId.push_back( eid );
 
    //   std::cout<<"FS= "<<EvtPDL::getStdHep(_vId[i])<<std::endl;
  }
}
 
std::vector<EvtVector4R> EvtFSPick::getP4() {
  EvtVector4R              ptep;
  int                      nfs = _vnames.size();
  std::vector<EvtVector4R> vp4;
  for ( int i = 0; i < nfs; i++ )
  {
    int xid = getIndex( _vId[i] );
    ptep    = _EvtFS[xid]->getP4();
    vp4.push_back( ptep );
  }
  return vp4;
}
 
std::vector<EvtVector4R> EvtFSPick::getP4Lab() {
  EvtVector4R              ptep;
  int                      nfs = _vnames.size();
  std::vector<EvtVector4R> vp4;
  for ( int i = 0; i < nfs; i++ )
  {
    int xid = getIndex( _vId[i] );
    ptep    = _EvtFS[xid]->getP4Lab();
    vp4.push_back( ptep );
  }
  return vp4;
}
 
int EvtFSPick::getIndex( EvtId eid ) {
  for ( int i = 0; i < _vId.size(); i++ )
  {
    EvtId pid = _fsid[i];
    if ( eid == pid )
    {
      EvtId did = EvtPDL::getId( "d" );
      _fsid[i]  = did;
      return i;
    }
  }
  std::cout << "EvtFSPick::Cannot find the particle " << EvtPDL::getStdHep( eid )
            << " in the generated final state" << std::endl;
  abort();
}