BlenderKit initial commit.

BlenderKit add-on is the official addon of the BlenderKit service for Blender 3d. (www.blenderkit.com)
It enables users to upload, search, download, and rate different assets for blender.
It works together with BlenderKit server.

1 files changed, 158 insertions, 0 deletions

diff --git a/blenderkit/autothumb_model_bg.py b/blenderkit/autothumb_model_bg.py
new file mode 100644
index 00000000..dc34d59c
--- /dev/null
+++ b/blenderkit/autothumb_model_bg.py
@@ -0,0 +1,158 @@
+# ##### 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 #####
+
+
+import sys, json, math
+from pathlib import Path
+import bpy
+from blenderkit import utils, append_link, bg_blender
+import mathutils
+
+BLENDERKIT_EXPORT_TEMP_DIR = sys.argv[-1]
+BLENDERKIT_THUMBNAIL_PATH = sys.argv[-2]
+BLENDERKIT_EXPORT_FILE_INPUT = sys.argv[-3]
+BLENDERKIT_EXPORT_DATA = sys.argv[-4]
+
+
+def get_obnames():
+    with open(BLENDERKIT_EXPORT_DATA, 'r') as s:
+        data = json.load(s)
+    obnames = eval(data['models'])
+    return obnames
+
+
+def center_obs_for_thumbnail(obs):
+    s = bpy.context.scene
+    # obs = bpy.context.selected_objects
+    parent = obs[0]
+
+    while parent.parent != None:
+        parent = parent.parent
+    # reset parent rotation, so we see how it really snaps.
+    parent.rotation_euler = (0, 0, 0)
+    bpy.context.scene.update()
+    minx, miny, minz, maxx, maxy, maxz = utils.get_bounds_worldspace(obs)
+
+    cx = (maxx - minx) / 2 + minx
+    cy = (maxy - miny) / 2 + miny
+    for ob in s.collection.objects:
+        ob.select_set(False)
+
+    bpy.context.view_layer.objects.active = parent
+    parent.location += mathutils.Vector((-cx, -cy, -minz))
+
+    camZ = s.camera.parent.parent
+    camZ.location.z = (maxz - minz) / 2
+    dx = (maxx - minx)
+    dy = (maxy - miny)
+    dz = (maxz - minz)
+    r = math.sqrt(dx * dx + dy * dy + dz * dz)
+
+    scaler = bpy.context.view_layer.objects['scaler']
+    scaler.scale = (r, r, r)
+    coef = .7
+    r *= coef
+    camZ.scale = (r, r, r)
+    s.update()
+
+
+def render_thumbnails():
+    bpy.ops.render.render(write_still=True, animation=False)
+
+
+if __name__ == "__main__":
+    try:
+        with open(BLENDERKIT_EXPORT_DATA, 'r') as s:
+            data = json.load(s)
+
+        user_preferences = bpy.context.preferences.addons['blenderkit'].preferences
+
+        bg_blender.progress('preparing thumbnail scene')
+        obnames = get_obnames()
+        main_object, allobs = append_link.append_objects(file_name=BLENDERKIT_EXPORT_FILE_INPUT,
+                                                         obnames=obnames,
+                                                         link=True)
+        bpy.context.scene.update()
+
+        camdict = {
+            'GROUND': 'camera ground',
+            'WALL': 'camera wall',
+            'CEILING': 'camera ceiling',
+            'FLOAT': 'camera float'
+        }
+
+        bpy.context.scene.camera = bpy.data.objects[camdict[data['thumbnail_snap_to']]]
+        center_obs_for_thumbnail(allobs)
+        bpy.context.scene.render.filepath = BLENDERKIT_THUMBNAIL_PATH
+        if user_preferences.thumbnail_use_gpu:
+            bpy.context.scene.cycles.device = 'GPU'
+
+        fdict = {
+            'DEFAULT': 1,
+            'FRONT': 2,
+            'SIDE': 3,
+            'TOP': 4,
+        }
+        s = bpy.context.scene
+        s.frame_set(fdict[data['thumbnail_angle']])
+
+        snapdict = {
+            'GROUND': 'Ground',
+            'WALL': 'Wall',
+            'CEILING': 'Ceiling',
+            'FLOAT': 'Float'
+        }
+
+        collection = bpy.context.scene.collection.children[snapdict[data['thumbnail_snap_to']]]
+        collection.hide_viewport = False
+        collection.hide_render = False
+        collection.hide_select = False
+
+        main_object.rotation_euler = (0, 0, 0)
+        bpy.data.materials['bkit background'].node_tree.nodes['Value'].outputs['Value'].default_value \
+            = data['thumbnail_background_lightness']
+        s.cycles.samples = data['thumbnail_samples']
+        bpy.context.view_layer.cycles.use_denoising = data['thumbnail_denoising']
+        bpy.context.scene.update()
+
+        # import blender's HDR here
+        hdr_path = Path('datafiles/studiolights/world/interior.exr')
+        bpath = Path(bpy.utils.resource_path('LOCAL'))
+        ipath = bpath / hdr_path
+        ipath = str(ipath)
+
+        # this  stuff is for mac and possibly linux. For blender // means relative path.
+        # for Mac, // means start of absolute path
+        if ipath.startswith('//'):
+            ipath = ipath[1:]
+
+        img = bpy.data.images['interior.exr']
+        img.filepath = ipath
+        img.reload()
+
+        bg_blender.progress('rendering thumbnail')
+        render_thumbnails()
+        fpath = BLENDERKIT_THUMBNAIL_PATH + '0001.jpg'
+        bg_blender.progress('background autothumbnailer finished successfully')
+
+
+    except:
+        import traceback
+
+        traceback.print_exc()
+        sys.exit(1)