__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, write to the Free Software Foundation,
#  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
# END GPL LICENSE BLOCK #####

bl_info = {
    "name": "Gaffer",
    "description": "Master your lighting workflow with easy access to light properties, HDRIs and other tools",
    "author": "Greg Zaal",
    "version": (3, 2, 2),
    "blender": (3, 4, 0),
    "location": "3D View > Sidebar  &  World Settings > HDRI",
    "warning": "",
    "wiki_url": "https://github.com/gregzaal/Gaffer/wiki",
    "tracker_url": "https://github.com/gregzaal/Gaffer/issues?q=is%3Aopen+is%3Aissue+label%3Abug",
    "category": "Lighting",
}

if "bpy" in locals():
    import imp

    imp.reload(constants)
    imp.reload(functions)
    imp.reload(operators)
    imp.reload(ui)
    imp.reload(addon_updater)
    imp.reload(addon_updater_ops)
else:
    from . import constants, functions, operators, ui  # noqa: F401 (imported but unused, needed for reload)

import bpy
import os
from . import addon_updater_ops
from collections import OrderedDict


class GafferPreferences(bpy.types.AddonPreferences):
    bl_idname = __package__

    # Add-on Updater Prefs
    auto_check_update: bpy.props.BoolProperty(
        name="Auto-check for Update",
        description="If enabled, auto-check for updates using an interval",
        default=True,
    )
    updater_interval_months: bpy.props.IntProperty(
        name="Months",
        description="Number of months between checking for updates",
        default=0,
        min=0,
    )
    updater_interval_days: bpy.props.IntProperty(
        name="Days",
        description="Number of days between checking for updates",
        default=1,
        min=0,
    )
    updater_interval_hours: bpy.props.IntProperty(
        name="Hours",
        description="Number of hours between checking for updates",
        default=0,
        min=0,
        max=23,
    )
    updater_interval_minutes: bpy.props.IntProperty(
        name="Minutes",
        description="Number of minutes between checking for updates",
        default=0,
        min=0,
        max=59,
    )
    updater_expand_prefs: bpy.props.BoolProperty(default=False)

    # Add-on Prefs
    show_hdri_list: bpy.props.BoolProperty(
        name="Show",
        description="List all the detected HDRIs and their detected resolutions/variants below",
        default=False,
    )
    include_8bit: bpy.props.BoolProperty(
        name="Detect 8-bit images as HDRIs",
        description=(
            "Include LDR images like JPGs and PNGs when searching for files in the HDRI folder. "
            "Sometimes example renders are put next to HDRI files which can confuse the HDRI detection "
            "process, so they are ignored by default. Enable this to include them"
        ),
        default=False,
        update=functions.update_hdri_path,
    )
    panel_category: bpy.props.StringProperty(
        name="Panel Category/Tab",
        description=("Select which sidebar category/tab to place Gaffer's panels in"),
        default="Gaffer",
        update=ui.update_category,
    )
    offline_mode: bpy.props.BoolProperty(
        name="Offline Mode",
        description=("Stop Gaffer from checking polyhaven.com for the latest HDRI and tag lists"),
        default=False,
        update=functions.update_offline_mode,
    )
    auto_refresh_light_list: bpy.props.BoolProperty(
        name="Auto-Refresh Light List",
        description=(
            "Watch for changes to the scene and automatically refresh the light list. "
            "May impact performance in heavy scenes. If disabled, "
            "you'll need to manually refresh the light list when you add or remove lights"
        ),
        default=True,
    )

    show_debug: bpy.props.BoolProperty(
        name="Show Debug Tools",
        description="Expand this box to show various debugging tools",
        default=False,
    )

    ForcePreviewsRefresh: bpy.props.BoolProperty(default=True, options={"HIDDEN"})
    RequestThumbGen: bpy.props.BoolProperty(default=False, options={"HIDDEN"})

    def draw(self, context):
        layout = self.layout

        main_col = layout.column()

        row = main_col.row(align=True)
        row.label(text="HDRI Folders:")
        hdri_paths = functions.get_persistent_setting("hdri_paths")
        if hdri_paths[0] != "":
            sub = row.column(align=True)
            sub.alignment = "RIGHT"
            sub.operator("gaffer.hdri_path_add", text="Add another", icon="ADD")
        hp_col = main_col.column(align=True)
        for i, hp in enumerate(hdri_paths):
            row = hp_col.row(align=True)
            row.operator(
                "gaffer.hdri_path_edit",
                text=hp + (" (missing!)" if not os.path.exists(hp) else ""),
                icon=("ERROR" if not os.path.exists(hp) else "NONE"),
            ).folder_index = i
            if len(hdri_paths) > 1:
                row.operator("gaffer.hdri_path_remove", text="", icon="X").folder_index = i
            row.operator("gaffer.hdri_path_edit", text="", icon="FILE_FOLDER").folder_index = i

        if hdri_paths[0] != "":
            hdris = functions.get_hdri_list()
            if hdris:
                num_files = sum(len(x) for x in hdris.values())
                hdris = OrderedDict(sorted(hdris.items(), key=lambda x: x[0].lower()))
                num_hdris = len(hdris)
                row = main_col.row()
                row.alignment = "RIGHT"
                row.label(text="Found {} HDRIs ({} files)".format(num_hdris, num_files))
                if num_hdris > 0:
                    row.prop(self, "show_hdri_list", toggle=True)
                row.operator("gaffer.detect_hdris", text="Refresh", icon="FILE_REFRESH")

                if self.show_hdri_list:
                    col = main_col.column(align=True)
                    for name in hdris:
                        col.label(text=name)
                        variations = hdris[name]
                        if len(variations) >= 10:
                            row = col.row()
                            row.alignment = "CENTER"
                            row.label(
                                text=(
                                    "There are quite a few varations of this HDRI, " "maybe they were wrongly detected?"
                                ),
                                icon="QUESTION",
                            )
                            row = col.row()
                            row.alignment = "CENTER"
                            op = row.operator(
                                "wm.url_open",
                                text="Click here to learn how to fix this",
                                icon="URL",
                            )
                            op.url = "https://github.com/gregzaal/Gaffer/wiki/HDRI-Detection-and-Grouping"
                        for v in variations:
                            col.label(text="    " + v)
        else:
            row = main_col.row()
            row.alignment = "RIGHT"
            row.label(text="Select the folder that contains all your HDRIs. Subfolders will be included.")

        row = main_col.row()
        row.alignment = "RIGHT"
        row.prop(self, "include_8bit")

        row = main_col.row()
        row.label(text="Settings:")
        col = main_col.column()
        col.prop(self, "panel_category")
        col.prop(self, "offline_mode")
        col.prop(self, "auto_refresh_light_list")

        addon_updater_ops.update_settings_ui(self, context)

        box = layout.box()
        col = box.column()
        row = col.row(align=True)
        row.alignment = "LEFT"
        row.prop(
            self,
            "show_debug",
            text="Debug Tools",
            emboss=False,
            icon="TRIA_DOWN" if self.show_debug else "TRIA_RIGHT",
        )
        if self.show_debug:
            col = box.column()
            col.operator("gaffer.dbg_delete_thumbs")
            col.operator("gaffer.dbg_upload_hdri_list")
            col.operator("gaffer.dbg_upload_logs")


class BlacklistedObject(bpy.types.PropertyGroup):
    name: bpy.props.StringProperty(default="")


class GafferProperties(bpy.types.PropertyGroup):
    Lights: bpy.props.StringProperty(name="Lights", default="", description="The objects to include in the isolation")
    ColTempExpand: bpy.props.BoolProperty(
        name="Color Temperature Presets",
        default=False,
        description="Preset color temperatures based on real-world light sources",
    )
    MoreExpand: bpy.props.StringProperty(
        name="Show more options",
        default="",
        description="Show settings such as MIS, falloff, ray visibility...",
    )
    MoreExpandAll: bpy.props.BoolProperty(
        name="Show more options",
        default=False,
        description="Show settings such as MIS, falloff, ray visibility...",
    )
    LightUIIndex: bpy.props.IntProperty(name="light index", default=0, min=0, description="light index")
    LightsHiddenRecord: bpy.props.StringProperty(name="hidden record", default="", description="hidden record")
    WorldHiddenRecord: bpy.props.StringProperty(name="hidden record", default="", description="hidden record")
    SoloActive: bpy.props.StringProperty(name="soloactive", default="", description="soloactive")
    VisibleCollectionsOnly: bpy.props.BoolProperty(
        name="Visible Collections Only",
        default=True,
        description="Only show lights that are in visible collections",
    )
    VisibleLightsOnly: bpy.props.BoolProperty(
        name="Visible Lights Only",
        default=False,
        description="Only show lights that are not hidden",
    )
    WorldVis: bpy.props.BoolProperty(
        name="Hide World lighting",
        default=True,
        description="Don't display (or render) the environment lighting",
        update=functions._update_world_vis,
    )
    WorldReflOnly: bpy.props.BoolProperty(
        name="Reflection Only",
        default=False,
        description="Only show the World lighting in reflections",
        update=functions._update_world_refl_only,
    )
    LightRadiusAlpha: bpy.props.FloatProperty(
        name="Alpha",
        default=0.6,
        min=0,
        max=1,
        description="The opacity of the overlaid circles",
    )
    LightRadiusUseColor: bpy.props.BoolProperty(
        name="Use Color",
        default=True,
        description="Draw the radius of each light in the same color as the light",
    )
    LabelUseColor: bpy.props.BoolProperty(
        name="Use Color",
        default=True,
        description="Draw the label of each light in the same color as the light",
    )
    LightRadiusSelectedOnly: bpy.props.BoolProperty(
        name="Selected Only",
        default=False,
        description="Draw the radius for every visible light, or only selected lights",
    )
    LightRadiusXray: bpy.props.BoolProperty(
        name="X-Ray",
        default=False,
        description="Draw the circle in front of all objects",
    )
    LightRadiusDrawType: bpy.props.EnumProperty(
        name="Draw Type",
        description="How should the radius display look?",
        default="dotted",
        items=(
            ("filled", "Filled", "Draw a circle filled with a solid color"),
            ("solid", "Solid", "Draw a solid outline of the circle"),
            ("dotted", "Dotted", "Draw a dotted outline of the circle"),
        ),
    )
    DefaultRadiusColor: bpy.props.FloatVectorProperty(
        name="Default Color",
        description=(
            "When 'Use Color' is disaled, or when the color of a light is unknown "
            "(such as when using a texture), this color is used instead"
        ),
        subtype="COLOR",
        size=3,
        min=0.0,
        max=1.0,
        default=(1.0, 1.0, 1.0),
    )
    DefaultLabelBGColor: bpy.props.FloatVectorProperty(
        name="Background Color",
        description=(
            "When 'Use Color' is disaled, or when the color of a light is unknown "
            "(such as when using a texture), this color is used instead"
        ),
        subtype="COLOR",
        size=3,
        min=0.0,
        max=1.0,
        default=(0.0, 0.0, 0.0),
    )
    LabelAlpha: bpy.props.FloatProperty(
        name="Alpha",
        default=0.5,
        min=0,
        max=1,
        description="The opacity of the drawn labels",
    )
    LabelFontSize: bpy.props.IntProperty(name="Font Size", default=14, min=1, description="How large the text is drawn")
    LabelDrawType: bpy.props.EnumProperty(
        name="Draw Type",
        description="How should the label look?",
        default="color_bg",
        items=(
            (
                "color_bg",
                "Colored background, plain text",
                "Show the label name on a colored background",
            ),
            (
                "plain_bg",
                "Colored text in plain background",
                "Show the label name in color, on a plain background",
            ),
            (
                "color_text",
                "Text only, no background",
                "Show the text without any background",
            ),
        ),
    )
    LabelTextColor: bpy.props.FloatVectorProperty(
        name="Text Color",
        description="The color of the label name text",
        subtype="COLOR",
        size=3,
        min=0.0,
        max=1.0,
        default=(1.0, 1.0, 1.0),
    )
    LabelAlign: bpy.props.EnumProperty(
        name="Alignment",
        description="The positioning of the label relative to the light",
        default="r",
        items=(
            ("c", "Centered", "Positioned exactly on the light"),
            ("t", "Top", "Positioned above the light"),
            ("b", "Bottom", "Positioned below the light"),
            ("l", "Left", "Positioned to the left of the light"),
            ("r", "Right", "Positioned to the right of the light"),
            ("bl", "Bottom Left", "Positioned below and to the left of the light"),
            ("tl", "Top Left", "Positioned above and to the left of the light"),
            ("tr", "Top Right", "Positioned below and to the right of the light"),
            ("br", "Bottom Right", "Positioned above and to the right of the light"),
        ),
    )
    LabelMargin: bpy.props.IntProperty(
        name="Margin",
        default=90,
        description="Draw the label this distance away from the light",
    )
    SunObject: bpy.props.StringProperty(
        name="Sun Obj",
        default="",
        description="The light object to use to drive the Sky rotation",
    )

    # Internal vars (not shown in UI)
    IsShowingRadius: bpy.props.BoolProperty(default=False, options={"HIDDEN"})
    IsShowingLabel: bpy.props.BoolProperty(default=False, options={"HIDDEN"})
    BlacklistIndex: bpy.props.IntProperty(default=0, options={"HIDDEN"})
    VarNameCounter: bpy.props.IntProperty(default=0, options={"HIDDEN"})
    HDRIList: bpy.props.StringProperty(default="", options={"HIDDEN"})
    RequestJPGGen: bpy.props.BoolProperty(default=False, options={"HIDDEN"})
    ShowProgress: bpy.props.BoolProperty(default=False, options={"HIDDEN"})
    Progress: bpy.props.FloatProperty(default=0.0, options={"HIDDEN"})
    ProgressText: bpy.props.StringProperty(default="", options={"HIDDEN"})
    ProgressBarText: bpy.props.StringProperty(default="", options={"HIDDEN"})
    ShowHDRIHaven: bpy.props.BoolProperty(default=False, options={"HIDDEN"})
    ThumbnailsBigHDRIFound: bpy.props.BoolProperty(default=False, options={"HIDDEN"})
    FileNotFoundError: bpy.props.BoolProperty(default=False, options={"HIDDEN"})
    Blacklist: bpy.props.CollectionProperty(type=BlacklistedObject)  # must be registered after classes


class GafferHDRIProperties(bpy.types.PropertyGroup):
    # HDRI Handler stuffs
    hdri_handler_enabled: bpy.props.BoolProperty(
        name="Enable",
        description="Turn on/off Gaffer's HDRI handler",
        default=False,
        update=functions.hdri_enable,
    )
    hdri: bpy.props.EnumProperty(name="HDRIs", items=functions.hdri_enum_previews, update=functions.switch_hdri)
    hdri_variation: bpy.props.EnumProperty(
        name="Resolution / Variation",
        items=functions.variation_enum_previews,
        update=functions.update_variation,
    )
    hdri_search: bpy.props.StringProperty(
        name="Search",
        description="Show only HDRIs matching this text - name, subfolder and tags will match",
        default="",
        update=functions.update_search,
    )
    hdri_favorite: bpy.props.BoolProperty(
        name="Show Only Favorites",
        description="Filter to show only your favorite HDRIs in the list",
        default=False,
        update=functions.update_search,
    )
    hdri_folder_filter: bpy.props.StringProperty(
        name="Folder Filter",
        description="Show only HDRIs from this subfolder",
        default="",
        update=functions.update_search,
    )
    hdri_rotation: bpy.props.FloatProperty(
        name="Rotation",
        description="Rotate the HDRI (in degrees) around the Z-axis",
        default=0,
        soft_min=-180,
        soft_max=180,
        update=functions.update_rotation,
    )
    hdri_brightness: bpy.props.FloatProperty(
        name="Brightness",
        description="Change the exposure of the HDRI to emit more or less light (measured in EVs)",
        default=0,
        soft_min=-10,
        soft_max=10,
        update=functions.update_brightness,
    )
    hdri_contrast: bpy.props.FloatProperty(
        name="Contrast",
        description="Adjust the Gamma, making light sources stonger or weaker compared to darker parts of the HDRI",
        default=1,
        min=0,
        soft_max=2,
        update=functions.update_contrast,
    )
    hdri_saturation: bpy.props.FloatProperty(
        name="Saturation",
        description="Control how strong the colours in the HDRI are",
        default=1,
        min=0,
        soft_max=2,
        update=functions.update_saturation,
    )
    hdri_warmth: bpy.props.FloatProperty(
        name="Warmth",
        description="Control the relative color temperature of the HDRI (blue/orange)",
        default=1,
        soft_min=0,
        soft_max=2,
        update=functions.update_warmth,
    )
    hdri_tint: bpy.props.FloatProperty(
        name="Purple/Green Tint",
        description="Control the purple/green color balance, usually used together with the warmth setting",
        default=1,
        soft_min=0,
        soft_max=2,
        update=functions.update_tint,
    )
    hdri_color: bpy.props.FloatVectorProperty(
        name="Mix Color",
        description="Tint, or fully color, the environment. Use Alpha to control strength/opacity",
        subtype="COLOR",
        size=4,
        soft_min=0.0,
        soft_max=1.0,
        default=(0.5, 0.15, 0.075, 0.0),
        update=functions.update_color,
    )
    hdri_horz_shift: bpy.props.FloatProperty(
        name="Horizon Shift",
        description="Move the horizon down to view more of the sky",
        default=0,
        soft_min=0,
        soft_max=1,
        update=functions.update_horizon,
    )
    hdri_horz_exp: bpy.props.FloatProperty(
        name="Warp",
        description="Adjust this if the sky looks distorted",
        default=0,
        soft_min=-1,
        soft_max=1,
        update=functions.update_horizon,
    )
    hdri_use_jpg_background: bpy.props.BoolProperty(
        name="High-res JPG background",
        default=False,
        description=(
            "Use a higher-res JPG image for the background, keeping the HDR just for lighting - "
            "enable this and set the main resolution to a low option to save memory"
        ),
        update=functions.setup_hdri,
    )
    hdri_use_darkened_jpg: bpy.props.BoolProperty(
        name="Pre-darkened",
        default=False,
        description=(
            "Use a darker version of the JPG to avoid clipped highlights (but at the cost of potential banding)"
        ),
        update=functions.setup_hdri,
    )
    hdri_use_bg_reflections: bpy.props.BoolProperty(
        name="Use for reflections",
        default=False,
        description="Use these settings for the appearance of reflections as well",
        update=functions.setup_hdri,
    )
    hdri_use_separate_rotation: bpy.props.BoolProperty(
        name="Rotation",
        default=False,
        description="Adjust the rotation for the background separately from the lighting",
        update=functions.setup_hdri,
    )
    hdri_background_rotation: bpy.props.FloatProperty(
        name="Value",
        description="Rotate the HDRI (in degrees) around the Z-axis",
        default=0,
        soft_min=-180,
        soft_max=180,
        update=functions.update_background_rotation,
    )
    hdri_use_separate_brightness: bpy.props.BoolProperty(
        name="Brightness",
        default=False,
        description="Adjust the brightness value for the background separately from the lighting",
        update=functions.setup_hdri,
    )
    hdri_background_brightness: bpy.props.FloatProperty(
        name="Value",
        description="Make the background image brighter or darker without affecting the lighting",
        default=0,
        soft_min=-10,
        soft_max=10,
        update=functions.update_background_brightness,
    )
    hdri_use_separate_contrast: bpy.props.BoolProperty(
        name="Contrast",
        default=False,
        description="Adjust the contrast value for the background separately from the lighting",
        update=functions.setup_hdri,
    )
    hdri_background_contrast: bpy.props.FloatProperty(
        name="Value",
        description="Give the background image more or less contrast without affecting the lighting",
        default=1,
        min=0,
        soft_max=2,
        update=functions.update_background_contrast,
    )
    hdri_use_separate_saturation: bpy.props.BoolProperty(
        name="Saturation",
        default=False,
        description="Adjust the saturation value for the background separately from the lighting",
        update=functions.setup_hdri,
    )
    hdri_background_saturation: bpy.props.FloatProperty(
        name="Value",
        description="Change the saturation of the background image without affecting the lighting",
        default=1,
        min=0,
        soft_max=2,
        update=functions.update_background_saturation,
    )
    hdri_use_separate_warmth: bpy.props.BoolProperty(
        name="Warmth",
        default=False,
        description="Adjust the warmth value for the background separately from the lighting",
        update=functions.setup_hdri,
    )
    hdri_background_warmth: bpy.props.FloatProperty(
        name="Value",
        description="Change the warmth of the background image without affecting the lighting",
        default=1,
        soft_min=0,
        soft_max=2,
        update=functions.update_background_warmth,
    )
    hdri_use_separate_tint: bpy.props.BoolProperty(
        name="Tint",
        default=False,
        description="Adjust the tint value for the background separately from the lighting",
        update=functions.setup_hdri,
    )
    hdri_background_tint: bpy.props.FloatProperty(
        name="Value",
        description="Change the tint of the background image without affecting the lighting",
        default=1,
        soft_min=0,
        soft_max=2,
        update=functions.update_background_tint,
    )
    hdri_use_separate_color: bpy.props.BoolProperty(
        name="Mix Color",
        default=False,
        description="Adjust the color value for the background separately from the lighting",
        update=functions.setup_hdri,
    )
    hdri_background_color: bpy.props.FloatVectorProperty(
        name="Background Color",
        description="Tint, or fully color, the background. Use Alpha to control strength/opacity",
        subtype="COLOR",
        size=4,
        soft_min=0.0,
        soft_max=1.0,
        default=(0.5, 0.15, 0.075, 0.0),
        update=functions.update_background_color,
    )
    hdri_clamp: bpy.props.FloatProperty(
        name="Clamp Brightness",
        description=(
            "Set any values brighter than this value to this value. "
            "Disabled when on 0. Use when bright lights (e.g. sun) are too bright"
        ),
        default=0,
        min=0,
        soft_max=50000,
        update=functions.update_clamp,
    )
    hdri_advanced: bpy.props.BoolProperty(name="Advanced", description="Show/hide advanced settings", default=False)
    hdri_jpg_gen_all: bpy.props.BoolProperty(
        name="Generate for ALL HDRIs",
        description="Generate the JPG and darkened JPG for all HDRIs that you have. This will probably take a while",
        default=False,
    )
    hdri_show_tags_ui: bpy.props.BoolProperty(
        name="Tags",
        description="Choose some tags for this HDRI to help you search for it later",
        default=False,
    )
    hdri_custom_tags: bpy.props.StringProperty(
        name="New Tag(s)",
        description="Add some of your own tags to this HDRI - separate by commas or semi-colons",
        default="",
        update=functions.set_custom_tags,
    )

    # Internal vars (not shown in UI)
    OldWorldSettings: bpy.props.StringProperty(default="", options={"HIDDEN"})


classes = [
    GafferPreferences,
    BlacklistedObject,
    GafferProperties,
    GafferHDRIProperties,
    operators.GAFFER_OT_rename,
    operators.GAFFER_OT_set_strength,
    operators.GAFFER_OT_set_temp,
    operators.GAFFER_OT_show_temp_list,
    operators.GAFFER_OT_hide_temp_list,
    operators.GAFFER_OT_show_more,
    operators.GAFFER_OT_hide_more,
    operators.GAFFER_OT_hide_show_light,
    operators.GAFFER_OT_select_light,
    operators.GAFFER_OT_solo,
    operators.GAFFER_OT_light_use_nodes,
    operators.GAFFER_OT_node_set_strength,
    operators.GAFFER_OT_refresh_light_list,
    operators.GAFFER_OT_set_light_data_user_names,
    operators.GAFFER_OT_apply_exposure,
    operators.GAFFER_OT_link_sky_to_sun,
    operators.GAFFER_OT_aim_light,
    operators.GAFFER_OT_aim_light_with_view,
    operators.GAFFER_OT_show_light_radius,
    operators.GAFFER_OT_show_light_label,
    operators.GAFFER_OT_refresh_bgl,
    operators.GAFFER_OT_add_blacklisted,
    operators.GAFFER_OT_remove_blacklisted,
    operators.GAFFER_OT_detect_hdris,
    operators.GAFFER_OT_hdri_path_edit,
    operators.GAFFER_OT_hdri_path_add,
    operators.GAFFER_OT_hdri_path_remove,
    operators.GAFFER_OT_hdri_thumb_gen,
    operators.GAFFER_OT_hdri_jpg_gen,
    operators.GAFFER_OT_hdri_clear_search,
    operators.GAFFER_OT_hdri_set_favorite,
    operators.GAFFER_OT_hdri_set_folder_filter,
    operators.GAFFER_OT_hdri_paddles,
    operators.GAFFER_OT_hdri_variation_paddles,
    operators.GAFFER_OT_hdri_add_tag,
    operators.GAFFER_OT_hdri_random,
    operators.GAFFER_OT_hdri_reset,
    operators.GAFFER_OT_hdri_save,
    operators.GAFFER_OT_fix_mis,
    operators.GAFFER_OT_get_hdrihaven,
    operators.GAFFER_OT_hide_hdrihaven,
    operators.GAFFER_OT_open_hdrihaven,
    operators.GAFFER_OT_hdri_open_data_folder,
    operators.GAFFER_OT_debug_delete_thumbs,
    operators.GAFFER_OT_debug_upload_hdri_list,
    operators.GAFFER_OT_debug_upload_logs,
    ui.GAFFER_PT_hdris,
    ui.GAFFER_MT_folder_filter,
    ui.OBJECT_UL_object_list,
]


def register():
    addon_updater_ops.register(bl_info)

    functions.previews_register()
    functions.cleanup_logs()

    bpy.types.NODE_PT_active_node_generic.append(ui.gaffer_node_menu_func)

    from bpy.utils import register_class

    for cls in classes:
        register_class(cls)
    ui.update_category(bpy.context.preferences.addons[__name__].preferences, bpy.context)

    bpy.types.Scene.gaf_props = bpy.props.PointerProperty(type=GafferProperties)
    bpy.types.World.gaf_hdri_props = bpy.props.PointerProperty(type=GafferHDRIProperties)
    bpy.app.handlers.load_post.append(operators.load_handler)
    bpy.app.handlers.depsgraph_update_post.append(functions.depsgraph_update_post_handler)


def unregister():
    addon_updater_ops.unregister()

    bpy.app.handlers.load_post.remove(operators.load_handler)
    bpy.app.handlers.depsgraph_update_post.remove(functions.depsgraph_update_post_handler)

    functions.previews_unregister()

    if operators.GAFFER_OT_show_light_radius._handle is not None:
        bpy.types.SpaceView3D.draw_handler_remove(operators.GAFFER_OT_show_light_radius._handle, "WINDOW")
        bpy.context.scene.gaf_props.IsShowingRadius = False
    if operators.GAFFER_OT_show_light_label._handle is not None:
        bpy.types.SpaceView3D.draw_handler_remove(operators.GAFFER_OT_show_light_label._handle, "WINDOW")
        bpy.context.scene.gaf_props.IsShowingLabel = False

    del bpy.types.Scene.gaf_props
    del bpy.types.World.gaf_hdri_props

    bpy.types.NODE_PT_active_node_generic.remove(ui.gaffer_node_menu_func)

    from bpy.utils import unregister_class

    for cls in reversed(classes):
        unregister_class(cls)


if __name__ == "__main__":
    register()