APPLICATIONS/geb2daq.cxx

// @(#)fROOT/APPLICATIONS:$Name:  $:$Id: geb2daq.cxx,v 1.10 2007/09/05 09:32:04 Diego_Faso Exp $
// Author: Diego Faso <mailto:faso@to.infn.it>, 2006/07/25

// the following lines are required in order to compile on Mac-OS
//#ifdef __CINT__
#if !defined(linux) && !defined(__linux) && !defined(__linux__)
#undef __GNUC__
#endif 

#include <sys/ioctl.h>
#include <sys/socket.h>
#include <netinet/in.h>

#include <Riostream.h>

#include "TRandom.h"

#include "TFndRun.h"
#include "FROOT.h"

Int_t verbose_level = 0;

TString db_host = 0;
TString data_path = 0;
TString run_type = 0;
Int_t first_run = 0;
Int_t last_run  = 0;
TFndRun *fRun = 0;

UInt_t Udp_Port = 2007;
Int_t  gebip = FROOT::Ip2Int(getenv("FND_GEBIP"));
Int_t  nevts = std::numeric_limits<Int_t>::max();

Int_t      fSdOnl;
UInt_t     fAddLen;
struct sockaddr_in ClAdd;
struct sockaddr_in MyAdd;
struct sockaddr *Cl = 0;
struct sockaddr *My = 0;
int l = 0;

UInt_t *raw;
Int_t cur_run=0;
UInt_t ec=0;
Int_t istop = 0;
Int_t evlen = 0;

void print_usage(){
  cout << endl << " ---------------------------------------------- " << endl; 
  cout << endl << "Hi, this is \"geb2daq\"" << endl;
  cout << " -> Available command-line arguments are:" << endl; 
  cout << "     - argv[1]: path to raw-data" << endl;
  cout << "     - argv[2]: name of the run-type to be processed (es: FINU-COSM-BHABHA-PEDE-...)" << endl;
  cout << "     - argv[3]: number of the first run to be processed " << endl;
  cout << "     - argv[4]: number of the last run to be processed " << endl;
  cout << "     - argv[5]: number of events to be processed " << endl;
  cout << "     - argv[6]: simulated DAQ-rate " << endl;
  cout << " -> Default arguments:" << endl;
  cout << "     - argv[4]: if empty the default is equal to argv[4] (single run processing)" << endl; 
  cout << "     - argv[5]: if empty the default is " << endl; 
  cout << "     - argv[6]: if empty the default is: random rate" << endl; 
  cout << " -> Example command: " << endl;   
  cout << " geb2daq $RDT FINU 1357 1359 1000" << endl << endl;   
  cout << " -> The DAQ-simulation will send data to the ip-address secified in \"$FND_GEBIP\"." << endl;
  cout << "     (actual value for is \"$FND_GEBIP\": 0x" << hex << gebip << dec << endl << endl; 
  cout << " ---------------------------------------------- " << endl << endl; 
}


void BuildFndRun(){

  //   cout << "Path to raw-data: \"" << data_path << "\"" << endl;
  fRun = new TFndRun();
  fRun->SetDataPath(data_path);
  fRun->SetRunType(run_type);
  fRun->SetMaxNumberOfEvents(nevts);

}

void create_socket(){

  cout << "Create Socket " << endl;
  Cl = (struct sockaddr*) &ClAdd;
  My = (struct sockaddr*) &MyAdd;
  
  fAddLen=sizeof(MyAdd);                
  bzero((char*)&MyAdd, fAddLen);
  MyAdd.sin_family      = AF_INET;
  MyAdd.sin_addr.s_addr = INADDR_ANY;     
  
  MyAdd.sin_port = htons(Udp_Port);
  fSdOnl = socket(AF_INET, SOCK_DGRAM, 0);
  bind(fSdOnl, My, fAddLen);
  
  l=0x10000;
  setsockopt(fSdOnl,SOL_SOCKET,SO_SNDBUF,&l,sizeof(l));
  
}

Int_t print_raw_ev(){
  
  cout << "Event: " << ec << endl;
  if(raw){
    evlen = raw[0];
    if(verbose_level > 1){
      for(Int_t i=0; i<5; i++) cout << "raw[ " << i << "]: " << raw[i] << endl;   
      //for(Int_t i=0; i<evlen; i++) cout << "raw[ " << i << "]: " << raw[i] << endl;   
    }
    return 0;
  }
  else{
    cout << " Raw event not available!" << endl;
    return -1;
  }
  
}


int check_socket(){

  Int_t check_result=0;
  //Int_t nn = 0;
  do {
    //cout << "check socket " << ++nn << endl;
    usleep(1000);
    ioctl(fSdOnl,FIONREAD,(Char_t *)&check_result);
  } while (check_result == 0);
  return check_result;
}


void send_one_event(){

  Int_t check_result=check_socket();
  if(check_result){
    if(verbose_level > 2) cout << "receive request" << endl;
    Int_t receive_err = recvfrom(fSdOnl ,(Char_t *)&check_result  ,sizeof(int) ,0 ,Cl ,&fAddLen);
    if(verbose_level > 2) cout << "get event: receive_err = " << receive_err << endl;
        
    ec++;

    // ---
    Int_t print_ret_val = 0;
    if(verbose_level > 1) print_ret_val = print_raw_ev();
    else if(verbose_level > 0 && ec%100==0) print_ret_val = print_raw_ev();
    if(print_ret_val !=0) cout << "PRINT-RAW error!" << endl;
    // ---
    sendto(fSdOnl ,(char*)raw,evlen*sizeof(int),0 ,Cl ,fAddLen);
  }
  else{
    cout << ".....................................\"check_result\"=0 ...!!!" << endl;
    sendto(fSdOnl ,&check_result ,sizeof(int) ,0 ,Cl ,fAddLen);  
  }
  return;
}

void event_loop(const Long_t &delay){
  if(verbose_level > 0) cout << "Start EVENT loop" << endl;

  Int_t initrun_val=0;

  for(cur_run=first_run;cur_run<=last_run;cur_run++){
    if(ec>nevts){
      cout  << "geb2daq message: Requested number of events reached." << endl;
      return;
    }
    if(verbose_level > 0) cout << "Initializing new run: run " << cur_run << " ev: " << ec << " ; requested evts: " << nevts << endl;
    initrun_val = fRun->InitNewRun(run_type,cur_run);
    if (initrun_val == -4) FROOT::TerminateFroot(0,TString("wrong argument received as \"run-type\" (\""+run_type)+"\").");

    Int_t exit_val = 0;
    while(exit_val == 0 &&  initrun_val==0) {
      raw = fRun->GetNextRawEvent(exit_val); 
      send_one_event();
      fRun->ClearEvent();
      
      if(delay > 0 ) usleep((ULong_t)delay);
      //      else usleep((ULong_t)(gRandom->Gaus(1000000,500000)));
      else usleep((ULong_t)(gRandom->Uniform(1,1500000)));
      
      usleep(100); // required anyway
    }
    
    if(verbose_level > 0) cout << "geb2daq message: End of Run Processing: " << ec << " events " << endl;   
    if(cur_run != last_run) fRun->FinishRun();
  }
}

Int_t main(Int_t argc, Char_t **argv) {

  //  for(Int_t i=0;i<argc;i++) cout << "argv[" << i << "]" << argv[i] << endl;

  verbose_level = 2;


  //  Float_t max_rate = std::numeric_limits<Float_t>::max(); // ev/s
  Float_t max_rate = -1;

  if(argc < 4){
    print_usage();
    return 0;
  }

  data_path = argv[1];
  run_type = argv[2];
  first_run = atoi(argv[3]);
  last_run = first_run;

  if(argc > 4) last_run = atoi(argv[4]);
  if(argc > 5) nevts = atoi(argv[5]);
  if(argc > 6) max_rate = atoi(argv[6]);

  Long_t Delay = (Long_t)(1e+6 * (Float_t)(1./max_rate)); // microseconds
  cout << "geb2daq message: " << nevts << " events will be sent to ip \"" << gebip << "\" (port " << Udp_Port << ")" << endl;
  if(Delay>0) cout << "geb2daq message: Maximum rate set to " << max_rate << " ev/s (delay: " << (ULong_t)(Delay/1000) << " ms)." << endl;
  else cout << "geb2daq message: simulated DAQ-rate set to \"random\" (delay range: 1ms - 10 s)." << endl;
  BuildFndRun();
  create_socket();
  event_loop(Delay);
  
  fRun->FinishRun(kTRUE);
  delete fRun;
  cout << endl << " ---> geb2daq process completed... bye bye ;)"<< endl;
}

---