AddinLib/CustomGraphics.py at main · Zxynine/AddinLib · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142

from __future__ import annotations
import adsk.core, adsk.fusion
from adsk.core import Color as Colour
from enum import Enum


from. import AppObjects,utils


class VisualEffects(Enum):
	BasicMonoChrome = adsk.fusion.CustomGraphicsSolidColorEffect.create
	Overlayed = adsk.fusion.CustomGraphicsShowThroughColorEffect.create
	FusionApperence = adsk.fusion.CustomGraphicsAppearanceColorEffect.create
	VertexShading = adsk.fusion.CustomGraphicsVertexColorEffect.create
	ShadedMaterial = adsk.fusion.CustomGraphicsBasicMaterialColorEffect.create


class LineStyles(Enum):
	Solid = adsk.fusion.LineStylePatterns.continuousLineStylePattern
	DashDotted = adsk.fusion.LineStylePatterns.centerLineStylePattern
	Dashed = adsk.fusion.LineStylePatterns.dashedLineStylePattern
	Dotted = adsk.fusion.LineStylePatterns.dotLineStylePattern
	DashDoubleDotted = adsk.fusion.LineStylePatterns.phantomLineStylePattern
	Ribbed = adsk.fusion.LineStylePatterns.tracksLineStylePattern
	ZigZagged = adsk.fusion.LineStylePatterns.zigzagLineStylePattern


def getMaterialLibNames(libFilter):
	materialLibs = AppObjects.GetApp().materialLibraries
	libNames = []
	for materialLib in materialLibs:
		if (not libFilter) or libFilter(materialLib): libNames.append(materialLib.name)
	return libNames


def hasAppearances(lib:adsk.core.MaterialLibrary): return (lib and lib.appearances.count > 0)
def getAppearance(libName, appearanceName):
		if not appearanceName or appearanceName == 'None': return
		appearance = AppObjects.GetDesign().appearances.itemByName(appearanceName)
		if appearance: return appearance
		matLib = AppObjects.GetApp().materialLibraries.itemByName(libName)
		if matLib: appearance = matLib.appearances.itemByName(appearanceName)
		return appearance


appearancesMap = {}
def getAppearancesFromLib(libName, filterExp:str):
		global appearancesMap
		appearanceList = None
		if libName in appearancesMap: appearanceList = appearancesMap[libName]
		else:
			materialLib = AppObjects.GetApp().materialLibraries.itemByName(libName)
			appearances = utils.Items(materialLib.appearances)
			appearanceList = [apperance.name for apperance in appearances]
			appearancesMap[libName] = appearanceList

		if filterExp and len(filterExp) > 0:
			filteredList = []
			for appearanceName in appearanceList:
				if appearanceName.lower().find(filterExp.lower()) >= 0: filteredList.append(appearanceName)
			return filteredList
		else: return appearanceList


TransparentRed = Colour.create(255,0,0,50)
RemovalRed = adsk.fusion.CustomGraphicsShowThroughColorEffect.create(TransparentRed,0.25)
SolidBlue = Colour.create(50,100,175,255)
SelectedBlue = adsk.fusion.CustomGraphicsShowThroughColorEffect.create(SolidBlue,0.25)


def ClearCustomGraphics(root:adsk.fusion.Component,returnNew=False):
	if root.customGraphicsGroups.count > 0:
		[group.deleteMe() for group in root.customGraphicsGroups]
		AppObjects.GetApp().activeViewport.refresh()
	if returnNew:return root.customGraphicsGroups.add()


def MeshFromBRep(graphics:adsk.fusion.CustomGraphicsGroup, object:adsk.fusion.BRepBody,colour:Colour=None):
	objectGroup = graphics.addGroup()
	allMeshes:list[adsk.fusion.TriangleMesh] = object.meshManager.displayMeshes
	for mesh in allMeshes:
		meshCoords = adsk.fusion.CustomGraphicsCoordinates.create(mesh.nodeCoordinatesAsDouble)
		graphicObj = objectGroup.addMesh(meshCoords,mesh.nodeIndices, [],[])
		if colour: graphicObj.color = colour
	return objectGroup


class CustomGraphics:
	def __init__(self, root: adsk.fusion.Component=None):
		if root is None: root = AppObjects.GetRoot()
		self.graphics = ClearCustomGraphics(root,True)
		self.root = root
	def clear(self):return self.graphics.deleteMe()


def DrawLine(curve):
	pass


# def DrawGraphics():
# 	des = adsk.fusion.Design.cast(app_.activeProduct)
# 	root = des.rootComponent
# 	# Check to see if a custom graphics groups already exists and delete it.
# 	graphics = ClearCustomGraphics(root,True)

# 	allFaces:'list[adsk.fusion.BRepFace]' = [face_selection_input.selection(face).entity for face in range(face_selection_input.selectionCount)]

# 	interSections:'list[adsk.core.InfiniteLine3D]' = []
# 	for i in range(-1,len(allFaces)-1):
# 		PlanarGeom : adsk.core.Plane= allFaces[i].geometry
# 		interSections.append(PlanarGeom.intersectWithPlane(allFaces[i+1].geometry))

# 	tryLines = []
# 	for line in interSections:
# 		start = line.origin
# 		end = start.asVector()
# 		direction = line.direction
# 		direction.scaleBy(30)
# 		end.add(direction)
# 		end = end.asPoint()

# 		tryLines.append(adsk.core.Line3D.create(line.origin,end))
# 		graphics.addCurve(tryLines[-1])