Add class and macro to save and auto-restore a variable

5 years ago · fdb97a3e9c
1 changed files with 13 additions and 0 deletions
--- a/Marlin/src/core/utility.h
+++ b/Marlin/src/core/utility.h
@ -121,3 +121,16 @@ inline void serial_delay(const millis_t ms) {
 #endif

 void print_bin(const uint16_t val);
+
+template<typename T>
+class restorer {
+  T& ref_;
+  T  val_;
+public:
+  restorer(T& perm) : ref_(perm), val_(perm) {}
+  ~restorer() { restore(); }
+  inline void restore() { ref_ = val_; }
+};
+
+#define REMEMBER(X) restorer<typeof(X)> X##_restorer(X)
+#define RESTORE(X) X##_restorer.restore()