mcapi::log_state() in run()

[simgrid.git] / src / mc / VisitedState.cpp

diff --git a/src/mc/VisitedState.cpp b/src/mc/VisitedState.cpp
index fa49142..8c4d035 100644 (file)
--- a/src/mc/VisitedState.cpp
+++ b/src/mc/VisitedState.cpp
@@ -23,7 +23,7 @@ namespace mc {
 VisitedState::VisitedState(unsigned long state_number) : num(state_number)
 {  
   this->heap_bytes_used = mcapi::get().get_remote_heap_bytes();
-  this->actors_count = mcapi::get().mc_get_remote_simulation().actors().size();
+  this->actors_count = mcapi::get().get_actors_size();
   this->system_state = std::make_shared<simgrid::mc::Snapshot>(state_number);
 }
 
@@ -51,7 +51,7 @@ VisitedStates::addVisitedState(unsigned long state_number, simgrid::mc::State* g
             new_state->num, graph_state->num_);
 
   auto range =
-      boost::range::equal_range(states_, new_state.get(), simgrid::mc::DerefAndCompareByActorsCountAndUsedHeap());
+      boost::range::equal_range(states_, new_state.get(), mcapi::get().compare_pair());
 
   if (compare_snapshots)
     for (auto i = range.first; i != range.second; ++i) {