Compare commits
144 Commits
version-1.
...
version-1.
Author | SHA1 | Date | |
---|---|---|---|
06126862d4 | |||
7b8713881e | |||
7c65851b75 | |||
ec72091268 | |||
cf7adb065f | |||
74220e4fc4 | |||
0ebd4435e5 | |||
c24501661e | |||
5b77ae50a1 | |||
74958cf217 | |||
5026dfc441 | |||
843667e612 | |||
cf3f7234eb | |||
4647175a7e | |||
33da5195f3 | |||
3814fb2683 | |||
15484a65cd | |||
d9e2b36204 | |||
cc690ec8c9 | |||
0422070d55 | |||
8cefb4fb07 | |||
23549fa676 | |||
cb73030e6a | |||
fbf02c3625 | |||
95699aca36 | |||
60018cd78c | |||
5f73837d3c | |||
9272e22129 | |||
e14a0aa53c | |||
51cf097c8f | |||
4608204f1d | |||
3f95249196 | |||
9df016da09 | |||
64e29e695b | |||
d5f285a381 | |||
e39429272d | |||
bdb00eeaaa | |||
3ef2ca0c07 | |||
35d4f85010 | |||
8151b952b9 | |||
2d2585b8d7 | |||
65204db228 | |||
570b1d4bfe | |||
68b046c714 | |||
cb20d6ee03 | |||
![]() |
645bdd950f | ||
![]() |
74a5830dae | ||
da4d4df5fb | |||
9c3098cc0d | |||
98beaf7fb7 | |||
3364371ac6 | |||
2723b07fa2 | |||
c2a037ca89 | |||
d3451d4de3 | |||
5094977614 | |||
![]() |
a11a55be22 | ||
68d2fc8e42 | |||
2de4a8e87c | |||
39b2bacdcc | |||
a1416f99dd | |||
0fa7d60028 | |||
c1b6480f9a | |||
56353d4177 | |||
469a9318af | |||
e265081131 | |||
115eea82c6 | |||
900068a6f5 | |||
c8229500d1 | |||
65ff9da428 | |||
fcba8a2e0f | |||
7ef5e522f8 | |||
ae570e5907 | |||
16b90d2ea8 | |||
875c92ee9d | |||
cbfc75a89c | |||
54e676b36f | |||
c94b0f5f2d | |||
a58bfe9a76 | |||
d332e1e50a | |||
![]() |
dd66d5ce93 | ||
965b02cec4 | |||
fd67675c12 | |||
06a661126b | |||
191b150280 | |||
feb62ddae0 | |||
603159f0d1 | |||
d2ae3f9cb7 | |||
079f8ff4c3 | |||
c97859ef33 | |||
f1ebea8948 | |||
1b82977c6e | |||
a7307bf7b5 | |||
11fd12e125 | |||
54dccb20ba | |||
61a8db3f96 | |||
0b2f0a3ec1 | |||
5117ec7cde | |||
74f61fa83a | |||
f73671c4f0 | |||
6f970a41e5 | |||
ccedb7cbb1 | |||
abcd8b0168 | |||
534a5a6ac4 | |||
6b5faa423e | |||
232e8f6167 | |||
af0dee0c9d | |||
baac86f59b | |||
19d54b7fd6 | |||
0067157251 | |||
4c84fbf339 | |||
6a5c392b5b | |||
cbaccaed49 | |||
cfc53e007c | |||
2768f0a59f | |||
b6c7ec1546 | |||
417b6e80f5 | |||
90259297ca | |||
3d9f4e893a | |||
4be497ed27 | |||
28fe6e8f96 | |||
22e4f2dc5e | |||
537dcf846a | |||
8ca4159fe8 | |||
d7bf001ffe | |||
6ea15d2bfe | |||
6fda496652 | |||
8dab01138e | |||
3da76ddb24 | |||
c57da7ab2b | |||
63b976cb44 | |||
73a62da8da | |||
2c70ceb489 | |||
38ccb54b50 | |||
1df113ca01 | |||
887a9cc697 | |||
143456ae1d | |||
f41ea8c5a3 | |||
7d90a92e24 | |||
2388f800dc | |||
38a3bcba71 | |||
2cf400a74c | |||
54ebb0bf5d | |||
9e84d2a416 | |||
772e6b0b1b |
4
.gitignore
vendored
4
.gitignore
vendored
@@ -1,10 +1,10 @@
|
|||||||
*.pyc
|
*.pyc
|
||||||
*.swp
|
*.swp
|
||||||
*.blend
|
/*.blend*
|
||||||
*.blend[1-9]
|
|
||||||
blender_cloud/wheels/*.whl
|
blender_cloud/wheels/*.whl
|
||||||
/textures*/
|
/textures*/
|
||||||
/test_*.py
|
/test_*.py
|
||||||
/dist/
|
/dist/
|
||||||
/build/
|
/build/
|
||||||
/addon-bundle/*.zip
|
/addon-bundle/*.zip
|
||||||
|
__pycache__
|
||||||
|
32
CHANGELOG.md
Normal file
32
CHANGELOG.md
Normal file
@@ -0,0 +1,32 @@
|
|||||||
|
# Blender Cloud changelog
|
||||||
|
|
||||||
|
## Version 1.6.3 (2017-03-21)
|
||||||
|
|
||||||
|
- Fixed bug where local project path wasn't shown for projects only set up for Flamenco
|
||||||
|
(and not Attract).
|
||||||
|
- Added this CHANGELOG.md file, which will contain user-relevant changes.
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.6.2 (2017-03-17)
|
||||||
|
|
||||||
|
- Flamenco: when opening non-existing file path, open parent instead - Sybren A. Stüvel
|
||||||
|
- Fix T50954: Improve Blender Cloud add-on project selector
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.6.1 (2017-03-07)
|
||||||
|
|
||||||
|
- Show error in GUI when Blender Cloud is unreachable
|
||||||
|
- Fixed sample count when using branched path tracing
|
||||||
|
|
||||||
|
|
||||||
|
## Version 1.6.0 (2017-02-14)
|
||||||
|
|
||||||
|
- Default to frame chunk size of 1 (instead of 10).
|
||||||
|
- Turn off "use overwrite" and "use placeholder" for Flamenco blend files.
|
||||||
|
- Fixed bugs when blendfile is outside the project directory
|
||||||
|
|
||||||
|
|
||||||
|
## Older versions
|
||||||
|
|
||||||
|
For the history of older versions, please refer to the
|
||||||
|
[Git history](https://developer.blender.org/diffusion/BCA/)
|
57
README-flamenco.md
Normal file
57
README-flamenco.md
Normal file
@@ -0,0 +1,57 @@
|
|||||||
|
# Flamenco
|
||||||
|
|
||||||
|
The Blender Cloud add-on has preliminary support for [Flamenco](https://flamenco.io/).
|
||||||
|
It requires a project on the [Blender Cloud](https://cloud.blender.org/) that is set up for
|
||||||
|
Flamenco, and it requires you to be logged in as a user with rights to use Flamenco.
|
||||||
|
|
||||||
|
|
||||||
|
## Quirks
|
||||||
|
|
||||||
|
Flamenco support is unpolished, so it has some quirks.
|
||||||
|
|
||||||
|
- Project selection happens through the Attract project selector. As a result, you can only
|
||||||
|
select Attract-enabled projects (even when they are not set up for Flamenco). Be careful
|
||||||
|
which project you select.
|
||||||
|
- The top level directory of the project is also set through the Attract properties.
|
||||||
|
- Defaults are heavily biased for our use in the Blender Institute.
|
||||||
|
- Settings that should be project-specific are not, i.e. are regular add-on preferences.
|
||||||
|
- Sending a project to Flamenco will check the "File Extensions" setting in the Output panel,
|
||||||
|
and save the blend file to the current filename.
|
||||||
|
|
||||||
|
## Render job file locations
|
||||||
|
|
||||||
|
Rendering via Flamenco roughly comprises of two steps:
|
||||||
|
|
||||||
|
1. Packing the file to render with its dependencies, and placing them in the "job file path".
|
||||||
|
2. Rendering, and placing the output files in the "job output path".
|
||||||
|
|
||||||
|
### Job file path
|
||||||
|
|
||||||
|
The "job file path" consists of the following path components:
|
||||||
|
|
||||||
|
1. The add-on preference "job file path", e.g. `/render/_flamenco/storage`
|
||||||
|
2. The current date and time, your Blender Cloud username, and the name of the current blend file.
|
||||||
|
3. The name of the current blend file.
|
||||||
|
|
||||||
|
For example:
|
||||||
|
|
||||||
|
`/render/_flamenco/storage/2017-01-18-104841.931387-sybren-03_02_A.layout/03_02_A.layout.blend`
|
||||||
|
|
||||||
|
### Job output path
|
||||||
|
|
||||||
|
The file path of output files consists of the following path components:
|
||||||
|
|
||||||
|
1. The add-on preference "job file path", e.g. `/render/agent327/frames`
|
||||||
|
2. The path of the current blend file, relative to the project directory. The first N components
|
||||||
|
of this path can be stripped; when N=1 it turns `scenes/03-searching/03_02_A-snooping/` into
|
||||||
|
`03-searching/03_02_A-snooping/`.
|
||||||
|
3. The name of the current blend file, without `.blend`.
|
||||||
|
4. The file name depends on the type of output:
|
||||||
|
- When rendering to image files: A 5-digit frame number with the required file extension.
|
||||||
|
- When rendering to a video file: The frame range with the required file extension.
|
||||||
|
|
||||||
|
For example:
|
||||||
|
|
||||||
|
`/render/agent327/frames/03-searching/03_02_A-snooping/03_02_A.layout/00441.exr`
|
||||||
|
|
||||||
|
`/render/agent327/frames/03-searching/03_02_A-snooping/03_02_A.layout/14-51,60-133.mkv`
|
@@ -22,6 +22,9 @@ cd $(dirname $(readlink -f $0))
|
|||||||
BCLOUD=$(ls ../dist/blender_cloud-*.addon.zip | tail -n 1)
|
BCLOUD=$(ls ../dist/blender_cloud-*.addon.zip | tail -n 1)
|
||||||
BID=$(ls ../../../blender-id-addon/dist/blender_id-*.addon.zip | tail -n 1)
|
BID=$(ls ../../../blender-id-addon/dist/blender_id-*.addon.zip | tail -n 1)
|
||||||
|
|
||||||
|
[ -z "$BCLOUD" ] && echo "BCloud addon not found" >&2
|
||||||
|
[ -z "$BID" ] && echo "B'ID addon not found" >&2
|
||||||
|
|
||||||
cp -va $BCLOUD $BID .
|
cp -va $BCLOUD $BID .
|
||||||
|
|
||||||
BUNDLE=$(basename $BCLOUD)
|
BUNDLE=$(basename $BCLOUD)
|
||||||
|
@@ -20,16 +20,15 @@
|
|||||||
|
|
||||||
bl_info = {
|
bl_info = {
|
||||||
'name': 'Blender Cloud',
|
'name': 'Blender Cloud',
|
||||||
'author': 'Sybren A. Stüvel and Francesco Siddi',
|
"author": "Sybren A. Stüvel, Francesco Siddi, Inês Almeida, Antony Riakiotakis",
|
||||||
'version': (1, 4, 1),
|
'version': (1, 6, 3),
|
||||||
'blender': (2, 77, 0),
|
'blender': (2, 77, 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.',
|
||||||
'wiki_url': 'http://wiki.blender.org/index.php/Extensions:2.6/Py/'
|
'wiki_url': 'https://wiki.blender.org/index.php/Extensions:2.6/Py/'
|
||||||
'Scripts/System/BlenderCloud',
|
'Scripts/System/BlenderCloud',
|
||||||
'category': 'System',
|
'category': 'System',
|
||||||
'support': 'OFFICIAL'
|
|
||||||
}
|
}
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
@@ -79,17 +78,23 @@ def register():
|
|||||||
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')
|
||||||
|
flamenco = reload_mod('flamenco')
|
||||||
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)
|
image_sharing, attract, flamenco)
|
||||||
|
|
||||||
async_loop.setup_asyncio_executor()
|
async_loop.setup_asyncio_executor()
|
||||||
async_loop.register()
|
async_loop.register()
|
||||||
|
|
||||||
|
flamenco.register()
|
||||||
texture_browser.register()
|
texture_browser.register()
|
||||||
blender.register()
|
blender.register()
|
||||||
settings_sync.register()
|
settings_sync.register()
|
||||||
image_sharing.register()
|
image_sharing.register()
|
||||||
|
attract.register()
|
||||||
|
|
||||||
|
blender.handle_project_update()
|
||||||
|
|
||||||
|
|
||||||
def _monkey_patch_requests():
|
def _monkey_patch_requests():
|
||||||
@@ -110,10 +115,13 @@ def _monkey_patch_requests():
|
|||||||
|
|
||||||
|
|
||||||
def unregister():
|
def unregister():
|
||||||
from . import blender, texture_browser, async_loop, settings_sync, image_sharing
|
from . import (blender, texture_browser, async_loop, settings_sync, image_sharing, attract,
|
||||||
|
flamenco)
|
||||||
|
|
||||||
image_sharing.unregister()
|
image_sharing.unregister()
|
||||||
|
attract.unregister()
|
||||||
settings_sync.unregister()
|
settings_sync.unregister()
|
||||||
blender.unregister()
|
blender.unregister()
|
||||||
texture_browser.unregister()
|
texture_browser.unregister()
|
||||||
async_loop.unregister()
|
async_loop.unregister()
|
||||||
|
flamenco.unregister()
|
||||||
|
@@ -39,9 +39,18 @@ def setup_asyncio_executor():
|
|||||||
calls that could be performed in parallel are queued, and thus we can
|
calls that could be performed in parallel are queued, and thus we can
|
||||||
reliably cancel them.
|
reliably cancel them.
|
||||||
"""
|
"""
|
||||||
|
import sys
|
||||||
|
|
||||||
executor = concurrent.futures.ThreadPoolExecutor()
|
executor = concurrent.futures.ThreadPoolExecutor()
|
||||||
loop = asyncio.get_event_loop()
|
|
||||||
|
if sys.platform == 'win32':
|
||||||
|
# On Windows, the default event loop is SelectorEventLoop, which does
|
||||||
|
# not support subprocesses. ProactorEventLoop should be used instead.
|
||||||
|
# Source: https://docs.python.org/3/library/asyncio-subprocess.html
|
||||||
|
loop = asyncio.ProactorEventLoop()
|
||||||
|
asyncio.set_event_loop(loop)
|
||||||
|
else:
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
loop.set_default_executor(executor)
|
loop.set_default_executor(executor)
|
||||||
# loop.set_debug(True)
|
# loop.set_debug(True)
|
||||||
|
|
||||||
@@ -178,12 +187,28 @@ class AsyncModalOperatorMixin:
|
|||||||
log = logging.getLogger('%s.AsyncModalOperatorMixin' % __name__)
|
log = logging.getLogger('%s.AsyncModalOperatorMixin' % __name__)
|
||||||
|
|
||||||
_state = 'INITIALIZING'
|
_state = 'INITIALIZING'
|
||||||
|
stop_upon_exception = False
|
||||||
|
|
||||||
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, context.window)
|
||||||
|
|
||||||
|
self.log.info('Starting')
|
||||||
|
self._new_async_task(self.async_execute(context))
|
||||||
|
|
||||||
return {'RUNNING_MODAL'}
|
return {'RUNNING_MODAL'}
|
||||||
|
|
||||||
|
async def async_execute(self, context):
|
||||||
|
"""Entry point of the asynchronous operator.
|
||||||
|
|
||||||
|
Implement in a subclass.
|
||||||
|
"""
|
||||||
|
return
|
||||||
|
|
||||||
|
def quit(self):
|
||||||
|
"""Signals the state machine to stop this operator from running."""
|
||||||
|
self._state = 'QUIT'
|
||||||
|
|
||||||
def execute(self, context):
|
def execute(self, context):
|
||||||
return self.invoke(context, None)
|
return self.invoke(context, None)
|
||||||
|
|
||||||
@@ -195,6 +220,11 @@ class AsyncModalOperatorMixin:
|
|||||||
if ex is not None:
|
if ex is not None:
|
||||||
self._state = 'EXCEPTION'
|
self._state = 'EXCEPTION'
|
||||||
self.log.error('Exception while running task: %s', ex)
|
self.log.error('Exception while running task: %s', ex)
|
||||||
|
if self.stop_upon_exception:
|
||||||
|
self.quit()
|
||||||
|
self._finish(context)
|
||||||
|
return {'FINISHED'}
|
||||||
|
|
||||||
return {'RUNNING_MODAL'}
|
return {'RUNNING_MODAL'}
|
||||||
|
|
||||||
if self._state == 'QUIT':
|
if self._state == 'QUIT':
|
||||||
|
1000
blender_cloud/attract/__init__.py
Normal file
1000
blender_cloud/attract/__init__.py
Normal file
File diff suppressed because it is too large
Load Diff
182
blender_cloud/attract/draw.py
Normal file
182
blender_cloud/attract/draw.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()
|
@@ -20,19 +20,21 @@
|
|||||||
|
|
||||||
Separated from __init__.py so that we can import & run from non-Blender environments.
|
Separated from __init__.py so that we can import & run from non-Blender environments.
|
||||||
"""
|
"""
|
||||||
|
import functools
|
||||||
import logging
|
import logging
|
||||||
import os.path
|
import os.path
|
||||||
|
|
||||||
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
|
from bpy.props import StringProperty, EnumProperty, PointerProperty, BoolProperty, IntProperty
|
||||||
import rna_prop_ui
|
import rna_prop_ui
|
||||||
|
|
||||||
from . import pillar
|
from . import pillar, async_loop, flamenco
|
||||||
|
from .utils import pyside_cache, redraw
|
||||||
|
|
||||||
PILLAR_SERVER_URL = 'https://cloudapi.blender.org/'
|
PILLAR_WEB_SERVER_URL = 'https://cloud.blender.org/'
|
||||||
# PILLAR_SERVER_URL = 'http://localhost:5000/'
|
# PILLAR_WEB_SERVER_URL = 'http://pillar-web:5001/'
|
||||||
|
PILLAR_SERVER_URL = '%sapi/' % PILLAR_WEB_SERVER_URL
|
||||||
|
|
||||||
ADDON_NAME = 'blender_cloud'
|
ADDON_NAME = 'blender_cloud'
|
||||||
log = logging.getLogger(__name__)
|
log = logging.getLogger(__name__)
|
||||||
@@ -40,11 +42,10 @@ log = logging.getLogger(__name__)
|
|||||||
icons = None
|
icons = None
|
||||||
|
|
||||||
|
|
||||||
def redraw(self, context):
|
@pyside_cache('version')
|
||||||
context.area.tag_redraw()
|
|
||||||
|
|
||||||
|
|
||||||
def blender_syncable_versions(self, context):
|
def blender_syncable_versions(self, context):
|
||||||
|
"""Returns the list of items used by SyncStatusProperties.version EnumProperty."""
|
||||||
|
|
||||||
bss = context.window_manager.blender_sync_status
|
bss = context.window_manager.blender_sync_status
|
||||||
versions = bss.available_blender_versions
|
versions = bss.available_blender_versions
|
||||||
if not versions:
|
if not versions:
|
||||||
@@ -104,6 +105,94 @@ class SyncStatusProperties(PropertyGroup):
|
|||||||
self['available_blender_versions'] = new_versions
|
self['available_blender_versions'] = new_versions
|
||||||
|
|
||||||
|
|
||||||
|
@pyside_cache('project')
|
||||||
|
def bcloud_available_projects(self, context):
|
||||||
|
"""Returns the list of items used by BlenderCloudProjectGroup.project EnumProperty."""
|
||||||
|
|
||||||
|
projs = preferences().project.available_projects
|
||||||
|
if not projs:
|
||||||
|
return [('', 'No projects available in your Blender Cloud', '')]
|
||||||
|
return [(p['_id'], p['name'], '') for p in projs]
|
||||||
|
|
||||||
|
|
||||||
|
@functools.lru_cache(1)
|
||||||
|
def project_extensions(project_id) -> set:
|
||||||
|
"""Returns the extensions the project is enabled for.
|
||||||
|
|
||||||
|
At the moment of writing these are 'attract' and 'flamenco'.
|
||||||
|
"""
|
||||||
|
|
||||||
|
log.debug('Finding extensions for project %s', project_id)
|
||||||
|
|
||||||
|
# We can't use our @property, since the preferences may be loaded from a
|
||||||
|
# preferences blend file, in which case it is not constructed from Python code.
|
||||||
|
available_projects = preferences().project.get('available_projects', [])
|
||||||
|
if not available_projects:
|
||||||
|
log.debug('No projects available.')
|
||||||
|
return set()
|
||||||
|
|
||||||
|
proj = next((p for p in available_projects
|
||||||
|
if p['_id'] == project_id), None)
|
||||||
|
if proj is None:
|
||||||
|
log.debug('Project %s not found in available projects.', project_id)
|
||||||
|
return set()
|
||||||
|
|
||||||
|
return set(proj.get('enabled_for', ()))
|
||||||
|
|
||||||
|
|
||||||
|
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.
|
||||||
|
"""
|
||||||
|
|
||||||
|
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):
|
||||||
|
status = EnumProperty(
|
||||||
|
items=[
|
||||||
|
('NONE', 'NONE', 'We have done nothing at all yet'),
|
||||||
|
('IDLE', 'IDLE', 'User requested something, which is done, and we are now idle'),
|
||||||
|
('FETCHING', 'FETCHING', 'Fetching available projects from Blender Cloud'),
|
||||||
|
],
|
||||||
|
name='status',
|
||||||
|
update=redraw)
|
||||||
|
|
||||||
|
project = EnumProperty(
|
||||||
|
items=bcloud_available_projects,
|
||||||
|
name='Cloud project',
|
||||||
|
description='Which Blender Cloud project to work with',
|
||||||
|
update=handle_project_update
|
||||||
|
)
|
||||||
|
|
||||||
|
# List of projects is stored in 'available_projects' ID property,
|
||||||
|
# because I don't know how to store a variable list of strings in a proper RNA property.
|
||||||
|
@property
|
||||||
|
def available_projects(self) -> list:
|
||||||
|
return self.get('available_projects', [])
|
||||||
|
|
||||||
|
@available_projects.setter
|
||||||
|
def available_projects(self, new_projects):
|
||||||
|
self['available_projects'] = new_projects
|
||||||
|
handle_project_update()
|
||||||
|
|
||||||
|
|
||||||
class BlenderCloudPreferences(AddonPreferences):
|
class BlenderCloudPreferences(AddonPreferences):
|
||||||
bl_idname = ADDON_NAME
|
bl_idname = ADDON_NAME
|
||||||
|
|
||||||
@@ -117,12 +206,54 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
)
|
)
|
||||||
|
|
||||||
local_texture_dir = StringProperty(
|
local_texture_dir = StringProperty(
|
||||||
name='Default Blender Cloud texture storage directory',
|
name='Default Blender Cloud Texture Storage Directory',
|
||||||
subtype='DIR_PATH',
|
subtype='DIR_PATH',
|
||||||
default='//textures')
|
default='//textures')
|
||||||
|
|
||||||
open_browser_after_share = BoolProperty(
|
open_browser_after_share = BoolProperty(
|
||||||
name='Open browser after sharing file',
|
name='Open Browser after Sharing File',
|
||||||
|
description='When enabled, Blender will open a webbrowser',
|
||||||
|
default=True
|
||||||
|
)
|
||||||
|
|
||||||
|
# TODO: store project-dependent properties with the project, so that people
|
||||||
|
# can switch projects and the Attract and Flamenco properties switch with it.
|
||||||
|
project = PointerProperty(type=BlenderCloudProjectGroup)
|
||||||
|
|
||||||
|
cloud_project_local_path = StringProperty(
|
||||||
|
name='Local Project Path',
|
||||||
|
description='Local path of your Attract project, used to search for blend files; '
|
||||||
|
'usually best to set to an absolute path',
|
||||||
|
subtype='DIR_PATH',
|
||||||
|
default='//../')
|
||||||
|
|
||||||
|
flamenco_manager = PointerProperty(type=flamenco.FlamencoManagerGroup)
|
||||||
|
# TODO: before making Flamenco public, change the defaults to something less Institute-specific.
|
||||||
|
# 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(
|
||||||
|
name='Job File Path',
|
||||||
|
description='Path where to store job files, should be accesible for Workers too',
|
||||||
|
subtype='DIR_PATH',
|
||||||
|
default='/render/_flamenco/storage')
|
||||||
|
|
||||||
|
# TODO: before making Flamenco public, change the defaults to something less Institute-specific.
|
||||||
|
flamenco_job_output_path = StringProperty(
|
||||||
|
name='Job Output Path',
|
||||||
|
description='Path where to store output files, should be accessible for Workers',
|
||||||
|
subtype='DIR_PATH',
|
||||||
|
default='/render/_flamenco/output')
|
||||||
|
flamenco_job_output_strip_components = IntProperty(
|
||||||
|
name='Job Output Path Strip Components',
|
||||||
|
description='The final output path comprises of the job output path, and the blend file '
|
||||||
|
'path relative to the project with this many path components stripped off '
|
||||||
|
'the front',
|
||||||
|
min=0,
|
||||||
|
default=0,
|
||||||
|
soft_max=4,
|
||||||
|
)
|
||||||
|
flamenco_open_browser_after_submit = BoolProperty(
|
||||||
|
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
|
||||||
)
|
)
|
||||||
@@ -140,8 +271,8 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
blender_id_profile = None
|
blender_id_profile = None
|
||||||
else:
|
else:
|
||||||
blender_id_profile = blender_id.get_active_profile()
|
blender_id_profile = blender_id.get_active_profile()
|
||||||
|
|
||||||
if blender_id is None:
|
if blender_id is None:
|
||||||
|
|
||||||
msg_icon = 'ERROR'
|
msg_icon = 'ERROR'
|
||||||
text = 'This add-on requires Blender ID'
|
text = 'This add-on requires Blender ID'
|
||||||
help_text = 'Make sure that the Blender ID add-on is installed and activated'
|
help_text = 'Make sure that the Blender ID add-on is installed and activated'
|
||||||
@@ -205,9 +336,20 @@ 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('Image Sharing on Blender Cloud', icon_value=icon('CLOUD'))
|
||||||
texture_box.enabled = msg_icon != 'ERROR'
|
|
||||||
share_box.prop(self, 'open_browser_after_share')
|
share_box.prop(self, 'open_browser_after_share')
|
||||||
|
|
||||||
|
# Project selector
|
||||||
|
project_box = layout.box()
|
||||||
|
project_box.enabled = self.project.status in {'NONE', 'IDLE'}
|
||||||
|
|
||||||
|
self.draw_project_selector(project_box, self.project)
|
||||||
|
extensions = project_extensions(self.project.project)
|
||||||
|
|
||||||
|
# Flamenco stuff
|
||||||
|
if 'flamenco' in extensions:
|
||||||
|
flamenco_box = project_box.column()
|
||||||
|
self.draw_flamenco_buttons(flamenco_box, self.flamenco_manager, context)
|
||||||
|
|
||||||
def draw_subscribe_button(self, layout):
|
def draw_subscribe_button(self, layout):
|
||||||
layout.operator('pillar.subscribe', icon='WORLD')
|
layout.operator('pillar.subscribe', icon='WORLD')
|
||||||
|
|
||||||
@@ -242,12 +384,96 @@ class BlenderCloudPreferences(AddonPreferences):
|
|||||||
else:
|
else:
|
||||||
row_pull.label('Cloud Sync is running.')
|
row_pull.label('Cloud Sync is running.')
|
||||||
|
|
||||||
|
def draw_project_selector(self, project_box, bcp: BlenderCloudProjectGroup):
|
||||||
|
project_row = project_box.row(align=True)
|
||||||
|
project_row.label('Project settings', icon_value=icon('CLOUD'))
|
||||||
|
|
||||||
|
row_buttons = project_row.row(align=True)
|
||||||
|
|
||||||
|
projects = bcp.available_projects
|
||||||
|
project = bcp.project
|
||||||
|
if bcp.status in {'NONE', 'IDLE'}:
|
||||||
|
if not projects or not project:
|
||||||
|
row_buttons.operator('pillar.projects',
|
||||||
|
text='Find project to load',
|
||||||
|
icon='FILE_REFRESH')
|
||||||
|
else:
|
||||||
|
row_buttons.prop(bcp, 'project')
|
||||||
|
row_buttons.operator('pillar.projects',
|
||||||
|
text='',
|
||||||
|
icon='FILE_REFRESH')
|
||||||
|
else:
|
||||||
|
row_buttons.label('Fetching available projects.')
|
||||||
|
|
||||||
|
enabled_for = project_extensions(project)
|
||||||
|
if not project:
|
||||||
|
return
|
||||||
|
|
||||||
|
if not enabled_for:
|
||||||
|
project_box.label('This project is not set up for Attract or Flamenco')
|
||||||
|
return
|
||||||
|
|
||||||
|
project_box.label('This project is set up for: %s' %
|
||||||
|
', '.join(sorted(enabled_for)))
|
||||||
|
|
||||||
|
# This is only needed when the project is set up for either Attract or Flamenco.
|
||||||
|
project_box.prop(self, 'cloud_project_local_path',
|
||||||
|
text='Local Cloud project path')
|
||||||
|
|
||||||
|
def draw_flamenco_buttons(self, flamenco_box, bcp: flamenco.FlamencoManagerGroup, context):
|
||||||
|
header_row = flamenco_box.row(align=True)
|
||||||
|
header_row.label('Flamenco:', icon_value=icon('CLOUD'))
|
||||||
|
|
||||||
|
manager_box = flamenco_box.row(align=True)
|
||||||
|
|
||||||
|
if bcp.status in {'NONE', 'IDLE'}:
|
||||||
|
if not bcp.available_managers or not bcp.manager:
|
||||||
|
manager_box.operator('flamenco.managers',
|
||||||
|
text='Find Flamenco Managers',
|
||||||
|
icon='FILE_REFRESH')
|
||||||
|
else:
|
||||||
|
manager_box.prop(bcp, 'manager', text='Manager')
|
||||||
|
manager_box.operator('flamenco.managers',
|
||||||
|
text='',
|
||||||
|
icon='FILE_REFRESH')
|
||||||
|
else:
|
||||||
|
manager_box.label('Fetching available managers.')
|
||||||
|
|
||||||
|
path_box = flamenco_box.row(align=True)
|
||||||
|
path_box.prop(self, 'flamenco_job_file_path')
|
||||||
|
props = path_box.operator('flamenco.explore_file_path', text='', icon='DISK_DRIVE')
|
||||||
|
props.path = self.flamenco_job_file_path
|
||||||
|
|
||||||
|
job_output_box = flamenco_box.column(align=True)
|
||||||
|
path_box = job_output_box.row(align=True)
|
||||||
|
path_box.prop(self, 'flamenco_job_output_path')
|
||||||
|
props = path_box.operator('flamenco.explore_file_path', text='', icon='DISK_DRIVE')
|
||||||
|
props.path = self.flamenco_job_output_path
|
||||||
|
|
||||||
|
job_output_box.prop(self, 'flamenco_job_output_strip_components',
|
||||||
|
text='Strip Components')
|
||||||
|
|
||||||
|
from .flamenco import render_output_path
|
||||||
|
|
||||||
|
path_box = job_output_box.row(align=True)
|
||||||
|
output_path = render_output_path(context)
|
||||||
|
if output_path:
|
||||||
|
path_box.label(str(output_path))
|
||||||
|
props = path_box.operator('flamenco.explore_file_path', text='', icon='DISK_DRIVE')
|
||||||
|
props.path = str(output_path.parent)
|
||||||
|
else:
|
||||||
|
path_box.label('Blend file is not in your project path, '
|
||||||
|
'unable to give output path example.')
|
||||||
|
|
||||||
|
flamenco_box.prop(self, 'flamenco_open_browser_after_submit')
|
||||||
|
|
||||||
|
|
||||||
class PillarCredentialsUpdate(pillar.PillarOperatorMixin,
|
class PillarCredentialsUpdate(pillar.PillarOperatorMixin,
|
||||||
Operator):
|
Operator):
|
||||||
"""Updates the Pillar URL and tests the new URL."""
|
"""Updates the Pillar URL and tests the new URL."""
|
||||||
bl_idname = 'pillar.credentials_update'
|
bl_idname = 'pillar.credentials_update'
|
||||||
bl_label = 'Update credentials'
|
bl_label = 'Update credentials'
|
||||||
|
bl_description = 'Resynchronises your Blender ID login with Blender Cloud'
|
||||||
|
|
||||||
log = logging.getLogger('bpy.ops.%s' % bl_idname)
|
log = logging.getLogger('bpy.ops.%s' % bl_idname)
|
||||||
|
|
||||||
@@ -294,6 +520,7 @@ class PILLAR_OT_subscribe(Operator):
|
|||||||
"""Opens a browser to subscribe the user to the Cloud."""
|
"""Opens a browser to subscribe the user to the Cloud."""
|
||||||
bl_idname = 'pillar.subscribe'
|
bl_idname = 'pillar.subscribe'
|
||||||
bl_label = 'Subscribe to the Cloud'
|
bl_label = 'Subscribe to the Cloud'
|
||||||
|
bl_description = "Opens a page in a web browser to subscribe to the Blender Cloud"
|
||||||
|
|
||||||
def execute(self, context):
|
def execute(self, context):
|
||||||
import webbrowser
|
import webbrowser
|
||||||
@@ -304,6 +531,75 @@ class PILLAR_OT_subscribe(Operator):
|
|||||||
return {'FINISHED'}
|
return {'FINISHED'}
|
||||||
|
|
||||||
|
|
||||||
|
class PILLAR_OT_projects(async_loop.AsyncModalOperatorMixin,
|
||||||
|
pillar.AuthenticatedPillarOperatorMixin,
|
||||||
|
Operator):
|
||||||
|
"""Fetches the projects available to the user"""
|
||||||
|
bl_idname = 'pillar.projects'
|
||||||
|
bl_label = 'Fetch available projects'
|
||||||
|
|
||||||
|
stop_upon_exception = True
|
||||||
|
_log = logging.getLogger('bpy.ops.%s' % bl_idname)
|
||||||
|
|
||||||
|
async def async_execute(self, context):
|
||||||
|
if not await self.authenticate(context):
|
||||||
|
return
|
||||||
|
|
||||||
|
import pillarsdk
|
||||||
|
from .pillar import pillar_call
|
||||||
|
|
||||||
|
self.log.info('Going to fetch projects for user %s', self.user_id)
|
||||||
|
|
||||||
|
preferences().project.status = 'FETCHING'
|
||||||
|
|
||||||
|
# Get all projects, except the home project.
|
||||||
|
projects_user = await pillar_call(
|
||||||
|
pillarsdk.Project.all,
|
||||||
|
{'where': {'user': self.user_id,
|
||||||
|
'category': {'$ne': 'home'}},
|
||||||
|
'sort': '-_created',
|
||||||
|
'projection': {'_id': True,
|
||||||
|
'name': True,
|
||||||
|
'extension_props': True},
|
||||||
|
})
|
||||||
|
|
||||||
|
projects_shared = await pillar_call(
|
||||||
|
pillarsdk.Project.all,
|
||||||
|
{'where': {'user': {'$ne': self.user_id},
|
||||||
|
'permissions.groups.group': {'$in': self.db_user.groups}},
|
||||||
|
'sort': '-_created',
|
||||||
|
'projection': {'_id': True,
|
||||||
|
'name': True,
|
||||||
|
'extension_props': True},
|
||||||
|
})
|
||||||
|
|
||||||
|
# We need to convert to regular dicts before storing in ID properties.
|
||||||
|
# Also don't store more properties than we need.
|
||||||
|
def reduce_properties(project_list):
|
||||||
|
for p in project_list:
|
||||||
|
p = p.to_dict()
|
||||||
|
extension_props = p.get('extension_props', {})
|
||||||
|
enabled_for = list(extension_props.keys())
|
||||||
|
|
||||||
|
self._log.debug('Project %r is enabled for %s', p['name'], enabled_for)
|
||||||
|
yield {
|
||||||
|
'_id': p['_id'],
|
||||||
|
'name': p['name'],
|
||||||
|
'enabled_for': enabled_for,
|
||||||
|
}
|
||||||
|
|
||||||
|
projects = list(reduce_properties(projects_user['_items'])) + \
|
||||||
|
list(reduce_properties(projects_shared['_items']))
|
||||||
|
|
||||||
|
preferences().project.available_projects = projects
|
||||||
|
|
||||||
|
self.quit()
|
||||||
|
|
||||||
|
def quit(self):
|
||||||
|
preferences().project.status = 'IDLE'
|
||||||
|
super().quit()
|
||||||
|
|
||||||
|
|
||||||
class PILLAR_PT_image_custom_properties(rna_prop_ui.PropertyPanel, bpy.types.Panel):
|
class PILLAR_PT_image_custom_properties(rna_prop_ui.PropertyPanel, bpy.types.Panel):
|
||||||
"""Shows custom properties in the image editor."""
|
"""Shows custom properties in the image editor."""
|
||||||
|
|
||||||
@@ -353,10 +649,12 @@ def icon(icon_name: str) -> int:
|
|||||||
|
|
||||||
|
|
||||||
def register():
|
def register():
|
||||||
|
bpy.utils.register_class(BlenderCloudProjectGroup)
|
||||||
bpy.utils.register_class(BlenderCloudPreferences)
|
bpy.utils.register_class(BlenderCloudPreferences)
|
||||||
bpy.utils.register_class(PillarCredentialsUpdate)
|
bpy.utils.register_class(PillarCredentialsUpdate)
|
||||||
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_PT_image_custom_properties)
|
bpy.utils.register_class(PILLAR_PT_image_custom_properties)
|
||||||
|
|
||||||
addon_prefs = preferences()
|
addon_prefs = preferences()
|
||||||
@@ -385,10 +683,12 @@ def register():
|
|||||||
def unregister():
|
def unregister():
|
||||||
unload_custom_icons()
|
unload_custom_icons()
|
||||||
|
|
||||||
|
bpy.utils.unregister_class(BlenderCloudProjectGroup)
|
||||||
bpy.utils.unregister_class(PillarCredentialsUpdate)
|
bpy.utils.unregister_class(PillarCredentialsUpdate)
|
||||||
bpy.utils.unregister_class(BlenderCloudPreferences)
|
bpy.utils.unregister_class(BlenderCloudPreferences)
|
||||||
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_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
|
||||||
|
692
blender_cloud/flamenco/__init__.py
Normal file
692
blender_cloud/flamenco/__init__.py
Normal file
@@ -0,0 +1,692 @@
|
|||||||
|
# ##### 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 #####
|
||||||
|
|
||||||
|
"""Flamenco interface.
|
||||||
|
|
||||||
|
The preferences are managed blender.py, the rest of the Flamenco-specific stuff is here.
|
||||||
|
"""
|
||||||
|
import functools
|
||||||
|
import logging
|
||||||
|
from pathlib import Path, PurePath
|
||||||
|
import typing
|
||||||
|
|
||||||
|
import bpy
|
||||||
|
from bpy.types import AddonPreferences, Operator, WindowManager, Scene, PropertyGroup
|
||||||
|
from bpy.props import StringProperty, EnumProperty, PointerProperty, BoolProperty, IntProperty
|
||||||
|
|
||||||
|
from .. import async_loop, pillar
|
||||||
|
from ..utils import pyside_cache, redraw
|
||||||
|
|
||||||
|
log = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
# Global flag used to determine whether panels etc. can be drawn.
|
||||||
|
flamenco_is_active = False
|
||||||
|
|
||||||
|
|
||||||
|
@pyside_cache('manager')
|
||||||
|
def available_managers(self, context):
|
||||||
|
"""Returns the list of items used by a manager-selector EnumProperty."""
|
||||||
|
|
||||||
|
from ..blender import preferences
|
||||||
|
|
||||||
|
mngrs = preferences().flamenco_manager.available_managers
|
||||||
|
if not mngrs:
|
||||||
|
return [('', 'No managers available in your Blender Cloud', '')]
|
||||||
|
return [(p['_id'], p['name'], '') for p in mngrs]
|
||||||
|
|
||||||
|
|
||||||
|
class FlamencoManagerGroup(PropertyGroup):
|
||||||
|
manager = EnumProperty(
|
||||||
|
items=available_managers,
|
||||||
|
name='Flamenco Manager',
|
||||||
|
description='Which Flamenco Manager to use for jobs')
|
||||||
|
|
||||||
|
status = EnumProperty(
|
||||||
|
items=[
|
||||||
|
('NONE', 'NONE', 'We have done nothing at all yet'),
|
||||||
|
('IDLE', 'IDLE', 'User requested something, which is done, and we are now idle'),
|
||||||
|
('FETCHING', 'FETCHING', 'Fetching available Flamenco managers from Blender Cloud'),
|
||||||
|
],
|
||||||
|
name='status',
|
||||||
|
update=redraw)
|
||||||
|
|
||||||
|
# List of managers is stored in 'available_managers' ID property,
|
||||||
|
# because I don't know how to store a variable list of strings in a proper RNA property.
|
||||||
|
@property
|
||||||
|
def available_managers(self) -> list:
|
||||||
|
return self.get('available_managers', [])
|
||||||
|
|
||||||
|
@available_managers.setter
|
||||||
|
def available_managers(self, new_managers):
|
||||||
|
self['available_managers'] = new_managers
|
||||||
|
|
||||||
|
|
||||||
|
class FlamencoPollMixin:
|
||||||
|
@classmethod
|
||||||
|
def poll(cls, context):
|
||||||
|
return flamenco_is_active
|
||||||
|
|
||||||
|
|
||||||
|
class FLAMENCO_OT_fmanagers(async_loop.AsyncModalOperatorMixin,
|
||||||
|
pillar.AuthenticatedPillarOperatorMixin,
|
||||||
|
FlamencoPollMixin,
|
||||||
|
Operator):
|
||||||
|
"""Fetches the Flamenco Managers available to the user"""
|
||||||
|
bl_idname = 'flamenco.managers'
|
||||||
|
bl_label = 'Fetch available Flamenco Managers'
|
||||||
|
|
||||||
|
stop_upon_exception = True
|
||||||
|
log = logging.getLogger('%s.FLAMENCO_OT_fmanagers' % __name__)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def mypref(self) -> FlamencoManagerGroup:
|
||||||
|
from ..blender import preferences
|
||||||
|
|
||||||
|
return preferences().flamenco_manager
|
||||||
|
|
||||||
|
async def async_execute(self, context):
|
||||||
|
if not await self.authenticate(context):
|
||||||
|
return
|
||||||
|
|
||||||
|
from .sdk import Manager
|
||||||
|
from ..pillar import pillar_call
|
||||||
|
|
||||||
|
self.log.info('Going to fetch managers for user %s', self.user_id)
|
||||||
|
|
||||||
|
self.mypref.status = 'FETCHING'
|
||||||
|
managers = await pillar_call(Manager.all)
|
||||||
|
|
||||||
|
# We need to convert to regular dicts before storing in ID properties.
|
||||||
|
# Also don't store more properties than we need.
|
||||||
|
as_list = [{'_id': p['_id'], 'name': p['name']} for p in managers['_items']]
|
||||||
|
|
||||||
|
self.mypref.available_managers = as_list
|
||||||
|
self.quit()
|
||||||
|
|
||||||
|
def quit(self):
|
||||||
|
self.mypref.status = 'IDLE'
|
||||||
|
super().quit()
|
||||||
|
|
||||||
|
|
||||||
|
class FLAMENCO_OT_render(async_loop.AsyncModalOperatorMixin,
|
||||||
|
pillar.AuthenticatedPillarOperatorMixin,
|
||||||
|
FlamencoPollMixin,
|
||||||
|
Operator):
|
||||||
|
"""Performs a Blender render on Flamenco."""
|
||||||
|
bl_idname = 'flamenco.render'
|
||||||
|
bl_label = 'Render on Flamenco'
|
||||||
|
bl_description = __doc__.rstrip('.')
|
||||||
|
|
||||||
|
stop_upon_exception = True
|
||||||
|
log = logging.getLogger('%s.FLAMENCO_OT_render' % __name__)
|
||||||
|
|
||||||
|
async def async_execute(self, context):
|
||||||
|
if not await self.authenticate(context):
|
||||||
|
return
|
||||||
|
|
||||||
|
from ..blender import preferences
|
||||||
|
|
||||||
|
scene = context.scene
|
||||||
|
|
||||||
|
# Save to a different file, specifically for Flamenco.
|
||||||
|
context.window_manager.flamenco_status = 'PACKING'
|
||||||
|
filepath = await self._save_blendfile(context)
|
||||||
|
|
||||||
|
# Determine where the render output will be stored.
|
||||||
|
render_output = render_output_path(context, filepath)
|
||||||
|
if render_output is None:
|
||||||
|
self.report({'ERROR'}, 'Current file is outside of project path.')
|
||||||
|
self.quit()
|
||||||
|
return
|
||||||
|
self.log.info('Will output render files to %s', render_output)
|
||||||
|
|
||||||
|
# BAM-pack the files to the destination directory.
|
||||||
|
outfile, missing_sources = await self.bam_pack(filepath)
|
||||||
|
if not outfile:
|
||||||
|
return
|
||||||
|
|
||||||
|
# Create the job at Flamenco Server.
|
||||||
|
prefs = preferences()
|
||||||
|
|
||||||
|
context.window_manager.flamenco_status = 'COMMUNICATING'
|
||||||
|
settings = {'blender_cmd': '{blender}',
|
||||||
|
'chunk_size': scene.flamenco_render_fchunk_size,
|
||||||
|
'filepath': str(outfile),
|
||||||
|
'frames': scene.flamenco_render_frame_range,
|
||||||
|
'render_output': str(render_output),
|
||||||
|
}
|
||||||
|
|
||||||
|
# Add extra settings specific to the job type
|
||||||
|
if scene.flamenco_render_job_type == 'blender-render-progressive':
|
||||||
|
if scene.cycles.progressive == 'BRANCHED_PATH':
|
||||||
|
samples = scene.cycles.aa_samples
|
||||||
|
else:
|
||||||
|
samples = scene.cycles.samples
|
||||||
|
|
||||||
|
if scene.cycles.use_square_samples:
|
||||||
|
samples **= 2
|
||||||
|
|
||||||
|
settings['cycles_num_chunks'] = scene.flamenco_render_schunk_count
|
||||||
|
settings['cycles_sample_count'] = samples
|
||||||
|
settings['format'] = 'EXR'
|
||||||
|
|
||||||
|
try:
|
||||||
|
job_info = await create_job(self.user_id,
|
||||||
|
prefs.project.project,
|
||||||
|
prefs.flamenco_manager.manager,
|
||||||
|
scene.flamenco_render_job_type,
|
||||||
|
settings,
|
||||||
|
'Render %s' % filepath.name,
|
||||||
|
priority=scene.flamenco_render_job_priority)
|
||||||
|
except Exception as ex:
|
||||||
|
self.report({'ERROR'}, 'Error creating Flamenco job: %s' % ex)
|
||||||
|
self.quit()
|
||||||
|
return
|
||||||
|
|
||||||
|
# Store the job ID in a file in the output dir.
|
||||||
|
with open(str(outfile.parent / 'jobinfo.json'), 'w', encoding='utf8') as outfile:
|
||||||
|
import json
|
||||||
|
|
||||||
|
job_info['missing_files'] = [str(mf) for mf in missing_sources]
|
||||||
|
json.dump(job_info, outfile, sort_keys=True, indent=4)
|
||||||
|
|
||||||
|
# We can now remove the local copy we made with bpy.ops.wm.save_as_mainfile().
|
||||||
|
# Strictly speaking we can already remove it after the BAM-pack, but it may come in
|
||||||
|
# handy in case of failures.
|
||||||
|
try:
|
||||||
|
self.log.info('Removing temporary file %s', filepath)
|
||||||
|
filepath.unlink()
|
||||||
|
except Exception as ex:
|
||||||
|
self.report({'ERROR'}, 'Unable to remove file: %s' % ex)
|
||||||
|
self.quit()
|
||||||
|
return
|
||||||
|
|
||||||
|
if prefs.flamenco_open_browser_after_submit:
|
||||||
|
import webbrowser
|
||||||
|
from urllib.parse import urljoin
|
||||||
|
from ..blender import PILLAR_WEB_SERVER_URL
|
||||||
|
|
||||||
|
url = urljoin(PILLAR_WEB_SERVER_URL, '/flamenco/jobs/%s/redir' % job_info['_id'])
|
||||||
|
webbrowser.open_new_tab(url)
|
||||||
|
|
||||||
|
# Do a final report.
|
||||||
|
if missing_sources:
|
||||||
|
names = (ms.name for ms in missing_sources)
|
||||||
|
self.report({'WARNING'}, 'Flamenco job created with missing files: %s' %
|
||||||
|
'; '.join(names))
|
||||||
|
else:
|
||||||
|
self.report({'INFO'}, 'Flamenco job created.')
|
||||||
|
|
||||||
|
self.quit()
|
||||||
|
|
||||||
|
async def _save_blendfile(self, context):
|
||||||
|
"""Save to a different file, specifically for Flamenco.
|
||||||
|
|
||||||
|
We shouldn't overwrite the artist's file.
|
||||||
|
We can compress, since this file won't be managed by SVN and doesn't need diffability.
|
||||||
|
"""
|
||||||
|
|
||||||
|
render = context.scene.render
|
||||||
|
|
||||||
|
# Remember settings we need to restore after saving.
|
||||||
|
old_use_file_extension = render.use_file_extension
|
||||||
|
old_use_overwrite = render.use_overwrite
|
||||||
|
old_use_placeholder = render.use_placeholder
|
||||||
|
|
||||||
|
try:
|
||||||
|
|
||||||
|
# The file extension should be determined by the render settings, not necessarily
|
||||||
|
# by the setttings in the output panel.
|
||||||
|
render.use_file_extension = True
|
||||||
|
|
||||||
|
# Rescheduling should not overwrite existing frames.
|
||||||
|
render.use_overwrite = False
|
||||||
|
render.use_placeholder = False
|
||||||
|
|
||||||
|
filepath = Path(context.blend_data.filepath).with_suffix('.flamenco.blend')
|
||||||
|
self.log.info('Saving copy to temporary file %s', filepath)
|
||||||
|
bpy.ops.wm.save_as_mainfile(filepath=str(filepath),
|
||||||
|
compress=True,
|
||||||
|
copy=True)
|
||||||
|
finally:
|
||||||
|
# Restore the settings we changed, even after an exception.
|
||||||
|
render.use_file_extension = old_use_file_extension
|
||||||
|
render.use_overwrite = old_use_overwrite
|
||||||
|
render.use_placeholder = old_use_placeholder
|
||||||
|
|
||||||
|
return filepath
|
||||||
|
|
||||||
|
def quit(self):
|
||||||
|
super().quit()
|
||||||
|
bpy.context.window_manager.flamenco_status = 'IDLE'
|
||||||
|
|
||||||
|
async def bam_pack(self, filepath: Path) -> (typing.Optional[Path], typing.List[Path]):
|
||||||
|
"""BAM-packs the blendfile to the destination directory.
|
||||||
|
|
||||||
|
Returns the path of the destination blend file.
|
||||||
|
|
||||||
|
:param filepath: the blend file to pack (i.e. the current blend file)
|
||||||
|
:returns: the destination blend file, or None if there were errors BAM-packing,
|
||||||
|
and a list of missing paths.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from datetime import datetime
|
||||||
|
from ..blender import preferences
|
||||||
|
from . import bam_interface
|
||||||
|
|
||||||
|
prefs = preferences()
|
||||||
|
|
||||||
|
# Create a unique directory that is still more or less identifyable.
|
||||||
|
# This should work better than a random ID.
|
||||||
|
# BAM doesn't like output directories that end in '.blend'.
|
||||||
|
unique_dir = '%s-%s-%s' % (datetime.now().isoformat('-').replace(':', ''),
|
||||||
|
self.db_user['username'],
|
||||||
|
filepath.stem)
|
||||||
|
outdir = Path(prefs.flamenco_job_file_path) / unique_dir
|
||||||
|
outfile = outdir / filepath.name
|
||||||
|
|
||||||
|
try:
|
||||||
|
outdir.mkdir(parents=True)
|
||||||
|
except Exception as ex:
|
||||||
|
self.log.exception('Unable to create output path %s', outdir)
|
||||||
|
self.report({'ERROR'}, 'Unable to create output path: %s' % ex)
|
||||||
|
self.quit()
|
||||||
|
return None, []
|
||||||
|
|
||||||
|
try:
|
||||||
|
missing_sources = await bam_interface.bam_copy(filepath, outfile)
|
||||||
|
except bam_interface.CommandExecutionError as ex:
|
||||||
|
self.log.exception('Unable to execute BAM pack')
|
||||||
|
self.report({'ERROR'}, 'Unable to execute BAM pack: %s' % ex)
|
||||||
|
self.quit()
|
||||||
|
return None, []
|
||||||
|
|
||||||
|
return outfile, missing_sources
|
||||||
|
|
||||||
|
|
||||||
|
class FLAMENCO_OT_scene_to_frame_range(FlamencoPollMixin, Operator):
|
||||||
|
"""Sets the scene frame range as the Flamenco render frame range."""
|
||||||
|
bl_idname = 'flamenco.scene_to_frame_range'
|
||||||
|
bl_label = 'Sets the scene frame range as the Flamenco render frame range'
|
||||||
|
bl_description = __doc__.rstrip('.')
|
||||||
|
|
||||||
|
def execute(self, context):
|
||||||
|
s = context.scene
|
||||||
|
s.flamenco_render_frame_range = '%i-%i' % (s.frame_start, s.frame_end)
|
||||||
|
return {'FINISHED'}
|
||||||
|
|
||||||
|
|
||||||
|
class FLAMENCO_OT_copy_files(Operator,
|
||||||
|
FlamencoPollMixin,
|
||||||
|
async_loop.AsyncModalOperatorMixin):
|
||||||
|
"""Uses BAM to copy the current blendfile + dependencies to the target directory."""
|
||||||
|
bl_idname = 'flamenco.copy_files'
|
||||||
|
bl_label = 'Copy files to target'
|
||||||
|
bl_description = __doc__.rstrip('.')
|
||||||
|
|
||||||
|
stop_upon_exception = True
|
||||||
|
|
||||||
|
async def async_execute(self, context):
|
||||||
|
from pathlib import Path
|
||||||
|
from . import bam_interface
|
||||||
|
from ..blender import preferences
|
||||||
|
|
||||||
|
context.window_manager.flamenco_status = 'PACKING'
|
||||||
|
|
||||||
|
missing_sources = await bam_interface.bam_copy(
|
||||||
|
Path(context.blend_data.filepath),
|
||||||
|
Path(preferences().flamenco_job_file_path),
|
||||||
|
)
|
||||||
|
|
||||||
|
if missing_sources:
|
||||||
|
names = (ms.name for ms in missing_sources)
|
||||||
|
self.report({'ERROR'}, 'Missing source files: %s' % '; '.join(names))
|
||||||
|
|
||||||
|
self.quit()
|
||||||
|
|
||||||
|
def quit(self):
|
||||||
|
super().quit()
|
||||||
|
bpy.context.window_manager.flamenco_status = 'IDLE'
|
||||||
|
|
||||||
|
|
||||||
|
class FLAMENCO_OT_explore_file_path(FlamencoPollMixin,
|
||||||
|
Operator):
|
||||||
|
"""Opens the Flamenco job storage path in a file explorer.
|
||||||
|
|
||||||
|
If the path cannot be found, this operator tries to open its parent.
|
||||||
|
"""
|
||||||
|
|
||||||
|
bl_idname = 'flamenco.explore_file_path'
|
||||||
|
bl_label = 'Open in file explorer'
|
||||||
|
bl_description = __doc__.rstrip('.')
|
||||||
|
|
||||||
|
path = StringProperty(name='Path', description='Path to explore', subtype='DIR_PATH')
|
||||||
|
|
||||||
|
def execute(self, context):
|
||||||
|
import platform
|
||||||
|
import pathlib
|
||||||
|
|
||||||
|
# Possibly open a parent of the path
|
||||||
|
to_open = pathlib.Path(self.path)
|
||||||
|
while to_open.parent != to_open: # while we're not at the root
|
||||||
|
if to_open.exists():
|
||||||
|
break
|
||||||
|
to_open = to_open.parent
|
||||||
|
else:
|
||||||
|
self.report({'ERROR'}, 'Unable to open %s or any of its parents.' % self.path)
|
||||||
|
return {'CANCELLED'}
|
||||||
|
to_open = str(to_open)
|
||||||
|
|
||||||
|
if platform.system() == "Windows":
|
||||||
|
import os
|
||||||
|
os.startfile(to_open)
|
||||||
|
|
||||||
|
elif platform.system() == "Darwin":
|
||||||
|
import subprocess
|
||||||
|
subprocess.Popen(["open", to_open])
|
||||||
|
|
||||||
|
else:
|
||||||
|
import subprocess
|
||||||
|
subprocess.Popen(["xdg-open", to_open])
|
||||||
|
|
||||||
|
return {'FINISHED'}
|
||||||
|
|
||||||
|
|
||||||
|
async def create_job(user_id: str,
|
||||||
|
project_id: str,
|
||||||
|
manager_id: str,
|
||||||
|
job_type: str,
|
||||||
|
job_settings: dict,
|
||||||
|
job_name: str = None,
|
||||||
|
*,
|
||||||
|
priority: int = 50,
|
||||||
|
job_description: str = None) -> dict:
|
||||||
|
"""Creates a render job at Flamenco Server, returning the job object as dictionary."""
|
||||||
|
|
||||||
|
import json
|
||||||
|
from .sdk import Job
|
||||||
|
from ..pillar import pillar_call
|
||||||
|
|
||||||
|
job_attrs = {
|
||||||
|
'status': 'queued',
|
||||||
|
'priority': priority,
|
||||||
|
'name': job_name,
|
||||||
|
'settings': job_settings,
|
||||||
|
'job_type': job_type,
|
||||||
|
'user': user_id,
|
||||||
|
'manager': manager_id,
|
||||||
|
'project': project_id,
|
||||||
|
}
|
||||||
|
if job_description:
|
||||||
|
job_attrs['description'] = job_description
|
||||||
|
|
||||||
|
log.info('Going to create Flamenco job:\n%s',
|
||||||
|
json.dumps(job_attrs, indent=4, sort_keys=True))
|
||||||
|
|
||||||
|
job = Job(job_attrs)
|
||||||
|
await pillar_call(job.create)
|
||||||
|
|
||||||
|
log.info('Job created succesfully: %s', job._id)
|
||||||
|
return job.to_dict()
|
||||||
|
|
||||||
|
|
||||||
|
def is_image_type(render_output_type: str) -> bool:
|
||||||
|
"""Determines whether the render output type is an image (True) or video (False)."""
|
||||||
|
|
||||||
|
# This list is taken from rna_scene.c:273, rna_enum_image_type_items.
|
||||||
|
video_types = {'AVI_JPEG', 'AVI_RAW', 'FRAMESERVER', 'FFMPEG', 'QUICKTIME'}
|
||||||
|
return render_output_type not in video_types
|
||||||
|
|
||||||
|
|
||||||
|
@functools.lru_cache(1)
|
||||||
|
def _render_output_path(
|
||||||
|
local_project_path: str,
|
||||||
|
blend_filepath: Path,
|
||||||
|
flamenco_job_output_strip_components: int,
|
||||||
|
flamenco_job_output_path: str,
|
||||||
|
render_image_format: str,
|
||||||
|
flamenco_render_frame_range: str,
|
||||||
|
) -> typing.Optional[PurePath]:
|
||||||
|
"""Cached version of render_output_path()
|
||||||
|
|
||||||
|
This ensures that redraws of the Flamenco Render and Add-on preferences panels
|
||||||
|
is fast.
|
||||||
|
"""
|
||||||
|
|
||||||
|
try:
|
||||||
|
project_path = Path(bpy.path.abspath(local_project_path)).resolve()
|
||||||
|
except FileNotFoundError:
|
||||||
|
# Path.resolve() will raise a FileNotFoundError if the project path doesn't exist.
|
||||||
|
return None
|
||||||
|
|
||||||
|
try:
|
||||||
|
proj_rel = blend_filepath.parent.relative_to(project_path)
|
||||||
|
except ValueError:
|
||||||
|
return None
|
||||||
|
|
||||||
|
rel_parts = proj_rel.parts[flamenco_job_output_strip_components:]
|
||||||
|
output_top = Path(flamenco_job_output_path)
|
||||||
|
|
||||||
|
# Strip off '.flamenco' too; we use 'xxx.flamenco.blend' as job file, but
|
||||||
|
# don't want to have all the output paths ending in '.flamenco'.
|
||||||
|
stem = blend_filepath.stem
|
||||||
|
if stem.endswith('.flamenco'):
|
||||||
|
stem = stem[:-9]
|
||||||
|
|
||||||
|
dir_components = output_top.joinpath(*rel_parts) / stem
|
||||||
|
|
||||||
|
# Blender will have to append the file extensions by itself.
|
||||||
|
if is_image_type(render_image_format):
|
||||||
|
return dir_components / '######'
|
||||||
|
return dir_components / flamenco_render_frame_range
|
||||||
|
|
||||||
|
|
||||||
|
def render_output_path(context, filepath: Path = None) -> typing.Optional[PurePath]:
|
||||||
|
"""Returns the render output path to be sent to Flamenco.
|
||||||
|
|
||||||
|
:param context: the Blender context (used to find Flamenco preferences etc.)
|
||||||
|
:param filepath: the Path of the blend file to render, or None for the current file.
|
||||||
|
|
||||||
|
Returns None when the current blend file is outside the project path.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from ..blender import preferences
|
||||||
|
|
||||||
|
scene = context.scene
|
||||||
|
prefs = preferences()
|
||||||
|
|
||||||
|
if filepath is None:
|
||||||
|
filepath = Path(context.blend_data.filepath)
|
||||||
|
|
||||||
|
return _render_output_path(
|
||||||
|
prefs.cloud_project_local_path,
|
||||||
|
filepath,
|
||||||
|
prefs.flamenco_job_output_strip_components,
|
||||||
|
prefs.flamenco_job_output_path,
|
||||||
|
scene.render.image_settings.file_format,
|
||||||
|
scene.flamenco_render_frame_range,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class FLAMENCO_PT_render(bpy.types.Panel, FlamencoPollMixin):
|
||||||
|
bl_label = "Flamenco Render"
|
||||||
|
bl_space_type = 'PROPERTIES'
|
||||||
|
bl_region_type = 'WINDOW'
|
||||||
|
bl_context = "render"
|
||||||
|
bl_options = {'DEFAULT_CLOSED'}
|
||||||
|
|
||||||
|
def draw(self, context):
|
||||||
|
layout = self.layout
|
||||||
|
|
||||||
|
from ..blender import preferences
|
||||||
|
|
||||||
|
prefs = preferences()
|
||||||
|
|
||||||
|
labeled_row = layout.split(0.25, align=True)
|
||||||
|
labeled_row.label('Job Type:')
|
||||||
|
labeled_row.prop(context.scene, 'flamenco_render_job_type', text='')
|
||||||
|
|
||||||
|
labeled_row = layout.split(0.25, align=True)
|
||||||
|
labeled_row.label('Frame Range:')
|
||||||
|
prop_btn_row = labeled_row.row(align=True)
|
||||||
|
prop_btn_row.prop(context.scene, 'flamenco_render_frame_range', text='')
|
||||||
|
prop_btn_row.operator('flamenco.scene_to_frame_range', text='', icon='ARROW_LEFTRIGHT')
|
||||||
|
|
||||||
|
layout.prop(context.scene, 'flamenco_render_job_priority')
|
||||||
|
layout.prop(context.scene, 'flamenco_render_fchunk_size')
|
||||||
|
|
||||||
|
if getattr(context.scene, 'flamenco_render_job_type', None) == 'blender-render-progressive':
|
||||||
|
layout.prop(context.scene, 'flamenco_render_schunk_count')
|
||||||
|
|
||||||
|
readonly_stuff = layout.column(align=True)
|
||||||
|
labeled_row = readonly_stuff.split(0.25, align=True)
|
||||||
|
labeled_row.label('Storage:')
|
||||||
|
prop_btn_row = labeled_row.row(align=True)
|
||||||
|
prop_btn_row.label(prefs.flamenco_job_file_path)
|
||||||
|
props = prop_btn_row.operator(FLAMENCO_OT_explore_file_path.bl_idname,
|
||||||
|
text='', icon='DISK_DRIVE')
|
||||||
|
props.path = prefs.flamenco_job_file_path
|
||||||
|
|
||||||
|
labeled_row = readonly_stuff.split(0.25, align=True)
|
||||||
|
labeled_row.label('Output:')
|
||||||
|
prop_btn_row = labeled_row.row(align=True)
|
||||||
|
render_output = render_output_path(context)
|
||||||
|
|
||||||
|
if render_output is None:
|
||||||
|
prop_btn_row.label('Unable to render with Flamenco, outside of project directory.')
|
||||||
|
else:
|
||||||
|
prop_btn_row.label(str(render_output))
|
||||||
|
props = prop_btn_row.operator(FLAMENCO_OT_explore_file_path.bl_idname,
|
||||||
|
text='', icon='DISK_DRIVE')
|
||||||
|
props.path = str(render_output.parent)
|
||||||
|
|
||||||
|
flamenco_status = context.window_manager.flamenco_status
|
||||||
|
if flamenco_status == 'IDLE':
|
||||||
|
layout.operator(FLAMENCO_OT_render.bl_idname,
|
||||||
|
text='Render on Flamenco',
|
||||||
|
icon='RENDER_ANIMATION')
|
||||||
|
elif flamenco_status == 'PACKING':
|
||||||
|
layout.label('Flamenco is packing your file + dependencies')
|
||||||
|
elif flamenco_status == 'COMMUNICATING':
|
||||||
|
layout.label('Communicating with Flamenco Server')
|
||||||
|
else:
|
||||||
|
layout.label('Unknown Flamenco status %s' % flamenco_status)
|
||||||
|
|
||||||
|
|
||||||
|
def activate():
|
||||||
|
"""Activates draw callbacks, menu items etc. for Flamenco."""
|
||||||
|
|
||||||
|
global flamenco_is_active
|
||||||
|
log.info('Activating Flamenco')
|
||||||
|
flamenco_is_active = True
|
||||||
|
|
||||||
|
|
||||||
|
def deactivate():
|
||||||
|
"""Deactivates draw callbacks, menu items etc. for Flamenco."""
|
||||||
|
|
||||||
|
global flamenco_is_active
|
||||||
|
log.info('Deactivating Flamenco')
|
||||||
|
flamenco_is_active = False
|
||||||
|
|
||||||
|
|
||||||
|
def register():
|
||||||
|
from ..utils import redraw
|
||||||
|
|
||||||
|
bpy.utils.register_class(FlamencoManagerGroup)
|
||||||
|
bpy.utils.register_class(FLAMENCO_OT_fmanagers)
|
||||||
|
bpy.utils.register_class(FLAMENCO_OT_render)
|
||||||
|
bpy.utils.register_class(FLAMENCO_OT_scene_to_frame_range)
|
||||||
|
bpy.utils.register_class(FLAMENCO_OT_copy_files)
|
||||||
|
bpy.utils.register_class(FLAMENCO_OT_explore_file_path)
|
||||||
|
bpy.utils.register_class(FLAMENCO_PT_render)
|
||||||
|
|
||||||
|
scene = bpy.types.Scene
|
||||||
|
scene.flamenco_render_fchunk_size = IntProperty(
|
||||||
|
name='Frame Chunk Size',
|
||||||
|
description='Maximum number of frames to render per task',
|
||||||
|
min=1,
|
||||||
|
default=1,
|
||||||
|
)
|
||||||
|
scene.flamenco_render_schunk_count = IntProperty(
|
||||||
|
name='Number of Sample Chunks',
|
||||||
|
description='Number of Cycles samples chunks to use per frame',
|
||||||
|
min=2,
|
||||||
|
default=3,
|
||||||
|
soft_max=10,
|
||||||
|
)
|
||||||
|
scene.flamenco_render_frame_range = StringProperty(
|
||||||
|
name='Frame Range',
|
||||||
|
description='Frames to render, in "printer range" notation'
|
||||||
|
)
|
||||||
|
scene.flamenco_render_job_type = EnumProperty(
|
||||||
|
name='Job Type',
|
||||||
|
items=[
|
||||||
|
('blender-render', 'Simple Render', 'Simple frame-by-frame render'),
|
||||||
|
('blender-render-progressive', 'Progressive Render',
|
||||||
|
'Each frame is rendered multiple times with different Cycles sample chunks, then combined'),
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
|
scene.flamenco_render_job_priority = IntProperty(
|
||||||
|
name='Job Priority',
|
||||||
|
min=0,
|
||||||
|
default=50,
|
||||||
|
max=100,
|
||||||
|
description='Higher numbers mean higher priority'
|
||||||
|
)
|
||||||
|
|
||||||
|
bpy.types.WindowManager.flamenco_status = EnumProperty(
|
||||||
|
items=[
|
||||||
|
('IDLE', 'IDLE', 'Not doing anything.'),
|
||||||
|
('PACKING', 'PACKING', 'BAM-packing all dependencies.'),
|
||||||
|
('COMMUNICATING', 'COMMUNICATING', 'Communicating with Flamenco Server.'),
|
||||||
|
],
|
||||||
|
name='flamenco_status',
|
||||||
|
default='IDLE',
|
||||||
|
description='Current status of the Flamenco add-on',
|
||||||
|
update=redraw)
|
||||||
|
|
||||||
|
|
||||||
|
def unregister():
|
||||||
|
deactivate()
|
||||||
|
bpy.utils.unregister_module(__name__)
|
||||||
|
|
||||||
|
try:
|
||||||
|
del bpy.types.Scene.flamenco_render_fchunk_size
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
try:
|
||||||
|
del bpy.types.Scene.flamenco_render_schunk_count
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
try:
|
||||||
|
del bpy.types.Scene.flamenco_render_frame_range
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
try:
|
||||||
|
del bpy.types.Scene.flamenco_render_job_type
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
try:
|
||||||
|
del bpy.types.Scene.flamenco_render_job_priority
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
try:
|
||||||
|
del bpy.types.WindowManager.flamenco_status
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
144
blender_cloud/flamenco/bam_interface.py
Normal file
144
blender_cloud/flamenco/bam_interface.py
Normal file
@@ -0,0 +1,144 @@
|
|||||||
|
"""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) -> 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',
|
||||||
|
]
|
||||||
|
|
||||||
|
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()
|
13
blender_cloud/flamenco/sdk.py
Normal file
13
blender_cloud/flamenco/sdk.py
Normal file
@@ -0,0 +1,13 @@
|
|||||||
|
from pillarsdk.resource import List, Find, Create
|
||||||
|
|
||||||
|
|
||||||
|
class Manager(List, Find):
|
||||||
|
"""Manager class wrapping the REST nodes endpoint"""
|
||||||
|
path = 'flamenco/managers'
|
||||||
|
|
||||||
|
|
||||||
|
class Job(List, Find, Create):
|
||||||
|
"""Job class wrapping the REST nodes endpoint
|
||||||
|
"""
|
||||||
|
path = 'flamenco/jobs'
|
||||||
|
ensure_query_projections = {'project': 1}
|
@@ -107,11 +107,7 @@ class PILLAR_OT_image_share(pillar.PillarOperatorMixin,
|
|||||||
self.report({'ERROR'}, 'Datablock is dirty, save it first.')
|
self.report({'ERROR'}, 'Datablock is dirty, save it first.')
|
||||||
return {'CANCELLED'}
|
return {'CANCELLED'}
|
||||||
|
|
||||||
async_loop.AsyncModalOperatorMixin.invoke(self, context, event)
|
return async_loop.AsyncModalOperatorMixin.invoke(self, context, event)
|
||||||
|
|
||||||
self.log.info('Starting sharing')
|
|
||||||
self._new_async_task(self.async_execute(context))
|
|
||||||
return {'RUNNING_MODAL'}
|
|
||||||
|
|
||||||
async def async_execute(self, context):
|
async def async_execute(self, context):
|
||||||
"""Entry point of the asynchronous operator."""
|
"""Entry point of the asynchronous operator."""
|
||||||
@@ -121,15 +117,15 @@ class PILLAR_OT_image_share(pillar.PillarOperatorMixin,
|
|||||||
try:
|
try:
|
||||||
# Refresh credentials
|
# Refresh credentials
|
||||||
try:
|
try:
|
||||||
self.user_id = await self.check_credentials(context,
|
db_user = await self.check_credentials(context, REQUIRES_ROLES_FOR_IMAGE_SHARING)
|
||||||
REQUIRES_ROLES_FOR_IMAGE_SHARING)
|
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:
|
||||||
self.log.exception('User not subscribed to cloud.')
|
self.log.exception('User not subscribed to cloud.')
|
||||||
self.report({'ERROR'}, 'Please subscribe to the Blender Cloud.')
|
self.report({'ERROR'}, 'Please subscribe to the Blender Cloud.')
|
||||||
self._state = 'QUIT'
|
self._state = 'QUIT'
|
||||||
return
|
return
|
||||||
except pillar.CredentialsNotSyncedError:
|
except pillar.UserNotLoggedInError:
|
||||||
self.log.exception('Error checking/refreshing credentials.')
|
self.log.exception('Error checking/refreshing credentials.')
|
||||||
self.report({'ERROR'}, 'Please log in on Blender ID first.')
|
self.report({'ERROR'}, 'Please log in on Blender ID first.')
|
||||||
self._state = 'QUIT'
|
self._state = 'QUIT'
|
||||||
@@ -287,7 +283,7 @@ class PILLAR_OT_image_share(pillar.PillarOperatorMixin,
|
|||||||
async def upload_screenshot(self, context) -> pillarsdk.Node:
|
async def upload_screenshot(self, context) -> pillarsdk.Node:
|
||||||
"""Takes a screenshot, saves it to a temp file, and uploads it."""
|
"""Takes a screenshot, saves it to a temp file, and uploads it."""
|
||||||
|
|
||||||
self.name = datetime.datetime.now().strftime('Screenshot-%Y-%m-%d-%H:%M:%S.png')
|
self.name = datetime.datetime.now().strftime('Screenshot-%Y-%m-%d-%H%M%S.png')
|
||||||
self.report({'INFO'}, "Uploading %s '%s'" % (self.target.lower(), self.name))
|
self.report({'INFO'}, "Uploading %s '%s'" % (self.target.lower(), self.name))
|
||||||
|
|
||||||
with tempfile.TemporaryDirectory() as tmpdir:
|
with tempfile.TemporaryDirectory() as tmpdir:
|
||||||
|
@@ -146,6 +146,13 @@ def blender_id_subclient() -> dict:
|
|||||||
return subclient
|
return subclient
|
||||||
|
|
||||||
|
|
||||||
|
def pillar_user_uuid() -> str:
|
||||||
|
"""Returns the UUID of the Pillar user."""
|
||||||
|
|
||||||
|
import blender_id
|
||||||
|
return blender_id.get_subclient_user_id(SUBCLIENT_ID)
|
||||||
|
|
||||||
|
|
||||||
def pillar_api(pillar_endpoint: str = None, caching=True) -> pillarsdk.Api:
|
def pillar_api(pillar_endpoint: str = None, caching=True) -> pillarsdk.Api:
|
||||||
"""Returns the Pillar SDK API object for the current user.
|
"""Returns the Pillar SDK API object for the current user.
|
||||||
|
|
||||||
@@ -198,6 +205,12 @@ pillar_semaphore = asyncio.Semaphore(3)
|
|||||||
|
|
||||||
|
|
||||||
async def pillar_call(pillar_func, *args, caching=True, **kwargs):
|
async def pillar_call(pillar_func, *args, caching=True, **kwargs):
|
||||||
|
"""Calls a Pillar function.
|
||||||
|
|
||||||
|
A semaphore is used to ensure that there won't be too many
|
||||||
|
calls to Pillar simultaneously.
|
||||||
|
"""
|
||||||
|
|
||||||
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()
|
||||||
|
|
||||||
@@ -205,6 +218,12 @@ async def pillar_call(pillar_func, *args, caching=True, **kwargs):
|
|||||||
return await loop.run_in_executor(None, partial)
|
return await loop.run_in_executor(None, partial)
|
||||||
|
|
||||||
|
|
||||||
|
def sync_call(pillar_func, *args, caching=True, **kwargs):
|
||||||
|
"""Synchronous call to Pillar, ensures the correct Api object is used."""
|
||||||
|
|
||||||
|
return pillar_func(*args, api=pillar_api(caching=caching), **kwargs)
|
||||||
|
|
||||||
|
|
||||||
async def check_pillar_credentials(required_roles: set):
|
async def check_pillar_credentials(required_roles: set):
|
||||||
"""Tries to obtain the user at Pillar using the user's credentials.
|
"""Tries to obtain the user at Pillar using the user's credentials.
|
||||||
|
|
||||||
@@ -241,7 +260,7 @@ async def check_pillar_credentials(required_roles: set):
|
|||||||
profile.save_json()
|
profile.save_json()
|
||||||
raise NotSubscribedToCloudError()
|
raise NotSubscribedToCloudError()
|
||||||
|
|
||||||
return pillar_user_id
|
return db_user
|
||||||
|
|
||||||
|
|
||||||
async def refresh_pillar_credentials(required_roles: set):
|
async def refresh_pillar_credentials(required_roles: set):
|
||||||
@@ -256,7 +275,7 @@ async def refresh_pillar_credentials(required_roles: set):
|
|||||||
import blender_id
|
import blender_id
|
||||||
|
|
||||||
from . import blender
|
from . import blender
|
||||||
pillar_endpoint = blender.preferences().pillar_server.rstrip('/')
|
pillar_endpoint = blender.preferences().pillar_server
|
||||||
|
|
||||||
# Create a subclient token and send it to Pillar.
|
# Create a subclient token and send it to Pillar.
|
||||||
# May raise a blender_id.BlenderIdCommError
|
# May raise a blender_id.BlenderIdCommError
|
||||||
@@ -780,15 +799,16 @@ def is_cancelled(future: asyncio.Future) -> bool:
|
|||||||
|
|
||||||
class PillarOperatorMixin:
|
class PillarOperatorMixin:
|
||||||
async def check_credentials(self, context, required_roles) -> bool:
|
async def check_credentials(self, context, required_roles) -> bool:
|
||||||
"""Checks credentials with Pillar, and if ok returns the user ID.
|
"""Checks credentials with Pillar, and if ok returns the user document from Pillar/MongoDB.
|
||||||
|
|
||||||
Returns None if the user cannot be found, or if the user is not a Cloud subscriber.
|
:raises UserNotLoggedInError: if the user is not logged in
|
||||||
|
:raises NotSubscribedToCloudError: if the user does not have any of the required roles
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# self.report({'INFO'}, 'Checking Blender Cloud credentials')
|
# self.report({'INFO'}, 'Checking Blender Cloud credentials')
|
||||||
|
|
||||||
try:
|
try:
|
||||||
user_id = await check_pillar_credentials(required_roles)
|
db_user = await check_pillar_credentials(required_roles)
|
||||||
except NotSubscribedToCloudError:
|
except NotSubscribedToCloudError:
|
||||||
self._log_subscription_needed()
|
self._log_subscription_needed()
|
||||||
raise
|
raise
|
||||||
@@ -796,20 +816,22 @@ class PillarOperatorMixin:
|
|||||||
self.log.info('Credentials not synced, re-syncing automatically.')
|
self.log.info('Credentials not synced, re-syncing automatically.')
|
||||||
else:
|
else:
|
||||||
self.log.info('Credentials okay.')
|
self.log.info('Credentials okay.')
|
||||||
return user_id
|
return db_user
|
||||||
|
|
||||||
try:
|
try:
|
||||||
user_id = await refresh_pillar_credentials(required_roles)
|
db_user = await refresh_pillar_credentials(required_roles)
|
||||||
except NotSubscribedToCloudError:
|
except NotSubscribedToCloudError:
|
||||||
self._log_subscription_needed()
|
self._log_subscription_needed()
|
||||||
raise
|
raise
|
||||||
|
except CredentialsNotSyncedError:
|
||||||
|
self.log.info('Credentials not synced after refreshing, handling as not logged in.')
|
||||||
|
raise UserNotLoggedInError('Not logged in.')
|
||||||
except UserNotLoggedInError:
|
except UserNotLoggedInError:
|
||||||
self.log.error('User not logged in on Blender ID.')
|
self.log.error('User not logged in on Blender ID.')
|
||||||
|
raise
|
||||||
else:
|
else:
|
||||||
self.log.info('Credentials refreshed and ok.')
|
self.log.info('Credentials refreshed and ok.')
|
||||||
return user_id
|
return db_user
|
||||||
|
|
||||||
return None
|
|
||||||
|
|
||||||
def _log_subscription_needed(self):
|
def _log_subscription_needed(self):
|
||||||
self.log.warning(
|
self.log.warning(
|
||||||
@@ -818,6 +840,39 @@ class PillarOperatorMixin:
|
|||||||
'Please subscribe to the blender cloud at https://cloud.blender.org/join')
|
'Please subscribe to the blender cloud at https://cloud.blender.org/join')
|
||||||
|
|
||||||
|
|
||||||
|
class AuthenticatedPillarOperatorMixin(PillarOperatorMixin):
|
||||||
|
"""Checks credentials, to be used at the start of async_execute().
|
||||||
|
|
||||||
|
Sets self.user_id to the current user's ID, and self.db_user to the user info dict,
|
||||||
|
if authentication was succesful; sets both to None if not.
|
||||||
|
"""
|
||||||
|
|
||||||
|
async def authenticate(self, context) -> bool:
|
||||||
|
from . import pillar
|
||||||
|
|
||||||
|
self.log.info('Checking credentials')
|
||||||
|
self.user_id = None
|
||||||
|
self.db_user = None
|
||||||
|
try:
|
||||||
|
self.db_user = await self.check_credentials(context, ())
|
||||||
|
except pillar.UserNotLoggedInError as ex:
|
||||||
|
self.log.info('Not logged in error raised: %s', ex)
|
||||||
|
self.report({'ERROR'}, 'Please log in on Blender ID first.')
|
||||||
|
self.quit()
|
||||||
|
return False
|
||||||
|
except requests.exceptions.ConnectionError:
|
||||||
|
self.log.exception('Error checking pillar credentials.')
|
||||||
|
self.report({'ERROR'}, 'Unable to connect to Blender Cloud, '
|
||||||
|
'check your internet connection.')
|
||||||
|
self.quit()
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
self.user_id = self.db_user['_id']
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
async def find_or_create_node(where: dict,
|
async def find_or_create_node(where: dict,
|
||||||
additional_create_props: dict = None,
|
additional_create_props: dict = None,
|
||||||
projection: dict = None,
|
projection: dict = None,
|
||||||
|
@@ -234,11 +234,7 @@ class PILLAR_OT_sync(pillar.PillarOperatorMixin,
|
|||||||
self.bss_report({'ERROR'}, 'No Blender version to sync for was given.')
|
self.bss_report({'ERROR'}, 'No Blender version to sync for was given.')
|
||||||
return {'CANCELLED'}
|
return {'CANCELLED'}
|
||||||
|
|
||||||
async_loop.AsyncModalOperatorMixin.invoke(self, context, event)
|
return async_loop.AsyncModalOperatorMixin.invoke(self, context, event)
|
||||||
|
|
||||||
self.log.info('Starting synchronisation')
|
|
||||||
self._new_async_task(self.async_execute(context))
|
|
||||||
return {'RUNNING_MODAL'}
|
|
||||||
|
|
||||||
def action_select(self, context):
|
def action_select(self, context):
|
||||||
"""Allows selection of the Blender version to use.
|
"""Allows selection of the Blender version to use.
|
||||||
@@ -285,14 +281,15 @@ class PILLAR_OT_sync(pillar.PillarOperatorMixin,
|
|||||||
try:
|
try:
|
||||||
# Refresh credentials
|
# Refresh credentials
|
||||||
try:
|
try:
|
||||||
self.user_id = 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']
|
||||||
log.debug('Found user ID: %s', self.user_id)
|
log.debug('Found user ID: %s', self.user_id)
|
||||||
except pillar.NotSubscribedToCloudError:
|
except pillar.NotSubscribedToCloudError:
|
||||||
self.log.exception('User not subscribed to cloud.')
|
self.log.exception('User not subscribed to cloud.')
|
||||||
self.bss_report({'SUBSCRIBE'}, 'Please subscribe to the Blender Cloud.')
|
self.bss_report({'SUBSCRIBE'}, 'Please subscribe to the Blender Cloud.')
|
||||||
self._state = 'QUIT'
|
self._state = 'QUIT'
|
||||||
return
|
return
|
||||||
except pillar.CredentialsNotSyncedError:
|
except pillar.UserNotLoggedInError:
|
||||||
self.log.exception('Error checking/refreshing credentials.')
|
self.log.exception('Error checking/refreshing credentials.')
|
||||||
self.bss_report({'ERROR'}, 'Please log in on Blender ID first.')
|
self.bss_report({'ERROR'}, 'Please log in on Blender ID first.')
|
||||||
self._state = 'QUIT'
|
self._state = 'QUIT'
|
||||||
|
@@ -258,8 +258,9 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
scroll_offset_space_left = 0
|
scroll_offset_space_left = 0
|
||||||
|
|
||||||
def invoke(self, context, event):
|
def invoke(self, context, event):
|
||||||
# Refuse to start if the file hasn't been saved.
|
# Refuse to start if the file hasn't been saved. It's okay if
|
||||||
if context.blend_data.is_dirty:
|
# it's dirty, we just need to know where '//' points to.
|
||||||
|
if not os.path.exists(context.blend_data.filepath):
|
||||||
self.report({'ERROR'}, 'Please save your Blend file before using '
|
self.report({'ERROR'}, 'Please save your Blend file before using '
|
||||||
'the Blender Cloud addon.')
|
'the Blender Cloud addon.')
|
||||||
return {'CANCELLED'}
|
return {'CANCELLED'}
|
||||||
@@ -288,10 +289,7 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
self._scroll_reset()
|
self._scroll_reset()
|
||||||
|
|
||||||
context.window.cursor_modal_set('DEFAULT')
|
context.window.cursor_modal_set('DEFAULT')
|
||||||
async_loop.AsyncModalOperatorMixin.invoke(self, context, event)
|
return async_loop.AsyncModalOperatorMixin.invoke(self, context, event)
|
||||||
self._new_async_task(self.async_execute(context))
|
|
||||||
|
|
||||||
return {'RUNNING_MODAL'}
|
|
||||||
|
|
||||||
def modal(self, context, event):
|
def modal(self, context, event):
|
||||||
result = async_loop.AsyncModalOperatorMixin.modal(self, context, event)
|
result = async_loop.AsyncModalOperatorMixin.modal(self, context, event)
|
||||||
@@ -366,13 +364,13 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
self.log.debug('Checking credentials')
|
self.log.debug('Checking credentials')
|
||||||
|
|
||||||
try:
|
try:
|
||||||
user_id = 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:
|
||||||
self.log.info('User not subscribed to Blender Cloud.')
|
self.log.info('User not subscribed to Blender Cloud.')
|
||||||
self._show_subscribe_screen()
|
self._show_subscribe_screen()
|
||||||
return None
|
return None
|
||||||
|
|
||||||
if user_id is None:
|
if db_user is None:
|
||||||
raise pillar.UserNotLoggedInError()
|
raise pillar.UserNotLoggedInError()
|
||||||
|
|
||||||
await self.async_download_previews()
|
await self.async_download_previews()
|
||||||
@@ -555,6 +553,7 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
"""Draws the GUI with OpenGL."""
|
"""Draws the GUI with OpenGL."""
|
||||||
|
|
||||||
drawers = {
|
drawers = {
|
||||||
|
'INITIALIZING': self._draw_initializing,
|
||||||
'CHECKING_CREDENTIALS': self._draw_checking_credentials,
|
'CHECKING_CREDENTIALS': self._draw_checking_credentials,
|
||||||
'BROWSING': self._draw_browser,
|
'BROWSING': self._draw_browser,
|
||||||
'DOWNLOADING_TEXTURE': self._draw_downloading,
|
'DOWNLOADING_TEXTURE': self._draw_downloading,
|
||||||
@@ -655,6 +654,13 @@ class BlenderCloudBrowser(pillar.PillarOperatorMixin,
|
|||||||
'Checking login credentials',
|
'Checking login credentials',
|
||||||
(0.0, 0.0, 0.2, 0.6))
|
(0.0, 0.0, 0.2, 0.6))
|
||||||
|
|
||||||
|
def _draw_initializing(self, context):
|
||||||
|
"""OpenGL drawing code for the INITIALIZING state."""
|
||||||
|
|
||||||
|
self._draw_text_on_colour(context,
|
||||||
|
'Initializing',
|
||||||
|
(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, 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)
|
||||||
@@ -850,13 +856,6 @@ class PILLAR_OT_switch_hdri(pillar.PillarOperatorMixin,
|
|||||||
file_uuid = bpy.props.StringProperty(name='file_uuid',
|
file_uuid = bpy.props.StringProperty(name='file_uuid',
|
||||||
description='File ID to download')
|
description='File ID to download')
|
||||||
|
|
||||||
def invoke(self, context, event):
|
|
||||||
async_loop.AsyncModalOperatorMixin.invoke(self, context, event)
|
|
||||||
|
|
||||||
self.log.info('Starting')
|
|
||||||
self._new_async_task(self.async_execute(context))
|
|
||||||
return {'RUNNING_MODAL'}
|
|
||||||
|
|
||||||
async def async_execute(self, context):
|
async def async_execute(self, context):
|
||||||
"""Entry point of the asynchronous operator."""
|
"""Entry point of the asynchronous operator."""
|
||||||
|
|
||||||
@@ -864,19 +863,20 @@ class PILLAR_OT_switch_hdri(pillar.PillarOperatorMixin,
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
try:
|
try:
|
||||||
user_id = 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']
|
||||||
except pillar.NotSubscribedToCloudError:
|
except pillar.NotSubscribedToCloudError:
|
||||||
self.log.exception('User not subscribed to cloud.')
|
self.log.exception('User not subscribed to cloud.')
|
||||||
self.report({'ERROR'}, 'Please subscribe to the Blender Cloud.')
|
self.report({'ERROR'}, 'Please subscribe to the Blender Cloud.')
|
||||||
self._state = 'QUIT'
|
self._state = 'QUIT'
|
||||||
return
|
return
|
||||||
except pillar.CredentialsNotSyncedError:
|
except pillar.UserNotLoggedInError:
|
||||||
self.log.exception('Error checking/refreshing credentials.')
|
self.log.exception('Error checking/refreshing credentials.')
|
||||||
self.report({'ERROR'}, 'Please log in on Blender ID first.')
|
self.report({'ERROR'}, 'Please log in on Blender ID first.')
|
||||||
self._state = 'QUIT'
|
self._state = 'QUIT'
|
||||||
return
|
return
|
||||||
|
|
||||||
if user_id is None:
|
if not user_id:
|
||||||
raise pillar.UserNotLoggedInError()
|
raise pillar.UserNotLoggedInError()
|
||||||
|
|
||||||
await self.download_and_replace(context)
|
await self.download_and_replace(context)
|
||||||
@@ -977,6 +977,11 @@ def _hdri_download_panel(self, current_image):
|
|||||||
props.file_uuid = current_image.hdri_variation
|
props.file_uuid = current_image.hdri_variation
|
||||||
|
|
||||||
|
|
||||||
|
# Storage for variation labels, as the strings in EnumProperty items
|
||||||
|
# MUST be kept in Python memory.
|
||||||
|
variation_label_storage = {}
|
||||||
|
|
||||||
|
|
||||||
def hdri_variation_choices(self, context):
|
def hdri_variation_choices(self, context):
|
||||||
if context.area.type == 'IMAGE_EDITOR':
|
if context.area.type == 'IMAGE_EDITOR':
|
||||||
image = context.edit_image
|
image = context.edit_image
|
||||||
@@ -988,8 +993,11 @@ def hdri_variation_choices(self, context):
|
|||||||
if 'bcloud_node' not in image:
|
if 'bcloud_node' not in image:
|
||||||
return []
|
return []
|
||||||
|
|
||||||
choices = [(file_doc['file'], file_doc['resolution'], '')
|
choices = []
|
||||||
for file_doc in image['bcloud_node']['properties']['files']]
|
for file_doc in image['bcloud_node']['properties']['files']:
|
||||||
|
label = file_doc['resolution']
|
||||||
|
variation_label_storage[label] = label
|
||||||
|
choices.append((file_doc['file'], label, ''))
|
||||||
|
|
||||||
return choices
|
return choices
|
||||||
|
|
||||||
|
@@ -16,6 +16,8 @@
|
|||||||
#
|
#
|
||||||
# ##### END GPL LICENSE BLOCK #####
|
# ##### END GPL LICENSE BLOCK #####
|
||||||
|
|
||||||
|
import pathlib
|
||||||
|
|
||||||
|
|
||||||
def sizeof_fmt(num: int, suffix='B') -> str:
|
def sizeof_fmt(num: int, suffix='B') -> str:
|
||||||
"""Returns a human-readable size.
|
"""Returns a human-readable size.
|
||||||
@@ -29,3 +31,72 @@ def sizeof_fmt(num: int, suffix='B') -> str:
|
|||||||
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:
|
||||||
|
"""Performs a breadth-first search for the filename.
|
||||||
|
|
||||||
|
Returns the path that contains the file, or None if not found.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import collections
|
||||||
|
|
||||||
|
# Be lenient on our input type.
|
||||||
|
if isinstance(path, str):
|
||||||
|
path = pathlib.Path(path)
|
||||||
|
|
||||||
|
if not path.exists():
|
||||||
|
return None
|
||||||
|
assert path.is_dir()
|
||||||
|
|
||||||
|
to_visit = collections.deque([path])
|
||||||
|
while to_visit:
|
||||||
|
this_path = to_visit.popleft()
|
||||||
|
|
||||||
|
for subpath in this_path.iterdir():
|
||||||
|
if subpath.is_dir():
|
||||||
|
to_visit.append(subpath)
|
||||||
|
continue
|
||||||
|
|
||||||
|
if subpath.name == filename:
|
||||||
|
return subpath
|
||||||
|
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def pyside_cache(propname):
|
||||||
|
"""Decorator, stores the result of the decorated callable in Python-managed memory.
|
||||||
|
|
||||||
|
This is to work around the warning at
|
||||||
|
https://www.blender.org/api/blender_python_api_master/bpy.props.html#bpy.props.EnumProperty
|
||||||
|
"""
|
||||||
|
|
||||||
|
if callable(propname):
|
||||||
|
raise TypeError('Usage: pyside_cache("property_name")')
|
||||||
|
|
||||||
|
def decorator(wrapped):
|
||||||
|
"""Stores the result of the callable in Python-managed memory.
|
||||||
|
|
||||||
|
This is to work around the warning at
|
||||||
|
https://www.blender.org/api/blender_python_api_master/bpy.props.html#bpy.props.EnumProperty
|
||||||
|
"""
|
||||||
|
|
||||||
|
import functools
|
||||||
|
|
||||||
|
@functools.wraps(wrapped)
|
||||||
|
# We can't use (*args, **kwargs), because EnumProperty explicitly checks
|
||||||
|
# for the number of fixed positional arguments.
|
||||||
|
def wrapper(self, context):
|
||||||
|
result = None
|
||||||
|
try:
|
||||||
|
result = wrapped(self, context)
|
||||||
|
return result
|
||||||
|
finally:
|
||||||
|
rna_type, rna_info = getattr(self.bl_rna, propname)
|
||||||
|
rna_info['_cached_result'] = result
|
||||||
|
return wrapper
|
||||||
|
return decorator
|
||||||
|
|
||||||
|
|
||||||
|
def redraw(self, context):
|
||||||
|
context.area.tag_redraw()
|
||||||
|
8
requirements-dev.txt
Normal file
8
requirements-dev.txt
Normal file
@@ -0,0 +1,8 @@
|
|||||||
|
-r requirements.txt
|
||||||
|
|
||||||
|
# Primary requirements
|
||||||
|
pytest==3.0.3
|
||||||
|
|
||||||
|
# Secondary requirements
|
||||||
|
py==1.4.31
|
||||||
|
|
@@ -1,7 +1,7 @@
|
|||||||
# 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.5.0
|
pillarsdk==1.6.1
|
||||||
wheel==0.29.0
|
wheel==0.29.0
|
||||||
|
|
||||||
# Secondary requirements:
|
# Secondary requirements:
|
||||||
|
35
setup.py
35
setup.py
@@ -18,11 +18,13 @@
|
|||||||
# ##### 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
|
||||||
import re
|
import re
|
||||||
import pathlib
|
import pathlib
|
||||||
|
import zipfile
|
||||||
|
|
||||||
from distutils import log
|
from distutils import log
|
||||||
from distutils.core import Command
|
from distutils.core import Command
|
||||||
@@ -168,6 +170,34 @@ class BlenderAddonBdist(bdist):
|
|||||||
super().run()
|
super().run()
|
||||||
|
|
||||||
|
|
||||||
|
# noinspection PyAttributeOutsideInit
|
||||||
|
class BlenderAddonFdist(BlenderAddonBdist):
|
||||||
|
"""Ensures that 'python setup.py fdist' creates a plain folder structure."""
|
||||||
|
|
||||||
|
user_options = [
|
||||||
|
('dest-path=', None, 'addon installation path'),
|
||||||
|
]
|
||||||
|
|
||||||
|
def initialize_options(self):
|
||||||
|
super().initialize_options()
|
||||||
|
self.dest_path = None # path that will contain the addon
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
super().run()
|
||||||
|
|
||||||
|
# dist_files is a list of tuples ('bdist', 'any', 'filepath')
|
||||||
|
filepath = self.distribution.dist_files[0][2]
|
||||||
|
|
||||||
|
# if dest_path is not specified use the filename as the dest_path (minus the .zip)
|
||||||
|
assert filepath.endswith('.zip')
|
||||||
|
target_folder = self.dest_path or filepath[:-4]
|
||||||
|
|
||||||
|
print('Unzipping the package on {}.'.format(target_folder))
|
||||||
|
|
||||||
|
with zipfile.ZipFile(filepath, 'r') as zip_ref:
|
||||||
|
zip_ref.extractall(target_folder)
|
||||||
|
|
||||||
|
|
||||||
# noinspection PyAttributeOutsideInit
|
# noinspection PyAttributeOutsideInit
|
||||||
class BlenderAddonInstall(install):
|
class BlenderAddonInstall(install):
|
||||||
"""Ensures the module is placed at the root of the zip file."""
|
"""Ensures the module is placed at the root of the zip file."""
|
||||||
@@ -191,16 +221,17 @@ class AvoidEggInfo(install_egg_info):
|
|||||||
|
|
||||||
setup(
|
setup(
|
||||||
cmdclass={'bdist': BlenderAddonBdist,
|
cmdclass={'bdist': BlenderAddonBdist,
|
||||||
|
'fdist': BlenderAddonFdist,
|
||||||
'install': BlenderAddonInstall,
|
'install': BlenderAddonInstall,
|
||||||
'install_egg_info': AvoidEggInfo,
|
'install_egg_info': AvoidEggInfo,
|
||||||
'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.4.1',
|
version='1.6.3',
|
||||||
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']),
|
data_files=[('blender_cloud', ['README.md', 'README-flamenco.md']),
|
||||||
('blender_cloud/icons', glob.glob('blender_cloud/icons/*'))],
|
('blender_cloud/icons', glob.glob('blender_cloud/icons/*'))],
|
||||||
scripts=[],
|
scripts=[],
|
||||||
url='https://developer.blender.org/diffusion/BCA/',
|
url='https://developer.blender.org/diffusion/BCA/',
|
||||||
|
25
tests/test_utils.py
Normal file
25
tests/test_utils.py
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
"""Unittests for blender_cloud.utils."""
|
||||||
|
|
||||||
|
import pathlib
|
||||||
|
import unittest
|
||||||
|
|
||||||
|
from blender_cloud import utils
|
||||||
|
|
||||||
|
|
||||||
|
class FindInPathTest(unittest.TestCase):
|
||||||
|
def test_nonexistant_path(self):
|
||||||
|
path = pathlib.Path('/doesnotexistreally')
|
||||||
|
self.assertFalse(path.exists())
|
||||||
|
self.assertIsNone(utils.find_in_path(path, 'jemoeder.blend'))
|
||||||
|
|
||||||
|
def test_really_breadth_first(self):
|
||||||
|
"""A depth-first test might find dir_a1/dir_a2/dir_a3/find_me.txt first."""
|
||||||
|
|
||||||
|
path = pathlib.Path(__file__).parent / 'test_really_breadth_first'
|
||||||
|
found = utils.find_in_path(path, 'find_me.txt')
|
||||||
|
self.assertEqual(path / 'dir_b1' / 'dir_b2' / 'find_me.txt', found)
|
||||||
|
|
||||||
|
def test_nonexistant_file(self):
|
||||||
|
path = pathlib.Path(__file__).parent / 'test_really_breadth_first'
|
||||||
|
found = utils.find_in_path(path, 'do_not_find_me.txt')
|
||||||
|
self.assertEqual(None, found)
|
19
update_version.sh
Executable file
19
update_version.sh
Executable file
@@ -0,0 +1,19 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
VERSION="${1/version-}"
|
||||||
|
|
||||||
|
if [ -z "$VERSION" ]; then
|
||||||
|
echo "Usage: $0 new-version" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
BL_INFO_VER=$(echo "$VERSION" | sed 's/\./, /g')
|
||||||
|
|
||||||
|
sed "s/version='[^']*'/version='$VERSION'/" -i setup.py
|
||||||
|
sed "s/'version': ([^)]*)/'version': ($BL_INFO_VER)/" -i blender_cloud/__init__.py
|
||||||
|
|
||||||
|
git diff
|
||||||
|
echo
|
||||||
|
echo "Don't forget to commit and tag:"
|
||||||
|
echo git commit -m \'Bumped version to $VERSION\' setup.py blender_cloud/__init__.py
|
||||||
|
echo git tag -a version-$VERSION -m \'Tagged version $VERSION\'
|
Reference in New Issue
Block a user