Skip to content
Snippets Groups Projects
render.py 73.8 KiB
Newer Older
# ***** BEGIN GPL LICENSE BLOCK *****
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
# #**** END GPL LICENSE BLOCK #****
Luca Bonavita's avatar
Luca Bonavita committed

# <pep8 compliant>
"""Write the POV file using this file's functions and some from other modules then render it."""
Luca Bonavita's avatar
Luca Bonavita committed
import bpy
Maurice Raybaud's avatar
Maurice Raybaud committed
from sys import platform
Maurice Raybaud's avatar
Maurice Raybaud committed
from math import (
    pi,
)  # maybe move to scenography.py and topology_*****_data.py respectively with smoke and matrix

import tempfile  # generate temporary files with random names
from bpy.types import Operator
Maurice Raybaud's avatar
Maurice Raybaud committed
from bpy.utils import register_class, unregister_class
Maurice Raybaud's avatar
Maurice Raybaud committed
from . import (
    scripting,
)  # for writing, importing and rendering directly POV Scene Description Language items
from . import scenography  # for atmosphere, environment, effects, lighting, camera
from . import shading  # for BI POV shaders emulation
Maurice Raybaud's avatar
Maurice Raybaud committed
from . import object_mesh_topology  # for mesh based geometry
from . import object_curve_topology  # for curves based geometry

# from . import object_primitives  # for import and export of POV specific primitives
Maurice Raybaud's avatar
Maurice Raybaud committed
from .scenography import image_format, img_map, img_map_transforms, path_image

from .shading import write_object_material_interior
Maurice Raybaud's avatar
Maurice Raybaud committed
from .object_primitives import write_object_modifiers
Maurice Raybaud's avatar
Maurice Raybaud committed

Campbell Barton's avatar
Campbell Barton committed
def string_strip_hyphen(name):
Maurice Raybaud's avatar
Maurice Raybaud committed
    """Remove hyphen characters from a string to avoid POV errors."""
Campbell Barton's avatar
Campbell Barton committed
    return name.replace("-", "")
Maurice Raybaud's avatar
Maurice Raybaud committed

Maurice Raybaud's avatar
Maurice Raybaud committed
def safety(name, ref_level_bound):
Maurice Raybaud's avatar
Maurice Raybaud committed
    """append suffix characters to names of various material declinations.
Maurice Raybaud's avatar
Maurice Raybaud committed
    Material declinations are necessary to POV syntax and used in shading.py
Maurice Raybaud's avatar
Maurice Raybaud committed
    by the pov_has_no_specular_maps function to create the finish map trick and
Maurice Raybaud's avatar
Maurice Raybaud committed
    the suffixes avoid name collisions.
    Keyword arguments:
    name -- the initial material name as a string
Maurice Raybaud's avatar
Maurice Raybaud committed
    ref_level_bound -- the enum number of the ref_level_bound being written:
        ref_level_bound=1 is for texture with No specular nor Mirror reflection
        ref_level_bound=2 is for texture with translation of spec and mir levels
Maurice Raybaud's avatar
Maurice Raybaud committed
        for when no map influences them
Maurice Raybaud's avatar
Maurice Raybaud committed
        ref_level_bound=3 is for texture with Maximum Spec and Mirror
Maurice Raybaud's avatar
Maurice Raybaud committed
    """
Maurice Raybaud's avatar
Maurice Raybaud committed
    # All the try except clause below seems useless as each time
    # prefix rewritten even after and outside of it what was the point?
    # It may not even be any longer possible to feed no arg from Blender UI
    # try:
    # if name: # if int(name) > 0: # could be zero if no argument provided
    # # and always triggered exception so is this similar ?
    # prefix = "shader"
    # except BaseException as e:
    # print(e.__doc__)
    # print('An exception occurred: {}'.format(e))
Maurice Raybaud's avatar
Maurice Raybaud committed
    # prefix = "" # rewritten below...
    prefix = "shader_"
Campbell Barton's avatar
Campbell Barton committed
    name = string_strip_hyphen(name)
Maurice Raybaud's avatar
Maurice Raybaud committed
    if ref_level_bound == 2:
        return prefix + name
Maurice Raybaud's avatar
Maurice Raybaud committed
    # implicit else-if (no return yet)
    if ref_level_bound == 1:
        return prefix + name + "0"  # used for 0 of specular map
Maurice Raybaud's avatar
Maurice Raybaud committed
    # implicit else-if (no return yet)
    if ref_level_bound == 3:
        return prefix + name + "1"  # used for 1 of specular map
# -------- end safety string name material

Luca Bonavita's avatar
Luca Bonavita committed

Maurice Raybaud's avatar
Maurice Raybaud committed
def is_renderable(ob):
    """test for objects flagged as hidden or boolean operands not to render"""
    return not ob.hide_render and ob not in csg_list
def renderable_objects(scene):
Maurice Raybaud's avatar
Maurice Raybaud committed
    """test for non hidden, non boolean operands objects to render"""
    return [ob for ob in bpy.data.objects if is_renderable(ob)]
Maurice Raybaud's avatar
Maurice Raybaud committed
def no_renderable_objects():
    """Boolean operands only. Not to render"""
    return list(csg_list)
Maurice Raybaud's avatar
Maurice Raybaud committed
tab_level = 0
user_dir = bpy.utils.resource_path('USER')
preview_dir = os.path.join(user_dir, "preview")

# Make sure Preview directory exists and is empty
Maurice Raybaud's avatar
Maurice Raybaud committed
smoke_path = os.path.join(preview_dir, "smoke.df3")

Maurice Raybaud's avatar
Maurice Raybaud committed
# below properties not added to __init__ yet to avoid conflicts with material sss scale
# unless it would override then should be interfaced also in scene units property tab
Maurice Raybaud's avatar
Maurice Raybaud committed
# if scene.pov.sslt_enable:
    # file.write("    mm_per_unit %s\n"%scene.pov.mm_per_unit)
    # file.write("    subsurface {\n")
    # file.write("        samples %s, %s\n"%(scene.pov.sslt_samples_max,scene.pov.sslt_samples_min))
    # if scene.pov.sslt_radiosity:
        # file.write("        radiosity on\n")
    # file.write("}\n")
Maurice Raybaud's avatar
Maurice Raybaud committed
'''
Maurice Raybaud's avatar
Maurice Raybaud committed
# def write_object_modifiers(scene, ob, File):
# """Translate some object level POV statements from Blender UI
# to POV syntax and write to exported file """

# # Maybe return that string to be added instead of directly written.

# '''XXX WIP
# onceCSG = 0
# for mod in ob.modifiers:
# if onceCSG == 0:
# if mod :
# if mod.type == 'BOOLEAN':
# if ob.pov.boolean_mod == "POV":
# File.write("\tinside_vector <%.6g, %.6g, %.6g>\n" %
# (ob.pov.inside_vector[0],
# ob.pov.inside_vector[1],
# ob.pov.inside_vector[2]))
# onceCSG = 1
# '''

# if ob.pov.hollow:
# File.write("\thollow\n")
# if ob.pov.double_illuminate:
# File.write("\tdouble_illuminate\n")
# if ob.pov.sturm:
# File.write("\tsturm\n")
# if ob.pov.no_shadow:
# File.write("\tno_shadow\n")
# if ob.pov.no_image:
# File.write("\tno_image\n")
# if ob.pov.no_reflection:
# File.write("\tno_reflection\n")
# if ob.pov.no_radiosity:
# File.write("\tno_radiosity\n")
# if ob.pov.inverse:
# File.write("\tinverse\n")
# if ob.pov.hierarchy:
# File.write("\thierarchy\n")

# # XXX, Commented definitions
# '''
# if scene.pov.photon_enable:
# File.write("photons {\n")
# if ob.pov.target:
# File.write("target %.4g\n"%ob.pov.target_value)
# if ob.pov.refraction:
# File.write("refraction on\n")
# if ob.pov.reflection:
# File.write("reflection on\n")
# if ob.pov.pass_through:
# File.write("pass_through\n")
# File.write("}\n")
# if ob.pov.object_ior > 1:
# File.write("interior {\n")
# File.write("ior %.4g\n"%ob.pov.object_ior)
# if scene.pov.photon_enable and ob.pov.target and ob.pov.refraction and ob.pov.dispersion:
# File.write("ior %.4g\n"%ob.pov.dispersion_value)
Maurice Raybaud's avatar
Maurice Raybaud committed
# File.write("ior %s\n"%ob.pov.dispersion_samples)
# if scene.pov.photon_enable == False:
# File.write("caustics %.4g\n"%ob.pov.fake_caustics_power)
# '''
Luca Bonavita's avatar
Luca Bonavita committed
def write_pov(filename, scene=None, info_callback=None):
Maurice Raybaud's avatar
Maurice Raybaud committed
    """Main export process from Blender UI to POV syntax and write to exported file """
    file = open(filename, "w")
Luca Bonavita's avatar
Luca Bonavita committed

    # Only for testing
    if not scene:
        scene = bpy.data.scenes[0]

    render = scene.render
    world = scene.world
    global_matrix = mathutils.Matrix.Rotation(-pi / 2.0, 4, 'X')
    comments = scene.pov.comments_enable and not scene.pov.tempfiles_enable
Maurice Raybaud's avatar
Maurice Raybaud committed
    linebreaksinlists = scene.pov.list_lf_enable and not scene.pov.tempfiles_enable
    feature_set = bpy.context.preferences.addons[__package__].preferences.branch_feature_set_povray
    using_uberpov = feature_set == 'uberpov'
    pov_binary = PovrayRender._locate_binary()
    if using_uberpov:
        print("Unofficial UberPOV feature set chosen in preferences")
    else:
        print("Official POV-Ray 3.7 feature set chosen in preferences")
Campbell Barton's avatar
Campbell Barton committed
    if 'uber' in pov_binary:
Maurice Raybaud's avatar
Maurice Raybaud committed
        print("The name of the binary suggests you are probably rendering with Uber POV engine")
Maurice Raybaud's avatar
Maurice Raybaud committed
        print("The name of the binary suggests you are probably rendering with standard POV engine")
Maurice Raybaud's avatar
Maurice Raybaud committed
    def set_tab(tabtype, spaces):
        tab_str = ""
Maurice Raybaud's avatar
Maurice Raybaud committed
            tab_str = ""
Maurice Raybaud's avatar
Maurice Raybaud committed
            tab_str = "\t"
Maurice Raybaud's avatar
Maurice Raybaud committed
            tab_str = spaces * " "
        return tab_str
Maurice Raybaud's avatar
Maurice Raybaud committed
    tab = set_tab(scene.pov.indentation_character, scene.pov.indentation_spaces)
    if not scene.pov.tempfiles_enable:
Maurice Raybaud's avatar
Maurice Raybaud committed
        def tab_write(str_o):
Maurice Raybaud's avatar
Maurice Raybaud committed
            """Indent POV syntax from brackets levels and write to exported file """
Maurice Raybaud's avatar
Maurice Raybaud committed
            global tab_level
            brackets = str_o.count("{") - str_o.count("}") + str_o.count("[") - str_o.count("]")
            if brackets < 0:
Maurice Raybaud's avatar
Maurice Raybaud committed
                tab_level = tab_level + brackets
            if tab_level < 0:
                print("Indentation Warning: tab_level = %s" % tab_level)
                tab_level = 0
            if tab_level >= 1:
                file.write("%s" % tab * tab_level)
            file.write(str_o)
            if brackets > 0:
Maurice Raybaud's avatar
Maurice Raybaud committed
                tab_level = tab_level + brackets
Maurice Raybaud's avatar
Maurice Raybaud committed
        def tab_write(str_o):
Maurice Raybaud's avatar
Maurice Raybaud committed
            """write directly to exported file if user checked autonamed temp files (faster)."""
            file.write(str_o)
Maurice Raybaud's avatar
Maurice Raybaud committed
    def unique_name(name, name_seq):
Maurice Raybaud's avatar
Maurice Raybaud committed
        """Increment any generated POV name that could get identical to avoid collisions"""
Luca Bonavita's avatar
Luca Bonavita committed

Maurice Raybaud's avatar
Maurice Raybaud committed
        if name not in name_seq:
Campbell Barton's avatar
Campbell Barton committed
            name = string_strip_hyphen(name)
Luca Bonavita's avatar
Luca Bonavita committed
            return name

        name_orig = name
        i = 1
Maurice Raybaud's avatar
Maurice Raybaud committed
        while name in name_seq:
Luca Bonavita's avatar
Luca Bonavita committed
            i += 1
Campbell Barton's avatar
Campbell Barton committed
        name = string_strip_hyphen(name)
Luca Bonavita's avatar
Luca Bonavita committed
        return name

Maurice Raybaud's avatar
Maurice Raybaud committed
    def write_matrix(matrix):
        """Translate some transform matrix from Blender UI
Maurice Raybaud's avatar
Maurice Raybaud committed
        to POV syntax and write to exported file """
Maurice Raybaud's avatar
Maurice Raybaud committed
        tab_write(
            "matrix <%.6f, %.6f, %.6f,  %.6f, %.6f, %.6f,  %.6f, %.6f, %.6f,  %.6f, %.6f, %.6f>\n"
            % (
                matrix[0][0],
                matrix[1][0],
                matrix[2][0],
                matrix[0][1],
                matrix[1][1],
                matrix[2][1],
                matrix[0][2],
                matrix[1][2],
                matrix[2][2],
                matrix[0][3],
                matrix[1][3],
                matrix[2][3],
            )
        )
Luca Bonavita's avatar
Luca Bonavita committed

Maurice Raybaud's avatar
Maurice Raybaud committed
    material_names_dictionary = {}
    DEF_MAT_NAME = ""  # or "Default"?
    # -----------------------------------------------------------------------------
Maurice Raybaud's avatar
Maurice Raybaud committed
    def export_meta(metas):
Maurice Raybaud's avatar
Maurice Raybaud committed
        """write all POV blob primitives and Blender Metas to exported file """
Luca Bonavita's avatar
Luca Bonavita committed
        # TODO - blenders 'motherball' naming is not supported.
        if comments and len(metas) >= 1:
            file.write("//--Blob objects--\n\n")
        # Get groups of metaballs by blender name prefix.
        meta_group = {}
        meta_elems = {}
Luca Bonavita's avatar
Luca Bonavita committed
        for ob in metas:
            if prefix not in meta_group:
                meta_group[prefix] = ob  # .data.threshold
            elems = [
                (elem, ob)
                for elem in ob.data.elements
Maurice Raybaud's avatar
Maurice Raybaud committed
                if elem.type in {'BALL', 'ELLIPSOID', 'CAPSULE', 'CUBE', 'PLANE'}
            if prefix in meta_elems:
                meta_elems[prefix].extend(elems)
                meta_elems[prefix] = elems
Maurice Raybaud's avatar
Maurice Raybaud committed
                tab_write("\n//dummy sphere to represent empty meta location\n")
                tab_write(
                    "sphere {<%.6g, %.6g, %.6g>,0 pigment{rgbt 1} "
                    "no_image no_reflection no_radiosity "
                    "photons{pass_through collect off} hollow}\n\n"
                    % (ob.location.x, ob.location.y, ob.location.z)
                )  # ob.name > povdataname)
Campbell Barton's avatar
Campbell Barton committed
            # other metaballs
            else:
Maurice Raybaud's avatar
Maurice Raybaud committed
                for mg, mob in meta_group.items():
                    if len(meta_elems[mg]) != 0:
Maurice Raybaud's avatar
Maurice Raybaud committed
                        tab_write("blob{threshold %.4g // %s \n" % (mob.data.threshold, mg))
                        for elems in meta_elems[mg]:
                            elem = elems[0]
                            loc = elem.co
                            stiffness = elem.stiffness
                            if elem.use_negative:
                                stiffness = -stiffness
Maurice Raybaud's avatar
Maurice Raybaud committed
                                tab_write(
                                    "sphere { <%.6g, %.6g, %.6g>, %.4g, %.4g "
Maurice Raybaud's avatar
Maurice Raybaud committed
                                    % (loc.x, loc.y, loc.z, elem.radius, stiffness)
Maurice Raybaud's avatar
Maurice Raybaud committed
                                write_matrix(global_matrix @ elems[1].matrix_world)
                                tab_write("}\n")
Maurice Raybaud's avatar
Maurice Raybaud committed
                                tab_write(
                                    "sphere{ <%.6g, %.6g, %.6g>,%.4g,%.4g "
                                    % (
                                        loc.x / elem.size_x,
                                        loc.y / elem.size_y,
                                        loc.z / elem.size_z,
                                        elem.radius,
                                        stiffness,
                                    )
                                )
Maurice Raybaud's avatar
Maurice Raybaud committed
                                tab_write(
                                    "scale <%.6g, %.6g, %.6g>"
                                    % (elem.size_x, elem.size_y, elem.size_z)
                                )
Maurice Raybaud's avatar
Maurice Raybaud committed
                                write_matrix(global_matrix @ elems[1].matrix_world)
                                tab_write("}\n")
                            elif elem.type == 'CAPSULE':
Maurice Raybaud's avatar
Maurice Raybaud committed
                                tab_write(
                                    "cylinder{ <%.6g, %.6g, %.6g>,<%.6g, %.6g, %.6g>,%.4g,%.4g "
                                    % (
                                        (loc.x - elem.size_x),
                                        (loc.y),
                                        (loc.z),
                                        (loc.x + elem.size_x),
                                        (loc.y),
                                        (loc.z),
                                        elem.radius,
                                        stiffness,
                                    )
                                )
Maurice Raybaud's avatar
Maurice Raybaud committed
                                # tab_write("scale <%.6g, %.6g, %.6g>" % (elem.size_x, elem.size_y, elem.size_z))
                                write_matrix(global_matrix @ elems[1].matrix_world)
                                tab_write("}\n")
Maurice Raybaud's avatar
Maurice Raybaud committed
                                tab_write(
                                    "cylinder { -x*8, +x*8,%.4g,%.4g translate<%.6g,%.6g,%.6g> scale  <1/4,1,1> scale <%.6g, %.6g, %.6g>\n"
                                    % (
                                        elem.radius * 2.0,
                                        stiffness / 4.0,
                                        loc.x,
                                        loc.y,
                                        loc.z,
                                        elem.size_x,
                                        elem.size_y,
                                        elem.size_z,
                                    )
                                )
Maurice Raybaud's avatar
Maurice Raybaud committed
                                write_matrix(global_matrix @ elems[1].matrix_world)
                                tab_write("}\n")
                                tab_write(
                                    "cylinder { -y*8, +y*8,%.4g,%.4g translate<%.6g,%.6g,%.6g> scale <1,1/4,1> scale <%.6g, %.6g, %.6g>\n"
                                    % (
                                        elem.radius * 2.0,
                                        stiffness / 4.0,
                                        loc.x,
                                        loc.y,
                                        loc.z,
                                        elem.size_x,
                                        elem.size_y,
                                        elem.size_z,
                                    )
                                )
Maurice Raybaud's avatar
Maurice Raybaud committed
                                write_matrix(global_matrix @ elems[1].matrix_world
Loading
Loading full blame...