forked from youen/assembly_handbook
removed old files
This commit is contained in:
Youen Toupin
parent
e57f90d5e5
commit
c8374e1ea2
25
InitGui.py
25
InitGui.py
@ -23,7 +23,6 @@ class AssemblyHandbookWorkbench(Gui.Workbench):
|
||||
|
||||
dev = True # indicates development mode (enables additional tools, log, debug checks, etc.)
|
||||
|
||||
context = None
|
||||
docObserver = None
|
||||
docLinkObserver = None
|
||||
partsCache = None
|
||||
@ -42,9 +41,6 @@ class AssemblyHandbookWorkbench(Gui.Workbench):
|
||||
"""
|
||||
code which should be computed when a user switch to this workbench
|
||||
"""
|
||||
if self.context is None:
|
||||
self.initializeContext()
|
||||
|
||||
import ahb_document_observer
|
||||
if self.docLinkObserver is None:
|
||||
self.docLinkObserver = ahb_document_observer.DocLinkObserver()
|
||||
@ -75,30 +71,14 @@ class AssemblyHandbookWorkbench(Gui.Workbench):
|
||||
self.appendContextMenu("", contextMenu)
|
||||
self.appendContextMenu("", "Separator")
|
||||
|
||||
def initializeContext(self):
|
||||
import ahb_context
|
||||
self.context = ahb_context.AHB_Context()
|
||||
|
||||
def registerCommands(self):
|
||||
toolbox = []
|
||||
|
||||
#self.importModule('ahb_cmd_export_csv')
|
||||
#toolbox.append("AHB_exportCsv")
|
||||
|
||||
#self.importModule('ahb_cmd_set_active_stage')
|
||||
#toolbox.append("AHB_setActiveStage")
|
||||
|
||||
#self.importModule('ahb_cmd_set_part_stage')
|
||||
#toolbox.append("AHB_setPartStage")
|
||||
|
||||
#self.importModule('ahb_cmd_switch_visibility_mode')
|
||||
#toolbox.append("AHB_switchVisibilityMode")
|
||||
|
||||
self.importModule('ahb_cmd_render')
|
||||
toolbox.append("AHB_render")
|
||||
|
||||
#self.importModule('ahb_cmd_animate')
|
||||
#toolbox.append("AHB_animate")
|
||||
|
||||
self.importModule('ahb_cmd_new_step')
|
||||
toolbox.append("AHB_new_step")
|
||||
@ -126,11 +106,7 @@ class AssemblyHandbookWorkbench(Gui.Workbench):
|
||||
self.removeToolbar("Assembly Handbook")
|
||||
self.removeMenu("Assembly Handbook")
|
||||
|
||||
self.context.setAnimationActive(False)
|
||||
|
||||
import importlib
|
||||
import ahb_context
|
||||
importlib.reload(ahb_context)
|
||||
|
||||
import ahb_parts_cache
|
||||
importlib.reload(ahb_parts_cache)
|
||||
@ -140,7 +116,6 @@ class AssemblyHandbookWorkbench(Gui.Workbench):
|
||||
importlib.reload(ahb_techdraw_extensions)
|
||||
self.techDrawExtensions = ahb_techdraw_extensions.TechDrawExtensions()
|
||||
|
||||
self.initializeContext()
|
||||
self.registerCommands()
|
||||
pass
|
||||
|
||||
|
||||
@ -1,21 +0,0 @@
|
||||
from typing import Optional
|
||||
|
||||
import FreeCADGui as Gui
|
||||
import FreeCAD as App
|
||||
|
||||
class AHB_Animate:
|
||||
def GetResources(self):
|
||||
return {"MenuText": "Animate",
|
||||
"ToolTip": "Animate active stage to show all build stages",
|
||||
"Pixmap": ""
|
||||
}
|
||||
|
||||
def IsActive(self):
|
||||
return True
|
||||
|
||||
def Activated(self, stageId: Optional[int] = None):
|
||||
workbench = Gui.getWorkbench("AssemblyHandbookWorkbench")
|
||||
workbench.context.setAnimationActive(not workbench.context.getAnimationActive())
|
||||
|
||||
from ahb_command import AHB_CommandWrapper
|
||||
AHB_CommandWrapper.addGuiCommand('AHB_animate', AHB_Animate())
|
||||
@ -1,29 +0,0 @@
|
||||
import FreeCADGui as Gui
|
||||
import FreeCAD as App
|
||||
from PySide import QtGui
|
||||
|
||||
|
||||
class AHB_SetActiveStage:
|
||||
def GetResources(self):
|
||||
return {"MenuText": "Set active stage",
|
||||
"ToolTip": "Sets the stage which is the active stage, or creates a new stage if it doesn't exist",
|
||||
"Pixmap": ""
|
||||
}
|
||||
|
||||
def IsActive(self):
|
||||
return True
|
||||
|
||||
def Activated(self, stageId = None):
|
||||
if stageId is None:
|
||||
reply = QtGui.QInputDialog.getText(None, "Set Active Stage", "Enter the stage number:")
|
||||
if reply[1]:
|
||||
# user clicked OK
|
||||
stageId = int(reply[0])
|
||||
else:
|
||||
return
|
||||
|
||||
workbench = Gui.getWorkbench("AssemblyHandbookWorkbench")
|
||||
workbench.context.setActiveStage(stageId)
|
||||
|
||||
from ahb_command import AHB_CommandWrapper
|
||||
AHB_CommandWrapper.addGuiCommand('AHB_setActiveStage', AHB_SetActiveStage())
|
||||
@ -1,43 +0,0 @@
|
||||
from typing import Optional
|
||||
|
||||
import FreeCADGui as Gui
|
||||
import FreeCAD as App
|
||||
|
||||
class AHB_SetPartStage:
|
||||
def GetResources(self):
|
||||
return {"MenuText": "Set part stage",
|
||||
"ToolTip": "Sets the stage for this part to the currently active stage",
|
||||
"Pixmap": ""
|
||||
}
|
||||
|
||||
def IsActive(self):
|
||||
return True
|
||||
|
||||
def Activated(self, stageId: Optional[int] = None):
|
||||
workbench = Gui.getWorkbench("AssemblyHandbookWorkbench")
|
||||
if stageId is None:
|
||||
stageId = workbench.context.getActiveStage()
|
||||
if stageId is None:
|
||||
raise BaseException("You must activate a stage before using this command")
|
||||
|
||||
obj: App.DocumentObject
|
||||
for obj in Gui.Selection.getSelection():
|
||||
if obj is not None and obj.TypeId in ['Part::Feature','App::Part']:
|
||||
if not "AssemblyHandbook_Stage" in obj.PropertiesList:
|
||||
obj.addProperty("App::PropertyInteger", "AssemblyHandbook_Stage", "AssemblyHandbook")
|
||||
if not "AssemblyHandbook_RenderLines" in obj.PropertiesList:
|
||||
obj.addProperty("App::PropertyBool", "AssemblyHandbook_RenderLines", "AssemblyHandbook")
|
||||
obj.AssemblyHandbook_RenderLines = True
|
||||
|
||||
obj.AssemblyHandbook_Stage = stageId
|
||||
|
||||
# if we are setting the stage on a part, remove the property from all features of that part
|
||||
if obj.TypeId == 'App::Part':
|
||||
for feature in obj.Group:
|
||||
if 'AssemblyHandbook_Stage' in feature.PropertiesList:
|
||||
feature.removeProperty('AssemblyHandbook_Stage')
|
||||
|
||||
workbench.context.onPartStageChanged(obj)
|
||||
|
||||
from ahb_command import AHB_CommandWrapper
|
||||
AHB_CommandWrapper.addGuiCommand('AHB_setPartStage', AHB_SetPartStage())
|
||||
@ -1,21 +0,0 @@
|
||||
from typing import Optional
|
||||
|
||||
import FreeCADGui as Gui
|
||||
import FreeCAD as App
|
||||
|
||||
class AHB_SwitchVisibilityMode:
|
||||
def GetResources(self):
|
||||
return {"MenuText": "Switch visibility",
|
||||
"ToolTip": "Switch visibility mode",
|
||||
"Pixmap": ""
|
||||
}
|
||||
|
||||
def IsActive(self):
|
||||
return True
|
||||
|
||||
def Activated(self, stageId: Optional[int] = None):
|
||||
workbench = Gui.getWorkbench("AssemblyHandbookWorkbench")
|
||||
workbench.context.setAllStagesVisible(not workbench.context.getAllStagesVisible())
|
||||
|
||||
from ahb_command import AHB_CommandWrapper
|
||||
AHB_CommandWrapper.addGuiCommand('AHB_switchVisibilityMode', AHB_SwitchVisibilityMode())
|
||||
127
ahb_context.py
127
ahb_context.py
@ -1,127 +0,0 @@
|
||||
from typing import Optional
|
||||
|
||||
import FreeCADGui as Gui
|
||||
import FreeCAD as App
|
||||
from PySide.QtCore import QTimer
|
||||
import random
|
||||
|
||||
def _rgb(r, g, b):
|
||||
return r/255.0, g/255.0, b/255.0, 0.0
|
||||
|
||||
_colors = [
|
||||
_rgb(94, 224, 174),
|
||||
_rgb(223, 234, 70),
|
||||
_rgb(35, 120, 200),
|
||||
_rgb(188, 80, 33),
|
||||
_rgb(26, 242, 69),
|
||||
_rgb(13, 41, 107),
|
||||
_rgb(153, 9, 52),
|
||||
_rgb(211, 119, 0),
|
||||
_rgb(34, 130, 7),
|
||||
_rgb(221, 79, 249)
|
||||
]
|
||||
|
||||
class AHB_Context:
|
||||
_activeStageId: int = None
|
||||
_allStagesVisible: bool = True
|
||||
_animationActive: bool = False
|
||||
_timeSinceLastStageUpdate: int = 0
|
||||
|
||||
def setActiveStage(self, stageId):
|
||||
self._activeStageId = stageId
|
||||
self._updateVisibility()
|
||||
|
||||
def getActiveStage(self):
|
||||
return self._activeStageId
|
||||
|
||||
def getAllStages(self):
|
||||
allStages = set()
|
||||
|
||||
doc = App.ActiveDocument
|
||||
for obj in doc.Objects:
|
||||
if obj.TypeId in ['Part::Feature', 'App::Part']:
|
||||
if 'AssemblyHandbook_Stage' in obj.PropertiesList:
|
||||
allStages.add(obj.AssemblyHandbook_Stage)
|
||||
|
||||
allStages = list(allStages)
|
||||
allStages.sort()
|
||||
return allStages
|
||||
|
||||
def onPartStageChanged(self, part: Optional[App.DocumentObject]):
|
||||
allStages = self.getAllStages()
|
||||
|
||||
doc = App.ActiveDocument
|
||||
|
||||
if part is not None and not 'AssemblyHandbook_Stage' in part.PropertiesList:
|
||||
part.ViewObject.ShapeColor = (0.8,0.8,0.8,0.0)
|
||||
|
||||
for obj in doc.Objects:
|
||||
if obj.TypeId in ['Part::Feature', 'App::Part']:
|
||||
if 'AssemblyHandbook_Stage' in obj.PropertiesList:
|
||||
stageIndex = allStages.index(obj.AssemblyHandbook_Stage)
|
||||
if obj.TypeId == 'Part::Feature':
|
||||
obj.ViewObject.ShapeColor = _colors[stageIndex % len(_colors)]
|
||||
elif obj.TypeId == 'App::Part':
|
||||
for feature in obj.Group:
|
||||
if feature.TypeId == 'Part::Feature':
|
||||
feature.ViewObject.ShapeColor = _colors[stageIndex % len(_colors)]
|
||||
|
||||
self._updateVisibility()
|
||||
|
||||
def setAllStagesVisible(self, allStagesVisible):
|
||||
self._allStagesVisible = allStagesVisible
|
||||
self._updateVisibility()
|
||||
|
||||
def getAllStagesVisible(self):
|
||||
return self._allStagesVisible
|
||||
|
||||
def _updateVisibility(self):
|
||||
activeStageId = self.getActiveStage()
|
||||
|
||||
doc = App.ActiveDocument
|
||||
for obj in doc.Objects:
|
||||
if obj.TypeId == 'Part::Feature' in ['Part::Feature', 'App::Part']:
|
||||
if 'AssemblyHandbook_Stage' in obj.PropertiesList:
|
||||
obj.ViewObject.Visibility = self._allStagesVisible or activeStageId is None or obj.AssemblyHandbook_Stage <= activeStageId
|
||||
|
||||
def setAnimationActive(self, animationActive):
|
||||
if self._animationActive == animationActive: return
|
||||
self._animationActive = animationActive
|
||||
self._updateAnimation()
|
||||
|
||||
def getAnimationActive(self):
|
||||
return self._animationActive
|
||||
|
||||
def _updateAnimation(self):
|
||||
if not self._animationActive: return
|
||||
|
||||
allStages = self.getAllStages()
|
||||
if len(allStages) > 1 and self._timeSinceLastStageUpdate >= 1000:
|
||||
self._timeSinceLastStageUpdate = 0
|
||||
activeStageId = self.getActiveStage() or 0
|
||||
stageIndex = allStages.index(activeStageId) if activeStageId in allStages else 0
|
||||
stageIndex = (stageIndex + 1) % len(allStages)
|
||||
self.setActiveStage(allStages[stageIndex])
|
||||
|
||||
tickResolution = 33
|
||||
|
||||
import math
|
||||
from pivy import coin
|
||||
cam = Gui.ActiveDocument.ActiveView.getCameraNode()
|
||||
"""rot = coin.SbRotation()
|
||||
rot.setValue(coin.SbVec3f(0, 0, 1), math.pi * tickResolution / 1000.0)
|
||||
nrot = cam.orientation.getValue() * rot
|
||||
cam.orientation = nrot"""
|
||||
|
||||
center = coin.SbVec3f(1000,0,750)
|
||||
v = cam.position.getValue() - center
|
||||
v[2] = 0
|
||||
dist = v.length()
|
||||
height = cam.position.getValue()[2]
|
||||
angle = math.atan2(v[1], v[0])
|
||||
angle = angle + 0.3 * tickResolution / 1000.0
|
||||
cam.position = coin.SbVec3f(center[0] + math.cos(angle)*dist, center[1] + math.sin(angle)*dist, height)
|
||||
cam.pointAt(center, coin.SbVec3f(0,0,1))
|
||||
|
||||
self._timeSinceLastStageUpdate = self._timeSinceLastStageUpdate + tickResolution
|
||||
QTimer.singleShot(tickResolution, self._updateAnimation)
|
||||
@ -0,0 +1 @@
|
||||
# this module is only used to help other modules to locate its folder
|
||||
Loading…
Reference in New Issue
Block a user