FWCore/PluginManager/interface/PluginFactoryBase.h - master - githubmirror/cmssw

FWCore/PluginManager/interface/PluginFactoryBase.h
113 строк4 KB
Chris Jones
Use oneapi namespace
22 дек 2021, 00:32
22 дек 2021, 00:326cf8c1a
#ifndef FWCore_PluginManager_PluginFactoryBase_h
#define FWCore_PluginManager_PluginFactoryBase_h
// -*- C++ -*-
//
// Package:     PluginManager
// Class  :     PluginFactoryBase
//
/**\class PluginFactoryBase PluginFactoryBase.h FWCore/PluginManager/interface/PluginFactoryBase.h

 Description: Base class for all plugin factories

 Usage:
    This interface provides access to the most generic information about a plugin factory:
    1) what is the name of the presently available plugins
    2) the file name of the loadable which contained the plugin
*/
//
// Original Author:  Chris Jones
//         Created:  Wed Apr  4 12:24:44 EDT 2007
//

// system include files
#include <string>
#include <vector>
#include <map>
#include <atomic>
#include "oneapi/tbb/concurrent_unordered_map.h"
#include "oneapi/tbb/concurrent_vector.h"
#include "FWCore/Utilities/interface/zero_allocator.h"
#include "FWCore/Utilities/interface/Signal.h"
#include "FWCore/Utilities/interface/thread_safety_macros.h"
// user include files
#include "FWCore/PluginManager/interface/PluginInfo.h"

// forward declarations
namespace edmplugin {
  class PluginFactoryBase {
  public:
    PluginFactoryBase() {}
    PluginFactoryBase(const PluginFactoryBase&) = delete;                   // stop default
    const PluginFactoryBase& operator=(const PluginFactoryBase&) = delete;  // stop default
    virtual ~PluginFactoryBase();

    struct PluginMakerInfo {
      PluginMakerInfo(void* iPtr, const std::string& iName) : m_name(iName), m_ptr() {
        m_ptr.store(iPtr, std::memory_order_release);
      }

      PluginMakerInfo(const PluginMakerInfo& iOther) : m_name(iOther.m_name), m_ptr() {
        m_ptr.store(iOther.m_ptr.load(std::memory_order_acquire), std::memory_order_release);
      }

      PluginMakerInfo& operator=(const PluginMakerInfo& iOther) {
        m_name = iOther.m_name;
        m_ptr.store(iOther.m_ptr.load(std::memory_order_acquire), std::memory_order_release);
        return *this;
      }
      std::string m_name;
      //NOTE: this has to be last since once it is non zero it signals
      // that the construction has finished
      std::atomic<void*> m_ptr;
    };

    typedef oneapi::tbb::concurrent_vector<PluginMakerInfo, edm::zero_allocator<PluginMakerInfo>> PMakers;
    typedef oneapi::tbb::concurrent_unordered_map<std::string, PMakers> Plugins;

    // ---------- const member functions ---------------------

    ///return info about all plugins which are already available in the program
    virtual std::vector<PluginInfo> available() const;

    ///returns the name of the category to which this plugin factory belongs
    virtual const std::string& category() const = 0;

    //The signal is only modified during a shared library load which is protected by a mutex by the operating system
    ///signal containing plugin category, and  plugin info for newly added plugin
    CMS_THREAD_SAFE mutable edm::signalslot::Signal<void(const std::string&, const PluginInfo&)> newPluginAdded_;

    // ---------- static member functions --------------------

    // ---------- member functions ---------------------------

  protected:
    /**call this as the last line in the constructor of inheriting classes
      this is done so that the virtual table will be properly initialized when the
      routine is called
      */
    void finishedConstruction();

    void newPlugin(const std::string& iName);

    //since each inheriting class has its own Container type to hold their PMakers
    // this function allows them to share the same code when doing the lookup
    // this routine will throw an exception if iName is unknown therefore the return value is always valid
    void* findPMaker(const std::string& iName) const;

    //similar to findPMaker but will return 'end()' if iName is known
    void* tryToFindPMaker(const std::string& iName) const;

    void fillInfo(const PMakers& makers, PluginInfo& iInfo, std::vector<PluginInfo>& iReturn) const;

    void fillAvailable(std::vector<PluginInfo>& iReturn) const;

    void registerPMaker(void* iPMaker, const std::string& iName);

  private:
    void checkProperLoadable(const std::string& iName, const std::string& iLoadedFrom) const;
    // ---------- member data --------------------------------
    Plugins m_plugins;
  };

}  // namespace edmplugin
#endif