pymodel.h

// -*- c++ -*-

// Copyright 2009-2020 NTESS. Under the terms
// of Contract DE-NA0003525 with NTESS, the U.S.
// Government retains certain rights in this software.
//
// Copyright (c) 2009-2020, NTESS
// 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_MODEL_PYTHON
#define SST_CORE_MODEL_PYTHON

// This only works if we have Python defined from configure, otherwise this is
// a compile time error.
#ifdef SST_CONFIG_HAVE_PYTHON

#include <map>
#include <string>
#include <vector>

#include "sst/core/model/sstmodel.h"
#include "sst/core/config.h"
#include "sst/core/rankInfo.h"
#include "sst/core/output.h"
#include "sst/core/configGraph.h"

using namespace SST;

namespace SST {
namespace Core {

class SSTPythonModelDefinition : public SSTModelDescription {

public:
    SSTPythonModelDefinition(const std::string& script_file, int verbosity, Config* config, int argc, char **argv);
    SSTPythonModelDefinition(const std::string& script_file, int verbosity, Config* config);
    virtual ~SSTPythonModelDefinition();
    
    ConfigGraph* createConfigGraph() override;
    
protected:
    void initModel(const std::string& script_file, int verbosity, Config* config, int argc, char** argv);
    std::string scriptName;
    Output* output;
    Config* config;
    ConfigGraph *graph;
    char *namePrefix;
    size_t namePrefixLen;
    std::vector<size_t> nameStack;
    std::map<std::string, ComponentId_t> compNameMap;
    ComponentId_t nextComponentId;
    

public:  /* Public, but private.  Called only from Python functions */
    Config* getConfig(void) const { return config; }

    ConfigGraph* getGraph(void) const { return graph; }

    Output* getOutput() const { return output; }

    ComponentId_t getNextComponentId() { return nextComponentId++; }

    ComponentId_t addComponent(const char *name, const char *type) {
        ComponentId_t id = getNextComponentId();
        graph->addComponent(id, name, type);
        compNameMap[std::string(name)] = id;
        return id;
    }

    ComponentId_t findComponentByName(const char *name) const {
        std::string origname(name);
        auto index = origname.find(":");
        std::string compname = origname.substr(0,index);
        auto itr = compNameMap.find(compname);
        
        // Check to see if component was found
        if ( itr == compNameMap.end() ) return UNSET_COMPONENT_ID;
        
        // If this was just a component name
        if ( index == std::string::npos ) return itr->second;
        
        // See if this is a valid subcomponent name
        ConfigComponent* cc = graph->findComponent(itr->second);
        cc = cc->findSubComponentByName(origname.substr(index+1,std::string::npos));
        if ( cc ) return cc->id;
        return UNSET_COMPONENT_ID;
    }
    
    void addLink(ComponentId_t id, const char *link_name, const char *port, const char *latency, bool no_cut) const {graph->addLink(id, link_name, port, latency, no_cut); }
    void setLinkNoCut(const char *link_name) const {graph->setLinkNoCut(link_name); }
    
    void pushNamePrefix(const char *name);
    void popNamePrefix(void);
    char* addNamePrefix(const char *name) const;
    
    void setStatisticOutput(const char* Name) { graph->setStatisticOutput(Name); }
    void addStatisticOutputParameter(const std::string& param, const std::string& value) { graph->addStatisticOutputParameter(param, value); }
    void setStatisticLoadLevel(uint8_t loadLevel) { graph->setStatisticLoadLevel(loadLevel); }
    
    void enableStatisticForComponentName(const std::string& compname, const std::string& statname, bool apply_to_children = false) const {
        graph->enableStatisticForComponentName(compname,statname,apply_to_children);
    }

    void enableStatisticForComponentType(const std::string& comptype, const std::string& statname, bool apply_to_children = false) const  {
        graph->enableStatisticForComponentType(comptype, statname, apply_to_children);
    }
    
    void addStatisticParameterForComponentName(const std::string& compname, const std::string& statname, const std::string& param, const std::string& value, bool apply_to_children = false) {
        graph->addStatisticParameterForComponentName(compname,statname,param,value,apply_to_children);
    }

    void addStatisticParameterForComponentType(const std::string& comptype, const std::string& statname, const std::string& param, const std::string& value, bool apply_to_children = false) {
        graph->addStatisticParameterForComponentType(comptype, statname, param, value, apply_to_children);
    }
};

std::map<std::string,std::string> generateStatisticParameters(PyObject* statParamDict);

}
}

#endif

#endif