archipack/__init__.py

   1 # -*- coding:utf-8 -*-
   2
   3 # ##### BEGIN GPL LICENSE BLOCK #####
   4 #
   5 #  This program is free software; you can redistribute it and/or
   6 #  modify it under the terms of the GNU General Public License
   7 #  as published by the Free Software Foundation; either version 2
   8 #  of the License, or (at your option) any later version.
   9 #
  10 #  This program is distributed in the hope that it will be useful,
  11 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 #  GNU General Public License for more details.
  14 #
  15 #  You should have received a copy of the GNU General Public License
  16 #  along with this program; if not, write to the Free Software Foundation,
  17 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18 #
  19 # ##### END GPL LICENSE BLOCK #####
  20
  21 # <pep8 compliant>
  22
  23 # ----------------------------------------------------------
  24 # Author: Stephen Leger (s-leger)
  25 #
  26 # ----------------------------------------------------------
  27
  28 bl_info = {
  29     'name': 'Archipack',
  30     'description': 'Architectural objects',
  31     'author': 's-leger',
  32     'license': 'GPL',
  33     'deps': '',
  34     'version': (1, 2, 8),
  35     'blender': (2, 7, 8),
  36     'location': 'View3D > Tools > Create > Archipack',
  37     'warning': '',
  38     'wiki_url': 'https://github.com/s-leger/archipack/wiki',
  39     'tracker_url': 'https://github.com/s-leger/archipack/issues',
  40     'link': 'https://github.com/s-leger/archipack',
  41     'support': 'COMMUNITY',
  42     'category': 'Add Mesh'
  43     }
  44
  45 import os
  46
  47 if "bpy" in locals():
  48     import importlib as imp
  49     imp.reload(archipack_progressbar)
  50     imp.reload(archipack_material)
  51     imp.reload(archipack_snap)
  52     imp.reload(archipack_manipulator)
  53     imp.reload(archipack_reference_point)
  54     imp.reload(archipack_autoboolean)
  55     imp.reload(archipack_door)
  56     imp.reload(archipack_window)
  57     imp.reload(archipack_stair)
  58     imp.reload(archipack_wall2)
  59     imp.reload(archipack_roof)
  60     imp.reload(archipack_slab)
  61     imp.reload(archipack_fence)
  62     imp.reload(archipack_truss)
  63     imp.reload(archipack_floor)
  64     imp.reload(archipack_rendering)
  65
  66     # print("archipack: reload ready")
  67 else:
  68     from . import archipack_progressbar
  69     from . import archipack_material
  70     from . import archipack_snap
  71     from . import archipack_manipulator
  72     from . import archipack_reference_point
  73     from . import archipack_autoboolean
  74     from . import archipack_door
  75     from . import archipack_window
  76     from . import archipack_stair
  77     from . import archipack_wall2
  78     from . import archipack_roof
  79     from . import archipack_slab
  80     from . import archipack_fence
  81     from . import archipack_truss
  82     from . import archipack_floor
  83     from . import archipack_rendering
  84
  85     # print("archipack: ready")
  86
  87 # noinspection PyUnresolvedReferences
  88 import bpy
  89 # noinspection PyUnresolvedReferences
  90 from bpy.types import (
  91     Panel, WindowManager, PropertyGroup,
  92     AddonPreferences, Menu
  93     )
  94 from bpy.props import (
  95     EnumProperty, PointerProperty,
  96     StringProperty, BoolProperty,
  97     IntProperty, FloatProperty, FloatVectorProperty
  98     )
  99
 100 from bpy.utils import previews
 101 icons_collection = {}
 102
 103
 104 # ----------------------------------------------------
 105 # Addon preferences
 106 # ----------------------------------------------------
 107
 108 def update_panel(self, context):
 109     try:
 110         bpy.utils.unregister_class(TOOLS_PT_Archipack_Tools)
 111         bpy.utils.unregister_class(TOOLS_PT_Archipack_Create)
 112     except:
 113         pass
 114     prefs = context.user_preferences.addons[__name__].preferences
 115     TOOLS_PT_Archipack_Tools.bl_category = prefs.tools_category
 116     bpy.utils.register_class(TOOLS_PT_Archipack_Tools)
 117     TOOLS_PT_Archipack_Create.bl_category = prefs.create_category
 118     bpy.utils.register_class(TOOLS_PT_Archipack_Create)
 119
 120
 121 class Archipack_Pref(AddonPreferences):
 122     bl_idname = __name__
 123
 124     tools_category = StringProperty(
 125         name="Tools",
 126         description="Choose a name for the category of the Tools panel",
 127         default="Tools",
 128         update=update_panel
 129     )
 130     create_category = StringProperty(
 131         name="Create",
 132         description="Choose a name for the category of the Create panel",
 133         default="Create",
 134         update=update_panel
 135     )
 136     create_submenu = BoolProperty(
 137         name="Use Sub-menu",
 138         description="Put Achipack's object into a sub menu (shift+a)",
 139         default=True
 140     )
 141     max_style_draw_tool = BoolProperty(
 142         name="Draw a wall use 3dsmax style",
 143         description="Reverse clic / release & drag cycle for Draw a wall",
 144         default=True
 145     )
 146     # Arrow sizes (world units)
 147     arrow_size = FloatProperty(
 148             name="Arrow",
 149             description="Manipulators arrow size (blender units)",
 150             default=0.05
 151             )
 152     # Handle area size (pixels)
 153     handle_size = IntProperty(
 154             name="Handle",
 155             description="Manipulators handle sensitive area size (pixels)",
 156             min=2,
 157             default=10
 158             )
 159     matlib_path = StringProperty(
 160             name="Folder path",
 161             description="absolute path to material library folder",
 162             default=""
 163             )
 164     # Font sizes and basic colour scheme
 165     feedback_size_main = IntProperty(
 166             name="Main",
 167             description="Main title font size (pixels)",
 168             min=2,
 169             default=16
 170             )
 171     feedback_size_title = IntProperty(
 172             name="Title",
 173             description="Tool name font size (pixels)",
 174             min=2,
 175             default=14
 176             )
 177     feedback_size_shortcut = IntProperty(
 178             name="Shortcut",
 179             description="Shortcuts font size (pixels)",
 180             min=2,
 181             default=11
 182             )
 183     feedback_shortcut_area = FloatVectorProperty(
 184             name="Background Shortcut",
 185             description="Shortcut area background color",
 186             subtype='COLOR_GAMMA',
 187             default=(0, 0.4, 0.6, 0.2),
 188             size=4,
 189             min=0, max=1
 190             )
 191     feedback_title_area = FloatVectorProperty(
 192             name="Background Main",
 193             description="Title area background color",
 194             subtype='COLOR_GAMMA',
 195             default=(0, 0.4, 0.6, 0.5),
 196             size=4,
 197             min=0, max=1
 198             )
 199     feedback_colour_main = FloatVectorProperty(
 200             name="Font Main",
 201             description="Title color",
 202             subtype='COLOR_GAMMA',
 203             default=(0.95, 0.95, 0.95, 1.0),
 204             size=4,
 205             min=0, max=1
 206             )
 207     feedback_colour_key = FloatVectorProperty(
 208             name="Font Shortcut key",
 209             description="KEY label color",
 210             subtype='COLOR_GAMMA',
 211             default=(0.67, 0.67, 0.67, 1.0),
 212             size=4,
 213             min=0, max=1
 214             )
 215     feedback_colour_shortcut = FloatVectorProperty(
 216             name="Font Shortcut hint",
 217             description="Shortcuts text color",
 218             subtype='COLOR_GAMMA',
 219             default=(0.51, 0.51, 0.51, 1.0),
 220             size=4,
 221             min=0, max=1
 222             )
 223
 224     def draw(self, context):
 225         layout = self.layout
 226         box = layout.box()
 227         row = box.row()
 228         col = row.column()
 229         col.label(text="Tab Category:")
 230         col.prop(self, "tools_category")
 231         col.prop(self, "create_category")
 232         col.prop(self, "create_submenu")
 233         box = layout.box()
 234         box.label("Features")
 235         box.prop(self, "max_style_draw_tool")
 236         box = layout.box()
 237         row = box.row()
 238         col = row.column()
 239         col.label(text="Material library:")
 240         col.prop(self, "matlib_path")
 241         box = layout.box()
 242         row = box.row()
 243         split = row.split(percentage=0.5)
 244         col = split.column()
 245         col.label(text="Colors:")
 246         row = col.row(align=True)
 247         row.prop(self, "feedback_title_area")
 248         row = col.row(align=True)
 249         row.prop(self, "feedback_shortcut_area")
 250         row = col.row(align=True)
 251         row.prop(self, "feedback_colour_main")
 252         row = col.row(align=True)
 253         row.prop(self, "feedback_colour_key")
 254         row = col.row(align=True)
 255         row.prop(self, "feedback_colour_shortcut")
 256         col = split.column()
 257         col.label(text="Font size:")
 258         col.prop(self, "feedback_size_main")
 259         col.prop(self, "feedback_size_title")
 260         col.prop(self, "feedback_size_shortcut")
 261         col.label(text="Manipulators:")
 262         col.prop(self, "arrow_size")
 263         col.prop(self, "handle_size")
 264         # layout.operator("archipack.render_thumbs")
 265
 266 # ----------------------------------------------------
 267 # Archipack panels
 268 # ----------------------------------------------------
 269
 270
 271 class TOOLS_PT_Archipack_Tools(Panel):
 272     bl_label = "Archipack Tools"
 273     bl_idname = "TOOLS_PT_Archipack_Tools"
 274     bl_space_type = "VIEW_3D"
 275     bl_region_type = "TOOLS"
 276     bl_category = "Tools"
 277     bl_context = "objectmode"
 278
 279     @classmethod
 280     def poll(self, context):
 281         return True
 282
 283     def draw(self, context):
 284         wm = context.window_manager
 285         layout = self.layout
 286         box = layout.box()
 287         box.label("Auto boolean")
 288         box.operator("archipack.auto_boolean", text="AutoBoolean", icon='AUTO').mode = 'HYBRID'
 289         box = layout.box()
 290         box.label("Rendering")
 291         box.prop(wm.archipack, 'render_type', text="")
 292         box.operator("archipack.render", icon='RENDER_STILL')
 293         box = layout.box()
 294         box.label("Render presets Thumbnails")
 295         box.operator("archipack.render_thumbs", icon='ERROR')
 296
 297
 298 class TOOLS_PT_Archipack_Create(Panel):
 299     bl_label = "Add Archipack"
 300     bl_idname = "TOOLS_PT_Archipack_Create"
 301     bl_space_type = "VIEW_3D"
 302     bl_region_type = "TOOLS"
 303     bl_category = "Create"
 304     bl_context = "objectmode"
 305
 306     @classmethod
 307     def poll(self, context):
 308         return True
 309
 310     def draw(self, context):
 311         global icons_collection
 312
 313         icons = icons_collection["main"]
 314         layout = self.layout
 315         row = layout.row(align=True)
 316         box = row.box()
 317         box.label("Objects")
 318         row = box.row(align=True)
 319         row.operator("archipack.window_preset_menu",
 320                     text="Window",
 321                     icon_value=icons["window"].icon_id
 322                     ).preset_operator = "archipack.window"
 323         row.operator("archipack.window_preset_menu",
 324                     text="",
 325                     icon='GREASEPENCIL'
 326                     ).preset_operator = "archipack.window_draw"
 327         row = box.row(align=True)
 328         row.operator("archipack.door_preset_menu",
 329                     text="Door",
 330                     icon_value=icons["door"].icon_id
 331                     ).preset_operator = "archipack.door"
 332         row.operator("archipack.door_preset_menu",
 333                     text="",
 334                     icon='GREASEPENCIL'
 335                     ).preset_operator = "archipack.door_draw"
 336         row = box.row(align=True)
 337         row.operator("archipack.stair_preset_menu",
 338                     text="Stair",
 339                     icon_value=icons["stair"].icon_id
 340                     ).preset_operator = "archipack.stair"
 341         row = box.row(align=True)
 342         row.operator("archipack.wall2",
 343                     icon_value=icons["wall"].icon_id
 344                     )
 345         row.operator("archipack.wall2_draw", text="Draw", icon='GREASEPENCIL')
 346         row.operator("archipack.wall2_from_curve", text="", icon='CURVE_DATA')
 347
 348         row = box.row(align=True)
 349         row.operator("archipack.fence_preset_menu",
 350                     text="Fence",
 351                     icon_value=icons["fence"].icon_id
 352                     ).preset_operator = "archipack.fence"
 353         row.operator("archipack.fence_from_curve", text="", icon='CURVE_DATA')
 354         row = box.row(align=True)
 355         row.operator("archipack.truss",
 356                     icon_value=icons["truss"].icon_id
 357                     )
 358         row = box.row(align=True)
 359         row.operator("archipack.slab_from_curve",
 360                     icon_value=icons["slab"].icon_id
 361                     )
 362         row = box.row(align=True)
 363         row.operator("archipack.wall2_from_slab",
 364                     icon_value=icons["wall"].icon_id)
 365         row.operator("archipack.slab_from_wall",
 366                     icon_value=icons["slab"].icon_id
 367                     ).ceiling = False
 368         row.operator("archipack.slab_from_wall",
 369                     text="->Ceiling",
 370                     icon_value=icons["slab"].icon_id
 371                     ).ceiling = True
 372         row = box.row(align=True)
 373         row.operator("archipack.roof_preset_menu",
 374                     text="Roof",
 375                     icon_value=icons["roof"].icon_id
 376                     ).preset_operator = "archipack.roof"
 377         row = box.row(align=True)
 378         row.operator("archipack.floor_preset_menu",
 379                     text="Floor",
 380                     icon_value=icons["floor"].icon_id
 381                     ).preset_operator = "archipack.floor"
 382         row.operator("archipack.floor_preset_menu",
 383                     text="->Wall",
 384                     icon_value=icons["floor"].icon_id
 385                     ).preset_operator = "archipack.floor_from_wall"
 386         row.operator("archipack.floor_preset_menu",
 387                     text="",
 388                     icon='CURVE_DATA').preset_operator = "archipack.floor_from_curve"
 389
 390
 391 # ----------------------------------------------------
 392 # ALT + A menu
 393 # ----------------------------------------------------
 394
 395
 396 def draw_menu(self, context):
 397     global icons_collection
 398     icons = icons_collection["main"]
 399     layout = self.layout
 400     layout.operator_context = 'INVOKE_REGION_WIN'
 401
 402     layout.operator("archipack.wall2",
 403                     text="Wall",
 404                     icon_value=icons["wall"].icon_id
 405                     )
 406     layout.operator("archipack.window_preset_menu",
 407                     text="Window",
 408                     icon_value=icons["window"].icon_id
 409                     ).preset_operator = "archipack.window"
 410     layout.operator("archipack.door_preset_menu",
 411                     text="Door",
 412                     icon_value=icons["door"].icon_id
 413                     ).preset_operator = "archipack.door"
 414     layout.operator("archipack.stair_preset_menu",
 415                     text="Stair",
 416                     icon_value=icons["stair"].icon_id
 417                     ).preset_operator = "archipack.stair"
 418     layout.operator("archipack.fence_preset_menu",
 419                     text="Fence",
 420                     icon_value=icons["fence"].icon_id
 421                     ).preset_operator = "archipack.fence"
 422     layout.operator("archipack.truss",
 423                     text="Truss",
 424                     icon_value=icons["truss"].icon_id
 425                     )
 426     layout.operator("archipack.floor_preset_menu",
 427                     text="Floor",
 428                     icon_value=icons["floor"].icon_id
 429                     ).preset_operator = "archipack.floor"
 430     layout.operator("archipack.roof_preset_menu",
 431                     text="Roof",
 432                     icon_value=icons["roof"].icon_id
 433                     ).preset_operator = "archipack.roof"
 434
 435
 436 class ARCHIPACK_MT_create(Menu):
 437     bl_label = 'Archipack'
 438
 439     def draw(self, context):
 440         draw_menu(self, context)
 441
 442
 443 def menu_func(self, context):
 444     layout = self.layout
 445     layout.separator()
 446     global icons_collection
 447     icons = icons_collection["main"]
 448
 449     # either draw sub menu or right at end of this one
 450     if context.user_preferences.addons[__name__].preferences.create_submenu:
 451         layout.operator_context = 'INVOKE_REGION_WIN'
 452         layout.menu("ARCHIPACK_MT_create", icon_value=icons["archipack"].icon_id)
 453     else:
 454         draw_menu(self, context)
 455
 456
 457 # ----------------------------------------------------
 458 # Datablock to store global addon variables
 459 # ----------------------------------------------------
 460
 461
 462 class archipack_data(PropertyGroup):
 463     render_type = EnumProperty(
 464         items=(
 465             ('1', "Draw over", "Draw over last rendered image"),
 466             ('2', "OpenGL", ""),
 467             ('3', "Animation OpenGL", ""),
 468             ('4', "Image", "Render image and draw over"),
 469             ('5', "Animation", "Draw on each frame")
 470             ),
 471         name="Render type",
 472         description="Render method"
 473         )
 474
 475
 476 def register():
 477     global icons_collection
 478     icons = previews.new()
 479     icons_dir = os.path.join(os.path.dirname(__file__), "icons")
 480     for icon in os.listdir(icons_dir):
 481         name, ext = os.path.splitext(icon)
 482         icons.load(name, os.path.join(icons_dir, icon), 'IMAGE')
 483     icons_collection["main"] = icons
 484
 485     archipack_progressbar.register()
 486     archipack_material.register()
 487     archipack_snap.register()
 488     archipack_manipulator.register()
 489     archipack_reference_point.register()
 490     archipack_autoboolean.register()
 491     archipack_door.register()
 492     archipack_window.register()
 493     archipack_stair.register()
 494     archipack_wall2.register()
 495     archipack_roof.register()
 496     archipack_slab.register()
 497     archipack_fence.register()
 498     archipack_truss.register()
 499     archipack_floor.register()
 500     archipack_rendering.register()
 501
 502     bpy.utils.register_class(archipack_data)
 503     WindowManager.archipack = PointerProperty(type=archipack_data)
 504     bpy.utils.register_class(Archipack_Pref)
 505     update_panel(None, bpy.context)
 506     bpy.utils.register_class(ARCHIPACK_MT_create)
 507     bpy.types.VIEW3D_MT_mesh_add.append(menu_func)
 508
 509
 510 def unregister():
 511     global icons_collection
 512     bpy.types.VIEW3D_MT_mesh_add.remove(menu_func)
 513     bpy.utils.unregister_class(ARCHIPACK_MT_create)
 514
 515     bpy.utils.unregister_class(TOOLS_PT_Archipack_Tools)
 516     bpy.utils.unregister_class(TOOLS_PT_Archipack_Create)
 517     bpy.utils.unregister_class(Archipack_Pref)
 518     archipack_progressbar.unregister()
 519     archipack_material.unregister()
 520     archipack_snap.unregister()
 521     archipack_manipulator.unregister()
 522     archipack_reference_point.unregister()
 523     archipack_autoboolean.unregister()
 524     archipack_door.unregister()
 525     archipack_window.unregister()
 526     archipack_stair.unregister()
 527     archipack_wall2.unregister()
 528     archipack_roof.unregister()
 529     archipack_slab.unregister()
 530     archipack_fence.unregister()
 531     archipack_truss.unregister()
 532     archipack_floor.unregister()
 533     archipack_rendering.unregister()
 534
 535     bpy.utils.unregister_class(archipack_data)
 536     del WindowManager.archipack
 537
 538     for icons in icons_collection.values():
 539         previews.remove(icons)
 540     icons_collection.clear()
 541
 542
 543 if __name__ == "__main__":
 544     register()