home *** CD-ROM | disk | FTP | other *** search
/ Windows News 2005 November / WNnov2005.iso / Windows / Equipement / Blender / blender-2.37a-windows.exe / $_5_ / .blender / scripts / tex2uvbaker.py < prev    next >
Text File  |  2005-06-13  |  16KB  |  499 lines
  1. #!BPY
  2.  
  3. """ Registration info for Blender menus:
  4. Name: 'Texture Baker'
  5. Blender: 237
  6. Group: 'UV'
  7. Tooltip: 'Procedural to uvmapped texture baker'
  8. """
  9.  
  10. __author__ = "Jean-Michel Soler (jms)"
  11. __url__ = ("blender", "elysiun",
  12. "Official Page, http://jmsoler.free.fr/didacticiel/blender/tutor/cpl_mesh3d2uv2d_en.htm",
  13. "Communicate problems and errors, http://www.zoo-logique.org/3D.Blender/newsportal/thread.php?group=3D.Blender")
  14. __version__ = "0.2.6 2005/5/29"
  15.  
  16. __bpydoc__ = """\
  17. Texture Baker "bakes" Blender procedural materials (including textures): it saves them as 2d uv-mapped images.
  18.  
  19. This script saves an uv texture layout of the chosen mesh, that can be used as
  20. an uv map for it.  It is a way to export procedurals from Blender as normal
  21. image textures that can be edited with a 2d image manipulation program or used
  22. with the mesh in games and other 3d applications.
  23.  
  24. Usage:
  25.  
  26. a) Enter face mode and define uv coordinates for your mesh (do not forget to choose a development shape);<br>
  27. b) Define its materials and textures;<br>
  28. c) Run this script and check the console.
  29.  
  30. Global variables:
  31.  
  32. a) FRAME (integer): the last frame of the animation, autodocumented.<br>
  33. b) LIMIT (integer): 0 or 1, uvcoords may exceed limits 0.0 to 1.0, this variable obliges the script to do a complete framing of the uvcoord.
  34.  
  35. Notes:<br>
  36.    This script was based on a suggestion by Martin (Theeth) Poirier.
  37. """
  38.  
  39. #---------------------------------------------
  40. # Last release : 0.2.6 ,  2005/05/29 , 22h00
  41. #---------------------------------------------
  42. #---------------------------------------------
  43. # (c) jm soler  07/2004 : 'Procedural Texture Baker'
  44. #     Based on a Martin 'Theeth' Poirier's really
  45. #     good idea : makes a rvk mesh with uv coords
  46. #     of the original mesh.
  47. #
  48. #     Released under Blender Artistic Licence
  49. #
  50. #   0.2.6
  51. #       -- Creation of LAMP object is removed and replaced
  52. #       by the use of the shadeless option  in material object
  53. #
  54. #       -- helpmsg corrected : the aim of the script
  55. #       is to bake any type of textures so we have not
  56. #       to mapinput its textures on UV .
  57. #
  58. #       --'pers' camera was replaced by an 'ortho' one. 
  59. #
  60. #   0.2.5
  61. #       -- if a image file  with the same name exits the
  62. #       system returns an error 
  63. #
  64. #   0.2.4
  65. #       -- a LIMIT variable is added to unlock the uvcoords
  66. #       autoframing
  67. #
  68. #
  69. #   0.2.3 :  
  70. #        Great thanks for Apollux  who sees a lot of these 
  71. #        problems
  72. #
  73. #        --Everytime you run the script a new set 
  74. #        of objects is created. File size and memory 
  75. #        consumption can go  pretty high if you are 
  76. #        not aware of that . 
  77. #        Now it ONLY creates 3 objects: a flattened 
  78. #         mesh, a camera and a lamp.
  79. #        --all the 3 objects was placed on layer 1, but if 
  80. #        that layer was not visible while you used the script 
  81. #        all you will get a is an empty render. 
  82. #        Now the layer is tst and activated befor the shoot 
  83. #        --The flattened mesh was really flattend only after 
  84. #        frame 100 (if you playbacked the animation, you can 
  85. #        actually see the mesh becoming flat on the first 100 
  86. #        frames). No more.
  87. #        -- When the script is run, it changes temporary to 
  88. #        the new cammera, set the render output to a square 
  89. #        (i.e. 1024 x 1024 or else), does the render, and then 
  90. #        resets the render output and the active camera to the 
  91. #        original one. But if no original camera was found
  92. #        this produce an error.    
  93. #
  94. #  0.2.2 : 
  95. #        if the uv mesh objet exists it used,
  96. #        no creation of a new one. As the lamp and  
  97. #        the camera
  98. #  0.2.1 : 
  99. #        This script automaticaly frame and shoot the
  100. #        new uv mesh . The image file is saved ine the
  101. #        /render folder.
  102. #
  103. #---------------------------------------------
  104. #  On user-friendly side :
  105. #---------------------------------------------
  106. #- Tadje Vobovnik adds the Select Image Size Menu
  107. #
  108. #---------------------------------------------
  109. # Official Page :
  110. #  http://jmsoler.free.fr/didacticiel/blender/tutor/cpl_mesh3d2uv2d_en.htm
  111. # For problems and errors:
  112. #   http://www.zoo-logique.org/3D.Blender/newsportal/thread.php?group=3D.Blender
  113. #---------------------------------------------
  114.  
  115. import Blender
  116. from Blender import NMesh, Draw, Object, Scene, Camera
  117.  
  118. #----------------------------------- 
  119. # Last release : 0.2.5 ,  2005/05/22 , 20h00
  120. #----------------------------------- 
  121. # la fonction Blender.sys.dirname pose un
  122. # probleme lorsque la memoire est trop encombree
  123. # ---
  124. # It seems that the Blender.sys.dirname function
  125. # poses a problem when the memory is too much encumbered 
  126. #----------------------------------- 
  127. try:
  128.   import nt
  129.   os = nt
  130.   os.sep='\\'
  131. except:
  132.   import posix
  133.   os = posix
  134.   os.sep='/'
  135. DIRNAME=Blender.Get('filename')
  136. #----------------------------------- 
  137. #  decoupage de la chaine en fragment
  138. #  de faτon a isoler le nom du fichier
  139. #  du repertoire
  140. # ---
  141. # split string in fragments to isolate
  142. # the file name from the path name
  143. #----------------------------------- 
  144.  
  145. if DIRNAME.find(os.sep):
  146.     k0=DIRNAME.split(os.sep)
  147. else:
  148.     k0=DIRNAME.split('/')   
  149. DIRNAME=DIRNAME.replace(k0[-1],'')
  150. #----------------------------------- 
  151. # Last release : 0.2.5 ,  2005/05/22 , end
  152. #----------------------------------- 
  153.  
  154. #----------------------------------- 
  155. # Last release : 0.2.4 ,  2005/05/22 , 15h00
  156. #----------------------------------- 
  157. FRAME = Blender.Get('endframe')
  158. #----------------------------------- 
  159. # Last release : 0.2.4 ,  2005/05/22 , end
  160. #----------------------------------- 
  161.  
  162. #----------------------------------- 
  163. # Last release : 0.2.4 ,  2005/05/18 , 11h00
  164. #
  165. # Si  LIMIT == 0 le script n'essaye pas de realiser
  166. # un nouveau cadrage pour que l'image presente toute les
  167. # coordonnΘes uv.
  168. # ---
  169. # if  LIMIT == 0 the script do not try to make
  170. # a new framing with all the uvcoord  in only one
  171. # shoot...
  172. #-----------------------------------
  173. LIMIT=0
  174. #----------------------------------- 
  175. # Last release : 0.2.4 ,  2005/05/18 , END
  176. #----------------------------------- 
  177.  
  178. XYLIMIT = [0.0, 0.0,1.0,1.0]    
  179. OBJPOS = 100.0
  180. DEBUG=1
  181.  
  182. helpmsg = """
  183. Texture Baker:
  184.  
  185. This script saves an uv texture layout of the chosen mesh, that can be used as
  186. an uv map for it.  It is a way to export procedural textures from Blender as
  187. normal image textures that can be edited with a 2d image manipulation program
  188. or used with the mesh in games and other 3d applications.
  189.  
  190. Basic instructions:
  191. - Enter face mode and define uv coordinates for your mesh (do not forget to
  192.   choose a development shape);
  193. - Define its materials and textures ;
  194. - Run this script and check the console.
  195.  
  196. """
  197.  
  198. def GET_newobject (TYPE,NAME):
  199.    """
  200. # ---------------------------
  201. # Function  GET_newobject 
  202. # 
  203. #  IN : TYPE   string , object type ('Mesh','Empty',...)
  204. #       NAME   string , name object  
  205. #  OUT: OBJECT  Blender objetc described in teh string TYPE
  206. #       SCENE   Blender current scene object
  207. # ---------------------------   
  208.     Return and object and the current scene
  209.    """
  210.    SCENE = Blender.Scene.getCurrent()
  211.    OBJECT = Blender.Object.New(TYPE,NAME)
  212.    SCENE.link(OBJECT)
  213.    return OBJECT, SCENE
  214.  
  215. def RenameImage(RDIR, MYDIR, FILENAME, name):
  216.     """
  217. # ---------------------------
  218. # Function  RenameImage
  219. #
  220. #  IN : RDIR      string , current render directory
  221. #       MYDIR     string , new render dir for this shoot
  222. #       FILENAME  string , last rendered image filename 
  223. #       name      string , new name for this image 
  224. #  OUT:  nothing
  225. # ---------------------------
  226.      Rename the file pointed by the string name
  227.      recall the  function if the file yet exists
  228.     """
  229.     newfname = RDIR + MYDIR + name
  230.     if newfname.find('.png', -4) < 0 : newfname += '.png'
  231.     if not Blender.sys.exists(newfname):
  232.        os.rename(FILENAME, newfname)
  233.     else:
  234.         name = Draw.PupStrInput ('ReName Image, please :', name, 32)
  235.         RenameImage(RDIR, MYDIR, FILENAME, name)
  236.  
  237. def SAVE_image (rc, name, FRAME):
  238.    """  
  239. # ---------------------------
  240. # Function  SAVE_image
  241. #
  242. #  IN : rc    current render context object
  243. #       name  string , image name
  244. #       FRAME  integer, last numbre of the curent animation  
  245. #  OUT: nothing  
  246. # ---------------------------   
  247.    """
  248.    rc.enableExtensions(1)   
  249.    MYDIR = ''
  250.    RENDERDIR = rc.getRenderPath().replace('\\','/')
  251.    if RENDERDIR.find('//')==0 : 
  252.       print 'filename', Blender.Get('filename'),'/n', Blender.sys.dirname(Blender.Get('filename')) 
  253.       RDIR=RENDERDIR.replace('//',DIRNAME)
  254.    else:
  255.         RDIR=RENDERDIR[:]
  256.    if DEBUG : print  'RDIR : ', RDIR
  257.     
  258.    HOMEDIR=Blender.Get('homedir')
  259.    if DEBUG : print  'HOMEDIR', HOMEDIR
  260.    rc.setRenderPath(RENDERDIR + MYDIR)
  261.    if DEBUG : print "Render folder:", RENDERDIR + MYDIR
  262.    IMAGETYPE = Blender.Scene.Render.PNG
  263.    if DEBUG : print  'IMAGETYPE : ',IMAGETYPE
  264.    rc.setImageType(IMAGETYPE)
  265.    NEWFRAME = FRAME
  266.    OLDEFRAME = rc.endFrame()
  267.    OLDSFRAME = rc.startFrame()
  268.    rc.startFrame(NEWFRAME)
  269.    rc.endFrame(NEWFRAME)
  270.    rc.renderAnim()
  271.    Blender.Scene.Render.CloseRenderWindow()
  272.  
  273.    FILENAME = "%04d" % NEWFRAME
  274.    FILENAME = FILENAME.replace (' ', '0')
  275.    FILENAME = RDIR + MYDIR + FILENAME + '.png'
  276.  
  277.    RenameImage(RDIR, MYDIR, FILENAME, name)
  278.  
  279.    rc.endFrame(OLDEFRAME)
  280.    rc.startFrame(OLDSFRAME)
  281.    rc.setRenderPath(RENDERDIR)
  282.  
  283. def SHOOT (XYlimit, frame, obj, name, FRAME):
  284.    """
  285. # ---------------------------
  286. # Function  SHOOT
  287. #
  288. #  IN : XYlimit  list of 4 floats, smallest and biggest 
  289. #                uvcoords
  290. #       frame    cureente frame
  291. #       obj      for object location
  292. #       name     image name
  293. #       FRAME    the last animation's frame 
  294. #  OUT:  nothing 
  295. # ---------------------------   
  296.       render and save the baked textures picture 
  297.    """
  298.    try:
  299.       CAM = Blender.Object.Get('UVCAMERA')
  300.       Cam = CAM.getData()
  301.       SC = Blender.Scene.getCurrent()
  302.    except:
  303.       Cam = Blender.Camera.New()
  304.       Cam.name = 'UVCamera'
  305.       CAM, SC = GET_newobject('Camera','UVCAMERA')
  306.       CAM.link(Cam)
  307.       CAM.setName('UVCAMERA')
  308.       Cam.lens = 30
  309.       Cam.name = 'UVCamera'
  310.  
  311.    Cam.setType('ortho')
  312.    Cam.setScale(1.0)
  313.  
  314.    CAM.setLocation(obj.getLocation())
  315.    CAM.LocX += XYlimit[2] * 0.500
  316.    CAM.LocY += XYlimit[3] * 0.500
  317.    CAM.LocZ += max (XYlimit[2], XYlimit[3])
  318.    CAM.setEuler (0.0, 0.0, 0.0)
  319.  
  320.    context = SC.getRenderingContext()
  321.    Camold = SC.getCurrentCamera()
  322.    SC.setCurrentCamera(CAM)
  323.  
  324.    OLDy = context.imageSizeY()
  325.    OLDx = context.imageSizeX()
  326.  
  327.    tres = Draw.PupMenu('TEXTURE OUT RESOLUTION : %t | 256 %x1 | 512 %x2 | 768 %x3 | 1024 %x4 | 2048 %x5 ')
  328.  
  329.    if (tres) == 1: res = 256
  330.    elif (tres) == 2: res = 512
  331.    elif (tres) == 3: res = 768
  332.    elif (tres) == 4: res = 1024
  333.    elif (tres) == 5: res = 2048
  334.    else: res = 512
  335.  
  336.    context.imageSizeY(res)
  337.    context.imageSizeX(res)
  338.    SAVE_image (context, name, FRAME)
  339.    context.imageSizeY(OLDy)
  340.    context.imageSizeX(OLDx)
  341.  
  342.    if Camold :SC.setCurrentCamera(Camold)
  343.  
  344.    Blender.Set ('curframe', frame)
  345.  
  346.  
  347. #----------------------------------- 
  348. # release : 0.2.6 ,  2005/05/29 , 00h00
  349. #----------------------------------- 
  350. def PROV_Shadeless(MATList):
  351.     """
  352. # ---------------------------
  353. # Function  PROV_Shadeless
  354. #
  355. #  IN : MATList  a list of the mesh's materials
  356. #  OUT: SHADEDict  a dictionnary  of the materials' shadeles value
  357. # ---------------------------   
  358.     """
  359.     SHADEDict={}
  360.     for mat in MATList:
  361.        SHADEDict[mat.name]=mat.mode
  362.        mat.mode |= Blender.Material.Modes.SHADELESS
  363.     return SHADEDict
  364. #----------------------------------- 
  365. # Last release : 0.2.6 ,  2005/05/29 , end
  366. #----------------------------------- 
  367.  
  368. #----------------------------------- 
  369. # release : 0.2.6 ,  2005/05/29 , 00h00
  370. #----------------------------------- 
  371. def REST_Shadeless(SHADEDict):
  372.     """
  373. # ---------------------------
  374. # Function  REST_Shadeless
  375. #
  376. #  IN : SHADEDict   a dictionnary  of the materials' shadeles value
  377. #  OUT : nothing
  378. # ---------------------------   
  379.     """
  380.     for m in SHADEDict.keys():
  381.        mat=Blender.Material.Get(m)
  382.        mat.mode=SHADEDict[m]
  383. #----------------------------------- 
  384. # release : 0.2.6 ,  2005/05/29 , end
  385. #----------------------------------- 
  386.  
  387. def Mesh2UVCoord (LIMIT):
  388.    """
  389. # ---------------------------
  390. # Function  Mesh2UVCoord
  391. #
  392. #  IN : LIMIT  integer, create or not a new framing for uvcoords 
  393. #  OUT:  nothing
  394. # ---------------------------   
  395.    """
  396.    try:
  397.       MESH3D = Object.GetSelected()[0]
  398.       if MESH3D.getType() == 'Mesh':
  399.          MESH = MESH3D.getData()
  400.  
  401.          try:
  402.             NewOBJECT=Blender.Object.Get('UVOBJECT')
  403.             CurSCENE=Blender.Scene.getCurrent()            
  404.             MESH2 = NewOBJECT.getData()
  405.             
  406.          except:
  407.             NewOBJECT, CurSCENE = GET_newobject('Mesh','UVOBJECT')
  408.             MESH2 = Blender.NMesh.GetRaw()
  409.  
  410.          MESH2.faces=[]
  411.          for f in MESH.faces:
  412.             f1 = Blender.NMesh.Face()
  413.  
  414.             for v in f.v:
  415.                v1 = Blender.NMesh.Vert (v.co[0], v.co[1], v.co[2])
  416.                MESH2.verts.append(v1)
  417.                f1.v.append(MESH2.verts[len(MESH2.verts) - 1])
  418.  
  419.             MESH2.faces.append(f1)
  420.             f1.uv = f.uv[:]
  421.             f1.col = f.col[:]
  422.             f1.smooth = f.smooth
  423.             f1.mode = f.mode
  424.             f1.flag = f.flag
  425.             f1.mat = f.mat
  426.  
  427.          MESH2.materials = MESH.materials[:]
  428.  
  429.          NewOBJECT.setLocation (OBJPOS, OBJPOS, 0.0)
  430.          NewOBJECT.setEuler (0.0, 0.0, 0.0)
  431.  
  432.          MESH2.removeAllKeys()
  433.  
  434.          MESH2.update()
  435.          MESH2.insertKey (1, 'absolute')
  436.          MESH2.update()
  437.  
  438.          for f in MESH2.faces:
  439.             for v in f.v:
  440.                for n in [0,1]:
  441.                   v.co[n] = f.uv[f.v.index(v)][n]
  442.                   exec "if v.co[%s] > XYLIMIT[%s]: XYLIMIT[%s] = v.co[%s]" % (n, n+2, n+2, n)
  443.                   exec "if v.co[%s] < XYLIMIT[%s]: XYLIMIT[%s] = v.co[%s]" % (n, n, n, n)
  444.                v.co[2] = 0.0
  445.  
  446.          if DEBUG: print XYLIMIT
  447.  
  448.          MESH2.update()
  449.          MESH2.insertKey (FRAME, 'absolute')
  450.          MESH2.update()
  451.  
  452.          imagename = 'uvtext'
  453.  
  454.          name = "CHANGE IMAGE NAME ? %t | Replace it | No replacing | Script help"
  455.          result = Draw.PupMenu(name)
  456.  
  457.          if result == 1:
  458.             imagename = Draw.PupStrInput ('Image Name:', imagename, 32)
  459.  
  460.          if result != 3:
  461.             #----------------------------------- 
  462.             # release : 0.2.6 ,  2005/05/29 , 00h00
  463.             #----------------------------------- 
  464.             SHADEDict=PROV_Shadeless(MESH2.materials)
  465.             #----------------------------------- 
  466.             # release : 0.2.6 ,  2005/05/29 , end
  467.             #----------------------------------- 
  468.  
  469.             if LIMIT :
  470.                  SHOOT(XYLIMIT, FRAME, NewOBJECT, imagename, FRAME)
  471.             else :
  472.                  SHOOT([0.0,0.0,1.0,1.0], FRAME, NewOBJECT, imagename, FRAME)
  473.             #----------------------------------- 
  474.             # release : 0.2.6,  2005/05/29 , 00h00
  475.             #----------------------------------- 
  476.             REST_Shadeless(SHADEDict)
  477.             #----------------------------------- 
  478.             # release : 0.2.6 ,  2005/05/29 , end
  479.             #----------------------------------- 
  480.  
  481.             Blender.Redraw()
  482.  
  483.          else:
  484.             Blender.ShowHelp('tex2uvbaker.py')
  485.             #Draw.PupMenu("Ready%t|Please check console for instructions")
  486.             if DEBUG: print helpmsg
  487.  
  488.       else:
  489.          name = "ERROR: active object is not a mesh or has no UV coordinates"
  490.          result = Draw.PupMenu(name)
  491.          print 'problem : no object selected or not mesh'
  492.  
  493.    except:
  494.       name = "ERROR: active object is not a mesh or has no UV coordinates"
  495.       result = Draw.PupMenu(name)
  496.       print 'problem : no object selected or not mesh'
  497.  
  498. Mesh2UVCoord(LIMIT)
  499.