finish half of virtools importer
This commit is contained in:
parent
ec0749f3e5
commit
8fad643dd9
@ -1,6 +1,9 @@
|
||||
import bpy
|
||||
import tempfile
|
||||
from . import PROP_preferences, UTIL_functions, UTIL_file_browser, UTIL_blender_mesh
|
||||
from bpy_extras.wm_utils.progress_report import ProgressReport
|
||||
import tempfile, os
|
||||
from . import PROP_preferences
|
||||
from . import UTIL_virtools_types, UTIL_functions, UTIL_file_browser, UTIL_blender_mesh, UTIL_ballance_texture
|
||||
from . import PROP_ballance_element, PROP_virtools_group, PROP_virtools_material
|
||||
from .PyBMap import bmap_wrapper as bmap
|
||||
|
||||
class BBP_OT_import_virtools(bpy.types.Operator, UTIL_file_browser.ImportVirtoolsFile):
|
||||
@ -36,10 +39,182 @@ def _import_virtools(file_name_: str, encodings_: tuple[str]) -> None:
|
||||
with tempfile.TemporaryDirectory() as vt_temp_folder:
|
||||
# create virtools reader context
|
||||
with bmap.BMFileReader(
|
||||
file_name_, vt_temp_folder,
|
||||
PROP_preferences.get_raw_preferences().mBallanceTextureFolder,
|
||||
encodings_) as reader:
|
||||
pass
|
||||
file_name_,
|
||||
vt_temp_folder,
|
||||
PROP_preferences.get_raw_preferences().mBallanceTextureFolder,
|
||||
encodings_) as reader:
|
||||
|
||||
# prepare progress reporter
|
||||
with ProgressReport(wm = bpy.context.window_manager) as progress:
|
||||
# import textures
|
||||
texture_cret_map: dict[bmap.BMTexture, bpy.types.Image] = _import_virtools_textures(
|
||||
reader, progress, texture_cret_map)
|
||||
# import materials
|
||||
material_cret_map: dict[bmap.BMMaterial, bpy.types.Material] = _import_virtools_materials(
|
||||
reader, progress, texture_cret_map)
|
||||
# import meshes
|
||||
mesh_cret_map: dict[bmap.BMMesh, bpy.types.Mesh] = _import_virtools_meshes(
|
||||
reader, progress, material_cret_map)
|
||||
# import 3dobjects
|
||||
obj3d_cret_map: dict[bmap.BM3dObject, bpy.types.Object] = _import_virtools_3dobjects(
|
||||
reader, progress, mesh_cret_map)
|
||||
# import groups
|
||||
_import_virtools_groups(reader, progress, obj3d_cret_map)
|
||||
|
||||
def _import_virtools_textures(
|
||||
reader: bmap.BMFileReader,
|
||||
progress:ProgressReport
|
||||
) -> dict[bmap.BMTexture, bpy.types.Image]:
|
||||
# create map
|
||||
texture_cret_map: dict[bmap.BMTexture, bpy.types.Image] = {}
|
||||
progress.enter_substeps(reader.get_texture_count(), "Loading Textures")
|
||||
|
||||
# create another temp folder for raw data virtools texture importing
|
||||
with tempfile.TemporaryDirectory() as rawdata_temp:
|
||||
|
||||
for vttexture in reader.get_textures():
|
||||
# if this image is raw data, save it in external folder before loading
|
||||
texpath_to_load: str = vttexture.get_file_name()
|
||||
if vttexture.get_save_options() == UTIL_virtools_types.CK_TEXTURE_SAVEOPTIONS.CKTEXTURE_RAWDATA:
|
||||
texpath_to_load = os.path.join(
|
||||
rawdata_temp,
|
||||
os.path.basename(vttexture.get_file_name())
|
||||
)
|
||||
vttexture.save_image(texpath_to_load)
|
||||
|
||||
# detect whether it is ballance texture and load
|
||||
try_blc_tex: str | None = UTIL_ballance_texture.get_ballance_texture_filename(texpath_to_load)
|
||||
tex: bpy.types.Image
|
||||
if try_blc_tex:
|
||||
# load as ballance texture
|
||||
tex = UTIL_ballance_texture.load_ballance_texture(try_blc_tex)
|
||||
else:
|
||||
# load as other textures
|
||||
tex = UTIL_ballance_texture.load_other_texture(texpath_to_load)
|
||||
|
||||
# rename and insert it to map
|
||||
tex.name = UTIL_functions.virtools_name_regulator(vttexture.get_name())
|
||||
texture_cret_map[vttexture] = tex
|
||||
|
||||
# inc steps
|
||||
progress.step()
|
||||
|
||||
# leave progress and return map
|
||||
progress.leave_substeps()
|
||||
return texture_cret_map
|
||||
|
||||
def _import_virtools_materials(
|
||||
reader: bmap.BMFileReader,
|
||||
progress: ProgressReport,
|
||||
texture_cret_map: dict[bmap.BMTexture, bpy.types.Image]
|
||||
) -> dict[bmap.BMMaterial, bpy.types.Material]:
|
||||
# create map and prepare progress
|
||||
material_cret_map: dict[bmap.BMMaterial, bpy.types.Material] = {}
|
||||
progress.enter_substeps(reader.get_material_count(), "Loading Materials")
|
||||
|
||||
for vtmaterial in reader.get_materials():
|
||||
# create new raw material
|
||||
rawmtl: PROP_virtools_material.RawVirtoolsMaterial = PROP_virtools_material.RawVirtoolsMaterial()
|
||||
|
||||
rawmtl.mDiffuse = vtmaterial.get_diffuse()
|
||||
rawmtl.mAmbient = vtmaterial.get_ambient()
|
||||
rawmtl.mSpecular = vtmaterial.get_specular()
|
||||
rawmtl.mEmissive = vtmaterial.get_emissive()
|
||||
rawmtl.mSpecularPower = vtmaterial.get_specular_power()
|
||||
|
||||
mtltex: bmap.BMTexture | None = vtmaterial.get_texture()
|
||||
if mtltex:
|
||||
rawmtl.mTexture = texture_cret_map.get(mtltex, None)
|
||||
else:
|
||||
rawmtl.mTexture = None
|
||||
rawmtl.mTextureBorderColor = vtmaterial.get_texture_border_color()
|
||||
|
||||
rawmtl.mTextureBlendMode = vtmaterial.get_texture_blend_mode()
|
||||
rawmtl.mTextureMinMode = vtmaterial.get_texture_min_mode()
|
||||
rawmtl.mTextureMagMode = vtmaterial.get_texture_mag_mode()
|
||||
rawmtl.mTextureAddressMode = vtmaterial.get_texture_address_mode()
|
||||
|
||||
rawmtl.mSourceBlend = vtmaterial.get_source_blend()
|
||||
rawmtl.mDestBlend = vtmaterial.get_dest_blend()
|
||||
rawmtl.mFillMode = vtmaterial.get_fill_mode()
|
||||
rawmtl.mShadeMode = vtmaterial.get_shade_mode()
|
||||
|
||||
rawmtl.mEnableAlphaTest = vtmaterial.get_alpha_test_enabled()
|
||||
rawmtl.mEnableAlphaBlend = vtmaterial.get_alpha_blend_enabled()
|
||||
rawmtl.mEnablePerspectiveCorrection = vtmaterial.get_perspective_correction_enabled()
|
||||
rawmtl.mEnableZWrite = vtmaterial.get_z_write_enabled()
|
||||
rawmtl.mEnableTwoSided = vtmaterial.get_two_sided_enabled()
|
||||
|
||||
rawmtl.mAlphaRef = vtmaterial.get_alpha_ref()
|
||||
rawmtl.mAlphaFunc = vtmaterial.get_alpha_func()
|
||||
rawmtl.mZFunc = vtmaterial.get_z_func()
|
||||
|
||||
# create mtl and apply it
|
||||
mtl: bpy.types.Material = bpy.data.materials.new(
|
||||
UTIL_functions.virtools_name_regulator(vtmaterial.get_name())
|
||||
)
|
||||
PROP_virtools_material.set_raw_virtools_material(mtl, rawmtl)
|
||||
PROP_virtools_material.apply_to_blender_material(mtl)
|
||||
|
||||
# add into map and step
|
||||
material_cret_map[vtmaterial] = mtl
|
||||
progress.step()
|
||||
|
||||
# leave progress and return
|
||||
progress.leave_substeps()
|
||||
return material_cret_map
|
||||
|
||||
def _import_virtools_meshes(
|
||||
reader: bmap.BMFileReader,
|
||||
progress: ProgressReport,
|
||||
material_cret_map: dict[bmap.BMMaterial, bpy.types.Material]
|
||||
) -> dict[bmap.BMMesh, bpy.types.Mesh]:
|
||||
# create map and prepare progress
|
||||
mesh_cret_map: dict[bmap.BMMesh, bpy.types.Mesh] = {}
|
||||
progress.enter_substeps(reader.get_material_count(), "Loading Meshes")
|
||||
|
||||
for vtmesh in reader.get_meshs():
|
||||
# add into map and step
|
||||
#mesh_cret_map[vtmaterial] = mtl
|
||||
progress.step()
|
||||
|
||||
# leave progress and return
|
||||
progress.leave_substeps()
|
||||
return mesh_cret_map
|
||||
|
||||
def _import_virtools_3dobjects(
|
||||
reader: bmap.BMFileReader,
|
||||
progress: ProgressReport,
|
||||
mesh_cret_map: dict[bmap.BMMesh, bpy.types.Mesh]
|
||||
) -> dict[bmap.BM3dObject, bpy.types.Object]:
|
||||
# create map and prepare progress
|
||||
obj3d_cret_map: dict[bmap.BM3dObject, bpy.types.Object] = {}
|
||||
progress.enter_substeps(reader.get_material_count(), "Loading 3dObjects")
|
||||
|
||||
for vt3dobj in reader.get_3dobjects():
|
||||
# add into map and step
|
||||
#obj3d_cret_map[vtmaterial] = mtl
|
||||
progress.step()
|
||||
|
||||
# leave progress and return
|
||||
progress.leave_substeps()
|
||||
return obj3d_cret_map
|
||||
|
||||
def _import_virtools_groups(
|
||||
reader: bmap.BMFileReader,
|
||||
progress: ProgressReport,
|
||||
obj3d_cret_map: dict[bmap.BM3dObject, bpy.types.Object]
|
||||
) -> dict[bmap.BM3dObject, bpy.types.Object]:
|
||||
# prepare progress
|
||||
progress.enter_substeps(reader.get_material_count(), "Loading Groups")
|
||||
|
||||
for vtgroup in reader.get_groups():
|
||||
# add into map and step
|
||||
#obj3d_cret_map[vtmaterial] = mtl
|
||||
progress.step()
|
||||
|
||||
# leave progress
|
||||
progress.leave_substeps()
|
||||
|
||||
def register() -> None:
|
||||
bpy.utils.register_class(BBP_OT_import_virtools)
|
||||
|
||||
@ -37,23 +37,9 @@ def clamp_int(v: int, min_val: int, max_val: int) -> int:
|
||||
elif (v > max_val): return max_val
|
||||
else: return v
|
||||
|
||||
|
||||
_TLimitIterator = typing.TypeVar('_TLimitIterator')
|
||||
def limit_iterator(it: typing.Iterator[_TLimitIterator], limit_count: int) -> typing.Iterator[_TLimitIterator]:
|
||||
"""!
|
||||
A generator wrapper for another generator to make sure the length of the given generator output entries
|
||||
is not greater (<= less equal) than given number.
|
||||
|
||||
@param it[in] A iterator need to be limited. Use iter(ls) if your passing value is not a generator (eg. tuple, list)
|
||||
@param limit_count[in] The count to limit. Must be a positive number. Can be 0.
|
||||
@return A generator with limited length. Use tuple(ret) if you just want to convert it to a tuple.
|
||||
"""
|
||||
counter: int = 0
|
||||
while counter < limit_count:
|
||||
# if no elements in given iterator, StopIteration will raise.
|
||||
# It is okey because it naturally stop the iteration of this generator.
|
||||
yield next(it)
|
||||
counter += 1
|
||||
def virtools_name_regulator(name: str | None) -> str:
|
||||
if name: return name
|
||||
else: return 'annoymous'
|
||||
|
||||
def message_box(message: tuple[str], title: str, icon: str):
|
||||
"""
|
||||
|
||||
Loading…
Reference in New Issue
Block a user