Changeset 36b8d8 for molecuilder/src/Helpers/Assert.cpp

Timestamp:

Mar 19, 2010, 4:30:57 PM (15 years ago)

Author:

Tillmann Crueger <crueger@…>

Children:

9f3025

Parents:

43ed42

git-author:

Tillmann Crueger <crueger@…> (03/19/10 10:50:06)

git-committer:

Tillmann Crueger <crueger@…> (03/19/10 16:30:57)

Message:

Added more functionality to custom asserts.

• Asserts allow now setting any choice as default behaviour
• Asserts allow setting of hooks that have to be performed before the exit is done

File:

• molecuilder/src/Helpers/Assert.cpp (modified) (4 diffs)

molecuilder/src/Helpers/Assert.cpp

@@ -11 +11 @@
 using namespace std;
 
-bool _my_assert::always_throw = false;
+namespace Assert{
+  AssertionFailure::AssertionFailure(std::string _condition,
+                                     std::string _file,
+                                     int _line,
+                                     std::string _message) :
+    condition(_condition),
+    file(_file),
+    line(_line),
+    message(_message)
+  {}
+
+  std::string AssertionFailure::getFile(){
+    return file;
+  }
+
+  int AssertionFailure::getLine(){
+    return line;
+  }
+
+  std::string AssertionFailure::getMessage(){
+    return message;
+  }
+
+  std::ostream& AssertionFailure::operator<<(std::ostream& out){
+    out << "Assertion \"" << condition << "\" failed in file " << file << " at line " << line << endl;
+    out << "Assertion Message: " << message << std::endl;
+    return out;
+  }
+
+  const char  ActionKeys[]  = {'\0','a','t','i'};
+  const char* ActionNames[] = {"Ask","Abort","Throw","Ignore"};
+}
+
+using namespace Assert;
+
+Action _my_assert::defaultAction = Ask;
+std::vector<Assert::hook_t> _my_assert::hooks;
 
 bool _my_assert::check(const bool res,
@@ -23 +59 @@
     cout << "Assertion " << condition << " failed in file " << filename << " at line " << line << endl;
     cout << "Assertion Message: " << message << std::endl;
-    if(always_throw){
-      throw AssertException(filename,line);
-    }
     while(true){
-      cout << "Please choose: (a)bbort, (t)hrow execption, (i)gnore, al(w)ays ignore" << endl;
       char choice;
-      cin >> choice;
+      if(defaultAction==Ask) {
+        cout << "Please choose: (a)bort, (t)hrow execption, (i)gnore, al(w)ays ignore" << endl;
+        cin >> choice;
+      }
+      else{
+        choice = ActionKeys[defaultAction];
+      }
       switch(choice){
         case 'a':
@@ -35 +73 @@
           break;
         case 't':
-          throw AssertException(filename,line);
+          throw AssertionFailure(condition,filename,line,message);
           break;
         case 'w':
@@ -48 +86 @@
   return false;
 }
+
+void _my_assert::doHooks(){
+  for(vector<hook_t>::reverse_iterator iter = hooks.rbegin(); iter!=hooks.rend(); ++iter ){
+    (*iter)();
+  }
+}
+
+void _my_assert::addHook(hook_t hook){
+  hooks.push_back(hook);
+}
+
+void _my_assert::removeHook(Assert::hook_t hook){
+  for(vector<hook_t>::iterator iter = hooks.begin(); iter!=hooks.end();){
+    if((*iter)==hook){
+      iter = hooks.erase(iter);
+    }
+    else{
+      ++iter;
+    }
+  }
+}
+
+void _my_assert::setDefault(Assert::Action action){
+  defaultAction = action;
+}
+Assert::Action _my_assert::getDefault(){
+  return defaultAction;
+}
+std::string _my_assert::printDefault(){
+  return ActionNames[defaultAction];
+}