yyc12345
4ba3ff9e5a
- fix the generic error of EnumPropHelper. - use EnumPropHelper in UTIL_ioport_shared.ExportParams field instead of raw Blender string. - remove useless type hint in various modules.
206 lines
9.6 KiB
Python
206 lines
9.6 KiB
Python
import bpy
|
|
import typing
|
|
from . import UTIL_virtools_types, UTIL_functions
|
|
|
|
class RawVirtoolsTexture():
|
|
|
|
# Instance Member Declarations
|
|
|
|
mSaveOptions: UTIL_virtools_types.CK_TEXTURE_SAVEOPTIONS
|
|
mVideoFormat: UTIL_virtools_types.VX_PIXELFORMAT
|
|
|
|
# Default Value Declarations
|
|
|
|
cDefaultSaveOptions: typing.ClassVar[UTIL_virtools_types.CK_TEXTURE_SAVEOPTIONS] = UTIL_virtools_types.CK_TEXTURE_SAVEOPTIONS.CKTEXTURE_RAWDATA
|
|
cDefaultVideoFormat: typing.ClassVar[UTIL_virtools_types.VX_PIXELFORMAT] = UTIL_virtools_types.VX_PIXELFORMAT._16_ARGB1555
|
|
|
|
def __init__(self, **kwargs):
|
|
# assign default value for each component
|
|
self.mSaveOptions = kwargs.get('mSaveOptions', RawVirtoolsTexture.cDefaultSaveOptions)
|
|
self.mVideoFormat = kwargs.get('mVideoFormat', RawVirtoolsTexture.cDefaultVideoFormat)
|
|
|
|
# blender enum prop helper defines
|
|
_g_Helper_CK_TEXTURE_SAVEOPTIONS = UTIL_virtools_types.EnumPropHelper(UTIL_virtools_types.CK_TEXTURE_SAVEOPTIONS)
|
|
_g_Helper_VX_PIXELFORMAT = UTIL_virtools_types.EnumPropHelper(UTIL_virtools_types.VX_PIXELFORMAT)
|
|
|
|
class BBP_PG_virtools_texture(bpy.types.PropertyGroup):
|
|
|
|
save_options: bpy.props.EnumProperty(
|
|
name = "Save Options",
|
|
description = "When saving a composition textures or sprites can be kept as reference to external files or converted to a given format and saved inside the composition file.",
|
|
items = _g_Helper_CK_TEXTURE_SAVEOPTIONS.generate_items(),
|
|
default = _g_Helper_CK_TEXTURE_SAVEOPTIONS.to_selection(RawVirtoolsTexture.cDefaultSaveOptions),
|
|
translation_context = 'BBP_PG_virtools_texture/property'
|
|
) # type: ignore
|
|
|
|
video_format: bpy.props.EnumProperty(
|
|
name = "Video Format",
|
|
description = "The desired surface pixel format in video memory.",
|
|
items = _g_Helper_VX_PIXELFORMAT.generate_items(),
|
|
default = _g_Helper_VX_PIXELFORMAT.to_selection(RawVirtoolsTexture.cDefaultVideoFormat),
|
|
translation_context = 'BBP_PG_virtools_texture/property'
|
|
) # type: ignore
|
|
|
|
#region Virtools Texture Getter Setter
|
|
|
|
def get_virtools_texture(img: bpy.types.Image) -> BBP_PG_virtools_texture:
|
|
return img.virtools_texture
|
|
|
|
def get_raw_virtools_texture(img: bpy.types.Image) -> RawVirtoolsTexture:
|
|
props: BBP_PG_virtools_texture = get_virtools_texture(img)
|
|
rawdata: RawVirtoolsTexture = RawVirtoolsTexture()
|
|
|
|
rawdata.mSaveOptions = _g_Helper_CK_TEXTURE_SAVEOPTIONS.get_selection(props.save_options)
|
|
rawdata.mVideoFormat = _g_Helper_VX_PIXELFORMAT.get_selection(props.video_format)
|
|
return rawdata
|
|
|
|
def set_raw_virtools_texture(img: bpy.types.Image, rawdata: RawVirtoolsTexture) -> None:
|
|
props: BBP_PG_virtools_texture = get_virtools_texture(img)
|
|
|
|
props.save_options = _g_Helper_CK_TEXTURE_SAVEOPTIONS.to_selection(rawdata.mSaveOptions)
|
|
props.video_format = _g_Helper_VX_PIXELFORMAT.to_selection(rawdata.mVideoFormat)
|
|
|
|
#endregion
|
|
|
|
#region Virtools Texture Drawer
|
|
|
|
# YYC MARK:
|
|
# Because Image do not have its unique properties window,
|
|
# so we only can draw Virtools Texture properties in other window.
|
|
# We provide various functions to help draw properties.
|
|
|
|
def draw_virtools_texture(img: bpy.types.Image, layout: bpy.types.UILayout):
|
|
props: BBP_PG_virtools_texture = get_virtools_texture(img)
|
|
|
|
layout.prop(props, 'save_options')
|
|
layout.prop(props, 'video_format')
|
|
|
|
#endregion
|
|
|
|
#region Ballance Texture Preset
|
|
|
|
_g_OpaqueBallanceTexturePreset: RawVirtoolsTexture = RawVirtoolsTexture(
|
|
mSaveOptions = UTIL_virtools_types.CK_TEXTURE_SAVEOPTIONS.CKTEXTURE_EXTERNAL,
|
|
mVideoFormat = UTIL_virtools_types.VX_PIXELFORMAT._16_ARGB1555,
|
|
)
|
|
_g_TransparentBallanceTexturePreset: RawVirtoolsTexture = RawVirtoolsTexture(
|
|
mSaveOptions = UTIL_virtools_types.CK_TEXTURE_SAVEOPTIONS.CKTEXTURE_EXTERNAL,
|
|
mVideoFormat = UTIL_virtools_types.VX_PIXELFORMAT._32_ARGB8888,
|
|
)
|
|
_g_NonBallanceTexturePreset: RawVirtoolsTexture = RawVirtoolsTexture(
|
|
mSaveOptions = UTIL_virtools_types.CK_TEXTURE_SAVEOPTIONS.CKTEXTURE_RAWDATA,
|
|
mVideoFormat = UTIL_virtools_types.VX_PIXELFORMAT._32_ARGB8888,
|
|
)
|
|
|
|
## The preset collection of all Ballance texture.
|
|
# Key is texture name and can be used as file name checking.
|
|
# Value is its preset which can be assigned.
|
|
_g_BallanceTexturePresets: dict[str, RawVirtoolsTexture] = {
|
|
# "atari.avi": _g_TransparentBallanceTexturePreset,
|
|
"atari.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Ball_LightningSphere1.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Ball_LightningSphere2.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Ball_LightningSphere3.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Ball_Paper.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Ball_Stone.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Ball_Wood.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Brick.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Button01_deselect.tga": _g_TransparentBallanceTexturePreset,
|
|
"Button01_select.tga": _g_TransparentBallanceTexturePreset,
|
|
"Button01_special.tga": _g_TransparentBallanceTexturePreset,
|
|
"Column_beige.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Column_beige_fade.tga": _g_TransparentBallanceTexturePreset,
|
|
"Column_blue.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Cursor.tga": _g_TransparentBallanceTexturePreset,
|
|
"Dome.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"DomeEnvironment.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"DomeShadow.tga": _g_TransparentBallanceTexturePreset,
|
|
"ExtraBall.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"ExtraParticle.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"E_Holzbeschlag.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"FloorGlow.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Floor_Side.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Floor_Top_Border.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Floor_Top_Borderless.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Floor_Top_Checkpoint.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Floor_Top_Flat.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Floor_Top_Profil.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Floor_Top_ProfilFlat.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Font_1.tga": _g_TransparentBallanceTexturePreset,
|
|
"Gravitylogo_intro.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"HardShadow.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Laterne_Glas.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Laterne_Schatten.tga": _g_TransparentBallanceTexturePreset,
|
|
"Laterne_Verlauf.tga": _g_TransparentBallanceTexturePreset,
|
|
"Logo.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Metal_stained.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Misc_Ufo.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Misc_UFO_Flash.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Modul03_Floor.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Modul03_Wall.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Modul11_13_Wood.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Modul11_Wood.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Modul15.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Modul16.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Modul18.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Modul18_Gitter.tga": _g_TransparentBallanceTexturePreset,
|
|
"Modul30_d_Seiten.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Particle_Flames.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Particle_Smoke.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"PE_Bal_balloons.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"PE_Bal_platform.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"PE_Ufo_env.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Pfeil.tga": _g_TransparentBallanceTexturePreset,
|
|
"P_Extra_Life_Oil.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"P_Extra_Life_Particle.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"P_Extra_Life_Shadow.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Rail_Environment.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"sandsack.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"SkyLayer.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Sky_Vortex.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Stick_Bottom.tga": _g_TransparentBallanceTexturePreset,
|
|
"Stick_Stripes.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Target.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Tower_Roof.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Trafo_Environment.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Trafo_FlashField.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Trafo_Shadow_Big.tga": _g_TransparentBallanceTexturePreset,
|
|
"Tut_Pfeil01.tga": _g_TransparentBallanceTexturePreset,
|
|
"Tut_Pfeil_Hoch.tga": _g_TransparentBallanceTexturePreset,
|
|
"Wolken_intro.tga": _g_TransparentBallanceTexturePreset,
|
|
"Wood_Metal.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Wood_MetalStripes.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Wood_Misc.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Wood_Nailed.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Wood_Old.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Wood_Panel.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Wood_Plain.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Wood_Plain2.bmp": _g_OpaqueBallanceTexturePreset,
|
|
"Wood_Raft.bmp": _g_OpaqueBallanceTexturePreset,
|
|
}
|
|
|
|
def get_ballance_texture_preset(texname: str) -> RawVirtoolsTexture:
|
|
try_preset: RawVirtoolsTexture | None = _g_BallanceTexturePresets.get(texname, None)
|
|
if try_preset is None:
|
|
# fallback to non-ballance one
|
|
try_preset = _g_NonBallanceTexturePreset
|
|
|
|
return try_preset
|
|
|
|
def get_nonballance_texture_preset() -> RawVirtoolsTexture:
|
|
return _g_NonBallanceTexturePreset
|
|
|
|
#endregion
|
|
|
|
def register() -> None:
|
|
bpy.utils.register_class(BBP_PG_virtools_texture)
|
|
|
|
# add into image metadata
|
|
bpy.types.Image.virtools_texture = bpy.props.PointerProperty(type = BBP_PG_virtools_texture)
|
|
|
|
def unregister() -> None:
|
|
# del from image metadata
|
|
del bpy.types.Image.virtools_texture
|
|
|
|
bpy.utils.unregister_class(BBP_PG_virtools_texture)
|