__init__.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, see <http://www.gnu.org/licenses/>
#  and write to the Free Software Foundation, Inc., 51 Franklin Street,
#  Fifth Floor, Boston, MA  02110-1301, USA..
#
#  The Original Code is Copyright (C) 2013-2014 by Gorodetskiy Nikita  ###
#  All rights reserved.
#
#  Contact:      sverchok-b3d@ya.ru    ###
#  Information:  http://nikitron.cc.ua/sverchok_en.html   ###
#
#  The Original Code is: all of this file.
#
#  Contributor(s):
#     Nedovizin Alexander (aka Cfyzzz)
#     Gorodetskiy Nikita (aka Nikitron)
#     Linus Yng (aka Ly29)
#     Agustin Jimenez (aka AgustinJB)
#     Dealga McArdle (aka Zeffii)
#     Konstantin Vorobiew (aka Kosvor)
#     Ilya Portnov (aka portnov)
#     Eleanor Howick (aka elfnor)
#
#  ***** END GPL LICENSE BLOCK *****
#
# -*- coding: utf-8 -*-

bl_info = {
    "name": "Sverchok",
    "author": (
        "sverchok-b3d@ya.ru, "
        "Cfyzzz, Nikitron, Ly29, "
        "AgustinJB, Zeffii, Kosvor, "
        "Portnov, Elfnor"
    ),
    "version": (0, 5, 4, 51),
    "blender": (2, 7, 2),
    "location": "Nodes > CustomNodesTree > Add user nodes",
    "description": "Parametric node-based geometry programming",
    "warning": "",
    "wiki_url": "http://nikitron.cc.ua/sverch/html/main.html",
    "tracker_url": (
        "http://www.blenderartists.org/forum/showthread.php?272679"
        "-Addon-WIP-Sverchok-parametric-tool-for-architects"),
    "category": "Node"}

import sys
import importlib


# monkey patch the sverchok name, I am sure there is a better way to do this.

if __name__ != "sverchok":
    sys.modules["sverchok"] = sys.modules[__name__]

# to store imported modules
imported_modules = []

# ugly hack, should make respective dict in __init__ like nodes
# or parse it
root_modules = ["menu", "node_tree", "data_structure", "core",
                "utils", "ui", "nodes", "old_nodes"]
core_modules = ["handlers", "update_system", "upgrade_nodes"]
utils_modules = [
    # non UI tools
    "cad_module", "sv_bmesh_utils", "sv_viewer_utils", "sv_curve_utils",
    "voronoi", "sv_script", "sv_itertools", "script_importhelper",
    "csg_core", "csg_geom", "sv_easing_functions",
    # UI text editor ui
    "text_editor_submenu", "text_editor_plugins",
    # UI operators
    "sv_panels_tools", "sv_IO_panel_tools", "group_tools",
]
ui_modules = [
    "color_def", "sv_IO_panel", "sv_templates_menu", "sv_panels", "nodeview_space_menu",
    # bgl modules
    "viewer_draw", "viewer_draw_mk2", "nodeview_bgl_viewer_draw",
    "index_viewer_draw"
]

# modules and pkg path, nodes are done separately.
mods_bases = [(root_modules, "sverchok"),
              (core_modules, "sverchok.core"),
              (utils_modules, "sverchok.utils"),
              (ui_modules, "sverchok.ui")]

#  settings have to be treated separately incase the folder name
#  is something else than sverchok...
settings = importlib.import_module(".settings", __name__)
imported_modules.append(settings)


def import_modules(modules, base, im_list):
    for m in modules:
        im = importlib.import_module('.{}'.format(m), base)
        im_list.append(im)


# parse the nodes/__init__.py dictionary and load all nodes
def make_node_list():
    node_list = []
    base_name = "sverchok.nodes"
    for category, names in nodes.nodes_dict.items():
        importlib.import_module('.{}'.format(category), base_name)
        import_modules(names, '{}.{}'.format(base_name, category), node_list)
    return node_list

for mods, base in mods_bases:
    import_modules(mods, base, imported_modules)

node_list = make_node_list()

reload_event = bool("bpy" in locals())

if reload_event:
    import nodeitems_utils
    #  reload the base modules
    #  then reload nodes after the node module as been reloaded
    for im in imported_modules:
        importlib.reload(im)
    node_list = make_node_list()
    for node in node_list:
        importlib.reload(node)
    old_nodes.reload_old()
    menu.reload_menu()

import bpy
from sverchok.utils import ascii_print


def register():
    for m in imported_modules + node_list:
        if hasattr(m, "register"):
            m.register()
    # this is used to access preferences, should/could be hidden
    # in an interface
    data_structure.SVERCHOK_NAME = __name__
    print("** version: ", bl_info['version']," **")
    print("** Have a nice day with sverchok  **\n")
    ascii_print.logo()

    if reload_event:
        data_structure.RELOAD_EVENT = True
        print("Sverchok is reloaded, press update")


def unregister():
    for m in reversed(imported_modules + node_list):
        if hasattr(m, "unregister"):
            m.unregister()