# -*- coding: utf-8 -*-
#
# ***** BEGIN GPL LICENSE BLOCK *****
#
# --------------------------------------------------------------------------
# Blender 2.5 Geographical Sun Add-On
# --------------------------------------------------------------------------
#
# Authors:
# Doug Hammond
#
# 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, see <http://www.gnu.org/licenses/>.
#
# ***** END GPL LICENCE BLOCK *****
#

# System imports
#-----------------------------------------------------------------------------
import datetime, math, time
today = datetime.datetime.now()

# Blender imports
#----------------------------------------------------------------------------- 
import bpy
from extensions_framework import Addon, declarative_property_group
from extensions_framework.ui import property_group_renderer

# Addon setup
#----------------------------------------------------------------------------- 
bl_info = {
    "name": "Geographical Sun",
    "author": "Doug Hammond (dougal2)",
    "version": (0, 0, 1),
    "blender": (2, 56, 0),
    "category": "Object",
    "location": "Lamp data > Geographical Sun",
    "warning": "",
    "wiki_url": "",
    "tracker_url": "https://developer.blender.org/maniphest/task/edit/form/2/",
    "description": "Set SUN Lamp rotation according to geographical time and location."
}
GeoSunAddon = Addon(bl_info)

# Sun rotation calculator implementation
#----------------------------------------------------------------------------- 
class sun_calculator(object):
    """
    Based on SunLight v1.0 by Miguel Kabantsov (miguelkab@gmail.com)
    Replaces the faulty sun position calculation algorythm with a precise calculation (Source for algorythm: http://de.wikipedia.org/wiki/Sonnenstand),
    Co-Ordinates: http://www.bcca.org/misc/qiblih/latlong.html
    Author: Nils-Peter Fischer (Nils-Peter.Fischer@web.de)
    """
    
    location_list = [
        ("EUROPE",[
            ("Antwerp, Belgium",            67),
            ("Berlin, Germany",             1),
            ("Bratislava, Slovak Republic", 70),
            ("Brno, Czech Republic",        72),
            ("Brussles, Belgium",           68),
            ("Geneva, Switzerland",         65),
            ("Helsinki, Finland",           7),
            ("Innsbruck, Austria",          62),
            ("Kyiv, Ukraine",               64),
            ("London, England",             10),
            ("Lyon, France",                66),
            ("Nitra, Slovak Republic",      69),
            ("Oslo, Norway",                58),
            ("Paris, France",               15),
            ("Praha, Czech Republic",       71),
            ("Rome, Italy",                 18),
            ("Telfs, Austria",              63),
            ("Warsaw, Poland",              74),
            ("Wroclaw, Poland",             73),
            ("Zurich, Switzerland",         21),
        ]),
        
        ("WORLD CITIES", [
            ("Beijing, China",              0),
            ("Bombay, India",               2),
            ("Buenos Aires, Argentina",     3),
            ("Cairo, Egypt",                4),
            ("Cape Town, South Africa",     5),
            ("Caracas, Venezuela",          6),
            ("Curitiba, Brazil",            60),
            ("Hong Kong, China",            8),
            ("Jerusalem, Israel",           9),
            ("Joinville, Brazil",           61),
            ("Mexico City, Mexico",         11),
            ("Moscow, Russia",              12),
            ("New Delhi, India",            13),
            ("Ottawa, Canada",              14),
            ("Rio de Janeiro, Brazil",      16),
            ("Riyadh, Saudi Arabia",        17),
            ("Sao Paulo, Brazil",           59),
            ("Sydney, Australia",           19),
            ("Tokyo, Japan",                20),
        ]),
        
        ("US CITIES", [
            ("Albuquerque, NM",             22),
            ("Anchorage, AK",               23),
            ("Atlanta, GA",                 24),
            ("Austin, TX",                  25),
            ("Birmingham, AL",              26),
            ("Bismarck, ND",                27),
            ("Boston, MA",                  28),
            ("Boulder, CO",                 29),
            ("Chicago, IL",                 30),
            ("Dallas, TX",                  31),
            ("Denver, CO",                  32),
            ("Detroit, MI",                 33),
            ("Honolulu, HI",                34),
            ("Houston, TX",                 35),
            ("Indianapolis, IN",            36),
            ("Jackson, MS",                 37),
            ("Kansas City, MO",             38),
            ("Los Angeles, CA",             39),
            ("Menomonee Falls, WI",         40),
            ("Miami, FL",                   41),
            ("Minneapolis, MN",             42),
            ("New Orleans, LA",             43),
            ("New York City, NY",           44),
            ("Oklahoma City, OK",           45),
            ("Philadelphia, PA",            46),
            ("Phoenix, AZ",                 47),
            ("Pittsburgh, PA",              48),
            ("Portland, ME",                49),
            ("Portland, OR",                50),
            ("Raleigh, NC",                 51),
            ("Richmond, VA",                52),
            ("Saint Louis, MO",             53),
            ("San Diego, CA",               54),
            ("San Francisco, CA",           55),
            ("Seattle, WA",                 56),
            ("Washington DC",               57),
        ])
    ]
    
    location_data = {
        # Europe
        67: ( 51.2167, -4.4, 1),
        1:  ( 52.33, -13.30, 1),
        70: ( 48.17, -17.17, 1),
        72: ( 49.2, -16.63, 1),
        68: ( 58.8467, -4.3525, 1),
        65: ( 46.217, -6.150, 1),
        7:  ( 60.1667, -24.9667,2),
        62: ( 47.2672, -11.3928, 1),
        64: ( 50.75, -30.0833, 2),
        10: ( 51.50, 0.0, 0),
        66: ( 45.767, -4.833, 1),
        69: ( 48.32, -18.07, 1),
        58: ( 59.56, -10.41, 1),
        15: ( 48.8667, -2.667, 1),
        71: ( 50.08, -14.46, 1),
        18: ( 41.90, -12.4833, 1),
        63: ( 47.3, -11.0667, 1),
        74: ( 52.232, -21.008, 1),
        73: ( 51.108, -17.038, 1),
        21: ( 47.3833, -8.5333, 1),
        
        # World Cities
        0:  ( 39.9167, -116.4167, 8),
        2:  ( 18.9333, -72.8333, 5.5),
        3:  (-34.60, 58.45, -3),
        4:  ( 30.10, -31.3667, 2),
        5:  (-33.9167, -18.3667, 2),
        6:  ( 10.50, 66.9333, -4),
        60: (-25.4278, 49.2731, -3),
        8:  ( 22.25, -114.1667, 8),
        9:  ( 31.7833, -35.2333, 2),
        61: (-29.3044, 48.8456, -3),
        11: ( 19.4, 99.15, -6),
        12: ( 55.75, -37.5833, 3),
        13: ( 28.6, -77.2, 5.5),
        14: ( 45.41667, 75.7, -5),
        16: (-22.90, 43.2333, -3),
        17: ( 24.633, -46.71667, 3),
        59: ( -23.5475, 46.6361, -3),
        19: (-33.8667, -151.2167,10),
        20: ( 35.70, -139.7667, 9), 
        
        # US Cities
        22: ( 35.0833, 106.65, -7),
        23: ( 61.217, 149.90, -9),
        24: ( 33.733, 84.383, -5),
        25: ( 30.283, 97.733, -6),
        26: ( 33.521, 86.8025, -6),
        27: ( 46.817, 100.783, -6),
        28: ( 42.35, 71.05, -5),
        29: ( 40.125, 105.237, -7),
        30: ( 41.85, 87.65, -6),
        31: ( 32.46, 96.47, -6),
        32: ( 39.733, 104.983, -7),
        33: ( 42.333, 83.05, -5),
        34: ( 21.30, 157.85, -10),
        35: ( 29.75, 95.35, -6),
        36: ( 39.767, 86.15, -5),
        37: ( 32.283, 90.183, -6),
        38: ( 39.083, 94.567, -6),
        39: ( 34.05, 118.233, -8),
        40: ( 43.11, 88.10, -6),
        41: ( 25.767, 80.183, -5),
        42: ( 44.967, 93.25, -6),
        43: ( 29.95, 90.067, -6),
        44: ( 40.7167, 74.0167, -5),
        45: ( 35.483, 97.533, -6),
        46: ( 39.95, 75.15, -5),
        47: ( 33.433, 112.067,-7),
        48: ( 40.433, 79.9833, -5),
        49: ( 43.666, 70.283, -5),
        50: ( 45.517, 122.65, -8),
        51: ( 35.783, 78.65, -5),
        52: ( 37.5667, 77.450, -5),
        53: ( 38.6167, 90.1833, -6),
        54: ( 32.7667, 117.2167, -8),
        55: ( 37.7667, 122.4167, -8),
        56: ( 47.60, 122.3167, -8),
        57: ( 38.8833, 77.0333, -5),
    }
    
    # mathematical helpers
    @staticmethod
    def sind(deg):
        return math.sin(math.radians(deg))
    
    @staticmethod
    def cosd(deg):
        return math.cos(math.radians(deg))
    
    @staticmethod
    def tand(deg):
        return math.tan(math.radians(deg))
    
    @staticmethod
    def asind(deg):
        return math.degrees(math.asin(deg))
    
    @staticmethod
    def atand(deg):
        return math.degrees(math.atan(deg))
    
    @staticmethod
    def geo_sun_astronomicJulianDate(Year, Month, Day, LocalTime, Timezone):
        if Month > 2.0:
            Y = Year
            M = Month
        else:
            Y = Year - 1.0
            M = Month + 12.0
            
        UT = LocalTime - Timezone
        hour = UT / 24.0
        A = int(Y/100.0)
        
        JD = math.floor(365.25*(Y+4716.0)) + math.floor(30.6001*(M+1.0)) + Day + hour - 1524.4
        
        # The following section is adopted from netCDF4 netcdftime implementation.
        # Copyright: 2008 by Jeffrey Whitaker
        # License: http://www.opensource.org/licenses/mit-license.php
        if JD >= 2299170.5:
            # 1582 October 15 (Gregorian Calendar)
            B = 2.0 - A + int(A/4.0)
        elif JD < 2299160.5:
            # 1582 October 5 (Julian Calendar)
            B = 0
        else:
            raise Exception('ERROR: Date falls in the gap between Julian and Gregorian calendars.')
            B = 0
        
        return JD+B
    
    @staticmethod
    def geoSunData(Latitude, Longitude, Year, Month, Day, LocalTime, Timezone):
        JD = sun_calculator.geo_sun_astronomicJulianDate(Year, Month, Day, LocalTime, Timezone)
        
        phi = Latitude
        llambda = Longitude
                
        n = JD - 2451545.0
        LDeg = (280.460 + 0.9856474*n) - (math.floor((280.460 + 0.9856474*n)/360.0) * 360.0)
        gDeg = (357.528 + 0.9856003*n) - (math.floor((357.528 + 0.9856003*n)/360.0) * 360.0)
        LambdaDeg = LDeg + 1.915 * sun_calculator.sind(gDeg) + 0.02 * sun_calculator.sind(2.0*gDeg)
        
        epsilonDeg = 23.439 - 0.0000004*n
        
        alphaDeg = sun_calculator.atand( (sun_calculator.cosd(epsilonDeg) * sun_calculator.sind(LambdaDeg)) / sun_calculator.cosd(LambdaDeg) )
        if sun_calculator.cosd(LambdaDeg) < 0.0:
            alphaDeg += 180.0
            
        deltaDeg = sun_calculator.asind( sun_calculator.sind(epsilonDeg) * sun_calculator.sind(LambdaDeg) )
        
        JDNull = sun_calculator.geo_sun_astronomicJulianDate(Year, Month, Day, 0.0, 0.0)
        
        TNull = (JDNull - 2451545.0) / 36525.0
        T = LocalTime - Timezone
        
        thetaGh = 6.697376 + 2400.05134*TNull + 1.002738*T
        thetaGh -= math.floor(thetaGh/24.0) * 24.0
        
        thetaG = thetaGh * 15.0
        theta = thetaG + llambda
        
        tau = theta - alphaDeg
        
        a = sun_calculator.atand( sun_calculator.sind(tau) / ( sun_calculator.cosd(tau)*sun_calculator.sind(phi) - sun_calculator.tand(deltaDeg)*sun_calculator.cosd(phi)) )
        if sun_calculator.cosd(tau)*sun_calculator.sind(phi) - sun_calculator.tand(deltaDeg)*sun_calculator.cosd(phi) < 0.0:
            a += 180.0
        
        h = sun_calculator.asind( sun_calculator.cosd(deltaDeg)*sun_calculator.cosd(tau)*sun_calculator.cosd(phi) + sun_calculator.sind(deltaDeg)*sun_calculator.sind(phi) )
        
        R = 1.02 / (sun_calculator.tand (h+(10.3/(h+5.11))))
        hR = h + R/60.0
        
        azimuth = a
        elevation = hR
        
        return azimuth, elevation

# Addon classes
#----------------------------------------------------------------------------- 
@GeoSunAddon.addon_register_class
class OBJECT_OT_set_geographical_sun_now(bpy.types.Operator):
    bl_idname = 'object.set_geographical_sun_now'
    bl_label = 'Set time to NOW'
    
    @classmethod
    def poll(cls, context):
        cl = context.lamp
        return cl and cl.type == 'SUN'
    
    def execute(self, context):
        GSP = context.lamp.GeoSunProperties
        
        now = datetime.datetime.now()
        for p in ("hour", "minute", "day", "month", "year"):
            setattr(
                GSP,
                p,
                getattr(now, p)
            )
        GSP.tz = time.timezone
        GSP.dst = False
        
        return {'FINISHED'}

@GeoSunAddon.addon_register_class
class OBJECT_OT_set_geographical_sun_pos(bpy.types.Operator):
    bl_idname = 'object.set_geographical_sun_pos'
    bl_label = 'Set SUN position'
    
    @classmethod
    def poll(cls, context):
        cl = context.lamp
        return cl and cl.type == 'SUN'
    
    def execute(self, context):
        try:
            GSP = context.lamp.GeoSunProperties
            
            dst = 1 if GSP.dst else 0
            
            az,el = sun_calculator.geoSunData(
                GSP.lat,
                GSP.long,
                GSP.year,
                GSP.month,
                GSP.day,
                GSP.hour + GSP.minute/60.0,
                -GSP.tz + dst
            )
            
            context.object.rotation_euler = ( math.radians(90-el), 0, math.radians(-az) )
            return {'FINISHED'}
        except Exception as err:
            self.report({'ERROR'}, str(err))
            return {'CANCELLED'}

@GeoSunAddon.addon_register_class
class OBJECT_OT_set_geographical_location_preset(bpy.types.Operator):
    bl_idname = 'object.set_geographical_location_preset'
    bl_label = 'Apply location preset'
    
    index = bpy.props.IntProperty()
    
    @classmethod
    def poll(cls, context):
        cl = context.lamp
        return cl and cl.type == 'SUN'
    
    def execute(self, context):
        GSP = context.lamp.GeoSunProperties
        GSP.lat, GSP.long, GSP.tz = sun_calculator.location_data[self.properties.index]
        return {'FINISHED'}

# Dynamic submenu magic !

def draw_generator(locations):
    def draw(self, context):
        sl = self.layout
        for location in locations:
            location_name, location_index = location
            sl.operator('OBJECT_OT_set_geographical_location_preset', text=location_name).index = location_index
    return draw

submenus = []
for label, locations in sun_calculator.location_list:
    submenu_idname = 'OBJECT_MT_geo_sun_location_cat%d'%len(submenus)
    submenu = type(
        submenu_idname,
        (bpy.types.Menu,),
        {
            'bl_idname': submenu_idname,
            'bl_label': label,
            'draw': draw_generator(locations)
        }
    )
    GeoSunAddon.addon_register_class(submenu)
    submenus.append(submenu)

@GeoSunAddon.addon_register_class
class OBJECT_MT_geo_sun_location(bpy.types.Menu):
    bl_label = 'Location preset'
    
    def draw(self, context):
        sl = self.layout
        for sm in submenus:
            sl.menu(sm.bl_idname)

@GeoSunAddon.addon_register_class
class GeoSunProperties(declarative_property_group):
    ef_attach_to = ['Lamp']
    
    controls = [
        ['hour', 'minute'],
        ['day', 'month', 'year'],
        ['tz', 'dst'],
        'location_menu',
        ['lat', 'long'],
        ['set_time', 'set_posn'],
    ]
    
    properties = [
        {
            'type': 'int',
            'attr': 'minute',
            'name': 'Minute',
            'min': 0,
            'soft_min': 0,
            'max': 59,
            'soft_max': 59,
            'default': today.minute
        },
        {
            'type': 'int',
            'attr': 'hour',
            'name': 'Hour',
            'min': 0,
            'soft_min': 0,
            'max': 24,
            'soft_max': 24,
            'default': today.hour
        },
        {
            'type': 'int',
            'attr': 'day',
            'name': 'Day',
            'min': 1,
            'soft_min': 1,
            'max': 31,
            'soft_max': 31,
            'default': today.day
        },
        {
            'type': 'int',
            'attr': 'month',
            'name': 'Month',
            'min': 1,
            'soft_min': 1,
            'max': 12,
            'soft_max': 12,
            'default': today.month
        },
        {
            'type': 'int',
            'attr': 'year',
            'name': 'Year',
            'min': datetime.MINYEAR,
            'soft_min': datetime.MINYEAR,
            'max': datetime.MAXYEAR,
            'soft_max': datetime.MAXYEAR,
            'default': today.year
        },
        {
            'type': 'int',
            'attr': 'tz',
            'name': 'Time zone',
            'min': -13,
            'soft_min': -13,
            'max': 13,
            'soft_max': 13,
            'default': time.timezone
        },
        {
            'type': 'bool',
            'attr': 'dst',
            'name': 'DST',
            'default': False
        },
        {
            'type': 'float',
            'attr': 'lat',
            'name': 'Lat.',
            'min': -180.0,
            'soft_min': -180.0,
            'max': 180.0,
            'soft_max': 180.0,
            'default': 0.0
        },
        {
            'type': 'float',
            'attr': 'long',
            'name': 'Long.',
            'min': -90.0,
            'soft_min': -90.0,
            'max': 90.0,
            'soft_max': 90.0,
            'default': 0.0
        },
        
        # draw operators and menus
        {
            'attr': 'location_menu',
            'type': 'menu',
            'menu': 'OBJECT_MT_geo_sun_location'
        },
        {
            'attr': 'set_time',
            'type': 'operator',
            'operator': 'object.set_geographical_sun_now',
            'icon': 'PREVIEW_RANGE'
        },
        {
            'attr': 'set_posn',
            'type': 'operator',
            'operator': 'object.set_geographical_sun_pos',
            'icon': 'WORLD_DATA'
        },
    ]

@GeoSunAddon.addon_register_class
class GeoSunPanel(property_group_renderer):
    bl_space_type = 'PROPERTIES'
    bl_region_type = 'WINDOW'
    bl_context = 'data'
    bl_label = 'Geographical Sun'
    
    display_property_groups = [
        ( ('lamp',), 'GeoSunProperties' )
    ]
    
    @classmethod
    def poll(cls, context):
        cl = context.lamp
        return cl and cl.type == 'SUN'

# Bootstrap the Addon
#----------------------------------------------------------------------------- 
register, unregister = GeoSunAddon.init_functions()
