release/io/engine_render_pov.py

   1 import bpy
   2
   3 from math import atan, pi, degrees
   4 import subprocess
   5 import os
   6 import sys
   7 import time
   8
   9 import platform as pltfrm
  10
  11 if pltfrm.architecture()[0] == '64bit':
  12         bitness = 64
  13 else:
  14         bitness = 32
  15
  16 def write_pov(filename, scene=None, info_callback = None):
  17         file = open(filename, 'w')
  18
  19         # Only for testing
  20         if not scene:
  21                 scene = bpy.data.scenes[0]
  22
  23         render = scene.render_data
  24         materialTable = {}
  25
  26         def saneName(name):
  27                 name = name.lower()
  28                 for ch in ' /\\+=-[]{}().,<>\'":;~!@#$%^&*|?':
  29                         name = name.replace(ch, '_')
  30                 return name
  31
  32         def writeMatrix(matrix):
  33                 file.write('\tmatrix <%.6f, %.6f, %.6f,  %.6f, %.6f, %.6f,  %.6f, %.6f, %.6f,  %.6f, %.6f, %.6f>\n' %\
  34                 (matrix[0][0], matrix[0][1], matrix[0][2],  matrix[1][0], matrix[1][1], matrix[1][2],  matrix[2][0], matrix[2][1], matrix[2][2],  matrix[3][0], matrix[3][1], matrix[3][2]) )
  35
  36         def exportCamera():
  37                 camera = scene.camera
  38                 matrix = camera.matrix
  39
  40                 # compute resolution
  41                 Qsize=float(render.resolution_x)/float(render.resolution_y)
  42
  43                 file.write('camera {\n')
  44                 file.write('\tlocation  <0, 0, 0>\n')
  45                 file.write('\tlook_at  <0, 0, -1>\n')
  46                 file.write('\tright <%s, 0, 0>\n' % -Qsize)
  47                 file.write('\tup <0, 1, 0>\n')
  48                 file.write('\tangle  %f \n' % (360.0*atan(16.0/camera.data.lens)/pi))
  49
  50                 file.write('\trotate  <%.6f, %.6f, %.6f>\n' % tuple([degrees(e) for e in matrix.rotationPart().toEuler()]))
  51                 file.write('\ttranslate <%.6f, %.6f, %.6f>\n' % (matrix[3][0], matrix[3][1], matrix[3][2]))
  52                 file.write('}\n')
  53
  54
  55
  56         def exportLamps(lamps):
  57                 # Get all lamps
  58                 for ob in lamps:
  59                         lamp = ob.data
  60
  61                         matrix = ob.matrix
  62
  63                         color = tuple([c * lamp.energy for c in lamp.color]) # Colour is modified by energy
  64
  65                         file.write('light_source {\n')
  66                         file.write('\t< 0,0,0 >\n')
  67                         file.write('\tcolor red %.6f green %.6f blue %.6f\n' % color)
  68
  69                         if lamp.type == 'POINT': # Point Lamp
  70                                 pass
  71                         elif lamp.type == 'SPOT': # Spot
  72                                 file.write('\tspotlight\n')
  73
  74                                 # Falloff is the main radius from the centre line
  75                                 file.write('\tfalloff %.2f\n' % (lamp.spot_size/2.0) ) # 1 TO 179 FOR BOTH
  76                                 file.write('\tradius %.6f\n' % ((lamp.spot_size/2.0) * (1-lamp.spot_blend)) )
  77
  78                                 # Blender does not have a tightness equivilent, 0 is most like blender default.
  79                                 file.write('\ttightness 0\n') # 0:10f
  80
  81                                 file.write('\tpoint_at  <0, 0, -1>\n')
  82                         elif lamp.type == 'AREA':
  83
  84                                 size_x = lamp.size
  85                                 samples_x = lamp.shadow_ray_samples_x
  86                                 if lamp.shape == 'SQUARE':
  87                                         size_y = size_x
  88                                         samples_y = samples_x
  89                                 else:
  90                                         size_y = lamp.size_y
  91                                         samples_y = lamp.shadow_ray_samples_y
  92
  93
  94
  95                                 file.write('\tarea_light <%d,0,0>,<0,0,%d> %d, %d\n' % (size_x, size_y, samples_x, samples_y))
  96                                 if lamp.shadow_ray_sampling_method == 'CONSTANT_JITTERED':
  97                                         if lamp.jitter:
  98                                                 file.write('\tjitter\n')
  99                                 else:
 100                                         file.write('\tadaptive 1\n')
 101                                         file.write('\tjitter\n')
 102
 103                         if lamp.shadow_method == 'NOSHADOW':
 104                                 file.write('\tshadowless\n')
 105
 106                         file.write('\tfade_distance %.6f\n' % lamp.distance)
 107                         file.write('\tfade_power %d\n' % 1) # Could use blenders lamp quad?
 108                         writeMatrix(matrix)
 109
 110                         file.write('}\n')
 111
 112         def exportMeshs(sel):
 113                 def bMat2PovString(material):
 114                         povstring = 'finish {'
 115                         if world != None:
 116                                 povstring += 'ambient <%.6f, %.6f, %.6f> ' % tuple([c*material.ambient for c in world.ambient_color])
 117
 118                         povstring += 'diffuse %.6f ' % material.diffuse_reflection
 119                         povstring += 'specular %.6f ' % material.specular_reflection
 120
 121
 122                         if material.raytrace_mirror.enabled:
 123                                 #povstring += 'interior { ior %.6f } ' % material.IOR
 124                                 raytrace_mirror= material.raytrace_mirror
 125                                 if raytrace_mirror.reflect:
 126                                         povstring += 'reflection {'
 127                                         povstring += '<%.6f, %.6f, %.6f>' % tuple(material.mirror_color) # Should ask for ray mirror flag
 128                                         povstring += 'fresnel 1 falloff %.6f exponent %.6f metallic %.6f} ' % (raytrace_mirror.fresnel, raytrace_mirror.fresnel_fac, raytrace_mirror.reflect)
 129
 130
 131
 132                         if material.raytrace_transparency.enabled:
 133                                 #povstring += 'interior { ior %.6f } ' % material.IOR
 134                                 pass
 135
 136                         #file.write('\t\troughness %.6f\n' % (material.hard*0.5))
 137                         #file.write('\t\t\tcrand 0.0\n') # Sand granyness
 138                         #file.write('\t\t\tmetallic %.6f\n' % material.spec)
 139                         #file.write('\t\t\tphong %.6f\n' % material.spec)
 140                         #file.write('\t\t\tphong_size %.6f\n' % material.spec)
 141                         povstring += 'brilliance %.6f ' % (material.specular_hardness/256.0) # Like hardness
 142                         povstring += '}'
 143                         #file.write('\t}\n')
 144                         return povstring
 145
 146
 147                 world = scene.world
 148
 149                 # Convert all materials to strings we can access directly per vertex.
 150                 for material in bpy.data.materials:
 151                         materialTable[material.name] = bMat2PovString(material)
 152
 153
 154                 ob_num = 0
 155
 156                 for ob in sel:
 157                         ob_num+= 1
 158
 159                         if ob.type in ('LAMP', 'CAMERA', 'EMPTY'):
 160                                 continue
 161
 162                         me = ob.data
 163                         me_materials= me.materials
 164
 165                         me = ob.create_render_mesh(scene)
 166
 167                         if not me:
 168                                 continue
 169
 170                         if info_callback:
 171                                 info_callback('Object %2.d of %2.d (%s)' % (ob_num, len(sel), ob.name))
 172
 173                         #if ob.type!='MESH':
 174                         #       continue
 175                         # me = ob.data
 176
 177                         matrix = ob.matrix
 178                         try:    uv_layer = me.active_uv_texture.data
 179                         except:uv_layer = None
 180
 181                         try:    vcol_layer = me.active_vertex_color.data
 182                         except:vcol_layer = None
 183
 184
 185                         def regular_face(f):
 186                                 fv = f.verts
 187                                 if fv[3]== 0:
 188                                         return fv[0], fv[1], fv[2]
 189                                 return fv[0], fv[1], fv[2], fv[3]
 190
 191                         faces_verts = [regular_face(f) for f in me.faces]
 192                         faces_normals = [tuple(f.normal) for f in me.faces]
 193                         verts_normals = [tuple(v.normal) for v in me.verts]
 194
 195                         # quads incur an extra face
 196                         quadCount = len([f for f in faces_verts if len(f)==4])
 197
 198                         file.write('mesh2 {\n')
 199                         file.write('\tvertex_vectors {\n')
 200                         file.write('\t\t%s' % (len(me.verts))) # vert count
 201                         for v in me.verts:
 202                                 file.write(',\n\t\t<%.6f, %.6f, %.6f>' % tuple(v.co)) # vert count
 203                         file.write('\n  }\n')
 204
 205
 206                         # Build unique Normal list
 207                         uniqueNormals = {}
 208                         for fi, f in enumerate(me.faces):
 209                                 fv = faces_verts[fi]
 210                                 # [-1] is a dummy index, use a list so we can modify in place
 211                                 if f.smooth: # Use vertex normals
 212                                         for v in fv:
 213                                                 key = verts_normals[v]
 214                                                 uniqueNormals[key] = [-1]
 215                                 else: # Use face normal
 216                                         key = faces_normals[fi]
 217                                         uniqueNormals[key] = [-1]
 218
 219                         file.write('\tnormal_vectors {\n')
 220                         file.write('\t\t%d' % len(uniqueNormals)) # vert count
 221                         idx = 0
 222                         for no, index in uniqueNormals.items():
 223                                 file.write(',\n\t\t<%.6f, %.6f, %.6f>' % no) # vert count
 224                                 index[0] = idx
 225                                 idx +=1
 226                         file.write('\n  }\n')
 227
 228
 229                         # Vertex colours
 230                         vertCols = {} # Use for material colours also.
 231
 232                         if uv_layer:
 233                                 # Generate unique UV's
 234                                 uniqueUVs = {}
 235
 236                                 for fi, uv in enumerate(uv_layer):
 237
 238                                         if len(faces_verts[fi])==4:
 239                                                 uvs = uv.uv1, uv.uv2, uv.uv3, uv.uv4
 240                                         else:
 241                                                 uvs = uv.uv1, uv.uv2, uv.uv3
 242
 243                                         for uv in uvs:
 244                                                 uniqueUVs[tuple(uv)] = [-1]
 245
 246                                 file.write('\tuv_vectors {\n')
 247                                 #print unique_uvs
 248                                 file.write('\t\t%s' % (len(uniqueUVs))) # vert count
 249                                 idx = 0
 250                                 for uv, index in uniqueUVs.items():
 251                                         file.write(',\n\t\t<%.6f, %.6f>' % uv)
 252                                         index[0] = idx
 253                                         idx +=1
 254                                 '''
 255                                 else:
 256                                         # Just add 1 dummy vector, no real UV's
 257                                         file.write('\t\t1') # vert count
 258                                         file.write(',\n\t\t<0.0, 0.0>')
 259                                 '''
 260                                 file.write('\n  }\n')
 261
 262
 263                         if me.vertex_colors:
 264
 265                                 for fi, f in enumerate(me.faces):
 266                                         material_index = f.material_index
 267                                         material = me_materials[material_index]
 268
 269                                         if material and material.vertex_color_paint:
 270
 271                                                 col = vcol_layer[fi]
 272
 273                                                 if len(faces_verts[fi])==4:
 274                                                         cols = col.color1, col.color2, col.color3, col.color4
 275                                                 else:
 276                                                         cols = col.color1, col.color2, col.color3
 277
 278                                                 for col in cols:
 279                                                         key = col[0], col[1], col[2], material_index # Material index!
 280                                                         vertCols[key] = [-1]
 281
 282                                         else:
 283                                                 if material:
 284                                                         diffuse_color = tuple(material.diffuse_color)
 285                                                         key = diffuse_color[0], diffuse_color[1], diffuse_color[2], material_index
 286                                                         vertCols[key] = [-1]
 287
 288
 289                         else:
 290                                 # No vertex colours, so write material colours as vertex colours
 291                                 for i, material in enumerate(me_materials):
 292
 293                                         if material:
 294                                                 diffuse_color = tuple(material.diffuse_color)
 295                                                 key = diffuse_color[0], diffuse_color[1], diffuse_color[2], i # i == f.mat
 296                                                 vertCols[key] = [-1]
 297
 298
 299                         # Vert Colours
 300                         file.write('\ttexture_list {\n')
 301                         file.write('\t\t%s' % (len(vertCols))) # vert count
 302                         idx=0
 303                         for col, index in vertCols.items():
 304
 305                                 if me_materials:
 306                                         material = me_materials[col[3]]
 307                                         materialString = materialTable[material.name]
 308                                 else:
 309                                         materialString = '' # Dont write anything
 310
 311                                 float_col = col[0], col[1], col[2], 1-material.alpha, materialString
 312                                 #print material.apl
 313                                 file.write(',\n\t\ttexture { pigment {rgbf<%.6f, %.6f, %.6f, %.6f>}%s}' % float_col)
 314                                 index[0] = idx
 315                                 idx+=1
 316
 317                         file.write( '\n  }\n' )
 318
 319                         # Face indicies
 320                         file.write('\tface_indices {\n')
 321                         file.write('\t\t%d' % (len(me.faces) + quadCount)) # faces count
 322                         for fi, f in enumerate(me.faces):
 323                                 fv = faces_verts[fi]
 324                                 material_index= f.material_index
 325                                 if len(fv) == 4:        indicies = (0,1,2), (0,2,3)
 326                                 else:                           indicies = ((0,1,2),)
 327
 328                                 if vcol_layer:
 329                                         col = vcol_layer[fi]
 330
 331                                         if len(fv) == 4:
 332                                                 cols = col.color1, col.color2, col.color3, col.color4
 333                                         else:
 334                                                 cols = col.color1, col.color2, col.color3
 335
 336
 337                                 if not me_materials or me_materials[material_index] == None: # No materials
 338                                         for i1, i2, i3 in indicies:
 339                                                 file.write(',\n\t\t<%d,%d,%d>' % (fv[i1], fv[i2], fv[i3])) # vert count
 340                                 else:
 341                                         material = me_materials[material_index]
 342                                         for i1, i2, i3 in indicies:
 343                                                 if me.vertex_colors and material.vertex_color_paint:
 344                                                         # Colour per vertex - vertex colour
 345
 346                                                         col1 = cols[i1]
 347                                                         col2 = cols[i2]
 348                                                         col3 = cols[i3]
 349
 350                                                         ci1 = vertCols[col1[0], col1[1], col1[2], material_index][0]
 351                                                         ci2 = vertCols[col2[0], col2[1], col2[2], material_index][0]
 352                                                         ci3 = vertCols[col3[0], col3[1], col3[2], material_index][0]
 353                                                 else:
 354                                                         # Colour per material - flat material colour
 355                                                         diffuse_color= material.diffuse_color
 356                                                         ci1 = ci2 = ci3 = vertCols[diffuse_color[0], diffuse_color[1], diffuse_color[2], f.material_index][0]
 357
 358                                                 file.write(',\n\t\t<%d,%d,%d>, %d,%d,%d' % (fv[i1], fv[i2], fv[i3], ci1, ci2, ci3)) # vert count
 359
 360
 361
 362                         file.write('\n  }\n')
 363
 364                         # normal_indices indicies
 365                         file.write('\tnormal_indices {\n')
 366                         file.write('\t\t%d' % (len(me.faces) + quadCount)) # faces count
 367                         for fi, fv in enumerate(faces_verts):
 368
 369                                 if len(fv) == 4:        indicies = (0,1,2), (0,2,3)
 370                                 else:                           indicies = ((0,1,2),)
 371
 372                                 for i1, i2, i3 in indicies:
 373                                         if f.smooth:
 374                                                 file.write(',\n\t\t<%d,%d,%d>' %\
 375                                                 (uniqueNormals[verts_normals[fv[i1]]][0],\
 376                                                  uniqueNormals[verts_normals[fv[i2]]][0],\
 377                                                  uniqueNormals[verts_normals[fv[i3]]][0])) # vert count
 378                                         else:
 379                                                 idx = uniqueNormals[faces_normals[fi]][0]
 380                                                 file.write(',\n\t\t<%d,%d,%d>' % (idx, idx, idx)) # vert count
 381
 382                         file.write('\n  }\n')
 383
 384                         if uv_layer:
 385                                 file.write('\tuv_indices {\n')
 386                                 file.write('\t\t%d' % (len(me.faces) + quadCount)) # faces count
 387                                 for fi, fv in enumerate(faces_verts):
 388
 389                                         if len(fv) == 4:        indicies = (0,1,2), (0,2,3)
 390                                         else:                           indicies = ((0,1,2),)
 391
 392                                         uv = uv_layer[fi]
 393                                         if len(faces_verts[fi])==4:
 394                                                 uvs = tuple(uv.uv1), tuple(uv.uv2), tuple(uv.uv3), tuple(uv.uv4)
 395                                         else:
 396                                                 uvs = tuple(uv.uv1), tuple(uv.uv2), tuple(uv.uv3)
 397
 398                                         for i1, i2, i3 in indicies:
 399                                                 file.write(',\n\t\t<%d,%d,%d>' %\
 400                                                 (uniqueUVs[uvs[i1]][0],\
 401                                                  uniqueUVs[uvs[i2]][0],\
 402                                                  uniqueUVs[uvs[i2]][0])) # vert count
 403                                 file.write('\n  }\n')
 404
 405                         if me.materials:
 406                                 material = me.materials[0] # dodgy
 407                                 if material and material.raytrace_transparency.enabled:
 408                                         file.write('\tinterior { ior %.6f }\n' % material.raytrace_transparency.ior)
 409
 410                         writeMatrix(matrix)
 411                         file.write('}\n')
 412
 413                         bpy.data.remove_mesh(me)
 414
 415
 416         exportCamera()
 417         #exportMaterials()
 418         sel = scene.objects
 419         lamps = [l for l in sel if l.type == 'LAMP']
 420         exportLamps(lamps)
 421         exportMeshs(sel)
 422
 423         file.close()
 424
 425
 426 def write_pov_ini(filename_ini, filename_pov, filename_image):
 427         scene = bpy.data.scenes[0]
 428         render = scene.render_data
 429
 430         x= int(render.resolution_x*render.resolution_percentage*0.01)
 431         y= int(render.resolution_y*render.resolution_percentage*0.01)
 432
 433         file = open(filename_ini, 'w')
 434
 435         file.write('Input_File_Name="%s"\n' % filename_pov)
 436         file.write('Output_File_Name="%s"\n' % filename_image)
 437
 438         file.write('Width=%d\n' % x)
 439         file.write('Height=%d\n' % y)
 440
 441         # Needed for border render.
 442         '''
 443         file.write('Start_Column=%d\n' % part.x)
 444         file.write('End_Column=%d\n' % (part.x+part.w))
 445
 446         file.write('Start_Row=%d\n' % (part.y))
 447         file.write('End_Row=%d\n' % (part.y+part.h))
 448         '''
 449
 450         file.write('Display=0\n')
 451         file.write('Pause_When_Done=0\n')
 452         file.write('Output_File_Type=C\n') # TGA, best progressive loading
 453         file.write('Output_Alpha=1\n')
 454
 455         if render.antialiasing:
 456                 aa_mapping = {'OVERSAMPLE_5':2, 'OVERSAMPLE_8':3, 'OVERSAMPLE_11':4, 'OVERSAMPLE_16':5} # method 1 assumed
 457                 file.write('Antialias=1\n')
 458                 file.write('Antialias_Depth=%d\n' % aa_mapping[render.antialiasing_samples])
 459         else:
 460                 file.write('Antialias=0\n')
 461
 462         file.close()
 463
 464
 465 class PovrayRenderEngine(bpy.types.RenderEngine):
 466         __label__ = "Povray"
 467         DELAY = 0.02
 468         def _export(self, scene):
 469                 import tempfile
 470
 471                 self.temp_file_in = tempfile.mktemp(suffix='.pov')
 472                 self.temp_file_out = tempfile.mktemp(suffix='.tga')
 473                 self.temp_file_ini = tempfile.mktemp(suffix='.ini')
 474
 475                 def info_callback(txt):
 476                         self.update_stats("", "POVRAY: " + txt)
 477
 478                 write_pov(self.temp_file_in, scene, info_callback)
 479
 480         def _render(self):
 481
 482                 try:            os.remove(self.temp_file_out) # so as not to load the old file
 483                 except: pass
 484
 485                 write_pov_ini(self.temp_file_ini, self.temp_file_in, self.temp_file_out)
 486
 487                 print ("***-STARTING-***")
 488                 # This works too but means we have to wait until its done
 489                 # os.system('povray %s' % self.temp_file_ini)
 490
 491                 pov_binary = "povray"
 492
 493                 if sys.platform=='win32':
 494                         if bitness == 64:
 495                                 pov_binary = "pvengine64"
 496                         else:
 497                                 pov_binary = "pvengine"
 498
 499                 self.process = subprocess.Popen([pov_binary, self.temp_file_ini]) # stdout=subprocess.PIPE, stderr=subprocess.PIPE
 500
 501                 print ("***-DONE-***")
 502
 503         def _cleanup(self):
 504                 for f in (self.temp_file_in, self.temp_file_ini, self.temp_file_out):
 505                         try:            os.remove(f)
 506                         except: pass
 507
 508                 self.update_stats("", "")
 509
 510         def render(self, scene):
 511
 512                 self.update_stats("", "POVRAY: Exporting data from Blender")
 513                 self._export(scene)
 514                 self.update_stats("", "POVRAY: Parsing File")
 515                 self._render()
 516
 517                 r = scene.render_data
 518
 519                 # compute resolution
 520                 x= int(r.resolution_x*r.resolution_percentage*0.01)
 521                 y= int(r.resolution_y*r.resolution_percentage*0.01)
 522
 523
 524
 525                 # Wait for the file to be created
 526                 while not os.path.exists(self.temp_file_out):
 527                         if self.test_break():
 528                                 try:            self.process.terminate()
 529                                 except: pass
 530                                 break
 531
 532                         if self.process.poll() != None:
 533                                 self.update_stats("", "POVRAY: Failed")
 534                                 break
 535
 536                         time.sleep(self.DELAY)
 537
 538                 if os.path.exists(self.temp_file_out):
 539
 540                         self.update_stats("", "POVRAY: Rendering")
 541
 542                         prev_size = -1
 543
 544                         def update_image():
 545                                 result = self.begin_result(0, 0, x, y)
 546                                 lay = result.layers[0]
 547                                 # possible the image wont load early on.
 548                                 try:            lay.rect_from_file(self.temp_file_out, 0, 0)
 549                                 except: pass
 550                                 self.end_result(result)
 551
 552                         # Update while povray renders
 553                         while True:
 554
 555                                 # test if povray exists
 556                                 if self.process.poll() != None:
 557                                         update_image();
 558                                         break
 559
 560                                 # user exit
 561                                 if self.test_break():
 562                                         try:            self.process.terminate()
 563                                         except: pass
 564
 565                                         break
 566
 567                                 # Would be nice to redirect the output
 568                                 # stdout_value, stderr_value = self.process.communicate() # locks
 569
 570
 571                                 # check if the file updated
 572                                 new_size = os.path.getsize(self.temp_file_out)
 573
 574                                 if new_size != prev_size:
 575                                         update_image()
 576                                         prev_size = new_size
 577
 578                                 time.sleep(self.DELAY)
 579
 580                 self._cleanup()
 581
 582
 583 bpy.types.register(PovrayRenderEngine)