Kpipi0pi0.cxx

Go to the documentation of this file.

//
//  Kpipi0pi0.cxx is the single D0 tag code to reconstruct D0 or anti-D0 through the final
//  states of kpipi0pi0 from D0 decays. Kpipi0pi0.cxx was transfered from the Fortran routine
//  "Kpipi0pi0.f" which was orignally used for study of the D0D0-bar production and D0 decays
//  at the BES-II experiment during the time period from 2002 to 2008.
//
//  The orignal Fortran routine "Kpipi0pi0.f" used at the BES-II experiment was coded by H.L.
//  Ma and G. Rong in 2003.
//
//  Kpipi0pi0.cxx was transfered by G. Rong and J. Liu in December, 2005.
//
//  Since 2008, G. Rong and L.L. Jiang have been working on developing this code to analyze of
//  the data taken at 3.773 GeV with the BES-III detector at the BEPC-II collider.
//
//  During developing this code, many People made significant contributions to this code. These
//  are
//          G. Rong, L.L. Jiang, J. Liu, H.L. Ma, J.C. Chen, D.H. Zhang,
//          M.G. Zhao, B. Zheng, L. Li, Y. Fang, Z.Y. Yi, H.H. Liu, Z.Q. Liu et al.
//
//                                       By G. Rong and L.L. Jiang
//                                       March, 2009
//
//  ==========================================================================================
 
#include "SD0TagAlg/Kpipi0pi0.h"
#include "SD0TagAlg/SingleBase.h"
 
Kpipi0pi0::Kpipi0pi0() {}
 
Kpipi0pi0::~Kpipi0pi0() {}
 
void Kpipi0pi0::MTotal( double event, SmartDataPtr<EvtRecTrackCol> evtRecTrkCol, Vint iGood,
                        Vint iGam, double Ebeam, int PID_flag, int Charge_candidate_D ) {
 
  int nGood = iGood.size();
  int nGam  = iGam.size();
 
  iGoodtag.clear();
  iGamtag.clear();
 
  double           mass_bcgg, delE_tag_temp;
  int              m_chargetag, m_chargek, m_chargepi;
  int              ika_temp, ipi_temp, iGam1_temp, iGam2_temp, iGam3_temp, iGam4_temp;
  HepLorentzVector pddd;
  HepLorentzVector pddd_temp;
 
  int               cqtm_temp;
  IDataProviderSvc* eventSvc = NULL;
  Gaudi::svcLocator()->service( "EventDataSvc", eventSvc );
  SmartDataPtr<EvtRecEvent>        evtRecEvent( eventSvc, EventModel::EvtRec::EvtRecEvent );
  SmartDataPtr<Event::EventHeader> eventHeader( eventSvc, "/Event/EventHeader" );
 
  int runNo = eventHeader->runNumber();
  int rec   = eventHeader->eventNumber();
 
  double xecm = 2 * Ebeam;
 
  kpipi0pi0md    = false;
  double tagmode = 0;
 
  if ( ( evtRecEvent->totalCharged() < 2 || nGam < 4 ) ) { return; }
 
  double ecms = xecm;
 
  ISimplePIDSvc* simple_pid;
  Gaudi::svcLocator()->service( "SimplePIDSvc", simple_pid );
 
  double deltaE_tem = 0.20;
  int    ncount1    = 0;
 
  Hep3Vector    xorigin( 0, 0, 0 );
  IVertexDbSvc* vtxsvc;
  Gaudi::svcLocator()->service( "VertexDbSvc", vtxsvc );
  if ( vtxsvc->isVertexValid() )
  {
    double* dbv = vtxsvc->PrimaryVertex();
    double* vv  = vtxsvc->SigmaPrimaryVertex();
    xorigin.setX( dbv[0] );
    xorigin.setY( dbv[1] );
    xorigin.setZ( dbv[2] );
  }
 
  double xv = xorigin.x();
  double yv = xorigin.y();
  double zv = xorigin.z();
 
  HepPoint3D point0( 0., 0., 0. );
  HepPoint3D IP( xorigin[0], xorigin[1], xorigin[2] );
  //////////////////////////////////////////////////////////////////
  HepLorentzVector p2gfit1, p2gfit2;
  HepLorentzVector p2gg;
 
  HepLorentzVector ptrk1_temp, ptrk2_temp, ptrk3_temp, ptrk4_temp, ptrk5_temp, ptrk6_temp,
      ptrk7_temp, ptrk8_temp;
  for ( int i = 0; i < evtRecEvent->totalCharged(); i++ )
  {
    EvtRecTrackIterator itTrk = evtRecTrkCol->begin() + i;
 
    int ika = ( *itTrk )->trackId();
 
    if ( !( *itTrk )->isMdcKalTrackValid() ) continue;
    RecMdcKalTrack* mdcKalTrk1 = ( *itTrk )->mdcKalTrack();
    RecMdcKalTrack::setPidType( RecMdcKalTrack::kaon );
    /////////////////////////////////////////
    m_chargek = mdcKalTrk1->charge();
    if ( Charge_candidate_D != 0 )
    {
      if ( m_chargek != -Charge_candidate_D ) continue;
    }
    if ( Charge_candidate_D == 0 )
    {
      if ( abs( m_chargek ) != 1 ) continue;
    }
    /////////////////////////////////////////
    HepVector    a1  = mdcKalTrk1->getZHelixK();
    HepSymMatrix Ea1 = mdcKalTrk1->getZErrorK();
    VFHelix      helixip3_1( point0, a1, Ea1 );
    helixip3_1.pivot( IP );
    HepVector vecipa1 = helixip3_1.a();
 
    double dr1       = fabs( vecipa1[0] );
    double dz1       = fabs( vecipa1[3] );
    double costheta1 = cos( mdcKalTrk1->theta() );
    if ( dr1 >= 1.0 ) continue;
    if ( dz1 >= 10.0 ) continue;
    if ( fabs( costheta1 ) >= 0.93 ) continue;
    /////////////////////////////////////////
    if ( PID_flag == 5 )
    {
      simple_pid->preparePID( *itTrk );
      if ( simple_pid->probKaon() < 0.0 || simple_pid->probKaon() < simple_pid->probPion() )
        continue;
    }
    /////////////////////////////////////////
 
    WTrackParameter kam( xmass[3], mdcKalTrk1->getZHelixK(), mdcKalTrk1->getZErrorK() );
 
    //
    // select pi
    //
    for ( int j = 0; j < evtRecEvent->totalCharged(); j++ )
    {
      EvtRecTrackIterator itTrk = evtRecTrkCol->begin() + j;
 
      int ipi = ( *itTrk )->trackId();
      if ( ipi == ika ) continue;
 
      if ( !( *itTrk )->isMdcKalTrackValid() ) continue;
      RecMdcKalTrack* mdcKalTrk2 = ( *itTrk )->mdcKalTrack();
      RecMdcKalTrack::setPidType( RecMdcKalTrack::pion );
      /////////////////////////////////////////
      m_chargepi = mdcKalTrk2->charge();
      if ( ( m_chargek + m_chargepi ) != 0 ) continue;
      /////////////////////////////////////////
      HepVector    a2  = mdcKalTrk2->getZHelix();
      HepSymMatrix Ea2 = mdcKalTrk2->getZError();
      VFHelix      helixip3_2( point0, a2, Ea2 );
      helixip3_2.pivot( IP );
      HepVector vecipa2 = helixip3_2.a();
 
      double dr2       = fabs( vecipa2[0] );
      double dz2       = fabs( vecipa2[3] );
      double costheta2 = cos( mdcKalTrk2->theta() );
      if ( dr2 >= 1.0 ) continue;
      if ( dz2 >= 10.0 ) continue;
      if ( fabs( costheta2 ) >= 0.93 ) continue;
      /////////////////////////////////////////
      if ( PID_flag == 5 )
      {
        simple_pid->preparePID( *itTrk );
        if ( simple_pid->probPion() < 0.0 || simple_pid->probPion() < simple_pid->probKaon() )
          continue;
      }
      /////////////////////////////////////////
 
      WTrackParameter pip( xmass[2], mdcKalTrk2->getZHelix(), mdcKalTrk2->getZError() );
 
      for ( int m = 0; m < nGam - 1; m++ )
      {
        if ( iGam[m] == -1 ) continue;
        RecEmcShower*    g1Trk = ( *( evtRecTrkCol->begin() + iGam[m] ) )->emcShower();
        double           eraw1 = g1Trk->energy();
        double           phi1  = g1Trk->phi();
        double           the1  = g1Trk->theta();
        HepLorentzVector ptrkg1, ptrkg10, ptrkg12;
        ptrkg1.setPx( eraw1 * sin( the1 ) * cos( phi1 ) );
        ptrkg1.setPy( eraw1 * sin( the1 ) * sin( phi1 ) );
        ptrkg1.setPz( eraw1 * cos( the1 ) );
        ptrkg1.setE( eraw1 );
        ptrkg10 = ptrkg1;
        ptrkg12 = ptrkg1.boost( -0.011, 0, 0 );
 
        for ( int n = m + 1; n < nGam; n++ )
        {
          if ( iGam[n] == -1 ) continue;
          RecEmcShower*    g2Trk = ( *( evtRecTrkCol->begin() + iGam[n] ) )->emcShower();
          double           eraw2 = g2Trk->energy();
          double           phi2  = g2Trk->phi();
          double           the2  = g2Trk->theta();
          HepLorentzVector ptrkg2, ptrkg20, ptrkg22;
          ptrkg2.setPx( eraw2 * sin( the2 ) * cos( phi2 ) );
          ptrkg2.setPy( eraw2 * sin( the2 ) * sin( phi2 ) );
          ptrkg2.setPz( eraw2 * cos( the2 ) );
          ptrkg2.setE( eraw2 );
          ptrkg20 = ptrkg2;
          ptrkg22 = ptrkg2.boost( -0.011, 0, 0 );
 
          /////////////////////////////////////////////////////////////
          HepLorentzVector ptrkpi0;
          ptrkpi0            = ptrkg12 + ptrkg22;
          double m_xmpi0_tem = ptrkpi0.mag();
          if ( m_xmpi0_tem > 0.150 || m_xmpi0_tem < 0.115 ) continue;
          /////////////////////////////////////////////////////////////
          bool IsEndcap1 = false;
          bool IsEndcap2 = false;
          if ( fabs( cos( the1 ) ) > 0.86 && fabs( cos( the1 ) ) < 0.92 ) IsEndcap1 = true;
          if ( fabs( cos( the2 ) ) > 0.86 && fabs( cos( the2 ) ) < 0.92 ) IsEndcap2 = true;
          if ( IsEndcap1 && IsEndcap2 ) continue;
          /////////////////////////////////////////////////////////////
 
          KalmanKinematicFit* kmfit = KalmanKinematicFit::instance();
          kmfit->init();
          kmfit->setChisqCut( 2500 );
          kmfit->AddTrack( 0, 0.0, g1Trk );
          kmfit->AddTrack( 1, 0.0, g2Trk );
          kmfit->AddResonance( 0, mpi0, 0, 1 );
          kmfit->Fit( 0 ); // Perform fit
          kmfit->BuildVirtualParticle( 0 );
 
          double pi0_chisq = kmfit->chisq( 0 );
          if ( pi0_chisq >= 2500 ) continue;
          HepLorentzVector p2gfit1 = kmfit->pfit( 0 ) + kmfit->pfit( 1 );
          p2gfit1.boost( -0.011, 0, 0 );
 
          for ( int s = 0; s < nGam - 1; s++ )
          {
            if ( iGam[s] == -1 ) continue;
            RecEmcShower* g3Trk = ( *( evtRecTrkCol->begin() + iGam[s] ) )->emcShower();
            if ( iGam[s] == iGam[m] || iGam[s] == iGam[n] ) continue;
            double           eraw3 = g3Trk->energy();
            double           phi3  = g3Trk->phi();
            double           the3  = g3Trk->theta();
            HepLorentzVector ptrkg3, ptrkg30, ptrkg32;
            ptrkg3.setPx( eraw3 * sin( the3 ) * cos( phi3 ) );
            ptrkg3.setPy( eraw3 * sin( the3 ) * sin( phi3 ) );
            ptrkg3.setPz( eraw3 * cos( the3 ) );
            ptrkg3.setE( eraw3 );
            ptrkg30 = ptrkg3;
            ptrkg32 = ptrkg3.boost( -0.011, 0, 0 );
 
            for ( int r = s + 1; r < nGam; r++ )
            {
              if ( iGam[r] == -1 ) continue;
              RecEmcShower* g4Trk = ( *( evtRecTrkCol->begin() + iGam[r] ) )->emcShower();
              if ( iGam[r] == iGam[m] || iGam[r] == iGam[n] ) continue;
              double           eraw4 = g4Trk->energy();
              double           phi4  = g4Trk->phi();
              double           the4  = g4Trk->theta();
              HepLorentzVector ptrkg4, ptrkg40, ptrkg42;
              ptrkg4.setPx( eraw4 * sin( the4 ) * cos( phi4 ) );
              ptrkg4.setPy( eraw4 * sin( the4 ) * sin( phi4 ) );
              ptrkg4.setPz( eraw4 * cos( the4 ) );
              ptrkg4.setE( eraw4 );
              ptrkg40 = ptrkg4;
              ptrkg42 = ptrkg4.boost( -0.011, 0, 0 );
 
              /////////////////////////////////////////////////////////////
              HepLorentzVector ptrkpi0_2;
              ptrkpi0_2            = ptrkg32 + ptrkg42;
              double m_xmpi0_2_tem = ptrkpi0_2.mag();
              if ( m_xmpi0_2_tem > 0.150 || m_xmpi0_2_tem < 0.115 ) continue;
              /////////////////////////////////////////////////////////////
              bool IsEndcap3 = false;
              bool IsEndcap4 = false;
              if ( fabs( cos( the3 ) ) > 0.86 && fabs( cos( the3 ) ) < 0.92 ) IsEndcap3 = true;
              if ( fabs( cos( the4 ) ) > 0.86 && fabs( cos( the4 ) ) < 0.92 ) IsEndcap4 = true;
              if ( IsEndcap3 && IsEndcap4 ) continue;
              /////////////////////////////////////////////////////////////
 
              KalmanKinematicFit* kmfit2 = KalmanKinematicFit::instance();
              kmfit2->init();
              kmfit2->setChisqCut( 2500 );
              kmfit2->AddTrack( 0, 0.0, g3Trk );
              kmfit2->AddTrack( 1, 0.0, g4Trk );
              kmfit2->AddResonance( 0, mpi0, 0, 1 );
 
              kmfit2->Fit( 0 ); // Perform fit
              kmfit2->BuildVirtualParticle( 0 );
 
              double pi0_2_chisq = kmfit2->chisq( 0 );
              if ( pi0_2_chisq >= 2500 ) continue;
              HepLorentzVector p2gfit2 = kmfit2->pfit( 0 ) + kmfit2->pfit( 1 );
              p2gfit2.boost( -0.011, 0, 0 );
 
              //////////////////////////////////////////////////////////////
              HepPoint3D   vx( xorigin.x(), xorigin.y(), xorigin.z() );
              HepSymMatrix Evx( 3, 0 );
              double       bx = 1E+6;
              Evx[0][0]       = bx * bx;
              double by       = 1E+6;
              Evx[1][1]       = by * by;
              double bz       = 1E+6;
              Evx[2][2]       = bz * bz;
              VertexParameter vxpar;
              vxpar.setVx( vx );
              vxpar.setEvx( Evx );
              //////////////////////////////////////////////////////////////
 
              VertexFit* vtxfit = VertexFit::instance();
              vtxfit->init();
              vtxfit->AddTrack( 0, kam );
              vtxfit->AddTrack( 1, pip );
              vtxfit->AddVertex( 0, vxpar, 0, 1 );
              if ( !vtxfit->Fit( 0 ) ) continue;
              vtxfit->Swim( 0 );
 
              WTrackParameter wkam = vtxfit->wtrk( 0 );
              WTrackParameter wpip = vtxfit->wtrk( 1 );
 
              HepVector kam_val = HepVector( 7, 0 );
              kam_val           = wkam.w();
              HepVector pip_val = HepVector( 7, 0 );
              pip_val           = wpip.w();
 
              HepLorentzVector P_KAM( kam_val[0], kam_val[1], kam_val[2], kam_val[3] );
              HepLorentzVector P_PIP( pip_val[0], pip_val[1], pip_val[2], pip_val[3] );
 
              P_KAM.boost( -0.011, 0, 0 );
              P_PIP.boost( -0.011, 0, 0 );
              pddd = P_KAM + P_PIP + p2gfit1 + p2gfit2;
 
              double pkpipi0pi0 = pddd.rho();
 
              double temp1 = ( ecms / 2 ) * ( ecms / 2 ) - pkpipi0pi0 * pkpipi0pi0;
              if ( temp1 < 0 ) temp1 = 0;
              double mass_bc_tem = sqrt( temp1 );
              if ( mass_bc_tem < 1.82 || mass_bc_tem > 1.89 ) continue;
 
              double delE_tag_tag = ecms / 2 - pddd.e();
 
              if ( fabs( delE_tag_tag ) < deltaE_tem )
              {
                deltaE_tem    = fabs( delE_tag_tag );
                delE_tag_temp = delE_tag_tag;
                mass_bcgg     = mass_bc_tem;
 
                pddd_temp = pddd;
                cqtm_temp = m_chargek;
 
                ika_temp = ika;
                ipi_temp = ipi;
 
                iGam1_temp = iGam[m];
                iGam2_temp = iGam[n];
                iGam3_temp = iGam[s];
                iGam4_temp = iGam[r];
                ncount1    = 1;
              }
            }
          }
        }
      }
    }
  }
 
  if ( ncount1 == 1 )
  {
    tagmode = 25;
    if ( cqtm_temp < 0 ) tagmode = -25;
    tagmd    = tagmode;
    mass_bc  = mass_bcgg;
    delE_tag = delE_tag_temp;
 
    cqtm = -1.0 * cqtm_temp;
 
    iGoodtag.push_back( ipi_temp );
    iGoodtag.push_back( ika_temp );
 
    iGamtag.push_back( iGam1_temp );
    iGamtag.push_back( iGam2_temp );
    iGamtag.push_back( iGam3_temp );
    iGamtag.push_back( iGam4_temp );
 
    ptag = pddd_temp;
 
    kpipi0pi0md = true;
  }
}