From f4de628c999810808d961899299686a9f0c50b5b Mon Sep 17 00:00:00 2001
From: Thales Lima Oliveira <thaleslima.ufu@gmail.com>
Date: Sat, 14 Jul 2018 13:08:45 -0300
Subject: Some vector pre-allocation and test curves implementation

---
 Project/Electromechanical.h | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'Project/Electromechanical.h')

diff --git a/Project/Electromechanical.h b/Project/Electromechanical.h
index 4c82bb6..573c0c2 100644
--- a/Project/Electromechanical.h
+++ b/Project/Electromechanical.h
@@ -57,6 +57,9 @@ class Electromechanical : public ElectricCalculation
     bool RunStabilityCalculation();
     wxString GetErrorMessage() const { return m_errorMsg; }
     std::vector<double> GetTimeVector() const { return m_timeVector; }
+    
+    // For tests
+    std::vector<double> GetIterationVector() const { return m_iterationsNumVector; }
 
    protected:
     void SetEventTimeList();
@@ -98,6 +101,7 @@ class Electromechanical : public ElectricCalculation
                                         double k = 1.0);
 
     void SaveData();
+    void PreallocateVectors();
 
     wxWindow* m_parent = NULL;
     wxString m_errorMsg = _("Unknown error");
@@ -127,6 +131,10 @@ class Electromechanical : public ElectricCalculation
     std::vector<bool> m_eventOccurrenceList;
 
     std::vector<double> m_timeVector;
+    
+    // For tests
+    int m_iterationsNum = 0.0;
+    std::vector<double> m_iterationsNumVector;
 };
 
 #endif  // ELECTROMECHANICAL_H
-- 
cgit