def invoke(self, context, event):
self.objs.clear()
settings = get_settings()
libpath = settings.drill_lib
assets = odcutils.obj_list_from_lib(libpath, exclude='Drill')
for asset_object_name in assets:
self.objs.add().name = asset_object_name
context.window_manager.invoke_search_popup(self)
return {'FINISHED'}
def invoke(self, context, event):
self.objs.clear()
#here we grab the asset library from the addon prefs
settings = get_settings()
libpath = settings.imp_lib
assets = odcutils.obj_list_from_lib(libpath, exclude='_')
for asset_object_name in assets:
self.objs.add().name = asset_object_name
#context.window_manager.invoke_search_popup(self.ob_list)
context.window_manager.invoke_search_popup(self)
return {'FINISHED'}
def invoke(self, context, event):
settings = get_settings()
libpath = settings.ortho_lib
assets = obj_list_from_lib(libpath)
if settings.bracket in assets:
current_obs = [ob.name for ob in bpy.data.objects]
obj_from_lib(settings.ortho_lib, settings.bracket)
for ob in bpy.data.objects:
if ob.name not in current_obs:
Bracket = ob
Bracket.hide = False
context.scene.objects.link(Bracket)
else:
Bracket = None
if context.object and context.object.type == 'MESH':
self.bracket_manager = BracketDataManager(
context,
snap_type='OBJECT',
snap_object=context.object,
name='Bracket',
bracket=Bracket)
self.bracket_slicer = BracektSlicer(context, self.bracket_manager)
else:
self.bracket_manager = BracketDataManager(context,
snap_type='SCENE',
snap_object=None,
name='Bracket',
bracket=Bracket)
self.bracket_slicer = None
help_txt = "DRAW MARGIN OUTLINE\n\nLeft Click on model to place bracket.\n G to grab \n S to show slice \n ENTER to confirm \n ESC to cancel"
self.help_box = TextBox(context, 500, 500, 300, 200, 10, 20, help_txt)
self.help_box.snap_to_corner(context, corner=[1, 1])
self.mode = 'start'
self._handle = bpy.types.SpaceView3D.draw_handler_add(
bracket_placement_draw_callback, (self, context), 'WINDOW',
'POST_PIXEL')
context.window_manager.modal_handler_add(self)
return {'RUNNING_MODAL'}
def update_link_operators():
#unregister get crown
bpy.utils.unregister_class(OPENDENTAL_OT_place_implant)
bpy.utils.unregister_class(OPENDENTAL_OT_implant_from_contour)
#redefinte list from user prefs prop
global lib_teeth_enum
global lib_teeth
lib_teeth_enum = []
lib_teeth = []
settings = get_settings()
dbg = settings.debug
lib_teeth = odcutils.obj_list_from_lib(settings.tooth_lib,
exclude='_',
debug=dbg)
for ind, obj in enumerate(lib_teeth):
lib_teeth_enum.append((str(ind), obj, str(ind)))
#reregister.
bpy.utils.register_class(OPENDENTAL_OT_place_implant)
bpy.utils.register_class(OPENDENTAL_OT_implant_from_contour)
def place_implant(context,
implant_space,
location,
orientation,
imp,
hardware=True):
'''
args:
context
implant_space - ODC Implant Space type
location - Vector
orientation - Matrix or Quaternion
lib_implants -
imp - string representing implant object name in link library
'''
#check if space already has an implant object.
#if so, delete, replace, print warning
sce = context.scene
if implant_space.implant and implant_space.implant in bpy.data.objects:
print("replacing the existing implant with the one you chose")
Implant = bpy.data.objects[implant_space.implant]
#unlink it from the scene, clear it's useres, remove it.
if Implant.children:
for child in Implant.children:
sce.objects.unlink(child)
child.user_clear
bpy.data.objects.remove(child)
sce.objects.unlink(Implant)
implant_mesh = Implant.data
Implant.user_clear()
#remove the object
bpy.data.objects.remove(Implant)
implant_mesh.user_clear()
bpy.data.meshes.remove(implant_mesh)
sce.update()
#TDOD what about the children/hardwares?
world_mx = Matrix.Identity(4)
world_mx[0][3] = location[0]
world_mx[1][3] = location[1]
world_mx[2][3] = location[2]
#mx_b = Matrix.Identity(4)
#mx_l = Matrix.Identity(4)
#is this more memory friendly than listing all objects?
current_obs = [ob.name for ob in bpy.data.objects]
#link the new implant from the library
settings = get_settings()
odcutils.obj_from_lib(settings.imp_lib, imp)
#this is slightly more robust than trusting we don't have duplicate names.
for ob in bpy.data.objects:
if ob.name not in current_obs:
Implant = ob
sce.objects.link(Implant)
#Implant.matrix_basis = mx_b
Implant.matrix_world = world_mx
Implant.update_tag()
sce.update()
Implant.rotation_mode = 'QUATERNION'
Implant.rotation_quaternion = orientation
sce.update()
#Implant.matrix_local = mx_l
#Implant.location = L
if sce.odc_props.master:
Master = bpy.data.objects[sce.odc_props.master]
odcutils.parent_in_place(Implant, Master)
else:
print(
'No Master Model, placing implant anyway, moving objects may not preserve spatial relationships'
)
#looks a little redundant, but it ensure if any
#duplicates exist our referencing stays accurate
Implant.name = implant_space.name + "_" + Implant.name
implant_space.implant = Implant.name
if hardware:
current_obs = [ob.name for ob in bpy.data.objects]
inc = imp + '_'
settings = get_settings()
hardware_list = odcutils.obj_list_from_lib(settings.imp_lib,
include=inc)
print(hardware_list)
for ob in hardware_list:
odcutils.obj_from_lib(settings.imp_lib, ob)
for ob in bpy.data.objects:
if ob.name not in current_obs:
sce.objects.link(ob)
ob.parent = Implant
ob.layers[11] = True #TODO: put this in layer management.
return Implant
def execute(self, context):
settings = get_settings()
dbg = settings.debug
#if bpy.context.mode != 'OBJECT':
# bpy.ops.object.mode_set(mode = 'OBJECT')
sce = context.scene
#n = sce.odc_implant_index
#implant_space = sce.odc_implants[n]
implants = odcutils.implant_selection(context)
layers_copy = [layer for layer in context.scene.collection.all_objects]
context.scene.collection.all_objects[0]
if implants != []:
for implant_space in implants:
#check if space already has an implant object.
#if so, delete, replace, print warning
if implant_space.implant and implant_space.implant in bpy.data.objects:
self.report({
'WARNING'
}, "replacing the existing implant with the one you chose")
Implant = bpy.data.objects[implant_space.implant]
#the origin/location of the implant is it's apex
L = Implant.location.copy()
world_mx = Implant.matrix_world.copy()
#the platorm is the length of the implant above the apex, in the local Z direction
#local Z positive is out the apex, soit's negative.
#Put the cursor there
sce.cursor.location = L - Implant.dimensions[
2] * world_mx.to_3x3() @ Vector((0, 0, 1))
#first get rid of children...so we can use the
#parent to find out who the children are
if Implant.children:
for child in Implant.children:
sce.objects.unlink(child)
child.user_clear()
bpy.data.objects.remove(child)
#unlink it from the scene, clear it's users, remove it.
sce.objects.unlink(Implant)
Implant.user_clear()
#remove the object
bpy.data.objects.remove(Implant)
#TDOD what about the children/hardwares?
else:
world_mx = Matrix.Identity(4)
world_mx[0][3] = sce.cursor.location[0]
world_mx[1][3] = sce.cursor.location[1]
world_mx[2][3] = sce.cursor.location[2]
#is this more memory friendly than listing all objects?
current_obs = [ob.name for ob in bpy.data.objects]
#link the new implant from the library
odcutils.obj_from_lib(settings.imp_lib, self.imp)
#this is slightly more robust than trusting we don't have duplicate names.
for ob in bpy.data.objects:
if ob.name not in current_obs:
Implant = ob
context.collection.objects.link(Implant)
#this relies on the associated hardware objects having the parent implant
#name inside them
if self.hardware:
current_obs = [ob.name for ob in bpy.data.objects]
inc = self.imp + '_'
hardware_list = odcutils.obj_list_from_lib(
settings.imp_lib, include=inc)
print(hardware_list)
for ob in hardware_list:
odcutils.obj_from_lib(settings.imp_lib, ob)
for ob in bpy.data.objects:
if ob.name not in current_obs:
context.collection.objects.link(ob)
ob.parent = Implant
ob.layers[11] = True
delta = Implant.dimensions[2] * world_mx.to_3x3() @ Vector(
(0, 0, 1))
print(delta.length)
world_mx[0][3] += delta[0]
world_mx[1][3] += delta[1]
world_mx[2][3] += delta[2]
Implant.matrix_world = world_mx
if sce.odc_props.master:
Master = bpy.data.objects[sce.odc_props.master]
odcutils.parent_in_place(Implant, Master)
else:
self.report({
'WARNING'
}, 'No Master Model, placing implant anyway, moving objects may not preserve spatial relationships'
)
#looks a little redundant, but it ensure if any
#duplicates exist our referencing stays accurate
Implant.name = implant_space.name + '_' + Implant.name
implant_space.implant = Implant.name
odcutils.layer_management(sce.odc_implants, debug=dbg)
else:
world_mx = Matrix.Identity(4)
world_mx[0][3] = sce.cursor.location[0]
world_mx[1][3] = sce.cursor.location[1]
world_mx[2][3] = sce.cursor.location[2]
#is this more memory friendly than listing all objects?
current_obs = [ob.name for ob in bpy.data.objects]
#link the new implant from the library
odcutils.obj_from_lib(settings.imp_lib, self.imp)
#this is slightly more robust than trusting we don't have duplicate names.
for ob in bpy.data.objects:
if ob.name not in current_obs:
Implant = ob
context.collection.objects.link(Implant)
Implant.matrix_world = world_mx
for i, layer in enumerate(layers_copy):
context.scene.collection.all_objects[i] = layer
context.scene.collection.all_objects[11] = True
return {'FINISHED'}
