fixing some types

Author: MarcG-TUE

include/maxplus/base/fsm/fsm.h

@@ -210,6 +210,7 @@ class FiniteStateMachine {
     [[nodiscard]] virtual const SetOfStateRefs &getInitialStates() const = 0;
     [[nodiscard]] virtual const SetOfStateRefs &getFinalStates() const = 0;
     [[nodiscard]] virtual const SetOfStates &getStates() const = 0;
+    [[nodiscard]] virtual const SetOfEdges &getEdges() const = 0;
 };
 
 //
@@ -507,7 +508,7 @@ class FiniteStateMachine : public Abstract::FiniteStateMachine {
 
 private:
     SetOfStates<StateLabelType, EdgeLabelType> states;
-    Abstract::SetOfEdges edges;
+    SetOfEdges<StateLabelType, EdgeLabelType> edges;
     // State<StateLabelType, EdgeLabelType> *initialState;
     SetOfStateRefs<StateLabelType, EdgeLabelType> initialStates;
     SetOfStateRefs<StateLabelType, EdgeLabelType> finalStates;
@@ -650,6 +651,11 @@ class FiniteStateMachine : public Abstract::FiniteStateMachine {
     [[nodiscard]] const SetOfStates<StateLabelType, EdgeLabelType> &getStates() const override {
         return this->states;
     };
+
+     [[nodiscard]] const SetOfEdges<StateLabelType, EdgeLabelType> &getEdges() const override {
+        return this->edges;
+    };
+
     Abstract::SetOfStateRefs getStateRefs() {
         Abstract::SetOfStateRefs result;
         for (auto i : this->states) {
@@ -658,7 +664,13 @@ class FiniteStateMachine : public Abstract::FiniteStateMachine {
         return result;
     };
 
-    Abstract::SetOfEdges &getEdges() { return this->edges; };
+    Abstract::SetOfEdgeRefs getEdgeRefs() {
+        Abstract::SetOfEdgeRefs result;
+        for (auto i : this->edges) {
+            result.insert(&(*(i.second)));
+        }
+        return result;
+    };
 
     EdgeRef<StateLabelType, EdgeLabelType>
     getEdge(const State<StateLabelType, EdgeLabelType> &source,

src/graph/smpls.cc

@@ -33,7 +33,7 @@ void EdgeLabeledModeFSM::removeDanglingStates() {
     ELSSetOfStateRefs statesToBeRemoved;
 
     // const ELSSetOfStates& elsStates = this->getStates();
-    auto &elsEdges = dynamic_cast<ELSSetOfEdges &>(this->getEdges());
+    auto &elsEdges = dynamic_cast<const ELSSetOfEdges &>(this->getEdges());
 
     /*go through all edges and find all edges that end in
     dangling states. Also store dangling states.*/
@@ -71,11 +71,11 @@ void EdgeLabeledModeFSM::removeDanglingStates() {
         statesToBeRemoved.clear();
 
         // elsStates = this->getStates();
-        elsEdges = this->getEdges();
+        const ELSSetOfEdges &elsEdges2 = this->getEdges();
 
         /*go through all edges and find all edges that end in
         dangling states. Also store dangling states.*/
-        for (const auto &it : elsEdges) {
+        for (const auto &it : elsEdges2) {
 
             auto e = dynamic_cast<ELSEdgeRef>(&(*(it.second)));
             const auto s = dynamic_cast<ELSStateRef>(e->getDestination());