InputSource.h

Go to the documentation of this file.

#ifndef DataFormats_FWLite_interface_InputSource_h
#define DataFormats_FWLite_interface_InputSource_h

#include <vector>
#include <string>
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/Exception.h"

namespace fwlite {

  class InputSource {
  
  public:
    InputSource() {
      throw cms::Exception("InvalidInput") << "Must specify a vstring fileNames" << std::endl;
    }
    InputSource(const edm::ParameterSet& cfg) :
      maxEvents_(-1), reportAfter_(10),
      files_( cfg.getParameterSet("fwliteInput").getParameter<std::vector<std::string> >("fileNames") )
      {
    // optional parameter
    if( cfg.getParameterSet("fwliteInput").existsAs<int>("maxEvents")){ 
      maxEvents_ = cfg.getParameterSet("fwliteInput").getParameter<int>("maxEvents"); 
    }
    // optional parameter
    if( cfg.getParameterSet("fwliteInput").existsAs<unsigned int>("outputEvery")){ 
      reportAfter_ = cfg.getParameterSet("fwliteInput").getParameter<unsigned int>("outputEvery"); 
    }
      }
      const std::vector<std::string>& files() const { return files_; }
      int maxEvents() const { return maxEvents_; }
      unsigned int reportAfter() const { return reportAfter_;}
      
  protected:
      int maxEvents_;
      unsigned int reportAfter_;
      std::vector<std::string> files_;
  };
}

#endif