Compare commits
214 Commits
version-1.
...
version-1.
Author | SHA1 | Date | |
---|---|---|---|
e32e75e3db | |||
6fa5ab5481 | |||
379580de86 | |||
db30b3df76 | |||
5de99baaef | |||
2184b39d27 | |||
23b1f7de7d | |||
28f68c6fbf | |||
b00cb233cc | |||
2142e9e7fc | |||
1dea802932 | |||
077bd1abdb | |||
5a2c528681 | |||
53b12376d1 | |||
8495868ea6 | |||
cf810de41b | |||
c457767edf | |||
985b3f6a7d | |||
a45bf3cd5c | |||
3789742cc8 | |||
58f374e175 | |||
99e90e1008 | |||
dd83d3ee60 | |||
e74e014c66 | |||
01541f181e | |||
a69f4d3fd9 | |||
3ffea46f23 | |||
94c5811e42 | |||
676ad1ed14 | |||
79e6fa37f4 | |||
e06fa3ea75 | |||
fb6352dc7d | |||
97ad8bf5ba | |||
b0f7719add | |||
dada275e32 | |||
6bce1ccf90 | |||
bbe524c099 | |||
462da038ec | |||
8d7799655e | |||
cb0393868e | |||
5a61a7a6c4 | |||
60d1fbff50 | |||
352fe239f2 | |||
09c1bf67b4 | |||
23235afe71 | |||
ff9624d4f3 | |||
48c60f73d7 | |||
12eaaa5bae | |||
f7396350db | |||
cc97288018 | |||
26105add9c | |||
ea81cc5769 | |||
25b6053836 | |||
65a05403dc | |||
770b0121fa | |||
2b155eac45 | |||
d36959e91b | |||
9028c38c68 | |||
f04f07eaf1 | |||
6c38a432bc | |||
53fa3e628a | |||
924fb45cb2 | |||
b5619757bc | |||
ae41745743 | |||
ffab83f921 | |||
8bef2f48a5 | |||
74b46ff0db | |||
e1934b20d9 | |||
0caf761863 | |||
bc864737ae | |||
f454a99c4b | |||
40732e0487 | |||
b86bffbdbb | |||
67f9d40fd3 | |||
c4de4e9990 | |||
6d2e6efa13 | |||
ff9ae0117d | |||
974d33e3a3 | |||
8de3a0bba2 | |||
6f705b917f | |||
02b694f5d4 | |||
663ebae572 | |||
cb5a116dff | |||
5821611d89 | |||
8bd1faa575 | |||
8899bff5e4 | |||
4fd4ad7448 | |||
4f32b49ad3 | |||
1f13b4d249 | |||
ef57dba5d3 | |||
419249ee19 | |||
113eb8f7ab | |||
85f911cb59 | |||
564c2589b1 | |||
80155ed4f4 | |||
d8c5c4eecd | |||
3972ce4543 | |||
d75a055149 | |||
649542daad | |||
1d99751d20 | |||
69028e0cfd | |||
dc7ad296bf | |||
3f2479067c | |||
6fefe4ffd8 | |||
62c1c966f6 | |||
57aadc1817 | |||
7204d4a24c | |||
641b51496a | |||
0562d57513 | |||
ac19e48895 | |||
73d96e5c89 | |||
4bfdac223a | |||
5d6777c74b | |||
f4322f1d1f | |||
13a8595cc0 | |||
af413059b0 | |||
4d26ad248e | |||
d019fd0cf0 | |||
fb9ffbbc23 | |||
f6d797512a | |||
8367abeeb9 | |||
2f5f82b1a8 | |||
a04137ec6a | |||
87c90a7f72 | |||
4de8122920 | |||
21d2257be0 | |||
bc4036573c | |||
87cf1e12fa | |||
b35d7bc5f3 | |||
973dafcc3a | |||
62d16fff35 | |||
ed3de414ef | |||
b0a03c81f5 | |||
99f0764986 | |||
f9c2dda9fa | |||
0a7dea568a | |||
40c31e8be2 | |||
394395a7f5 | |||
f1478bf3d9 | |||
2fce27f8cb | |||
59e6491110 | |||
afca7abe18 | |||
4aae107396 | |||
096a5f5803 | |||
79dc5c91f7 | |||
0a99b9e22e | |||
0452fd845b | |||
c0a8602e17 | |||
10c51b3af5 | |||
0be5c16926 | |||
4158c4eed5 | |||
de4a93de98 | |||
331e9e6ca0 | |||
1d81f4bc38 | |||
5f58f8b6f7 | |||
164f65f30c | |||
b82bc14fcf | |||
9e5dcd0b55 | |||
531ddad8f5 | |||
7e0dd0384d | |||
da0c8dd944 | |||
8065ab88a4 | |||
6baf43e53b | |||
f1fa273370 | |||
bf96638c88 | |||
bc8a985228 | |||
ba14c33b6d | |||
0a7e7195a2 | |||
ecab0f6163 | |||
3c91ccced6 | |||
c9ed6c7d23 | |||
5fa01daf9e | |||
77664fb6d7 | |||
45cffc5365 | |||
fb5433d473 | |||
a17fe45712 | |||
1bfba64bdc | |||
cdb4bf4f4f | |||
15254b8951 | |||
3ed5f2c187 | |||
0be3bf7f49 | |||
f207e14664 | |||
9932003400 | |||
e7035e6f0c | |||
014a36d24e | |||
068451a7aa | |||
56fb1ec3df | |||
e93094cb88 | |||
33718a1a35 | |||
db82dbe730 | |||
8d405330ee | |||
66ddc7b47b | |||
2fa8cb4054 | |||
e7b5c75046 | |||
1d93bd9e5e | |||
ac2d0c033c | |||
61fa63eb1d | |||
7022412889 | |||
b4f71745b0 | |||
1d41fce1ae | |||
e636fde4ce | |||
82a9dc5226 | |||
1f40915ac8 | |||
32693c0f64 | |||
c38748eb05 | |||
ac85bea111 | |||
7b5613ce77 | |||
ec5f317dac | |||
a51f61d9b5 | |||
13bc9a89c8 | |||
996b722813 | |||
e7f2567bfc | |||
ff8e71c542 | |||
543da5c8d8 |
160
CHANGELOG.md
160
CHANGELOG.md
@@ -1,9 +1,167 @@
|
|||||||
# Blender Cloud changelog
|
# Blender Cloud changelog
|
||||||
|
|
||||||
|
## Version 1.15 (2019-12-12)
|
||||||
|
|
||||||
|
- Avoid creating BAT pack when the to-be-rendered file is already inside the job storage
|
||||||
|
directory. This assumes that the paths are already correct for the Flamenco Workers.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.14 (2019-10-10)
|
||||||
|
|
||||||
|
- Upgraded BAT to 1.2 for missing smoke caches, compatibility with Blender 2.81, and some
|
||||||
|
Windows-specific fixes.
|
||||||
|
- Removed warnings on the terminal when running Blender 2.80+
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.13 (2019-04-18)
|
||||||
|
|
||||||
|
- Upgraded BAT to 1.1.1 for a compatibility fix with Blender 2.79
|
||||||
|
- Flamenco: Support for Flamenco Manager settings versioning + for settings version 2.
|
||||||
|
When using Blender Cloud Add-on 1.12 or older, Flamenco Server will automatically convert the
|
||||||
|
Manager settings to version 1.
|
||||||
|
- More Blender 2.80 compatibility fixes
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.12 (2019-03-25)
|
||||||
|
|
||||||
|
- Flamenco: Change how progressive render tasks are created. Instead of the artist setting a fixed
|
||||||
|
number of sample chunks, they can now set a maximum number of samples for each render task.
|
||||||
|
Initial render tasks are created with a low number of samples, and subsequent tasks have an
|
||||||
|
increasing number of samples, up to the set maximum. The total number of samples of the final
|
||||||
|
render is still equal to the number of samples configured in the blend file.
|
||||||
|
Requires Flamenco Server 2.2 or newer.
|
||||||
|
- Flamenco: Added a hidden "Submit & Quit" button. This button can be enabled in the add-on
|
||||||
|
preferences and and then be available on the Flamenco Render panel. Pressing the button will
|
||||||
|
silently close Blender after the job has been submitted to Flamenco (for example to click,
|
||||||
|
walk away, and free up memory for when the same machine is part of the render farm).
|
||||||
|
- Flamenco: Name render jobs just 'thefile' instead of 'Render thefile.flamenco.blend'.
|
||||||
|
This makes the job overview on Flamenco Server cleaner.
|
||||||
|
- Flamenco: support Shaman servers. See https://www.flamenco.io/docs/user_manual/shaman/
|
||||||
|
for more info.
|
||||||
|
- Flamenco: The 'blender-video-chunks' job type now also allows MP4 and MOV video containers.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.11.1 (2019-01-04)
|
||||||
|
|
||||||
|
- Bundled missing Texture Browser icons.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.11.0 (2019-01-04)
|
||||||
|
|
||||||
|
- Texture Browser now works on Blender 2.8.
|
||||||
|
- Blender Sync: Fixed compatibility issue with Blender 2.8.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.10.0 (2019-01-02)
|
||||||
|
|
||||||
|
- Bundles Blender-Asset-Tracer 0.8.
|
||||||
|
- Fix crashing Blender when running in background mode (e.g. without GUI).
|
||||||
|
- Flamenco: Include extra job parameters to allow for encoding a video at the end of a render
|
||||||
|
job that produced an image sequence.
|
||||||
|
- Flamenco: Compress all blend files, and not just the one we save from Blender.
|
||||||
|
- Flamenco: Store more info in the `jobinfo.json` file. This is mostly useful for debugging issues
|
||||||
|
on the render farm, as now things like the exclusion filter and Manager settings are logged too.
|
||||||
|
- Flamenco: Allow BAT-packing of only those assets that are referred to by relative path (e.g.
|
||||||
|
a path starting with `//`). Assets with an absolute path are ignored, and assumed to be reachable
|
||||||
|
at the same path by the Workers.
|
||||||
|
- Flamenco: Added 'blender-video-chunks' job type, meant for rendering the edit of a film from the
|
||||||
|
VSE. This job type requires that the file is configured for rendering to Matroska video
|
||||||
|
files.
|
||||||
|
|
||||||
|
Audio is only extracted when there is an audio codec configured. This is a bit arbitrary, but it's
|
||||||
|
at least a way to tell whether the artist is considering that there is audio of any relevance in
|
||||||
|
the current blend file.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.9.4 (2018-11-01)
|
||||||
|
|
||||||
|
- Fixed Python 3.6 and Blender 2.79b incompatibilities accidentally introduced in 1.9.3.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.9.3 (2018-10-30)
|
||||||
|
|
||||||
|
- Fix drawing of Attract strips in the VSE on Blender 2.8.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.9.2 (2018-09-17)
|
||||||
|
|
||||||
|
- No changes, just a different filename to force a refresh on our
|
||||||
|
hosting platform.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.9.1 (2018-09-17)
|
||||||
|
|
||||||
|
- Fix issue with Python 3.7, which is used by current daily builds of Blender.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.9 (2018-09-05)
|
||||||
|
|
||||||
|
- Last version to support Blender versions before 2.80!
|
||||||
|
- Replace BAM with BAT🦇.
|
||||||
|
- Don't crash the texture browser when an invalid texture is seen.
|
||||||
|
- Support colour strips as Attract shots.
|
||||||
|
- Flamenco: allow jobs to be created in 'paused' state.
|
||||||
|
- Flamenco: only show Flamenco Managers that are linked to the currently selected project.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.8 (2018-01-03)
|
||||||
|
|
||||||
|
- Distinguish between 'please subscribe' (to get a new subscription) and 'please renew' (to renew an
|
||||||
|
existing subscription).
|
||||||
|
- When re-opening the Texture Browser it now opens in the same folder as where it was when closed.
|
||||||
|
- In the texture browser, draw the components of the texture (i.e. which map types are available),
|
||||||
|
such as 'bump, normal, specular'.
|
||||||
|
- Use Interface Scale setting from user preferences to draw the Texture Browser text.
|
||||||
|
- Store project-specific settings in the preferences, such as filesystem paths, for each project,
|
||||||
|
and restore those settings when the project is selected again. Does not touch settings that
|
||||||
|
haven't been set for the newly selected project. These settings are only saved when a setting
|
||||||
|
is updated, so to save your current settings need to update a single setting; this saves all
|
||||||
|
settings for the project.
|
||||||
|
- Added button in the User Preferences to open a Cloud project in your webbrowser.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.7.5 (2017-10-06)
|
||||||
|
|
||||||
|
- Sorting the project list alphabetically.
|
||||||
|
- Renamed 'Job File Path' to 'Job Storage Path' so it's more explicit.
|
||||||
|
- Allow overriding the render output path on a per-scene basis.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.7.4 (2017-09-05)
|
||||||
|
|
||||||
|
- Fix [T52621](https://developer.blender.org/T52621): Fixed class name collision upon add-on
|
||||||
|
registration. This is checked since Blender 2.79.
|
||||||
|
- Fix [T48852](https://developer.blender.org/T48852): Screenshot no longer shows "Communicating with
|
||||||
|
Blender Cloud".
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.7.3 (2017-08-08)
|
||||||
|
|
||||||
|
- Default to scene frame range when no frame range is given.
|
||||||
|
- Refuse to render on Flamenco before blend file is saved at least once.
|
||||||
|
- Fixed some Windows-specific issues.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.7.2 (2017-06-22)
|
||||||
|
|
||||||
|
- Fixed compatibility with Blender 2.78c.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.7.1 (2017-06-13)
|
||||||
|
|
||||||
|
- Fixed asyncio issues on Windows
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.7.0 (2017-06-09)
|
||||||
|
|
||||||
|
- Fixed reloading after upgrading from 1.4.4 (our last public release).
|
||||||
|
- Fixed bug handling a symlinked project path.
|
||||||
|
- Added support for Manager-defined path replacement variables.
|
||||||
|
|
||||||
|
|
||||||
## Version 1.6.4 (2017-04-21)
|
## Version 1.6.4 (2017-04-21)
|
||||||
|
|
||||||
- Added file exclusion filter for Flamenco. A filter like "*.abc;*.mkv;*.mov" can be
|
- Added file exclusion filter for Flamenco. A filter like `*.abc;*.mkv;*.mov` can be
|
||||||
used to prevent certain files from being copied to the job storage directory.
|
used to prevent certain files from being copied to the job storage directory.
|
||||||
Requires a Blender that is bundled with BAM 1.1.7 or newer.
|
Requires a Blender that is bundled with BAM 1.1.7 or newer.
|
||||||
|
|
||||||
|
@@ -21,8 +21,8 @@
|
|||||||
bl_info = {
|
bl_info = {
|
||||||
'name': 'Blender Cloud',
|
'name': 'Blender Cloud',
|
||||||
"author": "Sybren A. Stüvel, Francesco Siddi, Inês Almeida, Antony Riakiotakis",
|
"author": "Sybren A. Stüvel, Francesco Siddi, Inês Almeida, Antony Riakiotakis",
|
||||||
'version': (1, 6, 4),
|
'version': (1, 15),
|
||||||
'blender': (2, 77, 0),
|
'blender': (2, 80, 0),
|
||||||
'location': 'Addon Preferences panel, and Ctrl+Shift+Alt+A anywhere for texture browser',
|
'location': 'Addon Preferences panel, and Ctrl+Shift+Alt+A anywhere for texture browser',
|
||||||
'description': 'Texture library browser and Blender Sync. Requires the Blender ID addon '
|
'description': 'Texture library browser and Blender Sync. Requires the Blender ID addon '
|
||||||
'and Blender 2.77a or newer.',
|
'and Blender 2.77a or newer.',
|
||||||
@@ -65,36 +65,45 @@ def register():
|
|||||||
|
|
||||||
def reload_mod(name):
|
def reload_mod(name):
|
||||||
modname = '%s.%s' % (__name__, name)
|
modname = '%s.%s' % (__name__, name)
|
||||||
module = importlib.reload(sys.modules[modname])
|
try:
|
||||||
sys.modules[modname] = module
|
old_module = sys.modules[modname]
|
||||||
return module
|
except KeyError:
|
||||||
|
# Wasn't loaded before -- can happen after an upgrade.
|
||||||
|
new_module = importlib.import_module(modname)
|
||||||
|
else:
|
||||||
|
new_module = importlib.reload(old_module)
|
||||||
|
|
||||||
|
sys.modules[modname] = new_module
|
||||||
|
return new_module
|
||||||
|
|
||||||
reload_mod('blendfile')
|
reload_mod('blendfile')
|
||||||
reload_mod('home_project')
|
reload_mod('home_project')
|
||||||
reload_mod('utils')
|
reload_mod('utils')
|
||||||
|
reload_mod('pillar')
|
||||||
|
|
||||||
blender = reload_mod('blender')
|
|
||||||
async_loop = reload_mod('async_loop')
|
async_loop = reload_mod('async_loop')
|
||||||
|
flamenco = reload_mod('flamenco')
|
||||||
|
attract = reload_mod('attract')
|
||||||
texture_browser = reload_mod('texture_browser')
|
texture_browser = reload_mod('texture_browser')
|
||||||
settings_sync = reload_mod('settings_sync')
|
settings_sync = reload_mod('settings_sync')
|
||||||
image_sharing = reload_mod('image_sharing')
|
image_sharing = reload_mod('image_sharing')
|
||||||
attract = reload_mod('attract')
|
blender = reload_mod('blender')
|
||||||
flamenco = reload_mod('flamenco')
|
project_specific = reload_mod('project_specific')
|
||||||
else:
|
else:
|
||||||
from . import (blender, texture_browser, async_loop, settings_sync, blendfile, home_project,
|
from . import (blender, texture_browser, async_loop, settings_sync, blendfile, home_project,
|
||||||
image_sharing, attract, flamenco)
|
image_sharing, attract, flamenco, project_specific)
|
||||||
|
|
||||||
async_loop.setup_asyncio_executor()
|
async_loop.setup_asyncio_executor()
|
||||||
async_loop.register()
|
async_loop.register()
|
||||||
|
|
||||||
flamenco.register()
|
flamenco.register()
|
||||||
|
attract.register()
|
||||||
texture_browser.register()
|
texture_browser.register()
|
||||||
blender.register()
|
|
||||||
settings_sync.register()
|
settings_sync.register()
|
||||||
image_sharing.register()
|
image_sharing.register()
|
||||||
attract.register()
|
blender.register()
|
||||||
|
|
||||||
blender.handle_project_update()
|
project_specific.handle_project_update()
|
||||||
|
|
||||||
|
|
||||||
def _monkey_patch_requests():
|
def _monkey_patch_requests():
|
||||||
|
@@ -511,7 +511,7 @@ if system == "win32":
|
|||||||
_get_win_folder = _get_win_folder_with_pywin32
|
_get_win_folder = _get_win_folder_with_pywin32
|
||||||
except ImportError:
|
except ImportError:
|
||||||
try:
|
try:
|
||||||
from ctypes import windll
|
from ctypes import windll # type: ignore
|
||||||
_get_win_folder = _get_win_folder_with_ctypes
|
_get_win_folder = _get_win_folder_with_ctypes
|
||||||
except ImportError:
|
except ImportError:
|
||||||
try:
|
try:
|
||||||
|
@@ -23,6 +23,7 @@ import traceback
|
|||||||
import concurrent.futures
|
import concurrent.futures
|
||||||
import logging
|
import logging
|
||||||
import gc
|
import gc
|
||||||
|
import typing
|
||||||
|
|
||||||
import bpy
|
import bpy
|
||||||
|
|
||||||
@@ -33,17 +34,12 @@ _loop_kicking_operator_running = False
|
|||||||
|
|
||||||
|
|
||||||
def setup_asyncio_executor():
|
def setup_asyncio_executor():
|
||||||
"""Sets up AsyncIO to run on a single thread.
|
"""Sets up AsyncIO to run properly on each platform."""
|
||||||
|
|
||||||
This ensures that only one Pillar HTTP call is performed at the same time. Other
|
|
||||||
calls that could be performed in parallel are queued, and thus we can
|
|
||||||
reliably cancel them.
|
|
||||||
"""
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
executor = concurrent.futures.ThreadPoolExecutor()
|
|
||||||
|
|
||||||
if sys.platform == 'win32':
|
if sys.platform == 'win32':
|
||||||
|
asyncio.get_event_loop().close()
|
||||||
# On Windows, the default event loop is SelectorEventLoop, which does
|
# On Windows, the default event loop is SelectorEventLoop, which does
|
||||||
# not support subprocesses. ProactorEventLoop should be used instead.
|
# not support subprocesses. ProactorEventLoop should be used instead.
|
||||||
# Source: https://docs.python.org/3/library/asyncio-subprocess.html
|
# Source: https://docs.python.org/3/library/asyncio-subprocess.html
|
||||||
@@ -51,9 +47,15 @@ def setup_asyncio_executor():
|
|||||||
asyncio.set_event_loop(loop)
|
asyncio.set_event_loop(loop)
|
||||||
else:
|
else:
|
||||||
loop = asyncio.get_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
|
|
||||||
|
executor = concurrent.futures.ThreadPoolExecutor(max_workers=10)
|
||||||
loop.set_default_executor(executor)
|
loop.set_default_executor(executor)
|
||||||
# loop.set_debug(True)
|
# loop.set_debug(True)
|
||||||
|
|
||||||
|
from . import pillar
|
||||||
|
# No more than this many Pillar calls should be made simultaneously
|
||||||
|
pillar.pillar_semaphore = asyncio.Semaphore(3, loop=loop)
|
||||||
|
|
||||||
|
|
||||||
def kick_async_loop(*args) -> bool:
|
def kick_async_loop(*args) -> bool:
|
||||||
"""Performs a single iteration of the asyncio event loop.
|
"""Performs a single iteration of the asyncio event loop.
|
||||||
@@ -152,7 +154,7 @@ class AsyncLoopModalOperator(bpy.types.Operator):
|
|||||||
_loop_kicking_operator_running = True
|
_loop_kicking_operator_running = True
|
||||||
|
|
||||||
wm = context.window_manager
|
wm = context.window_manager
|
||||||
self.timer = wm.event_timer_add(0.00001, context.window)
|
self.timer = wm.event_timer_add(0.00001, window=context.window)
|
||||||
|
|
||||||
return {'RUNNING_MODAL'}
|
return {'RUNNING_MODAL'}
|
||||||
|
|
||||||
@@ -191,7 +193,7 @@ class AsyncModalOperatorMixin:
|
|||||||
|
|
||||||
def invoke(self, context, event):
|
def invoke(self, context, event):
|
||||||
context.window_manager.modal_handler_add(self)
|
context.window_manager.modal_handler_add(self)
|
||||||
self.timer = context.window_manager.event_timer_add(1 / 15, context.window)
|
self.timer = context.window_manager.event_timer_add(1 / 15, window=context.window)
|
||||||
|
|
||||||
self.log.info('Starting')
|
self.log.info('Starting')
|
||||||
self._new_async_task(self.async_execute(context))
|
self._new_async_task(self.async_execute(context))
|
||||||
@@ -237,7 +239,7 @@ class AsyncModalOperatorMixin:
|
|||||||
self._stop_async_task()
|
self._stop_async_task()
|
||||||
context.window_manager.event_timer_remove(self.timer)
|
context.window_manager.event_timer_remove(self.timer)
|
||||||
|
|
||||||
def _new_async_task(self, async_task: asyncio.coroutine, future: asyncio.Future = None):
|
def _new_async_task(self, async_task: typing.Coroutine, future: asyncio.Future = None):
|
||||||
"""Stops the currently running async task, and starts another one."""
|
"""Stops the currently running async task, and starts another one."""
|
||||||
|
|
||||||
self.log.debug('Setting up a new task %r, so any existing task must be stopped', async_task)
|
self.log.debug('Setting up a new task %r, so any existing task must be stopped', async_task)
|
||||||
|
@@ -46,9 +46,16 @@ if "bpy" in locals():
|
|||||||
draw = importlib.reload(draw)
|
draw = importlib.reload(draw)
|
||||||
pillar = importlib.reload(pillar)
|
pillar = importlib.reload(pillar)
|
||||||
async_loop = importlib.reload(async_loop)
|
async_loop = importlib.reload(async_loop)
|
||||||
|
blender = importlib.reload(blender)
|
||||||
|
compatibility = importlib.reload(compatibility)
|
||||||
|
else:
|
||||||
|
import bpy
|
||||||
|
|
||||||
|
if bpy.app.version < (2, 80):
|
||||||
|
from . import draw_27 as draw
|
||||||
else:
|
else:
|
||||||
from . import draw
|
from . import draw
|
||||||
from .. import pillar, async_loop
|
from .. import pillar, async_loop, blender, compatibility
|
||||||
|
|
||||||
import bpy
|
import bpy
|
||||||
import pillarsdk
|
import pillarsdk
|
||||||
@@ -57,6 +64,7 @@ from pillarsdk.projects import Project
|
|||||||
from pillarsdk import exceptions as sdk_exceptions
|
from pillarsdk import exceptions as sdk_exceptions
|
||||||
|
|
||||||
from bpy.types import Operator, Panel, AddonPreferences
|
from bpy.types import Operator, Panel, AddonPreferences
|
||||||
|
import bl_ui.space_sequencer
|
||||||
|
|
||||||
log = logging.getLogger(__name__)
|
log = logging.getLogger(__name__)
|
||||||
|
|
||||||
@@ -173,10 +181,11 @@ class AttractPollMixin:
|
|||||||
return attract_is_active
|
return attract_is_active
|
||||||
|
|
||||||
|
|
||||||
class ToolsPanel(AttractPollMixin, Panel):
|
class ATTRACT_PT_tools(AttractPollMixin, Panel):
|
||||||
bl_label = 'Attract'
|
bl_label = 'Attract'
|
||||||
bl_space_type = 'SEQUENCE_EDITOR'
|
bl_space_type = 'SEQUENCE_EDITOR'
|
||||||
bl_region_type = 'UI'
|
bl_region_type = 'UI'
|
||||||
|
bl_category = 'Strip'
|
||||||
|
|
||||||
def draw_header(self, context):
|
def draw_header(self, context):
|
||||||
strip = active_strip(context)
|
strip = active_strip(context)
|
||||||
@@ -186,30 +195,44 @@ class ToolsPanel(AttractPollMixin, Panel):
|
|||||||
def draw(self, context):
|
def draw(self, context):
|
||||||
strip = active_strip(context)
|
strip = active_strip(context)
|
||||||
layout = self.layout
|
layout = self.layout
|
||||||
strip_types = {'MOVIE', 'IMAGE', 'META'}
|
strip_types = {'MOVIE', 'IMAGE', 'META', 'COLOR'}
|
||||||
|
|
||||||
selshots = list(selected_shots(context))
|
|
||||||
if strip and strip.type in strip_types and strip.atc_object_id:
|
if strip and strip.type in strip_types and strip.atc_object_id:
|
||||||
|
self._draw_attractstrip_buttons(context, strip)
|
||||||
|
elif context.selected_sequences:
|
||||||
|
if len(context.selected_sequences) > 1:
|
||||||
|
noun = 'Selected Strips'
|
||||||
|
else:
|
||||||
|
noun = 'This Strip'
|
||||||
|
layout.operator(ATTRACT_OT_submit_selected.bl_idname,
|
||||||
|
text='Submit %s as New Shot' % noun)
|
||||||
|
layout.operator('attract.shot_relink')
|
||||||
|
else:
|
||||||
|
layout.operator(ATTRACT_OT_submit_all.bl_idname)
|
||||||
|
layout.operator(ATTRACT_OT_project_open_in_browser.bl_idname, icon='WORLD')
|
||||||
|
|
||||||
|
def _draw_attractstrip_buttons(self, context, strip):
|
||||||
|
"""Draw buttons when selected strips are Attract shots."""
|
||||||
|
|
||||||
|
layout = self.layout
|
||||||
|
selshots = list(selected_shots(context))
|
||||||
|
|
||||||
if len(selshots) > 1:
|
if len(selshots) > 1:
|
||||||
noun = '%i Shots' % len(selshots)
|
noun = '%i Shots' % len(selshots)
|
||||||
else:
|
else:
|
||||||
noun = 'This Shot'
|
noun = 'This Shot'
|
||||||
|
|
||||||
if strip.atc_object_id_conflict:
|
if strip.atc_object_id_conflict:
|
||||||
warnbox = layout.box()
|
warnbox = layout.box()
|
||||||
warnbox.alert = True
|
warnbox.alert = True
|
||||||
warnbox.label('Warning: This shot is linked to multiple sequencer strips.',
|
warnbox.label(text='Warning: This shot is linked to multiple sequencer strips.',
|
||||||
icon='ERROR')
|
icon='ERROR')
|
||||||
|
|
||||||
layout.prop(strip, 'atc_name', text='Name')
|
layout.prop(strip, 'atc_name', text='Name')
|
||||||
layout.prop(strip, 'atc_status', text='Status')
|
layout.prop(strip, 'atc_status', text='Status')
|
||||||
|
|
||||||
# Create a special sub-layout for read-only properties.
|
# Create a special sub-layout for read-only properties.
|
||||||
ro_sub = layout.column(align=True)
|
ro_sub = layout.column(align=True)
|
||||||
ro_sub.enabled = False
|
ro_sub.enabled = False
|
||||||
ro_sub.prop(strip, 'atc_description', text='Description')
|
ro_sub.prop(strip, 'atc_description', text='Description')
|
||||||
ro_sub.prop(strip, 'atc_notes', text='Notes')
|
ro_sub.prop(strip, 'atc_notes', text='Notes')
|
||||||
|
|
||||||
if strip.atc_is_synced:
|
if strip.atc_is_synced:
|
||||||
sub = layout.column(align=True)
|
sub = layout.column(align=True)
|
||||||
row = sub.row(align=True)
|
row = sub.row(align=True)
|
||||||
@@ -219,7 +242,7 @@ class ToolsPanel(AttractPollMixin, Panel):
|
|||||||
icon='TRIA_UP')
|
icon='TRIA_UP')
|
||||||
else:
|
else:
|
||||||
row.operator(ATTRACT_OT_submit_all.bl_idname)
|
row.operator(ATTRACT_OT_submit_all.bl_idname)
|
||||||
row.operator(AttractShotFetchUpdate.bl_idname,
|
row.operator(ATTRACT_OT_shot_fetch_update.bl_idname,
|
||||||
text='', icon='FILE_REFRESH')
|
text='', icon='FILE_REFRESH')
|
||||||
row.operator(ATTRACT_OT_shot_open_in_browser.bl_idname,
|
row.operator(ATTRACT_OT_shot_open_in_browser.bl_idname,
|
||||||
text='', icon='WORLD')
|
text='', icon='WORLD')
|
||||||
@@ -230,25 +253,14 @@ class ToolsPanel(AttractPollMixin, Panel):
|
|||||||
icon='RENDER_STILL')
|
icon='RENDER_STILL')
|
||||||
|
|
||||||
# Group more dangerous operations.
|
# Group more dangerous operations.
|
||||||
dangerous_sub = layout.split(0.6, align=True)
|
dangerous_sub = layout.split(**compatibility.factor(0.6), align=True)
|
||||||
dangerous_sub.operator('attract.strip_unlink',
|
dangerous_sub.operator('attract.strip_unlink',
|
||||||
text='Unlink %s' % noun,
|
text='Unlink %s' % noun,
|
||||||
icon='PANEL_CLOSE')
|
icon='PANEL_CLOSE')
|
||||||
dangerous_sub.operator(AttractShotDelete.bl_idname,
|
dangerous_sub.operator(ATTRACT_OT_shot_delete.bl_idname,
|
||||||
text='Delete %s' % noun,
|
text='Delete %s' % noun,
|
||||||
icon='CANCEL')
|
icon='CANCEL')
|
||||||
|
|
||||||
elif context.selected_sequences:
|
|
||||||
if len(context.selected_sequences) > 1:
|
|
||||||
noun = 'Selected Strips'
|
|
||||||
else:
|
|
||||||
noun = 'This Strip'
|
|
||||||
layout.operator(AttractShotSubmitSelected.bl_idname,
|
|
||||||
text='Submit %s as New Shot' % noun)
|
|
||||||
layout.operator('attract.shot_relink')
|
|
||||||
else:
|
|
||||||
layout.operator(ATTRACT_OT_submit_all.bl_idname)
|
|
||||||
|
|
||||||
|
|
||||||
class AttractOperatorMixin(AttractPollMixin):
|
class AttractOperatorMixin(AttractPollMixin):
|
||||||
"""Mix-in class for all Attract operators."""
|
"""Mix-in class for all Attract operators."""
|
||||||
@@ -378,7 +390,7 @@ class AttractOperatorMixin(AttractPollMixin):
|
|||||||
draw.tag_redraw_all_sequencer_editors()
|
draw.tag_redraw_all_sequencer_editors()
|
||||||
|
|
||||||
|
|
||||||
class AttractShotFetchUpdate(AttractOperatorMixin, Operator):
|
class ATTRACT_OT_shot_fetch_update(AttractOperatorMixin, Operator):
|
||||||
bl_idname = "attract.shot_fetch_update"
|
bl_idname = "attract.shot_fetch_update"
|
||||||
bl_label = "Fetch Update From Attract"
|
bl_label = "Fetch Update From Attract"
|
||||||
bl_description = 'Update status, description & notes from Attract'
|
bl_description = 'Update status, description & notes from Attract'
|
||||||
@@ -397,7 +409,8 @@ class AttractShotFetchUpdate(AttractOperatorMixin, Operator):
|
|||||||
return {'FINISHED'}
|
return {'FINISHED'}
|
||||||
|
|
||||||
|
|
||||||
class AttractShotRelink(AttractShotFetchUpdate):
|
@compatibility.convert_properties
|
||||||
|
class ATTRACT_OT_shot_relink(AttractOperatorMixin, Operator):
|
||||||
bl_idname = "attract.shot_relink"
|
bl_idname = "attract.shot_relink"
|
||||||
bl_label = "Relink With Attract"
|
bl_label = "Relink With Attract"
|
||||||
|
|
||||||
@@ -466,7 +479,8 @@ class ATTRACT_OT_shot_open_in_browser(AttractOperatorMixin, Operator):
|
|||||||
return {'FINISHED'}
|
return {'FINISHED'}
|
||||||
|
|
||||||
|
|
||||||
class AttractShotDelete(AttractOperatorMixin, Operator):
|
@compatibility.convert_properties
|
||||||
|
class ATTRACT_OT_shot_delete(AttractOperatorMixin, Operator):
|
||||||
bl_idname = 'attract.shot_delete'
|
bl_idname = 'attract.shot_delete'
|
||||||
bl_label = 'Delete Shot'
|
bl_label = 'Delete Shot'
|
||||||
bl_description = 'Remove this shot from Attract'
|
bl_description = 'Remove this shot from Attract'
|
||||||
@@ -521,7 +535,7 @@ class AttractShotDelete(AttractOperatorMixin, Operator):
|
|||||||
col.prop(self, 'confirm', text="I hereby confirm: delete %s from The Edit." % noun)
|
col.prop(self, 'confirm', text="I hereby confirm: delete %s from The Edit." % noun)
|
||||||
|
|
||||||
|
|
||||||
class AttractStripUnlink(AttractOperatorMixin, Operator):
|
class ATTRACT_OT_strip_unlink(AttractOperatorMixin, Operator):
|
||||||
bl_idname = 'attract.strip_unlink'
|
bl_idname = 'attract.strip_unlink'
|
||||||
bl_label = 'Unlink Shot From This Strip'
|
bl_label = 'Unlink Shot From This Strip'
|
||||||
bl_description = 'Remove Attract props from the selected strip(s)'
|
bl_description = 'Remove Attract props from the selected strip(s)'
|
||||||
@@ -565,7 +579,7 @@ class AttractStripUnlink(AttractOperatorMixin, Operator):
|
|||||||
return {'FINISHED'}
|
return {'FINISHED'}
|
||||||
|
|
||||||
|
|
||||||
class AttractShotSubmitSelected(AttractOperatorMixin, Operator):
|
class ATTRACT_OT_submit_selected(AttractOperatorMixin, Operator):
|
||||||
bl_idname = 'attract.submit_selected'
|
bl_idname = 'attract.submit_selected'
|
||||||
bl_label = 'Submit All Selected'
|
bl_label = 'Submit All Selected'
|
||||||
bl_description = 'Submits all selected strips to Attract'
|
bl_description = 'Submits all selected strips to Attract'
|
||||||
@@ -629,8 +643,7 @@ class ATTRACT_OT_open_meta_blendfile(AttractOperatorMixin, Operator):
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def poll(cls, context):
|
def poll(cls, context):
|
||||||
return AttractOperatorMixin.poll(context) and \
|
return bool(any(cls.filename_from_metadata(s) for s in context.selected_sequences))
|
||||||
bool(any(cls.filename_from_metadata(s) for s in context.selected_sequences))
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def filename_from_metadata(strip):
|
def filename_from_metadata(strip):
|
||||||
@@ -905,7 +918,45 @@ class ATTRACT_OT_copy_id_to_clipboard(AttractOperatorMixin, Operator):
|
|||||||
return {'FINISHED'}
|
return {'FINISHED'}
|
||||||
|
|
||||||
|
|
||||||
def draw_strip_movie_meta(self, context):
|
@compatibility.convert_properties
|
||||||
|
class ATTRACT_OT_project_open_in_browser(Operator):
|
||||||
|
bl_idname = 'attract.project_open_in_browser'
|
||||||
|
bl_label = 'Open Project in Browser'
|
||||||
|
bl_description = 'Opens a webbrowser to show the project in Attract'
|
||||||
|
|
||||||
|
project_id = bpy.props.StringProperty(name='Project ID', default='')
|
||||||
|
|
||||||
|
def execute(self, context):
|
||||||
|
import webbrowser
|
||||||
|
import urllib.parse
|
||||||
|
|
||||||
|
import pillarsdk
|
||||||
|
from ..pillar import sync_call
|
||||||
|
from ..blender import PILLAR_WEB_SERVER_URL, preferences
|
||||||
|
|
||||||
|
if not self.project_id:
|
||||||
|
self.project_id = preferences().project.project
|
||||||
|
|
||||||
|
project = sync_call(pillarsdk.Project.find, self.project_id, {'projection': {'url': True}})
|
||||||
|
|
||||||
|
if log.isEnabledFor(logging.DEBUG):
|
||||||
|
import pprint
|
||||||
|
log.debug('found project: %s', pprint.pformat(project.to_dict()))
|
||||||
|
|
||||||
|
url = urllib.parse.urljoin(PILLAR_WEB_SERVER_URL, 'attract/' + project.url)
|
||||||
|
webbrowser.open_new_tab(url)
|
||||||
|
self.report({'INFO'}, 'Opened a browser at %s' % url)
|
||||||
|
|
||||||
|
return {'FINISHED'}
|
||||||
|
|
||||||
|
|
||||||
|
class ATTRACT_PT_strip_metadata(bl_ui.space_sequencer.SequencerButtonsPanel, Panel):
|
||||||
|
bl_label = "Metadata"
|
||||||
|
bl_parent_id = "SEQUENCER_PT_source"
|
||||||
|
bl_category = "Strip"
|
||||||
|
bl_options = {'DEFAULT_CLOSED'}
|
||||||
|
|
||||||
|
def draw(self, context):
|
||||||
strip = active_strip(context)
|
strip = active_strip(context)
|
||||||
if not strip:
|
if not strip:
|
||||||
return
|
return
|
||||||
@@ -918,12 +969,12 @@ def draw_strip_movie_meta(self, context):
|
|||||||
row = box.row(align=True)
|
row = box.row(align=True)
|
||||||
fname = meta.get('BLEND_FILE', None) or None
|
fname = meta.get('BLEND_FILE', None) or None
|
||||||
if fname:
|
if fname:
|
||||||
row.label('Original Blendfile: %s' % fname)
|
row.label(text='Original Blendfile: %s' % fname)
|
||||||
row.operator(ATTRACT_OT_open_meta_blendfile.bl_idname,
|
row.operator(ATTRACT_OT_open_meta_blendfile.bl_idname,
|
||||||
text='', icon='FILE_BLEND')
|
text='', icon='FILE_BLEND')
|
||||||
sfra = meta.get('START_FRAME', '?')
|
sfra = meta.get('START_FRAME', '?')
|
||||||
efra = meta.get('END_FRAME', '?')
|
efra = meta.get('END_FRAME', '?')
|
||||||
box.label('Original Frame Range: %s-%s' % (sfra, efra))
|
box.label(text='Original Frame Range: %s-%s' % (sfra, efra))
|
||||||
|
|
||||||
|
|
||||||
def activate():
|
def activate():
|
||||||
@@ -931,6 +982,9 @@ def activate():
|
|||||||
|
|
||||||
log.info('Activating Attract')
|
log.info('Activating Attract')
|
||||||
attract_is_active = True
|
attract_is_active = True
|
||||||
|
|
||||||
|
# TODO: properly fix 2.8 compatibility; this is just a workaround.
|
||||||
|
if hasattr(bpy.app.handlers, 'scene_update_post'):
|
||||||
bpy.app.handlers.scene_update_post.append(scene_update_post_handler)
|
bpy.app.handlers.scene_update_post.append(scene_update_post_handler)
|
||||||
draw.callback_enable()
|
draw.callback_enable()
|
||||||
|
|
||||||
@@ -942,6 +996,8 @@ def deactivate():
|
|||||||
attract_is_active = False
|
attract_is_active = False
|
||||||
draw.callback_disable()
|
draw.callback_disable()
|
||||||
|
|
||||||
|
# TODO: properly fix 2.8 compatibility; this is just a workaround.
|
||||||
|
if hasattr(bpy.app.handlers, 'scene_update_post'):
|
||||||
try:
|
try:
|
||||||
bpy.app.handlers.scene_update_post.remove(scene_update_post_handler)
|
bpy.app.handlers.scene_update_post.remove(scene_update_post_handler)
|
||||||
except ValueError:
|
except ValueError:
|
||||||
@@ -949,6 +1005,11 @@ def deactivate():
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
_rna_classes = [cls for cls in locals().values()
|
||||||
|
if isinstance(cls, type) and cls.__name__.startswith('ATTRACT')]
|
||||||
|
log.info('RNA classes:\n%s', '\n'.join([repr(cls) for cls in _rna_classes]))
|
||||||
|
|
||||||
|
|
||||||
def register():
|
def register():
|
||||||
bpy.types.Sequence.atc_is_synced = bpy.props.BoolProperty(name="Is Synced")
|
bpy.types.Sequence.atc_is_synced = bpy.props.BoolProperty(name="Is Synced")
|
||||||
bpy.types.Sequence.atc_object_id = bpy.props.StringProperty(name="Attract Object ID")
|
bpy.types.Sequence.atc_object_id = bpy.props.StringProperty(name="Attract Object ID")
|
||||||
@@ -972,24 +1033,18 @@ def register():
|
|||||||
name="Status")
|
name="Status")
|
||||||
bpy.types.Sequence.atc_order = bpy.props.IntProperty(name="Order")
|
bpy.types.Sequence.atc_order = bpy.props.IntProperty(name="Order")
|
||||||
|
|
||||||
bpy.types.SEQUENCER_PT_edit.append(draw_strip_movie_meta)
|
for cls in _rna_classes:
|
||||||
|
bpy.utils.register_class(cls)
|
||||||
bpy.utils.register_class(ToolsPanel)
|
|
||||||
bpy.utils.register_class(AttractShotRelink)
|
|
||||||
bpy.utils.register_class(AttractShotDelete)
|
|
||||||
bpy.utils.register_class(AttractStripUnlink)
|
|
||||||
bpy.utils.register_class(AttractShotFetchUpdate)
|
|
||||||
bpy.utils.register_class(AttractShotSubmitSelected)
|
|
||||||
bpy.utils.register_class(ATTRACT_OT_submit_all)
|
|
||||||
bpy.utils.register_class(ATTRACT_OT_open_meta_blendfile)
|
|
||||||
bpy.utils.register_class(ATTRACT_OT_shot_open_in_browser)
|
|
||||||
bpy.utils.register_class(ATTRACT_OT_make_shot_thumbnail)
|
|
||||||
bpy.utils.register_class(ATTRACT_OT_copy_id_to_clipboard)
|
|
||||||
|
|
||||||
|
|
||||||
def unregister():
|
def unregister():
|
||||||
deactivate()
|
deactivate()
|
||||||
bpy.utils.unregister_module(__name__)
|
for cls in _rna_classes:
|
||||||
|
try:
|
||||||
|
bpy.utils.unregister_class(cls)
|
||||||
|
except RuntimeError:
|
||||||
|
log.warning('Unable to unregister class %r, probably already unregistered', cls)
|
||||||
|
|
||||||
del bpy.types.Sequence.atc_is_synced
|
del bpy.types.Sequence.atc_is_synced
|
||||||
del bpy.types.Sequence.atc_object_id
|
del bpy.types.Sequence.atc_object_id
|
||||||
del bpy.types.Sequence.atc_object_id_conflict
|
del bpy.types.Sequence.atc_object_id_conflict
|
||||||
|
@@ -18,9 +18,12 @@
|
|||||||
|
|
||||||
# <pep8 compliant>
|
# <pep8 compliant>
|
||||||
|
|
||||||
import bpy
|
|
||||||
import logging
|
import logging
|
||||||
import collections
|
import typing
|
||||||
|
|
||||||
|
import bpy
|
||||||
|
import bgl
|
||||||
|
import gpu
|
||||||
|
|
||||||
log = logging.getLogger(__name__)
|
log = logging.getLogger(__name__)
|
||||||
|
|
||||||
@@ -34,10 +37,74 @@ strip_status_colour = {
|
|||||||
'todo': (1.0, 0.5019607843137255, 0.5019607843137255)
|
'todo': (1.0, 0.5019607843137255, 0.5019607843137255)
|
||||||
}
|
}
|
||||||
|
|
||||||
CONFLICT_COLOUR = (0.576, 0.118, 0.035) # RGB tuple
|
CONFLICT_COLOUR = (0.576, 0.118, 0.035, 1.0) # RGBA tuple
|
||||||
|
|
||||||
|
gpu_vertex_shader = '''
|
||||||
|
uniform mat4 ModelViewProjectionMatrix;
|
||||||
|
|
||||||
|
layout (location = 0) in vec2 pos;
|
||||||
|
layout (location = 1) in vec4 color;
|
||||||
|
|
||||||
|
out vec4 lineColor; // output to the fragment shader
|
||||||
|
|
||||||
|
void main()
|
||||||
|
{
|
||||||
|
gl_Position = ModelViewProjectionMatrix * vec4(pos.x, pos.y, 0.0, 1.0);
|
||||||
|
lineColor = color;
|
||||||
|
}
|
||||||
|
'''
|
||||||
|
|
||||||
|
gpu_fragment_shader = '''
|
||||||
|
out vec4 fragColor;
|
||||||
|
in vec4 lineColor;
|
||||||
|
|
||||||
|
void main()
|
||||||
|
{
|
||||||
|
fragColor = lineColor;
|
||||||
|
}
|
||||||
|
'''
|
||||||
|
|
||||||
|
Float2 = typing.Tuple[float, float]
|
||||||
|
Float3 = typing.Tuple[float, float, float]
|
||||||
|
Float4 = typing.Tuple[float, float, float, float]
|
||||||
|
|
||||||
|
|
||||||
def get_strip_rectf(strip):
|
class AttractLineDrawer:
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self._format = gpu.types.GPUVertFormat()
|
||||||
|
self._pos_id = self._format.attr_add(
|
||||||
|
id="pos",
|
||||||
|
comp_type="F32",
|
||||||
|
len=2,
|
||||||
|
fetch_mode="FLOAT")
|
||||||
|
self._color_id = self._format.attr_add(
|
||||||
|
id="color",
|
||||||
|
comp_type="F32",
|
||||||
|
len=4,
|
||||||
|
fetch_mode="FLOAT")
|
||||||
|
|
||||||
|
self.shader = gpu.types.GPUShader(gpu_vertex_shader, gpu_fragment_shader)
|
||||||
|
|
||||||
|
def draw(self,
|
||||||
|
coords: typing.List[Float2],
|
||||||
|
colors: typing.List[Float4]):
|
||||||
|
if not coords:
|
||||||
|
return
|
||||||
|
|
||||||
|
bgl.glEnable(bgl.GL_BLEND)
|
||||||
|
bgl.glLineWidth(2.0)
|
||||||
|
|
||||||
|
vbo = gpu.types.GPUVertBuf(len=len(coords), format=self._format)
|
||||||
|
vbo.attr_fill(id=self._pos_id, data=coords)
|
||||||
|
vbo.attr_fill(id=self._color_id, data=colors)
|
||||||
|
|
||||||
|
batch = gpu.types.GPUBatch(type="LINES", buf=vbo)
|
||||||
|
batch.program_set(self.shader)
|
||||||
|
batch.draw()
|
||||||
|
|
||||||
|
|
||||||
|
def get_strip_rectf(strip) -> Float4:
|
||||||
# Get x and y in terms of the grid's frames and channels
|
# Get x and y in terms of the grid's frames and channels
|
||||||
x1 = strip.frame_final_start
|
x1 = strip.frame_final_start
|
||||||
x2 = strip.frame_final_end
|
x2 = strip.frame_final_end
|
||||||
@@ -47,59 +114,56 @@ def get_strip_rectf(strip):
|
|||||||
return x1, y1, x2, y2
|
return x1, y1, x2, y2
|
||||||
|
|
||||||
|
|
||||||
def draw_underline_in_strip(strip_coords, pixel_size_x, color):
|
def underline_in_strip(strip_coords: Float4,
|
||||||
from bgl import glColor4f, glRectf, glEnable, glDisable, GL_BLEND
|
pixel_size_x: float,
|
||||||
import bgl
|
color: Float4,
|
||||||
|
out_coords: typing.List[Float2],
|
||||||
context = bpy.context
|
out_colors: typing.List[Float4]):
|
||||||
|
|
||||||
# Strip coords
|
# Strip coords
|
||||||
s_x1, s_y1, s_x2, s_y2 = strip_coords
|
s_x1, s_y1, s_x2, s_y2 = strip_coords
|
||||||
|
|
||||||
# be careful not to draw over the current frame line
|
# be careful not to draw over the current frame line
|
||||||
cf_x = context.scene.frame_current_final
|
cf_x = bpy.context.scene.frame_current_final
|
||||||
|
|
||||||
bgl.glPushAttrib(bgl.GL_COLOR_BUFFER_BIT | bgl.GL_LINE_BIT)
|
# TODO(Sybren): figure out how to pass one colour per line,
|
||||||
|
# instead of one colour per vertex.
|
||||||
|
out_coords.append((s_x1, s_y1))
|
||||||
|
out_colors.append(color)
|
||||||
|
|
||||||
glColor4f(*color)
|
|
||||||
glEnable(GL_BLEND)
|
|
||||||
bgl.glLineWidth(2)
|
|
||||||
bgl.glBegin(bgl.GL_LINES)
|
|
||||||
|
|
||||||
bgl.glVertex2f(s_x1, s_y1)
|
|
||||||
if s_x1 < cf_x < s_x2:
|
if s_x1 < cf_x < s_x2:
|
||||||
# Bad luck, the line passes our strip
|
# Bad luck, the line passes our strip, so draw two lines.
|
||||||
bgl.glVertex2f(cf_x - pixel_size_x, s_y1)
|
out_coords.append((cf_x - pixel_size_x, s_y1))
|
||||||
bgl.glVertex2f(cf_x + pixel_size_x, s_y1)
|
out_colors.append(color)
|
||||||
bgl.glVertex2f(s_x2, s_y1)
|
|
||||||
|
|
||||||
bgl.glEnd()
|
out_coords.append((cf_x + pixel_size_x, s_y1))
|
||||||
bgl.glPopAttrib()
|
out_colors.append(color)
|
||||||
|
|
||||||
|
out_coords.append((s_x2, s_y1))
|
||||||
|
out_colors.append(color)
|
||||||
|
|
||||||
|
|
||||||
def draw_strip_conflict(strip_coords, pixel_size_x):
|
def strip_conflict(strip_coords: Float4,
|
||||||
|
out_coords: typing.List[Float2],
|
||||||
|
out_colors: typing.List[Float4]):
|
||||||
"""Draws conflicting states between strips."""
|
"""Draws conflicting states between strips."""
|
||||||
|
|
||||||
import bgl
|
|
||||||
|
|
||||||
s_x1, s_y1, s_x2, s_y2 = strip_coords
|
s_x1, s_y1, s_x2, s_y2 = strip_coords
|
||||||
bgl.glPushAttrib(bgl.GL_COLOR_BUFFER_BIT | bgl.GL_LINE_BIT)
|
|
||||||
|
|
||||||
# Always draw the full rectangle, the conflict should be resolved and thus stand out.
|
# TODO(Sybren): draw a rectangle instead of a line.
|
||||||
bgl.glColor3f(*CONFLICT_COLOUR)
|
out_coords.append((s_x1, s_y2))
|
||||||
bgl.glLineWidth(2)
|
out_colors.append(CONFLICT_COLOUR)
|
||||||
|
|
||||||
bgl.glBegin(bgl.GL_LINE_LOOP)
|
out_coords.append((s_x2, s_y1))
|
||||||
bgl.glVertex2f(s_x1, s_y1)
|
out_colors.append(CONFLICT_COLOUR)
|
||||||
bgl.glVertex2f(s_x2, s_y1)
|
|
||||||
bgl.glVertex2f(s_x2, s_y2)
|
|
||||||
bgl.glVertex2f(s_x1, s_y2)
|
|
||||||
bgl.glEnd()
|
|
||||||
|
|
||||||
bgl.glPopAttrib()
|
out_coords.append((s_x2, s_y2))
|
||||||
|
out_colors.append(CONFLICT_COLOUR)
|
||||||
|
|
||||||
|
out_coords.append((s_x1, s_y1))
|
||||||
|
out_colors.append(CONFLICT_COLOUR)
|
||||||
|
|
||||||
|
|
||||||
def draw_callback_px():
|
def draw_callback_px(line_drawer: AttractLineDrawer):
|
||||||
context = bpy.context
|
context = bpy.context
|
||||||
|
|
||||||
if not context.scene.sequence_editor:
|
if not context.scene.sequence_editor:
|
||||||
@@ -115,6 +179,10 @@ def draw_callback_px():
|
|||||||
|
|
||||||
strips = shown_strips(context)
|
strips = shown_strips(context)
|
||||||
|
|
||||||
|
coords = [] # type: typing.List[Float2]
|
||||||
|
colors = [] # type: typing.List[Float4]
|
||||||
|
|
||||||
|
# Collect all the lines (vertex coords + vertex colours) to draw.
|
||||||
for strip in strips:
|
for strip in strips:
|
||||||
if not strip.atc_object_id:
|
if not strip.atc_object_id:
|
||||||
continue
|
continue
|
||||||
@@ -136,9 +204,11 @@ def draw_callback_px():
|
|||||||
|
|
||||||
alpha = 1.0 if strip.atc_is_synced else 0.5
|
alpha = 1.0 if strip.atc_is_synced else 0.5
|
||||||
|
|
||||||
draw_underline_in_strip(strip_coords, pixel_size_x, color + (alpha,))
|
underline_in_strip(strip_coords, pixel_size_x, color + (alpha,), coords, colors)
|
||||||
if strip.atc_is_synced and strip.atc_object_id_conflict:
|
if strip.atc_is_synced and strip.atc_object_id_conflict:
|
||||||
draw_strip_conflict(strip_coords, pixel_size_x)
|
strip_conflict(strip_coords, coords, colors)
|
||||||
|
|
||||||
|
line_drawer.draw(coords, colors)
|
||||||
|
|
||||||
|
|
||||||
def tag_redraw_all_sequencer_editors():
|
def tag_redraw_all_sequencer_editors():
|
||||||
@@ -162,8 +232,13 @@ def callback_enable():
|
|||||||
if cb_handle:
|
if cb_handle:
|
||||||
return
|
return
|
||||||
|
|
||||||
|
# Doing GPU stuff in the background crashes Blender, so let's not.
|
||||||
|
if bpy.app.background:
|
||||||
|
return
|
||||||
|
|
||||||
|
line_drawer = AttractLineDrawer()
|
||||||
cb_handle[:] = bpy.types.SpaceSequenceEditor.draw_handler_add(
|
cb_handle[:] = bpy.types.SpaceSequenceEditor.draw_handler_add(
|
||||||
draw_callback_px, (), 'WINDOW', 'POST_VIEW'),
|
draw_callback_px, (line_drawer,), 'WINDOW', 'POST_VIEW'),
|
||||||
|
|
||||||
tag_redraw_all_sequencer_editors()
|
tag_redraw_all_sequencer_editors()
|
||||||
|
|
||||||
|
182
blender_cloud/attract/draw_27.py
Normal file
182
blender_cloud/attract/draw_27.py
Normal file
@@ -0,0 +1,182 @@
|
|||||||
|
# ##### 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 #####
|
||||||
|
|
||||||
|
# <pep8 compliant>
|
||||||
|
|
||||||
|
import bpy
|
||||||
|
import logging
|
||||||
|
import collections
|
||||||
|
|
||||||
|
log = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
strip_status_colour = {
|
||||||
|
None: (0.7, 0.7, 0.7),
|
||||||
|
'approved': (0.6392156862745098, 0.8784313725490196, 0.30196078431372547),
|
||||||
|
'final': (0.9058823529411765, 0.9607843137254902, 0.8274509803921568),
|
||||||
|
'in_progress': (1.0, 0.7450980392156863, 0.0),
|
||||||
|
'on_hold': (0.796078431372549, 0.6196078431372549, 0.08235294117647059),
|
||||||
|
'review': (0.8941176470588236, 0.9607843137254902, 0.9764705882352941),
|
||||||
|
'todo': (1.0, 0.5019607843137255, 0.5019607843137255)
|
||||||
|
}
|
||||||
|
|
||||||
|
CONFLICT_COLOUR = (0.576, 0.118, 0.035) # RGB tuple
|
||||||
|
|
||||||
|
|
||||||
|
def get_strip_rectf(strip):
|
||||||
|
# Get x and y in terms of the grid's frames and channels
|
||||||
|
x1 = strip.frame_final_start
|
||||||
|
x2 = strip.frame_final_end
|
||||||
|
y1 = strip.channel + 0.2
|
||||||
|
y2 = strip.channel - 0.2 + 1
|
||||||
|
|
||||||
|
return x1, y1, x2, y2
|
||||||
|
|
||||||
|
|
||||||
|
def draw_underline_in_strip(strip_coords, pixel_size_x, color):
|
||||||
|
from bgl import glColor4f, glRectf, glEnable, glDisable, GL_BLEND
|
||||||
|
import bgl
|
||||||
|
|
||||||
|
context = bpy.context
|
||||||
|
|
||||||
|
# Strip coords
|
||||||
|
s_x1, s_y1, s_x2, s_y2 = strip_coords
|
||||||
|
|
||||||
|
# be careful not to draw over the current frame line
|
||||||
|
cf_x = context.scene.frame_current_final
|
||||||
|
|
||||||
|
bgl.glPushAttrib(bgl.GL_COLOR_BUFFER_BIT | bgl.GL_LINE_BIT)
|
||||||
|
|
||||||
|
glColor4f(*color)
|
||||||
|
glEnable(GL_BLEND)
|
||||||
|
bgl.glLineWidth(2)
|
||||||
|
bgl.glBegin(bgl.GL_LINES)
|
||||||
|
|
||||||
|
bgl.glVertex2f(s_x1, s_y1)
|
||||||
|
if s_x1 < cf_x < s_x2:
|
||||||
|
# Bad luck, the line passes our strip
|
||||||
|
bgl.glVertex2f(cf_x - pixel_size_x, s_y1)
|
||||||
|
bgl.glVertex2f(cf_x + pixel_size_x, s_y1)
|
||||||
|
bgl.glVertex2f(s_x2, s_y1)
|
||||||
|
|
||||||
|
bgl.glEnd()
|
||||||
|
bgl.glPopAttrib()
|
||||||
|
|
||||||
|
|
||||||
|
def draw_strip_conflict(strip_coords, pixel_size_x):
|
||||||
|
"""Draws conflicting states between strips."""
|
||||||
|
|
||||||
|
import bgl
|
||||||
|
|
||||||
|
s_x1, s_y1, s_x2, s_y2 = strip_coords
|
||||||
|
bgl.glPushAttrib(bgl.GL_COLOR_BUFFER_BIT | bgl.GL_LINE_BIT)
|
||||||
|
|
||||||
|
# Always draw the full rectangle, the conflict should be resolved and thus stand out.
|
||||||
|
bgl.glColor3f(*CONFLICT_COLOUR)
|
||||||
|
bgl.glLineWidth(2)
|
||||||
|
|
||||||
|
bgl.glBegin(bgl.GL_LINE_LOOP)
|
||||||
|
bgl.glVertex2f(s_x1, s_y1)
|
||||||
|
bgl.glVertex2f(s_x2, s_y1)
|
||||||
|
bgl.glVertex2f(s_x2, s_y2)
|
||||||
|
bgl.glVertex2f(s_x1, s_y2)
|
||||||
|
bgl.glEnd()
|
||||||
|
|
||||||
|
bgl.glPopAttrib()
|
||||||
|
|
||||||
|
|
||||||
|
def draw_callback_px():
|
||||||
|
context = bpy.context
|
||||||
|
|
||||||
|
if not context.scene.sequence_editor:
|
||||||
|
return
|
||||||
|
|
||||||
|
from . import shown_strips
|
||||||
|
|
||||||
|
region = context.region
|
||||||
|
xwin1, ywin1 = region.view2d.region_to_view(0, 0)
|
||||||
|
xwin2, ywin2 = region.view2d.region_to_view(region.width, region.height)
|
||||||
|
one_pixel_further_x, one_pixel_further_y = region.view2d.region_to_view(1, 1)
|
||||||
|
pixel_size_x = one_pixel_further_x - xwin1
|
||||||
|
|
||||||
|
strips = shown_strips(context)
|
||||||
|
|
||||||
|
for strip in strips:
|
||||||
|
if not strip.atc_object_id:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Get corners (x1, y1), (x2, y2) of the strip rectangle in px region coords
|
||||||
|
strip_coords = get_strip_rectf(strip)
|
||||||
|
|
||||||
|
# check if any of the coordinates are out of bounds
|
||||||
|
if strip_coords[0] > xwin2 or strip_coords[2] < xwin1 or strip_coords[1] > ywin2 or \
|
||||||
|
strip_coords[3] < ywin1:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Draw
|
||||||
|
status = strip.atc_status
|
||||||
|
if status in strip_status_colour:
|
||||||
|
color = strip_status_colour[status]
|
||||||
|
else:
|
||||||
|
color = strip_status_colour[None]
|
||||||
|
|
||||||
|
alpha = 1.0 if strip.atc_is_synced else 0.5
|
||||||
|
|
||||||
|
draw_underline_in_strip(strip_coords, pixel_size_x, color + (alpha,))
|
||||||
|
if strip.atc_is_synced and strip.atc_object_id_conflict:
|
||||||
|
draw_strip_conflict(strip_coords, pixel_size_x)
|
||||||
|
|
||||||
|
|
||||||
|
def tag_redraw_all_sequencer_editors():
|
||||||
|
context = bpy.context
|
||||||
|
|
||||||
|
# Py cant access notifiers
|
||||||
|
for window in context.window_manager.windows:
|
||||||
|
for area in window.screen.areas:
|
||||||
|
if area.type == 'SEQUENCE_EDITOR':
|
||||||
|
for region in area.regions:
|
||||||
|
if region.type == 'WINDOW':
|
||||||
|
region.tag_redraw()
|
||||||
|
|
||||||
|
|
||||||
|
# This is a list so it can be changed instead of set
|
||||||
|
# if it is only changed, it does not have to be declared as a global everywhere
|
||||||
|
cb_handle = []
|
||||||
|
|
||||||
|
|
||||||
|
def callback_enable():
|
||||||
|
if cb_handle:
|
||||||
|
return
|
||||||
|
|
||||||
|
cb_handle[:] = bpy.types.SpaceSequenceEditor.draw_handler_add(
|
||||||
|
draw_callback_px, (), 'WINDOW', 'POST_VIEW'),
|
||||||
|
|
||||||
|
tag_redraw_all_sequencer_editors()
|
||||||
|
|
||||||
|
|
||||||
|
def callback_disable():
|
||||||
|
if not cb_handle:
|
||||||
|
return
|
||||||
|
|
||||||
|
try:
|
||||||
|
bpy.types.SpaceSequenceEditor.draw_handler_remove(cb_handle[0], 'WINDOW')
|
||||||
|
except ValueError:
|
||||||
|
# Thrown when already removed.
|
||||||
|
pass
|
||||||
|
cb_handle.clear()
|
||||||
|
|
||||||
|
tag_redraw_all_sequencer_editors()
|
@@ -23,22 +23,21 @@ Separated from __init__.py so that we can import & run from non-Blender environm
|
|||||||
import functools
|
import functools
|
||||||
import logging
|
import logging
|
||||||
import os.path
|
import os.path
|
||||||
|
import tempfile
|
||||||
|
|
||||||
import bpy
|
import bpy
|
||||||
from bpy.types import AddonPreferences, Operator, WindowManager, Scene, PropertyGroup
|
from bpy.types import AddonPreferences, Operator, WindowManager, Scene, PropertyGroup
|
||||||
from bpy.props import StringProperty, EnumProperty, PointerProperty, BoolProperty, IntProperty
|
from bpy.props import StringProperty, EnumProperty, PointerProperty, BoolProperty, IntProperty
|
||||||
import rna_prop_ui
|
import rna_prop_ui
|
||||||
|
|
||||||
from . import pillar, async_loop, flamenco
|
from . import compatibility, pillar, async_loop, flamenco, project_specific
|
||||||
from .utils import pyside_cache, redraw
|
from .utils import pyside_cache, redraw
|
||||||
|
|
||||||
PILLAR_WEB_SERVER_URL = 'https://cloud.blender.org/'
|
PILLAR_WEB_SERVER_URL = os.environ.get('BCLOUD_SERVER', 'https://cloud.blender.org/')
|
||||||
# PILLAR_WEB_SERVER_URL = 'http://pillar-web:5001/'
|
|
||||||
PILLAR_SERVER_URL = '%sapi/' % PILLAR_WEB_SERVER_URL
|
PILLAR_SERVER_URL = '%sapi/' % PILLAR_WEB_SERVER_URL
|
||||||
|
|
||||||
ADDON_NAME = 'blender_cloud'
|
ADDON_NAME = 'blender_cloud'
|
||||||
log = logging.getLogger(__name__)
|
log = logging.getLogger(__name__)
|
||||||
|
|
||||||
icons = None
|
icons = None
|
||||||
|
|
||||||
|
|
||||||
@@ -53,6 +52,7 @@ def blender_syncable_versions(self, context):
|
|||||||
return [(v, v, '') for v in versions]
|
return [(v, v, '') for v in versions]
|
||||||
|
|
||||||
|
|
||||||
|
@compatibility.convert_properties
|
||||||
class SyncStatusProperties(PropertyGroup):
|
class SyncStatusProperties(PropertyGroup):
|
||||||
status = EnumProperty(
|
status = EnumProperty(
|
||||||
items=[
|
items=[
|
||||||
@@ -140,30 +140,7 @@ def project_extensions(project_id) -> set:
|
|||||||
return set(proj.get('enabled_for', ()))
|
return set(proj.get('enabled_for', ()))
|
||||||
|
|
||||||
|
|
||||||
def handle_project_update(_=None, _2=None):
|
@compatibility.convert_properties
|
||||||
"""Handles changing projects, which may cause extensions to be disabled/enabled.
|
|
||||||
|
|
||||||
Ignores arguments so that it can be used as property update callback.
|
|
||||||
"""
|
|
||||||
|
|
||||||
project_id = preferences().project.project
|
|
||||||
log.info('Updating internal state to reflect extensions enabled on current project %s.',
|
|
||||||
project_id)
|
|
||||||
|
|
||||||
project_extensions.cache_clear()
|
|
||||||
|
|
||||||
from blender_cloud import attract, flamenco
|
|
||||||
attract.deactivate()
|
|
||||||
flamenco.deactivate()
|
|
||||||
|
|
||||||
enabled_for = project_extensions(project_id)
|
|
||||||
log.info('Project extensions: %s', enabled_for)
|
|
||||||
if 'attract' in enabled_for:
|
|
||||||
attract.activate()
|
|
||||||
if 'flamenco' in enabled_for:
|
|
||||||
flamenco.activate()
|
|
||||||
|
|
||||||
|
|
||||||
class BlenderCloudProjectGroup(PropertyGroup):
|
class BlenderCloudProjectGroup(PropertyGroup):
|
||||||
status = EnumProperty(
|
status = EnumProperty(
|
||||||
items=[
|
items=[
|
||||||
@@ -178,7 +155,7 @@ class BlenderCloudProjectGroup(PropertyGroup):
|
|||||||
items=bcloud_available_projects,
|
items=bcloud_available_projects,
|
||||||
name='Cloud project',
|
name='Cloud project',
|
||||||
description='Which Blender Cloud project to work with',
|
description='Which Blender Cloud project to work with',
|
||||||
update=handle_project_update
|
update=project_specific.handle_project_update
|
||||||
)
|
)
|
||||||
|
|
||||||
# List of projects is stored in 'available_projects' ID property,
|
# List of projects is stored in 'available_projects' ID property,
|
||||||
@@ -190,13 +167,14 @@ class BlenderCloudProjectGroup(PropertyGroup):
|
|||||||
@available_projects.setter
|
@available_projects.setter
|
||||||
def available_projects(self, new_projects):
|
def available_projects(self, new_projects):
|
||||||
self['available_projects'] = new_projects
|
self['available_projects'] = new_projects
|
||||||
handle_project_update()
|
project_specific.handle_project_update()
|
||||||
|
|
||||||
|
|
||||||
|
@compatibility.convert_properties
|
||||||
class BlenderCloudPreferences(AddonPreferences):
|
class BlenderCloudPreferences(AddonPreferences):
|
||||||
bl_idname = ADDON_NAME
|
bl_idname = ADDON_NAME
|
||||||
|
|
||||||
# The following two properties are read-only to limit the scope of the
|
# The following property is read-only to limit the scope of the
|
||||||
# addon and allow for proper testing within this scope.
|
# addon and allow for proper testing within this scope.
|
||||||
pillar_server = StringProperty(
|
pillar_server = StringProperty(
|
||||||
name='Blender Cloud Server',
|
name='Blender Cloud Server',
|
||||||
@@ -225,29 +203,32 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
description='Local path of your Attract project, used to search for blend files; '
|
description='Local path of your Attract project, used to search for blend files; '
|
||||||
'usually best to set to an absolute path',
|
'usually best to set to an absolute path',
|
||||||
subtype='DIR_PATH',
|
subtype='DIR_PATH',
|
||||||
default='//../')
|
default='//../',
|
||||||
|
update=project_specific.store,
|
||||||
|
)
|
||||||
|
|
||||||
flamenco_manager = PointerProperty(type=flamenco.FlamencoManagerGroup)
|
flamenco_manager = PointerProperty(type=flamenco.FlamencoManagerGroup)
|
||||||
flamenco_exclude_filter = StringProperty(
|
flamenco_exclude_filter = StringProperty(
|
||||||
name='File Exclude Filter',
|
name='File Exclude Filter',
|
||||||
description='Filter like "*.abc;*.mkv" to prevent certain files to be packed '
|
description='Space-separated list of filename filters, like "*.abc *.mkv", to prevent '
|
||||||
'into the output directory',
|
'matching files from being packed into the output directory',
|
||||||
default='')
|
default='',
|
||||||
# TODO: before making Flamenco public, change the defaults to something less Institute-specific.
|
update=project_specific.store,
|
||||||
# NOTE: The assumption is that the workers can also find the files in the same path.
|
)
|
||||||
# This assumption is true for the Blender Institute.
|
|
||||||
flamenco_job_file_path = StringProperty(
|
flamenco_job_file_path = StringProperty(
|
||||||
name='Job File Path',
|
name='Job Storage Path',
|
||||||
description='Path where to store job files, should be accesible for Workers too',
|
description='Path where to store job files, should be accesible for Workers too',
|
||||||
subtype='DIR_PATH',
|
subtype='DIR_PATH',
|
||||||
default='/render/_flamenco/storage')
|
default=tempfile.gettempdir(),
|
||||||
|
update=project_specific.store,
|
||||||
# TODO: before making Flamenco public, change the defaults to something less Institute-specific.
|
)
|
||||||
flamenco_job_output_path = StringProperty(
|
flamenco_job_output_path = StringProperty(
|
||||||
name='Job Output Path',
|
name='Job Output Path',
|
||||||
description='Path where to store output files, should be accessible for Workers',
|
description='Path where to store output files, should be accessible for Workers',
|
||||||
subtype='DIR_PATH',
|
subtype='DIR_PATH',
|
||||||
default='/render/_flamenco/output')
|
default=tempfile.gettempdir(),
|
||||||
|
update=project_specific.store,
|
||||||
|
)
|
||||||
flamenco_job_output_strip_components = IntProperty(
|
flamenco_job_output_strip_components = IntProperty(
|
||||||
name='Job Output Path Strip Components',
|
name='Job Output Path Strip Components',
|
||||||
description='The final output path comprises of the job output path, and the blend file '
|
description='The final output path comprises of the job output path, and the blend file '
|
||||||
@@ -256,11 +237,28 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
min=0,
|
min=0,
|
||||||
default=0,
|
default=0,
|
||||||
soft_max=4,
|
soft_max=4,
|
||||||
|
update=project_specific.store,
|
||||||
)
|
)
|
||||||
|
flamenco_relative_only = BoolProperty(
|
||||||
|
name='Relative Paths Only',
|
||||||
|
description='When enabled, only assets that are referred to with a relative path are '
|
||||||
|
'packed, and assets referred to by an absolute path are excluded from the '
|
||||||
|
'BAT pack. When disabled, all assets are packed',
|
||||||
|
default=False,
|
||||||
|
update=project_specific.store,
|
||||||
|
)
|
||||||
|
|
||||||
flamenco_open_browser_after_submit = BoolProperty(
|
flamenco_open_browser_after_submit = BoolProperty(
|
||||||
name='Open Browser after Submitting Job',
|
name='Open Browser after Submitting Job',
|
||||||
description='When enabled, Blender will open a webbrowser',
|
description='When enabled, Blender will open a webbrowser',
|
||||||
default=True
|
default=True,
|
||||||
|
)
|
||||||
|
flamenco_show_quit_after_submit_button = BoolProperty(
|
||||||
|
name='Show "Submit & Quit" button',
|
||||||
|
description='When enabled, next to the "Render on Flamenco" button there will be a button '
|
||||||
|
'"Submit & Quit" that silently quits Blender after submitting the render job '
|
||||||
|
'to Flamenco',
|
||||||
|
default=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
def draw(self, context):
|
def draw(self, context):
|
||||||
@@ -319,8 +317,8 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
bss = context.window_manager.blender_sync_status
|
bss = context.window_manager.blender_sync_status
|
||||||
bsync_box = layout.box()
|
bsync_box = layout.box()
|
||||||
bsync_box.enabled = msg_icon != 'ERROR'
|
bsync_box.enabled = msg_icon != 'ERROR'
|
||||||
row = bsync_box.row().split(percentage=0.33)
|
row = bsync_box.row().split(**compatibility.factor(0.33))
|
||||||
row.label('Blender Sync with Blender Cloud', icon_value=icon('CLOUD'))
|
row.label(text='Blender Sync with Blender Cloud', icon_value=icon('CLOUD'))
|
||||||
|
|
||||||
icon_for_level = {
|
icon_for_level = {
|
||||||
'INFO': 'NONE',
|
'INFO': 'NONE',
|
||||||
@@ -330,7 +328,7 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
}
|
}
|
||||||
msg_icon = icon_for_level[bss.level] if bss.message else 'NONE'
|
msg_icon = icon_for_level[bss.level] if bss.message else 'NONE'
|
||||||
message_container = row.row()
|
message_container = row.row()
|
||||||
message_container.label(bss.message, icon=msg_icon)
|
message_container.label(text=bss.message, icon=msg_icon)
|
||||||
|
|
||||||
sub = bsync_box.column()
|
sub = bsync_box.column()
|
||||||
|
|
||||||
@@ -340,7 +338,7 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
|
|
||||||
# Image Share stuff
|
# Image Share stuff
|
||||||
share_box = layout.box()
|
share_box = layout.box()
|
||||||
share_box.label('Image Sharing on Blender Cloud', icon_value=icon('CLOUD'))
|
share_box.label(text='Image Sharing on Blender Cloud', icon_value=icon('CLOUD'))
|
||||||
share_box.prop(self, 'open_browser_after_share')
|
share_box.prop(self, 'open_browser_after_share')
|
||||||
|
|
||||||
# Project selector
|
# Project selector
|
||||||
@@ -362,7 +360,7 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
layout.enabled = bss.status in {'NONE', 'IDLE'}
|
layout.enabled = bss.status in {'NONE', 'IDLE'}
|
||||||
|
|
||||||
buttons = layout.column()
|
buttons = layout.column()
|
||||||
row_buttons = buttons.row().split(percentage=0.5)
|
row_buttons = buttons.row().split(**compatibility.factor(0.5))
|
||||||
row_push = row_buttons.row()
|
row_push = row_buttons.row()
|
||||||
row_pull = row_buttons.row(align=True)
|
row_pull = row_buttons.row(align=True)
|
||||||
|
|
||||||
@@ -371,34 +369,33 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
icon='TRIA_UP').action = 'PUSH'
|
icon='TRIA_UP').action = 'PUSH'
|
||||||
|
|
||||||
versions = bss.available_blender_versions
|
versions = bss.available_blender_versions
|
||||||
version = bss.version
|
|
||||||
if bss.status in {'NONE', 'IDLE'}:
|
if bss.status in {'NONE', 'IDLE'}:
|
||||||
if not versions or not version:
|
if not versions:
|
||||||
row_pull.operator('pillar.sync',
|
row_pull.operator('pillar.sync',
|
||||||
text='Find version to load',
|
text='Find version to load',
|
||||||
icon='TRIA_DOWN').action = 'REFRESH'
|
icon='TRIA_DOWN').action = 'REFRESH'
|
||||||
else:
|
else:
|
||||||
props = row_pull.operator('pillar.sync',
|
props = row_pull.operator('pillar.sync',
|
||||||
text='Load %s settings' % version,
|
text='Load %s settings' % bss.version,
|
||||||
icon='TRIA_DOWN')
|
icon='TRIA_DOWN')
|
||||||
props.action = 'PULL'
|
props.action = 'PULL'
|
||||||
props.blender_version = version
|
props.blender_version = bss.version
|
||||||
row_pull.operator('pillar.sync',
|
row_pull.operator('pillar.sync',
|
||||||
text='',
|
text='',
|
||||||
icon='DOTSDOWN').action = 'SELECT'
|
icon=compatibility.SYNC_SELECT_VERSION_ICON).action = 'SELECT'
|
||||||
else:
|
else:
|
||||||
row_pull.label('Cloud Sync is running.')
|
row_pull.label(text='Cloud Sync is running.')
|
||||||
|
|
||||||
def draw_project_selector(self, project_box, bcp: BlenderCloudProjectGroup):
|
def draw_project_selector(self, project_box, bcp: BlenderCloudProjectGroup):
|
||||||
project_row = project_box.row(align=True)
|
project_row = project_box.row(align=True)
|
||||||
project_row.label('Project settings', icon_value=icon('CLOUD'))
|
project_row.label(text='Project settings', icon_value=icon('CLOUD'))
|
||||||
|
|
||||||
row_buttons = project_row.row(align=True)
|
row_buttons = project_row.row(align=True)
|
||||||
|
|
||||||
projects = bcp.available_projects
|
projects = bcp.available_projects
|
||||||
project = bcp.project
|
project = bcp.project
|
||||||
if bcp.status in {'NONE', 'IDLE'}:
|
if bcp.status in {'NONE', 'IDLE'}:
|
||||||
if not projects or not project:
|
if not projects:
|
||||||
row_buttons.operator('pillar.projects',
|
row_buttons.operator('pillar.projects',
|
||||||
text='Find project to load',
|
text='Find project to load',
|
||||||
icon='FILE_REFRESH')
|
icon='FILE_REFRESH')
|
||||||
@@ -407,34 +404,38 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
row_buttons.operator('pillar.projects',
|
row_buttons.operator('pillar.projects',
|
||||||
text='',
|
text='',
|
||||||
icon='FILE_REFRESH')
|
icon='FILE_REFRESH')
|
||||||
|
props = row_buttons.operator('pillar.project_open_in_browser',
|
||||||
|
text='',
|
||||||
|
icon='WORLD')
|
||||||
|
props.project_id = project
|
||||||
else:
|
else:
|
||||||
row_buttons.label('Fetching available projects.')
|
row_buttons.label(text='Fetching available projects.')
|
||||||
|
|
||||||
enabled_for = project_extensions(project)
|
enabled_for = project_extensions(project)
|
||||||
if not project:
|
if not project:
|
||||||
return
|
return
|
||||||
|
|
||||||
if not enabled_for:
|
if not enabled_for:
|
||||||
project_box.label('This project is not set up for Attract or Flamenco')
|
project_box.label(text='This project is not set up for Attract or Flamenco')
|
||||||
return
|
return
|
||||||
|
|
||||||
project_box.label('This project is set up for: %s' %
|
project_box.label(text='This project is set up for: %s' %
|
||||||
', '.join(sorted(enabled_for)))
|
', '.join(sorted(enabled_for)))
|
||||||
|
|
||||||
# This is only needed when the project is set up for either Attract or Flamenco.
|
# This is only needed when the project is set up for either Attract or Flamenco.
|
||||||
project_box.prop(self, 'cloud_project_local_path',
|
project_box.prop(self, 'cloud_project_local_path',
|
||||||
text='Local Cloud Project Path')
|
text='Local Project Path')
|
||||||
|
|
||||||
def draw_flamenco_buttons(self, flamenco_box, bcp: flamenco.FlamencoManagerGroup, context):
|
def draw_flamenco_buttons(self, flamenco_box, bcp: flamenco.FlamencoManagerGroup, context):
|
||||||
header_row = flamenco_box.row(align=True)
|
header_row = flamenco_box.row(align=True)
|
||||||
header_row.label('Flamenco:', icon_value=icon('CLOUD'))
|
header_row.label(text='Flamenco:', icon_value=icon('CLOUD'))
|
||||||
|
|
||||||
manager_split = flamenco_box.split(0.32, align=True)
|
manager_split = flamenco_box.split(**compatibility.factor(0.32), align=True)
|
||||||
manager_split.label('Manager:')
|
manager_split.label(text='Manager:')
|
||||||
manager_box = manager_split.row(align=True)
|
manager_box = manager_split.row(align=True)
|
||||||
|
|
||||||
if bcp.status in {'NONE', 'IDLE'}:
|
if bcp.status in {'NONE', 'IDLE'}:
|
||||||
if not bcp.available_managers or not bcp.manager:
|
if not bcp.available_managers:
|
||||||
manager_box.operator('flamenco.managers',
|
manager_box.operator('flamenco.managers',
|
||||||
text='Find Flamenco Managers',
|
text='Find Flamenco Managers',
|
||||||
icon='FILE_REFRESH')
|
icon='FILE_REFRESH')
|
||||||
@@ -444,9 +445,9 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
text='',
|
text='',
|
||||||
icon='FILE_REFRESH')
|
icon='FILE_REFRESH')
|
||||||
else:
|
else:
|
||||||
manager_box.label('Fetching available managers.')
|
manager_box.label(text='Fetching available managers.')
|
||||||
|
|
||||||
path_split = flamenco_box.split(0.32, align=True)
|
path_split = flamenco_box.split(**compatibility.factor(0.32), align=True)
|
||||||
path_split.label(text='Job File Path:')
|
path_split.label(text='Job File Path:')
|
||||||
path_box = path_split.row(align=True)
|
path_box = path_split.row(align=True)
|
||||||
path_box.prop(self, 'flamenco_job_file_path', text='')
|
path_box.prop(self, 'flamenco_job_file_path', text='')
|
||||||
@@ -454,17 +455,16 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
props.path = self.flamenco_job_file_path
|
props.path = self.flamenco_job_file_path
|
||||||
|
|
||||||
job_output_box = flamenco_box.column(align=True)
|
job_output_box = flamenco_box.column(align=True)
|
||||||
path_split = job_output_box.split(0.32, align=True)
|
path_split = job_output_box.split(**compatibility.factor(0.32), align=True)
|
||||||
path_split.label(text='Job Output Path:')
|
path_split.label(text='Job Output Path:')
|
||||||
path_box = path_split.row(align=True)
|
path_box = path_split.row(align=True)
|
||||||
path_box.prop(self, 'flamenco_job_output_path', text='')
|
path_box.prop(self, 'flamenco_job_output_path', text='')
|
||||||
props = path_box.operator('flamenco.explore_file_path', text='', icon='DISK_DRIVE')
|
props = path_box.operator('flamenco.explore_file_path', text='', icon='DISK_DRIVE')
|
||||||
props.path = self.flamenco_job_output_path
|
props.path = self.flamenco_job_output_path
|
||||||
|
|
||||||
job_output_box.prop(self, 'flamenco_exclude_filter')
|
job_output_box.prop(self, 'flamenco_exclude_filter')
|
||||||
|
|
||||||
prop_split = job_output_box.split(0.32, align=True)
|
prop_split = job_output_box.split(**compatibility.factor(0.32), align=True)
|
||||||
prop_split.label('Strip Components:')
|
prop_split.label(text='Strip Components:')
|
||||||
prop_split.prop(self, 'flamenco_job_output_strip_components', text='')
|
prop_split.prop(self, 'flamenco_job_output_strip_components', text='')
|
||||||
|
|
||||||
from .flamenco import render_output_path
|
from .flamenco import render_output_path
|
||||||
@@ -472,14 +472,16 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
path_box = job_output_box.row(align=True)
|
path_box = job_output_box.row(align=True)
|
||||||
output_path = render_output_path(context)
|
output_path = render_output_path(context)
|
||||||
if output_path:
|
if output_path:
|
||||||
path_box.label(str(output_path))
|
path_box.label(text=str(output_path))
|
||||||
props = path_box.operator('flamenco.explore_file_path', text='', icon='DISK_DRIVE')
|
props = path_box.operator('flamenco.explore_file_path', text='', icon='DISK_DRIVE')
|
||||||
props.path = str(output_path.parent)
|
props.path = str(output_path.parent)
|
||||||
else:
|
else:
|
||||||
path_box.label('Blend file is not in your project path, '
|
path_box.label(text='Blend file is not in your project path, '
|
||||||
'unable to give output path example.')
|
'unable to give output path example.')
|
||||||
|
|
||||||
|
flamenco_box.prop(self, 'flamenco_relative_only')
|
||||||
flamenco_box.prop(self, 'flamenco_open_browser_after_submit')
|
flamenco_box.prop(self, 'flamenco_open_browser_after_submit')
|
||||||
|
flamenco_box.prop(self, 'flamenco_show_quit_after_submit_button')
|
||||||
|
|
||||||
|
|
||||||
class PillarCredentialsUpdate(pillar.PillarOperatorMixin,
|
class PillarCredentialsUpdate(pillar.PillarOperatorMixin,
|
||||||
@@ -545,6 +547,37 @@ class PILLAR_OT_subscribe(Operator):
|
|||||||
return {'FINISHED'}
|
return {'FINISHED'}
|
||||||
|
|
||||||
|
|
||||||
|
@compatibility.convert_properties
|
||||||
|
class PILLAR_OT_project_open_in_browser(Operator):
|
||||||
|
bl_idname = 'pillar.project_open_in_browser'
|
||||||
|
bl_label = 'Open in Browser'
|
||||||
|
bl_description = 'Opens a webbrowser to show the project'
|
||||||
|
|
||||||
|
project_id = StringProperty(name='Project ID')
|
||||||
|
|
||||||
|
def execute(self, context):
|
||||||
|
if not self.project_id:
|
||||||
|
return {'CANCELLED'}
|
||||||
|
|
||||||
|
import webbrowser
|
||||||
|
import urllib.parse
|
||||||
|
|
||||||
|
import pillarsdk
|
||||||
|
from .pillar import sync_call
|
||||||
|
|
||||||
|
project = sync_call(pillarsdk.Project.find, self.project_id, {'projection': {'url': True}})
|
||||||
|
|
||||||
|
if log.isEnabledFor(logging.DEBUG):
|
||||||
|
import pprint
|
||||||
|
log.debug('found project: %s', pprint.pformat(project.to_dict()))
|
||||||
|
|
||||||
|
url = urllib.parse.urljoin(PILLAR_WEB_SERVER_URL, 'p/' + project.url)
|
||||||
|
webbrowser.open_new_tab(url)
|
||||||
|
self.report({'INFO'}, 'Opened a browser at %s' % url)
|
||||||
|
|
||||||
|
return {'FINISHED'}
|
||||||
|
|
||||||
|
|
||||||
class PILLAR_OT_projects(async_loop.AsyncModalOperatorMixin,
|
class PILLAR_OT_projects(async_loop.AsyncModalOperatorMixin,
|
||||||
pillar.AuthenticatedPillarOperatorMixin,
|
pillar.AuthenticatedPillarOperatorMixin,
|
||||||
Operator):
|
Operator):
|
||||||
@@ -571,7 +604,7 @@ class PILLAR_OT_projects(async_loop.AsyncModalOperatorMixin,
|
|||||||
pillarsdk.Project.all,
|
pillarsdk.Project.all,
|
||||||
{'where': {'user': self.user_id,
|
{'where': {'user': self.user_id,
|
||||||
'category': {'$ne': 'home'}},
|
'category': {'$ne': 'home'}},
|
||||||
'sort': '-_created',
|
'sort': '-name',
|
||||||
'projection': {'_id': True,
|
'projection': {'_id': True,
|
||||||
'name': True,
|
'name': True,
|
||||||
'extension_props': True},
|
'extension_props': True},
|
||||||
@@ -581,7 +614,7 @@ class PILLAR_OT_projects(async_loop.AsyncModalOperatorMixin,
|
|||||||
pillarsdk.Project.all,
|
pillarsdk.Project.all,
|
||||||
{'where': {'user': {'$ne': self.user_id},
|
{'where': {'user': {'$ne': self.user_id},
|
||||||
'permissions.groups.group': {'$in': self.db_user.groups}},
|
'permissions.groups.group': {'$in': self.db_user.groups}},
|
||||||
'sort': '-_created',
|
'sort': '-name',
|
||||||
'projection': {'_id': True,
|
'projection': {'_id': True,
|
||||||
'name': True,
|
'name': True,
|
||||||
'extension_props': True},
|
'extension_props': True},
|
||||||
@@ -605,7 +638,10 @@ class PILLAR_OT_projects(async_loop.AsyncModalOperatorMixin,
|
|||||||
projects = list(reduce_properties(projects_user['_items'])) + \
|
projects = list(reduce_properties(projects_user['_items'])) + \
|
||||||
list(reduce_properties(projects_shared['_items']))
|
list(reduce_properties(projects_shared['_items']))
|
||||||
|
|
||||||
preferences().project.available_projects = projects
|
def proj_sort_key(project):
|
||||||
|
return project.get('name')
|
||||||
|
|
||||||
|
preferences().project.available_projects = sorted(projects, key=proj_sort_key)
|
||||||
|
|
||||||
self.quit()
|
self.quit()
|
||||||
|
|
||||||
@@ -625,8 +661,16 @@ class PILLAR_PT_image_custom_properties(rna_prop_ui.PropertyPanel, bpy.types.Pan
|
|||||||
_property_type = bpy.types.Image
|
_property_type = bpy.types.Image
|
||||||
|
|
||||||
|
|
||||||
|
def ctx_preferences():
|
||||||
|
"""Returns bpy.context.preferences in a 2.79-compatible way."""
|
||||||
|
try:
|
||||||
|
return bpy.context.preferences
|
||||||
|
except AttributeError:
|
||||||
|
return bpy.context.user_preferences
|
||||||
|
|
||||||
|
|
||||||
def preferences() -> BlenderCloudPreferences:
|
def preferences() -> BlenderCloudPreferences:
|
||||||
return bpy.context.user_preferences.addons[ADDON_NAME].preferences
|
return ctx_preferences().addons[ADDON_NAME].preferences
|
||||||
|
|
||||||
|
|
||||||
def load_custom_icons():
|
def load_custom_icons():
|
||||||
@@ -669,6 +713,7 @@ def register():
|
|||||||
bpy.utils.register_class(SyncStatusProperties)
|
bpy.utils.register_class(SyncStatusProperties)
|
||||||
bpy.utils.register_class(PILLAR_OT_subscribe)
|
bpy.utils.register_class(PILLAR_OT_subscribe)
|
||||||
bpy.utils.register_class(PILLAR_OT_projects)
|
bpy.utils.register_class(PILLAR_OT_projects)
|
||||||
|
bpy.utils.register_class(PILLAR_OT_project_open_in_browser)
|
||||||
bpy.utils.register_class(PILLAR_PT_image_custom_properties)
|
bpy.utils.register_class(PILLAR_PT_image_custom_properties)
|
||||||
|
|
||||||
addon_prefs = preferences()
|
addon_prefs = preferences()
|
||||||
@@ -703,6 +748,7 @@ def unregister():
|
|||||||
bpy.utils.unregister_class(SyncStatusProperties)
|
bpy.utils.unregister_class(SyncStatusProperties)
|
||||||
bpy.utils.unregister_class(PILLAR_OT_subscribe)
|
bpy.utils.unregister_class(PILLAR_OT_subscribe)
|
||||||
bpy.utils.unregister_class(PILLAR_OT_projects)
|
bpy.utils.unregister_class(PILLAR_OT_projects)
|
||||||
|
bpy.utils.unregister_class(PILLAR_OT_project_open_in_browser)
|
||||||
bpy.utils.unregister_class(PILLAR_PT_image_custom_properties)
|
bpy.utils.unregister_class(PILLAR_PT_image_custom_properties)
|
||||||
|
|
||||||
del WindowManager.last_blender_cloud_location
|
del WindowManager.last_blender_cloud_location
|
||||||
|
70
blender_cloud/compatibility.py
Normal file
70
blender_cloud/compatibility.py
Normal file
@@ -0,0 +1,70 @@
|
|||||||
|
"""Compatibility functions to support Blender 2.79 and 2.80+ in one code base."""
|
||||||
|
import functools
|
||||||
|
|
||||||
|
import bpy
|
||||||
|
|
||||||
|
|
||||||
|
if bpy.app.version < (2, 80):
|
||||||
|
SYNC_SELECT_VERSION_ICON = 'DOTSDOWN'
|
||||||
|
else:
|
||||||
|
SYNC_SELECT_VERSION_ICON = 'DOWNARROW_HLT'
|
||||||
|
|
||||||
|
|
||||||
|
# Get references to all property definition functions in bpy.props,
|
||||||
|
# so that they can be used to replace 'x = IntProperty()' to 'x: IntProperty()'
|
||||||
|
# dynamically when working on Blender 2.80+
|
||||||
|
__all_prop_funcs = {
|
||||||
|
getattr(bpy.props, propname)
|
||||||
|
for propname in dir(bpy.props)
|
||||||
|
if propname.endswith('Property')
|
||||||
|
}
|
||||||
|
|
||||||
|
def convert_properties(class_):
|
||||||
|
"""Class decorator to avoid warnings in Blender 2.80+
|
||||||
|
|
||||||
|
This decorator replaces property definitions like this:
|
||||||
|
|
||||||
|
someprop = bpy.props.IntProperty()
|
||||||
|
|
||||||
|
to annotations, as introduced in Blender 2.80:
|
||||||
|
|
||||||
|
someprop: bpy.props.IntProperty()
|
||||||
|
|
||||||
|
No-op if running on Blender 2.79 or older.
|
||||||
|
"""
|
||||||
|
|
||||||
|
if bpy.app.version < (2, 80):
|
||||||
|
return class_
|
||||||
|
|
||||||
|
if not hasattr(class_, '__annotations__'):
|
||||||
|
class_.__annotations__ = {}
|
||||||
|
|
||||||
|
attrs_to_delete = []
|
||||||
|
for name, value in class_.__dict__.items():
|
||||||
|
if not isinstance(value, tuple) or len(value) != 2:
|
||||||
|
continue
|
||||||
|
|
||||||
|
prop_func, kwargs = value
|
||||||
|
if prop_func not in __all_prop_funcs:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# This is a property definition, replace it with annotation.
|
||||||
|
attrs_to_delete.append(name)
|
||||||
|
class_.__annotations__[name] = value
|
||||||
|
|
||||||
|
for attr_name in attrs_to_delete:
|
||||||
|
delattr(class_, attr_name)
|
||||||
|
|
||||||
|
return class_
|
||||||
|
|
||||||
|
|
||||||
|
@functools.lru_cache()
|
||||||
|
def factor(factor: float) -> dict:
|
||||||
|
"""Construct keyword argument for UILayout.split().
|
||||||
|
|
||||||
|
On Blender 2.8 this returns {'factor': factor}, and on earlier Blenders it returns
|
||||||
|
{'percentage': factor}.
|
||||||
|
"""
|
||||||
|
if bpy.app.version < (2, 80, 0):
|
||||||
|
return {'percentage': factor}
|
||||||
|
return {'factor': factor}
|
File diff suppressed because it is too large
Load Diff
@@ -1,148 +0,0 @@
|
|||||||
"""BAM packing interface for Flamenco."""
|
|
||||||
|
|
||||||
import logging
|
|
||||||
from pathlib import Path
|
|
||||||
import typing
|
|
||||||
|
|
||||||
# Timeout of the BAM subprocess, in seconds.
|
|
||||||
SUBPROC_READLINE_TIMEOUT = 600
|
|
||||||
log = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
class CommandExecutionError(Exception):
|
|
||||||
"""Raised when there was an error executing a BAM command."""
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
async def bam_copy(base_blendfile: Path, target_blendfile: Path,
|
|
||||||
exclusion_filter: str) -> typing.List[Path]:
|
|
||||||
"""Uses BAM to copy the given file and dependencies to the target blendfile.
|
|
||||||
|
|
||||||
Due to the way blendfile_pack.py is programmed/structured, we cannot import it
|
|
||||||
and call a function; it has to be run in a subprocess.
|
|
||||||
|
|
||||||
:raises: asyncio.CanceledError if the task was cancelled.
|
|
||||||
:raises: asyncio.TimeoutError if reading a line from the BAM process timed out.
|
|
||||||
:raises: CommandExecutionError if the subprocess failed or output invalid UTF-8.
|
|
||||||
:returns: a list of missing sources; hopefully empty.
|
|
||||||
"""
|
|
||||||
|
|
||||||
import asyncio
|
|
||||||
import shlex
|
|
||||||
import subprocess
|
|
||||||
|
|
||||||
import bpy
|
|
||||||
import io_blend_utils
|
|
||||||
|
|
||||||
args = [
|
|
||||||
bpy.app.binary_path_python,
|
|
||||||
'-m', 'bam.pack',
|
|
||||||
'--input', str(base_blendfile),
|
|
||||||
'--output', str(target_blendfile),
|
|
||||||
'--mode', 'FILE',
|
|
||||||
]
|
|
||||||
|
|
||||||
if exclusion_filter:
|
|
||||||
args.extend(['--exclude', exclusion_filter])
|
|
||||||
|
|
||||||
cmd_to_log = ' '.join(shlex.quote(s) for s in args)
|
|
||||||
log.info('Executing %s', cmd_to_log)
|
|
||||||
|
|
||||||
proc = await asyncio.create_subprocess_exec(
|
|
||||||
*args,
|
|
||||||
env={'PYTHONPATH': io_blend_utils.pythonpath()},
|
|
||||||
stdin=subprocess.DEVNULL,
|
|
||||||
stdout=subprocess.PIPE,
|
|
||||||
stderr=subprocess.STDOUT,
|
|
||||||
)
|
|
||||||
|
|
||||||
missing_sources = []
|
|
||||||
|
|
||||||
try:
|
|
||||||
while not proc.stdout.at_eof():
|
|
||||||
line = await asyncio.wait_for(proc.stdout.readline(),
|
|
||||||
SUBPROC_READLINE_TIMEOUT)
|
|
||||||
if not line:
|
|
||||||
# EOF received, so let's bail.
|
|
||||||
break
|
|
||||||
|
|
||||||
try:
|
|
||||||
line = line.decode('utf8')
|
|
||||||
except UnicodeDecodeError as ex:
|
|
||||||
raise CommandExecutionError('Command produced non-UTF8 output, '
|
|
||||||
'aborting: %s' % ex)
|
|
||||||
|
|
||||||
line = line.rstrip()
|
|
||||||
if 'source missing:' in line:
|
|
||||||
path = parse_missing_source(line)
|
|
||||||
missing_sources.append(path)
|
|
||||||
log.warning('Source is missing: %s', path)
|
|
||||||
|
|
||||||
log.info(' %s', line)
|
|
||||||
finally:
|
|
||||||
if proc.returncode is None:
|
|
||||||
# Always wait for the process, to avoid zombies.
|
|
||||||
try:
|
|
||||||
proc.kill()
|
|
||||||
except ProcessLookupError:
|
|
||||||
# The process is already stopped, so killing is impossible. That's ok.
|
|
||||||
log.debug("The process was already stopped, aborting is impossible. That's ok.")
|
|
||||||
await proc.wait()
|
|
||||||
log.info('The process stopped with status code %i', proc.returncode)
|
|
||||||
|
|
||||||
if proc.returncode:
|
|
||||||
raise CommandExecutionError('Process stopped with status %i' % proc.returncode)
|
|
||||||
|
|
||||||
return missing_sources
|
|
||||||
|
|
||||||
|
|
||||||
def parse_missing_source(line: str) -> Path:
|
|
||||||
r"""Parses a "missing source" line into a pathlib.Path.
|
|
||||||
|
|
||||||
>>> parse_missing_source(r" source missing: b'D\xc3\xaffficult \xc3\x9cTF-8 filename'")
|
|
||||||
PosixPath('Dïfficult ÜTF-8 filename')
|
|
||||||
>>> parse_missing_source(r" source missing: b'D\xfffficult Win1252 f\xeflen\xe6me'")
|
|
||||||
PosixPath('D<EFBFBD>fficult Win1252 f<>len<65>me')
|
|
||||||
"""
|
|
||||||
|
|
||||||
_, missing_source = line.split(': ', 1)
|
|
||||||
missing_source_as_bytes = parse_byte_literal(missing_source.strip())
|
|
||||||
|
|
||||||
# The file could originate from any platform, so UTF-8 and the current platform's
|
|
||||||
# filesystem encodings are just guesses.
|
|
||||||
try:
|
|
||||||
missing_source = missing_source_as_bytes.decode('utf8')
|
|
||||||
except UnicodeDecodeError:
|
|
||||||
import sys
|
|
||||||
try:
|
|
||||||
missing_source = missing_source_as_bytes.decode(sys.getfilesystemencoding())
|
|
||||||
except UnicodeDecodeError:
|
|
||||||
missing_source = missing_source_as_bytes.decode('ascii', errors='replace')
|
|
||||||
|
|
||||||
path = Path(missing_source)
|
|
||||||
|
|
||||||
return path
|
|
||||||
|
|
||||||
|
|
||||||
def parse_byte_literal(bytes_literal: str) -> bytes:
|
|
||||||
r"""Parses a repr(bytes) output into a bytes object.
|
|
||||||
|
|
||||||
>>> parse_byte_literal(r"b'D\xc3\xaffficult \xc3\x9cTF-8 filename'")
|
|
||||||
b'D\xc3\xaffficult \xc3\x9cTF-8 filename'
|
|
||||||
>>> parse_byte_literal(r"b'D\xeffficult Win1252 f\xeflen\xe6me'")
|
|
||||||
b'D\xeffficult Win1252 f\xeflen\xe6me'
|
|
||||||
"""
|
|
||||||
|
|
||||||
# Some very basic assertions to make sure we have a proper bytes literal.
|
|
||||||
assert bytes_literal[0] == "b"
|
|
||||||
assert bytes_literal[1] in {'"', "'"}
|
|
||||||
assert bytes_literal[-1] == bytes_literal[1]
|
|
||||||
|
|
||||||
import ast
|
|
||||||
return ast.literal_eval(bytes_literal)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
import doctest
|
|
||||||
|
|
||||||
doctest.testmod()
|
|
190
blender_cloud/flamenco/bat_interface.py
Normal file
190
blender_cloud/flamenco/bat_interface.py
Normal file
@@ -0,0 +1,190 @@
|
|||||||
|
"""BAT🦇 packing interface for Flamenco."""
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
import logging
|
||||||
|
import pathlib
|
||||||
|
import re
|
||||||
|
import threading
|
||||||
|
import typing
|
||||||
|
import urllib.parse
|
||||||
|
|
||||||
|
import bpy
|
||||||
|
from blender_asset_tracer import pack
|
||||||
|
from blender_asset_tracer.pack import progress, transfer, shaman
|
||||||
|
|
||||||
|
log = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
_running_packer = None # type: pack.Packer
|
||||||
|
_packer_lock = threading.RLock()
|
||||||
|
|
||||||
|
# For using in other parts of the add-on, so only this file imports BAT.
|
||||||
|
Aborted = pack.Aborted
|
||||||
|
FileTransferError = transfer.FileTransferError
|
||||||
|
parse_shaman_endpoint = shaman.parse_endpoint
|
||||||
|
|
||||||
|
|
||||||
|
class BatProgress(progress.Callback):
|
||||||
|
"""Report progress of BAT Packing to the UI.
|
||||||
|
|
||||||
|
Uses asyncio.run_coroutine_threadsafe() to ensure the UI is only updated
|
||||||
|
from the main thread. This is required since we run the BAT Pack in a
|
||||||
|
background thread.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self) -> None:
|
||||||
|
super().__init__()
|
||||||
|
self.loop = asyncio.get_event_loop()
|
||||||
|
|
||||||
|
def _set_attr(self, attr: str, value):
|
||||||
|
|
||||||
|
async def do_it():
|
||||||
|
setattr(bpy.context.window_manager, attr, value)
|
||||||
|
|
||||||
|
asyncio.run_coroutine_threadsafe(do_it(), loop=self.loop)
|
||||||
|
|
||||||
|
def _txt(self, msg: str):
|
||||||
|
"""Set a text in a thread-safe way."""
|
||||||
|
self._set_attr('flamenco_status_txt', msg)
|
||||||
|
|
||||||
|
def _status(self, status: str):
|
||||||
|
"""Set the flamenco_status property in a thread-safe way."""
|
||||||
|
self._set_attr('flamenco_status', status)
|
||||||
|
|
||||||
|
def _progress(self, progress: int):
|
||||||
|
"""Set the flamenco_progress property in a thread-safe way."""
|
||||||
|
self._set_attr('flamenco_progress', progress)
|
||||||
|
|
||||||
|
def pack_start(self) -> None:
|
||||||
|
self._txt('Starting BAT Pack operation')
|
||||||
|
|
||||||
|
def pack_done(self,
|
||||||
|
output_blendfile: pathlib.Path,
|
||||||
|
missing_files: typing.Set[pathlib.Path]) -> None:
|
||||||
|
if missing_files:
|
||||||
|
self._txt('There were %d missing files' % len(missing_files))
|
||||||
|
else:
|
||||||
|
self._txt('Pack of %s done' % output_blendfile.name)
|
||||||
|
|
||||||
|
def pack_aborted(self, reason: str):
|
||||||
|
self._txt('Aborted: %s' % reason)
|
||||||
|
self._status('ABORTED')
|
||||||
|
|
||||||
|
def trace_blendfile(self, filename: pathlib.Path) -> None:
|
||||||
|
"""Called for every blendfile opened when tracing dependencies."""
|
||||||
|
self._txt('Inspecting %s' % filename.name)
|
||||||
|
|
||||||
|
def trace_asset(self, filename: pathlib.Path) -> None:
|
||||||
|
if filename.stem == '.blend':
|
||||||
|
return
|
||||||
|
self._txt('Found asset %s' % filename.name)
|
||||||
|
|
||||||
|
def rewrite_blendfile(self, orig_filename: pathlib.Path) -> None:
|
||||||
|
self._txt('Rewriting %s' % orig_filename.name)
|
||||||
|
|
||||||
|
def transfer_file(self, src: pathlib.Path, dst: pathlib.Path) -> None:
|
||||||
|
self._txt('Transferring %s' % src.name)
|
||||||
|
|
||||||
|
def transfer_file_skipped(self, src: pathlib.Path, dst: pathlib.Path) -> None:
|
||||||
|
self._txt('Skipped %s' % src.name)
|
||||||
|
|
||||||
|
def transfer_progress(self, total_bytes: int, transferred_bytes: int) -> None:
|
||||||
|
self._progress(round(100 * transferred_bytes / total_bytes))
|
||||||
|
|
||||||
|
def missing_file(self, filename: pathlib.Path) -> None:
|
||||||
|
# TODO(Sybren): report missing files in a nice way
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class ShamanPacker(shaman.ShamanPacker):
|
||||||
|
"""Packer with support for getting an auth token from Flamenco Server."""
|
||||||
|
|
||||||
|
def __init__(self,
|
||||||
|
bfile: pathlib.Path,
|
||||||
|
project: pathlib.Path,
|
||||||
|
target: str,
|
||||||
|
endpoint: str,
|
||||||
|
checkout_id: str,
|
||||||
|
*,
|
||||||
|
manager_id: str,
|
||||||
|
**kwargs) -> None:
|
||||||
|
self.manager_id = manager_id
|
||||||
|
super().__init__(bfile, project, target, endpoint, checkout_id, **kwargs)
|
||||||
|
|
||||||
|
def _get_auth_token(self) -> str:
|
||||||
|
"""get a token from Flamenco Server"""
|
||||||
|
|
||||||
|
from ..blender import PILLAR_SERVER_URL
|
||||||
|
from ..pillar import blender_id_subclient, uncached_session, SUBCLIENT_ID
|
||||||
|
|
||||||
|
url = urllib.parse.urljoin(PILLAR_SERVER_URL,
|
||||||
|
'flamenco/jwt/generate-token/%s' % self.manager_id)
|
||||||
|
auth_token = blender_id_subclient()['token']
|
||||||
|
|
||||||
|
resp = uncached_session.get(url, auth=(auth_token, SUBCLIENT_ID))
|
||||||
|
resp.raise_for_status()
|
||||||
|
return resp.text
|
||||||
|
|
||||||
|
|
||||||
|
async def copy(context,
|
||||||
|
base_blendfile: pathlib.Path,
|
||||||
|
project: pathlib.Path,
|
||||||
|
target: str,
|
||||||
|
exclusion_filter: str,
|
||||||
|
*,
|
||||||
|
relative_only: bool,
|
||||||
|
packer_class=pack.Packer,
|
||||||
|
**packer_args) \
|
||||||
|
-> typing.Tuple[pathlib.Path, typing.Set[pathlib.Path]]:
|
||||||
|
"""Use BAT🦇 to copy the given file and dependencies to the target location.
|
||||||
|
|
||||||
|
:raises: FileTransferError if a file couldn't be transferred.
|
||||||
|
:returns: the path of the packed blend file, and a set of missing sources.
|
||||||
|
"""
|
||||||
|
global _running_packer
|
||||||
|
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
|
wm = bpy.context.window_manager
|
||||||
|
|
||||||
|
packer = packer_class(base_blendfile, project, target,
|
||||||
|
compress=True, relative_only=relative_only, **packer_args)
|
||||||
|
with packer:
|
||||||
|
with _packer_lock:
|
||||||
|
if exclusion_filter:
|
||||||
|
# There was a mistake in an older version of the property tooltip,
|
||||||
|
# showing semicolon-separated instead of space-separated. We now
|
||||||
|
# just handle both.
|
||||||
|
filter_parts = re.split('[ ;]+', exclusion_filter.strip(' ;'))
|
||||||
|
packer.exclude(*filter_parts)
|
||||||
|
|
||||||
|
packer.progress_cb = BatProgress()
|
||||||
|
_running_packer = packer
|
||||||
|
|
||||||
|
log.debug('awaiting strategise')
|
||||||
|
wm.flamenco_status = 'INVESTIGATING'
|
||||||
|
await loop.run_in_executor(None, packer.strategise)
|
||||||
|
|
||||||
|
log.debug('awaiting execute')
|
||||||
|
wm.flamenco_status = 'TRANSFERRING'
|
||||||
|
await loop.run_in_executor(None, packer.execute)
|
||||||
|
|
||||||
|
log.debug('done')
|
||||||
|
wm.flamenco_status = 'DONE'
|
||||||
|
|
||||||
|
with _packer_lock:
|
||||||
|
_running_packer = None
|
||||||
|
|
||||||
|
return packer.output_path, packer.missing_files
|
||||||
|
|
||||||
|
|
||||||
|
def abort() -> None:
|
||||||
|
"""Abort a running copy() call.
|
||||||
|
|
||||||
|
No-op when there is no running copy(). Can be called from any thread.
|
||||||
|
"""
|
||||||
|
|
||||||
|
with _packer_lock:
|
||||||
|
if _running_packer is None:
|
||||||
|
log.debug('No running packer, ignoring call to bat_abort()')
|
||||||
|
return
|
||||||
|
log.info('Aborting running packer')
|
||||||
|
_running_packer.abort()
|
@@ -1,9 +1,94 @@
|
|||||||
|
import functools
|
||||||
|
import pathlib
|
||||||
|
import typing
|
||||||
|
|
||||||
from pillarsdk.resource import List, Find, Create
|
from pillarsdk.resource import List, Find, Create
|
||||||
|
|
||||||
|
|
||||||
class Manager(List, Find):
|
class Manager(List, Find):
|
||||||
"""Manager class wrapping the REST nodes endpoint"""
|
"""Manager class wrapping the REST nodes endpoint"""
|
||||||
path = 'flamenco/managers'
|
path = 'flamenco/managers'
|
||||||
|
PurePlatformPath = pathlib.PurePath
|
||||||
|
|
||||||
|
@functools.lru_cache(maxsize=1)
|
||||||
|
def _path_replacements(self) -> list:
|
||||||
|
"""Defer to _path_replacements_vN() to get path replacement vars.
|
||||||
|
|
||||||
|
Returns a list of tuples (variable name, variable value).
|
||||||
|
"""
|
||||||
|
settings_version = self.settings_version or 1
|
||||||
|
try:
|
||||||
|
settings_func = getattr(self, '_path_replacements_v%d' % settings_version)
|
||||||
|
except AttributeError:
|
||||||
|
raise RuntimeError('This manager has unsupported settings version %d; '
|
||||||
|
'upgrade Blender Cloud add-on')
|
||||||
|
|
||||||
|
def longest_value_first(item):
|
||||||
|
var_name, var_value = item
|
||||||
|
return -len(var_value), var_value, var_name
|
||||||
|
|
||||||
|
replacements = settings_func()
|
||||||
|
replacements.sort(key=longest_value_first)
|
||||||
|
return replacements
|
||||||
|
|
||||||
|
def _path_replacements_v1(self) -> typing.List[typing.Tuple[str, str]]:
|
||||||
|
import platform
|
||||||
|
|
||||||
|
if self.path_replacement is None:
|
||||||
|
return []
|
||||||
|
|
||||||
|
items = self.path_replacement.to_dict().items()
|
||||||
|
|
||||||
|
this_platform = platform.system().lower()
|
||||||
|
return [(varname, platform_replacements[this_platform])
|
||||||
|
for varname, platform_replacements in items
|
||||||
|
if this_platform in platform_replacements]
|
||||||
|
|
||||||
|
def _path_replacements_v2(self) -> typing.List[typing.Tuple[str, str]]:
|
||||||
|
import platform
|
||||||
|
|
||||||
|
if not self.variables:
|
||||||
|
return []
|
||||||
|
|
||||||
|
this_platform = platform.system().lower()
|
||||||
|
audiences = {'users', 'all'}
|
||||||
|
|
||||||
|
replacements = []
|
||||||
|
for var_name, variable in self.variables.to_dict().items():
|
||||||
|
# Path replacement requires bidirectional variables.
|
||||||
|
if variable.get('direction') != 'twoway':
|
||||||
|
continue
|
||||||
|
|
||||||
|
for var_value in variable.get('values', []):
|
||||||
|
if var_value.get('audience') not in audiences:
|
||||||
|
continue
|
||||||
|
if var_value.get('platform', '').lower() != this_platform:
|
||||||
|
continue
|
||||||
|
|
||||||
|
replacements.append((var_name, var_value.get('value')))
|
||||||
|
return replacements
|
||||||
|
|
||||||
|
def replace_path(self, some_path: pathlib.PurePath) -> str:
|
||||||
|
"""Performs path variable replacement.
|
||||||
|
|
||||||
|
Tries to find platform-specific path prefixes, and replaces them with
|
||||||
|
variables.
|
||||||
|
"""
|
||||||
|
assert isinstance(some_path, pathlib.PurePath), \
|
||||||
|
'some_path should be a PurePath, not %r' % some_path
|
||||||
|
|
||||||
|
for varname, path in self._path_replacements():
|
||||||
|
replacement = self.PurePlatformPath(path)
|
||||||
|
try:
|
||||||
|
relpath = some_path.relative_to(replacement)
|
||||||
|
except ValueError:
|
||||||
|
# Not relative to each other, so no replacement possible
|
||||||
|
continue
|
||||||
|
|
||||||
|
replacement_root = self.PurePlatformPath('{%s}' % varname)
|
||||||
|
return (replacement_root / relpath).as_posix()
|
||||||
|
|
||||||
|
return some_path.as_posix()
|
||||||
|
|
||||||
|
|
||||||
class Job(List, Find, Create):
|
class Job(List, Find, Create):
|
||||||
@@ -11,3 +96,14 @@ class Job(List, Find, Create):
|
|||||||
"""
|
"""
|
||||||
path = 'flamenco/jobs'
|
path = 'flamenco/jobs'
|
||||||
ensure_query_projections = {'project': 1}
|
ensure_query_projections = {'project': 1}
|
||||||
|
|
||||||
|
def patch(self, payload: dict, api=None):
|
||||||
|
import pillarsdk.utils
|
||||||
|
|
||||||
|
api = api or self.api
|
||||||
|
|
||||||
|
url = pillarsdk.utils.join_url(self.path, str(self['_id']))
|
||||||
|
headers = pillarsdk.utils.merge_dict(self.http_headers(),
|
||||||
|
{'Content-Type': 'application/json'})
|
||||||
|
response = api.patch(url, payload, headers=headers)
|
||||||
|
return response
|
||||||
|
@@ -25,7 +25,7 @@ import bpy
|
|||||||
import pillarsdk
|
import pillarsdk
|
||||||
from pillarsdk import exceptions as sdk_exceptions
|
from pillarsdk import exceptions as sdk_exceptions
|
||||||
from .pillar import pillar_call
|
from .pillar import pillar_call
|
||||||
from . import async_loop, pillar, home_project, blender
|
from . import async_loop, compatibility, pillar, home_project, blender
|
||||||
|
|
||||||
REQUIRES_ROLES_FOR_IMAGE_SHARING = {'subscriber', 'demo'}
|
REQUIRES_ROLES_FOR_IMAGE_SHARING = {'subscriber', 'demo'}
|
||||||
IMAGE_SHARING_GROUP_NODE_NAME = 'Image sharing'
|
IMAGE_SHARING_GROUP_NODE_NAME = 'Image sharing'
|
||||||
@@ -53,6 +53,7 @@ async def find_image_sharing_group_id(home_project_id, user_id):
|
|||||||
return share_group['_id']
|
return share_group['_id']
|
||||||
|
|
||||||
|
|
||||||
|
@compatibility.convert_properties
|
||||||
class PILLAR_OT_image_share(pillar.PillarOperatorMixin,
|
class PILLAR_OT_image_share(pillar.PillarOperatorMixin,
|
||||||
async_loop.AsyncModalOperatorMixin,
|
async_loop.AsyncModalOperatorMixin,
|
||||||
bpy.types.Operator):
|
bpy.types.Operator):
|
||||||
@@ -112,6 +113,10 @@ class PILLAR_OT_image_share(pillar.PillarOperatorMixin,
|
|||||||
async def async_execute(self, context):
|
async def async_execute(self, context):
|
||||||
"""Entry point of the asynchronous operator."""
|
"""Entry point of the asynchronous operator."""
|
||||||
|
|
||||||
|
# We don't want to influence what is included in the screen shot.
|
||||||
|
if self.target == 'SCREENSHOT':
|
||||||
|
print('Blender Cloud add-on is communicating with Blender Cloud')
|
||||||
|
else:
|
||||||
self.report({'INFO'}, 'Communicating with Blender Cloud')
|
self.report({'INFO'}, 'Communicating with Blender Cloud')
|
||||||
|
|
||||||
try:
|
try:
|
||||||
@@ -120,9 +125,8 @@ class PILLAR_OT_image_share(pillar.PillarOperatorMixin,
|
|||||||
db_user = await self.check_credentials(context, REQUIRES_ROLES_FOR_IMAGE_SHARING)
|
db_user = await self.check_credentials(context, REQUIRES_ROLES_FOR_IMAGE_SHARING)
|
||||||
self.user_id = db_user['_id']
|
self.user_id = db_user['_id']
|
||||||
self.log.debug('Found user ID: %s', self.user_id)
|
self.log.debug('Found user ID: %s', self.user_id)
|
||||||
except pillar.NotSubscribedToCloudError:
|
except pillar.NotSubscribedToCloudError as ex:
|
||||||
self.log.exception('User not subscribed to cloud.')
|
self._log_subscription_needed(can_renew=ex.can_renew)
|
||||||
self.report({'ERROR'}, 'Please subscribe to the Blender Cloud.')
|
|
||||||
self._state = 'QUIT'
|
self._state = 'QUIT'
|
||||||
return
|
return
|
||||||
except pillar.UserNotLoggedInError:
|
except pillar.UserNotLoggedInError:
|
||||||
@@ -320,15 +324,25 @@ def window_menu(self, context):
|
|||||||
props.screenshot_full = True
|
props.screenshot_full = True
|
||||||
|
|
||||||
|
|
||||||
|
def get_topbar_menu():
|
||||||
|
"""Return the topbar menu in a Blender 2.79 and 2.80 compatible way."""
|
||||||
|
try:
|
||||||
|
menu = bpy.types.TOPBAR_MT_window
|
||||||
|
except AttributeError:
|
||||||
|
# Blender < 2.80
|
||||||
|
menu = bpy.types.INFO_MT_window
|
||||||
|
return menu
|
||||||
|
|
||||||
|
|
||||||
def register():
|
def register():
|
||||||
bpy.utils.register_class(PILLAR_OT_image_share)
|
bpy.utils.register_class(PILLAR_OT_image_share)
|
||||||
|
|
||||||
bpy.types.IMAGE_MT_image.append(image_editor_menu)
|
bpy.types.IMAGE_MT_image.append(image_editor_menu)
|
||||||
bpy.types.INFO_MT_window.append(window_menu)
|
get_topbar_menu().append(window_menu)
|
||||||
|
|
||||||
|
|
||||||
def unregister():
|
def unregister():
|
||||||
bpy.utils.unregister_class(PILLAR_OT_image_share)
|
bpy.utils.unregister_class(PILLAR_OT_image_share)
|
||||||
|
|
||||||
bpy.types.IMAGE_MT_image.remove(image_editor_menu)
|
bpy.types.IMAGE_MT_image.remove(image_editor_menu)
|
||||||
bpy.types.INFO_MT_window.remove(window_menu)
|
get_topbar_menu().remove(window_menu)
|
||||||
|
102
blender_cloud/pillar.py
Normal file → Executable file
102
blender_cloud/pillar.py
Normal file → Executable file
@@ -26,7 +26,8 @@ from contextlib import closing, contextmanager
|
|||||||
import urllib.parse
|
import urllib.parse
|
||||||
import pathlib
|
import pathlib
|
||||||
|
|
||||||
import requests
|
import requests.adapters
|
||||||
|
import requests.packages.urllib3.util.retry
|
||||||
import requests.structures
|
import requests.structures
|
||||||
import pillarsdk
|
import pillarsdk
|
||||||
import pillarsdk.exceptions
|
import pillarsdk.exceptions
|
||||||
@@ -42,7 +43,16 @@ RFC1123_DATE_FORMAT = '%a, %d %b %Y %H:%M:%S GMT'
|
|||||||
|
|
||||||
_pillar_api = {} # will become a mapping from bool (cached/non-cached) to pillarsdk.Api objects.
|
_pillar_api = {} # will become a mapping from bool (cached/non-cached) to pillarsdk.Api objects.
|
||||||
log = logging.getLogger(__name__)
|
log = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
_retries = requests.packages.urllib3.util.retry.Retry(
|
||||||
|
total=10,
|
||||||
|
backoff_factor=0.05,
|
||||||
|
)
|
||||||
|
_http_adapter = requests.adapters.HTTPAdapter(max_retries=_retries)
|
||||||
uncached_session = requests.session()
|
uncached_session = requests.session()
|
||||||
|
uncached_session.mount('https://', _http_adapter)
|
||||||
|
uncached_session.mount('http://', _http_adapter)
|
||||||
|
|
||||||
_testing_blender_id_profile = None # Just for testing, overrides what is returned by blender_id_profile.
|
_testing_blender_id_profile = None # Just for testing, overrides what is returned by blender_id_profile.
|
||||||
_downloaded_urls = set() # URLs we've downloaded this Blender session.
|
_downloaded_urls = set() # URLs we've downloaded this Blender session.
|
||||||
|
|
||||||
@@ -62,7 +72,16 @@ class CredentialsNotSyncedError(UserNotLoggedInError):
|
|||||||
|
|
||||||
|
|
||||||
class NotSubscribedToCloudError(UserNotLoggedInError):
|
class NotSubscribedToCloudError(UserNotLoggedInError):
|
||||||
"""Raised when the user may be logged in on Blender ID, but has no Blender Cloud token."""
|
"""Raised when the user does not have an active Cloud subscription.
|
||||||
|
|
||||||
|
:ivar can_renew: True when the user has an inactive subscription that can be renewed,
|
||||||
|
or False when the user has no subscription at all.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, can_renew: bool):
|
||||||
|
super().__init__()
|
||||||
|
self.can_renew = can_renew
|
||||||
|
log.warning('Not subscribed to cloud, can_renew=%s', can_renew)
|
||||||
|
|
||||||
|
|
||||||
class PillarError(RuntimeError):
|
class PillarError(RuntimeError):
|
||||||
@@ -87,18 +106,18 @@ class CloudPath(pathlib.PurePosixPath):
|
|||||||
def project_uuid(self) -> str:
|
def project_uuid(self) -> str:
|
||||||
assert self.parts[0] == '/'
|
assert self.parts[0] == '/'
|
||||||
if len(self.parts) <= 1:
|
if len(self.parts) <= 1:
|
||||||
return None
|
return ''
|
||||||
return self.parts[1]
|
return self.parts[1]
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def node_uuids(self) -> list:
|
def node_uuids(self) -> tuple:
|
||||||
assert self.parts[0] == '/'
|
assert self.parts[0] == '/'
|
||||||
return self.parts[2:]
|
return self.parts[2:]
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def node_uuid(self) -> str:
|
def node_uuid(self) -> str:
|
||||||
if len(self.parts) <= 2:
|
if len(self.parts) <= 2:
|
||||||
return None
|
return ''
|
||||||
|
|
||||||
return self.parts[-1]
|
return self.parts[-1]
|
||||||
|
|
||||||
@@ -115,6 +134,12 @@ def with_existing_dir(filename: str, open_mode: str, encoding=None):
|
|||||||
yield file_object
|
yield file_object
|
||||||
|
|
||||||
|
|
||||||
|
def _shorten(somestr: str, maxlen=40) -> str:
|
||||||
|
"""Shortens strings for logging"""
|
||||||
|
|
||||||
|
return (somestr[:maxlen - 3] + '...') if len(somestr) > maxlen else somestr
|
||||||
|
|
||||||
|
|
||||||
def save_as_json(pillar_resource, json_filename):
|
def save_as_json(pillar_resource, json_filename):
|
||||||
with with_existing_dir(json_filename, 'w') as outfile:
|
with with_existing_dir(json_filename, 'w') as outfile:
|
||||||
log.debug('Saving metadata to %r' % json_filename)
|
log.debug('Saving metadata to %r' % json_filename)
|
||||||
@@ -200,8 +225,11 @@ def pillar_api(pillar_endpoint: str = None, caching=True) -> pillarsdk.Api:
|
|||||||
return _pillar_api[caching]
|
return _pillar_api[caching]
|
||||||
|
|
||||||
|
|
||||||
# No more than this many Pillar calls should be made simultaneously
|
# This is an asyncio.Semaphore object, which is late-instantiated to be sure
|
||||||
pillar_semaphore = asyncio.Semaphore(3)
|
# the asyncio loop has been created properly. On Windows we create a new one,
|
||||||
|
# which can cause this semaphore to still be linked against the old default
|
||||||
|
# loop.
|
||||||
|
pillar_semaphore = None
|
||||||
|
|
||||||
|
|
||||||
async def pillar_call(pillar_func, *args, caching=True, **kwargs):
|
async def pillar_call(pillar_func, *args, caching=True, **kwargs):
|
||||||
@@ -214,8 +242,21 @@ async def pillar_call(pillar_func, *args, caching=True, **kwargs):
|
|||||||
partial = functools.partial(pillar_func, *args, api=pillar_api(caching=caching), **kwargs)
|
partial = functools.partial(pillar_func, *args, api=pillar_api(caching=caching), **kwargs)
|
||||||
loop = asyncio.get_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
|
|
||||||
async with pillar_semaphore:
|
# Use explicit calls to acquire() and release() so that we have more control over
|
||||||
|
# how long we wait and how we handle timeouts.
|
||||||
|
try:
|
||||||
|
await asyncio.wait_for(pillar_semaphore.acquire(), timeout=10, loop=loop)
|
||||||
|
except asyncio.TimeoutError:
|
||||||
|
log.info('Waiting for semaphore to call %s', pillar_func.__name__)
|
||||||
|
try:
|
||||||
|
await asyncio.wait_for(pillar_semaphore.acquire(), timeout=50, loop=loop)
|
||||||
|
except asyncio.TimeoutError:
|
||||||
|
raise RuntimeError('Timeout waiting for Pillar Semaphore!')
|
||||||
|
|
||||||
|
try:
|
||||||
return await loop.run_in_executor(None, partial)
|
return await loop.run_in_executor(None, partial)
|
||||||
|
finally:
|
||||||
|
pillar_semaphore.release()
|
||||||
|
|
||||||
|
|
||||||
def sync_call(pillar_func, *args, caching=True, **kwargs):
|
def sync_call(pillar_func, *args, caching=True, **kwargs):
|
||||||
@@ -251,14 +292,15 @@ async def check_pillar_credentials(required_roles: set):
|
|||||||
except (pillarsdk.UnauthorizedAccess, pillarsdk.ResourceNotFound, pillarsdk.ForbiddenAccess):
|
except (pillarsdk.UnauthorizedAccess, pillarsdk.ResourceNotFound, pillarsdk.ForbiddenAccess):
|
||||||
raise CredentialsNotSyncedError()
|
raise CredentialsNotSyncedError()
|
||||||
|
|
||||||
roles = db_user.roles or set()
|
roles = set(db_user.roles or set())
|
||||||
log.debug('User has roles %r', roles)
|
log.getChild('check_pillar_credentials').debug('user has roles %r', roles)
|
||||||
if required_roles and not required_roles.intersection(set(roles)):
|
if required_roles and not required_roles.intersection(roles):
|
||||||
# Delete the subclient info. This forces a re-check later, which can
|
# Delete the subclient info. This forces a re-check later, which can
|
||||||
# then pick up on the user's new status.
|
# then pick up on the user's new status.
|
||||||
del profile.subclients[SUBCLIENT_ID]
|
del profile.subclients[SUBCLIENT_ID]
|
||||||
profile.save_json()
|
profile.save_json()
|
||||||
raise NotSubscribedToCloudError()
|
|
||||||
|
raise NotSubscribedToCloudError(can_renew='has_subscription' in roles)
|
||||||
|
|
||||||
return db_user
|
return db_user
|
||||||
|
|
||||||
@@ -441,9 +483,9 @@ async def download_to_file(url, filename, *,
|
|||||||
log.debug('Downloading was cancelled before doing the GET')
|
log.debug('Downloading was cancelled before doing the GET')
|
||||||
raise asyncio.CancelledError('Downloading was cancelled')
|
raise asyncio.CancelledError('Downloading was cancelled')
|
||||||
|
|
||||||
log.debug('Performing GET %s', url)
|
log.debug('Performing GET %s', _shorten(url))
|
||||||
response = await loop.run_in_executor(None, perform_get_request)
|
response = await loop.run_in_executor(None, perform_get_request)
|
||||||
log.debug('Status %i from GET %s', response.status_code, url)
|
log.debug('Status %i from GET %s', response.status_code, _shorten(url))
|
||||||
response.raise_for_status()
|
response.raise_for_status()
|
||||||
|
|
||||||
if response.status_code == 304:
|
if response.status_code == 304:
|
||||||
@@ -457,9 +499,9 @@ async def download_to_file(url, filename, *,
|
|||||||
log.debug('Downloading was cancelled before downloading the GET response')
|
log.debug('Downloading was cancelled before downloading the GET response')
|
||||||
raise asyncio.CancelledError('Downloading was cancelled')
|
raise asyncio.CancelledError('Downloading was cancelled')
|
||||||
|
|
||||||
log.debug('Downloading response of GET %s', url)
|
log.debug('Downloading response of GET %s', _shorten(url))
|
||||||
await loop.run_in_executor(None, download_loop)
|
await loop.run_in_executor(None, download_loop)
|
||||||
log.debug('Done downloading response of GET %s', url)
|
log.debug('Done downloading response of GET %s', _shorten(url))
|
||||||
|
|
||||||
# We're done downloading, now we have something cached we can use.
|
# We're done downloading, now we have something cached we can use.
|
||||||
log.debug('Saving header cache to %s', header_store)
|
log.debug('Saving header cache to %s', header_store)
|
||||||
@@ -534,7 +576,8 @@ async def fetch_texture_thumbs(parent_node_uuid: str, desired_size: str,
|
|||||||
for texture_node in texture_nodes)
|
for texture_node in texture_nodes)
|
||||||
|
|
||||||
# raises any exception from failed handle_texture_node() calls.
|
# raises any exception from failed handle_texture_node() calls.
|
||||||
await asyncio.gather(*coros)
|
loop = asyncio.get_event_loop()
|
||||||
|
await asyncio.gather(*coros, loop=loop)
|
||||||
|
|
||||||
log.info('fetch_texture_thumbs: Done downloading texture thumbnails')
|
log.info('fetch_texture_thumbs: Done downloading texture thumbnails')
|
||||||
|
|
||||||
@@ -599,7 +642,11 @@ async def download_texture_thumbnail(texture_node, desired_size: str,
|
|||||||
# Cached headers are stored next to thumbnails in sidecar files.
|
# Cached headers are stored next to thumbnails in sidecar files.
|
||||||
header_store = '%s.headers' % thumb_path
|
header_store = '%s.headers' % thumb_path
|
||||||
|
|
||||||
|
try:
|
||||||
await download_to_file(thumb_url, thumb_path, header_store=header_store, future=future)
|
await download_to_file(thumb_url, thumb_path, header_store=header_store, future=future)
|
||||||
|
except requests.exceptions.HTTPError as ex:
|
||||||
|
log.error('Unable to download %s: %s', thumb_url, ex)
|
||||||
|
thumb_path = 'ERROR'
|
||||||
|
|
||||||
loop.call_soon_threadsafe(thumbnail_loaded, texture_node, file_desc, thumb_path)
|
loop.call_soon_threadsafe(thumbnail_loaded, texture_node, file_desc, thumb_path)
|
||||||
|
|
||||||
@@ -747,7 +794,8 @@ async def download_texture(texture_node,
|
|||||||
future=future)
|
future=future)
|
||||||
downloaders.append(dlr)
|
downloaders.append(dlr)
|
||||||
|
|
||||||
return await asyncio.gather(*downloaders, return_exceptions=True)
|
loop = asyncio.get_event_loop()
|
||||||
|
return await asyncio.gather(*downloaders, return_exceptions=True, loop=loop)
|
||||||
|
|
||||||
|
|
||||||
async def upload_file(project_id: str, file_path: pathlib.Path, *,
|
async def upload_file(project_id: str, file_path: pathlib.Path, *,
|
||||||
@@ -773,9 +821,9 @@ async def upload_file(project_id: str, file_path: pathlib.Path, *,
|
|||||||
log.debug('Uploading was cancelled before doing the POST')
|
log.debug('Uploading was cancelled before doing the POST')
|
||||||
raise asyncio.CancelledError('Uploading was cancelled')
|
raise asyncio.CancelledError('Uploading was cancelled')
|
||||||
|
|
||||||
log.debug('Performing POST %s', url)
|
log.debug('Performing POST %s', _shorten(url))
|
||||||
response = await loop.run_in_executor(None, upload)
|
response = await loop.run_in_executor(None, upload)
|
||||||
log.debug('Status %i from POST %s', response.status_code, url)
|
log.debug('Status %i from POST %s', response.status_code, _shorten(url))
|
||||||
response.raise_for_status()
|
response.raise_for_status()
|
||||||
|
|
||||||
resp = response.json()
|
resp = response.json()
|
||||||
@@ -810,7 +858,6 @@ class PillarOperatorMixin:
|
|||||||
try:
|
try:
|
||||||
db_user = await check_pillar_credentials(required_roles)
|
db_user = await check_pillar_credentials(required_roles)
|
||||||
except NotSubscribedToCloudError:
|
except NotSubscribedToCloudError:
|
||||||
self._log_subscription_needed()
|
|
||||||
raise
|
raise
|
||||||
except CredentialsNotSyncedError:
|
except CredentialsNotSyncedError:
|
||||||
self.log.info('Credentials not synced, re-syncing automatically.')
|
self.log.info('Credentials not synced, re-syncing automatically.')
|
||||||
@@ -821,7 +868,6 @@ class PillarOperatorMixin:
|
|||||||
try:
|
try:
|
||||||
db_user = await refresh_pillar_credentials(required_roles)
|
db_user = await refresh_pillar_credentials(required_roles)
|
||||||
except NotSubscribedToCloudError:
|
except NotSubscribedToCloudError:
|
||||||
self._log_subscription_needed()
|
|
||||||
raise
|
raise
|
||||||
except CredentialsNotSyncedError:
|
except CredentialsNotSyncedError:
|
||||||
self.log.info('Credentials not synced after refreshing, handling as not logged in.')
|
self.log.info('Credentials not synced after refreshing, handling as not logged in.')
|
||||||
@@ -833,11 +879,13 @@ class PillarOperatorMixin:
|
|||||||
self.log.info('Credentials refreshed and ok.')
|
self.log.info('Credentials refreshed and ok.')
|
||||||
return db_user
|
return db_user
|
||||||
|
|
||||||
def _log_subscription_needed(self):
|
def _log_subscription_needed(self, *, can_renew: bool, level='ERROR'):
|
||||||
self.log.warning(
|
if can_renew:
|
||||||
'Please subscribe to the blender cloud at https://cloud.blender.org/join')
|
msg = 'Please renew your Blender Cloud subscription at https://cloud.blender.org/renew'
|
||||||
self.report({'INFO'},
|
else:
|
||||||
'Please subscribe to the blender cloud at https://cloud.blender.org/join')
|
msg = 'Please subscribe to the blender cloud at https://cloud.blender.org/join'
|
||||||
|
self.log.warning(msg)
|
||||||
|
self.report({level}, msg)
|
||||||
|
|
||||||
|
|
||||||
class AuthenticatedPillarOperatorMixin(PillarOperatorMixin):
|
class AuthenticatedPillarOperatorMixin(PillarOperatorMixin):
|
||||||
|
163
blender_cloud/project_specific.py
Normal file
163
blender_cloud/project_specific.py
Normal file
@@ -0,0 +1,163 @@
|
|||||||
|
"""Handle saving and loading project-specific settings."""
|
||||||
|
|
||||||
|
import contextlib
|
||||||
|
import logging
|
||||||
|
import typing
|
||||||
|
|
||||||
|
# Names of BlenderCloudPreferences properties that are both project-specific
|
||||||
|
# and simple enough to store directly in a dict.
|
||||||
|
PROJECT_SPECIFIC_SIMPLE_PROPS = (
|
||||||
|
'cloud_project_local_path',
|
||||||
|
)
|
||||||
|
|
||||||
|
# Names of BlenderCloudPreferences properties that are project-specific and
|
||||||
|
# Flamenco Manager-specific, and simple enough to store in a dict.
|
||||||
|
FLAMENCO_PER_PROJECT_PER_MANAGER = (
|
||||||
|
'flamenco_exclude_filter',
|
||||||
|
'flamenco_job_file_path',
|
||||||
|
'flamenco_job_output_path',
|
||||||
|
'flamenco_job_output_strip_components',
|
||||||
|
'flamenco_relative_only',
|
||||||
|
)
|
||||||
|
|
||||||
|
log = logging.getLogger(__name__)
|
||||||
|
project_settings_loading = 0 # counter, if > 0 then we're loading stuff.
|
||||||
|
|
||||||
|
|
||||||
|
@contextlib.contextmanager
|
||||||
|
def mark_as_loading():
|
||||||
|
"""Sets project_settings_loading > 0 while the context is active.
|
||||||
|
|
||||||
|
A counter is used to allow for nested mark_as_loading() contexts.
|
||||||
|
"""
|
||||||
|
global project_settings_loading
|
||||||
|
project_settings_loading += 1
|
||||||
|
try:
|
||||||
|
yield
|
||||||
|
finally:
|
||||||
|
project_settings_loading -= 1
|
||||||
|
|
||||||
|
|
||||||
|
def update_preferences(prefs, names_to_update: typing.Iterable[str],
|
||||||
|
new_values: typing.Mapping[str, typing.Any]):
|
||||||
|
for name in names_to_update:
|
||||||
|
if not hasattr(prefs, name):
|
||||||
|
log.debug('not setting %r, property cannot be found', name)
|
||||||
|
continue
|
||||||
|
|
||||||
|
if name in new_values:
|
||||||
|
log.debug('setting %r = %r', name, new_values[name])
|
||||||
|
setattr(prefs, name, new_values[name])
|
||||||
|
else:
|
||||||
|
# The property wasn't stored, so set the default value instead.
|
||||||
|
bl_type, args = getattr(prefs.bl_rna, name)
|
||||||
|
log.debug('finding default value for %r', name)
|
||||||
|
if 'default' not in args:
|
||||||
|
log.debug('no default value for %r, not touching', name)
|
||||||
|
continue
|
||||||
|
log.debug('found default value for %r = %r', name, args['default'])
|
||||||
|
setattr(prefs, name, args['default'])
|
||||||
|
|
||||||
|
|
||||||
|
def handle_project_update(_=None, _2=None):
|
||||||
|
"""Handles changing projects, which may cause extensions to be disabled/enabled.
|
||||||
|
|
||||||
|
Ignores arguments so that it can be used as property update callback.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from .blender import preferences, project_extensions
|
||||||
|
|
||||||
|
with mark_as_loading():
|
||||||
|
prefs = preferences()
|
||||||
|
project_id = prefs.project.project
|
||||||
|
log.info('Updating internal state to reflect extensions enabled on current project %s.',
|
||||||
|
project_id)
|
||||||
|
|
||||||
|
project_extensions.cache_clear()
|
||||||
|
|
||||||
|
from blender_cloud import attract, flamenco
|
||||||
|
attract.deactivate()
|
||||||
|
flamenco.deactivate()
|
||||||
|
|
||||||
|
enabled_for = project_extensions(project_id)
|
||||||
|
log.info('Project extensions: %s', enabled_for)
|
||||||
|
if 'attract' in enabled_for:
|
||||||
|
attract.activate()
|
||||||
|
if 'flamenco' in enabled_for:
|
||||||
|
flamenco.activate()
|
||||||
|
|
||||||
|
# Load project-specific settings from the last time we visited this project.
|
||||||
|
ps = prefs.get('project_settings', {}).get(project_id, {})
|
||||||
|
if not ps:
|
||||||
|
log.debug('no project-specific settings are available, '
|
||||||
|
'only resetting available Flamenco Managers')
|
||||||
|
# The Flamenco Manager should really be chosen explicitly out of the available
|
||||||
|
# Managers.
|
||||||
|
prefs.flamenco_manager.available_managers = []
|
||||||
|
return
|
||||||
|
|
||||||
|
if log.isEnabledFor(logging.DEBUG):
|
||||||
|
from pprint import pformat
|
||||||
|
log.debug('loading project-specific settings:\n%s', pformat(ps.to_dict()))
|
||||||
|
|
||||||
|
# Restore simple properties.
|
||||||
|
update_preferences(prefs, PROJECT_SPECIFIC_SIMPLE_PROPS, ps)
|
||||||
|
|
||||||
|
# Restore Flamenco settings.
|
||||||
|
prefs.flamenco_manager.available_managers = ps.get('flamenco_available_managers', [])
|
||||||
|
flamenco_manager_id = ps.get('flamenco_manager_id')
|
||||||
|
if flamenco_manager_id:
|
||||||
|
log.debug('setting flamenco manager to %s', flamenco_manager_id)
|
||||||
|
try:
|
||||||
|
# This will trigger a load of Project+Manager-specfic settings.
|
||||||
|
prefs.flamenco_manager.manager = flamenco_manager_id
|
||||||
|
except TypeError:
|
||||||
|
log.warning('manager %s for this project could not be found', flamenco_manager_id)
|
||||||
|
elif prefs.flamenco_manager.available_managers:
|
||||||
|
prefs.flamenco_manager.manager = prefs.flamenco_manager.available_managers[0]['_id']
|
||||||
|
|
||||||
|
|
||||||
|
def store(_=None, _2=None):
|
||||||
|
"""Remember project-specific settings as soon as one of them changes.
|
||||||
|
|
||||||
|
Ignores arguments so that it can be used as property update callback.
|
||||||
|
|
||||||
|
No-op when project_settings_loading=True, to prevent saving project-
|
||||||
|
specific settings while they are actually being loaded.
|
||||||
|
"""
|
||||||
|
from .blender import preferences
|
||||||
|
|
||||||
|
global project_settings_loading
|
||||||
|
if project_settings_loading:
|
||||||
|
return
|
||||||
|
|
||||||
|
prefs = preferences()
|
||||||
|
project_id = prefs.project.project
|
||||||
|
all_settings = prefs.get('project_settings', {})
|
||||||
|
ps = all_settings.get(project_id, {}) # either a dict or bpy.types.IDPropertyGroup
|
||||||
|
|
||||||
|
for name in PROJECT_SPECIFIC_SIMPLE_PROPS:
|
||||||
|
ps[name] = getattr(prefs, name)
|
||||||
|
|
||||||
|
# Store project-specific Flamenco settings
|
||||||
|
ps['flamenco_manager_id'] = prefs.flamenco_manager.manager
|
||||||
|
ps['flamenco_available_managers'] = prefs.flamenco_manager.available_managers
|
||||||
|
|
||||||
|
# Store per-project, per-manager settings for the current Manager.
|
||||||
|
pppm = ps.get('flamenco_managers_settings', {})
|
||||||
|
pppm[prefs.flamenco_manager.manager] = {
|
||||||
|
name: getattr(prefs, name) for name in FLAMENCO_PER_PROJECT_PER_MANAGER
|
||||||
|
}
|
||||||
|
ps['flamenco_managers_settings'] = pppm # IDPropertyGroup has no setdefault() method.
|
||||||
|
|
||||||
|
# Store this project's settings in the preferences.
|
||||||
|
all_settings[project_id] = ps
|
||||||
|
prefs['project_settings'] = all_settings
|
||||||
|
|
||||||
|
if log.isEnabledFor(logging.DEBUG):
|
||||||
|
from pprint import pformat
|
||||||
|
if hasattr(all_settings, 'to_dict'):
|
||||||
|
to_log = all_settings.to_dict()
|
||||||
|
else:
|
||||||
|
to_log = all_settings
|
||||||
|
log.debug('Saving project-specific settings:\n%s', pformat(to_log))
|
@@ -25,6 +25,7 @@ import functools
|
|||||||
import logging
|
import logging
|
||||||
import pathlib
|
import pathlib
|
||||||
import tempfile
|
import tempfile
|
||||||
|
import typing
|
||||||
import shutil
|
import shutil
|
||||||
|
|
||||||
import bpy
|
import bpy
|
||||||
@@ -34,7 +35,7 @@ import asyncio
|
|||||||
import pillarsdk
|
import pillarsdk
|
||||||
from pillarsdk import exceptions as sdk_exceptions
|
from pillarsdk import exceptions as sdk_exceptions
|
||||||
from .pillar import pillar_call
|
from .pillar import pillar_call
|
||||||
from . import async_loop, pillar, cache, blendfile, home_project
|
from . import async_loop, blender, compatibility, pillar, cache, blendfile, home_project
|
||||||
|
|
||||||
SETTINGS_FILES_TO_UPLOAD = ['userpref.blend', 'startup.blend']
|
SETTINGS_FILES_TO_UPLOAD = ['userpref.blend', 'startup.blend']
|
||||||
|
|
||||||
@@ -100,7 +101,7 @@ async def find_sync_group_id(home_project_id: str,
|
|||||||
user_id: str,
|
user_id: str,
|
||||||
blender_version: str,
|
blender_version: str,
|
||||||
*,
|
*,
|
||||||
may_create=True) -> str:
|
may_create=True) -> typing.Tuple[str, str]:
|
||||||
"""Finds the group node in which to store sync assets.
|
"""Finds the group node in which to store sync assets.
|
||||||
|
|
||||||
If the group node doesn't exist and may_create=True, it creates it.
|
If the group node doesn't exist and may_create=True, it creates it.
|
||||||
@@ -122,7 +123,7 @@ async def find_sync_group_id(home_project_id: str,
|
|||||||
|
|
||||||
if not may_create and sync_group is None:
|
if not may_create and sync_group is None:
|
||||||
log.info("Sync folder doesn't exist, and not creating it either.")
|
log.info("Sync folder doesn't exist, and not creating it either.")
|
||||||
return None, None
|
return '', ''
|
||||||
|
|
||||||
# Find/create the sub-group for the requested Blender version
|
# Find/create the sub-group for the requested Blender version
|
||||||
try:
|
try:
|
||||||
@@ -144,7 +145,7 @@ async def find_sync_group_id(home_project_id: str,
|
|||||||
if not may_create and sub_sync_group is None:
|
if not may_create and sub_sync_group is None:
|
||||||
log.info("Sync folder for Blender version %s doesn't exist, "
|
log.info("Sync folder for Blender version %s doesn't exist, "
|
||||||
"and not creating it either.", blender_version)
|
"and not creating it either.", blender_version)
|
||||||
return sync_group['_id'], None
|
return sync_group['_id'], ''
|
||||||
|
|
||||||
return sync_group['_id'], sub_sync_group['_id']
|
return sync_group['_id'], sub_sync_group['_id']
|
||||||
|
|
||||||
@@ -195,6 +196,7 @@ async def available_blender_versions(home_project_id: str, user_id: str) -> list
|
|||||||
|
|
||||||
|
|
||||||
# noinspection PyAttributeOutsideInit
|
# noinspection PyAttributeOutsideInit
|
||||||
|
@compatibility.convert_properties
|
||||||
class PILLAR_OT_sync(pillar.PillarOperatorMixin,
|
class PILLAR_OT_sync(pillar.PillarOperatorMixin,
|
||||||
async_loop.AsyncModalOperatorMixin,
|
async_loop.AsyncModalOperatorMixin,
|
||||||
bpy.types.Operator):
|
bpy.types.Operator):
|
||||||
@@ -203,9 +205,9 @@ class PILLAR_OT_sync(pillar.PillarOperatorMixin,
|
|||||||
bl_description = 'Synchronises Blender settings with Blender Cloud'
|
bl_description = 'Synchronises Blender settings with Blender Cloud'
|
||||||
|
|
||||||
log = logging.getLogger('bpy.ops.%s' % bl_idname)
|
log = logging.getLogger('bpy.ops.%s' % bl_idname)
|
||||||
home_project_id = None
|
home_project_id = ''
|
||||||
sync_group_id = None # top-level sync group node ID
|
sync_group_id = '' # top-level sync group node ID
|
||||||
sync_group_versioned_id = None # sync group node ID for the given Blender version.
|
sync_group_versioned_id = '' # sync group node ID for the given Blender version.
|
||||||
|
|
||||||
action = bpy.props.EnumProperty(
|
action = bpy.props.EnumProperty(
|
||||||
items=[
|
items=[
|
||||||
@@ -284,9 +286,8 @@ class PILLAR_OT_sync(pillar.PillarOperatorMixin,
|
|||||||
db_user = await self.check_credentials(context, REQUIRES_ROLES_FOR_SYNC)
|
db_user = await self.check_credentials(context, REQUIRES_ROLES_FOR_SYNC)
|
||||||
self.user_id = db_user['_id']
|
self.user_id = db_user['_id']
|
||||||
log.debug('Found user ID: %s', self.user_id)
|
log.debug('Found user ID: %s', self.user_id)
|
||||||
except pillar.NotSubscribedToCloudError:
|
except pillar.NotSubscribedToCloudError as ex:
|
||||||
self.log.exception('User not subscribed to cloud.')
|
self._log_subscription_needed(can_renew=ex.can_renew)
|
||||||
self.bss_report({'SUBSCRIBE'}, 'Please subscribe to the Blender Cloud.')
|
|
||||||
self._state = 'QUIT'
|
self._state = 'QUIT'
|
||||||
return
|
return
|
||||||
except pillar.UserNotLoggedInError:
|
except pillar.UserNotLoggedInError:
|
||||||
@@ -388,12 +389,12 @@ class PILLAR_OT_sync(pillar.PillarOperatorMixin,
|
|||||||
"""Loads files from the Pillar server."""
|
"""Loads files from the Pillar server."""
|
||||||
|
|
||||||
# If the sync group node doesn't exist, offer a list of groups that do.
|
# If the sync group node doesn't exist, offer a list of groups that do.
|
||||||
if self.sync_group_id is None:
|
if not self.sync_group_id:
|
||||||
self.bss_report({'ERROR'},
|
self.bss_report({'ERROR'},
|
||||||
'There are no synced Blender settings in your Blender Cloud.')
|
'There are no synced Blender settings in your Blender Cloud.')
|
||||||
return
|
return
|
||||||
|
|
||||||
if self.sync_group_versioned_id is None:
|
if not self.sync_group_versioned_id:
|
||||||
self.bss_report({'ERROR'}, 'Therre are no synced Blender settings for version %s' %
|
self.bss_report({'ERROR'}, 'Therre are no synced Blender settings for version %s' %
|
||||||
self.blender_version)
|
self.blender_version)
|
||||||
return
|
return
|
||||||
@@ -416,10 +417,19 @@ class PILLAR_OT_sync(pillar.PillarOperatorMixin,
|
|||||||
bss = bpy.context.window_manager.blender_sync_status
|
bss = bpy.context.window_manager.blender_sync_status
|
||||||
bss.available_blender_versions = versions
|
bss.available_blender_versions = versions
|
||||||
|
|
||||||
if versions:
|
if not versions:
|
||||||
# There are versions to sync, so we can remove the status message.
|
# There are versions to sync, so we can remove the status message.
|
||||||
# However, if there aren't any, the status message shows why, and
|
# However, if there aren't any, the status message shows why, and
|
||||||
# shouldn't be erased.
|
# shouldn't be erased.
|
||||||
|
return
|
||||||
|
|
||||||
|
# Prevent warnings that the current value of the EnumProperty isn't valid.
|
||||||
|
current_version = '%d.%d' % bpy.app.version[:2]
|
||||||
|
if current_version in versions:
|
||||||
|
bss.version = current_version
|
||||||
|
else:
|
||||||
|
bss.version = versions[0]
|
||||||
|
|
||||||
self.bss_report({'INFO'}, '')
|
self.bss_report({'INFO'}, '')
|
||||||
|
|
||||||
async def download_settings_file(self, fname: str, temp_dir: str):
|
async def download_settings_file(self, fname: str, temp_dir: str):
|
||||||
@@ -477,13 +487,13 @@ class PILLAR_OT_sync(pillar.PillarOperatorMixin,
|
|||||||
self.log.info('Overriding machine-local settings in %s', file_path)
|
self.log.info('Overriding machine-local settings in %s', file_path)
|
||||||
|
|
||||||
# Remember some settings that should not be overwritten from the Cloud.
|
# Remember some settings that should not be overwritten from the Cloud.
|
||||||
up = bpy.context.user_preferences
|
prefs = blender.ctx_preferences()
|
||||||
remembered = {}
|
remembered = {}
|
||||||
for rna_key, python_key in LOCAL_SETTINGS_RNA:
|
for rna_key, python_key in LOCAL_SETTINGS_RNA:
|
||||||
assert '.' in python_key, 'Sorry, this code assumes there is a dot in the Python key'
|
assert '.' in python_key, 'Sorry, this code assumes there is a dot in the Python key'
|
||||||
|
|
||||||
try:
|
try:
|
||||||
value = up.path_resolve(python_key)
|
value = prefs.path_resolve(python_key)
|
||||||
except ValueError:
|
except ValueError:
|
||||||
# Setting doesn't exist. This can happen, for example Cycles
|
# Setting doesn't exist. This can happen, for example Cycles
|
||||||
# settings on a build that doesn't have Cycles enabled.
|
# settings on a build that doesn't have Cycles enabled.
|
||||||
@@ -492,7 +502,7 @@ class PILLAR_OT_sync(pillar.PillarOperatorMixin,
|
|||||||
# Map enums from strings (in Python) to ints (in DNA).
|
# Map enums from strings (in Python) to ints (in DNA).
|
||||||
dot_index = python_key.rindex('.')
|
dot_index = python_key.rindex('.')
|
||||||
parent_key, prop_key = python_key[:dot_index], python_key[dot_index + 1:]
|
parent_key, prop_key = python_key[:dot_index], python_key[dot_index + 1:]
|
||||||
parent = up.path_resolve(parent_key)
|
parent = prefs.path_resolve(parent_key)
|
||||||
prop = parent.bl_rna.properties[prop_key]
|
prop = parent.bl_rna.properties[prop_key]
|
||||||
if prop.type == 'ENUM':
|
if prop.type == 'ENUM':
|
||||||
log.debug('Rewriting %s from %r to %r',
|
log.debug('Rewriting %s from %r to %r',
|
||||||
|
@@ -18,212 +18,35 @@
|
|||||||
|
|
||||||
import asyncio
|
import asyncio
|
||||||
import logging
|
import logging
|
||||||
import threading
|
|
||||||
import os
|
import os
|
||||||
|
import threading
|
||||||
|
import typing
|
||||||
|
|
||||||
import bpy
|
import bpy
|
||||||
import bgl
|
import bgl
|
||||||
import blf
|
|
||||||
|
|
||||||
import pillarsdk
|
import pillarsdk
|
||||||
from . import async_loop, pillar, cache, blender, utils
|
from .. import async_loop, compatibility, pillar, cache, blender, utils
|
||||||
|
from . import menu_item as menu_item_mod # so that we can have menu items called 'menu_item'
|
||||||
|
from . import nodes
|
||||||
|
|
||||||
|
if bpy.app.version < (2, 80):
|
||||||
|
from . import draw_27 as draw
|
||||||
|
else:
|
||||||
|
from . import draw
|
||||||
|
|
||||||
REQUIRED_ROLES_FOR_TEXTURE_BROWSER = {'subscriber', 'demo'}
|
REQUIRED_ROLES_FOR_TEXTURE_BROWSER = {'subscriber', 'demo'}
|
||||||
MOUSE_SCROLL_PIXELS_PER_TICK = 50
|
MOUSE_SCROLL_PIXELS_PER_TICK = 50
|
||||||
|
|
||||||
ICON_WIDTH = 128
|
|
||||||
ICON_HEIGHT = 128
|
|
||||||
TARGET_ITEM_WIDTH = 400
|
TARGET_ITEM_WIDTH = 400
|
||||||
TARGET_ITEM_HEIGHT = 128
|
TARGET_ITEM_HEIGHT = 128
|
||||||
ITEM_MARGIN_X = 5
|
ITEM_MARGIN_X = 5
|
||||||
ITEM_MARGIN_Y = 5
|
ITEM_MARGIN_Y = 5
|
||||||
ITEM_PADDING_X = 5
|
ITEM_PADDING_X = 5
|
||||||
|
|
||||||
library_path = '/tmp'
|
|
||||||
library_icons_path = os.path.join(os.path.dirname(__file__), "icons")
|
|
||||||
log = logging.getLogger(__name__)
|
log = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class SpecialFolderNode(pillarsdk.Node):
|
|
||||||
NODE_TYPE = 'SPECIAL'
|
|
||||||
|
|
||||||
|
|
||||||
class UpNode(SpecialFolderNode):
|
|
||||||
NODE_TYPE = 'UP'
|
|
||||||
|
|
||||||
def __init__(self):
|
|
||||||
super().__init__()
|
|
||||||
self['_id'] = 'UP'
|
|
||||||
self['node_type'] = self.NODE_TYPE
|
|
||||||
|
|
||||||
|
|
||||||
class ProjectNode(SpecialFolderNode):
|
|
||||||
NODE_TYPE = 'PROJECT'
|
|
||||||
|
|
||||||
def __init__(self, project):
|
|
||||||
super().__init__()
|
|
||||||
|
|
||||||
assert isinstance(project, pillarsdk.Project), 'wrong type for project: %r' % type(project)
|
|
||||||
|
|
||||||
self.merge(project.to_dict())
|
|
||||||
self['node_type'] = self.NODE_TYPE
|
|
||||||
|
|
||||||
|
|
||||||
class MenuItem:
|
|
||||||
"""GUI menu item for the 3D View GUI."""
|
|
||||||
|
|
||||||
icon_margin_x = 4
|
|
||||||
icon_margin_y = 4
|
|
||||||
text_margin_x = 6
|
|
||||||
|
|
||||||
text_height = 16
|
|
||||||
text_width = 72
|
|
||||||
|
|
||||||
DEFAULT_ICONS = {
|
|
||||||
'FOLDER': os.path.join(library_icons_path, 'folder.png'),
|
|
||||||
'SPINNER': os.path.join(library_icons_path, 'spinner.png'),
|
|
||||||
}
|
|
||||||
|
|
||||||
FOLDER_NODE_TYPES = {'group_texture', 'group_hdri', UpNode.NODE_TYPE, ProjectNode.NODE_TYPE}
|
|
||||||
SUPPORTED_NODE_TYPES = {'texture', 'hdri'}.union(FOLDER_NODE_TYPES)
|
|
||||||
|
|
||||||
def __init__(self, node, file_desc, thumb_path: str, label_text):
|
|
||||||
self.log = logging.getLogger('%s.MenuItem' % __name__)
|
|
||||||
if node['node_type'] not in self.SUPPORTED_NODE_TYPES:
|
|
||||||
self.log.info('Invalid node type in node: %s', node)
|
|
||||||
raise TypeError('Node of type %r not supported; supported are %r.' % (
|
|
||||||
node['node_type'], self.SUPPORTED_NODE_TYPES))
|
|
||||||
|
|
||||||
assert isinstance(node, pillarsdk.Node), 'wrong type for node: %r' % type(node)
|
|
||||||
assert isinstance(node['_id'], str), 'wrong type for node["_id"]: %r' % type(node['_id'])
|
|
||||||
self.node = node # pillarsdk.Node, contains 'node_type' key to indicate type
|
|
||||||
self.file_desc = file_desc # pillarsdk.File object, or None if a 'folder' node.
|
|
||||||
self.label_text = label_text
|
|
||||||
self._thumb_path = ''
|
|
||||||
self.icon = None
|
|
||||||
self._is_folder = node['node_type'] in self.FOLDER_NODE_TYPES
|
|
||||||
self._is_spinning = False
|
|
||||||
|
|
||||||
# Determine sorting order.
|
|
||||||
# by default, sort all the way at the end and folders first.
|
|
||||||
self._order = 0 if self._is_folder else 10000
|
|
||||||
if node and node.properties and node.properties.order is not None:
|
|
||||||
self._order = node.properties.order
|
|
||||||
|
|
||||||
self.thumb_path = thumb_path
|
|
||||||
|
|
||||||
# Updated when drawing the image
|
|
||||||
self.x = 0
|
|
||||||
self.y = 0
|
|
||||||
self.width = 0
|
|
||||||
self.height = 0
|
|
||||||
|
|
||||||
def sort_key(self):
|
|
||||||
"""Key for sorting lists of MenuItems."""
|
|
||||||
return self._order, self.label_text
|
|
||||||
|
|
||||||
@property
|
|
||||||
def thumb_path(self) -> str:
|
|
||||||
return self._thumb_path
|
|
||||||
|
|
||||||
@thumb_path.setter
|
|
||||||
def thumb_path(self, new_thumb_path: str):
|
|
||||||
self._is_spinning = new_thumb_path == 'SPINNER'
|
|
||||||
|
|
||||||
self._thumb_path = self.DEFAULT_ICONS.get(new_thumb_path, new_thumb_path)
|
|
||||||
if self._thumb_path:
|
|
||||||
self.icon = bpy.data.images.load(filepath=self._thumb_path)
|
|
||||||
else:
|
|
||||||
self.icon = None
|
|
||||||
|
|
||||||
@property
|
|
||||||
def node_uuid(self) -> str:
|
|
||||||
return self.node['_id']
|
|
||||||
|
|
||||||
def represents(self, node) -> bool:
|
|
||||||
"""Returns True iff this MenuItem represents the given node."""
|
|
||||||
|
|
||||||
node_uuid = node['_id']
|
|
||||||
return self.node_uuid == node_uuid
|
|
||||||
|
|
||||||
def update(self, node, file_desc, thumb_path: str, label_text=None):
|
|
||||||
# We can get updated information about our Node, but a MenuItem should
|
|
||||||
# always represent one node, and it shouldn't be shared between nodes.
|
|
||||||
if self.node_uuid != node['_id']:
|
|
||||||
raise ValueError("Don't change the node ID this MenuItem reflects, "
|
|
||||||
"just create a new one.")
|
|
||||||
self.node = node
|
|
||||||
self.file_desc = file_desc # pillarsdk.File object, or None if a 'folder' node.
|
|
||||||
self.thumb_path = thumb_path
|
|
||||||
|
|
||||||
if label_text is not None:
|
|
||||||
self.label_text = label_text
|
|
||||||
|
|
||||||
@property
|
|
||||||
def is_folder(self) -> bool:
|
|
||||||
return self._is_folder
|
|
||||||
|
|
||||||
@property
|
|
||||||
def is_spinning(self) -> bool:
|
|
||||||
return self._is_spinning
|
|
||||||
|
|
||||||
def update_placement(self, x, y, width, height):
|
|
||||||
"""Use OpenGL to draw this one menu item."""
|
|
||||||
|
|
||||||
self.x = x
|
|
||||||
self.y = y
|
|
||||||
self.width = width
|
|
||||||
self.height = height
|
|
||||||
|
|
||||||
def draw(self, highlighted: bool):
|
|
||||||
bgl.glEnable(bgl.GL_BLEND)
|
|
||||||
if highlighted:
|
|
||||||
bgl.glColor4f(0.555, 0.555, 0.555, 0.8)
|
|
||||||
else:
|
|
||||||
bgl.glColor4f(0.447, 0.447, 0.447, 0.8)
|
|
||||||
|
|
||||||
bgl.glRectf(self.x, self.y, self.x + self.width, self.y + self.height)
|
|
||||||
|
|
||||||
texture = self.icon
|
|
||||||
err = texture.gl_load(filter=bgl.GL_NEAREST, mag=bgl.GL_NEAREST)
|
|
||||||
assert not err, 'OpenGL error: %i' % err
|
|
||||||
|
|
||||||
bgl.glColor4f(0.0, 0.0, 1.0, 0.5)
|
|
||||||
# bgl.glLineWidth(1.5)
|
|
||||||
|
|
||||||
# ------ TEXTURE ---------#
|
|
||||||
bgl.glBindTexture(bgl.GL_TEXTURE_2D, texture.bindcode[0])
|
|
||||||
bgl.glEnable(bgl.GL_TEXTURE_2D)
|
|
||||||
bgl.glBlendFunc(bgl.GL_SRC_ALPHA, bgl.GL_ONE_MINUS_SRC_ALPHA)
|
|
||||||
|
|
||||||
bgl.glColor4f(1, 1, 1, 1)
|
|
||||||
bgl.glBegin(bgl.GL_QUADS)
|
|
||||||
bgl.glTexCoord2d(0, 0)
|
|
||||||
bgl.glVertex2d(self.x + self.icon_margin_x, self.y)
|
|
||||||
bgl.glTexCoord2d(0, 1)
|
|
||||||
bgl.glVertex2d(self.x + self.icon_margin_x, self.y + ICON_HEIGHT)
|
|
||||||
bgl.glTexCoord2d(1, 1)
|
|
||||||
bgl.glVertex2d(self.x + self.icon_margin_x + ICON_WIDTH, self.y + ICON_HEIGHT)
|
|
||||||
bgl.glTexCoord2d(1, 0)
|
|
||||||
bgl.glVertex2d(self.x + self.icon_margin_x + ICON_WIDTH, self.y)
|
|
||||||
bgl.glEnd()
|
|
||||||
bgl.glDisable(bgl.GL_TEXTURE_2D)
|
|
||||||
bgl.glDisable(bgl.GL_BLEND)
|
|
||||||
|
|
||||||
texture.gl_free()
|
|
||||||
|
|
||||||
# draw some text
|
|
||||||
font_id = 0
|
|
||||||
blf.position(font_id,
|
|
||||||
self.x + self.icon_margin_x + ICON_WIDTH + self.text_margin_x,
|
|
||||||
self.y + ICON_HEIGHT * 0.5 - 0.25 * self.text_height, 0)
|
|
||||||
blf.size(font_id, self.text_height, self.text_width)
|
|
||||||
blf.draw(font_id, self.label_text)
|
|
||||||
|
|
||||||
def hits(self, mouse_x: int, mouse_y: int) -> bool:
|
|
||||||
return self.x < mouse_x < self.x + self.width and self.y < mouse_y < self.y + self.height
|
|
||||||
|
|
||||||
|
|
||||||
class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
||||||
async_loop.AsyncModalOperatorMixin,
|
async_loop.AsyncModalOperatorMixin,
|
||||||
bpy.types.Operator):
|
bpy.types.Operator):
|
||||||
@@ -236,17 +59,17 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
project_name = ''
|
project_name = ''
|
||||||
|
|
||||||
# This contains a stack of Node objects that lead up to the currently browsed node.
|
# This contains a stack of Node objects that lead up to the currently browsed node.
|
||||||
path_stack = []
|
path_stack = [] # type: typing.List[pillarsdk.Node]
|
||||||
|
|
||||||
# This contains a stack of MenuItem objects that lead up to the currently browsed node.
|
# This contains a stack of MenuItem objects that lead up to the currently browsed node.
|
||||||
menu_item_stack = []
|
menu_item_stack = [] # type: typing.List[menu_item_mod.MenuItem]
|
||||||
|
|
||||||
timer = None
|
timer = None
|
||||||
log = logging.getLogger('%s.BlenderCloudBrowser' % __name__)
|
log = logging.getLogger('%s.BlenderCloudBrowser' % __name__)
|
||||||
|
|
||||||
_menu_item_lock = threading.Lock()
|
_menu_item_lock = threading.Lock()
|
||||||
current_display_content = [] # list of MenuItems currently displayed
|
current_display_content = [] # type: typing.List[menu_item_mod.MenuItem]
|
||||||
loaded_images = set()
|
loaded_images = set() # type: typing.Set[str]
|
||||||
thumbnails_cache = ''
|
thumbnails_cache = ''
|
||||||
maximized_area = False
|
maximized_area = False
|
||||||
|
|
||||||
@@ -311,8 +134,8 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
self.mouse_y = event.mouse_y
|
self.mouse_y = event.mouse_y
|
||||||
|
|
||||||
left_mouse_release = event.type == 'LEFTMOUSE' and event.value == 'RELEASE'
|
left_mouse_release = event.type == 'LEFTMOUSE' and event.value == 'RELEASE'
|
||||||
if self._state == 'PLEASE_SUBSCRIBE' and left_mouse_release:
|
if left_mouse_release and self._state in {'PLEASE_SUBSCRIBE', 'PLEASE_RENEW'}:
|
||||||
self.open_browser_subscribe()
|
self.open_browser_subscribe(renew=self._state == 'PLEASE_RENEW')
|
||||||
self._finish(context)
|
self._finish(context)
|
||||||
return {'FINISHED'}
|
return {'FINISHED'}
|
||||||
|
|
||||||
@@ -365,9 +188,9 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
db_user = await self.check_credentials(context, REQUIRED_ROLES_FOR_TEXTURE_BROWSER)
|
db_user = await self.check_credentials(context, REQUIRED_ROLES_FOR_TEXTURE_BROWSER)
|
||||||
except pillar.NotSubscribedToCloudError:
|
except pillar.NotSubscribedToCloudError as ex:
|
||||||
self.log.info('User not subscribed to Blender Cloud.')
|
self._log_subscription_needed(can_renew=ex.can_renew, level='INFO')
|
||||||
self._show_subscribe_screen()
|
self._show_subscribe_screen(can_renew=ex.can_renew)
|
||||||
return None
|
return None
|
||||||
|
|
||||||
if db_user is None:
|
if db_user is None:
|
||||||
@@ -375,13 +198,17 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
|
|
||||||
await self.async_download_previews()
|
await self.async_download_previews()
|
||||||
|
|
||||||
def _show_subscribe_screen(self):
|
def _show_subscribe_screen(self, *, can_renew: bool):
|
||||||
"""Shows the "You need to subscribe" screen."""
|
"""Shows the "You need to subscribe" screen."""
|
||||||
|
|
||||||
|
if can_renew:
|
||||||
|
self._state = 'PLEASE_RENEW'
|
||||||
|
else:
|
||||||
self._state = 'PLEASE_SUBSCRIBE'
|
self._state = 'PLEASE_SUBSCRIBE'
|
||||||
|
|
||||||
bpy.context.window.cursor_set('HAND')
|
bpy.context.window.cursor_set('HAND')
|
||||||
|
|
||||||
def descend_node(self, menu_item: MenuItem):
|
def descend_node(self, menu_item: menu_item_mod.MenuItem):
|
||||||
"""Descends the node hierarchy by visiting this menu item's node.
|
"""Descends the node hierarchy by visiting this menu item's node.
|
||||||
|
|
||||||
Also keeps track of the current node, so that we know where the "up" button should go.
|
Also keeps track of the current node, so that we know where the "up" button should go.
|
||||||
@@ -390,7 +217,7 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
node = menu_item.node
|
node = menu_item.node
|
||||||
assert isinstance(node, pillarsdk.Node), 'Wrong type %s' % node
|
assert isinstance(node, pillarsdk.Node), 'Wrong type %s' % node
|
||||||
|
|
||||||
if isinstance(node, UpNode):
|
if isinstance(node, nodes.UpNode):
|
||||||
# Going up.
|
# Going up.
|
||||||
self.log.debug('Going up to %r', self.current_path)
|
self.log.debug('Going up to %r', self.current_path)
|
||||||
self.current_path = self.current_path.parent
|
self.current_path = self.current_path.parent
|
||||||
@@ -402,7 +229,7 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
self.project_name = ''
|
self.project_name = ''
|
||||||
else:
|
else:
|
||||||
# Going down, keep track of where we were
|
# Going down, keep track of where we were
|
||||||
if isinstance(node, ProjectNode):
|
if isinstance(node, nodes.ProjectNode):
|
||||||
self.project_name = node['name']
|
self.project_name = node['name']
|
||||||
|
|
||||||
self.current_path /= node['_id']
|
self.current_path /= node['_id']
|
||||||
@@ -445,19 +272,20 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
self.loaded_images.clear()
|
self.loaded_images.clear()
|
||||||
self.current_display_content.clear()
|
self.current_display_content.clear()
|
||||||
|
|
||||||
def add_menu_item(self, *args) -> MenuItem:
|
def add_menu_item(self, *args) -> menu_item_mod.MenuItem:
|
||||||
menu_item = MenuItem(*args)
|
menu_item = menu_item_mod.MenuItem(*args)
|
||||||
|
|
||||||
# Just make this thread-safe to be on the safe side.
|
# Just make this thread-safe to be on the safe side.
|
||||||
with self._menu_item_lock:
|
with self._menu_item_lock:
|
||||||
self.current_display_content.append(menu_item)
|
self.current_display_content.append(menu_item)
|
||||||
|
if menu_item.icon is not None:
|
||||||
self.loaded_images.add(menu_item.icon.filepath_raw)
|
self.loaded_images.add(menu_item.icon.filepath_raw)
|
||||||
|
|
||||||
self.sort_menu()
|
self.sort_menu()
|
||||||
|
|
||||||
return menu_item
|
return menu_item
|
||||||
|
|
||||||
def update_menu_item(self, node, *args) -> MenuItem:
|
def update_menu_item(self, node, *args):
|
||||||
node_uuid = node['_id']
|
node_uuid = node['_id']
|
||||||
|
|
||||||
# Just make this thread-safe to be on the safe side.
|
# Just make this thread-safe to be on the safe side.
|
||||||
@@ -479,7 +307,7 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
return
|
return
|
||||||
|
|
||||||
with self._menu_item_lock:
|
with self._menu_item_lock:
|
||||||
self.current_display_content.sort(key=MenuItem.sort_key)
|
self.current_display_content.sort(key=menu_item_mod.MenuItem.sort_key)
|
||||||
|
|
||||||
async def async_download_previews(self):
|
async def async_download_previews(self):
|
||||||
self._state = 'BROWSING'
|
self._state = 'BROWSING'
|
||||||
@@ -509,17 +337,17 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
self.log.debug('No node UUID and no project UUID, listing available projects')
|
self.log.debug('No node UUID and no project UUID, listing available projects')
|
||||||
children = await pillar.get_texture_projects()
|
children = await pillar.get_texture_projects()
|
||||||
for proj_dict in children:
|
for proj_dict in children:
|
||||||
self.add_menu_item(ProjectNode(proj_dict), None, 'FOLDER', proj_dict['name'])
|
self.add_menu_item(nodes.ProjectNode(proj_dict), None, 'FOLDER', proj_dict['name'])
|
||||||
return
|
return
|
||||||
|
|
||||||
# Make sure we can go up again.
|
# Make sure we can go up again.
|
||||||
self.add_menu_item(UpNode(), None, 'FOLDER', '.. up ..')
|
self.add_menu_item(nodes.UpNode(), None, 'FOLDER', '.. up ..')
|
||||||
|
|
||||||
# Download all child nodes
|
# Download all child nodes
|
||||||
self.log.debug('Iterating over child nodes of %r', self.current_path)
|
self.log.debug('Iterating over child nodes of %r', self.current_path)
|
||||||
for child in children:
|
for child in children:
|
||||||
# print(' - %(_id)s = %(name)s' % child)
|
# print(' - %(_id)s = %(name)s' % child)
|
||||||
if child['node_type'] not in MenuItem.SUPPORTED_NODE_TYPES:
|
if child['node_type'] not in menu_item_mod.MenuItem.SUPPORTED_NODE_TYPES:
|
||||||
self.log.debug('Skipping node of type %r', child['node_type'])
|
self.log.debug('Skipping node of type %r', child['node_type'])
|
||||||
continue
|
continue
|
||||||
self.add_menu_item(child, None, 'FOLDER', child['name'])
|
self.add_menu_item(child, None, 'FOLDER', child['name'])
|
||||||
@@ -538,6 +366,7 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
self.add_menu_item(node, None, 'SPINNER', texture_node['name'])
|
self.add_menu_item(node, None, 'SPINNER', texture_node['name'])
|
||||||
|
|
||||||
def thumbnail_loaded(node, file_desc, thumb_path):
|
def thumbnail_loaded(node, file_desc, thumb_path):
|
||||||
|
self.log.debug('Node %s thumbnail loaded', node['_id'])
|
||||||
self.update_menu_item(node, file_desc, thumb_path)
|
self.update_menu_item(node, file_desc, thumb_path)
|
||||||
|
|
||||||
await pillar.fetch_texture_thumbs(node_uuid, 's', directory,
|
await pillar.fetch_texture_thumbs(node_uuid, 's', directory,
|
||||||
@@ -547,6 +376,7 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
|
|
||||||
def browse_assets(self):
|
def browse_assets(self):
|
||||||
self.log.debug('Browsing assets at %r', self.current_path)
|
self.log.debug('Browsing assets at %r', self.current_path)
|
||||||
|
bpy.context.window_manager.last_blender_cloud_location = str(self.current_path)
|
||||||
self._new_async_task(self.async_download_previews())
|
self._new_async_task(self.async_download_previews())
|
||||||
|
|
||||||
def draw_menu(self, context):
|
def draw_menu(self, context):
|
||||||
@@ -559,6 +389,7 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
'DOWNLOADING_TEXTURE': self._draw_downloading,
|
'DOWNLOADING_TEXTURE': self._draw_downloading,
|
||||||
'EXCEPTION': self._draw_exception,
|
'EXCEPTION': self._draw_exception,
|
||||||
'PLEASE_SUBSCRIBE': self._draw_subscribe,
|
'PLEASE_SUBSCRIBE': self._draw_subscribe,
|
||||||
|
'PLEASE_RENEW': self._draw_renew,
|
||||||
}
|
}
|
||||||
|
|
||||||
if self._state in drawers:
|
if self._state in drawers:
|
||||||
@@ -566,12 +397,9 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
drawer(context)
|
drawer(context)
|
||||||
|
|
||||||
# For debugging: draw the state
|
# For debugging: draw the state
|
||||||
font_id = 0
|
draw.text((5, 5),
|
||||||
bgl.glColor4f(1.0, 1.0, 1.0, 1.0)
|
'%s %s' % (self._state, self.project_name),
|
||||||
blf.size(font_id, 20, 72)
|
rgba=(1.0, 1.0, 1.0, 1.0), fsize=12)
|
||||||
blf.position(font_id, 5, 5, 0)
|
|
||||||
blf.draw(font_id, '%s %s' % (self._state, self.project_name))
|
|
||||||
bgl.glDisable(bgl.GL_BLEND)
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _window_region(context):
|
def _window_region(context):
|
||||||
@@ -582,6 +410,12 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
|
|
||||||
def _draw_browser(self, context):
|
def _draw_browser(self, context):
|
||||||
"""OpenGL drawing code for the BROWSING state."""
|
"""OpenGL drawing code for the BROWSING state."""
|
||||||
|
from . import draw
|
||||||
|
|
||||||
|
if not self.current_display_content:
|
||||||
|
self._draw_text_on_colour(context, "Communicating with Blender Cloud",
|
||||||
|
(0.0, 0.0, 0.0, 0.6))
|
||||||
|
return
|
||||||
|
|
||||||
window_region = self._window_region(context)
|
window_region = self._window_region(context)
|
||||||
content_width = window_region.width - ITEM_MARGIN_X * 2
|
content_width = window_region.width - ITEM_MARGIN_X * 2
|
||||||
@@ -599,10 +433,9 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
block_height = item_height + ITEM_MARGIN_Y
|
block_height = item_height + ITEM_MARGIN_Y
|
||||||
|
|
||||||
bgl.glEnable(bgl.GL_BLEND)
|
bgl.glEnable(bgl.GL_BLEND)
|
||||||
bgl.glColor4f(0.0, 0.0, 0.0, 0.6)
|
draw.aabox((0, 0), (window_region.width, window_region.height),
|
||||||
bgl.glRectf(0, 0, window_region.width, window_region.height)
|
(0.0, 0.0, 0.0, 0.6))
|
||||||
|
|
||||||
if self.current_display_content:
|
|
||||||
bottom_y = float('inf')
|
bottom_y = float('inf')
|
||||||
|
|
||||||
# The -1 / +2 are for extra rows that are drawn only half at the top/bottom.
|
# The -1 / +2 are for extra rows that are drawn only half at the top/bottom.
|
||||||
@@ -626,19 +459,7 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
self.scroll_offset_space_left +
|
self.scroll_offset_space_left +
|
||||||
0.25 * block_height)
|
0.25 * block_height)
|
||||||
|
|
||||||
else:
|
|
||||||
font_id = 0
|
|
||||||
text = "Communicating with Blender Cloud"
|
|
||||||
bgl.glColor4f(1.0, 1.0, 1.0, 1.0)
|
|
||||||
blf.size(font_id, 20, 72)
|
|
||||||
text_width, text_height = blf.dimensions(font_id, text)
|
|
||||||
blf.position(font_id,
|
|
||||||
content_x + content_width * 0.5 - text_width * 0.5,
|
|
||||||
content_y - content_height * 0.3 + text_height * 0.5, 0)
|
|
||||||
blf.draw(font_id, text)
|
|
||||||
|
|
||||||
bgl.glDisable(bgl.GL_BLEND)
|
bgl.glDisable(bgl.GL_BLEND)
|
||||||
# bgl.glColor4f(0.0, 0.0, 0.0, 1.0)
|
|
||||||
|
|
||||||
def _draw_downloading(self, context):
|
def _draw_downloading(self, context):
|
||||||
"""OpenGL drawing code for the DOWNLOADING_TEXTURE state."""
|
"""OpenGL drawing code for the DOWNLOADING_TEXTURE state."""
|
||||||
@@ -661,21 +482,15 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
'Initializing',
|
'Initializing',
|
||||||
(0.0, 0.0, 0.2, 0.6))
|
(0.0, 0.0, 0.2, 0.6))
|
||||||
|
|
||||||
def _draw_text_on_colour(self, context, text, bgcolour):
|
def _draw_text_on_colour(self, context, text: str, bgcolour):
|
||||||
content_height, content_width = self._window_size(context)
|
content_height, content_width = self._window_size(context)
|
||||||
|
|
||||||
bgl.glEnable(bgl.GL_BLEND)
|
bgl.glEnable(bgl.GL_BLEND)
|
||||||
bgl.glColor4f(*bgcolour)
|
|
||||||
bgl.glRectf(0, 0, content_width, content_height)
|
|
||||||
|
|
||||||
font_id = 0
|
draw.aabox((0, 0), (content_width, content_height), bgcolour)
|
||||||
bgl.glColor4f(1.0, 1.0, 1.0, 1.0)
|
draw.text((content_width * 0.5, content_height * 0.7),
|
||||||
blf.size(font_id, 20, 72)
|
text, fsize=20, align='C')
|
||||||
text_width, text_height = blf.dimensions(font_id, text)
|
|
||||||
|
|
||||||
blf.position(font_id,
|
|
||||||
content_width * 0.5 - text_width * 0.5,
|
|
||||||
content_height * 0.7 + text_height * 0.5, 0)
|
|
||||||
blf.draw(font_id, text)
|
|
||||||
bgl.glDisable(bgl.GL_BLEND)
|
bgl.glDisable(bgl.GL_BLEND)
|
||||||
|
|
||||||
def _window_size(self, context):
|
def _window_size(self, context):
|
||||||
@@ -692,33 +507,21 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
content_height, content_width = self._window_size(context)
|
content_height, content_width = self._window_size(context)
|
||||||
|
|
||||||
bgl.glEnable(bgl.GL_BLEND)
|
bgl.glEnable(bgl.GL_BLEND)
|
||||||
bgl.glColor4f(0.2, 0.0, 0.0, 0.6)
|
draw.aabox((0, 0), (content_width, content_height), (0.2, 0.0, 0.0, 0.6))
|
||||||
bgl.glRectf(0, 0, content_width, content_height)
|
|
||||||
|
|
||||||
font_id = 0
|
|
||||||
ex = self.async_task.exception()
|
ex = self.async_task.exception()
|
||||||
if isinstance(ex, pillar.UserNotLoggedInError):
|
if isinstance(ex, pillar.UserNotLoggedInError):
|
||||||
ex_msg = 'You are not logged in on Blender ID. Please log in at User Preferences, ' \
|
ex_msg = 'You are not logged in on Blender ID. Please log in at User Preferences, ' \
|
||||||
'System, Blender ID.'
|
'Add-ons, Blender ID Authentication.'
|
||||||
else:
|
else:
|
||||||
ex_msg = str(ex)
|
ex_msg = str(ex)
|
||||||
if not ex_msg:
|
if not ex_msg:
|
||||||
ex_msg = str(type(ex))
|
ex_msg = str(type(ex))
|
||||||
text = "An error occurred:\n%s" % ex_msg
|
text = "An error occurred:\n%s" % ex_msg
|
||||||
lines = textwrap.wrap(text)
|
lines = textwrap.wrap(text, width=100)
|
||||||
|
|
||||||
bgl.glColor4f(1.0, 1.0, 1.0, 1.0)
|
draw.text((content_width * 0.1, content_height * 0.9), lines, fsize=16)
|
||||||
blf.size(font_id, 20, 72)
|
|
||||||
_, text_height = blf.dimensions(font_id, 'yhBp')
|
|
||||||
|
|
||||||
def position(line_nr):
|
|
||||||
blf.position(font_id,
|
|
||||||
content_width * 0.1,
|
|
||||||
content_height * 0.8 - line_nr * text_height, 0)
|
|
||||||
|
|
||||||
for line_idx, line in enumerate(lines):
|
|
||||||
position(line_idx)
|
|
||||||
blf.draw(font_id, line)
|
|
||||||
bgl.glDisable(bgl.GL_BLEND)
|
bgl.glDisable(bgl.GL_BLEND)
|
||||||
|
|
||||||
def _draw_subscribe(self, context):
|
def _draw_subscribe(self, context):
|
||||||
@@ -726,7 +529,12 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
'Click to subscribe to the Blender Cloud',
|
'Click to subscribe to the Blender Cloud',
|
||||||
(0.0, 0.0, 0.2, 0.6))
|
(0.0, 0.0, 0.2, 0.6))
|
||||||
|
|
||||||
def get_clicked(self) -> MenuItem:
|
def _draw_renew(self, context):
|
||||||
|
self._draw_text_on_colour(context,
|
||||||
|
'Click to renew your Blender Cloud subscription',
|
||||||
|
(0.0, 0.0, 0.2, 0.6))
|
||||||
|
|
||||||
|
def get_clicked(self) -> typing.Optional[menu_item_mod.MenuItem]:
|
||||||
|
|
||||||
for item in self.current_display_content:
|
for item in self.current_display_content:
|
||||||
if item.hits(self.mouse_x, self.mouse_y):
|
if item.hits(self.mouse_x, self.mouse_y):
|
||||||
@@ -734,7 +542,7 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
|
|
||||||
return None
|
return None
|
||||||
|
|
||||||
def handle_item_selection(self, context, item: MenuItem):
|
def handle_item_selection(self, context, item: menu_item_mod.MenuItem):
|
||||||
"""Called when the user clicks on a menu item that doesn't represent a folder."""
|
"""Called when the user clicks on a menu item that doesn't represent a folder."""
|
||||||
|
|
||||||
from pillarsdk.utils import sanitize_filename
|
from pillarsdk.utils import sanitize_filename
|
||||||
@@ -806,11 +614,11 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
future=signalling_future))
|
future=signalling_future))
|
||||||
self.async_task.add_done_callback(texture_download_completed)
|
self.async_task.add_done_callback(texture_download_completed)
|
||||||
|
|
||||||
def open_browser_subscribe(self):
|
def open_browser_subscribe(self, *, renew: bool):
|
||||||
import webbrowser
|
import webbrowser
|
||||||
|
|
||||||
webbrowser.open_new_tab('https://cloud.blender.org/join')
|
url = 'renew' if renew else 'join'
|
||||||
|
webbrowser.open_new_tab('https://cloud.blender.org/%s' % url)
|
||||||
self.report({'INFO'}, 'We just started a browser for you.')
|
self.report({'INFO'}, 'We just started a browser for you.')
|
||||||
|
|
||||||
def _scroll_smooth(self):
|
def _scroll_smooth(self):
|
||||||
@@ -840,6 +648,7 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
self.scroll_offset_target = self.scroll_offset = 0
|
self.scroll_offset_target = self.scroll_offset = 0
|
||||||
|
|
||||||
|
|
||||||
|
@compatibility.convert_properties
|
||||||
class PILLAR_OT_switch_hdri(pillar.PillarOperatorMixin,
|
class PILLAR_OT_switch_hdri(pillar.PillarOperatorMixin,
|
||||||
async_loop.AsyncModalOperatorMixin,
|
async_loop.AsyncModalOperatorMixin,
|
||||||
bpy.types.Operator):
|
bpy.types.Operator):
|
||||||
@@ -865,9 +674,8 @@ class PILLAR_OT_switch_hdri(pillar.PillarOperatorMixin,
|
|||||||
try:
|
try:
|
||||||
db_user = await self.check_credentials(context, REQUIRED_ROLES_FOR_TEXTURE_BROWSER)
|
db_user = await self.check_credentials(context, REQUIRED_ROLES_FOR_TEXTURE_BROWSER)
|
||||||
user_id = db_user['_id']
|
user_id = db_user['_id']
|
||||||
except pillar.NotSubscribedToCloudError:
|
except pillar.NotSubscribedToCloudError as ex:
|
||||||
self.log.exception('User not subscribed to cloud.')
|
self._log_subscription_needed(can_renew=ex.can_renew)
|
||||||
self.report({'ERROR'}, 'Please subscribe to the Blender Cloud.')
|
|
||||||
self._state = 'QUIT'
|
self._state = 'QUIT'
|
||||||
return
|
return
|
||||||
except pillar.UserNotLoggedInError:
|
except pillar.UserNotLoggedInError:
|
||||||
@@ -887,13 +695,11 @@ class PILLAR_OT_switch_hdri(pillar.PillarOperatorMixin,
|
|||||||
self._state = 'QUIT'
|
self._state = 'QUIT'
|
||||||
|
|
||||||
async def download_and_replace(self, context):
|
async def download_and_replace(self, context):
|
||||||
from .pillar import sanitize_filename
|
|
||||||
|
|
||||||
self._state = 'DOWNLOADING_TEXTURE'
|
self._state = 'DOWNLOADING_TEXTURE'
|
||||||
|
|
||||||
current_image = bpy.data.images[self.image_name]
|
current_image = bpy.data.images[self.image_name]
|
||||||
node = current_image['bcloud_node']
|
node = current_image['bcloud_node']
|
||||||
filename = '%s.taken_from_file' % sanitize_filename(node['name'])
|
filename = '%s.taken_from_file' % pillar.sanitize_filename(node['name'])
|
||||||
|
|
||||||
local_path = os.path.dirname(bpy.path.abspath(current_image.filepath))
|
local_path = os.path.dirname(bpy.path.abspath(current_image.filepath))
|
||||||
top_texture_directory = bpy.path.abspath(context.scene.local_texture_dir)
|
top_texture_directory = bpy.path.abspath(context.scene.local_texture_dir)
|
||||||
@@ -903,21 +709,27 @@ class PILLAR_OT_switch_hdri(pillar.PillarOperatorMixin,
|
|||||||
resolution = next(file_ref['resolution'] for file_ref in node['properties']['files']
|
resolution = next(file_ref['resolution'] for file_ref in node['properties']['files']
|
||||||
if file_ref['file'] == file_uuid)
|
if file_ref['file'] == file_uuid)
|
||||||
|
|
||||||
self.log.info('Downloading file %r-%s to %s', file_uuid, resolution, local_path)
|
my_log = self.log
|
||||||
self.log.debug('Metadata will be stored at %s', meta_path)
|
my_log.info('Downloading file %r-%s to %s', file_uuid, resolution, local_path)
|
||||||
|
my_log.debug('Metadata will be stored at %s', meta_path)
|
||||||
|
|
||||||
def file_loading(file_path, file_desc, map_type):
|
def file_loading(file_path, file_desc, map_type):
|
||||||
self.log.info('Texture downloading to %s (%s)',
|
my_log.info('Texture downloading to %s (%s)',
|
||||||
file_path, utils.sizeof_fmt(file_desc['length']))
|
file_path, utils.sizeof_fmt(file_desc['length']))
|
||||||
|
|
||||||
async def file_loaded(file_path, file_desc, map_type):
|
async def file_loaded(file_path, file_desc, map_type):
|
||||||
if context.scene.local_texture_dir.startswith('//'):
|
if context.scene.local_texture_dir.startswith('//'):
|
||||||
file_path = bpy.path.relpath(file_path)
|
file_path = bpy.path.relpath(file_path)
|
||||||
|
|
||||||
self.log.info('Texture downloaded to %s', file_path)
|
my_log.info('Texture downloaded to %s', file_path)
|
||||||
current_image['bcloud_file_uuid'] = file_uuid
|
current_image['bcloud_file_uuid'] = file_uuid
|
||||||
current_image.filepath = file_path # This automatically reloads the image from disk.
|
current_image.filepath = file_path # This automatically reloads the image from disk.
|
||||||
|
|
||||||
|
# This forces users of the image to update.
|
||||||
|
for datablocks in bpy.data.user_map({current_image}).values():
|
||||||
|
for datablock in datablocks:
|
||||||
|
datablock.update_tag()
|
||||||
|
|
||||||
await pillar.download_file_by_uuid(file_uuid,
|
await pillar.download_file_by_uuid(file_uuid,
|
||||||
local_path,
|
local_path,
|
||||||
meta_path,
|
meta_path,
|
||||||
@@ -965,8 +777,8 @@ def _hdri_download_panel(self, current_image):
|
|||||||
current_image.name)
|
current_image.name)
|
||||||
return
|
return
|
||||||
|
|
||||||
row = self.layout.row(align=True).split(0.3)
|
row = self.layout.row(align=True).split(**compatibility.factor(0.3))
|
||||||
row.label('HDRi', icon_value=blender.icon('CLOUD'))
|
row.label(text='HDRi', icon_value=blender.icon('CLOUD'))
|
||||||
row.prop(current_image, 'hdri_variation', text='')
|
row.prop(current_image, 'hdri_variation', text='')
|
||||||
|
|
||||||
if current_image.hdri_variation != current_variation:
|
if current_image.hdri_variation != current_variation:
|
112
blender_cloud/texture_browser/draw.py
Normal file
112
blender_cloud/texture_browser/draw.py
Normal file
@@ -0,0 +1,112 @@
|
|||||||
|
"""OpenGL drawing code for the texture browser.
|
||||||
|
|
||||||
|
Requires Blender 2.80 or newer.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import typing
|
||||||
|
|
||||||
|
import bgl
|
||||||
|
import blf
|
||||||
|
import bpy
|
||||||
|
import gpu
|
||||||
|
from gpu_extras.batch import batch_for_shader
|
||||||
|
|
||||||
|
if bpy.app.background:
|
||||||
|
shader = None
|
||||||
|
texture_shader = None
|
||||||
|
else:
|
||||||
|
shader = gpu.shader.from_builtin('2D_UNIFORM_COLOR')
|
||||||
|
texture_shader = gpu.shader.from_builtin('2D_IMAGE')
|
||||||
|
|
||||||
|
Float2 = typing.Tuple[float, float]
|
||||||
|
Float3 = typing.Tuple[float, float, float]
|
||||||
|
Float4 = typing.Tuple[float, float, float, float]
|
||||||
|
|
||||||
|
|
||||||
|
def text(pos2d: Float2, display_text: typing.Union[str, typing.List[str]],
|
||||||
|
rgba: Float4 = (1.0, 1.0, 1.0, 1.0),
|
||||||
|
fsize=12,
|
||||||
|
align='L'):
|
||||||
|
"""Draw text with the top-left corner at 'pos2d'."""
|
||||||
|
|
||||||
|
dpi = bpy.context.preferences.system.dpi
|
||||||
|
gap = 12
|
||||||
|
x_pos, y_pos = pos2d
|
||||||
|
font_id = 0
|
||||||
|
blf.size(font_id, fsize, dpi)
|
||||||
|
|
||||||
|
# Compute the height of one line.
|
||||||
|
mwidth, mheight = blf.dimensions(font_id, "Tp") # Use high and low letters.
|
||||||
|
mheight *= 1.5
|
||||||
|
|
||||||
|
# Split text into lines.
|
||||||
|
if isinstance(display_text, str):
|
||||||
|
mylines = display_text.split("\n")
|
||||||
|
else:
|
||||||
|
mylines = display_text
|
||||||
|
maxwidth = 0
|
||||||
|
maxheight = len(mylines) * mheight
|
||||||
|
|
||||||
|
for idx, line in enumerate(mylines):
|
||||||
|
text_width, text_height = blf.dimensions(font_id, line)
|
||||||
|
if align == 'C':
|
||||||
|
newx = x_pos - text_width / 2
|
||||||
|
elif align == 'R':
|
||||||
|
newx = x_pos - text_width - gap
|
||||||
|
else:
|
||||||
|
newx = x_pos
|
||||||
|
|
||||||
|
# Draw
|
||||||
|
blf.position(font_id, newx, y_pos - mheight * idx, 0)
|
||||||
|
blf.color(font_id, rgba[0], rgba[1], rgba[2], rgba[3])
|
||||||
|
blf.draw(font_id, " " + line)
|
||||||
|
|
||||||
|
# saves max width
|
||||||
|
if maxwidth < text_width:
|
||||||
|
maxwidth = text_width
|
||||||
|
|
||||||
|
return maxwidth, maxheight
|
||||||
|
|
||||||
|
|
||||||
|
def aabox(v1: Float2, v2: Float2, rgba: Float4):
|
||||||
|
"""Draw an axis-aligned box."""
|
||||||
|
coords = [
|
||||||
|
(v1[0], v1[1]),
|
||||||
|
(v1[0], v2[1]),
|
||||||
|
(v2[0], v2[1]),
|
||||||
|
(v2[0], v1[1]),
|
||||||
|
]
|
||||||
|
shader.bind()
|
||||||
|
shader.uniform_float("color", rgba)
|
||||||
|
|
||||||
|
batch = batch_for_shader(shader, 'TRI_FAN', {"pos": coords})
|
||||||
|
batch.draw(shader)
|
||||||
|
|
||||||
|
|
||||||
|
def aabox_with_texture(v1: Float2, v2: Float2):
|
||||||
|
"""Draw an axis-aligned box with a texture."""
|
||||||
|
coords = [
|
||||||
|
(v1[0], v1[1]),
|
||||||
|
(v1[0], v2[1]),
|
||||||
|
(v2[0], v2[1]),
|
||||||
|
(v2[0], v1[1]),
|
||||||
|
]
|
||||||
|
texture_shader.bind()
|
||||||
|
texture_shader.uniform_int("image", 0)
|
||||||
|
|
||||||
|
batch = batch_for_shader(texture_shader, 'TRI_FAN', {
|
||||||
|
"pos": coords,
|
||||||
|
"texCoord": ((0, 0), (0, 1), (1, 1), (1, 0)),
|
||||||
|
})
|
||||||
|
batch.draw(texture_shader)
|
||||||
|
|
||||||
|
|
||||||
|
def bind_texture(texture: bpy.types.Image):
|
||||||
|
"""Bind a Blender image to a GL texture slot."""
|
||||||
|
bgl.glActiveTexture(bgl.GL_TEXTURE0)
|
||||||
|
bgl.glBindTexture(bgl.GL_TEXTURE_2D, texture.bindcode)
|
||||||
|
|
||||||
|
|
||||||
|
def load_texture(texture: bpy.types.Image) -> int:
|
||||||
|
"""Load the texture, return OpenGL error code."""
|
||||||
|
return texture.gl_load()
|
95
blender_cloud/texture_browser/draw_27.py
Normal file
95
blender_cloud/texture_browser/draw_27.py
Normal file
@@ -0,0 +1,95 @@
|
|||||||
|
"""OpenGL drawing code for the texture browser.
|
||||||
|
|
||||||
|
Requires Blender 2.79 or older.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import typing
|
||||||
|
|
||||||
|
import bgl
|
||||||
|
import blf
|
||||||
|
import bpy
|
||||||
|
|
||||||
|
Float2 = typing.Tuple[float, float]
|
||||||
|
Float3 = typing.Tuple[float, float, float]
|
||||||
|
Float4 = typing.Tuple[float, float, float, float]
|
||||||
|
|
||||||
|
|
||||||
|
def text(pos2d: Float2, display_text: typing.Union[str, typing.List[str]],
|
||||||
|
rgba: Float4 = (1.0, 1.0, 1.0, 1.0),
|
||||||
|
fsize=12,
|
||||||
|
align='L'):
|
||||||
|
"""Draw text with the top-left corner at 'pos2d'."""
|
||||||
|
|
||||||
|
dpi = bpy.context.user_preferences.system.dpi
|
||||||
|
gap = 12
|
||||||
|
x_pos, y_pos = pos2d
|
||||||
|
font_id = 0
|
||||||
|
blf.size(font_id, fsize, dpi)
|
||||||
|
|
||||||
|
# Compute the height of one line.
|
||||||
|
mwidth, mheight = blf.dimensions(font_id, "Tp") # Use high and low letters.
|
||||||
|
mheight *= 1.5
|
||||||
|
|
||||||
|
# Split text into lines.
|
||||||
|
if isinstance(display_text, str):
|
||||||
|
mylines = display_text.split("\n")
|
||||||
|
else:
|
||||||
|
mylines = display_text
|
||||||
|
maxwidth = 0
|
||||||
|
maxheight = len(mylines) * mheight
|
||||||
|
|
||||||
|
for idx, line in enumerate(mylines):
|
||||||
|
text_width, text_height = blf.dimensions(font_id, line)
|
||||||
|
if align == 'C':
|
||||||
|
newx = x_pos - text_width / 2
|
||||||
|
elif align == 'R':
|
||||||
|
newx = x_pos - text_width - gap
|
||||||
|
else:
|
||||||
|
newx = x_pos
|
||||||
|
|
||||||
|
# Draw
|
||||||
|
blf.position(font_id, newx, y_pos - mheight * idx, 0)
|
||||||
|
bgl.glColor4f(*rgba)
|
||||||
|
blf.draw(font_id, " " + line)
|
||||||
|
|
||||||
|
# saves max width
|
||||||
|
if maxwidth < text_width:
|
||||||
|
maxwidth = text_width
|
||||||
|
|
||||||
|
return maxwidth, maxheight
|
||||||
|
|
||||||
|
|
||||||
|
def aabox(v1: Float2, v2: Float2, rgba: Float4):
|
||||||
|
"""Draw an axis-aligned box."""
|
||||||
|
|
||||||
|
bgl.glColor4f(*rgba)
|
||||||
|
bgl.glRectf(*v1, *v2)
|
||||||
|
|
||||||
|
|
||||||
|
def aabox_with_texture(v1: Float2, v2: Float2):
|
||||||
|
"""Draw an axis-aligned box with a texture."""
|
||||||
|
|
||||||
|
bgl.glColor4f(1.0, 1.0, 1.0, 1.0)
|
||||||
|
|
||||||
|
bgl.glEnable(bgl.GL_TEXTURE_2D)
|
||||||
|
bgl.glBegin(bgl.GL_QUADS)
|
||||||
|
bgl.glTexCoord2d(0, 0)
|
||||||
|
bgl.glVertex2d(v1[0], v1[1])
|
||||||
|
bgl.glTexCoord2d(0, 1)
|
||||||
|
bgl.glVertex2d(v1[0], v2[1])
|
||||||
|
bgl.glTexCoord2d(1, 1)
|
||||||
|
bgl.glVertex2d(v2[0], v2[1])
|
||||||
|
bgl.glTexCoord2d(1, 0)
|
||||||
|
bgl.glVertex2d(v2[0], v1[1])
|
||||||
|
bgl.glEnd()
|
||||||
|
bgl.glDisable(bgl.GL_TEXTURE_2D)
|
||||||
|
|
||||||
|
|
||||||
|
def bind_texture(texture: bpy.types.Image):
|
||||||
|
"""Bind a Blender image to a GL texture slot."""
|
||||||
|
bgl.glBindTexture(bgl.GL_TEXTURE_2D, texture.bindcode[0])
|
||||||
|
|
||||||
|
|
||||||
|
def load_texture(texture: bpy.types.Image) -> int:
|
||||||
|
"""Load the texture, return OpenGL error code."""
|
||||||
|
return texture.gl_load(filter=bgl.GL_NEAREST, mag=bgl.GL_NEAREST)
|
BIN
blender_cloud/texture_browser/icons/error.png
Normal file
BIN
blender_cloud/texture_browser/icons/error.png
Normal file
Binary file not shown.
After Width: | Height: | Size: 11 KiB |
Before Width: | Height: | Size: 16 KiB After Width: | Height: | Size: 16 KiB |
Before Width: | Height: | Size: 2.4 KiB After Width: | Height: | Size: 2.4 KiB |
192
blender_cloud/texture_browser/menu_item.py
Normal file
192
blender_cloud/texture_browser/menu_item.py
Normal file
@@ -0,0 +1,192 @@
|
|||||||
|
import logging
|
||||||
|
import os.path
|
||||||
|
|
||||||
|
import bpy
|
||||||
|
import bgl
|
||||||
|
|
||||||
|
import pillarsdk
|
||||||
|
from . import nodes
|
||||||
|
|
||||||
|
if bpy.app.version < (2, 80):
|
||||||
|
from . import draw_27 as draw
|
||||||
|
else:
|
||||||
|
from . import draw
|
||||||
|
|
||||||
|
|
||||||
|
library_icons_path = os.path.join(os.path.dirname(__file__), "icons")
|
||||||
|
|
||||||
|
ICON_WIDTH = 128
|
||||||
|
ICON_HEIGHT = 128
|
||||||
|
|
||||||
|
|
||||||
|
class MenuItem:
|
||||||
|
"""GUI menu item for the 3D View GUI."""
|
||||||
|
|
||||||
|
icon_margin_x = 4
|
||||||
|
icon_margin_y = 4
|
||||||
|
text_margin_x = 6
|
||||||
|
|
||||||
|
text_size = 12
|
||||||
|
text_size_small = 10
|
||||||
|
|
||||||
|
DEFAULT_ICONS = {
|
||||||
|
'FOLDER': os.path.join(library_icons_path, 'folder.png'),
|
||||||
|
'SPINNER': os.path.join(library_icons_path, 'spinner.png'),
|
||||||
|
'ERROR': os.path.join(library_icons_path, 'error.png'),
|
||||||
|
}
|
||||||
|
|
||||||
|
FOLDER_NODE_TYPES = {'group_texture', 'group_hdri',
|
||||||
|
nodes.UpNode.NODE_TYPE, nodes.ProjectNode.NODE_TYPE}
|
||||||
|
SUPPORTED_NODE_TYPES = {'texture', 'hdri'}.union(FOLDER_NODE_TYPES)
|
||||||
|
|
||||||
|
def __init__(self, node, file_desc, thumb_path: str, label_text):
|
||||||
|
self.log = logging.getLogger('%s.MenuItem' % __name__)
|
||||||
|
if node['node_type'] not in self.SUPPORTED_NODE_TYPES:
|
||||||
|
self.log.info('Invalid node type in node: %s', node)
|
||||||
|
raise TypeError('Node of type %r not supported; supported are %r.' % (
|
||||||
|
node['node_type'], self.SUPPORTED_NODE_TYPES))
|
||||||
|
|
||||||
|
assert isinstance(node, pillarsdk.Node), 'wrong type for node: %r' % type(node)
|
||||||
|
assert isinstance(node['_id'], str), 'wrong type for node["_id"]: %r' % type(node['_id'])
|
||||||
|
self.node = node # pillarsdk.Node, contains 'node_type' key to indicate type
|
||||||
|
self.file_desc = file_desc # pillarsdk.File object, or None if a 'folder' node.
|
||||||
|
self.label_text = label_text
|
||||||
|
self.small_text = self._small_text_from_node()
|
||||||
|
self._thumb_path = ''
|
||||||
|
self.icon = None
|
||||||
|
self._is_folder = node['node_type'] in self.FOLDER_NODE_TYPES
|
||||||
|
self._is_spinning = False
|
||||||
|
|
||||||
|
# Determine sorting order.
|
||||||
|
# by default, sort all the way at the end and folders first.
|
||||||
|
self._order = 0 if self._is_folder else 10000
|
||||||
|
if node and node.properties and node.properties.order is not None:
|
||||||
|
self._order = node.properties.order
|
||||||
|
|
||||||
|
self.thumb_path = thumb_path
|
||||||
|
|
||||||
|
# Updated when drawing the image
|
||||||
|
self.x = 0
|
||||||
|
self.y = 0
|
||||||
|
self.width = 0
|
||||||
|
self.height = 0
|
||||||
|
|
||||||
|
def _small_text_from_node(self) -> str:
|
||||||
|
"""Return the components of the texture (i.e. which map types are available)."""
|
||||||
|
|
||||||
|
if not self.node:
|
||||||
|
return ''
|
||||||
|
|
||||||
|
try:
|
||||||
|
node_files = self.node.properties.files
|
||||||
|
except AttributeError:
|
||||||
|
# Happens for nodes that don't have .properties.files.
|
||||||
|
return ''
|
||||||
|
if not node_files:
|
||||||
|
return ''
|
||||||
|
|
||||||
|
map_types = {f.map_type for f in node_files if f.map_type}
|
||||||
|
map_types.discard('color') # all textures have colour
|
||||||
|
if not map_types:
|
||||||
|
return ''
|
||||||
|
return ', '.join(sorted(map_types))
|
||||||
|
|
||||||
|
def sort_key(self):
|
||||||
|
"""Key for sorting lists of MenuItems."""
|
||||||
|
return self._order, self.label_text
|
||||||
|
|
||||||
|
@property
|
||||||
|
def thumb_path(self) -> str:
|
||||||
|
return self._thumb_path
|
||||||
|
|
||||||
|
@thumb_path.setter
|
||||||
|
def thumb_path(self, new_thumb_path: str):
|
||||||
|
self._is_spinning = new_thumb_path == 'SPINNER'
|
||||||
|
|
||||||
|
self._thumb_path = self.DEFAULT_ICONS.get(new_thumb_path, new_thumb_path)
|
||||||
|
if self._thumb_path:
|
||||||
|
self.icon = bpy.data.images.load(filepath=self._thumb_path)
|
||||||
|
else:
|
||||||
|
self.icon = None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def node_uuid(self) -> str:
|
||||||
|
return self.node['_id']
|
||||||
|
|
||||||
|
def represents(self, node) -> bool:
|
||||||
|
"""Returns True iff this MenuItem represents the given node."""
|
||||||
|
|
||||||
|
node_uuid = node['_id']
|
||||||
|
return self.node_uuid == node_uuid
|
||||||
|
|
||||||
|
def update(self, node, file_desc, thumb_path: str, label_text=None):
|
||||||
|
# We can get updated information about our Node, but a MenuItem should
|
||||||
|
# always represent one node, and it shouldn't be shared between nodes.
|
||||||
|
if self.node_uuid != node['_id']:
|
||||||
|
raise ValueError("Don't change the node ID this MenuItem reflects, "
|
||||||
|
"just create a new one.")
|
||||||
|
self.node = node
|
||||||
|
self.file_desc = file_desc # pillarsdk.File object, or None if a 'folder' node.
|
||||||
|
self.thumb_path = thumb_path
|
||||||
|
|
||||||
|
if label_text is not None:
|
||||||
|
self.label_text = label_text
|
||||||
|
|
||||||
|
if thumb_path == 'ERROR':
|
||||||
|
self.small_text = 'This open is broken'
|
||||||
|
else:
|
||||||
|
self.small_text = self._small_text_from_node()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_folder(self) -> bool:
|
||||||
|
return self._is_folder
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_spinning(self) -> bool:
|
||||||
|
return self._is_spinning
|
||||||
|
|
||||||
|
def update_placement(self, x, y, width, height):
|
||||||
|
"""Use OpenGL to draw this one menu item."""
|
||||||
|
|
||||||
|
self.x = x
|
||||||
|
self.y = y
|
||||||
|
self.width = width
|
||||||
|
self.height = height
|
||||||
|
|
||||||
|
def draw(self, highlighted: bool):
|
||||||
|
bgl.glEnable(bgl.GL_BLEND)
|
||||||
|
if highlighted:
|
||||||
|
color = (0.555, 0.555, 0.555, 0.8)
|
||||||
|
else:
|
||||||
|
color = (0.447, 0.447, 0.447, 0.8)
|
||||||
|
|
||||||
|
draw.aabox((self.x, self.y), (self.x + self.width, self.y + self.height), color)
|
||||||
|
|
||||||
|
texture = self.icon
|
||||||
|
if texture:
|
||||||
|
err = draw.load_texture(texture)
|
||||||
|
assert not err, 'OpenGL error: %i' % err
|
||||||
|
|
||||||
|
# ------ TEXTURE ---------#
|
||||||
|
if texture:
|
||||||
|
draw.bind_texture(texture)
|
||||||
|
bgl.glBlendFunc(bgl.GL_SRC_ALPHA, bgl.GL_ONE_MINUS_SRC_ALPHA)
|
||||||
|
|
||||||
|
draw.aabox_with_texture(
|
||||||
|
(self.x + self.icon_margin_x, self.y),
|
||||||
|
(self.x + self.icon_margin_x + ICON_WIDTH, self.y + ICON_HEIGHT),
|
||||||
|
)
|
||||||
|
bgl.glDisable(bgl.GL_BLEND)
|
||||||
|
|
||||||
|
if texture:
|
||||||
|
texture.gl_free()
|
||||||
|
|
||||||
|
# draw some text
|
||||||
|
text_x = self.x + self.icon_margin_x + ICON_WIDTH + self.text_margin_x
|
||||||
|
text_y = self.y + ICON_HEIGHT * 0.5 - 0.25 * self.text_size
|
||||||
|
draw.text((text_x, text_y), self.label_text, fsize=self.text_size)
|
||||||
|
draw.text((text_x, self.y + 0.5 * self.text_size_small), self.small_text,
|
||||||
|
fsize=self.text_size_small, rgba=(1.0, 1.0, 1.0, 0.5))
|
||||||
|
|
||||||
|
def hits(self, mouse_x: int, mouse_y: int) -> bool:
|
||||||
|
return self.x < mouse_x < self.x + self.width and self.y < mouse_y < self.y + self.height
|
26
blender_cloud/texture_browser/nodes.py
Normal file
26
blender_cloud/texture_browser/nodes.py
Normal file
@@ -0,0 +1,26 @@
|
|||||||
|
import pillarsdk
|
||||||
|
|
||||||
|
|
||||||
|
class SpecialFolderNode(pillarsdk.Node):
|
||||||
|
NODE_TYPE = 'SPECIAL'
|
||||||
|
|
||||||
|
|
||||||
|
class UpNode(SpecialFolderNode):
|
||||||
|
NODE_TYPE = 'UP'
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
super().__init__()
|
||||||
|
self['_id'] = 'UP'
|
||||||
|
self['node_type'] = self.NODE_TYPE
|
||||||
|
|
||||||
|
|
||||||
|
class ProjectNode(SpecialFolderNode):
|
||||||
|
NODE_TYPE = 'PROJECT'
|
||||||
|
|
||||||
|
def __init__(self, project):
|
||||||
|
super().__init__()
|
||||||
|
|
||||||
|
assert isinstance(project, pillarsdk.Project), 'wrong type for project: %r' % type(project)
|
||||||
|
|
||||||
|
self.merge(project.to_dict())
|
||||||
|
self['node_type'] = self.NODE_TYPE
|
@@ -16,7 +16,9 @@
|
|||||||
#
|
#
|
||||||
# ##### END GPL LICENSE BLOCK #####
|
# ##### END GPL LICENSE BLOCK #####
|
||||||
|
|
||||||
|
import json
|
||||||
import pathlib
|
import pathlib
|
||||||
|
import typing
|
||||||
|
|
||||||
|
|
||||||
def sizeof_fmt(num: int, suffix='B') -> str:
|
def sizeof_fmt(num: int, suffix='B') -> str:
|
||||||
@@ -28,12 +30,12 @@ def sizeof_fmt(num: int, suffix='B') -> str:
|
|||||||
for unit in ['', 'Ki', 'Mi', 'Gi', 'Ti', 'Pi', 'Ei', 'Zi']:
|
for unit in ['', 'Ki', 'Mi', 'Gi', 'Ti', 'Pi', 'Ei', 'Zi']:
|
||||||
if abs(num) < 1024:
|
if abs(num) < 1024:
|
||||||
return '%.1f %s%s' % (num, unit, suffix)
|
return '%.1f %s%s' % (num, unit, suffix)
|
||||||
num /= 1024
|
num //= 1024
|
||||||
|
|
||||||
return '%.1f Yi%s' % (num, suffix)
|
return '%.1f Yi%s' % (num, suffix)
|
||||||
|
|
||||||
|
|
||||||
def find_in_path(path: pathlib.Path, filename: str) -> pathlib.Path:
|
def find_in_path(path: pathlib.Path, filename: str) -> typing.Optional[pathlib.Path]:
|
||||||
"""Performs a breadth-first search for the filename.
|
"""Performs a breadth-first search for the filename.
|
||||||
|
|
||||||
Returns the path that contains the file, or None if not found.
|
Returns the path that contains the file, or None if not found.
|
||||||
@@ -92,6 +94,9 @@ def pyside_cache(propname):
|
|||||||
result = wrapped(self, context)
|
result = wrapped(self, context)
|
||||||
return result
|
return result
|
||||||
finally:
|
finally:
|
||||||
|
try:
|
||||||
|
rna_type, rna_info = self.bl_rna.__annotations__[propname]
|
||||||
|
except AttributeError:
|
||||||
rna_type, rna_info = getattr(self.bl_rna, propname)
|
rna_type, rna_info = getattr(self.bl_rna, propname)
|
||||||
rna_info['_cached_result'] = result
|
rna_info['_cached_result'] = result
|
||||||
return wrapper
|
return wrapper
|
||||||
@@ -99,4 +104,15 @@ def pyside_cache(propname):
|
|||||||
|
|
||||||
|
|
||||||
def redraw(self, context):
|
def redraw(self, context):
|
||||||
|
if context.area is None:
|
||||||
|
return
|
||||||
context.area.tag_redraw()
|
context.area.tag_redraw()
|
||||||
|
|
||||||
|
|
||||||
|
class JSONEncoder(json.JSONEncoder):
|
||||||
|
"""JSON encoder with support for some Blender types."""
|
||||||
|
|
||||||
|
def default(self, o):
|
||||||
|
if o.__class__.__name__ == 'IDPropertyGroup' and hasattr(o, 'to_dict'):
|
||||||
|
return o.to_dict()
|
||||||
|
return super().default(o)
|
||||||
|
@@ -44,6 +44,12 @@ def load_wheel(module_name, fname_prefix):
|
|||||||
module_name, module.__file__, fname_prefix)
|
module_name, module.__file__, fname_prefix)
|
||||||
return
|
return
|
||||||
|
|
||||||
|
sys.path.append(wheel_filename(fname_prefix))
|
||||||
|
module = __import__(module_name)
|
||||||
|
log.debug('Loaded %s from %s', module_name, module.__file__)
|
||||||
|
|
||||||
|
|
||||||
|
def wheel_filename(fname_prefix: str) -> str:
|
||||||
path_pattern = os.path.join(my_dir, '%s*.whl' % fname_prefix)
|
path_pattern = os.path.join(my_dir, '%s*.whl' % fname_prefix)
|
||||||
wheels = glob.glob(path_pattern)
|
wheels = glob.glob(path_pattern)
|
||||||
if not wheels:
|
if not wheels:
|
||||||
@@ -51,12 +57,11 @@ def load_wheel(module_name, fname_prefix):
|
|||||||
|
|
||||||
# If there are multiple wheels that match, load the latest one.
|
# If there are multiple wheels that match, load the latest one.
|
||||||
wheels.sort()
|
wheels.sort()
|
||||||
sys.path.append(wheels[-1])
|
return wheels[-1]
|
||||||
module = __import__(module_name)
|
|
||||||
log.debug('Loaded %s from %s', module_name, module.__file__)
|
|
||||||
|
|
||||||
|
|
||||||
def load_wheels():
|
def load_wheels():
|
||||||
|
load_wheel('blender_asset_tracer', 'blender_asset_tracer')
|
||||||
load_wheel('lockfile', 'lockfile')
|
load_wheel('lockfile', 'lockfile')
|
||||||
load_wheel('cachecontrol', 'CacheControl')
|
load_wheel('cachecontrol', 'CacheControl')
|
||||||
load_wheel('pillarsdk', 'pillarsdk')
|
load_wheel('pillarsdk', 'pillarsdk')
|
||||||
|
13
deploy-to-shared.sh
Executable file
13
deploy-to-shared.sh
Executable file
@@ -0,0 +1,13 @@
|
|||||||
|
#!/bin/bash -e
|
||||||
|
|
||||||
|
FULLNAME="$(python3 setup.py --fullname)"
|
||||||
|
echo "Press [ENTER] to deploy $FULLNAME to /shared"
|
||||||
|
read dummy
|
||||||
|
|
||||||
|
./clear_wheels.sh
|
||||||
|
python3 setup.py wheels bdist
|
||||||
|
|
||||||
|
DISTDIR=$(pwd)/dist
|
||||||
|
cd /shared/software/addons
|
||||||
|
rm -vf blender_cloud/wheels/*.whl # remove obsolete wheel files
|
||||||
|
unzip -o $DISTDIR/$FULLNAME.addon.zip
|
@@ -1,15 +1,17 @@
|
|||||||
# Primary requirements:
|
# Primary requirements:
|
||||||
-e git+https://github.com/sybrenstuvel/cachecontrol.git@sybren-filecache-delete-crash-fix#egg=CacheControl
|
-e git+https://github.com/sybrenstuvel/cachecontrol.git@sybren-filecache-delete-crash-fix#egg=CacheControl
|
||||||
lockfile==0.12.2
|
lockfile==0.12.2
|
||||||
pillarsdk==1.6.1
|
pillarsdk==1.8.0
|
||||||
wheel==0.29.0
|
wheel==0.29.0
|
||||||
|
blender-asset-tracer==1.2.1
|
||||||
|
|
||||||
# Secondary requirements:
|
# Secondary requirements:
|
||||||
cffi==1.6.0
|
asn1crypto==0.24.0
|
||||||
cryptography==1.3.1
|
cffi==1.11.2
|
||||||
idna==2.1
|
cryptography==2.1.4
|
||||||
|
idna==2.6
|
||||||
pyasn1==0.1.9
|
pyasn1==0.1.9
|
||||||
pycparser==2.14
|
pycparser==2.18
|
||||||
pyOpenSSL==16.0.0
|
pyOpenSSL==17.5.0
|
||||||
requests==2.10.0
|
requests==2.10.0
|
||||||
six==1.10.0
|
six==1.11.0
|
||||||
|
47
setup.py
47
setup.py
@@ -18,7 +18,6 @@
|
|||||||
# ##### END GPL LICENSE BLOCK #####
|
# ##### END GPL LICENSE BLOCK #####
|
||||||
|
|
||||||
import glob
|
import glob
|
||||||
import os
|
|
||||||
import sys
|
import sys
|
||||||
import shutil
|
import shutil
|
||||||
import subprocess
|
import subprocess
|
||||||
@@ -29,13 +28,18 @@ import zipfile
|
|||||||
from distutils import log
|
from distutils import log
|
||||||
from distutils.core import Command
|
from distutils.core import Command
|
||||||
from distutils.command.bdist import bdist
|
from distutils.command.bdist import bdist
|
||||||
from distutils.command.install import install
|
from distutils.command.install import install, INSTALL_SCHEMES
|
||||||
from distutils.command.install_egg_info import install_egg_info
|
from distutils.command.install_egg_info import install_egg_info
|
||||||
from setuptools import setup, find_packages
|
from setuptools import setup, find_packages
|
||||||
|
|
||||||
requirement_re = re.compile('[><=]+')
|
requirement_re = re.compile('[><=]+')
|
||||||
sys.dont_write_bytecode = True
|
sys.dont_write_bytecode = True
|
||||||
|
|
||||||
|
# Download wheels from pypi. The specific versions are taken from requirements.txt
|
||||||
|
wheels = [
|
||||||
|
'lockfile', 'pillarsdk', 'blender-asset-tracer',
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
def set_default_path(var, default):
|
def set_default_path(var, default):
|
||||||
"""convert CLI-arguments (string) to Paths"""
|
"""convert CLI-arguments (string) to Paths"""
|
||||||
@@ -60,6 +64,7 @@ class BuildWheels(Command):
|
|||||||
self.wheels_path = None # path that will contain the installed wheels.
|
self.wheels_path = None # path that will contain the installed wheels.
|
||||||
self.deps_path = None # path in which dependencies are built.
|
self.deps_path = None # path in which dependencies are built.
|
||||||
self.cachecontrol_path = None # subdir of deps_path containing CacheControl
|
self.cachecontrol_path = None # subdir of deps_path containing CacheControl
|
||||||
|
self.bat_path = None # subdir of deps_path containing Blender-Asset-Tracer
|
||||||
|
|
||||||
def finalize_options(self):
|
def finalize_options(self):
|
||||||
self.my_path = pathlib.Path(__file__).resolve().parent
|
self.my_path = pathlib.Path(__file__).resolve().parent
|
||||||
@@ -69,6 +74,7 @@ class BuildWheels(Command):
|
|||||||
self.deps_path = set_default_path(self.deps_path, self.my_path / 'build/deps')
|
self.deps_path = set_default_path(self.deps_path, self.my_path / 'build/deps')
|
||||||
self.cachecontrol_path = set_default_path(self.cachecontrol_path,
|
self.cachecontrol_path = set_default_path(self.cachecontrol_path,
|
||||||
self.deps_path / 'cachecontrol')
|
self.deps_path / 'cachecontrol')
|
||||||
|
self.bat_path = self.deps_path / 'bat'
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
log.info('Storing wheels in %s', self.wheels_path)
|
log.info('Storing wheels in %s', self.wheels_path)
|
||||||
@@ -90,16 +96,11 @@ class BuildWheels(Command):
|
|||||||
# log.info(' - %s = %s / %s', package, line, line_req[-1])
|
# log.info(' - %s = %s / %s', package, line, line_req[-1])
|
||||||
|
|
||||||
self.wheels_path.mkdir(parents=True, exist_ok=True)
|
self.wheels_path.mkdir(parents=True, exist_ok=True)
|
||||||
|
for package in wheels:
|
||||||
# Download lockfile, as there is a suitable wheel on pypi.
|
pattern = package.replace('-', '_') + '*.whl'
|
||||||
if not list(self.wheels_path.glob('lockfile*.whl')):
|
if list(self.wheels_path.glob(pattern)):
|
||||||
log.info('Downloading lockfile wheel')
|
continue
|
||||||
self.download_wheel(requirements['lockfile'])
|
self.download_wheel(requirements[package])
|
||||||
|
|
||||||
# Download Pillar Python SDK from pypi.
|
|
||||||
if not list(self.wheels_path.glob('pillarsdk*.whl')):
|
|
||||||
log.info('Downloading Pillar Python SDK wheel')
|
|
||||||
self.download_wheel(requirements['pillarsdk'])
|
|
||||||
|
|
||||||
# Build CacheControl.
|
# Build CacheControl.
|
||||||
if not list(self.wheels_path.glob('CacheControl*.whl')):
|
if not list(self.wheels_path.glob('CacheControl*.whl')):
|
||||||
@@ -122,8 +123,8 @@ class BuildWheels(Command):
|
|||||||
"""Downloads a wheel from PyPI and saves it in self.wheels_path."""
|
"""Downloads a wheel from PyPI and saves it in self.wheels_path."""
|
||||||
|
|
||||||
subprocess.check_call([
|
subprocess.check_call([
|
||||||
'pip', 'download',
|
sys.executable, '-m', 'pip',
|
||||||
'--no-deps',
|
'download', '--no-deps',
|
||||||
'--dest', str(self.wheels_path),
|
'--dest', str(self.wheels_path),
|
||||||
requirement[0]
|
requirement[0]
|
||||||
])
|
])
|
||||||
@@ -164,6 +165,14 @@ class BlenderAddonBdist(bdist):
|
|||||||
super().initialize_options()
|
super().initialize_options()
|
||||||
self.formats = ['zip']
|
self.formats = ['zip']
|
||||||
self.plat_name = 'addon' # use this instead of 'linux-x86_64' or similar.
|
self.plat_name = 'addon' # use this instead of 'linux-x86_64' or similar.
|
||||||
|
self.fix_local_prefix()
|
||||||
|
|
||||||
|
def fix_local_prefix(self):
|
||||||
|
"""Place data files in blender_cloud instead of local/blender_cloud."""
|
||||||
|
for key in INSTALL_SCHEMES:
|
||||||
|
if 'data' not in INSTALL_SCHEMES[key]:
|
||||||
|
continue
|
||||||
|
INSTALL_SCHEMES[key]['data'] = '$base'
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
self.run_command('wheels')
|
self.run_command('wheels')
|
||||||
@@ -227,12 +236,16 @@ setup(
|
|||||||
'wheels': BuildWheels},
|
'wheels': BuildWheels},
|
||||||
name='blender_cloud',
|
name='blender_cloud',
|
||||||
description='The Blender Cloud addon allows browsing the Blender Cloud from Blender.',
|
description='The Blender Cloud addon allows browsing the Blender Cloud from Blender.',
|
||||||
version='1.6.4',
|
version='1.15',
|
||||||
author='Sybren A. Stüvel',
|
author='Sybren A. Stüvel',
|
||||||
author_email='sybren@stuvel.eu',
|
author_email='sybren@stuvel.eu',
|
||||||
packages=find_packages('.'),
|
packages=find_packages('.'),
|
||||||
data_files=[('blender_cloud', ['README.md', 'README-flamenco.md', 'CHANGELOG.md']),
|
data_files=[
|
||||||
('blender_cloud/icons', glob.glob('blender_cloud/icons/*'))],
|
('blender_cloud', ['README.md', 'README-flamenco.md', 'CHANGELOG.md']),
|
||||||
|
('blender_cloud/icons', glob.glob('blender_cloud/icons/*')),
|
||||||
|
('blender_cloud/texture_browser/icons',
|
||||||
|
glob.glob('blender_cloud/texture_browser/icons/*'))
|
||||||
|
],
|
||||||
scripts=[],
|
scripts=[],
|
||||||
url='https://developer.blender.org/diffusion/BCA/',
|
url='https://developer.blender.org/diffusion/BCA/',
|
||||||
license='GNU General Public License v2 or later (GPLv2+)',
|
license='GNU General Public License v2 or later (GPLv2+)',
|
||||||
|
95
tests/test_path_replacement.py
Normal file
95
tests/test_path_replacement.py
Normal file
@@ -0,0 +1,95 @@
|
|||||||
|
"""Unittests for blender_cloud.utils.
|
||||||
|
|
||||||
|
This unittest requires bpy to be importable, so build Blender as a module and install it
|
||||||
|
into your virtualenv. See https://stuvel.eu/files/bconf2016/#/10 for notes how.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import datetime
|
||||||
|
import pathlib
|
||||||
|
import unittest.mock
|
||||||
|
|
||||||
|
import pillarsdk.utils
|
||||||
|
|
||||||
|
from blender_cloud.flamenco import sdk
|
||||||
|
|
||||||
|
|
||||||
|
class PathReplacementTest(unittest.TestCase):
|
||||||
|
def setUp(self):
|
||||||
|
self.test_manager = sdk.Manager({
|
||||||
|
'_created': datetime.datetime(2017, 5, 31, 15, 12, 32, tzinfo=pillarsdk.utils.utc),
|
||||||
|
'_etag': 'c39942ee4bcc4658adcc21e4bcdfb0ae',
|
||||||
|
'_id': '592edd609837732a2a272c62',
|
||||||
|
'_updated': datetime.datetime(2017, 6, 8, 14, 51, 3, tzinfo=pillarsdk.utils.utc),
|
||||||
|
'description': 'Manager formerly known as "testman"',
|
||||||
|
'job_types': {'sleep': {'vars': {}}},
|
||||||
|
'name': '<script>alert("this is a manager")</script>',
|
||||||
|
'owner': '592edd609837732a2a272c63',
|
||||||
|
'path_replacement': {'job_storage': {'darwin': '/Volume/shared',
|
||||||
|
'linux': '/shared',
|
||||||
|
'windows': 's:/'},
|
||||||
|
'render': {'darwin': '/Volume/render/',
|
||||||
|
'linux': '/render/',
|
||||||
|
'windows': 'r:/'},
|
||||||
|
'longrender': {'darwin': '/Volume/render/long',
|
||||||
|
'linux': '/render/long',
|
||||||
|
'windows': 'r:/long'},
|
||||||
|
},
|
||||||
|
'projects': ['58cbdd5698377322d95eb55e'],
|
||||||
|
'service_account': '592edd609837732a2a272c60',
|
||||||
|
'stats': {'nr_of_workers': 3},
|
||||||
|
'url': 'http://192.168.3.101:8083/',
|
||||||
|
'user_groups': ['58cbdd5698377322d95eb55f'],
|
||||||
|
'variables': {'blender': {'darwin': '/opt/myblenderbuild/blender',
|
||||||
|
'linux': '/home/sybren/workspace/build_linux/bin/blender '
|
||||||
|
'--enable-new-depsgraph --factory-startup',
|
||||||
|
'windows': 'c:/temp/blender.exe'}}}
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_linux(self):
|
||||||
|
# (expected result, input)
|
||||||
|
test_paths = [
|
||||||
|
('/doesnotexistreally', '/doesnotexistreally'),
|
||||||
|
('{render}/agent327/scenes/A_01_03_B', '/render/agent327/scenes/A_01_03_B'),
|
||||||
|
('{job_storage}/render/agent327/scenes', '/shared/render/agent327/scenes'),
|
||||||
|
('{longrender}/agent327/scenes', '/render/long/agent327/scenes'),
|
||||||
|
]
|
||||||
|
|
||||||
|
self._do_test(test_paths, 'linux', pathlib.PurePosixPath)
|
||||||
|
|
||||||
|
def test_windows(self):
|
||||||
|
# (expected result, input)
|
||||||
|
test_paths = [
|
||||||
|
('c:/doesnotexistreally', 'c:/doesnotexistreally'),
|
||||||
|
('c:/some/path', r'c:\some\path'),
|
||||||
|
('{render}/agent327/scenes/A_01_03_B', r'R:\agent327\scenes\A_01_03_B'),
|
||||||
|
('{render}/agent327/scenes/A_01_03_B', r'r:\agent327\scenes\A_01_03_B'),
|
||||||
|
('{render}/agent327/scenes/A_01_03_B', r'r:/agent327/scenes/A_01_03_B'),
|
||||||
|
('{job_storage}/render/agent327/scenes', 's:/render/agent327/scenes'),
|
||||||
|
('{longrender}/agent327/scenes', 'r:/long/agent327/scenes'),
|
||||||
|
]
|
||||||
|
|
||||||
|
self._do_test(test_paths, 'windows', pathlib.PureWindowsPath)
|
||||||
|
|
||||||
|
def test_darwin(self):
|
||||||
|
# (expected result, input)
|
||||||
|
test_paths = [
|
||||||
|
('/Volume/doesnotexistreally', '/Volume/doesnotexistreally'),
|
||||||
|
('{render}/agent327/scenes/A_01_03_B', r'/Volume/render/agent327/scenes/A_01_03_B'),
|
||||||
|
('{job_storage}/render/agent327/scenes', '/Volume/shared/render/agent327/scenes'),
|
||||||
|
('{longrender}/agent327/scenes', '/Volume/render/long/agent327/scenes'),
|
||||||
|
]
|
||||||
|
|
||||||
|
self._do_test(test_paths, 'darwin', pathlib.PurePosixPath)
|
||||||
|
|
||||||
|
def _do_test(self, test_paths, platform, pathclass):
|
||||||
|
self.test_manager.PurePlatformPath = pathclass
|
||||||
|
|
||||||
|
def mocked_system():
|
||||||
|
return platform
|
||||||
|
|
||||||
|
with unittest.mock.patch('platform.system', mocked_system):
|
||||||
|
for expected_result, input_path in test_paths:
|
||||||
|
as_path_instance = pathclass(input_path)
|
||||||
|
self.assertEqual(expected_result,
|
||||||
|
self.test_manager.replace_path(as_path_instance),
|
||||||
|
'for input %r on platform %s' % (as_path_instance, platform))
|
@@ -17,3 +17,6 @@ echo
|
|||||||
echo "Don't forget to commit and tag:"
|
echo "Don't forget to commit and tag:"
|
||||||
echo git commit -m \'Bumped version to $VERSION\' setup.py blender_cloud/__init__.py
|
echo git commit -m \'Bumped version to $VERSION\' setup.py blender_cloud/__init__.py
|
||||||
echo git tag -a version-$VERSION -m \'Tagged version $VERSION\'
|
echo git tag -a version-$VERSION -m \'Tagged version $VERSION\'
|
||||||
|
echo
|
||||||
|
echo "To build a distribution ZIP:"
|
||||||
|
echo python3 setup.py bdist
|
||||||
|
Reference in New Issue
Block a user