factory.h

// Copyright 2009-2016 Sandia Corporation. Under the terms
// of Contract DE-AC04-94AL85000 with Sandia Corporation, the U.S.
// Government retains certain rights in this software.
//
// Copyright (c) 2009-2016, Sandia Corporation
// All rights reserved.
//
// This file is part of the SST software package. For license
// information, see the LICENSE file in the top level directory of the
// distribution.

#ifndef _SST_CORE_FACTORY_H
#define _SST_CORE_FACTORY_H

#include <sst/core/sst_types.h>

#include <stdio.h>
#include <mutex>

#include <sst/core/params.h>
#include <sst/core/elemLoader.h>
#include <sst/core/element.h>
#include <sst/core/statapi/statoutput.h>

/* Forward declare for Friendship */
extern int main(int argc, char **argv);

using namespace SST::Statistics;

namespace SST {

class Component;
class Introspector;

/**
 * Class for instantiating Components, Links and the like out
 * of element libraries.
 */
class Factory {
public:

    static Factory* getFactory() { return instance; }

    /** Get a list of allowed ports for a given component type.
     * @param type - Name of component in lib.name format
     * @return Vector of allowed port names 
     */
    const std::vector<std::string>* GetComponentAllowedPorts(std::string type);


    /** Attempt to create a new Component instantiation
     * @param id - The unique ID of the component instantiation
     * @param componentname - The fully qualified elementlibname.componentname type of component
     * @param params - The params to pass to the component's constructor
     * @return Newly created component
     */
    Component* CreateComponent(ComponentId_t id, std::string &componentname,
                               Params& params);
    
    /** Attempt to create a new Introspector instantiation
     * @param introspectorname - The fully qualified elementlibname.introspectorname type of introspector
     * @param params - The params to pass to the introspectors's constructor
     * @return Newly created introspector
     */
    Introspector* CreateIntrospector(std::string &introspectorname,
                               Params& params);

    /** Ensure that an element library containing the required event is loaded
     * @param eventname - The fully qualified elementlibname.eventname type
     */
    void RequireEvent(std::string eventname);

    /** Instatiate a new Module
     * @param type - Fully qualified elementlibname.modulename type
     * @param params - Parameters to pass to the Module's constructor
     */
    Module* CreateModule(std::string type, Params& params);

    /** Instatiate a new Module
     * @param type - Fully qualified elementlibname.modulename type
     * @param comp - Component instance to pass to the Module's constructor
     * @param params - Parameters to pass to the Module's constructor
     */
    Module* CreateModuleWithComponent(std::string type, Component* comp, Params& params);

    /** Instantiate a new Module from within the SST core
     * @param type - Name of the module to load (just modulename, not element.modulename)
     * @param params - Parameters to pass to the module at constructor time
     */
    Module* CreateCoreModule(std::string type, Params& params);

    /** Instantiate a new Module from within the SST core
     * @param type - Name of the module to load (just modulename, not element.modulename)
     * @param params - Parameters to pass to the module at constructor time
     */
    Module* CreateCoreModuleWithComponent(std::string type, Component* comp, Params& params);

    /** Instatiate a new Module
     * @param type - Fully qualified elementlibname.modulename type
     * @param comp - Component instance to pass to the SubComponent's constructor
     * @param params - Parameters to pass to the SubComponent's constructor
     */
    SubComponent* CreateSubComponent(std::string type, Component* comp, Params& params);

    /** Return partitioner function
     * @param name - Fully qualified elementlibname.partitioner type name
     */
    partitionFunction GetPartitioner(std::string name);

    /** Return generator function
     * @param name - Fully qualified elementlibname.generator type name
     */
    generateFunction GetGenerator(std::string name);

//    /** Instatiate a new StatisticBase Object
//     * @param type - Fully qualified elementlibname.statisticname type
//     * @param params - Parameters to pass to the Statistics's constructor
//     */
//    StatisticBase* CreateStatisticBase(std::string type, Params& params);
//    
//    /** Instatiate a new Statistic
//     * @param type - Fully qualified elementlibname.statisticname type
//     * @param params - Parameters to pass to the Statistics's constructor
//     */
//    template <typename T>
//    Statistic<T>* CreateStatistic(std::string type, Params& params)
//    {
//        StatisticBase *statbase = CreateStatisticBase(type,params);
//        // Now cast it to an actual Templated Statistic 
//        Statistic<T> *ret = dynamic_cast<Statistic<T>*>(statbase);
//        return ret;
//    }
    
//Module* 
//Factory::LoadCoreModule_Statistics(std::string& type, Params& params)
//{
//    // Names of sst.xxx Statistic Modules
//    if ("AccumulatorStatistic" == type) {
//        return new AccumulatorStatistic(params);
//    }
//    if ("HistogramStatistic" == type) {
//        return new HistogramStatistic(params);
//    }
//    if ("NullStatistic" == type) {
//        return new NullStatistic(params);
//    }
//    return NULL;
//}
    
    
    /** Return Python Module creation function
     * @param name - Fully qualified elementlibname.pythonModName type name
     */
    genPythonModuleFunction getPythonModule(std::string name);
    /** Checks to see if library exists and can be loaded */
    bool hasLibrary(std::string elemlib);
    void requireLibrary(std::string &elemlib);

    void getLoadedLibraryNames(std::set<std::string>& lib_names);
    void loadUnloadedLibraries(const std::set<std::string>& lib_names);

    /** Attempt to create a new Statistic Output instantiation
     * @param statOutputType - The name of the Statistic Output to create (Module Name)
     * @param statOutputParams - The params to pass to the statistic output's constructor
     * @return Newly created Statistic Output
     */
    StatisticOutput* CreateStatisticOutput(const std::string& statOutputType, const Params& statOutputParams);

    /** Determine if a statistic is defined in a components ElementInfoStatistic
     * @param type - The name of the component
     * @param statisticName - The name of the statistic 
     * @return True if the statistic is defined in the component's ElementInfoStatistic
     */
    bool DoesComponentInfoStatisticNameExist(const std::string& type, const std::string& statisticName);

    /** Determine if a statistic is defined in a subcomponents ElementInfoStatistic
     * @param type - The name of the subcomponent
     * @param statisticName - The name of the statistic 
     * @return True if the statistic is defined in the component's ElementInfoStatistic
     */
    bool DoesSubComponentInfoStatisticNameExist(const std::string& type, const std::string& statisticName);

    /** Get the enable level of a statistic defined in the component's ElementInfoStatistic
     * @param componentname - The name of the component
     * @param statisticName - The name of the statistic 
     * @return The Enable Level of the statistic from the ElementInfoStatistic
     */
    uint8_t GetComponentInfoStatisticEnableLevel(const std::string& type, const std::string& statisticName);
    
    /** Get the units of a statistic defined in the component's ElementInfoStatistic
     * @param componentname - The name of the component
     * @param statisticName - The name of the statistic 
     * @return The units string of the statistic from the ElementInfoStatistic
     */
    std::string GetComponentInfoStatisticUnits(const std::string& type, const std::string& statisticName);

private:
    Module* LoadCoreModule_StatisticOutputs(std::string& type, Params& params);
    
    friend int ::main(int argc, char **argv);

    struct ComponentInfo {
        const ElementInfoComponent* component;
        Params::KeySet_t            params;
        std::vector<std::string>    ports;
        std::vector<std::string>    statNames;
        std::vector<std::string>    statUnits;
        std::vector<uint8_t>        statEnableLevels;

        ComponentInfo() {}

        ComponentInfo(const ElementInfoComponent *component, Params::KeySet_t params) :
            component(component), params(params)
        {
            const ElementInfoPort *p = component->ports;
            while ( NULL != p && NULL != p->name ) {
                ports.push_back(p->name);
                p++;
            }

            const ElementInfoStatistic *s = component->stats;
            while ( NULL != s && NULL != s->name ) {
                statNames.push_back(s->name);
                statUnits.push_back(s->units);
                statEnableLevels.push_back(s->enableLevel);
                s++;
            }
        }

        ComponentInfo(const ComponentInfo& old) : component(old.component), params(old.params), ports(old.ports), 
                                                  statNames(old.statNames), statUnits(old.statUnits), statEnableLevels(old.statEnableLevels)
        { }

        ComponentInfo& operator=(const ComponentInfo& old)
        {
            component = old.component;
            params = old.params;
            ports = old.ports;
            statNames = old.statNames;
            statUnits = old.statUnits;
            statEnableLevels = old.statEnableLevels;
            return *this;
        }
    };

    struct IntrospectorInfo {
        const ElementInfoIntrospector* introspector;
        Params::KeySet_t params;

        IntrospectorInfo() {}

        IntrospectorInfo(const ElementInfoIntrospector *introspector,
                         Params::KeySet_t params) : introspector(introspector), params(params)
        { }

        IntrospectorInfo(const IntrospectorInfo& old) : introspector(old.introspector), params(old.params)
        { }

        IntrospectorInfo& operator=(const IntrospectorInfo& old)
        {
            introspector = old.introspector;
            params = old.params;
            return *this;
        }
    };


    struct ModuleInfo {
        const ElementInfoModule* module;
        Params::KeySet_t params;

        ModuleInfo() {}

        ModuleInfo(const ElementInfoModule *module,
                         Params::KeySet_t params) : module(module), params(params)
        { }

        ModuleInfo(const ModuleInfo& old) : module(old.module), params(old.params)
        { }

        ModuleInfo& operator=(const ModuleInfo& old)
        {
            module = old.module;
            params = old.params;
            return *this;
        }
    };

    struct SubComponentInfo {
        const ElementInfoSubComponent* subcomponent;
        Params::KeySet_t params;
        std::vector<std::string>  statNames;
        std::vector<std::string>  statUnits;
        std::vector<uint8_t>      statEnableLevels;

        SubComponentInfo() {}

        SubComponentInfo(const ElementInfoSubComponent *subcomponent,
                         Params::KeySet_t params) : subcomponent(subcomponent), params(params)
        {
            const ElementInfoStatistic *s = subcomponent->stats;
            while ( NULL != s && NULL != s->name ) {
                statNames.push_back(s->name);
                statUnits.push_back(s->units);
                statEnableLevels.push_back(s->enableLevel);
                s++;
            }
        }

    SubComponentInfo(const SubComponentInfo& old) : subcomponent(old.subcomponent), params(old.params), statNames(old.statNames), statUnits(old.statUnits), statEnableLevels(old.statEnableLevels)
        { }

        SubComponentInfo& operator=(const SubComponentInfo& old)
        {
            subcomponent = old.subcomponent;
            params = old.params;
            statNames = old.statNames;
            statUnits = old.statUnits;
            statEnableLevels = old.statEnableLevels;
            return *this;
        }
    };

    typedef std::map<std::string, const ElementLibraryInfo*> eli_map_t;
    typedef std::map<std::string, ComponentInfo> eic_map_t;
    typedef std::map<std::string, const ElementInfoEvent*> eie_map_t;
    typedef std::map<std::string, IntrospectorInfo> eii_map_t;
    typedef std::map<std::string, ModuleInfo> eim_map_t;
    typedef std::map<std::string, SubComponentInfo> eis_map_t;
    typedef std::map<std::string, const ElementInfoPartitioner*> eip_map_t;
    typedef std::map<std::string, const ElementInfoGenerator*> eig_map_t;

    Factory(std::string searchPaths);
    ~Factory();

    Factory();                      // Don't Implement
    Factory(Factory const&);        // Don't Implement
    void operator=(Factory const&); // Don't Implement

    static Factory *instance;

    Params::KeySet_t create_params_set(const ElementInfoParam *params);

    // find library information for name
    const ElementLibraryInfo* findLibrary(std::string name, bool showErrors=true);
    // handle low-level loading of name
    const ElementLibraryInfo* loadLibrary(std::string name, bool showErrors=true);

    eli_map_t loaded_libraries;
    eic_map_t found_components;
    eii_map_t found_introspectors;
    eie_map_t found_events;
    eim_map_t found_modules;
    eis_map_t found_subcomponents;
    eip_map_t found_partitioners;
    eig_map_t found_generators;
    std::string searchPaths;
    ElemLoader *loader;
    std::string loadingComponentType;

    std::pair<std::string, std::string> parseLoadName(const std::string& wholename);

    std::recursive_mutex factoryMutex;


protected:
   Output &out;
};

} // namespace SST

#endif // SST_CORE_FACTORY_H