X-Git-Url: https://git.teslayout.com/public/public/public/?a=blobdiff_plain;f=source%2Fstatistics%2FEngine.cpp;h=622e58ce860b57dbee5a05d8f7f7fb8c1d8e869c;hb=af9c86ffb0e28d35ad94d99c5f77e41578c972b4;hp=ebde06c7fbc2748252db21216af76a0d7a6377b1;hpb=c6b62fd814b6e5b4e3ba65b6f092f6aa4e1d298d;p=anna.git

diff --git a/source/statistics/Engine.cpp b/source/statistics/Engine.cpp
index ebde06c..622e58c 100644
--- a/source/statistics/Engine.cpp
+++ b/source/statistics/Engine.cpp
@@ -39,8 +39,8 @@ Engine::Engine() {
 //------------------------------------------------------------ Engine::~Engine()
 //------------------------------------------------------------------------------
 Engine::~Engine() {
-  for (_accumulator_vector_it it = a_accumulators.begin(); it != a_accumulators.end(); it++)
-    delete (*it);
+  for (_accumulator_map_it it = a_accumulators.begin(); it != a_accumulators.end(); it++)
+    delete (it->second);
 }
 
 
@@ -99,12 +99,9 @@ bool Engine::enableSampleLog(const int & id, const char *sampleFileName) throw()
   // For all concepts:
   _concept_identification_map_nc_iter it_min(a_concept_identification_map.begin());
   _concept_identification_map_nc_iter it_max(a_concept_identification_map.end());
-
   for(it = it_min; it != it_max; it++) {
     realName = "";
-
     if(providedName != "") realName = anna::functions::asString("%s.%d.csv", providedName.c_str(), (*it).first);
-
     SampleFile_ptr = &((*it).second.SampleFile);
     *SampleFile_ptr = realName;
   }
@@ -117,14 +114,20 @@ bool Engine::enableSampleLog(const int & id, const char *sampleFileName) throw()
 //--------------------------------------------------- Engine::disableSampleLog()
 //------------------------------------------------------------------------------
 bool Engine::disableSampleLog(const int & id) throw() {
-  _concept_identification_map_nc_iter it = a_concept_identification_map.find(id);
 
-  if(it == a_concept_identification_map.end()) return false;
+  if(id != -1) {
+    _concept_identification_map_nc_iter it = a_concept_identification_map.find(id);
 
-  // Access to map
-  std::string *SampleFile_ptr = &((*it).second.SampleFile);
-  *SampleFile_ptr = "";
-  return true;
+    if(it == a_concept_identification_map.end()) return false;
+
+    // Access to map
+    std::string *SampleFile_ptr = &((*it).second.SampleFile);
+    *SampleFile_ptr = "";
+    return true;
+  }
+
+  // For all concepts:
+  return enableSampleLog(id /* -1 */, "");
 }
 
 
@@ -156,13 +159,26 @@ bool Engine::logSample(const int & conceptId, const anna::Millisecond & unixTime
 
 
 //------------------------------------------------------------------------------
-//----------------------------------------------------------- Engine::asString()
+//-------------------------------------------------- Engine::createAccumulator()
 //------------------------------------------------------------------------------
-Accumulator *Engine::createAccumulator(const std::string &name) throw() {
-  Accumulator *result;
+Accumulator *Engine::createAccumulator(const std::string &name) throw(anna::RuntimeException) {
+  Accumulator *result = getAccumulator(name);
+
+  if (result)
+    throw anna::RuntimeException(anna::functions::asString("Cannot register another accumulator with the same name: %s", name.c_str()), ANNA_FILE_LOCATION);
 
   result = new Accumulator(name);
-  a_accumulators.push_back(result);
+  a_accumulators[name] = result;
+  return result;
+}
+
+Accumulator* Engine::getAccumulator(const std::string &name) throw() {
+  Accumulator *result = NULL;
+
+  _accumulator_map_nc_it it = a_accumulators.find(name);
+  if (it != a_accumulators.end())
+    result = it->second;
+
   return result;
 }
 
@@ -193,7 +209,7 @@ std::string Engine::asString(void) const throw() {
     trace += "\n";
     trace += "\n   Concept id:            "; trace += anna::functions::asString((*iter).first);
 
-    if((*iter).second.SampleFile != "") trace += "\n   Sample file:           "; trace += (*iter).second.SampleFile;
+    if((*iter).second.SampleFile != "") { trace += "\n   Sample file:           "; trace += (*iter).second.SampleFile; }
 
     trace += "\n   Description:           "; trace += (*iter).second.Description;
     trace += "\n   Unit:                  "; trace += (*iter).second.Unit;
@@ -204,9 +220,9 @@ std::string Engine::asString(void) const throw() {
 
   // Accumulators:
   trace += "\nNumber of accumulators= "; trace += anna::functions::asString(a_accumulators.size());
-  for (_accumulator_vector_it it = a_accumulators.begin(); it != a_accumulators.end(); it++) {
+  for (_accumulator_map_it it = a_accumulators.begin(); it != a_accumulators.end(); it++) {
     trace += "\n";
-    trace += (*it)->asString();
+    trace += it->second->asString();
   }
 
   return (trace);
@@ -240,8 +256,8 @@ anna::xml::Node* Engine::asXML(anna::xml::Node* parent, const int & numberOfDeci
   // Accumulators:
   anna::xml::Node* accumulators = result->createChild("anna.statistics.Accumulators");
   accumulators->createAttribute("RegisteredAccumulators", a_accumulators.size());
-  for (_accumulator_vector_it it = a_accumulators.begin(); it != a_accumulators.end(); it++)
-    (*it)->asXML(accumulators);
+  for (_accumulator_map_it it = a_accumulators.begin(); it != a_accumulators.end(); it++)
+    it->second->asXML(accumulators);
 
   return result;
 }