3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-05-31 01:46:31 +02:00
Code Issues Projects Releases Packages Wiki Activity

[blender] preview: handle background image not found

Browse source
This commit is contained in:
Loïc Vital 2023-06-09 15:33:04 +02:00
parent 349f488fbd
commit 649a45249c
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
meshroom/nodes/blender/scripts/preview.py
View file

@ -149,10 +149,14 @@ def setupBackground(view, folderUndistorted):
if len(matches) == 0: if len(matches) == 0:
baseImgName = os.path.splitext(os.path.basename(view['path']))[0] baseImgName = os.path.splitext(os.path.basename(view['path']))[0]
matches = glob.glob(folderUndistorted + '/*' + baseImgName + "*") # try with image name matches = glob.glob(folderUndistorted + '/*' + baseImgName + "*") # try with image name
if len(matches) == 0:
# no background image found
return False
undistortedImgPath = matches[0] undistortedImgPath = matches[0]
bpy.ops.image.open(filepath=undistortedImgPath) bpy.ops.image.open(filepath=undistortedImgPath)
undistortedImgName = os.path.basename(undistortedImgPath) undistortedImgName = os.path.basename(undistortedImgPath)
bpy.context.scene.node_tree.nodes["Image"].image = bpy.data.images[undistortedImgName] bpy.context.scene.node_tree.nodes["Image"].image = bpy.data.images[undistortedImgName]
return True
def loadModel(filename): def loadModel(filename):
@ -332,7 +336,10 @@ def main():
print("Rendering view " + view['viewId']) print("Rendering view " + view['viewId'])
if args.useBackground: if args.useBackground:
setupBackground(view, args.undistortedImages) if not setupBackground(view, args.undistortedImages):
# background setup failed
# do not render this frame
continue
setupRender(view, intrinsic, pose, args.output) setupRender(view, intrinsic, pose, args.output)
bpy.ops.render.render(write_still=True) bpy.ops.render.render(write_still=True)