1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
import bpy
from dataclasses import dataclass
from itertools import islice
from pathlib import Path
GEO_NODES_PASS = [
"GeometryNodeCustomGroup",
"GeometryNodeGroup",
"GeometryNodeRepeatInput",
"GeometryNodeRepeatOutput",
"GeometryNodeSimulationInput",
"GeometryNodeSimulationOutput"
]
@dataclass
class Rectangle:
x: float
y: float
width: float
height: float
def cut_image(old_path, new_path, rect):
image = bpy.data.images.load(old_path)
total_width, total_height = image.size
assert total_width >= rect.x + rect.width
assert total_height >= rect.y + rect.height
old_pixels = image.pixels
new_pixels = []
for y in range(rect.y, rect.y + rect.height):
row_start = y * total_width + rect.x
row_end = row_start + rect.width
new_pixels.extend(old_pixels[4 * row_start:4 * row_end])
image.scale(rect.width, rect.height)
image.pixels = new_pixels
image.filepath_raw = new_path
image.save()
bpy.data.images.remove(image)
def node_region_rect(region, node):
location = node.location
dimensions = node.dimensions
view_to_region = region.view2d.view_to_region
bottom_left = view_to_region(
location.x, location.y - dimensions.y, clip=False)
top_right = view_to_region(
location.x + dimensions.x, location.y, clip=False)
return Rectangle(bottom_left[0], bottom_left[1], top_right[0] - bottom_left[0], top_right[1] - bottom_left[1])
def iter_node_names(tree_type):
if tree_type == 'GEOMETRY':
for cls in bpy.types.GeometryNode.__subclasses__():
if cls.__name__ in GEO_NODES_PASS:
continue
yield cls.__name__
for cls in bpy.types.FunctionNode.__subclasses__():
yield cls.__name__
elif tree_type == 'COMPOSITING':
for cls in bpy.types.CompositorNode.__subclasses__():
yield cls.__name__
elif tree_type == 'SHADER':
for cls in bpy.types.ShaderNode.__subclasses__():
yield cls.__name__
elif tree_type == 'TEXTURE':
for cls in bpy.types.TextureNode.__subclasses__():
yield cls.__name__
class MakeScreenshotsOperator(bpy.types.Operator):
bl_idname = "test.make_screenshots"
bl_label = "Make Screenshots"
def invoke(self, context, event):
context.window_manager.modal_handler_add(self)
tree_type = context.space_data.node_tree.type
self.node_names_iterator = islice(iter_node_names(tree_type), 10000)
return self.prepare_next_node(context)
def prepare_next_node(self, context):
try:
node_name = next(self.node_names_iterator)
except:
return {'FINISHED'}
node_tree = context.space_data.node_tree
for node in node_tree.nodes:
node.location.x = 10000
node = node_tree.nodes.new(node_name)
node.select = False
node.show_preview = False
self.current_node = node
self.current_name = node_name
return {'RUNNING_MODAL'}
def modal(self, context, event):
base_path = Path.home() / "Downloads" / "node_screenshots"
temp_path = str(base_path / "temp.png")
filepath = str(base_path / f"node-types_{self.current_name}.png")
bpy.ops.screen.screenshot_area(filepath=temp_path)
rect = node_region_rect(context.region, self.current_node)
margin = 15
rect.x -= margin
rect.y -= margin
rect.width += margin * 2
rect.height += margin * 2
bpy.ops.screen.screenshot_area(filepath=filepath)
cut_image(temp_path, filepath, rect)
context.area.tag_redraw()
return self.prepare_next_node(context)
bpy.utils.register_class(MakeScreenshotsOperator)
|
