nodes.py

# ##### 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 #####

# <pep8 compliant>

import bpy

from bpy.utils import register_class
from bpy.types import Node, ShaderNodeTree, CompositorNodeTree, TextureNodeTree#, NodeSocket
from bpy.props import (
        StringProperty,
        BoolProperty,
        IntProperty,
        FloatProperty,
        FloatVectorProperty,
        EnumProperty,
        #PointerProperty,
        #CollectionProperty,
        )



############### object

class ObjectNodeTree(bpy.types.NodeTree):
    '''Povray Material Nodes'''

    bl_idname = 'ObjectNodeTree'
    bl_label = 'Povray Object Nodes'
    bl_icon = 'PLUGIN'

    @classmethod
    def poll(cls, context):
        return context.scene.render.engine == 'POVRAY_RENDER'

    @classmethod
    def get_from_context(cls, context):
        ob = context.active_object
        if ob and ob.type not in {'LIGHT'}:
            ma = ob.active_material
            if ma is not None:
                nt_name = ma.node_tree
                if nt_name != '':
                    return nt_name, ma, ma
        return (None, None, None)

    def update(self):
        self.refresh = True
################### output #############################################################################################

class PovrayOutputNode(Node, ObjectNodeTree):
    '''Output'''
    bl_idname = 'PovrayOutputNode'
    bl_label = 'Output'
    bl_icon = 'SOUND'

    def init(self, context):

        self.inputs.new('PovraySocketTexture', "Texture")

    def draw_buttons(self, context, layout):

        ob=context.object
        layout.prop(ob.pov, "object_ior",slider=True)

    def draw_buttons_ext(self, context, layout):

        ob=context.object
        layout.prop(ob.pov, "object_ior",slider=True)

    def draw_label(self):
        return "Output"



################### material ###########################################################################################
class PovrayTextureNode(Node, ObjectNodeTree):
    '''Texture'''
    bl_idname = 'PovrayTextureNode'
    bl_label = 'Simple texture'
    bl_icon = 'SOUND'

    def init(self, context):

        color=self.inputs.new('PovraySocketColor', "Pigment")
        color.default_value=(1,1,1)
        normal=self.inputs.new('NodeSocketFloat', "Normal")
        normal.hide_value=True
        finish=self.inputs.new('NodeSocketVector', "Finish")
        finish.hide_value=True

        self.outputs.new('PovraySocketTexture', "Texture")

    def draw_label(self):
        return "Simple texture"

class PovrayFinishNode(Node, ObjectNodeTree):
    '''Finish'''
    bl_idname = 'PovrayFinishNode'
    bl_label = 'Finish'
    bl_icon = 'SOUND'

    def init(self, context):

        self.inputs.new('PovraySocketFloat_0_1', "Emission")
        ambient=self.inputs.new('NodeSocketVector', "Ambient")
        ambient.hide_value=True
        diffuse=self.inputs.new('NodeSocketVector', "Diffuse")
        diffuse.hide_value=True
        specular=self.inputs.new('NodeSocketVector', "Highlight")
        specular.hide_value=True
        mirror=self.inputs.new('NodeSocketVector', "Mirror")
        mirror.hide_value=True
        iridescence=self.inputs.new('NodeSocketVector', "Iridescence")
        iridescence.hide_value=True
        subsurface=self.inputs.new('NodeSocketVector', "Translucency")
        subsurface.hide_value=True
        self.outputs.new('NodeSocketVector', "Finish")

    def draw_label(self):
        return "Finish"

class PovrayDiffuseNode(Node, ObjectNodeTree):
    '''Diffuse'''
    bl_idname = 'PovrayDiffuseNode'
    bl_label = 'Diffuse'
    bl_icon = 'SOUND'

    def init(self, context):

        intensity=self.inputs.new('PovraySocketFloat_0_1', "Intensity")
        intensity.default_value=0.8
        albedo=self.inputs.new('NodeSocketBool', "Albedo")
        albedo.default_value=False
        brilliance=self.inputs.new('PovraySocketFloat_0_10', "Brilliance")
        brilliance.default_value=1.8
        self.inputs.new('PovraySocketFloat_0_1', "Crand")
        self.outputs.new('NodeSocketVector', "Diffuse")

    def draw_label(self):
        return "Diffuse"

class PovrayPhongNode(Node, ObjectNodeTree):
    '''Phong'''
    bl_idname = 'PovrayPhongNode'
    bl_label = 'Phong'
    bl_icon = 'SOUND'

    def init(self, context):

        albedo=self.inputs.new('NodeSocketBool', "Albedo")
        intensity=self.inputs.new('PovraySocketFloat_0_1', "Intensity")
        intensity.default_value=0.8
        phong_size=self.inputs.new('PovraySocketInt_0_256', "Size")
        phong_size.default_value=60
        metallic=self.inputs.new('PovraySocketFloat_0_1', "Metallic")

        self.outputs.new('NodeSocketVector', "Phong")

    def draw_label(self):
        return "Phong"

class PovraySpecularNode(Node, ObjectNodeTree):
    '''Specular'''
    bl_idname = 'PovraySpecularNode'
    bl_label = 'Specular'
    bl_icon = 'SOUND'

    def init(self, context):

        albedo=self.inputs.new('NodeSocketBool', "Albedo")
        intensity=self.inputs.new('PovraySocketFloat_0_1', "Intensity")
        intensity.default_value=0.8
        roughness=self.inputs.new('PovraySocketFloat_0_1', "Roughness")
        roughness.default_value=0.02
        metallic=self.inputs.new('PovraySocketFloat_0_1', "Metallic")

        self.outputs.new('NodeSocketVector', "Specular")

    def draw_label(self):
        return "Specular"

class PovrayMirrorNode(Node, ObjectNodeTree):
    '''Mirror'''
    bl_idname = 'PovrayMirrorNode'