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

EvtRunMap

typedef map<int, vector<uint32_t> > EvtRunMap

Definition at line 16 of file evt_filter.cxx.

Function Documentation

getEvtRunMap()

EvtRunMap getEvtRunMap

(

const char *

fconf

)

Definition at line 19 of file evt_filter.cxx.

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

Referenced by main().

getFnamesAtBkk()

vector< string > getFnamesAtBkk

(

int

run

)

Definition at line 34 of file evt_filter.cxx.

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

Referenced by main().

listFnames()

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

listNotFound()

void listNotFound

(

EvtRunMap &

map

)

Definition at line 81 of file evt_filter.cxx.

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

Referenced by main().

main()

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 96 of file evt_filter.cxx.

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