CM3D2 Converter.mate_export

  1import bpy
  2from . import common
  3from . import compat
  4from . import cm3d2_data
  5from .translations.pgettext_functions import *
  6
  7
  8@compat.BlRegister()
  9class CNV_OT_export_cm3d2_mate(bpy.types.Operator):
 10    bl_idname = 'material.export_cm3d2_mate'
 11    bl_label = "mateとして保存"
 12    bl_description = "表示しているマテリアルをmateファイルとして保存します"
 13    bl_options = {'REGISTER', 'UNDO'}
 14
 15    filepath = bpy.props.StringProperty(subtype='FILE_PATH')
 16    filename_ext = ".mate"
 17    filter_glob = bpy.props.StringProperty(default="*.mate", options={'HIDDEN'})
 18
 19    is_backup = bpy.props.BoolProperty(name="ファイルをバックアップ", default=True, description="ファイルに上書きする場合にバックアップファイルを複製します")
 20
 21    version = bpy.props.IntProperty(name="ファイルバージョン", default=1000, min=1000, max=1111, soft_min=1000, soft_max=1111, step=1)
 22    name1 = bpy.props.StringProperty(name="名前1")
 23    name2 = bpy.props.StringProperty(name="名前2")
 24
 25    @classmethod
 26    def poll(cls, context):
 27        mate = getattr(context, 'material')
 28        if mate:
 29            if 'shader1' in mate and 'shader2' in mate:
 30                return True
 31        return False
 32
 33    def invoke(self, context, event):
 34        prefs = common.preferences()
 35        mate = context.material
 36        if prefs.mate_default_path:
 37            self.filepath = common.default_cm3d2_dir(prefs.mate_default_path, mate.name.lower(), "mate")
 38        else:
 39            self.filepath = common.default_cm3d2_dir(prefs.mate_export_path, mate.name.lower(), "mate")
 40        self.is_backup = bool(prefs.backup_ext)
 41        self.name1 = common.remove_serial_number(mate.name.lower())
 42        self.name2 = common.remove_serial_number(mate.name)
 43        context.window_manager.fileselect_add(self)
 44        return {'RUNNING_MODAL'}
 45
 46    def draw(self, context):
 47        row = self.layout.row()
 48        row.prop(self, 'is_backup', icon='FILE_BACKUP')
 49        if not common.preferences().backup_ext:
 50            row.enabled = False
 51        self.layout.prop(self, 'version', icon='LINENUMBERS_ON')
 52        self.layout.prop(self, 'name1', icon='SORTALPHA')
 53        self.layout.prop(self, 'name2', icon='SORTALPHA')
 54
 55    def execute(self, context):
 56        common.preferences().mate_export_path = self.filepath
 57
 58        try:
 59            writer = common.open_temporary(self.filepath, 'wb', is_backup=self.is_backup)
 60        except:
 61            self.report(type={'ERROR'}, message=f_tip_("ファイルを開くのに失敗しました、アクセス不可かファイルが存在しません。file={}", self.filepath))
 62            return {'CANCELLED'}
 63
 64        try:
 65            with writer:
 66                mate = context.material
 67                if compat.IS_LEGACY:
 68                    mat_data = cm3d2_data.MaterialHandler.parse_mate_old(mate, remove_serial=True)
 69                else:
 70                    mat_data = cm3d2_data.MaterialHandler.parse_mate(mate, remove_serial=True)
 71
 72                mat_data.version = self.version
 73                mat_data.name1 = self.name1
 74                mat_data.name2 = self.name2
 75
 76                mat_data.write(writer)
 77
 78        except common.CM3D2ExportException as e:
 79            self.report(type={'ERROR'}, message=str(e))
 80            return {'CANCELLED'}
 81
 82        return {'FINISHED'}
 83
 84
 85@compat.BlRegister()
 86class CNV_OT_export_cm3d2_mate_text(bpy.types.Operator):
 87    bl_idname = 'text.export_cm3d2_mate_text'
 88    bl_label = "mateとして保存"
 89    bl_description = "表示しているテキストデータをmateファイルとして保存します"
 90    bl_options = {'REGISTER', 'UNDO'}
 91
 92    filepath = bpy.props.StringProperty(subtype='FILE_PATH')
 93    filename_ext = ".mate"
 94    filter_glob = bpy.props.StringProperty(default="*.mate", options={'HIDDEN'})
 95
 96    is_backup = bpy.props.BoolProperty(name="ファイルをバックアップ", default=True, description="ファイルに上書きする場合にバックアップファイルを複製します")
 97
 98    version = bpy.props.IntProperty(name="ファイルバージョン", default=1000, min=1000, max=1111, soft_min=1000, soft_max=1111, step=1)
 99    name1 = bpy.props.StringProperty(name="名前1")
100    name2 = bpy.props.StringProperty(name="名前2")
101
102    @classmethod
103    def poll(cls, context):
104        edit_text = getattr(context, 'edit_text')
105        if edit_text:
106            data = edit_text.as_string()
107            if len(data) < 32:
108                return False
109            if "\ntex\n" in data or "\ncol\n" in data or "\nf\n" in data:
110                return True
111
112        return False
113
114    def invoke(self, context, event):
115        txt = context.edit_text
116        lines = txt.as_string().split('\n')
117        mate_name = lines[1]
118        if common.preferences().mate_default_path:
119            self.filepath = common.default_cm3d2_dir(common.preferences().mate_default_path, mate_name.lower(), "mate")
120        else:
121            self.filepath = common.default_cm3d2_dir(common.preferences().mate_export_path, mate_name.lower(), "mate")
122        try:
123            self.version = int(lines[0])
124        except:
125            self.version = 1000
126        if lines[1] != '***':
127            self.name1 = lines[1]
128        else:
129            self.name1 = lines[2]
130        self.name2 = lines[2]
131        context.window_manager.fileselect_add(self)
132        return {'RUNNING_MODAL'}
133
134    def draw(self, context):
135        row = self.layout.row()
136        row.prop(self, 'is_backup', icon='FILE_BACKUP')
137        if not common.preferences().backup_ext:
138            row.enabled = False
139        self.layout.prop(self, 'version', icon='LINENUMBERS_ON')
140        self.layout.prop(self, 'name1', icon='SORTALPHA')
141        self.layout.prop(self, 'name2', icon='SORTALPHA')
142
143    def execute(self, context):
144        common.preferences().mate_export_path = self.filepath
145
146        try:
147            text = context.edit_text.as_string()
148            mat_data = cm3d2_data.MaterialHandler.parse_text(text)
149        except Exception as e:
150            self.report(type={'ERROR'}, message='マテリアル情報の貼付けを中止します。' + str(e))
151            return {'CANCELLED'}
152
153        try:
154            file = common.open_temporary(self.filepath, 'wb', is_backup=self.is_backup)
155        except:
156            self.report(type={'ERROR'}, message=f_tip_("ファイルを開くのに失敗しました、アクセス不可かファイルが存在しません。file={}", self.filepath))
157            return {'CANCELLED'}
158
159        try:
160            with file:
161                mat_data.write(file, write_header=True)
162        except Exception as e:
163            self.report(type={'ERROR'}, message="mateファイルの出力に失敗、中止します。 構文を見直して下さい。" + str(e))
164            return {'CANCELLED'}
165
166        return {'FINISHED'}
167
168
169# テキストメニューに項目を登録
170def TEXT_MT_text(self, context):
171    self.layout.operator(CNV_OT_export_cm3d2_mate_text.bl_idname, icon_value=common.kiss_icon())