CYCLES_RENDER_PT_bake.py

# 「プロパティ」エリア > 「レンダー」タブ > 「ベイク」パネル (Cyclesのみ)
# "Propaties" Area > "Render" Tab > "Bake" Panel (Cycles Only)

import bpy
from bpy.props import *

################
# オペレーター #
################

class NewBakeImage(bpy.types.Operator):
	bl_idname = "image.new_bake_image"
	bl_label = "Crate New Image for Bake"
	bl_description = "Create a new image to use for bake"
	bl_options = {'REGISTER', 'UNDO'}

	name : StringProperty(name="Name", default="Bake")
	width : IntProperty(name="Width", default=1024, min=1, max=8192, soft_min=1, soft_max=8192, step=1, subtype='PIXEL')
	height : IntProperty(name="Height", default=1024, min=1, max=8192, soft_min=1, soft_max=8192, step=1, subtype='PIXEL')
	alpha : BoolProperty(name="Use alpha", default=True)
	float : BoolProperty(name="Use floating point color", default=False)
	show_image : BoolProperty(name="Show Image", default=True)

	@classmethod
	def poll(cls, context):
		if (context.active_object):
			if (context.active_object.type == 'MESH'):
				if (len(context.active_object.data.uv_layers)):
					return True
		return False

	def invoke(self, context, event):
		return context.window_manager.invoke_props_dialog(self)

	def execute(self, context):
		new_image = bpy.data.images.new(self.name, self.width, self.height, alpha=self.alpha, float_buffer=self.float)
		for mslot in bpy.context.active_object.material_slots:
			mslot.material.use_nodes = True
			node_tex = mslot.material.node_tree.nodes.new('ShaderNodeTexImage')
			node_tex.image = new_image
			node_tex.location = [node_tex.location[0]-300, node_tex.location[1]+300]
		if (self.show_image):
			max = -1
			for area in context.screen.areas:
				if (area.type == 'IMAGE_EDITOR'):
					image_area = area
					image_area.spaces[0].image = new_image
					break
				elif (area.type != 'VIEW_3D'):
					size = area.width * area.height
					if (max < size):
						image_area = area
						max = size
			else:
				bpy.ops.screen.userpref_show()
				new_window = context.window_manager.windows[-1]
				image_area = new_window.screen.areas[-1]
				image_area.type = 'IMAGE_EDITOR'
				image_area.spaces[0].image = new_image
				bpy.ops.image.view_zoom_ratio(ratio=0.5)
		return {'FINISHED'}

################
# クラスの登録 #
################

classes = [
	NewBakeImage
]

def register():
	for cls in classes:
		bpy.utils.register_class(cls)

def unregister():
	for cls in classes:
		bpy.utils.unregister_class(cls)


################
# メニュー追加 #
################

# メニューのオン/オフの判定
def IsMenuEnable(self_id):
	for id in bpy.context.preferences.addons[__name__.partition('.')[0]].preferences.disabled_menu.split(','):
		if (id == self_id):
			return False
	else:
		return True

# メニューを登録する関数
def menu(self, context):
	if (IsMenuEnable(__name__.split('.')[-1])):
		self.layout.operator(NewBakeImage.bl_idname, icon='IMAGE_DATA')
	if (context.preferences.addons[__name__.partition('.')[0]].preferences.use_disabled_menu):
		self.layout.operator('wm.toggle_menu_enable', icon='CANCEL').id = __name__.split('.')[-1]