Merge pull request #156 from MaximeHerpin/develop

replaced icons

__init__.py

@@ -41,8 +41,8 @@
 bl_info = {
     "name": "Modular trees",
     "author": "Herpin Maxime, Jake Dube",
-    "version": (2, 9, 0),
-    "blender": (2, 77, 0),
+    "version": (2, 9, 1),
+    "blender": (2, 79, 0),
     "location": "View3D > Tools > Tree > Make Tree",
     "description": "Generates an organic tree with correctly modeled branching.",
     "warning": "May take a long time to generate! Save your file before generating!",

icons.py

@@ -0,0 +1,46 @@
+# Copyright 2016 Maxime Herpin, Jake Dube
+#
+# ##### BEGIN GPL LICENSE BLOCK ######
+# This file is part of Modular Tree.
+#
+# Modular Tree 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 3 of the License, or
+# (at your option) any later version.
+#
+# Modular Tree 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 Modular Tree.  If not, see <http://www.gnu.org/licenses/>.
+# ##### END GPL LICENSE BLOCK #####
+
+from os.path import join, dirname
+from bpy.utils import previews
+
+# global variables
+ICONS = {'TREE', 'TREE_UPDATE', 'BATCH_TREE', 'TWIG'}
+ICON_COLLECTION = {}
+
+
+def register_icons():
+    preview_coll = previews.new()
+    icons_folder = join(dirname(__file__), "custom_icons")
+
+    for icon in ICONS:
+        preview_coll.load(icon, join(icons_folder, icon + ".png"), 'IMAGE')
+
+    ICON_COLLECTION["main"] = preview_coll
+
+
+def unregister_icons():
+    for collection in ICON_COLLECTION.values():
+        previews.remove(collection)
+    ICON_COLLECTION.clear()
+
+
+def get_icon(icon):
+    preview_coll = ICON_COLLECTION["main"]
+    return preview_coll[icon].icon_id