FWCore/TestProcessor/interface/RunFromSource.h - master - githubmirror/cmssw

FWCore/TestProcessor/interface/RunFromSource.h

71 строка2 KB

Chris Jones

Fix meta-data and Service issues in TestSourceProcessor

17 июл 2024, 16:34

17 июл 2024, 16:3487efee3

#ifndef FWCore_TestProcessor_RunFromSource_h
#define FWCore_TestProcessor_RunFromSource_h
// -*- C++ -*-
//
// Package:     FWCore/TestProcessor
// Class  :     RunFromSource
//
/**\class RunFromSource RunFromSource.h "RunFromSource.h"

 Description: [one line class summary]

 Usage:
    <usage>

*/
//
// Original Author:  Chris Jones
//         Created:  Mon, 30 Apr 2018 18:51:27 GMT
//

// system include files
#include <string>

// user include files
#include "FWCore/TestProcessor/interface/TestHandle.h"
#include "FWCore/Framework/interface/RunPrincipal.h"
#include "FWCore/Utilities/interface/TypeID.h"
#include "FWCore/ServiceRegistry/interface/ServiceToken.h"
#include "FWCore/ServiceRegistry/interface/ServiceRegistry.h"

// forward declarations

namespace edm {

  namespace test {

    class RunFromSource {
    public:
      RunFromSource(std::shared_ptr<RunPrincipal const> iPrincipal, edm::ServiceToken iToken)
          : principal_(iPrincipal), token_(iToken) {}

      // ---------- const member functions ---------------------
      template <typename T>
      TestHandle<T> get(std::string const& iModule,
                        std::string const& iInstanceLabel,
                        std::string const& iProcess) const {
        ServiceRegistry::Operate operate(token_);

        auto h = principal_->getByLabel(
            edm::PRODUCT_TYPE, edm::TypeID(typeid(T)), iModule, iInstanceLabel, iProcess, nullptr, nullptr, nullptr);
        if (h.failedToGet()) {
          return TestHandle<T>(std::move(h.whyFailedFactory()));
        }
        void const* basicWrapper = h.wrapper();
        assert(basicWrapper);
        Wrapper<T> const* wrapper = static_cast<Wrapper<T> const*>(basicWrapper);
        return TestHandle<T>(wrapper->product());
      }

      RunNumber_t run() const { return principal_->run(); }
      RunAuxiliary const& aux() const { return principal_->aux(); }

    private:
      // ---------- member data --------------------------------
      std::shared_ptr<RunPrincipal const> principal_;
      edm::ServiceToken token_;
    };
  }  // namespace test
}  // namespace edm

#endif