Changes in src/Actions/ParserAction/SetTremoloAtomdataAction.cpp [be21fa:ca331c]

File:

• src/Actions/ParserAction/SetTremoloAtomdataAction.cpp (modified) (1 diff)

src/Actions/ParserAction/SetTremoloAtomdataAction.cpp

@@ -41 +41 @@
   FormatParser<tremolo> &parser = FormatParserStorage::getInstance().getParser<tremolo>();
 
-  LOG(1, "Setting Tremolo's ATOMDATA to: '" << params.atomdata_string << "'");
+  LOG(1, "Setting Tremolo's ATOMDATA to: '" << params.atomdata_string.get() << "'");
 
-  parser.setAtomData(params.atomdata_string);
+  const std::string old_atomdata = parser.getAtomData();
 
-  return Action::success;
+  if (params.atomdata_reset.get())
+    parser.resetAtomData(params.atomdata_string.get());
+  else
+    parser.setAtomData(params.atomdata_string.get());
+
+  return Action::state_ptr(new ParserSetTremoloAtomdataState(old_atomdata, params));
 }
 
 Action::state_ptr ParserSetTremoloAtomdataAction::performUndo(Action::state_ptr _state) {
-//  ParserLoadXyzState *state = assert_cast<ParserLoadXyzState*>(_state.get());
+  ParserSetTremoloAtomdataState *state = assert_cast<ParserSetTremoloAtomdataState*>(_state.get());
 
-  return Action::failure;
-//  string newName = state->mol->getName();
-//  state->mol->setName(state->lastName);
-//
-//  return Action::state_ptr(new ParserLoadXyzState(state->mol,newName));
+  FormatParser<tremolo> &parser = FormatParserStorage::getInstance().getParser<tremolo>();
+  LOG(1, "INFO: Reverting to 'ATOMDATA " << state->old_atomdata << "'.");
+  parser.resetAtomData(state->old_atomdata);
+
+  return Action::state_ptr(_state);
 }
 
 Action::state_ptr ParserSetTremoloAtomdataAction::performRedo(Action::state_ptr _state){
-  return Action::failure;
+  ParserSetTremoloAtomdataState *state = assert_cast<ParserSetTremoloAtomdataState*>(_state.get());
+
+  FormatParser<tremolo> &parser = FormatParserStorage::getInstance().getParser<tremolo>();
+
+  if (state->params.atomdata_reset.get())
+    parser.resetAtomData(state->params.atomdata_string.get());
+  else
+    parser.setAtomData(state->params.atomdata_string.get());
+
+  return Action::state_ptr(_state);
 }
 
 bool ParserSetTremoloAtomdataAction::canUndo() {
-  return false;
+  return true;
 }
 
 bool ParserSetTremoloAtomdataAction::shouldUndo() {
-  return false;
+  return true;
 }
 /** =========== end of function ====================== */