Reconstruction/TrkReco/TrkReco-00-08-59-p04/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 <iostream>
#include <fstream>
#include <vector>
 
#include "TrkReco/TUpdater.h"
#include "TrkReco/Helix.h"
#include "TrkReco/Bfield.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;
 
}