SST  10.1.0
StructuralSimulationToolkit
pymodel.h
1 // -*- c++ -*-
2 
3 // Copyright 2009-2020 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-2020, 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 
37 class SSTPythonModelDefinition : public SSTModelDescription {
38 
39 public:
40  SSTPythonModelDefinition(const std::string& script_file, int verbosity, Config* config, int argc, char **argv);
41  SSTPythonModelDefinition(const std::string& script_file, int verbosity, Config* config);
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 
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  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); }
78  void setLinkNoCut(const char *link_name) const {graph->setLinkNoCut(link_name); }
79 
80  void pushNamePrefix(const char *name);
81  void popNamePrefix(void);
82  char* addNamePrefix(const char *name) const;
83 
84  void setStatisticOutput(const char* Name) { graph->setStatisticOutput(Name); }
85  void addStatisticOutputParameter(const std::string& param, const std::string& value) { graph->addStatisticOutputParameter(param, value); }
86  void setStatisticLoadLevel(uint8_t loadLevel) { graph->setStatisticLoadLevel(loadLevel); }
87 
88  void enableStatisticForComponentName(const std::string& compname, const std::string& statname, bool apply_to_children = false) const {
89  graph->enableStatisticForComponentName(compname,statname,apply_to_children);
90  }
91 
92  void enableStatisticForComponentType(const std::string& comptype, const std::string& statname, bool apply_to_children = false) const {
93  graph->enableStatisticForComponentType(comptype, statname, apply_to_children);
94  }
95 
96  void addStatisticParameterForComponentName(const std::string& compname, const std::string& statname, const std::string& param, const std::string& value, bool apply_to_children = false) {
97  graph->addStatisticParameterForComponentName(compname,statname,param,value,apply_to_children);
98  }
99 
100  void addStatisticParameterForComponentType(const std::string& comptype, const std::string& statname, const std::string& param, const std::string& value, bool apply_to_children = false) {
101  graph->addStatisticParameterForComponentType(comptype, statname, param, value, apply_to_children);
102  }
103 };
104 
105 std::map<std::string,std::string> generateStatisticParameters(PyObject* statParamDict);
106 
107 }
108 }
109 
110 #endif
111 
112 #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:193
A Configuration Graph A graph representing Components and Links.
Definition: configGraph.h:321
Base class for Model Generation.
Definition: sstmodel.h:22