/home/bes3soft/bes3soft/Boss/7.0.2/dist/7.0.2/Event/RawFile/RawFile-00-00-10/test/evt_filter.cxx

Go to the documentation of this file.

#include "RawFile/RawFileWriter.h"
#include "RawFile/RawFileReader.h"
#include "IRawFile/RawFileExceptions.h"
#include <xmlrpc-c/base.hpp>
#include <xmlrpc-c/client_simple.hpp>
#include <iostream>
#include <fstream>
#include <algorithm>
#include <vector>
#include <map>
#include <string>
#include <unistd.h>

using namespace std;

typedef map<int, vector<uint32_t> >  EvtRunMap;


EvtRunMap getEvtRunMap(const char* fconf) {
   EvtRunMap map;

   int      run;
   uint32_t evtId;
   ifstream cfs(fconf);

   while ( !(cfs>>run>>evtId).eof() ) {
      //std::cout << "run: " << run << "  evt: " << evtId << std::endl;
      map[run].push_back(evtId);
   }

   return map;
}

vector<string> getFnamesAtBkk(int run)
{
   xmlrpc_c::value result;
   xmlrpc_c::clientSimple aClient;

   try {
      aClient.call( "http://bes3db.ihep.ac.cn:8080/bemp/xmlrpc",
                    "FileFinder.getFileByRun",
                    "siss",
                    &result,
                    "REAL",
                    run,
                    "Full",
                    "disk" );
   }
   catch (...) {
      cerr << "Failed to lookup the Bookkeeping Server !!!" << endl;
      exit(1);
   }

   vector<string> fnames;
   vector<xmlrpc_c::value> vItems = ((xmlrpc_c::value_array)result).vectorValueValue();

   for ( vector<xmlrpc_c::value>::iterator it = vItems.begin();
         it != vItems.end(); ++it ) {
      //xmlrpc_c::value_struct* item = (xmlrpc_c::value_struct*)(&(*it));
      map<string, xmlrpc_c::value> item = (xmlrpc_c::value_struct)(*it);
      string fname = (xmlrpc_c::value_string)(item["Replica"]);
      fnames.push_back(fname);
   }

   if ( fnames.empty() ) {
      cout << "No files found in Bookkeeping !!!" << endl;
      exit(1);
   }

   return fnames;
}

void listFnames(const vector<string>& fnames)
{
   vector<string>::const_iterator it = fnames.begin();
   for ( ; it != fnames.end(); ++it ) {
      std::cout << (*it) << std::endl;
   }
}

void listNotFound(EvtRunMap& map)
{
   EvtRunMap::iterator it = map.begin();

   while ( it != map.end() ) {
      vector<uint32_t>::iterator iit = it->second.begin();

      while ( iit != it->second.end() ) {
         std::cout << "Event " << *iit << " not found in run " << it->first << std::endl;
         ++iit;
      }
      ++it;
   }
}

int main(int argc, char* argv[])
{
   if ( argc != 2 && argc != 3 ) {
      cout << "Usage: " << argv[0] << "  EvtList.txt  [data.raw]" << endl;
      exit(0);
   }

   if ( access( argv[1], F_OK ) < 0 ) {
      std::cerr << "Invalid file: " << argv[1] << std::endl;
      exit(1);
   }
   EvtRunMap map = getEvtRunMap(argv[1]);

   string ofname = string(argv[1]) + ".raw";
   RawFileWriter* pfw = new RawFileWriter(ofname.c_str());

   const uint32_t* data;

   if ( argc == 2 ) {

      EvtRunMap::iterator it = map.begin();
      while( it != map.end() ) {
         RawFileReader* pfr = new RawFileReader( getFnamesAtBkk(it->first) );
         /*
          * can be optimized by pfr->findEventById() when IDX exist!!!
          */
         while ( ! it->second.empty() ) {
            try {
               data = pfr->nextEvent();
            }
            catch ( RawFileException& e ) {
               e.print();
               break;
            }

            uint32_t evtId = data[8 + data[5]];
            vector<uint32_t>::iterator iit = find(it->second.begin(), it->second.end(), evtId);

            if ( iit != it->second.end() ) {
               pfw->writeEvent(data);
               it->second.erase(iit);
            }
         }

         delete pfr;

         ++it;
      }
   }
   else {

      RawFileReader* pfr = new RawFileReader( argv[2] );
      while ( true ) {
         try {
            data = pfr->nextEvent();
         }
         catch ( RawFileException& e ) {
            e.print();
            break;
         }

         uint32_t evtId = data[8 + data[5]];
         uint32_t runId = data[9 + data[5]];
         EvtRunMap::iterator it = map.find( runId );
         if ( it != map.end() ) {
            vector<uint32_t>::iterator iit = find(it->second.begin(), it->second.end(), evtId);

            if ( iit != it->second.end() ) {
               pfw->writeEvent(data);
               it->second.erase(iit);

               if ( it->second.empty() ) {
                  map.erase( it );
                  if ( map.empty() ) {
                     break;
                  }
               }
            }
         }
      }

      delete pfr;
   }

   delete pfw;

   // list events NOT found
   listNotFound(map);

   return 0;
}

---