API tools faq
paste
TheMusiKid

Blender to Maya

TheMusiKid
Feb 10th, 2013
208
0
Never
Add comment
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 39.74 KB | None | 0 0
raw download clone embed print report
  1. #!BPY
  2. """Registration info for Blender menus:
  3. Name: 'OBJ...'
  4. Blender: 232
  5. Group: 'Import'
  6. Tip: 'Import Wavefront File Format (*.obj)'
  7. """
  8.  
  9. #=========================================================================
  10. # Wavefront OBJ Importer/Exporter v1.2
  11. # This is a complete OBJ importer and exporter script
  12. # All Rights Reserved
  13. # [email protected]
  14.  
  15.  
  16. #Version mise a jour pour Blender 228 par jm soler
  17. #
  18.  
  19. #=========================================================================
  20. # DESCRIPTION: This script allows for full importing and exporting of
  21. #      .obj files. uv texture coordinates and normals can be exported and
  22. #      imported. .obj groups and materials can also be converted to and
  23. #      from material indexes in Blender.
  24. #
  25. # INSTALLATION:
  26. #      You need the FULL python installation to run this script. You can
  27. #      down load the latest version of PYTHON from http://www.python.org.
  28. #
  29. # INSTRUCTIONS (You definitely want to read this!):
  30. #      Once the script is loaded in Blender, hit alt-p. This will bring up
  31. #      the main interface panel. You will have a choise of exporting or
  32. #      importing an .obj module. If you are exporting you must have at least
  33. #      one mesh selected in Blender, or you will get an error.  
  34. #      You can change the export filename by entering the path and filename
  35. #      in the dialog.  If you do not enter a path, the path will default to
  36. #      your blender directory. You can change the default path in the script <==== NOTE
  37. #      itself by modifying the variable 'Filename' at the top of the script.  
  38. #
  39. #    EXPORTING:
  40. #      There are 4 different export options: Default, Groups, Material Layers,
  41. #      and Standard. "Default" will export your mesh using Material Layers if
  42. #      the mesh has material indexes. "Groups" and "Material Layers" are
  43. #      logically equivalent, but are a different .obj format. If you are
  44. #      exporting a Poser morph target, you must select "Material Layers".   <===== NOTE
  45. #      "Standard" will ignore grouping information, even if your mesh has
  46. #      material indexes.
  47. #
  48. #      There is also a choice between export using "mesh coordinates" or
  49. #      "object coordinates". "Object coordinates" are any location, rotation,
  50. #      or scaling values created outside of mesh edit. They belong to the object
  51. #      rather than the mesh. If you export using mesh coordinates (the default)
  52. #      the center of the object will always be at 0, 0, 0. Export using "mesh
  53. #      coordinates is definintely what you want to use if you are working with
  54. #      a Poser morph target. If you are exporting a group of objects, you will
  55. #      automatically be placed in "object coordinate" mode.
  56. #
  57. #    IMPORTING:
  58. #      If your OBJ model has uv mapping coordinates, and you want to use them  <===== NOTE
  59. #      in Blender, you can access them in two ways. The best way is through Blender's
  60. #      realtime UV coordinates which you enable simply by selecting the UV option in
  61. #      the material edit window. This gives you an exact version of the uv coordinates.
  62. #      An older method is to select the "stick" option in the material edit window. I
  63. #      really don't know why anyone would want to use this option since it cannot handle
  64. #      seams and texture overlap, but I left it in just in case someone could use it for
  65. #      something.
  66. #    
  67. #      If your OBJ contains groups, once it has been imported, it may still appear
  68. #      to lack any material indexes. If this happens, it can be remedied      <=== NOTE
  69. #      by going to the mesh editor window, clicking on the mesh selection button, and
  70. #      reselecting the mesh you have just imported. You will now have as many
  71. #      materials attached to your object as there are groups. You can then select
  72. #      different groups by doing a material select when you are in edit mode.
  73. #
  74. #      Finally, you might have problems with certain parts of the object not displaying
  75. #      after you go in and out of edit mode the first time. To fix this, simply go into
  76. #      edit mode again, and select the "remove doubles" option.
  77. #
  78. #
  79. # HISTORY:
  80. #   Nov 13, 2001: Initial Release
  81. #   Nov 16, 2001: Version 1.1 - no longer need to pre-define dummy materials
  82. #   Dec 13, 2001: Version 1.2 - now imports into realtime UV (the UV button in the material edit window), and
  83. #       exports realtime UV. This format is more compatible with the native .OBJ uv format. Should eliminate
  84. #       texture misalignments and seams. Simply press the UV button in the material edit window after importing.
  85. #
  86. #  GetRaw
  87. #================================
  88.  
  89.  
  90. # ===============================
  91. #   Setup our runtime constants
  92. # ===============================
  93.  
  94. DEBUG=1         #Set this to "1" to see extra messages
  95. MESHVERSION=3   # If the export file doesn't work,
  96. FILEVERSION=3   # try changing these to "2"
  97.  
  98. EVENT_PATHCHANGE=     1
  99. EVENT_IMPORT=         2
  100. EVENT_IMPORT_CONT=    3
  101. EVENT_OPTIONS=        4
  102. EVENT_EXPORT=         7
  103. EVENT_EXPORT_CHK=     5
  104. EVENT_EXPORT_CANCEL=  6
  105. EVENT_QUIT=           8
  106. EVENT_EXPORT_ERR=     9
  107. EVENT_TYPE=           10
  108. EVENT_DONE=           11
  109. EVENT_IMPORT_SELECT=  12
  110.  
  111. # ===============================
  112. # Import our libraries
  113. # ===============================
  114.  
  115. #import string
  116. #import os
  117. #import struct
  118.  
  119. try:
  120.     import nt
  121.     os=nt
  122.     os.sep='\\'
  123. except:    
  124.     import posix
  125.     os=posix
  126.     os.sep='/'
  127.  
  128. def isdir(path):
  129.     try:
  130.         st = os.stat(path)
  131.         return 1
  132.     except:
  133.         return 0
  134.    
  135. def split(pathname):
  136.          k0=pathname.split(os.sep)
  137.          directory=pathname.replace(k0[len(k0)-1],'')
  138.          Name=k0[len(k0)-1]
  139.          return directory, Name
  140.        
  141. def join(l0,l1):        
  142.      return  l0+os.sep+l1
  143.    
  144. os.isdir=isdir
  145. os.split=split
  146. os.join=join
  147.  
  148. import math
  149. import Blender
  150. #import Blender210
  151. from Blender import *
  152. from Blender import NMesh
  153. from Blender.Draw import *
  154. from Blender.BGL import *
  155. from Blender import Material
  156. from Blender import Window
  157.  
  158.  
  159.  
  160. # ===============================
  161. # Input Variables
  162. # ===============================
  163.  
  164. Filename = "G:\\tmp\\t\\t\\quip.obj"
  165.  
  166. gFilename=Create(Filename)
  167. gAlert   = 0
  168. type     = 1
  169. exporttype = 1
  170. returncode = 0
  171. operation = "Export"
  172. center = [0,0,0]
  173. rotation = [0,0,0]
  174. Transform = []
  175. multiflag = 0
  176.  
  177. #================================
  178. # def Fileselect function:
  179. #================================
  180. def ImportFunctionselet(filename):
  181.              global gFilename
  182.              global ExportOptions
  183.              global ExportType
  184.              global type
  185.              global exporttype
  186.              global operation
  187.              global gAlert
  188.              gFilename.val=filename
  189.              ImportFunction(filename, type)
  190.              operation = "Import"
  191.  
  192. #================================
  193. def ExitGUI ():
  194. #================================
  195.     Exit()
  196.  
  197. #================================
  198. def EventGUI (event):
  199. #================================
  200.       global gFilename
  201.       global ExportOptions
  202.       global ExportType
  203.       global type
  204.       global exporttype
  205.       global operation
  206.       global gAlert
  207.  
  208.       if (event==EVENT_IMPORT):
  209.          ImportFunction(gFilename.val, type)
  210.          operation = "Import"
  211.  
  212.       if (event==EVENT_IMPORT_SELECT):        
  213.          Window.FileSelector (ImportFunctionselet, 'IMPORT FILE')
  214.  
  215.  
  216.       if (event==EVENT_IMPORT_CONT):
  217.          gAlert = 0
  218.          operation = "Import"
  219.          Draw ()
  220.  
  221.       if (event==EVENT_EXPORT):
  222.          ExportFunction(gFilename.val, type)
  223.          operation = "Export"
  224.  
  225.       if (event==EVENT_EXPORT_CHK):
  226.          ExportFunctionOK(gFilename.val, type)
  227.          Draw ()
  228.       if (event==EVENT_EXPORT_CANCEL):
  229.          gAlert = 0
  230.          Draw ()
  231.       if (event==EVENT_OPTIONS):
  232.          type = ExportOptions.val
  233.          Draw ()
  234.       if (event==EVENT_TYPE):
  235.          exporttype = ExportType.val
  236.          Draw ()
  237.       if (event==EVENT_EXPORT_ERR):
  238.          gAlert = 0
  239.          Draw ()
  240.       if (event==EVENT_DONE):  
  241.          gAlert = 0
  242.          Draw ()
  243.       if (event==EVENT_QUIT):
  244.          ExitGUI()
  245.  
  246. #================================
  247. def DrawGUI():
  248. #================================
  249.       global type
  250.       global exporttype
  251.       global operation
  252.  
  253.       glClearColor (0.6,0.6,0.6,0)
  254.       glClear (GL_COLOR_BUFFER_BIT)
  255.  
  256.       global gFilename
  257.       global gAlert
  258.       global ExportOptions
  259.       global ExportType
  260.  
  261.       if (gAlert==0):
  262.          # Add in the copyright notice and title
  263.          glRasterPos2d(32, 380)
  264.          Text("Wavefront OBJ Importer/Exporter")
  265.          glRasterPos2d(32, 350)
  266.          Text("Copyright (C) Chris Lynch 2001")
  267.  
  268.          gFilename=String ("Filename: ",EVENT_PATHCHANGE,32,250,320,32,gFilename.val,255,"Full pathname and filename")
  269.          Button ("Export",EVENT_EXPORT,32,200,100,32)
  270.          Button ("Import",EVENT_IMPORT,252,200,100,32)
  271.          Button ("Select Import",EVENT_IMPORT_SELECT,355,200,100,32)
  272.          glRasterPos2d(32, 165)
  273.          Text("Select Export Options:")
  274.          options = "Export Options %t| Default %x1| Material Layers %x2| Obj Groups %x3| Standard %x4"
  275.          ExportOptions = Menu (options,EVENT_OPTIONS,200,150,150,32, type)
  276.          Button ("Done",EVENT_QUIT,142,50,100,32)
  277.          glRasterPos2d(32, 115)
  278.          Text("Export using ")
  279.          options = "Export Type %t| Mesh Coordinates %x1| Object Coordinates %x2"
  280.          ExportType = Menu (options,EVENT_TYPE,170,100,180,32, exporttype)
  281.          Button ("Done",EVENT_QUIT,142,50,100,32)
  282.  
  283.       elif (gAlert==1):
  284.          glRasterPos2i (32,250)
  285.          Text (gFilename.val+ " already exists. Save anyway?")
  286.          Button ("Save",EVENT_EXPORT_CHK,150,200,50,32)
  287.          Button ("Cancel",EVENT_EXPORT_CANCEL,250,200,50,32)
  288.          gAlert = 0
  289.       elif (gAlert==2):
  290.          glRasterPos2i (32,250)
  291.          Text (gFilename.val+ " cannot be found. Check directory and filename.")
  292.          Button ("Continue",EVENT_IMPORT_CONT,32,190,70,32)
  293.          gAlert = 0
  294.       elif gAlert == 3:
  295.          glRasterPos2i (32,250)
  296.          Text ("No objects selected to export. You must select one or more objects.")
  297.          Button ("Continue",EVENT_EXPORT_ERR,192,200,70,32)
  298.          gAlert = 0
  299.       elif gAlert == 5:
  300.          glRasterPos2i (32,250)
  301.          Text ("Invalid directory path.")
  302.          Button ("Continue",EVENT_EXPORT_ERR,192,200,70,32)
  303.          gAlert = 0
  304.       else:
  305.          glRasterPos2i (32,250)
  306.          Text (str(operation)+ " of " +str(gFilename.val)+ " done.")
  307.          Button ("Continue",EVENT_DONE,192,200,70,32)
  308.          
  309. #================================
  310. def RegisterGUI ():
  311. #================================
  312.     Register (DrawGUI,None,EventGUI)
  313.  
  314. #================================
  315. # MAIN SCRIPT
  316. #================================
  317. # Opens a file, writes data in it
  318. # and closes it up.
  319. #================================
  320. RegisterGUI()
  321.  
  322. #================================
  323. def ImportFunction (importName, type):
  324. #================================      
  325.       global gFilename
  326.       global gAlert
  327.  
  328.       try:
  329.          FILE=open (importName,"r")
  330.          
  331.          directory, Name = os.split(gFilename.val)
  332.          
  333.          
  334.          words = Name.split(".")
  335.          Name = words[0]
  336.          ObjImport(FILE, Name, gFilename.val)
  337.          FILE.close()
  338.          gAlert = 4
  339.          Draw ()
  340.       except IOError:
  341.          gAlert=2
  342.          Draw ()
  343.  
  344. #================================
  345. def ExportFunction (exportName, type):
  346. #================================      
  347.       global gFilename
  348.       global gAlert
  349.  
  350.       try:
  351.          FILE=open (exportName,"r")
  352.          FILE.close()
  353.          gAlert = 1
  354.          Draw ()
  355.       except IOError:
  356.  
  357.          directory, Name = os.split(gFilename.val)
  358.  
  359.          
  360.          if os.isdir(directory):
  361.             ExportFunctionOK(exportName, type)
  362.             Draw ()
  363.          else:
  364.             gAlert = 5
  365.             Draw ()
  366.  
  367. #================================
  368. def ExportFunctionOK (exportName, type):
  369. #================================      
  370.       global gFilename
  371.       global gAlert
  372.       global returncode
  373.  
  374.       FILE=open (exportName,"w")
  375.  
  376.       directory, Name = os.split(gFilename.val)
  377.      
  378.       words = Name.split(".")
  379.       Name = words[0]
  380.       ObjExport(FILE, Name, type)
  381.       if returncode > 0:
  382.          gAlert = 3
  383.       else:
  384.          gAlert = 4
  385.       FILE.flush()
  386.       FILE.close()
  387.  
  388. #=========================
  389. def ObjImport(file, Name, filename):
  390. #=========================  
  391.     vcount     = 0
  392.     vncount    = 0
  393.     vtcount    = 0
  394.     fcount     = 0
  395.     gcount     = 0
  396.     setcount   = 0
  397.     groupflag  = 0
  398.     objectflag = 0
  399.     mtlflag    = 0
  400.     baseindex  = 0
  401.     basevtcount = 0
  402.     basevncount = 0
  403.     matindex   = 0
  404.  
  405.     pointList    = []
  406.     uvList       = []
  407.     normalList   = []
  408.     faceList     = []
  409.     materialList = []
  410.     imagelist    = []
  411.    
  412.     uv = []
  413.     lines = file.readlines()
  414.     linenumber = 1
  415.  
  416.     for line in lines:
  417.         words = line.split()
  418.         if words and words[0] == "#":
  419.             pass # ignore comments
  420.         elif words and words[0] == "v":
  421.             vcount = vcount + 1
  422.             x = float(words[1])
  423.             y = float(words[2])
  424.             z = float(words[3])
  425.             pointList.append([x, y, z])
  426.  
  427.         elif words and words[0] == "vt":
  428.             vtcount = vtcount + 1
  429.             u = float(words[1])
  430.             v = float(words[2])
  431.             uvList.append([u, v])
  432.  
  433.         elif words and words[0] == "vn":
  434.             vncount = vncount + 1
  435.             i = float(words[1])
  436.             j = float(words[2])
  437.             k = float(words[3])
  438.             normalList.append([i, j, k])
  439.  
  440.         elif words and words[0] == "f":
  441.             fcount = fcount + 1
  442.             vi = [] # vertex  indices
  443.             ti = [] # texture indices
  444.             ni = [] # normal  indices
  445.             words = words[1:]
  446.             lcount = len(words)
  447.             for index in (xrange(lcount)):
  448.                if words[index].find( "/") == -1:
  449.                      vindex = int(words[index])
  450.                      if vindex < 0: vindex = baseindex + vindex + 1  
  451.                      vi.append(vindex)
  452.                else:
  453.                    vtn = words[index].split( "/")
  454.                    vindex = int(vtn[0])
  455.                    if vindex < 0: vindex = baseindex + vindex + 1
  456.                    vi.append(vindex)
  457.            
  458.                    if len(vtn) > 1 and vtn[1]:
  459.                       tindex = int(vtn[1])
  460.                       if tindex < 0: tindex = basevtcount +tindex + 1
  461.                       ti.append(tindex)
  462.  
  463.                    if len(vtn) > 2 and vtn[2]:
  464.                       nindex = int(vtn[2])
  465.                       if nindex < 0: nindex = basevncount +nindex + 1
  466.                       ni.append(nindex)
  467.             faceList.append([vi, ti, ni, matindex])
  468.  
  469.         elif words and words[0] == "o":
  470.             ObjectName = words[1]
  471.             objectflag = 1
  472.             #print "Name is %s" % ObjectName
  473.  
  474.         elif words and words[0] == "g":
  475.             groupflag = 1
  476.             index = len(words)
  477.             if objectflag == 0:
  478.                objectflag = 1
  479.                if index > 1:
  480.                   ObjectName = words[1].join("_")
  481.                   GroupName = words[1].join("_")
  482.                else:
  483.                   ObjectName = "Default"
  484.                   GroupName = "Default"
  485.                #print "Object name is %s" % ObjectName
  486.                #print "Group name is %s" % GroupName
  487.             else:
  488.                if index > 1:
  489.                   GroupName = join(words[1],"_")
  490.                else:
  491.                   GroupName = "Default"
  492.                #print "Group name is %s" % GroupName
  493.                  
  494.             if mtlflag == 0:
  495.                matindex = AddMeshMaterial(GroupName,materialList, matindex)
  496.             gcount = gcount + 1
  497.                
  498.             if fcount > 0:
  499.                baseindex = vcount
  500.                basevncount = vncount
  501.                basevtcount = vtcount
  502.  
  503.         elif words and words[0] == "mtllib":
  504.             # try to export materials
  505.             directory, dummy = os.split(filename)
  506.             filename = os.join(directory, words[1])
  507.             print  "try to import : ",filename
  508.             try:
  509.                 file = open(filename, "r")
  510.             except:
  511.                 print "no material file %s" % filename
  512.             else:
  513.                 mtlflag = 0
  514.                 file = open(filename, "r")
  515.                 line = file.readline()
  516.                 mtlflag = 1
  517.                 while line:
  518.                     words = line.split()
  519.                     if words and words[0] == "newmtl":
  520.                       name = words[1]
  521.                       line = file.readline()  # Ns ?
  522.                       words = line.split()
  523.                       while words[0] not in ["Ka","Kd","Ks","map_Kd"]:
  524.                           line = file.readline()
  525.                           words = line.split()
  526.                              
  527.                       if words[0] == "Ka":
  528.                         Ka = [float(words[1]),
  529.                               float(words[2]),
  530.                               float(words[3])]
  531.                         line = file.readline()  # Kd
  532.                         words = line.split()
  533.                        
  534.                       if words[0] == "Kd":
  535.                         Kd = [float(words[1]),
  536.                               float(words[2]),
  537.                               float(words[3])]
  538.                         line = file.readline()  # Ks
  539.                         words = line.split()
  540.                          
  541.                       if words[0] == "Ks":
  542.                         Ks = [float(words[1]),
  543.                               float(words[2]),
  544.                               float(words[3])]
  545.                          
  546.                       if words[0] == "map_Kd":
  547.                         Kmap= words[1]
  548.                         img=os.join(directory, Kmap)
  549.                         im=Blender.Image.Load(img)
  550.                         words = line.split()
  551.                          
  552.                       matindex = AddGlobalMaterial(name, matindex)                            
  553.                       matlist = Material.Get()
  554.                        
  555.                       if len(matlist) > 0:
  556.                          if name!='defaultMat':
  557.                              material = matlist[matindex]
  558.                              material.R = Kd[0]
  559.                              material.G = Kd[1]
  560.                              material.B = Kd[2]
  561.                              try:
  562.                                   material.specCol[0] = Ks[0]
  563.                                   material.specCol[1] = Ks[1]
  564.                                   material.specCol[2] = Ks[2]
  565.                              except:
  566.                                   pass
  567.                              try:
  568.                                   alpha = 1 - ((Ka[0]+Ka[1]+Ka[2])/3)
  569.                              except:
  570.                                   pass
  571.                              try:
  572.                                   material.alpha = alpha
  573.                              except:
  574.                                   pass
  575.  
  576.                              try:
  577.                                  
  578.                                  img=os.join(directory, Kmap)
  579.                                  im=Blender.Image.Load(img)
  580.                                  imagelist.append(im)
  581.                              
  582.                                  t=Blender.Texture.New(Kmap)
  583.                                  t.setType('Image')
  584.                                  t.setImage(im)
  585.                              
  586.                                  material.setTexture(0,t)
  587.                                  material.getTextures()[0].texco=16
  588.                              except:
  589.                                   pass
  590.                                
  591.                          else:
  592.                              material = matlist[matindex]
  593.                              
  594.                              material.R = 0.8
  595.                              material.G = 0.8
  596.                              material.B = 0.8
  597.                              material.specCol[0] = 0.5
  598.                              material.specCol[1] = 0.5
  599.                              material.specCol[2] = 0.5
  600.                              
  601.                              img=os.join(directory, Kmap)
  602.                              im=Blender.Image.Load(img)
  603.                              imagelist.append(im)
  604.                              
  605.                              t=Blender.Texture.New(Kmap)
  606.                              t.setType('Image')
  607.                              t.setImage(im)
  608.                              
  609.                              material.setTexture(0,t)
  610.                              material.getTextures()[0].texco=16
  611.                        
  612.                       else:
  613.                          mtlflag = 0
  614.                              
  615.                     line = file.readline()
  616.                          
  617.                        
  618.                 file.close()
  619.                  
  620.         elif words and words[0] == "usemtl":
  621.             if mtlflag == 1:
  622.                name = words[1]
  623.                matindex = AddMeshMaterial(name, materialList, matindex)
  624.         elif words:  
  625.             print "%s: %s" % (linenumber, words)
  626.         linenumber = linenumber + 1
  627.     file.close()
  628.  
  629.     # import in Blender
  630.  
  631.     print "import into Blender ..."
  632.     mesh   = NMesh.GetRaw ()
  633.  
  634.     i = 0
  635.     while i < vcount:
  636.       x, y, z = pointList[i]
  637.       vert=NMesh.Vert(x, y, z)
  638.       mesh.verts.append(vert)
  639.       i=i+1
  640.  
  641.     if vtcount > 0:
  642.        #mesh.hasFaceUV() = 1
  643.        print ("Object has uv coordinates")
  644.  
  645.     if len(materialList) > 0:
  646.        for m in materialList:
  647.           try:
  648.             M=Material.Get(m)
  649.             mesh.materials.append(M)
  650.           except:
  651.             pass
  652.  
  653.     total = len(faceList)
  654.     i = 0
  655.  
  656.     for f in faceList:
  657.         if i%1000 == 0:
  658.           print ("Progress = "+ str(i)+"/"+ str(total))
  659.  
  660.         i = i + 1
  661.         vi, ti, ni, matindex = f
  662.         face=NMesh.Face()
  663.         if len(materialList) > 0:
  664.            face.mat = matindex
  665.  
  666.         limit = len(vi)
  667.         setcount = setcount + len(vi)
  668.         c = 0    
  669.    
  670.         while c < limit:
  671.           m = vi[c]-1
  672.           if vtcount > 0 and len(ti) > c:
  673.              n = ti[c]-1
  674.           if vncount > 0 and len(ni) > c:
  675.              p = ni[c]-1
  676.  
  677.           if vtcount > 0:
  678.              try:
  679.                   u, v = uvList[n]
  680.              except:
  681.                   pass
  682.  
  683.              """
  684.        #  multiply uv coordinates by 2 and add 1. Apparently blender uses uv range of 1 to 3 (not 0 to 1).
  685.             mesh.verts[m].uvco[0] = (u*2)+1
  686.             mesh.verts[m].uvco[1] = (v*2)+1
  687.            """
  688.  
  689.           if vncount > 0:
  690.              if p > len(normalList):
  691.                 print("normal len = " +str(len(normalList))+ " vector len = " +str(len(pointList)))
  692.                 print("p = " +str(p))
  693.              x, y, z = normalList[p]  
  694.              mesh.verts[m].no[0] = x
  695.              mesh.verts[m].no[1] = y
  696.              mesh.verts[m].no[2] = z
  697.           c = c+1  
  698.      
  699.         if len(vi) < 5:
  700.           for index in vi:
  701.             face.v.append (mesh.verts[index-1])
  702.  
  703.           if vtcount > 0:  
  704.             for index in ti:
  705.                u, v = uvList[index-1]
  706.                face.uv.append((u,v))
  707.                
  708.             if len(imagelist)>0:
  709.                 face.image=imagelist[0]
  710.                 #print
  711.                
  712.           if vcount>0:
  713.              face.smooth=1
  714.  
  715.           mesh.faces.append(face)
  716.  
  717.     print "all other (general) polygons ..."
  718.     for f in faceList:
  719.         vi, ti, ni, matindex = f
  720.         if len(vi) > 4:
  721.             # export the polygon as edges
  722.             print ("Odd face, vertices = "+ str(len(vi)))
  723.             for i in range(len(vi)-2):
  724.                face = NMesh.Face()
  725.                if len(materialList) > 0:
  726.                   face.mat = matindex
  727.                face.v.append(mesh.verts[vi[0]-1])
  728.                face.v.append(mesh.verts[vi[i+1]-1])
  729.                face.v.append(mesh.verts[vi[i+2]-1])
  730.  
  731.                if vtcount > 0:
  732.                   if len(ti) > i+2:
  733.                      u, v = uvList[ti[0]-1]
  734.                      face.uv.append((u,v))
  735.                      u, v = uvList[ti[i+1]-1]
  736.                      face.uv.append((u,v))
  737.                      u, v = uvList[ti[i+2]-1]
  738.                      face.uv.append((u,v))
  739.  
  740.                mesh.faces.append(face)
  741.      
  742.     NMesh.PutRaw(mesh, Name,1)
  743.  
  744.     print ("Total number of vertices is "+ str(vcount))
  745.     print ("Total number of faces is "+ str(len(faceList)))
  746.     print ("Total number of sets is "+ str(setcount))
  747.  
  748.  
  749.     print("Finished importing " +str(Name)+ ".obj")
  750.  
  751. #=========================================
  752. def AddMeshMaterial(name, materialList, matindex):
  753. #=========================================
  754.    
  755.    index = 0
  756.    found = 0
  757.    limit = len(materialList)
  758.  
  759.    while index < limit:
  760.      if materialList[index] == name:
  761.         matindex = index
  762.         found = 1
  763.         index = limit
  764.      index = index + 1
  765.    
  766.    if found == 0:      
  767.       materialList.append(name)
  768.       matindex = len(materialList)-1
  769.        
  770.    return matindex
  771.  
  772. #=========================================
  773. def AddGlobalMaterial (name, matindex):
  774. #=========================================
  775.    
  776.    index = 0
  777.    found = 0
  778.    matindex  = 0
  779.    MatList = Material.Get()
  780.    limit = len(MatList)
  781.  
  782.    while index < limit:
  783.      if MatList[index].name == name:
  784.         matindex = index
  785.         found = 1
  786.         index = limit
  787.      index = index + 1
  788.  
  789.    if found == 0:
  790.       material = Material.New(name)
  791.       matindex = index
  792.    
  793.    return matindex
  794.  
  795. #================================
  796. def ObjExport(FILE, Name, type):
  797. #================================
  798.   global returncode
  799.   global vertexcount
  800.   global uvcount
  801.   global Transform
  802.   global multiflag
  803.   global exporttype
  804.  
  805.   vertexcount = 0
  806.   uvcount = 0
  807.   returncode = 0
  808.   print("Writing %s..." % Name)
  809.   FILE.write("# Wavefront OBJ (1.0) exported by lynx's OBJ import/export script\n\n")
  810.  
  811.   Objects = Object.GetSelected()
  812.   if Objects == []:
  813.      print("You have not selected an object!")
  814.      returncode = 4
  815.   else:
  816.      for object in Objects:
  817.         MtlList = []
  818.         if len(Objects) > 1 or exporttype > 1:
  819.            Transform = CreateMatrix(object, Transform)
  820.            multiflag = 1
  821.            
  822.         mesh = NMesh.GetRawFromObject(object.name)
  823.         ObjName = mesh.name
  824.         has_uvco = mesh.hasVertexUV()
  825.  
  826.         FILE.write("# Meshname:\t%s\n" % ObjName)
  827.  
  828.         faces = mesh.faces
  829.         materials = mesh.materials
  830.         Vertices = mesh.verts
  831.         GlobalMaterials = Material.Get()
  832.  
  833.         if len(materials) > 1 and len(GlobalMaterials) > 0 and type < 4:
  834.            CreateMtlFile(Name, materials, MtlList)
  835.  
  836.         # Total Vertices and faces; comment if not useful
  837.         FILE.write("# Total number of Faces:\t%s\n" % len(faces))
  838.         FILE.write("# Total number of Vertices:\t%s\n" % len(Vertices))
  839.  
  840.         FILE.write("\n")
  841.  
  842.         # print first image map for uvcoords to use
  843.         # to be updated when we get access to other textures
  844.         if mesh.hasFaceUV(): FILE.write("# UV Texture:\t%s\n\n" % mesh.hasFaceUV())
  845.  
  846.         if len(materials) > 1 and len(GlobalMaterials) > 0 and type < 3:
  847.            UseLayers(faces, Vertices, MtlList, has_uvco, FILE, ObjName, Name)
  848.         elif len(materials) > 1 and len(GlobalMaterials) > 0 and type == 3:
  849.            UseMtl(faces, Vertices, MtlList, has_uvco, FILE, ObjName, Name)
  850.         else:
  851.            Standard(faces, Vertices, has_uvco, FILE, ObjName)
  852.  
  853. #================================================
  854. def CreateMtlFile (name, MeshMaterials, MtlList):
  855. #================================================
  856.       global gFilename
  857.  
  858.     # try to export materials
  859.       directory, mtlname = os.split(gFilename.val)
  860.       mtlname = name + ".mtl"
  861.       filename = os.join(directory, mtlname)
  862.       file = open(filename, "w")
  863.  
  864.       file.write("# Materials for %s.\n" % (name + ".obj"))
  865.       file.write("# Created by Blender.\n")
  866.       file.write("# These files must be in the same directory for the materials to be read correctly.\n\n")
  867.  
  868.       MatList = Material.Get()
  869.       print str(MeshMaterials)
  870.  
  871.       MtlNList=[]
  872.       for m in  MatList:
  873.          MtlNList.append(m.name)
  874.  
  875.       counter = 1
  876.       found = 0  
  877.  
  878.       for material in MeshMaterials:
  879.          for mtl in MtlList:
  880.             if material == mtl:
  881.                 found = 1
  882.  
  883.          MtlList.append(material)
  884.  
  885.          if found == 0:
  886.             file.write("newmtl %s \n" % material.name)
  887.             index = 0
  888.             print material, MatList
  889.             while index < len(MatList):
  890.                if material.name == MatList[index].name:
  891.                   mtl = MatList[index]
  892.                   index = len(MatList)
  893.                   found = 1
  894.                index = index + 1
  895.  
  896.             if found == 1:
  897.                alpha = mtl.getAlpha()
  898.                file.write("       Ka %s %s %s \n" % (round(1-alpha,5), round(1-alpha,5), round(1-alpha,5)))
  899.                file.write("       Kd %s %s %s \n" % (round(mtl.R,5), round(mtl.G,5), round(mtl.B,5)))
  900.                file.write("       Ks %s %s %s \n" % (round(mtl.specCol[0],5), round(mtl.specCol[1],5), round(mtl.specCol[2],5)))
  901.                file.write("       illum 1\n")
  902.                
  903.             else:
  904.                file.write("       Ka %s %s %s \n" % (0, 0, 0))
  905.                file.write("       Kd %s %s %s \n" % (1, 1, 1))
  906.                file.write("       Ks %s %s %s \n" % (1, 1, 1))
  907.                file.write("       illum 1\n")
  908.  
  909.          found = 0
  910.  
  911.       file.flush()
  912.       file.close()
  913.  
  914. #===========================================================
  915. def Standard(faces, Vertices, has_uvco, FILE, ObjName):
  916. #===========================================================
  917.        global vertexcount
  918.        global uvcount
  919.        global multiflag
  920.  
  921.        uvPtrs = []
  922.        uvList = []
  923.  
  924.        FILE.write("o %s\n\n" % (ObjName))
  925.        FILE.write("g %s\n\n" % (ObjName))
  926.  
  927.        for v in Vertices:
  928.           vert = v.co
  929.           if multiflag  == 1:
  930.              vert = Alter(vert, Transform)
  931.           x, y, z = vert
  932.                
  933.           FILE.write("v %s %s %s\n" % (x, y, z))
  934.  
  935.        uv_flag = 0
  936.        for face in faces:
  937.          for uv in face.uv:
  938.             found = 0
  939.             index = len(uvList)
  940.             limit = 0
  941.             if len(uvList)-200 > 0:
  942.                limit = len(uvList)-200
  943.             while index > limit and found == 0:
  944.                uv_value = uvList[index-1]
  945.                if uv[0] == uv_value[0] and uv[1] == uv_value[1]:
  946.                   uvPtrs.append(index+uvcount)
  947.                   found = 1
  948.                index = index - 1
  949.             if found == 0:
  950.                uvList.append(uv)
  951.                index = len(uvList)
  952.                uvPtrs.append(index+uvcount)
  953.                u, v = uv
  954.                FILE.write("vt %s %s\n" % (u, v))
  955.                uv_flag = 1
  956.  
  957.        if has_uvco and uv_flag == 0:
  958.          for v in Vertices:
  959.             u, v, z = v.uvco
  960.             u = (u-1)/2
  961.             v = (v-1)/2
  962.             FILE.write("vt %s %s\n" % (u, v))
  963.  
  964.        for v in Vertices:
  965.           x, y, z = v.no
  966.           FILE.write("vn %s %s %s\n" % (x, y, z))
  967.  
  968.        p = 0
  969.        uvindex = 0
  970.        total = len(faces)
  971.  
  972.        for face in faces:
  973.           p = p+1
  974.           if (p%1000) == 0:
  975.               print ("Progress = "+ str(p)+ " of "+ str(total) +" faces")
  976.  
  977.           FILE.write("f ")
  978.           for index in range(len(face.v)):
  979.              v = face.v[index].index + vertexcount
  980.              if len(face.uv) > 0:
  981.                 FILE.write("%s/%s/%s " % (v+1, uvPtrs[uvindex], v+1))
  982.                 uvindex = uvindex+1
  983.              elif has_uvco:
  984.                 FILE.write("%s/%s/%s " % (v+1, v+1, v+1))
  985.              else:                    
  986.                 FILE.write("%s//%s " % (v+1, v+1))
  987.           FILE.write("\n")
  988.  
  989.        vertexcount = vertexcount + len(Vertices)
  990.        uvcount = uvcount + len(uvList)
  991.  
  992.        print("Export of " +str(ObjName)+ ".obj finished.\n")
  993.  
  994. #=====================================================================
  995. def UseLayers(faces, Vertices, MtlList, has_uvco, FILE, ObjName, Name):
  996. #=====================================================================
  997.        global vertexcount
  998.        global uvcount
  999.        global multiflag
  1000.  
  1001.        uvPtrs = []
  1002.        uvList = []
  1003.  
  1004.        FILE.write("mtllib %s\n\n" % (Name + ".mtl"))
  1005.        FILE.write("g %s\n\n" % (ObjName))
  1006.  
  1007.        for v in Vertices:
  1008.           vert = v.co
  1009.           if multiflag  == 1:
  1010.              vert = Alter(vert, Transform)  
  1011.           x, y, z = vert
  1012.           FILE.write("v %s %s %s\n" % (x, y, z))
  1013.  
  1014.        uv_flag = 0
  1015.        for m in range(len(MtlList)):
  1016.           for face in faces:
  1017.               if face.mat == m:
  1018.                  for uv in face.uv:
  1019.                     found = 0
  1020.                     index = len(uvList)
  1021.                     limit = 0
  1022.                     if len(uvList)-200 > 0:
  1023.                        limit = len(uvList)-200
  1024.                     while index > limit and found == 0:
  1025.                        uv_value = uvList[index-1]
  1026.                        if uv[0] == uv_value[0] and uv[1] == uv_value[1]:
  1027.                           uvPtrs.append(index+uvcount)
  1028.                           found = 1
  1029.                        index = index - 1
  1030.                     if found == 0:
  1031.                        uvList.append(uv)
  1032.                        index = len(uvList)
  1033.                        uvPtrs.append(index+uvcount)
  1034.                        u, v = uv
  1035.                        FILE.write("vt %s %s\n" % (u, v))
  1036.                        uv_flag = 1
  1037.  
  1038.        if has_uvco and uv_flag == 0:
  1039.          for v in Vertices:
  1040.             u, v, z = v.uvco
  1041.             u = (u-1)/2
  1042.             v = (v-1)/2
  1043.             FILE.write("vt %s %s\n" % (u, v))
  1044.  
  1045.        for v in Vertices:
  1046.           x, y, z = v.no
  1047.           FILE.write("vn %s %s %s\n" % (x, y, z))
  1048.  
  1049.        total = len(faces)
  1050.        p = 0
  1051.        uvindex = 0
  1052.        for m in range(len(MtlList)):        
  1053.           FILE.write("usemtl %s\n" % (MtlList[m].name))
  1054.           for face in faces:
  1055.               if face.mat == m:
  1056.                 p = p+1
  1057.                 if (p%1000) == 0:
  1058.                    print ("Progress = "+ str(p)+ " of "+ str(total) +" faces")
  1059.  
  1060.                 FILE.write("f ")
  1061.                 for index in range(len(face.v)):
  1062.                    v = face.v[index].index + vertexcount
  1063.                    if len(face.uv) > 0:
  1064.                       FILE.write("%s/%s/%s " % (v+1, uvPtrs[uvindex], v+1))
  1065.                       uvindex = uvindex+1
  1066.                    elif has_uvco:
  1067.                       FILE.write("%s/%s/%s " % (v+1, v+1, v+1))
  1068.                    else:
  1069.                       FILE.write("%s//%s " % (v+1, v+1))
  1070.                 FILE.write("\n")
  1071.  
  1072.        vertexcount = vertexcount + len(Vertices)
  1073.        print("Export of " +str(ObjName)+ ".obj using material layers finished.\n")
  1074.  
  1075. #==================================================================
  1076. def UseMtl(faces, Vertices, MtlList, has_uvco, FILE, ObjName, Name):
  1077. #==================================================================
  1078.        global vertexcount
  1079.        global multiflag
  1080.  
  1081.        FILE.write("mtllib %s\n\n" % (Name + ".mtl"))
  1082.        FILE.write("o %s\n\n" % (ObjName))
  1083.        
  1084.        index = 0
  1085.        VertexList = []
  1086.        for vertex in Vertices:
  1087.           VertexList.append(-1)
  1088.           index = index + 1
  1089.        print("number of vertices is " +str(len(VertexList)))
  1090.  
  1091.        Totalindex = 0
  1092.        ix = 0
  1093.        NewVertexList = []
  1094.        NewVertexCo = []
  1095.        for m in range(len(MtlList)):
  1096.            # Group name is the name of the mesh
  1097.            if MtlList[m]:
  1098.               FILE.write("g %s\n" % (MtlList[m].name+str(m+1)))
  1099.            else:
  1100.               FILE.write("g %s\n" % ("Null"+str(m+1)))
  1101.            FILE.write("s off\n\n")
  1102.          
  1103.            FILE.write("usemtl %s\n\n" % (MtlList[m].name))
  1104.  
  1105.            for face in faces:
  1106.               if face.mat == m:
  1107.                  for vertex in face.v:
  1108.                     v = vertex.index
  1109.                     if VertexList[v] < 0:
  1110.                        VertexList[v] = Totalindex
  1111.                        NewVertexList.append(v)
  1112.                        Totalindex = Totalindex + 1
  1113.  
  1114.            for v_old in NewVertexList:
  1115.               vert = Vertices[v_old].co
  1116.               if multiflag  == 1:
  1117.                 vert = Alter(vert, Transform)
  1118.               x, y, z = vert
  1119.               FILE.write("v %s %s %s\n" % (x, y, z))
  1120.               NewVertexCo.append([x,y,z])
  1121.  
  1122.            if has_uvco:
  1123.               for v_old in NewVertexList:
  1124.                  u, v, z = Vertices[v_old].uvco
  1125.                  u = (u-1)/2
  1126.                  v = (v-1)/2              
  1127.                  FILE.write("vt %s %s\n" % (u, v))
  1128.  
  1129.            for v_old in NewVertexList:
  1130.               x, y, z = Vertices[v_old].no
  1131.               FILE.write("vn %s %s %s\n" % (x, y, z))
  1132.  
  1133.            for face in faces:
  1134.              if face.mat == m:
  1135.                 FILE.write("f ")
  1136.                 for index in range(len(face.v)):
  1137.                    v = face.v[index].index
  1138.                    v_new = VertexList[v]
  1139.                    if has_uvco:
  1140.                       FILE.write("%s/%s/%s " % (v_new+1, v_new+1, v_new+1))
  1141.                    else:
  1142.                       FILE.write("%s//%s " % (v_new+1, v_new+1))
  1143.                 FILE.write("\n")
  1144.  
  1145.            FILE.write("\n")
  1146.  
  1147.            NewVertexList = []
  1148.            print("Group " +str(m+1)+ " of " +str(len(MtlList))+ " finished.")
  1149.  
  1150.        print("Export of " +str(ObjName)+ ".obj using groups finished.\n")
  1151.  
  1152. #========================================
  1153. def CreateMatrix(object, Transform):
  1154. #========================================
  1155.    Mx = []
  1156.    My = []
  1157.    Mz = []
  1158.    T1 = []
  1159.    Transform = []
  1160.  
  1161.    angle = object.RotX
  1162.    Mx.append([1, 0, 0])
  1163.    y = math.cos(angle)
  1164.    z = -math.sin(angle)
  1165.    Mx.append([0, y, z])
  1166.    y = math.sin(angle)
  1167.    z = math.cos(angle)
  1168.    Mx.append([0, y, z])
  1169.  
  1170.    angle = object.RotY
  1171.    x = math.cos(angle)
  1172.    z = math.sin(angle)
  1173.    My.append([x, 0, z])
  1174.    My.append([0, 1, 0])
  1175.    x = -math.sin(angle)
  1176.    z = math.cos(angle)
  1177.    My.append([x, 0, z])
  1178.  
  1179.    angle = object.RotZ
  1180.    x = math.cos(angle)
  1181.    y = -math.sin(angle)
  1182.    Mz.append([x, y, 0])
  1183.    x = math.sin(angle)
  1184.    y = math.cos(angle)
  1185.    Mz.append([x, y, 0])
  1186.    Mz.append([0, 0, 1])
  1187.  
  1188.    m0 = Mx[0]
  1189.    m1 = Mx[1]
  1190.    m2 = Mx[2]
  1191.    for row in My:
  1192.       x, y, z = row
  1193.       nx = x*m0[0] + y*m1[0] + z*m2[0]
  1194.       ny = x*m0[1] + y*m1[1] + z*m2[1]
  1195.       nz = x*m0[2] + y*m1[2] + z*m2[2]
  1196.       T1.append([nx, ny, nz])
  1197.  
  1198.    m0 = T1[0]
  1199.    m1 = T1[1]
  1200.    m2 = T1[2]
  1201.    for row in Mz:
  1202.      x, y, z = row
  1203.      nx = x*m0[0] + y*m1[0] + z*m2[0]
  1204.      ny = x*m0[1] + y*m1[1] + z*m2[1]
  1205.      nz = x*m0[2] + y*m1[2] + z*m2[2]
  1206.      Transform.append([nx, ny, nz])
  1207.  
  1208.    Transform.append([object.SizeX, object.SizeY, object.SizeZ])
  1209.    Transform.append([object.LocX, object.LocY, object.LocZ])
  1210.  
  1211.    return Transform
  1212.  
  1213. #======================================
  1214. def Alter(vect, Transform):
  1215. #======================================
  1216.    v2 = []
  1217.    nv = []
  1218.  
  1219.    x, y, z = vect
  1220.    sx, sy, sz = Transform[3]
  1221.    lx, ly, lz = Transform[4]
  1222.  
  1223.    v2.append(x*sx)
  1224.    v2.append(y*sy)
  1225.    v2.append(z*sz)
  1226.  
  1227.    for index in range(len(vect)):
  1228.       t = Transform[index]
  1229.       nv.append(v2[0]*t[0] + v2[1]*t[1] +v2[2]*t[2])
  1230.  
  1231.    nv[0] = nv[0]+lx
  1232.    nv[1] = nv[1]+ly
  1233.    nv[2] = nv[2]+lz
  1234.  
  1235.    return nv
Advertisement
Add Comment
Please, Sign In to add comment
Public Pastes
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy.  OK, I Understand
Not a member of Pastebin yet?
Sign Up, it unlocks many cool features!