EvtMassH2.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: EvtMassH1.cc
//
// Description: Routine to decay a particle  using a Dalitz
//    plot.
//
// Modification history:
//
//    Ping R.-G.       December, 2006       Module created
//
//------------------------------------------------------------------------
//
#include "EvtMassH2.hh"
#include "../EvtGenBase/EvtDiracParticle.hh"
#include "../EvtGenBase/EvtGenKine.hh"
#include "../EvtGenBase/EvtHelSys.hh"
#include "../EvtGenBase/EvtHighSpinParticle.hh"
#include "../EvtGenBase/EvtNeutrinoParticle.hh"
#include "../EvtGenBase/EvtPDL.hh"
#include "../EvtGenBase/EvtParticle.hh"
#include "../EvtGenBase/EvtPatches.hh"
#include "../EvtGenBase/EvtPhotonParticle.hh"
#include "../EvtGenBase/EvtRandom.hh"
#include "../EvtGenBase/EvtRaritaSchwingerParticle.hh"
#include "../EvtGenBase/EvtReport.hh"
#include "../EvtGenBase/EvtScalarParticle.hh"
#include "../EvtGenBase/EvtStringParticle.hh"
#include "../EvtGenBase/EvtTensor4C.hh"
#include "../EvtGenBase/EvtTensorParticle.hh"
#include "../EvtGenBase/EvtVector4C.hh"
#include "../EvtGenBase/EvtVector4R.hh"
#include "../EvtGenBase/EvtVectorParticle.hh"
#include <stdlib.h>
#include <string>
 
#include "TApplication.h"
#include "TAxis.h"
#include "TFile.h"
#include "TH1.h"
#include "TH2.h"
#include "TROOT.h"
// #include "CLHEP/config/CLHEP.h"
// #include "CLHEP/config/TemplateFunctions.h"
 
using std::endl;
 
EvtMassH2::~EvtMassH2() {}
 
void EvtMassH2::getName( std::string& model_name ) { model_name = "MassH2"; }
 
EvtDecayBase* EvtMassH2::clone() { return new EvtMassH2; }
 
void EvtMassH2::init() {
 
  // check that there are 4 arguments: Invariant mass part. Index: i,j, histor. file name, Hid
  checkNArg( 0 );
  EvtSpinType::spintype parenttype = EvtPDL::getSpinType( getParentId() );
}
void EvtMassH2::initProbMax() { noProbMax(); }
 
void EvtMassH2::decay( EvtParticle* p ) {
 
  const char* fl = setFileName();
  const char* hp = setHpoint();
  int*        dp;
  dp     = setDaugPair();
  int d1 = dp[0]; // m(d1,d2) pair at X axis
  int d2 = dp[1];
  int d3 = dp[2]; // m(d3,d4) pair at Y axis
  int d4 = dp[3];
 
  TFile  f( fl );
  TH2F*  hid   = (TH2F*)f.Get( hp );
  TAxis* xaxis = hid->GetXaxis();
  TAxis* yaxis = hid->GetYaxis();
 
  int    BINSx = xaxis->GetLast();
  int    BINSy = yaxis->GetLast();
  int    BINS  = BINSx * BINSy;
  double yvalue, ymax = 0.0;
  int    i, j, binxy;
 
  for ( i = 1; i < BINSx + 1; i++ )
  {
    for ( j = 1; j < BINSy + 1; j++ )
    {
      binxy  = hid->GetBin( i, j );
      yvalue = hid->GetBinContent( binxy );
      //      cout<<"binxy,yvalue = "<<binxy<<"; "<<yvalue<<endl;
      if ( yvalue > ymax ) ymax = yvalue;
    }
  }
 
loop:
  p->initializePhaseSpace( getNDaug(), getDaugs() );
 
  EvtParticle *id1, *id2, *id3, *id4, *s1;
  EvtVector4R  pd1, pd2, pd3, pd4, ps;
  double       xmass2, ymass2;
 
  id1 = p->getDaug( d1 );
  id2 = p->getDaug( d2 );
  id3 = p->getDaug( d3 );
  id4 = p->getDaug( d4 );
 
  pd1 = id1->getP4Lab();
  pd2 = id2->getP4Lab();
  pd3 = id3->getP4Lab();
  pd4 = id4->getP4Lab();
 
  xmass2 = ( pd1 + pd2 ).mass2();
  ymass2 = ( pd3 + pd4 ).mass2();
 
  int    xbin   = hid->GetXaxis()->FindBin( xmass2 );
  int    ybin   = hid->GetYaxis()->FindBin( ymass2 );
  int    xybin  = hid->GetBin( xbin, ybin );
  double zvalue = hid->GetBinContent( xybin );
  double xratio = zvalue / ymax;
  //  cout<<"***************zvalue,ymax,xratio= "<<zvalue<<"; "<<ymax<<"; "<<xratio<<endl;
  double rd1 = EvtRandom::Flat( 0.0, 1.0 );
  if ( rd1 > xratio ) goto loop;
  return;
}