diff options
Diffstat (limited to 'Project/ElementPlotData.h')
| -rw-r--r-- | Project/ElementPlotData.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/Project/ElementPlotData.h b/Project/ElementPlotData.h index 1434664..5c67575 100644 --- a/Project/ElementPlotData.h +++ b/Project/ElementPlotData.h @@ -49,6 +49,7 @@ class ElementPlotData CT_SHUNT_INDUCTOR, CT_SHUNT_CAPACITOR, CT_LOAD, + CT_TEST, NUM_ELEMENTS, CT_TIME }; @@ -59,12 +60,14 @@ class ElementPlotData wxString GetName() const { return m_name; } void SetName(wxString name) { m_name = name; } CurveType GetCurveType() const { return m_curveType; } + void SetCurveType(CurveType type) { m_curveType = type; } PlotData* GetPlotData(int index) const { return m_elementData[index]; } void AddData(std::vector<double> values, wxString name); int GetElementDataNumber() const { return static_cast<int>(m_elementData.size()); } std::vector<double> GetValues(int index) const { return m_elementData[index]->GetValues(); } void SetValues(int index, std::vector<double> values) { m_elementData[index]->SetValues(values); } + void SetPlot(int index, bool plot = true) { m_elementData[index]->SetPlot(plot); } wxString GetDataName(int index) const { return m_elementData[index]->GetName(); } void SetDataName(int index, wxString name) { m_elementData[index]->SetName(name); } wxColour GetColour(int index) const { return m_elementData[index]->GetColour(); } |