437 lines
15 KiB
Python
437 lines
15 KiB
Python
# #### 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 #####
|
|
|
|
from enum import IntEnum
|
|
from typing import Tuple, Optional
|
|
|
|
from bpy.types import Operator
|
|
import bpy.utils.previews
|
|
|
|
from ..modules.poliigon_core.api_remote_control_params import (
|
|
CATEGORY_ALL,
|
|
KEY_TAB_IMPORTED,
|
|
KEY_TAB_MY_ASSETS,
|
|
KEY_TAB_RECENT_DOWNLOADS,
|
|
KEY_TAB_ONLINE,
|
|
KEY_TAB_LOCAL)
|
|
from ..modules.poliigon_core.multilingual import _t
|
|
from ..asset_browser.asset_browser import create_poliigon_library
|
|
from ..dialogs.utils_dlg import get_ui_scale
|
|
from ..constants import HDRI_RESOLUTIONS
|
|
from ..toolbox import get_context
|
|
from ..toolbox_settings import save_settings
|
|
from ..utils import f_MDir
|
|
from .. import reporting
|
|
|
|
|
|
class ModeUpdate(IntEnum):
|
|
NO_UPDATE = 0
|
|
# Any update includes UI refresh
|
|
IMPORTED_AND_GET_ASSETS_AND_PAGE_1 = 1
|
|
IMPORTED_AND_GET_ASSETS = 2
|
|
IMPORTED_ONLY = 3
|
|
|
|
|
|
class POLIIGON_OT_setting(Operator):
|
|
bl_idname = "poliigon.poliigon_setting"
|
|
bl_label = ""
|
|
bl_description = _t("Edit Poliigon Addon Settings")
|
|
bl_space_type = "VIEW_3D"
|
|
bl_region_type = "UI"
|
|
bl_options = {"REGISTER", "INTERNAL"}
|
|
|
|
tooltip: bpy.props.StringProperty(default="", options={"HIDDEN"}) # noqa: F722, F821
|
|
mode: bpy.props.StringProperty(default="", options={"HIDDEN"}) # noqa: F722, F821
|
|
|
|
@staticmethod
|
|
def init_context(addon_version: str) -> None:
|
|
"""Called from operators.py to init global addon context."""
|
|
|
|
global cTB
|
|
cTB = get_context(addon_version)
|
|
|
|
@classmethod
|
|
def description(cls, context, properties):
|
|
return properties.tooltip
|
|
|
|
@staticmethod
|
|
def _set_library() -> int:
|
|
path_library_new = cTB.settings["set_library"]
|
|
path_library_old = cTB.get_library_path(primary=True)
|
|
f_MDir(path_library_new)
|
|
|
|
cTB.replace_library_path(
|
|
path_library_old,
|
|
path_library_new,
|
|
primary=True,
|
|
update_local_assets=True)
|
|
|
|
if bpy.app.version >= (3, 0) and cTB.user_legacy_own_assets():
|
|
create_poliigon_library(force=True)
|
|
|
|
bpy.ops.poliigon.popup_welcome("INVOKE_DEFAULT")
|
|
|
|
# do_update: update and switch to page 1
|
|
return ModeUpdate.IMPORTED_AND_GET_ASSETS_AND_PAGE_1
|
|
|
|
def _set_area(self, area: Optional[str] = None) -> int:
|
|
with cTB.lock_thumbs:
|
|
cTB.thumbs.clear()
|
|
|
|
if area is None:
|
|
area = self.mode.replace("asset_filter_", "").replace("area_", "")
|
|
cTB.settings["area"] = area
|
|
|
|
# This caused a delay when switching between Poliigon/My Assets
|
|
# vClearCache = 1
|
|
|
|
cTB.settings["show_settings"] = 0
|
|
cTB.settings["show_user"] = 0
|
|
cTB.vActiveAsset = None
|
|
|
|
cTB.track_screen_from_area()
|
|
|
|
# do_update: update, but skip switching to page 1 and getting assets
|
|
update = ModeUpdate.IMPORTED_ONLY
|
|
|
|
# If local assets, always run get_assets in api_rc (no api call involved)
|
|
if area == KEY_TAB_LOCAL:
|
|
update = ModeUpdate.IMPORTED_AND_GET_ASSETS
|
|
|
|
# When changing filters we are always resetting to page 1
|
|
if "asset_filter" in self.mode:
|
|
update = ModeUpdate.IMPORTED_AND_GET_ASSETS_AND_PAGE_1
|
|
|
|
return update
|
|
|
|
@staticmethod
|
|
def _show_my_account() -> None:
|
|
cTB.settings["show_settings"] = 0
|
|
cTB.settings["show_user"] = 1
|
|
cTB.vActiveAsset = None
|
|
cTB.refresh_ui()
|
|
|
|
@staticmethod
|
|
def _show_settings() -> None:
|
|
# TODO(Andreas): Does this code still have a purpose?
|
|
cTB.settings["show_settings"] = 1
|
|
cTB.settings["show_user"] = 0
|
|
cTB.vActiveAsset = None
|
|
cTB.refresh_ui()
|
|
|
|
def _set_category(self) -> int:
|
|
props = bpy.context.window_manager.poliigon_props
|
|
area = cTB.settings["area"]
|
|
|
|
mode_parts = self.mode.split("_")
|
|
idx_category = int(mode_parts[1])
|
|
button = mode_parts[2]
|
|
if idx_category < len(cTB.settings["category"]):
|
|
cTB.settings["category"][idx_category] = button
|
|
else:
|
|
cTB.settings["category"].append(button)
|
|
cTB.settings["category"] = cTB.settings[
|
|
"category"][: idx_category + 1]
|
|
|
|
categories = cTB.settings["category"]
|
|
if len(categories) > 1 and categories[-1].startswith("All "):
|
|
categories = categories[:-1]
|
|
cTB.settings["category"] = categories
|
|
|
|
if categories != [CATEGORY_ALL]:
|
|
cTB.signal_category_filter(" > ".join(categories))
|
|
|
|
cTB.vActiveAsset = None
|
|
cTB.vActiveMat = None
|
|
cTB.vActiveMode = None
|
|
# Do we want to clear searches when switching between areas?
|
|
cTB.vSearch[KEY_TAB_ONLINE] = props.search
|
|
cTB.vSearch[KEY_TAB_MY_ASSETS] = props.search
|
|
cTB.vSearch[KEY_TAB_RECENT_DOWNLOADS] = props.search
|
|
cTB.vSearch[KEY_TAB_IMPORTED] = props.search
|
|
cTB.vSearch[KEY_TAB_LOCAL] = props.search
|
|
|
|
self._clear_search_tab(tab=area)
|
|
self._set_area(KEY_TAB_ONLINE)
|
|
|
|
# do_update: update and switch to page 1
|
|
return ModeUpdate.IMPORTED_AND_GET_ASSETS_AND_PAGE_1
|
|
|
|
def _set_page(self) -> int:
|
|
with cTB.lock_thumbs:
|
|
cTB.thumbs.clear()
|
|
|
|
area = cTB.settings["area"]
|
|
idx_page = self.mode.split("_")[-1]
|
|
|
|
if cTB.vPage[area] == idx_page:
|
|
return {"FINISHED"}
|
|
elif idx_page == "-":
|
|
if cTB.vPage[area] > 0:
|
|
cTB.vPage[area] -= 1
|
|
elif idx_page == "+":
|
|
if cTB.vPage[area] < cTB.vPages[area]:
|
|
cTB.vPage[area] += 1
|
|
else:
|
|
cTB.vPage[area] = int(idx_page)
|
|
|
|
# do_update: update, but skip switching to page 1 and getting assets
|
|
return ModeUpdate.IMPORTED_ONLY
|
|
|
|
def _set_page_size(self) -> Tuple[int, bool]:
|
|
per_page = int(self.mode.split("@")[1])
|
|
if cTB.settings["page"] == per_page:
|
|
return 0, False
|
|
|
|
cTB.settings["page"] = per_page
|
|
# do_update and clear_cache
|
|
# do_update: update and switch to page 1
|
|
return ModeUpdate.IMPORTED_AND_GET_ASSETS_AND_PAGE_1, True
|
|
|
|
def _clear_search_tab(self, tab: str) -> None:
|
|
props = bpy.context.window_manager.poliigon_props
|
|
props.search = ""
|
|
|
|
cTB.vLastSearch[tab] = ""
|
|
cTB.vSearch[tab] = ""
|
|
|
|
def _clear_search(self) -> None:
|
|
if self.mode.endswith(KEY_TAB_ONLINE):
|
|
self._clear_search_tab(tab=KEY_TAB_ONLINE)
|
|
elif self.mode.endswith(KEY_TAB_MY_ASSETS):
|
|
self._clear_search_tab(tab=KEY_TAB_MY_ASSETS)
|
|
elif self.mode.endswith(KEY_TAB_IMPORTED):
|
|
self._clear_search_tab(tab=KEY_TAB_IMPORTED)
|
|
elif self.mode.endswith(KEY_TAB_RECENT_DOWNLOADS):
|
|
self._clear_search_tab(tab=KEY_TAB_RECENT_DOWNLOADS)
|
|
elif self.mode.endswith(KEY_TAB_LOCAL):
|
|
self._clear_search_tab(tab=KEY_TAB_LOCAL)
|
|
cTB.flush_thumb_prefetch_queue()
|
|
|
|
def _set_thumb_size(self) -> None:
|
|
size = self.mode.split("@")[1]
|
|
if cTB.settings["thumbsize"] == size:
|
|
return
|
|
cTB.settings["thumbsize"] = size
|
|
cTB.refresh_ui()
|
|
|
|
def _set_mode(self) -> int:
|
|
cTB.settings[self.mode] = not cTB.settings[self.mode]
|
|
|
|
# Update the session reference of this setting too.
|
|
do_update = ModeUpdate.NO_UPDATE
|
|
if self.mode == "download_link_blend":
|
|
cTB.link_blend_session = cTB.settings[self.mode]
|
|
elif self.mode == "download_prefer_blend":
|
|
cTB._asset_index.flush_is_local()
|
|
cTB._asset_index.update_all_local_assets(
|
|
library_dirs=cTB.get_library_paths())
|
|
cTB.refresh_ui()
|
|
elif self.mode == "hdri_use_jpg_bg":
|
|
do_update = ModeUpdate.IMPORTED_ONLY
|
|
return do_update
|
|
|
|
def _set_default(self) -> int:
|
|
key = self.mode.split("_")[1]
|
|
value = self.mode.split("_")[2]
|
|
cTB.settings[key] = value
|
|
|
|
if not self.mode.startswith("default_hdri"):
|
|
cTB.refresh_ui()
|
|
return 0 # do_update: no update
|
|
|
|
idx_size_exr = HDRI_RESOLUTIONS.index(
|
|
cTB.settings["hdri"])
|
|
idx_size_jpg = HDRI_RESOLUTIONS.index(
|
|
cTB.settings["hdrib"])
|
|
if idx_size_jpg <= idx_size_exr:
|
|
idx_size_jpg_new = min(idx_size_exr + 1,
|
|
len(HDRI_RESOLUTIONS) - 1)
|
|
cTB.settings["hdrib"] = HDRI_RESOLUTIONS[idx_size_jpg_new]
|
|
# do_update: update, but skip switching to page 1
|
|
return ModeUpdate.IMPORTED_AND_GET_ASSETS
|
|
|
|
def _disable_library_directory(self) -> None:
|
|
directory = self.mode.replace("disable_dir_", "")
|
|
if directory in cTB.settings["disabled_dirs"]:
|
|
cTB.settings["disabled_dirs"].remove(directory)
|
|
cTB.logger.info(f"Enabled directory: {directory}")
|
|
cTB.add_library_path(
|
|
directory, primary=False, update_local_assets=True)
|
|
else:
|
|
cTB.settings["disabled_dirs"].append(directory)
|
|
cTB.logger.info(f"Disabled directory: {directory}")
|
|
cTB.remove_library_path(
|
|
directory, update_local_assets=True)
|
|
cTB.refresh_ui()
|
|
|
|
def _forget_library_directory(self) -> None:
|
|
directory = self.mode.replace("del_dir_", "")
|
|
cTB.remove_library_path(
|
|
directory, update_local_assets=True)
|
|
cTB.refresh_ui()
|
|
|
|
def _toggle_material_property(self) -> None:
|
|
prop = self.mode.split("@")[1]
|
|
if prop in cTB.settings["mat_props"]:
|
|
cTB.settings["mat_props"].remove(prop)
|
|
else:
|
|
cTB.settings["mat_props"].append(prop)
|
|
|
|
@staticmethod
|
|
def _view_more() -> int:
|
|
props = bpy.context.window_manager.poliigon_props
|
|
area = cTB.settings["area"]
|
|
|
|
prev_area = area
|
|
area = KEY_TAB_ONLINE
|
|
|
|
cTB.settings["area"] = area
|
|
cat_area = cTB.settings["category"]
|
|
cTB.settings["category"] = cat_area
|
|
cTB.settings["show_settings"] = 0
|
|
cTB.settings["show_user"] = 0
|
|
cTB.vSearch[KEY_TAB_ONLINE] = cTB.vSearch[prev_area]
|
|
props.search = cTB.vSearch[prev_area]
|
|
cTB.vActiveAsset = None
|
|
# do_update: update, but skip switching to page 1
|
|
return ModeUpdate.IMPORTED_AND_GET_ASSETS
|
|
|
|
@staticmethod
|
|
def _do_clear_cache(clear_cache: bool) -> None:
|
|
if not clear_cache:
|
|
return
|
|
# TODO(Andreas): I assume somebody planned some cTB.thumbs refresh
|
|
# or something?
|
|
|
|
@staticmethod
|
|
def _do_update(do_update: int) -> None:
|
|
if not do_update:
|
|
return
|
|
|
|
area = cTB.settings["area"]
|
|
cTB.flush_thumb_prefetch_queue()
|
|
|
|
if do_update == ModeUpdate.IMPORTED_AND_GET_ASSETS_AND_PAGE_1:
|
|
cTB.vPage[area] = 0
|
|
cTB.vPages[area] = 1
|
|
|
|
# Not setting cursor as it can lead to being stuck on "wait".
|
|
# bpy.context.window.cursor_set("WAIT")
|
|
|
|
# TODO(SOFT-762): refactor to cache raw API request, also validate
|
|
# if this needs re-requesting (has calls to f_GetCategoryChildren).
|
|
|
|
# TODO(Andreas): disabled, when going addon-core
|
|
# cTB.f_GetCategories()
|
|
|
|
# TODO(Andreas): redundant?
|
|
cTB.f_GetSceneAssets()
|
|
|
|
if do_update < ModeUpdate.IMPORTED_ONLY:
|
|
if area in [KEY_TAB_ONLINE,
|
|
KEY_TAB_MY_ASSETS,
|
|
KEY_TAB_RECENT_DOWNLOADS,
|
|
KEY_TAB_IMPORTED,
|
|
KEY_TAB_LOCAL]:
|
|
cTB.f_GetAssets(area=area,
|
|
force_request=area == KEY_TAB_LOCAL)
|
|
|
|
cTB.refresh_ui()
|
|
|
|
@reporting.handle_operator()
|
|
def execute(self, context):
|
|
cTB.logger.debug(f"POLIIGON_OT_setting: mode='{self.mode}'")
|
|
|
|
get_ui_scale(cTB) # Force update DPI check for scale.
|
|
do_update = ModeUpdate.NO_UPDATE
|
|
clear_cache = False
|
|
|
|
if self.mode in ["none", ""]:
|
|
return {"FINISHED"}
|
|
elif self.mode == "set_library":
|
|
do_update = self._set_library()
|
|
elif self.mode.startswith("area_"):
|
|
do_update = self._set_area()
|
|
elif self.mode == "my_account":
|
|
self._show_my_account()
|
|
return {"FINISHED"}
|
|
elif self.mode == "settings":
|
|
self._show_settings()
|
|
return {"FINISHED"}
|
|
elif self.mode.startswith("category_"):
|
|
do_update = self._set_category()
|
|
elif self.mode.startswith("page_"):
|
|
do_update = self._set_page()
|
|
elif self.mode.startswith("page@"):
|
|
do_update, clear_cache = self._set_page_size()
|
|
elif self.mode.startswith("clear_search_"):
|
|
self._clear_search()
|
|
elif self.mode.startswith("thumbsize@"):
|
|
self._set_thumb_size()
|
|
elif self.mode in [
|
|
"apply_subdiv",
|
|
"auto_download",
|
|
"download_lods",
|
|
"download_prefer_blend",
|
|
"download_link_blend",
|
|
"hdri_use_jpg_bg",
|
|
"mat_props_edit",
|
|
"new_top",
|
|
"one_click_purchase",
|
|
"show_active",
|
|
"show_add_dir",
|
|
"show_asset_info",
|
|
"show_credits",
|
|
"show_default_prefs",
|
|
"show_display_prefs",
|
|
"show_import_prefs",
|
|
"show_asset_browser_prefs",
|
|
"show_mat_ops",
|
|
"show_mat_props",
|
|
"show_mat_texs",
|
|
"show_plan",
|
|
"show_feedback",
|
|
"show_settings",
|
|
"show_user",
|
|
"use_16"
|
|
]:
|
|
do_update = self._set_mode()
|
|
elif self.mode.startswith("default_"):
|
|
do_update = self._set_default()
|
|
elif self.mode.startswith("disable_dir_"):
|
|
self._disable_library_directory()
|
|
elif self.mode.startswith("del_dir_"):
|
|
self._forget_library_directory()
|
|
elif self.mode.startswith("prop@"):
|
|
self._toggle_material_property()
|
|
elif self.mode == "view_more":
|
|
do_update = self._view_more()
|
|
elif self.mode.startswith("asset_filter_"):
|
|
do_update = self._set_area()
|
|
else:
|
|
reporting.capture_message("invalid_setting_mode", self.mode)
|
|
self.report(
|
|
{"WARNING"}, _t("Invalid setting mode {0}").format(self.mode))
|
|
return {'CANCELLED'}
|
|
|
|
self._do_clear_cache(clear_cache)
|
|
self._do_update(do_update)
|
|
save_settings(cTB)
|
|
return {"FINISHED"}
|