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
|
# SPDX-License-Identifier: GPL-3.0-or-later
# Copyright (C) 2016-2020 by Nathan Lovato, Daniel Oakey, Razvan Radulescu, and contributors
import bpy
class InfoProgressBar:
"""
Draws a progress bar in the info header area
"""
def __init__(self, progress_min=0, progress_max=100):
assert progress_min < progress_max
self.progress_min = progress_min
self.progress_max = progress_max
self._progress = bpy.props.FloatProperty(
default=self.progress_min,
min=self.progress_min,
max=self.progress_max,
subtype="PERCENTAGE",
)
self._visible = False
def update(self, context):
for area in context.screen.areas:
if area.type == "INFO":
area.tag_redraw()
def draw(self):
if self.progress >= self.progress_max:
self.visible = False
else:
self.layout.prop(self, "_progress", text="Progress", slider=True)
@property
def progress(self):
return self._progress
@progress.setter
def progress(self, value):
self._progress = min(max(self.progress_min, value), self.progress_max)
@property
def visible(self):
return self._visible
@visible.setter
def visible(self, value):
self._visible = value
if self._visible:
bpy.types.INFO_HT_header.append(self.draw)
bpy.app.handlers.scene_update_post.add(self.update)
else:
bpy.types.INFO_HT_header.remove(self.draw)
bpy.app.handlers.scene_update_post.remove(self.update)
|