Tests: Automated geometry nodes benchmark

Add a script for a very simple object evaluation benchmark.
There could be more advanced ways of measuring the time
per-node or per modifier, but this just loads the file, tags
the active object for a reevaluation, and times how long
that takes.

Differential Revision: https://developer.blender.org/D16604
This commit is contained in:
Hans Goudey 2022-11-26 17:15:55 -06:00
parent 828525b268
commit 3a41e0f611
Notes: blender-bot 2023-06-07 10:31:13 +02:00
Referenced by issue #92969, Automated performance benchmarks for geometry nodes/modifiers
1 changed files with 49 additions and 0 deletions

View File

@ -0,0 +1,49 @@
# SPDX-License-Identifier: Apache-2.0
import api
import os
def _run(args):
import bpy
import time
# Evaluate objects once first, to avoid any possible lazy evaluation later.
bpy.context.view_layer.update()
# Tag all objects with geometry nodes modifiers to be recalculated.
for ob in bpy.context.view_layer.objects:
for modifier in ob.modifiers:
if modifier.type == 'NODES':
ob.update_tag()
break
start_time = time.time()
bpy.context.view_layer.update()
elapsed_time = time.time() - start_time
result = {'time': elapsed_time}
return result
class GeometryNodesTest(api.Test):
def __init__(self, filepath):
self.filepath = filepath
def name(self):
return self.filepath.stem
def category(self):
return "geometry_nodes"
def run(self, env, device_id):
args = {}
result, _ = env.run_in_blender(_run, args, [self.filepath])
return result
def generate(env):
filepaths = env.find_blend_files('geometry_nodes/*')
return [GeometryNodesTest(filepath) for filepath in filepaths]