Reconstruction/TrkReco/test/Test.cxx

Go to the documentation of this file.

#include "math.h"
#include "stdio.h"
 
#include "AsciiDmp/AsciiData.h"
#include "AsciiDmp/dmplib.h"
 
#include <fstream>
#include <iostream>
#include <vector>
 
#include "TrkReco/Bfield.h"
#include "TrkReco/Helix.h"
#include "TrkReco/TUpdater.h"
#include "TrkReco/TrkReco.h"
// #include "BesKernel/BesEvent.h"
#include "MdcRecGeo/MdcRecGeo.h"
#include "MdcTables/MdcTables.h"
 
int main() {
 
  TrkReco* _trkreco = TrkReco::getTrkReco();
 
  _trkreco->init( 0 );
 
  Bfield* fBfield = Bfield::getBfield( 5 );
  //   cout<<"Bfield:"<<endl;
  //   cout<<"bz(0.1,0.1,0.5):"<<fBfield->bz(0.1,0.1,0.5)<<endl;
  delete fBfield;
 
  HepPoint3D fPivot( 0, 0, 0 );
  Hep3Vector fmomentum( 1, 1, 1 );
  Helix      fHelix( fPivot, fmomentum, 1.0 );
  //   cout<<"Helix:"<<endl;
  //   cout<<"radius: "<<fHelix.radius()<<endl;
 
  TUpdater fUpd;
  //   cout <<"TUpdater"<<endl;
  fUpd.update();
  fUpd.dump();
  //   cout<<"Hi, event routine is running"<<endl;
 
  ifstream is( "AsciiDmp_out.dat" );
 
  if ( !is.good() )
  {
    std::cout << " The input file is wrong! " << std::endl;
    return 0;
  }
 
  // EVENT list;
  EVENT         evt;
  vector<EVENT> evt_list;
 
  // event loop ..
  for ( int ix = 0; ix < 100; ix++ )
  {
 
    // initialisation
    evt.set_initialized();
 
    try
    { is >> evt; } catch ( AsciiWrongTag& ex )
    {
      std::cerr << "wrong tag, got " << ex.got() << " expected: " << ex.expected()
                << std::endl;
      return 0;
    } catch ( AsciiDumpException& ex )
    { std::cerr << "AsciiDumpException was caught!" << std::endl; }
 
    // Output MC info.
    KINE mckn  = evt.kine;
    kine mctrk = mckn.kineList[0];
    cout << "==============New Event, MC info ================" << endl;
    cout << "nTrk:" << mckn.nKine << "  idPart" << mctrk.idPart << "  chrg:" << mctrk.q
         << "   pt:" << mctrk.pt << "  theta0:" << mctrk.theta0 << "  phi0:" << mctrk.phi0
         << endl;
    //   cout<<"Total mdcDigi of this event=
    //   "<<evt.mdcDigi.nDigi<<";"<<evt.mdcDigi.digi.size()<<endl;
 
    // Clear the old MdcRec_wirhit tables and create the hits' info for the new event.
    unsigned nt = MdcRecWirhitCol::getMdcRecWirhitCol()->size();
    //   cout<<"Col size = "<<nt<<endl;
    MdcRecWirhitCol::getMdcRecWirhitCol()->clear();
 
    MDCDIGI    mdigi = evt.mdcDigi;
    MdcRecGeo* fGeom = MdcRecGeo::getMdcRecGeo();
    //   cout<<fGeom->Wire(7,72)->Layer()<<"; "<<fGeom->Wire(7,72)->Cell()<<"; "
    //       <<fGeom->Wire(7,72)->Lyr()->NCell()<<endl;
    for ( unsigned i = 0; i < mdigi.nDigi; i++ )
    {
      //    MdcRec_wirhit* mhit = new MdcRec_wirhit;
      //    mhit->tdc = evt.mdcDigi.digi[i].driftTime;
      //    mhit->adc = evt.mdcDigi.digi[i].energyDeposit;
      //    MdcRecWirhitCol::getMdcRecWirhitCol()->push_back(*mhit);
      MdcRec_wirhit mhit;
      mhit.geo   = fGeom->Wire( mdigi.digi[i].layerNo - 1, mdigi.digi[i].cellNo - 1 );
      mhit.tdc   = mdigi.digi[i].driftTime;
      mhit.adc   = mdigi.digi[i].energyDeposit;
      mhit.ddl   = mhit.tdc * 40 / 10000;
      mhit.ddr   = mhit.ddl;
      mhit.erddl = 0.013;
      mhit.erddr = 0.013;
      mhit.lr    = 2;
      mhit.stat  = 0;
      mhit.stat  = mhit.stat |= 1048576;    // bit20
      mhit.stat  = mhit.stat |= 2097152;    // bit21
      mhit.stat  = mhit.stat |= 4194304;    // bit22
      mhit.stat  = mhit.stat |= 1073741824; // bit30
      //    cout<<"layerNo = "<<mhit.geo->Layer()<<"; "<<mdigi.digi[i].layerNo<<endl;
      //    cout<<"cellNo = "<<mhit.geo->Cell()<<"; "<<mdigi.digi[i].cellNo<<endl;
      //    cout<<"NCell of this layer = "<<mhit.geo->Lyr()->NCell()<<endl;
      MdcRecWirhitCol::getMdcRecWirhitCol()->push_back( mhit );
    }
    //   _trkreco->event(BesEvent * ,int * );
    _trkreco->event();
 
    std::cout << "Event No = " << evt.header.eventNo << std::endl;
    evt_list.push_back( evt );
  }
 
  // Clear the MdcRec_wirhit tables
  unsigned nt = MdcRecWirhitCol::getMdcRecWirhitCol()->size();
  for ( unsigned i = 0; i < nt; i++ )
  {
    //       delete &(*MdcRecWirhitCol::getMdcRecWirhitCol())[i];
  }
  MdcRecWirhitCol::getMdcRecWirhitCol()->clear();
 
  is.close();
 
  /*
  cout<<"Total Event = "<<evt_list.size()<<endl;
  cout<<"event1-mdc-ndigi :"<<evt_list[0].mdcDigi.nDigi<<endl;
  cout<<"event2-mdc-ndigi :"<<evt_list[1].mdcDigi.nDigi<<endl;
  cout<<"event3-mdc-ndigi :"<<evt_list[2].mdcDigi.nDigi<<endl;
  cout<<"event1-mdc-digi15-cell :"<<evt_list[0].mdcDigi.digi[14].cellNo<<endl;
  cout<<"event1-mdc-digi15-cell :"<<evt_list[1].mdcDigi.digi[14].cellNo<<endl;
  cout<<"event1-mdc-digi15-cell :"<<evt_list[2].mdcDigi.digi[14].cellNo<<endl;
 */
  _trkreco->term();
 
  return 1;
}