2025-07-01

2026-03-17 14:30:01 -06:00
parent f9a22056dd
commit 62b5978595
4579 changed files with 1257472 additions and 0 deletions
@@ -0,0 +1,167 @@
+import bpy
+import os
+
+from . import utilities_bake
+from . import op_bake
+from .t3dn_bip import previews
+
+from bpy.props import StringProperty
+
+class op_popup(bpy.types.Operator):
+	bl_idname = "ui.textools_popup"
+	bl_label = "Message"
+
+	message : StringProperty()
+
+	def execute(self, context):
+		self.report({'INFO'}, self.message)
+		print(self.message)
+		return {'FINISHED'}
+
+	def invoke(self, context, event):
+		wm = context.window_manager
+		return wm.invoke_popup(self, width=200)
+
+	def draw(self, context):
+		self.layout.label(text=self.message)
+
+size_textures = [
+		('32', '32', ''), 
+		('64', '64', ''), 
+		('128', '128', ''), 
+		('256', '256', ''), 
+		('512', '512', ''), 
+		('1024', '1024', ''), 
+		('2048', '2048', ''), 
+		('4096', '4096', ''),
+		('8192', '8192', '')
+	]
+
+# Preview collections created in the register function.
+preview_icons = previews.new(max_size=(32, 32))
+thumbnail_previews = None
+
+
+def icon_register(fileName):
+	name = fileName.split('.')[0]	# Don't include file extension
+	icons_location = os.path.join(os.path.dirname(__file__), "icons_bip")
+	preview_icons.load_safe(name, os.path.join(icons_location, fileName), 'IMAGE')
+
+
+def icon_get(name):
+	return preview_icons[name].icon_id
+
+
+def generate_bake_mode_previews():
+	image_location = os.path.join(os.path.dirname(__file__), "resources/bake_modes_bip")	
+	enum_items = []
+	
+	# Generate the thumbnails
+	for i, image in enumerate(os.listdir(image_location)):
+		mode = image[0:-4]
+		if mode in op_bake.modes:
+			filepath = os.path.join(image_location, image)
+			thumb = thumbnail_previews.load_safe(filepath, filepath, 'IMAGE')
+			enum_items.append((image, mode, "", thumb.icon_id, i))
+	
+	return enum_items
+
+
+
+def GetContextView3D():
+	for window in bpy.context.window_manager.windows:
+		screen = window.screen
+		for area in screen.areas:
+			if area.type == 'VIEW_3D': 
+				for region in area.regions:
+					if region.type == 'WINDOW':
+						override = {'window': window, 'screen': screen, 'area': area, 'region': region, 'scene': bpy.context.scene, 'edit_object': bpy.context.edit_object, 'active_object': bpy.context.active_object, 'selected_objects': bpy.context.selected_objects}   # Stuff the override context with very common requests by operators.  MORE COULD BE NEEDED!
+						return override					
+	return None
+
+
+
+def GetContextViewUV():
+	for window in bpy.context.window_manager.windows:
+		screen = window.screen
+		for area in screen.areas:
+			if area.ui_type == 'UV':
+				for region in area.regions:
+					if region.type == 'WINDOW': 
+						override = {'window': window, 'screen': screen, 'area': area, 'region': region, 'scene': bpy.context.scene, 'edit_object': bpy.context.edit_object, 'active_object': bpy.context.active_object, 'selected_objects': bpy.context.selected_objects}   # Stuff the override context with very common requests by operators.  MORE COULD BE NEEDED!
+						return override			
+	return None
+
+
+
+def get_padding():
+	size_min = min(bpy.context.scene.texToolsSettings.size[0],bpy.context.scene.texToolsSettings.size[1])
+	return bpy.context.scene.texToolsSettings.padding / size_min
+
+
+
+def get_bake_mode():
+	return str(bpy.context.scene.TT_bake_mode).replace(".bip","").lower()
+
+
+
+def set_bake_color_space_int(bake_mode):
+	preferences = bpy.context.preferences.addons[__package__].preferences
+	if "normal_" in bake_mode:
+		if preferences.bake_color_space_def == 'ASTANDARD' or preferences.bake_color_space_def == 'APBR':
+			return 3
+		else:
+			return 1
+	elif preferences.bake_color_space_def == 'STANDARD':
+		return 0
+	elif preferences.bake_color_space_def == 'PBR':
+		if op_bake.modes[bake_mode].material != "" or (bake_mode =='transmission' and not preferences.bool_clean_transmission) or bake_mode in {'diffuse','base_color','sss_color','emission','environment','combined'}:
+			return 0
+		return 1
+	elif preferences.bake_color_space_def == 'ASTANDARD':
+		return 2
+	elif preferences.bake_color_space_def == 'APBR':
+		if op_bake.modes[bake_mode].material != "" or (bake_mode =='transmission' and not preferences.bool_clean_transmission) or bake_mode in {'diffuse','base_color','sss_color','emission','environment','combined'}:
+			return 2
+		return 3
+
+
+
+def on_bakemode_set(self, context):
+	bake_mode = get_bake_mode()
+	if set_bake_color_space_int(bake_mode) == 1:
+		bpy.context.scene.texToolsSettings.bake_color_space = 'Non-Color'
+	elif set_bake_color_space_int(bake_mode) == 0:
+		bpy.context.scene.texToolsSettings.bake_color_space = 'sRGB'
+	elif set_bake_color_space_int(bake_mode) == 3:
+		bpy.context.scene.texToolsSettings.bake_color_space = 'Utility - Linear - sRGB'
+	else:
+		bpy.context.scene.texToolsSettings.bake_color_space = 'Utility - sRGB - Texture'
+	utilities_bake.on_select_bake_mode(bake_mode)
+
+
+
+def register():
+	global thumbnail_previews
+	thumbnail_previews = previews.new(max_size=(128, 128))
+
+	# This is an EnumProperty to hold all of the images
+	# You really can save it anywhere in bpy.types.*  Just make sure the location makes sense
+	bpy.types.Scene.TT_bake_mode = bpy.props.EnumProperty(
+		items=generate_bake_mode_previews(),
+		update = on_bakemode_set,
+		default = 'normal_tangent.bip'
+	)
+
+
+def unregister():
+	previews.remove(thumbnail_previews)
+	previews.remove(preview_icons)
+
+	del bpy.types.Scene.TT_bake_mode
+
+
+if __name__ == "__main__":
+	register()
+
+bpy.utils.register_class(op_popup)