From 5320e99276d8261fc838e05126f2f862b1bb2caa Mon Sep 17 00:00:00 2001 From: DarkPhoenix Date: Wed, 26 Jun 2019 20:06:51 +0300 Subject: [PATCH] Restore values for vectors (just in case!) --- gui/graphFrame/panel.py | 28 ++++++++++++++++++++-------- 1 file changed, 20 insertions(+), 8 deletions(-) diff --git a/gui/graphFrame/panel.py b/gui/graphFrame/panel.py index 2c4b772b1..48cb45c0c 100644 --- a/gui/graphFrame/panel.py +++ b/gui/graphFrame/panel.py @@ -173,16 +173,28 @@ class GraphControlPanel(wx.Panel): self.inputs[(inputDef.handle, inputDef.unit)] = (fieldTextBox, fieldIcon, fieldLabel) self.inputsSizer.Add(fieldSizer, 0, wx.EXPAND | wx.BOTTOM, 5) + def handleVector(vectorDef, vector, handledHandles): + handledHandles.add(vectorDef.lengthHandle) + handledHandles.add(vectorDef.angleHandle) + try: + storedLength = self._storedConsts[(vectorDef.lengthHandle, vectorDef.lengthUnit)] + except KeyError: + pass + else: + vector.SetLength(storedLength / 100) + try: + storedAngle = self._storedConsts[(vectorDef.angleHandle, vectorDef.angleUnit)] + except KeyError: + pass + else: + vector.SetAngle(storedAngle) + view = self.graphFrame.getView() handledHandles = set() - srcVectorDef = view.srcVectorDef - if srcVectorDef is not None: - handledHandles.add(srcVectorDef.lengthHandle) - handledHandles.add(srcVectorDef.angleHandle) - tgtVectorDef = view.tgtVectorDef - if tgtVectorDef is not None: - handledHandles.add(tgtVectorDef.lengthHandle) - handledHandles.add(tgtVectorDef.angleHandle) + if view.srcVectorDef is not None: + handleVector(view.srcVectorDef, self.srcVector, handledHandles) + if view.tgtVectorDef is not None: + handleVector(view.tgtVectorDef, self.tgtVector, handledHandles) selectedX = view.xDefMap[self.xType] # Always add main input addInputField(view.inputMap[selectedX.mainInput], handledHandles, mainInput=True)