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

#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>

Go to the source code of this file.

Typedefs
typedef map< int, vector< uint32_t > >	EvtRunMap
Functions
EvtRunMap	getEvtRunMap (const char *fconf)
vector< string >	getFnamesAtBkk (int run)
void	listFnames (const vector< string > &fnames)
void	listNotFound (EvtRunMap &map)
int	main (int argc, char *argv[])

---

Typedef Documentation

typedef map<int, vector<uint32_t> > EvtRunMap

Definition at line 16 of file evt_filter.cxx.

---

Function Documentation

EvtRunMap getEvtRunMap

(

const char *

fconf

)

Definition at line 19 of file evt_filter.cxx.

Referenced by main().

                                          {
   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

)

Definition at line 34 of file evt_filter.cxx.

References fname, and deljobs::string.

Referenced by main().

{
   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

)

Definition at line 73 of file evt_filter.cxx.

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

void listNotFound

(

EvtRunMap &

map

)

Definition at line 81 of file evt_filter.cxx.

Referenced by main().

{
   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[]
	)

Definition at line 96 of file evt_filter.cxx.

References getEvtRunMap(), getFnamesAtBkk(), listNotFound(), RawFileReader::nextEvent(), RawFileException::print(), deljobs::string, and RawFileWriter::writeEvent().

{
   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;
}

---