import os, re, bpy, struct, os.path, shutil
from . import common

class import_cm3d2_mate(bpy.types.Operator):
	bl_idname = 'material.import_cm3d2_mate'
	bl_label = "Import Mate"
	bl_description = "Open a .mate file as a material."
	bl_options = {'REGISTER', 'UNDO'}
	
	filepath = bpy.props.StringProperty(subtype='FILE_PATH')
	filename_ext = ".mate"
	filter_glob = bpy.props.StringProperty(default="*.mate", options={'HIDDEN'})
	
	is_decorate = bpy.props.BoolProperty(name="Decorate the material according to its type", default=True)
	is_replace_cm3d2_tex = bpy.props.BoolProperty(name="Find textures", default=True, description="Will search for the textures.")
	
	@classmethod
	def poll(cls, context):
		if 'material_slot' in dir(context):
			if 'material' in dir(context):
				return True
		return False
	
	def invoke(self, context, event):
		if common.preferences().mate_default_path:
			self.filepath = common.default_cm3d2_dir(common.preferences().mate_default_path, "", "mate")
		else:
			self.filepath = common.default_cm3d2_dir(common.preferences().mate_import_path, "", "mate")
		self.is_replace_cm3d2_tex = common.preferences().is_replace_cm3d2_tex
		context.window_manager.fileselect_add(self)
		return {'RUNNING_MODAL'}
	
	def draw(self, context):
		self.layout.prop(self, 'is_decorate', icon='TEXTURE_SHADED')
		self.layout.prop(self, 'is_replace_cm3d2_tex', icon='BORDERMOVE')
	
	def execute(self, context):
		common.preferences().mate_import_path = self.filepath
		
		ob = context.active_object
		me = ob.data
		
		try:
			file = open(self.filepath, 'rb')
		except:
			self.report(type={'ERROR'}, message="Failed to import file, File is possibly inaccessible or non-existent.")
			return {'CANCELLED'}
		if common.read_str(file) != 'CM3D2_MATERIAL':
			self.report(type={'ERROR'}, message="This is not a .mate file for CM3D2")
			return {'CANCELLED'}
		struct.unpack('<i', file.read(4))[0]
		common.read_str(file)
		mate_name = common.read_str(file)
		
		if not context.material_slot:
			bpy.ops.object.material_slot_add()
		root, ext = os.path.splitext(os.path.basename(self.filepath))
		mate = context.blend_data.materials.new(mate_name)
		context.material_slot.material = mate
		
		mate['shader1'] = common.read_str(file)
		mate['shader2'] = common.read_str(file)
		
		slot_index = 0
		already_texs = []
		for i in range(99999):
			type = common.read_str(file)
			if type == 'tex':
				slot = mate.texture_slots.create(slot_index)
				tex_name = common.read_str(file)
				tex = context.blend_data.textures.new(tex_name, 'IMAGE')
				slot.texture = tex
				sub_type = common.read_str(file)
				if sub_type == 'tex2d':
					img = context.blend_data.images.new(common.read_str(file), 128, 128)
					img['cm3d2_path'] = common.read_str(file)
					img.filepath = img['cm3d2_path']
					img.source = 'FILE'
					tex.image = img
					slot.color = struct.unpack('<3f', file.read(4*3))
					slot.diffuse_color_factor = struct.unpack('<f', file.read(4))[0]
					
					# tex探し
					if self.is_replace_cm3d2_tex:
						if common.replace_cm3d2_tex(img) and tex_name=='_MainTex':
							for face in me.polygons:
								if face.material_index == ob.active_material_index:
									me.uv_textures.active.data[face.index].image = img
			
			elif type == 'col':
				slot = mate.texture_slots.create(slot_index)
				tex_name = common.read_str(file)
				tex = context.blend_data.textures.new(tex_name, 'BLEND')
				mate.use_textures[slot_index] = False
				slot.use_rgb_to_intensity = True
				slot.color = struct.unpack('<3f', file.read(4*3))
				slot.diffuse_color_factor = struct.unpack('<f', file.read(4))[0]
				slot.texture = tex
			
			elif type == 'f':
				slot = mate.texture_slots.create(slot_index)
				tex_name = common.read_str(file)
				tex = context.blend_data.textures.new(tex_name, 'BLEND')
				mate.use_textures[slot_index] = False
				slot.diffuse_color_factor = struct.unpack('<f', file.read(4))[0]
				slot.texture = tex
			
			elif type == 'end':
				break
			else:
				self.report(type={'ERROR'}, message="Unknown value was found, mission failed")
				return {'CANCELLED'}
			
			if common.preferences().mate_unread_same_value:
				if tex_name in already_texs:
					mate.texture_slots.clear(slot_index)
					slot_index -= 1
				already_texs.append(tex_name)
			
			slot_index += 1
		
		file.close()
		common.decorate_material(mate, self.is_decorate, me, ob.active_material_index)
		return {'FINISHED'}

class import_cm3d2_mate_text(bpy.types.Operator):
	bl_idname = 'text.import_cm3d2_mate_text'
	bl_label = "Import a mate"
	bl_description = "Open a mate file in the text editor as text"
	bl_options = {'REGISTER', 'UNDO'}
	
	filepath = bpy.props.StringProperty(subtype='FILE_PATH')
	filename_ext = ".mate"
	filter_glob = bpy.props.StringProperty(default="*.mate", options={'HIDDEN'})
	
	is_overwrite = bpy.props.BoolProperty(name="Overwrites current text in the text editor.", default=False)
	
	@classmethod
	def poll(cls, context):
		return True
	
	def invoke(self, context, event):
		if common.preferences().mate_default_path:
			self.filepath = common.default_cm3d2_dir(common.preferences().mate_default_path, "", "mate")
		else:
			self.filepath = common.default_cm3d2_dir(common.preferences().mate_import_path, "", "mate")
		context.window_manager.fileselect_add(self)
		return {'RUNNING_MODAL'}
	
	def draw(self, context):
		self.layout.prop(self, 'is_overwrite', icon='SAVE_COPY')
	
	def execute(self, context):
		common.preferences().mate_import_path = self.filepath
		
		txt = None
		if self.is_overwrite:
			if 'edit_text' not in dir(context):
				self.report(type={'ERROR'}, message="Text data could not be overwritten")
				return {'CANCELLED'}
			if not context.edit_text:
				self.report(type={'ERROR'}, message="Text data could not be overwritten")
				return {'CANCELLED'}
			txt = context.edit_text
			txt.clear()
		
		try:
			file = open(self.filepath, 'rb')
		except:
			self.report(type={'ERROR'}, message="Failed to open the file, File does not exist or is not accessible")
			return {'CANCELLED'}
		if common.read_str(file) != 'CM3D2_MATERIAL':
			self.report(type={'ERROR'}, message="This is not a CM3D2 Mate file")
			return {'CANCELLED'}
		
		version = str(struct.unpack('<i', file.read(4))[0])
		name1 = common.read_str(file)
		name2 = common.read_str(file)
		if not txt:
			txt = context.blend_data.texts.new(os.path.basename(name2))
			context.area.type = 'TEXT_EDITOR'
			context.space_data.text = txt
		txt.write( version + "\n" )
		txt.write( name1 + "\n" )
		txt.write( name2 + "\n" )
		txt.write( common.read_str(file) + "\n" )
		txt.write( common.read_str(file) + "\n" )
		txt.write("\n")
		
		for i in range(99999):
			type = common.read_str(file)
			if type == 'tex':
				txt.write( type + "\n" )
				txt.write( "\t" + common.read_str(file) + "\n" )
				tex_type = common.read_str(file)
				txt.write( "\t" + tex_type + "\n" )
				if tex_type == 'tex2d':
					txt.write( "\t" + common.read_str(file) + "\n" )
					txt.write( "\t" + common.read_str(file) + "\n" )
					fs = struct.unpack('<4f', file.read(4*4))
					txt.write( "\t" + " ".join([str(fs[0]), str(fs[1]), str(fs[2]), str(fs[3])]) + "\n" )
			elif type == 'col':
				txt.write( type + "\n" )
				txt.write( "\t" + common.read_str(file) + "\n" )
				fs = struct.unpack('<4f', file.read(4*4))
				txt.write( "\t" + " ".join([str(fs[0]), str(fs[1]), str(fs[2]), str(fs[3])]) + "\n" )
			elif type == 'f':
				txt.write( type + "\n" )
				txt.write( "\t" + common.read_str(file) + "\n" )
				txt.write( "\t" + str(struct.unpack('<f', file.read(4))[0]) + "\n" )
			elif type == 'end':
				break
			else:
				self.report(type={'ERROR'}, message="Unknown value was found, mission failed.")
				return {'CANCELLED'}
		
		file.close()
		txt.current_line_index = 0
		return {'FINISHED'}

# テキストメニューに項目を登録
def TEXT_MT_text(self, context):
	self.layout.separator()
	self.layout.operator(import_cm3d2_mate_text.bl_idname, icon_value=common.preview_collections['main']['KISS'].icon_id)