Add 'apply' and 'remove' buttons to the Center 2D window.

Author: david-cattermole

python/mmSolver/tools/centertwodee/tool.py

@@ -60,6 +60,8 @@ def main():
         return
 
     try:
+        # Set the solver 'running' flag so that the Solver UI does not
+        # update.
         mmapi.set_solver_running(True)
 
         save_sel = maya.cmds.ls(selection=True, long=True) or []

python/mmSolver/tools/centertwodee/ui/centertwodee_window.py

@@ -85,10 +85,16 @@ def __init__(self, parent=None, name=None):
 
         # Standard Buttons
         self.baseHideStandardButtons()
+        self.createBtn.show()
+        self.applyBtn.show()
         self.resetBtn.show()
         self.closeBtn.show()
+        self.createBtn.setText('Apply')
+        self.applyBtn.setText('Remove')
         self.resetBtn.setText('Reset All')
 
+        self.createBtn.clicked.connect(self.apply_center)
+        self.applyBtn.clicked.connect(self.remove_center)
         self.resetBtn.clicked.connect(self.reset_options)
 
         # Hide irrelevant stuff
@@ -143,6 +149,14 @@ def camera_shape_update(self, value):
             lib.set_zoom(self.camera_shape, output)
         return
 
+    def apply_center(self):
+        import mmSolver.tools.centertwodee.tool as tool
+        tool.main()
+
+    def remove_center(self):
+        import mmSolver.tools.centertwodee.tool as tool
+        tool.remove()
+
     def reset_options(self):
         self.form.reset_options()