Fix function return types and remove unused variable

janjurca · Jan 17, 2024 · 771a811 · 771a811
1 parent 0e3fec1
commit 771a811
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/include/filestorm/filetree.h b/include/filestorm/filetree.h
@@ -42,8 +42,8 @@ class FileTree {
 
   Node* getRoot() const;
 
-  const int getDirectoryCount() const { return directory_count; }
-  const int getFileCount() const { return file_count; }
+  int getDirectoryCount() const { return directory_count; }
+  int getFileCount() const { return file_count; }
 
 private:
   void printRec(const Node* node, int depth) const;

diff --git a/include/filestorm/utils/psm.h b/include/filestorm/utils/psm.h
@@ -31,7 +31,7 @@ class Transition {
 
 class ProbabilisticStateMachine {
 public:
-  ProbabilisticStateMachine(std::map<std::string, State>& states, std::map<std::string, Transition>& transitions, State& init) : _states(states), _transitions(transitions), _currentState(init) {}
+  ProbabilisticStateMachine(std::map<std::string, Transition>& transitions, State& init) : _transitions(transitions), _currentState(init) {}
 
   void performTransition() {
     // Choose a transition based on probabilities
@@ -58,6 +58,5 @@ class ProbabilisticStateMachine {
 
 private:
   State& _currentState;
-  std::map<std::string, State>& _states;
   std::map<std::string, Transition>& _transitions;
 };
diff --git a/source/scenarios/aging.cpp b/source/scenarios/aging.cpp
@@ -71,7 +71,7 @@ void AgingScenario::run() {
   int iteration = 0;
   progressbar bar(getParameter("iterations").get_int());
 
-  ProbabilisticStateMachine psm(states, transtions, states.at("S"));
+  ProbabilisticStateMachine psm(transtions, states.at("S"));
 
   while (iteration < getParameter("iterations").get_int()) {
     bar.update();