SST  11.0.0
StructuralSimulationToolkit
pymodel.h
1 // -*- c++ -*-
2 
3 // Copyright 2009-2021 NTESS. Under the terms
4 // of Contract DE-NA0003525 with NTESS, the U.S.
5 // Government retains certain rights in this software.
6 //
7 // Copyright (c) 2009-2021, NTESS
8 // All rights reserved.
9 //
10 // This file is part of the SST software package. For license
11 // information, see the LICENSE file in the top level directory of the
12 // distribution.
13 
14 
15 #ifndef SST_CORE_MODEL_PYTHON
16 #define SST_CORE_MODEL_PYTHON
17 
18 // This only works if we have Python defined from configure, otherwise this is
19 // a compile time error.
20 //#ifdef SST_CONFIG_HAVE_PYTHON
21 
22 #include <map>
23 #include <string>
24 #include <vector>
25 
26 #include "sst/core/model/sstmodel.h"
27 #include "sst/core/config.h"
28 #include "sst/core/rankInfo.h"
29 #include "sst/core/output.h"
30 #include "sst/core/configGraph.h"
31 
32 using namespace SST;
33 
34 namespace SST {
35 namespace Core {
36 
38 
39 public:
40  SSTPythonModelDefinition(const std::string& script_file, int verbosity, Config* config, double start_time, int argc, char **argv);
41  SSTPythonModelDefinition(const std::string& script_file, int verbosity, Config* config, double start_time);
42  virtual ~SSTPythonModelDefinition();
43 
44  ConfigGraph* createConfigGraph() override;
45 
46 protected:
47  void initModel(const std::string& script_file, int verbosity, Config* config, int argc, char** argv);
48  std::string scriptName;
49  Output* output;
50  Config* config;
51  ConfigGraph *graph;
52  char *namePrefix;
53  size_t namePrefixLen;
54  std::vector<size_t> nameStack;
55  std::map<std::string, ComponentId_t> compNameMap;
56  ComponentId_t nextComponentId;
57  double start_time;
58 
59 public: /* Public, but private. Called only from Python functions */
60  Config* getConfig(void) const { return config; }
61 
62  ConfigGraph* getGraph(void) const { return graph; }
63 
64  Output* getOutput() const { return output; }
65 
66  ComponentId_t getNextComponentId() { return nextComponentId++; }
67 
68  ComponentId_t addComponent(const char *name, const char *type) {
69  auto id = graph->addComponent(name, type);
70  return id;
71  }
72 
73  ConfigComponent* findComponentByName(const char *name) const {
74  return graph->findComponentByName(std::string(name));
75  }
76 
77  ConfigComponentMap_t& components() { return graph->getComponentMap(); }
78 
79  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); }
80  void setLinkNoCut(const char *link_name) const {graph->setLinkNoCut(link_name); }
81 
82  void pushNamePrefix(const char *name);
83  void popNamePrefix(void);
84  char* addNamePrefix(const char *name) const;
85 
86  void setStatisticOutput(const char* Name) { graph->setStatisticOutput(Name); }
87  void addStatisticOutputParameter(const std::string& param, const std::string& value) { graph->addStatisticOutputParameter(param, value); }
88  void setStatisticLoadLevel(uint8_t loadLevel) { graph->setStatisticLoadLevel(loadLevel); }
89 
90  void addGlobalParameter(const char* set, const char* key, const char* value, bool overwrite) {
91  Params::insert_global(set,key,value,overwrite);
92  }
93 
94  UnitAlgebra getElapsedExecutionTime() const;
95  UnitAlgebra getLocalMemoryUsage() const;
96 };
97 
98 std::map<std::string,std::string> generateStatisticParameters(PyObject* statParamDict);
99 SST::Params pythonToCppParams(PyObject* statParamDict);
100 PyObject* buildStatisticObject(StatisticId_t id);
101 PyObject* buildEnabledStatistic(ConfigComponent* cc, const char* statName, PyObject* statParamDict,
102  bool apply_to_children);
103 PyObject* buildEnabledStatistics(ConfigComponent* cc, PyObject* statList, PyObject* paramDict, bool apply_to_children);
104 }
105 }
106 
107 #endif
Output object provides consistent method for outputting data to stdout, stderr and/or sst debug file...
Definition: output.h:54
Class to contain SST Simulation Configuration variables.
Definition: config.h:31
Represents the configuration of a generic component.
Definition: configGraph.h:219
A Configuration Graph A graph representing Components and Links.
Definition: configGraph.h:358
Parameter store.
Definition: params.h:44
static void insert_global(const std::string &set, const key_type &key, const key_type &value, bool overwrite=true)
Adds a key/value pair to the specified global set.
Definition: params.cc:342
Base class for Model Generation.
Definition: sstmodel.h:22
Performs Unit math in full precision.
Definition: unitAlgebra.h:107
Definition: pymodel.h:37