object.py

#############################################
# THUG1/2 OBJECT CONFIGURE
#############################################
import bpy
import struct
import mathutils
import math
import os, sys
from bpy.props import *
from . constants import *
from . helpers import *
from . autorail import *
from . collision import *
from . import scene_props
from . import qb 
from . qb import *
from . import script_template

# METHODS
#############################################
def _thug_object_settings_draw(self, context):
    if not context.object: return
    ob = context.object
    # ********************************************************
    # * LEVEL LIGHT
    # ********************************************************
    if ob.type == "LAMP" and ob.data.type in ["POINT", "AREA"]:
        self.layout.row().prop(ob, "thug_export_scene", toggle=True, icon='SCENE_DATA', text='Export to Scene')
        row = self.layout.row()
        row.column().prop(ob, "thug_created_at_start", toggle=True, icon='MOD_BUILD')
        row.column().prop(ob, "thug_network_option", text='')
        row = self.layout.row()
        row.column().row().prop(ob, "thug_tod_values", expand=True)
        box = self.layout.box().column()
        box.row().prop(ob.data.thug_light_props, "light_type")
        if ob.data.thug_light_props.light_type == 'TUBE':
            box.row().prop(ob.data.thug_light_props, "light_end_pos")
        if ob.data.thug_light_props.light_type == 'TUBE' or  ob.data.thug_light_props.light_type == 'SPHERE':
            box.row().prop(ob.data.thug_light_props, "light_radius")
        elif ob.data.thug_light_props.light_type == 'AREA':
            box.row().prop(ob.data.thug_light_props, "light_area")
        else:
            box.row().prop(ob.data.thug_light_props, "light_radius")
        box.row().prop(ob.data, "color")
        box.row().prop(ob.data, "energy")
        row = box.row()
        row.column().prop(ob.data.thug_light_props, "light_excludeskater")
        row.column().prop(ob.data.thug_light_props, "light_excludelevel")
        
    elif ob.type == "EMPTY":
        self.layout.row().prop(ob, "thug_export_scene", toggle=True, icon='SCENE_DATA', text='Export to Scene')
        self.layout.row().prop(ob.thug_empty_props, "empty_type")
        # ********************************************************
        # * RESTART
        # ********************************************************
        if ob.thug_empty_props.empty_type == "Restart":
            box = self.layout.box().column()
            box.row().prop(ob.thug_restart_props, "restart_type")
            box.row().prop(ob.thug_restart_props, "restart_name")
            row = box.row()
            row.column().prop(ob.thug_restart_props, "restart_p1")
            row.column().prop(ob.thug_restart_props, "restart_p2")
            row = box.row()
            row.column().prop(ob.thug_restart_props, "restart_team")
            row.column().prop(ob.thug_restart_props, "restart_gen")
            row = box.row()
            row.column().prop(ob.thug_restart_props, "restart_multi")
            row.column().prop(ob.thug_restart_props, "restart_horse")
            row = box.row()
            row.column().prop(ob.thug_restart_props, "restart_ctf")
            row.column().prop(ob.thug_restart_props, "primary_restart")
        # ********************************************************
        # * REFLECTION PROBE 
        # ********************************************************
        if ob.thug_empty_props.empty_type == 'CubemapProbe':
            box = self.layout.box().column(align=True)
            box.row().prop(ob.thug_cubemap_props, "resolution")
            box.row().prop(ob.thug_cubemap_props, "box_size")
            
        # ********************************************************
        # * LIGHT PROBE 
        # ********************************************************
        if ob.thug_empty_props.empty_type == 'LightProbe':
            box = self.layout.box().column(align=True)
            box.row().prop(ob.thug_cubemap_props, "resolution")
            
        # ********************************************************
        # * LIGHT VOLUME 
        # ********************************************************
        if ob.thug_empty_props.empty_type == 'LightVolume':
            box = self.layout.box().column(align=True)
            box.row().prop(ob.thug_lightvolume_props, "box_size")
            
        # ********************************************************
        # * PROXIMNODE 
        # ********************************************************
        elif ob.thug_empty_props.empty_type == "ProximNode":
            box = self.layout.box().column()
            box.row().prop(ob.thug_proxim_props, "proxim_type")
            box.row().prop(ob.thug_proxim_props, "proxim_shape")
            box.row().prop(ob.thug_proxim_props, "proxim_radius")
        # ********************************************************
        # * EMITTER OBJECT
        # ********************************************************
        elif ob.thug_empty_props.empty_type == "EmitterObject":
            box = self.layout.box().column()
            box.row().prop(ob.thug_emitter_props, "emit_type")
            box.row().prop(ob.thug_emitter_props, "emit_radius")
        # ********************************************************
        # * GENERIC
        # ********************************************************
        elif ob.thug_empty_props.empty_type == "GenericNode":
            self.layout.row().prop(ob.thug_generic_props, "generic_type")
        # ********************************************************
        # * GAME OBJECT
        # ********************************************************
        elif ob.thug_empty_props.empty_type == "GameObject":
            box = self.layout.box().column()
            box.row().prop(ob.thug_go_props, "go_type")
            if ob.thug_go_props.go_type == "Custom":
                box.row().prop(ob.thug_go_props, "go_type_other")
                box.row().prop_search(ob.thug_go_props, "go_model",
                context.window_manager.thug_game_assets, "models")
                #box.row().prop(ob.thug_go_props, "go_model")
            box.row().prop(ob.thug_go_props, "go_suspend")
        # ********************************************************
        # * PEDESTRIAN
        # ********************************************************
        elif ob.thug_empty_props.empty_type == "Pedestrian":
            box = self.layout.box().column()
            box.row().prop(ob.thug_ped_props, "ped_type")
            box.row().prop(ob.thug_ped_props, "ped_source", expand=True)
            if ob.thug_ped_props.ped_source == "Profile":
                box.row().prop(ob.thug_ped_props, "ped_profile")
            else:
                box.row().prop_search(
                ob.thug_ped_props, "ped_model",
                context.window_manager.thug_game_assets, "skins")#, icon='SCRIPT')
                #box.row().prop(ob.thug_ped_props, "ped_model")
            box.row().prop(ob.thug_ped_props, "ped_skeleton")
            box.row().prop(ob.thug_ped_props, "ped_animset")
            box.row().prop(ob.thug_ped_props, "ped_extra_anims")
            box.row().prop(ob.thug_ped_props, "ped_suspend")
        # ********************************************************
        # * VEHICLE
        # ********************************************************
        elif ob.thug_empty_props.empty_type == "Vehicle":
            box = self.layout.box().column()
            box.row().prop(ob.thug_veh_props, "veh_type")
            box.row().prop_search(
            ob.thug_veh_props, "veh_model",
            context.window_manager.thug_game_assets, "models")
            #box.row().prop(ob.thug_veh_props, "veh_model")
            if ob.thug_veh_props.veh_model != "" and ob.thug_veh_props.veh_model != "none":
                box.row().prop(ob.thug_veh_props, "veh_usemodellights")
                box.row().prop(ob.thug_veh_props, "veh_allowreplacetex")
                box.row().prop(ob.thug_veh_props, "veh_skeleton")
            box.row().prop(ob.thug_veh_props, "veh_suspend")
            row = box.row()
            row.column().prop(ob.thug_veh_props, "veh_norail")
            row.column().prop(ob.thug_veh_props, "veh_noskitch")
            
        # ********************************************************
        # * PARTICLE OBJECT
        # ********************************************************
        elif ob.thug_empty_props.empty_type == "ParticleObject":
            box = self.layout.box().column()
            box.row().label(text='Emitter dimensions')
            box.row().prop(ob.thug_particle_props, "particle_boxdimsstart", text='Start')
            box.row().prop(ob.thug_particle_props, "particle_boxdimsmid", text='Mid')
            box.row().prop(ob.thug_particle_props, "particle_boxdimsend", text='End')
            
            box = self.layout.box().column()
            box.row().label(text='Emitter locations')
            #box.row().prop(ob.thug_particle_props, "particle_usestartpos")
            #if ob.thug_particle_props.particle_usestartpos == True:
            box.row().prop(ob.thug_particle_props, "particle_startposition", text='Start')
            box.row().prop(ob.thug_particle_props, "particle_midposition", text='Mid')
            box.row().prop(ob.thug_particle_props, "particle_endposition", text='End')
            box.row().prop(ob.thug_particle_props, "particle_suspend")
            
            box = self.layout.box().column()
            box.row().prop_search(ob.thug_particle_props, "particle_texture",
            context.window_manager.thug_game_assets, "particle_textures")
            #box.row().prop(ob.thug_particle_props, "particle_texture")
            box.row().prop(ob.thug_particle_props, "particle_type")
            box.row().prop(ob.thug_particle_props, "particle_blendmode")
            row = box.row()
            row.column().prop(ob.thug_particle_props, "particle_fixedalpha")
            row.column().prop(ob.thug_particle_props, "particle_alphacutoff")
            
            box = self.layout.box().column()
            box.row().prop(ob.thug_particle_props, "particle_startcolor")
            box.row().prop(ob.thug_particle_props, "particle_usecolormidtime")
            if ob.thug_particle_props.particle_usecolormidtime == True:
                box.row().prop(ob.thug_particle_props, "particle_colormidtime")
            box.row().prop(ob.thug_particle_props, "particle_midcolor")
            box.row().prop(ob.thug_particle_props, "particle_endcolor")
            
            box = self.layout.box().column()
            box.row().prop(ob.thug_particle_props, "particle_maxstreams")
            box.row().prop(ob.thug_particle_props, "particle_emitrate")
            box.row().prop(ob.thug_particle_props, "particle_lifetime")
            box.row().prop(ob.thug_particle_props, "particle_usemidpoint")
            box.row().prop(ob.thug_particle_props, "particle_midpointpct")
            box.row().prop(ob.thug_particle_props, "EmitScript")
            box.row().prop(ob.thug_particle_props, "Force")
            box.row().prop(ob.thug_particle_props, "Speed")
            box.row().prop(ob.thug_particle_props, "EmitTarget")
            box.row().prop(ob.thug_particle_props, "UsePulseEmit")
            box.row().prop(ob.thug_particle_props, "RandomEmitRate")
            box.row().prop(ob.thug_particle_props, "RandomEmitDelay")
            box.row().prop(ob.thug_particle_props, "EmitRate1")
            box.row().prop(ob.thug_particle_props, "EmitRate1Delay")
            box.row().prop(ob.thug_particle_props, "EmitRate2")
            box.row().prop(ob.thug_particle_props, "EmitRate2Delay")
            
            box = self.layout.box().column()
            box.row().prop(ob.thug_particle_props, "particle_radius")
            box.row().prop(ob.thug_particle_props, "particle_radiusspread")
            box.row().prop(ob.thug_particle_props, "Size")
            box.row().prop(ob.thug_particle_props, "Width")
            box.row().prop(ob.thug_particle_props, "AngleSpread")
            box.row().prop(ob.thug_particle_props, "UseMidTime")
            if ob.thug_particle_props.UseMidTime:
                box.row().prop(ob.thug_particle_props, "MidTime")
        
    if ob.type == "EMPTY" and ob.thug_empty_props.empty_type in ( "Pedestrian", "Vehicle" ):
        self.layout.row().prop_search(
            ob, "thug_rail_connects_to",
            context.window_manager.thug_all_nodes, "paths")
        if (ob.thug_rail_connects_to and
                ob.thug_rail_connects_to in bpy.data.objects and
                bpy.data.objects[ob.thug_rail_connects_to].type != "CURVE"):
            self.layout.label(text=ob.thug_rail_connects_to + " is not a curve!", icon="ERROR")

    if ob.type == "MESH":
        box = self.layout.column()
        box.row().prop(ob, "thug_object_class")
        row = box.row()
        row.column().prop(ob, "thug_export_collision", toggle=True, icon='OBJECT_DATA', text='Collision')
        row.column().prop(ob, "thug_export_scene", toggle=True, icon='SCENE_DATA', text='Scene')
        
        if ob.thug_object_class == "LevelObject":
            box = self.layout.box().column()
            box.row().prop(ob.thug_levelobj_props, "obj_type")
            box.row().prop(ob.thug_levelobj_props, "obj_bouncy")
            if ob.thug_levelobj_props.obj_bouncy:
                box.row().prop(ob.thug_levelobj_props, "center_of_mass")
                box.row().prop(ob.thug_levelobj_props, "contacts")
                box.row().prop(ob.thug_levelobj_props, "coeff_restitution")
                box.row().prop(ob.thug_levelobj_props, "coeff_friction")
                box.row().prop(ob.thug_levelobj_props, "skater_collision_impulse_factor")
                box.row().prop(ob.thug_levelobj_props, "skater_collision_rotation_factor")
                box.row().prop(ob.thug_levelobj_props, "skater_collision_assent")
                box.row().prop(ob.thug_levelobj_props, "skater_collision_radius")
                box.row().prop(ob.thug_levelobj_props, "mass_over_moment")
                box.row().prop(ob.thug_levelobj_props, "SoundType")
                
                box.row().prop_search(ob.thug_levelobj_props, "stuckscript", 
                        context.window_manager.thug_all_nodes, "scripts", icon='SCRIPT' )       

            box.row().prop_search(ob, "thug_rail_connects_to", context.window_manager.thug_all_nodes, "paths")
            
        elif ob.thug_object_class == "LevelGeometry" and ob.thug_export_scene:  
            box = self.layout.box().column()          
            box.row().prop(ob, "thug_lightgroup")
            row = box.row()
            row.column().prop(ob, "thug_tod_controlled")
            row.column().prop(ob, "thug_is_shadow_volume")
            row = box.row()
            row.column().prop(ob, "thug_no_skater_shadow")
            row.column().prop(ob, "thug_material_blend")
            row = box.row()
            #row.column().prop(ob, "thug_allow_uv_scale")
            row.column().prop(ob.data.thug_billboard_props, "is_billboard")
            row.column().prop(ob, "thug_lighting_mesh")
            if ob.data.thug_billboard_props.is_billboard:
                box.row().prop(ob.data.thug_billboard_props, "type", expand=True)
                if ob.data.thug_billboard_props.type == 'AXIS':
                    #box.row().prop(ob.data.thug_billboard_props, "pivot_origin")
                    #box.row().prop(ob.data.thug_billboard_props, "pivot_pos")
                    box.row().prop(ob.data.thug_billboard_props, "pivot_axis")
            if ob.thug_tod_controlled:
                box.row().prop(ob, "thug_tod_values", expand=True)
                
                
                
        box = self.layout.column()    
        row = box.row()
        row.column().prop(ob, "thug_occluder", toggle=True, icon='MOD_BEVEL')
        row.column().prop(ob, "thug_do_autosplit", toggle=True, text='AutoSplit', icon='MOD_EDGESPLIT')
        if ob.thug_do_autosplit:
            row = box.row()
            row.column().prop(ob, "thug_do_autosplit_faces_per_subobject", text='Faces')
            row.column().prop(ob, "thug_do_autosplit_max_radius", text='Radius')
        
    if ob.type == "CURVE":
        self.layout.row().prop(ob, "thug_export_scene", toggle=True, icon='SCENE_DATA', text='Export to Scene')
        self.layout.row().prop(ob, "thug_path_type")
        if ob.thug_path_type == 'Waypoint':
            box = self.layout.box().column()
            box.row().label(text="Waypoint type")
            box.row().prop(ob.thug_waypoint_props, "waypt_type", expand=True)
            if ob.thug_waypoint_props.waypt_type == 'PedAI':
                box.row().prop(ob.thug_waypoint_props, "PedType", expand=True)
        
    if ob.type == "MESH" or (ob.type == "CURVE" and ob.thug_path_type != "") or ob.type == "EMPTY":
        box = self.layout.column()
        row = box.row()
        row.column().prop(ob, "thug_created_at_start", toggle=True, icon='MOD_BUILD')
        row.column().prop(ob, "thug_network_option", text='')
                
        if ob.type == "MESH" or (ob.type == "CURVE" and ob.thug_path_type == "Rail"):
            self.layout.row().prop(ob, "thug_is_trickobject", toggle=True, text='Trick Object', icon='MOD_DYNAMICPAINT')
            if ob.thug_is_trickobject:
                self.layout.row().prop(ob, "thug_cluster_name")
                if not is_string_clean(ob.thug_cluster_name):
                    box = self.layout.box().column(True)
                    box.label(text="Bad cluster name!", icon="ERROR")
                    box.label(text="Only valid characters are small and large letters")
                    box.label(text="digits, and underscores.")
                    
        # New template system below!
        box = self.layout.box().column()
        box.row().prop(ob.thug_triggerscript_props, "template_name")
        if ob.thug_triggerscript_props.template_name not in [ "None", "Custom", "Template" ]:
            #print("attempting to show template params")
            tmpl = script_template.get_template(ob.thug_triggerscript_props.template_name)
            #print(tmpl)
            paramindex = 0
            
            for prm in tmpl['Parameters']:
                paramindex += 1
                if prm['Name'] and prm['Type']:
                    if prm['Type'] == 'String' or prm['Type'] == 'Name':
                        box.row().prop(ob.thug_triggerscript_props, "param" + str(paramindex) + "_string", text=prm['Name'])
                    elif prm['Type'] == 'Integer' or prm['Type'] == 'Int':
                        box.row().prop(ob.thug_triggerscript_props, "param" + str(paramindex) + "_int", text=prm['Name'])
                    elif prm['Type'] == 'Float':
                        box.row().prop(ob.thug_triggerscript_props, "param" + str(paramindex) + "_float", text=prm['Name'])
                    elif prm['Type'] == 'Boolean':
                        box.row().prop(ob.thug_triggerscript_props, "param" + str(paramindex) + "_bool", text=prm['Name'])
                    elif prm['Type'] == 'Enum':
                        box.row().prop(ob.thug_triggerscript_props, "param" + str(paramindex) + "_enum", text=prm['Name'])
                    elif prm['Type'] == 'Flags':
                        box.row().prop_menu_enum(ob.thug_triggerscript_props, "param" + str(paramindex) + "_flags", text=prm['Name'], icon='SETTINGS')
                    elif prm['Type'] == 'Restart':
                        box.row().prop_search(ob.thug_triggerscript_props, "param" + str(paramindex) + "_string", 
                        context.window_manager.thug_all_nodes, "restarts", text=prm['Name'])
                    elif prm['Type'] == 'Rail' or prm['Type'] == 'Path':
                        box.row().prop_search(ob.thug_triggerscript_props, "param" + str(paramindex) + "_string", 
                        context.window_manager.thug_all_nodes, "paths", text=prm['Name'])
                    elif prm['Type'] == 'Script':
                        box.row().prop_search(ob.thug_triggerscript_props, "param" + str(paramindex) + "_string", 
                        context.window_manager.thug_all_nodes, "scripts", text=prm['Name'])
                        
        elif ob.thug_triggerscript_props.template_name == "Custom":
            box.row().prop_search(ob.thug_triggerscript_props, "custom_name", 
                        context.window_manager.thug_all_nodes, "scripts", icon='SCRIPT' )
            box.row().operator(THUGCreateTriggerScript.bl_idname, text=THUGCreateTriggerScript.bl_label)
        elif ob.thug_triggerscript_props.template_name == "Template":
            box.row().prop_search(ob.thug_triggerscript_props, "custom_name", 
                        context.window_manager.thug_all_nodes, "templates", icon='SCRIPT' )
        # End new template system
        
    if (ob.type == "CURVE" and ob.thug_path_type in ("Rail", "Ladder", "Waypoint", "Custom")):
        # context.window_manager.thug_rail_objects = [obj for obj in context.scene.objects if obj.type == "CURVE"]
        if ob.thug_path_type == "Rail":
            self.layout.row().prop(ob, "thug_rail_terrain_type")
            self.layout.row().operator(AutoRailMesh.bl_idname, text=AutoRailMesh.bl_label)
            
        self.layout.row().prop_search(
            ob, "thug_rail_connects_to",
            context.window_manager.thug_all_nodes, "paths")
        if (ob.thug_rail_connects_to and
                ob.thug_rail_connects_to in bpy.data.objects and
                bpy.data.objects[ob.thug_rail_connects_to].type != "CURVE"):
            self.layout.label(text=ob.thug_rail_connects_to + " is not a curve!", icon="ERROR")

    self.layout.row().prop(ob, "thug_node_expansion")
    scene_props.update_node_collection(self, context)

                            
#----------------------------------------------------------------------------------
def _update_rails_collection(self, context):
    context.window_manager.thug_all_rails.clear()
    for ob in bpy.data.objects:
        if ob.type == "CURVE" and ob.thug_path_type in ("Rail", "Ladder", "Waypoint", "Custom"):
            entry = context.window_manager.thug_all_rails.add()
            entry.name = ob.name

#----------------------------------------------------------------------------------
def _update_restart_collection(self, context):
    context.window_manager.thug_all_restarts.clear()
    for ob in bpy.data.objects:
        if ob.type == "EMPTY" and ob.thug_empty_props.empty_type in ("Restart"):
            entry = context.window_manager.thug_all_restarts.add()
            entry.name = ob.name



# PROPERTIES
#############################################
class THUG_PT_ObjectSettingsTools(bpy.types.Panel):
    bl_label = "TH Object Settings"
    bl_region_type = "UI"
    bl_space_type = "VIEW_3D"
    bl_category = "THUG Tools"

    @classmethod
    def poll(cls, context):
        return context.object and context.preferences.addons[ADDON_NAME].preferences.object_settings_tools

    def draw(self, context):
        _thug_object_settings_draw(self, context)

#----------------------------------------------------------------------------------
class THUG_PT_ObjectSettings(bpy.types.Panel):
    bl_label = "TH Object Settings"
    bl_region_type = "WINDOW"
    bl_space_type = "PROPERTIES"
    bl_context = "object"

    @classmethod
    def poll(cls, context):
        return context.object

    def draw(self, context):
        _thug_object_settings_draw(self, context)