diff options
Diffstat (limited to 'io_coat3D/coat.py')
-rw-r--r-- | io_coat3D/coat.py | 463 |
1 files changed, 463 insertions, 0 deletions
diff --git a/io_coat3D/coat.py b/io_coat3D/coat.py new file mode 100644 index 00000000..6e9cad3d --- /dev/null +++ b/io_coat3D/coat.py @@ -0,0 +1,463 @@ +# scene_blend_info.py Copyright (C) 2010, Mariano Hidalgo +# +# Show Information About the Blend. +# ***** BEGIN GPL LICENSE BLOCK ***** +# +# +# This program is free software; you can redistribute it and/or +# modify it under the terms of the GNU General Public License +# as published by the Free Software Foundation; either version 2 +# of the License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software Foundation, +# Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. +# +# ***** END GPL LICENCE BLOCK ***** + +import bpy +from bpy.props import * +from io_coat3D import tex +import os +import linecache + +class ObjectButtonsPanel(): + bl_space_type = 'PROPERTIES' + bl_region_type = 'WINDOW' + bl_context = "object" + +class SCENE_PT_Borgleader(ObjectButtonsPanel,bpy.types.Panel): + bl_label = "3D-Coat Applink" + bl_space_type = "PROPERTIES" + bl_region_type = "WINDOW" + bl_context = "scene" + + def draw(self, context): + layout = self.layout + scene = context.scene + me = context.scene.objects + mat_list = [] + import_no = 0 + coat = bpy.coat3D + coat3D = bpy.context.scene.coat3D + + if(os.path.isdir(coat3D.exchangedir)): + foldder = coat3D.exchangedir + if(foldder.rfind('Exchange') >= 0): + coat['exchange'] = foldder + coat['status'] = 1 + else: + coat['status'] = 0 + else: + coat['status'] = 0 + #Here you add your GUI + row = layout.row() + row.prop(coat3D,"type",text = "") + row = layout.row() + colL = row.column() + colR = row.column() + if(context.selected_objects): + if(context.selected_objects[0].type == 'MESH'): + colL.active = True + else: + colL.active = False + else: + colL.active = False + colL.operator("exportbutton", text="Export") + colL.label(text="Export Settings:") + + colL.prop(coat3D,"exportover") + if(coat3D.exportover): + colL.prop(coat3D,"exportmod") + colL.prop(coat3D,"exportfile") + colL.prop(coat3D,"export_pos") + + + if(bpy.context.active_object): + colR.active = True + else: + colR.active = False + + colR.operator("importbutton", text="Import") + colR.label(text="Import Settings:") + colR.prop(coat3D,"importmesh") + colR.prop(coat3D,"importmod") + colR.prop(coat3D,"smooth_on") + colR.prop(coat3D,"importtextures") + row = layout.row() + colL = row.column() + colM = row.column() + colR = row.column() + colL.operator("deltex",text="Del Tex") + if(bpy.context.active_object): + colM.active = True + if(bpy.context.active_object.coat3D.coatpath and os.path.isfile(bpy.context.active_object.coat3D.coatpath)): + colR.active = True + if(coat['active_coat'] == bpy.context.active_object.coat3D.coatpath): + colR.operator("load3b", text="Active 3b") + else: + colR.operator("load3b", text="Load 3b") + else: + colR.active = False + colR.operator("no3b",text="No 3b") + else: + colM.active = False + colR.active = False + colR.operator("no3b",text="") + colM.operator("pickname",text="Object name") + row = layout.row() + row.label(text="Object Path:") + row = layout.row() + row.prop(coat3D,"objectdir",text="") + row = layout.row() + + if(coat['status'] == 1): + row.label(text="Exchange Folder: connected") + Blender_folder = ("%s%sBlender"%(coat3D.exchangedir,os.sep)) + Blender_export = Blender_folder + Blender_export += ('%sexport.txt'%(os.sep)) + + if(not(os.path.isdir(Blender_folder))): + os.makedirs(Blender_folder) + Blender_folder = os.path.join(Blender_folder,"run.txt") + file = open(Blender_folder, "w") + file.close() + + if(os.path.isfile(Blender_export)): + obj_path ='' + obj_pathh = open(Blender_export) + for line in obj_pathh: + obj_path = line + break + obj_pathh.close() + print("%s"%obj_path) + export = obj_path + mod_time = os.path.getmtime(obj_path) + mtl_list = obj_path.replace('.obj','.mtl') + if(os.path.isfile(mtl_list)): + os.remove(mtl_list) + + for palikka in bpy.context.scene.objects: + if(palikka.type == 'MESH'): + if(palikka.coat3D.objpath == export): + import_no = 1 + target = palikka + break + + if(import_no): + new_obj = palikka + import_no = 0 + else: + bpy.ops.import_scene.obj(filepath=obj_path) + new_obj = scene.objects[0] + os.remove(Blender_export) + + bpy.context.scene.objects.active = new_obj + bpy.context.active_object.coat3D.objpath = obj_path + + if(coat3D.smooth_on): + bpy.ops.object.shade_smooth() + else: + bpy.ops.object.shade_flat() + + Blender_tex = ("%s%stextures.txt"%(coat3D.exchangedir,os.sep)) + mat_list.append(new_obj.material_slots[0].material) + tex.gettex(mat_list, new_obj, scene,export) + + + + + + if(coat['status'] == 0): + row.label(text="Exchange Folder: not connected") + row = layout.row() + row.prop(coat3D,"exchangedir",text="") + row = layout.row() + row.label(text="Author: haikalle@gmail.com") + + +class SCENE_OT_export(bpy.types.Operator): + bl_idname = "exportbutton" + bl_label = "Export your custom property" + bl_description = "Export your custom property" + + + def invoke(self, context, event): + checkname = '' + coat3D = bpy.context.scene.coat3D + coat = bpy.coat3D + scene = context.scene + coat3D.export_on = False + activeobj = bpy.context.active_object.name + obj = scene.objects[activeobj] + + importfile = coat3D.exchangedir + texturefile = coat3D.exchangedir + importfile += ('%simport.txt'%(os.sep)) + texturefile += ('%stextures.txt'%(os.sep)) + if(os.path.isfile(texturefile)): + os.remove(texturefile) + + checkname = coat3D.objectdir + + if(coat3D.objectdir[-4:] != '.obj'): + checkname += ('%s.obj'%(activeobj)) + + if(not(os.path.isfile(checkname)) or coat3D.exportover): + + + bpy.ops.export_scene.obj(filepath=checkname,use_selection=True, + use_modifiers=coat3D.exportmod,use_blen_objects=False, group_by_material= True, + use_materials = False,keep_vertex_order = True) + coat3D.export_on = True + + if(not(coat3D.exportover)): + coat3D.loca = obj.location + coat3D.rota = obj.rotation_euler + coat3D.scal = obj.scale + coat['export_off'] = 1 + else: + coat['export_off'] = 0 + + + + if(coat3D.exportfile == False): + file = open(importfile, "w") + file.write("%s"%(checkname)) + file.write("\n%s"%(checkname)) + file.write("\n[%s]"%(coat3D.type)) + file.close() + coat3D.objectdir = checkname + bpy.context.active_object.coat3D.objpath = coat3D.objectdir + + return('FINISHED') + + +class SCENE_OT_import(bpy.types.Operator): + bl_idname = "importbutton" + bl_label = "import your custom property" + bl_description = "import your custom property" + + def invoke(self, context, event): + scene = context.scene + coat3D = bpy.context.scene.coat3D + coat = bpy.coat3D + activeobj = bpy.context.active_object.name + mat_list = [] + scene.objects[activeobj].select = True + objekti = scene.objects[activeobj] + coat3D.loca = objekti.location + coat3D.rota = objekti.rotation_euler + + exportfile = coat3D.exchangedir + exportfile += ('%sexport.txt'%(os.sep)) + if(os.path.isfile(exportfile)): + export_file = open(exportfile) + for line in export_file: + if line.rfind('.3b'): + objekti.coat3D.coatpath = line + coat['active_coat'] = line + export_file.close() + os.remove(exportfile) + + + + + if(objekti.material_slots): + for obj_mat in objekti.material_slots: + mat_list.append(obj_mat.material) + act_mat_index = objekti.active_material_index + + + if(coat3D.importmesh and os.path.isfile(coat3D.objectdir)): + mtl = coat3D.objectdir + mtl = mtl.replace('.obj','.mtl') + if(os.path.isfile(mtl)): + os.remove(mtl) + + + bpy.ops.import_scene.obj(filepath=coat3D.objectdir) + obj_proxy = scene.objects[0] + proxy_mat = obj_proxy.material_slots[0].material + obj_proxy.data.materials.pop(0) + proxy_mat.user_clear() + bpy.data.materials.remove(proxy_mat) + bpy.ops.object.select_all(action='TOGGLE') + if(coat3D.export_pos): + scene.objects.active = objekti + objekti.select = True + coat3D.cursor = bpy.context.scene.cursor_location + bpy.context.scene.cursor_location = (0.0,0.0,0.0) + bpy.ops.object.origin_set(type='ORIGIN_CURSOR', center='MEDIAN') + + scene.objects.active = obj_proxy + + obj_data = objekti.data.id_data + objekti.data = obj_proxy.data.id_data + + if(coat3D.export_on): + objekti.scale = (1,1,1) + objekti.rotation_euler = (0,0,0) + + if(bpy.data.meshes[obj_data.name].users == 0): + bpy.data.meshes.remove(obj_data) + objekti.data.id_data.name = obj_data.name + + bpy.ops.object.select_all(action='TOGGLE') + + obj_proxy.select = True + bpy.ops.object.delete() + objekti.select = True + bpy.context.scene.objects.active = objekti + if(coat3D.export_on): + bpy.ops.object.origin_set(type='ORIGIN_GEOMETRY', center='MEDIAN') + coat3D.export_on = False + + else: + objekti.location = coat3D.loca + objekti.rotation_euler = coat3D.rota + bpy.ops.object.origin_set(type='GEOMETRY_ORIGIN', center='MEDIAN') + else: + scene.objects.active = obj_proxy + + obj_data = objekti.data.id_data + objekti.data = obj_proxy.data.id_data + if(bpy.data.meshes[obj_data.name].users == 0): + bpy.data.meshes.remove(obj_data) + objekti.data.id_data.name = obj_data.name + + obj_proxy.select = True + bpy.ops.object.delete() + objekti.select = True + bpy.context.scene.objects.active = objekti + + + + if(coat3D.smooth_on): + bpy.ops.object.shade_smooth() + else: + bpy.ops.object.shade_flat() + + if(coat3D.importmesh and not(os.path.isfile(coat3D.objectdir))): + coat3D.importmesh = False + + if(mat_list and coat3D.importmesh): + for mat_one in mat_list: + objekti.data.materials.append(mat_one) + objekti.active_material_index = act_mat_index + + if(mat_list): + for obj_mate in objekti.material_slots: + for tex_slot in obj_mate.material.texture_slots: + if(hasattr(tex_slot,'texture')): + if(tex_slot.texture.type == 'IMAGE'): + tex_slot.texture.image.reload() + + + if(coat3D.importmod): + mod_list = [] + for mod_index in objekti.modifiers: + objekti.modifiers.remove(mod_index) + + + + if(coat3D.importtextures): + export = '' + tex.gettex(mat_list,objekti,scene,export) + + if(coat['export_off']): + objekti.location = coat3D.loca + objekti.rotation_euler = coat3D.rota + objekti.scale = coat3D.scal + coat['export_off'] = 0 + + return('FINISHED') + +class SCENE_OT_load3b(bpy.types.Operator): + bl_idname = "load3b" + bl_label = "Loads 3b linked into object" + bl_description = "Loads 3b linked into object" + + + def invoke(self, context, event): + checkname = '' + coat3D = bpy.context.scene.coat3D + scene = context.scene + importfile = coat3D.exchangedir + importfile += ('%simport.txt'%(os.sep)) + + coat_path = bpy.context.active_object.coat3D.coatpath + + file = open(importfile, "w") + file.write("%s"%(coat_path)) + file.write("\n%s"%(coat_path)) + file.write("\n[3B]") + file.close() + + + coat['active_coat'] = coat_path + + + return('FINISHED') + +class SCENE_OT_no3b(bpy.types.Operator): + bl_idname = "no3b" + bl_label = "Loads 3b linked into object" + bl_description = "Loads 3b linked into object" + + + def invoke(self, context, event): + scene = context.scene + + return('FINISHED') + +class SCENE_OT_pickname(bpy.types.Operator): + bl_idname = "pickname" + bl_label = "Picks Object's name into path" + bl_description = "Loads 3b linked into object" + + + def invoke(self, context, event): + coat3D = bpy.context.scene.coat3D + scene = context.scene + new_name = os.path.dirname(coat3D.objectdir) + os.sep + new_name += ("%s.obj"%(bpy.context.active_object.name)) + coat3D.objectdir = new_name + + return('FINISHED') + +class SCENE_OT_deltex(bpy.types.Operator): + bl_idname = "deltex" + bl_label = "Picks Object's name into path" + bl_description = "Loads 3b linked into object" + + + def invoke(self, context, event): + coat3D = bpy.context.scene.coat3D + scene = context.scene + nimi = tex.objname(coat3D.objectdir) + osoite = os.path.dirname(coat3D.objectdir) + os.sep + just_nimi = tex.justname(nimi) + just_nimi += '_' + + files = os.listdir(osoite) + for i in files: + if(i.rfind(just_nimi) >= 0): + del_osoite = osoite + i + os.remove(del_osoite) + + return('FINISHED') + +def register(): + pass + +def unregister(): + pass + +if __name__ == "__main__": + register() |