def add_flash_animation(widget: QWidget,
duration: int = 300,
color: Array = (0.5, 0.5, 0.5, 0.5)):
"""Add flash animation to widget to highlight certain action (e.g. taking a screenshot).
Parameters
----------
widget : QWidget
Any Qt widget.
duration : int
Duration of the flash animation.
color : Array
Color of the flash animation. By default, we use light gray.
"""
color = transform_color(color)[0]
color = (255 * color).astype("int")
effect = QGraphicsColorizeEffect(widget)
widget.setGraphicsEffect(effect)
widget._flash_animation = QPropertyAnimation(effect, b"color")
widget._flash_animation.setStartValue(QColor(0, 0, 0, 0))
widget._flash_animation.setEndValue(QColor(0, 0, 0, 0))
widget._flash_animation.setLoopCount(1)
# let's make sure to remove the animation from the widget because
# if we don't, the widget will actually be black and white.
widget._flash_animation.finished.connect(
partial(remove_flash_animation, weakref.ref(widget)))
widget._flash_animation.start()
# now set an actual time for the flashing and an intermediate color
widget._flash_animation.setDuration(duration)
widget._flash_animation.setKeyValueAt(0.1, QColor(*color))
