mirror of
https://github.com/blender/blender-addons.git
synced 2025-07-25 16:05:20 +00:00
291 lines
9.5 KiB
Python
291 lines
9.5 KiB
Python
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
import bpy
|
|
from bpy.types import Operator, Menu
|
|
from bl_operators.presets import AddPresetBase
|
|
import os
|
|
|
|
from .sun_calc import (format_lat_long, format_time, format_hms, sun)
|
|
|
|
|
|
# -------------------------------------------------------------------
|
|
# Choice list of places, month and day at 12:00 noon
|
|
# -------------------------------------------------------------------
|
|
|
|
|
|
class SUNPOS_MT_Presets(Menu):
|
|
bl_label = "Sun Position Presets"
|
|
preset_subdir = "operator/sun_position"
|
|
preset_operator = "script.execute_preset"
|
|
draw = Menu.draw_preset
|
|
|
|
|
|
class SUNPOS_OT_AddPreset(AddPresetBase, Operator):
|
|
'''Add Sun Position preset'''
|
|
bl_idname = "world.sunpos_add_preset"
|
|
bl_label = "Add Sun Position preset"
|
|
preset_menu = "SUNPOS_MT_Presets"
|
|
|
|
# variable used for all preset values
|
|
preset_defines = [
|
|
"sun_props = bpy.context.scene.sun_pos_properties"
|
|
]
|
|
|
|
# properties to store in the preset
|
|
preset_values = [
|
|
"sun_props.day",
|
|
"sun_props.month",
|
|
"sun_props.time",
|
|
"sun_props.year",
|
|
"sun_props.UTC_zone",
|
|
"sun_props.use_daylight_savings",
|
|
"sun_props.latitude",
|
|
"sun_props.longitude",
|
|
]
|
|
|
|
# where to store the preset
|
|
preset_subdir = "operator/sun_position"
|
|
|
|
|
|
# -------------------------------------------------------------------
|
|
#
|
|
# Draw the Sun Panel, sliders, et. al.
|
|
#
|
|
# -------------------------------------------------------------------
|
|
|
|
class SUNPOS_PT_Panel(bpy.types.Panel):
|
|
bl_space_type = "PROPERTIES"
|
|
bl_region_type = "WINDOW"
|
|
bl_context = "world"
|
|
bl_label = "Sun Position"
|
|
bl_options = {'DEFAULT_CLOSED'}
|
|
|
|
def draw(self, context):
|
|
sp = context.scene.sun_pos_properties
|
|
p = context.preferences.addons[__package__].preferences
|
|
layout = self.layout
|
|
self.draw_panel(context, sp, p, layout)
|
|
|
|
def draw_panel(self, context, sp, p, layout):
|
|
col = self.layout.column(align=True)
|
|
col.label(text="Usage Mode")
|
|
row = col.row()
|
|
row.prop(sp, "usage_mode", expand=True)
|
|
col.separator()
|
|
if sp.usage_mode == "HDR":
|
|
self.draw_environ_mode_panel(context, sp, p, layout)
|
|
else:
|
|
self.draw_normal_mode_panel(context, sp, p, layout)
|
|
|
|
def draw_environ_mode_panel(self, context, sp, p, layout):
|
|
flow = layout.grid_flow(row_major=True, columns=0, even_columns=True,
|
|
even_rows=False, align=False)
|
|
|
|
col = flow.column(align=True)
|
|
col.label(text="Environment Texture")
|
|
|
|
if context.scene.world is not None:
|
|
if context.scene.world.node_tree is not None:
|
|
col.prop_search(sp, "hdr_texture",
|
|
context.scene.world.node_tree, "nodes", text="")
|
|
else:
|
|
col.label(text="Please activate Use Nodes in the World panel.",
|
|
icon="ERROR")
|
|
else:
|
|
col.label(text="Please select World in the World panel.",
|
|
icon="ERROR")
|
|
|
|
col.separator()
|
|
|
|
col = flow.column(align=True)
|
|
col.label(text="Sun Object")
|
|
col.prop_search(sp, "sun_object",
|
|
context.view_layer, "objects", text="")
|
|
col.separator()
|
|
|
|
col = flow.column(align=True)
|
|
col.prop(sp, "sun_distance")
|
|
if not sp.bind_to_sun:
|
|
col.prop(sp, "hdr_elevation")
|
|
col.prop(sp, "hdr_azimuth")
|
|
col.separator()
|
|
|
|
col = flow.column(align=True)
|
|
if sp.bind_to_sun:
|
|
col.prop(sp, "bind_to_sun", toggle=True, icon="CONSTRAINT",
|
|
text="Release binding")
|
|
else:
|
|
col.prop(sp, "bind_to_sun", toggle=True, icon="CONSTRAINT",
|
|
text="Bind Texture to Sun")
|
|
|
|
row = col.row(align=True)
|
|
row.enabled = not sp.bind_to_sun
|
|
row.operator("world.sunpos_show_hdr", icon='LIGHT_SUN')
|
|
|
|
def draw_normal_mode_panel(self, context, sp, p, layout):
|
|
if p.show_time_place:
|
|
row = layout.row(align=True)
|
|
row.menu(SUNPOS_MT_Presets.__name__, text=SUNPOS_MT_Presets.bl_label)
|
|
row.operator(SUNPOS_OT_AddPreset.bl_idname, text="", icon='ADD')
|
|
row.operator(SUNPOS_OT_AddPreset.bl_idname, text="", icon='REMOVE').remove_active = True
|
|
|
|
col = layout.column(align=True)
|
|
col.use_property_split = True
|
|
col.use_property_decorate = False
|
|
col.prop(sp, "sun_object")
|
|
col.separator()
|
|
|
|
col.prop(sp, "object_collection")
|
|
if sp.object_collection:
|
|
col.prop(sp, "object_collection_type")
|
|
if sp.object_collection_type == 'DIURNAL':
|
|
col.prop(sp, "time_spread")
|
|
col.separator()
|
|
|
|
if context.scene.world is not None:
|
|
if context.scene.world.node_tree is not None:
|
|
col.prop_search(sp, "sky_texture",
|
|
context.scene.world.node_tree, "nodes")
|
|
else:
|
|
col.label(text="Please activate Use Nodes in the World panel.",
|
|
icon="ERROR")
|
|
else:
|
|
col.label(text="Please select World in the World panel.",
|
|
icon="ERROR")
|
|
|
|
class SUNPOS_PT_Location(bpy.types.Panel):
|
|
bl_space_type = "PROPERTIES"
|
|
bl_region_type = "WINDOW"
|
|
bl_context = "world"
|
|
bl_label = "Location"
|
|
bl_parent_id = "SUNPOS_PT_Panel"
|
|
|
|
@classmethod
|
|
def poll(self, context):
|
|
sp = context.scene.sun_pos_properties
|
|
return sp.usage_mode != "HDR"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
sp = context.scene.sun_pos_properties
|
|
p = context.preferences.addons[__package__].preferences
|
|
|
|
col = layout.column(align=True)
|
|
col.label(text="Enter Coordinates")
|
|
col.prop(sp, "co_parser", text='', icon='URL')
|
|
|
|
layout.separator()
|
|
|
|
flow = layout.grid_flow(row_major=True, columns=0, even_columns=True, even_rows=False, align=False)
|
|
|
|
col = flow.column(align=True)
|
|
col.prop(sp, "latitude")
|
|
if p.show_dms:
|
|
row = col.row()
|
|
row.alignment = 'RIGHT'
|
|
row.label(text=format_lat_long(sp.latitude, True))
|
|
|
|
col = flow.column(align=True)
|
|
col.prop(sp, "longitude")
|
|
if p.show_dms:
|
|
row = col.row()
|
|
row.alignment = 'RIGHT'
|
|
row.label(text=format_lat_long(sp.longitude, False))
|
|
col.separator()
|
|
|
|
if p.show_north:
|
|
col = flow.column(align=True)
|
|
col.prop(sp, "show_north", toggle=True)
|
|
col.prop(sp, "north_offset")
|
|
col.separator()
|
|
|
|
if p.show_az_el:
|
|
col = flow.column(align=True)
|
|
split = col.split(factor=0.4, align=True)
|
|
split.label(text="Azimuth:")
|
|
split.label(text=str(round(sun.azimuth, 3)) + "°")
|
|
split = col.split(factor=0.4, align=True)
|
|
split.label(text="Elevation:")
|
|
split.label(text=str(round(sun.elevation, 3)) + "°")
|
|
col.separator()
|
|
|
|
if p.show_refraction:
|
|
col = flow.column()
|
|
col.prop(sp, "use_refraction")
|
|
col.separator()
|
|
|
|
col = flow.column()
|
|
col.prop(sp, "sun_distance")
|
|
|
|
|
|
class SUNPOS_PT_Time(bpy.types.Panel):
|
|
bl_space_type = "PROPERTIES"
|
|
bl_region_type = "WINDOW"
|
|
bl_context = "world"
|
|
bl_label = "Time"
|
|
bl_parent_id = "SUNPOS_PT_Panel"
|
|
|
|
@classmethod
|
|
def poll(self, context):
|
|
sp = context.scene.sun_pos_properties
|
|
return sp.usage_mode != "HDR"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
sp = context.scene.sun_pos_properties
|
|
p = context.preferences.addons[__package__].preferences
|
|
|
|
flow = layout.grid_flow(row_major=True, columns=0, even_columns=True, even_rows=False, align=False)
|
|
|
|
col = flow.column(align=True)
|
|
col.prop(sp, "use_day_of_year",
|
|
icon='SORTTIME')
|
|
if sp.use_day_of_year:
|
|
col.prop(sp, "day_of_year")
|
|
else:
|
|
col.prop(sp, "day")
|
|
col.prop(sp, "month")
|
|
col.prop(sp, "year")
|
|
col.separator()
|
|
|
|
col = flow.column(align=True)
|
|
col.prop(sp, "time")
|
|
col.prop(sp, "UTC_zone")
|
|
if p.show_daylight_savings:
|
|
col.prop(sp, "use_daylight_savings")
|
|
col.separator()
|
|
|
|
col = flow.column(align=True)
|
|
lt = format_time(sp.time,
|
|
p.show_daylight_savings and sp.use_daylight_savings,
|
|
sp.longitude)
|
|
ut = format_time(sp.time,
|
|
p.show_daylight_savings and sp.use_daylight_savings,
|
|
sp.longitude,
|
|
sp.UTC_zone)
|
|
col.alignment = 'CENTER'
|
|
|
|
split = col.split(factor=0.5, align=True)
|
|
split.label(text="Local:", icon='TIME')
|
|
split.label(text=lt)
|
|
split = col.split(factor=0.5, align=True)
|
|
split.label(text="UTC:", icon='PREVIEW_RANGE')
|
|
split.label(text=ut)
|
|
col.separator()
|
|
|
|
|
|
col = flow.column(align=True)
|
|
col.alignment = 'CENTER'
|
|
if p.show_rise_set:
|
|
sr = format_hms(sun.sunrise.time)
|
|
ss = format_hms(sun.sunset.time)
|
|
|
|
split = col.split(factor=0.5, align=True)
|
|
split.label(text="Sunrise:", icon='LIGHT_SUN')
|
|
split.label(text=sr)
|
|
split = col.split(factor=0.5, align=True)
|
|
split.label(text="Sunset:", icon='SOLO_ON')
|
|
split.label(text=ss)
|
|
|
|
col.separator()
|