/data/refman/pasoursint/CMSSW_6_1_2_SLHC2_patch1/src/EventFilter/Modules/src/ShmOutputModuleRegistry.cc

Go to the documentation of this file.

#include "EventFilter/Modules/interface/ShmOutputModuleRegistry.h"
#include "EventFilter/Modules/src/FUShmOutputModule.h"

#include <iostream>

namespace evf{

  ShmOutputModuleRegistry::ShmOutputModuleRegistry(const edm::ParameterSet &ps){
  }

  void ShmOutputModuleRegistry::registerModule(std::string &name, OutputModule *op)
  {
    clm_.insert(std::pair<std::string, OutputModule*>(name,op));
  }
  

  OutputModule* ShmOutputModuleRegistry::get(std::string &name)
  {
    OutputModule* retval = 0;
    idct it= clm_.find(name);
    if(it!=clm_.end())
      retval = (it->second);
    return retval;
  }
  void ShmOutputModuleRegistry::dumpRegistry(){
    idct it= clm_.begin();
    while(it!=clm_.end()){
      std::cout << "name " << it->first << "add " 
                << (unsigned long)(it->second) << std::endl;
      it++;
    }
  }
  std::vector<edm::FUShmOutputModule *> ShmOutputModuleRegistry::getShmOutputModules()
  {
    std::vector<edm::FUShmOutputModule *> outputs;
    idct it= clm_.begin();
    while(it!=clm_.end()){
      edm::FUShmOutputModule * sho = dynamic_cast<edm::FUShmOutputModule *> ((*it).second);
      if (sho!=NULL) {
        outputs.push_back(sho);
      }
      it++;
    }
    return outputs;
  }
  void ShmOutputModuleRegistry::clear()
  {
     clm_.clear();
     shmOutputsWithDatasets_.clear();
     listOfDatasets_.clear();
  }

  void ShmOutputModuleRegistry::updateDatasetInfo()
  {
    shmOutputsWithDatasets_.clear();
    listOfDatasets_.clear();
    std::vector<edm::FUShmOutputModule *> outputs = getShmOutputModules();
    for (unsigned int i=0;i<outputs.size();i++) {
      edm::FUShmOutputModule * output = outputs[i];
      if (output->getStreamId().size()) {
        std::vector<std::string> datasets  = output->getDatasetNames();
        listOfDatasets_.insert(listOfDatasets_.end(),datasets.begin(),datasets.end());
        if (datasets.size())
          shmOutputsWithDatasets_.push_back(output);
      }
    }
  }
  std::string ShmOutputModuleRegistry::getDatasetCSV()
  {
    std::string datasetNameString;
    for (unsigned int i=0;i<listOfDatasets_.size();i++) {
      if (i)
        datasetNameString+=",";
      datasetNameString+=listOfDatasets_[i];
    }
    return datasetNameString;
  }

  void ShmOutputModuleRegistry::insertStreamAndDatasetInfo(edm::ParameterSet & streams, edm::ParameterSet & datasets)
  {
    idct it= clm_.begin();
    while(it!=clm_.end()){
      edm::FUShmOutputModule * sho = dynamic_cast<edm::FUShmOutputModule *> ((*it).second);
      if (sho!=NULL) {
        sho->insertStreamAndDatasetInfo(streams,datasets);
      }
      it++;
    }
  }

} //end namespace evf