#include "mfacq.h"

//---------------------------------------------------------------
// Programme d'acquisition BAORadio multi-fibres/multi-threads 
//  LAL -   2009 - 2010
//     R. Ansari, M.Taurigna
//---------------------------------------------------------------

static RAcqMemZoneMgr* pMmgr=NULL;
static  PCIEMultiReader* pPcierThr=NULL;

void Stop(int s)
{ 
  if (s == 9765) cout << " Stop after exception ..." << endl;
  else printf("............. MAIN ... receive signal %d \n",s);
  if (pMmgr != NULL) pMmgr->Stop();
  if (pPcierThr !=NULL) pPcierThr->Stop();

}


//-----------------------------------------------------------------------
//-------------------- le programme principal ---------------------------
//-----------------------------------------------------------------------
int main(int narg, char* arg[])
{ 
  
  int rc = 0;
  cout << " ============= BAORadio / Acquisition : mfacq  =================" << endl;
  cout << " ===============================================================" << endl;
  cout << " =========  " <<BAOR_ACQ_VER_STR <<BAOR_ACQ_VER << " ===========" << endl;
  cout << " ===============================================================" << endl;
  
 
  InitTim();
 
  // Initialisation 
  TArrayInitiator _arri;
  FitsIOServerInitiator _fiosi;


  if ((narg < 2)||(strcmp(arg[1],"-h"))==0) {
    Usage(false);
    return 1;
  }
  
 
  try {
    // Creation/initialisation parametres Acq
    BRAcqConfig acpar;
    acpar.ReadDCFile(arg[1]);
    // Creation des repertoires 
    if (acpar.CreateOutputDirectories()!=0)  return 9;
    acpar.Print(cout);
    struct sigaction act;  
    if (!acpar.GetParams().fg_hard_ctrlc) {
      act.sa_handler=Stop;
      sigaction(SIGINT,&act,NULL);        
    }
    rc = MultiFibreAcq();    
  }
  catch (MiniFITSException& exc) {
    cerr << " mfacq.cc catched MiniFITSException " << exc.Msg() << endl;
    rc = 77;
  } 
  catch (PCIEWException& exc) {
    cerr << " mfacq.cc catched MiniFITSException " << exc.Msg() << endl;
    rc = 75;
  }  
  catch (PThrowable& exc) {
    cerr << " mfacq.cc catched Exception " << exc.Msg() << endl;
    rc = 76;
  }  
  catch (std::exception& sex) {
    cerr << "\n mfacq.cc std::exception :" 
         << (string)typeid(sex).name() << "\n msg= " 
         << sex.what() << endl;
    rc = 78;
  }
  catch (...) {
    cerr << "  mfacq.cc : Catched ... exception " << endl;
    rc = 79;
  }
  cout << "mfacq[9] ----- END --- stopping acq program " << endl;
  return rc;
}


/* --Nouvelle-Fonction-- */
void Usage(bool fgshort) 
{
  if (fgshort) {
    cout << " Usage: mfacq DataCardFile \n" << endl;
    cout << "parameter in DataCard :Fibres PaquetSize NFrameDMA NFiles NBlocPerFile \n"
         << "                 NMaxProc DataDirPath acqmode nZones nPaqZones fg_hard_ctrlc" << endl;
    cout << "   type mfacq -h for detailed usage " << endl;
    return;
  }
  return;
}
 
 
/* --Nouvelle-Fonction-- */
int MultiFibreAcq() 
{
  Timer tm("mfacq/MultiFibre");
  cout << " ---------- mfacq/ MultiFibreAcq()  ------------- " << endl;
  PCIEWrapperInterface* pciwp[4]={NULL,NULL,NULL,NULL};
    
  BRAcqConfig bpar;
  BRParList& acpar=bpar.GetParams();
  
  cout << " mfacq[1]/Info: Creating RAcqMemZoneMgr for" << acpar.NbFibers << " fibers , nZones=" 
       << acpar.nZones << " NbPaquet/Zone=" << acpar.nPaqZone 
       << " mmgrPaqSize=" << acpar.MMgrPaquetSize() << endl;
  RAcqMemZoneMgr mmgr(acpar.nZones, bpar.NFibers(), acpar.nPaqZone, acpar.MMgrPaquetSize());
  pMmgr=&mmgr;  
  int stepproc = 2;  
  int nmean = acpar.nPaqZone*acpar.NBlocPerFile/stepproc;

#ifndef NOPCIECARD  
  for (int i=0 ;i <acpar.NFibers() ;i++) {
    UINT32 card=(acpar.FiberNum[i]-1)/2+1;
    UINT32 cardfiber=(acpar.FiberNum[i]-1)%2
    cout <<"mfacq[2] CreatePCIEWrapperV6- indice " << i <<  "INIT card " << card 
	 << " fibre " << cardfiber << endl;
    pciwp[i] =  CreatePCIEWrapperV6(card,patternSZ,dmaSize,activate_pattern,cardfiber);
  }
#else 
  TestPCIWrapperNODMA pciw1(bpar.RecvPaquetSize(), acpar.nopciLossRate);
  TestPCIWrapperNODMA pciw2(bpar.RecvPaquetSize(), acpar.nopciLossRate);
  TestPCIWrapperNODMA pciw3(bpar.RecvPaquetSize(), acpar.nopciLossRate);
  TestPCIWrapperNODMA pciw4(bpar.RecvPaquetSize(), acpar.nopciLossRate);
  pciwp[0] = &pciw1;
  pciwp[1] = &pciw2;
  pciwp[2] = &pciw3;
  pciwp[3] = &pciw4;  
#endif 

  cout <<"mfacq[3] Creating DataSaver and DataProc2C objects ... " << endl;
  MultiDataSaver DsThr(mmgr, bpar.OutputDirectory(), bpar.MaxNbFiles(), 
		       bpar.BlocPerFile(), bpar.GetSaveFits());
  string ppath=bpar.OutputDirectory();
  DataProc2C PrThr(mmgr, ppath, 1280, stepproc, acpar.NMaxProc);
  vector<PCIEWrapperInterface*> vec_pciw;
  for  (int i=0 ;i<bpar.NFibers();i++)  vec_pciw.push_back( pciwp[i]);
  cout << "mfacq[3] Creating   PCIEMultiReader       " << endl;
  PCIEMultiReader PcierThr(vec_pciw,bpar.DMASizeBytes(),bpar.RecvPaquetSize(),
			   mmgr, bpar.MaxNbBlocs(), bpar.GetDataConvFg());
  usleep(200);  // attente au cas ou ...
  pPcierThr=&PcierThr;
  tm.Split("Threads created");  
  cout << "mfacq[4] - starting 3 threads pcir, ds, pr ... " << endl;
  PcierThr.start();
  DsThr.start();
  if (acpar.NMaxProc>0) { // On ne demarre que si au moins NMaxProc>0
    PrThr.start();
  }
  
  // On attend avant de declencher la terminaison des threads
  usleep(100000);
  
  cout << "mfacq[5] -  Waiting for threads to finish ... " << endl;  
  PcierThr.join();
  DsThr.join();
  mmgr.Stop();
  if (acpar.NMaxProc>0) { // On ne demarre que si au moins NMaxProc>0
    PrThr.join();
  }
  pMmgr=NULL;  
  cout << "mfacq[6] ---------- threads finished ---------------- " << endl;
  tm.Split("Threads Finished");  
 
#ifndef NOPCIECARD
  for (int i=0 ;i <acpar.NbFibers ;i++) {
    UINT32 card=(acpar.FiberNum[i]-1)/2+1;
    UINT32 cardfiber=(acpar.FiberNum[i]-1)%2;
    DeletePCIEWrapperV6(card,cardfiber);
  }
#endif    
return 0;

}