Fixed Timing / Data Update Logic / Low Health Overlay Issues / Comparison Node and Math Node issues.
This commit is contained in:
@ -1,177 +1,133 @@
|
||||
import time
|
||||
import sys
|
||||
from OdenGraphQt import BaseNode
|
||||
from Qt import QtWidgets, QtCore, QtGui
|
||||
#!/usr/bin/env python3
|
||||
|
||||
"""
|
||||
Standardized Flyff Low Health Alert Node:
|
||||
- Monitors an input value (1 = health alert, 0 = normal).
|
||||
- Displays a visual alert and plays a sound if enabled.
|
||||
- Uses a global update timer for processing.
|
||||
- Automatically processes float, int, and string values.
|
||||
"""
|
||||
|
||||
import time
|
||||
from OdenGraphQt import BaseNode
|
||||
from Qt import QtCore, QtWidgets, QtGui
|
||||
|
||||
# Attempt to import winsound (Windows-only)
|
||||
try:
|
||||
import winsound
|
||||
HAS_WINSOUND = True
|
||||
except ImportError:
|
||||
HAS_WINSOUND = False
|
||||
|
||||
|
||||
class OverlayCanvas(QtWidgets.QWidget):
|
||||
"""
|
||||
UI overlay for displaying a red warning box, which can be repositioned by dragging.
|
||||
"""
|
||||
|
||||
def __init__(self, parent=None):
|
||||
super().__init__(parent)
|
||||
|
||||
# **Full-screen overlay**
|
||||
screen_geo = QtWidgets.QApplication.primaryScreen().geometry()
|
||||
self.setGeometry(screen_geo) # Set to full screen
|
||||
|
||||
self.setGeometry(screen_geo)
|
||||
self.setWindowFlags(QtCore.Qt.FramelessWindowHint | QtCore.Qt.WindowStaysOnTopHint)
|
||||
self.setAttribute(QtCore.Qt.WA_TranslucentBackground, True)
|
||||
self.setAttribute(QtCore.Qt.WA_NoSystemBackground, True)
|
||||
self.setAttribute(QtCore.Qt.WA_OpaquePaintEvent, False)
|
||||
self.setAttribute(QtCore.Qt.WA_AlwaysStackOnTop, True)
|
||||
|
||||
# **Draggable Low Health Alert**
|
||||
self.helper_LowHealthAlert = QtCore.QRect(250, 300, 900, 35) # Default Position
|
||||
self.setVisible(False)
|
||||
self.helper_LowHealthAlert = QtCore.QRect(250, 300, 900, 35)
|
||||
self.dragging = False
|
||||
self.drag_offset = QtCore.QPoint()
|
||||
|
||||
self.setVisible(False) # Initially hidden
|
||||
self.drag_offset = None
|
||||
|
||||
def paintEvent(self, event):
|
||||
"""Draw the helper overlay objects."""
|
||||
if not self.isVisible():
|
||||
return # Don't draw anything if invisible
|
||||
|
||||
return
|
||||
painter = QtGui.QPainter(self)
|
||||
painter.setPen(QtCore.Qt.NoPen)
|
||||
painter.setBrush(QtGui.QColor(255, 0, 0)) # Solid red rectangle
|
||||
painter.setBrush(QtGui.QColor(255, 0, 0))
|
||||
painter.drawRect(self.helper_LowHealthAlert)
|
||||
|
||||
# Draw bold white text centered within the rectangle
|
||||
font = QtGui.QFont("Arial", 14, QtGui.QFont.Bold)
|
||||
painter.setFont(font)
|
||||
painter.setPen(QtGui.QColor(255, 255, 255))
|
||||
|
||||
text = "LOW HEALTH"
|
||||
metrics = QtGui.QFontMetrics(font)
|
||||
text_width = metrics.horizontalAdvance(text)
|
||||
text_height = metrics.height()
|
||||
text_x = self.helper_LowHealthAlert.center().x() - text_width // 2
|
||||
text_y = self.helper_LowHealthAlert.center().y() + text_height // 4
|
||||
|
||||
painter.drawText(text_x, text_y, text)
|
||||
text_x = self.helper_LowHealthAlert.center().x() - 50
|
||||
text_y = self.helper_LowHealthAlert.center().y() + 5
|
||||
painter.drawText(text_x, text_y, "LOW HEALTH")
|
||||
|
||||
def toggle_alert(self, state):
|
||||
"""
|
||||
Show or hide the overlay based on the state (1 = show, 0 = hide).
|
||||
"""
|
||||
self.setVisible(state == 1)
|
||||
self.update()
|
||||
|
||||
def mousePressEvent(self, event):
|
||||
"""Detect clicks inside the red box and allow dragging."""
|
||||
if event.button() == QtCore.Qt.LeftButton and self.helper_LowHealthAlert.contains(event.pos()):
|
||||
self.dragging = True
|
||||
self.drag_offset = event.pos() - self.helper_LowHealthAlert.topLeft()
|
||||
if event.button() == QtCore.Qt.LeftButton:
|
||||
if self.helper_LowHealthAlert.contains(event.pos()):
|
||||
self.dragging = True
|
||||
self.drag_offset = event.pos() - self.helper_LowHealthAlert.topLeft()
|
||||
super().mousePressEvent(event)
|
||||
|
||||
def mouseMoveEvent(self, event):
|
||||
"""Handle dragging movement."""
|
||||
if self.dragging:
|
||||
new_x = event.pos().x() - self.drag_offset.x()
|
||||
new_y = event.pos().y() - self.drag_offset.y()
|
||||
self.helper_LowHealthAlert.moveTopLeft(QtCore.QPoint(new_x, new_y))
|
||||
new_top_left = event.pos() - self.drag_offset
|
||||
self.helper_LowHealthAlert.moveTo(new_top_left)
|
||||
self.update()
|
||||
super().mouseMoveEvent(event)
|
||||
|
||||
def mouseReleaseEvent(self, event):
|
||||
"""Stop dragging when the mouse button is released."""
|
||||
self.dragging = False
|
||||
|
||||
if event.button() == QtCore.Qt.LeftButton:
|
||||
self.dragging = False
|
||||
super().mouseReleaseEvent(event)
|
||||
|
||||
class FlyffLowHealthAlertNode(BaseNode):
|
||||
"""
|
||||
Custom OdenGraphQt node that toggles a visual alert overlay and plays a beep when health is low.
|
||||
"""
|
||||
|
||||
__identifier__ = 'bunny-lab.io.flyff_low_health_alert_node'
|
||||
NODE_NAME = 'Flyff - Low Health Alert'
|
||||
|
||||
overlay_instance = None # Shared overlay instance
|
||||
last_beep_time = 0 # Time tracking for beep interval
|
||||
BEEP_INTERVAL_SECONDS = 2 # Beep every 2 seconds
|
||||
overlay_instance = None
|
||||
last_beep_time = 0
|
||||
BEEP_INTERVAL_SECONDS = 2
|
||||
|
||||
def __init__(self):
|
||||
super(FlyffLowHealthAlertNode, self).__init__()
|
||||
|
||||
# Create checkboxes to decide which kind of alert(s) to utilize
|
||||
self.add_checkbox('cb_1', '', 'Sound Alert', True)
|
||||
self.add_checkbox('cb_2', '', 'Visual Alert', True)
|
||||
|
||||
# Create Input Port
|
||||
self.add_input('Toggle (1 = On | 0 = Off)', color=(200, 100, 0))
|
||||
|
||||
# Add text input widget to display received value
|
||||
self.add_text_input('value', 'Current Value', text='0')
|
||||
self.add_combo_menu('beep_interval', 'Beep Interval', items=["0.5s", "1.0s", "2.0s"])
|
||||
|
||||
# Ensure only one overlay instance exists
|
||||
if not FlyffLowHealthAlertNode.overlay_instance:
|
||||
FlyffLowHealthAlertNode.overlay_instance = OverlayCanvas()
|
||||
FlyffLowHealthAlertNode.overlay_instance.show()
|
||||
|
||||
def process_input(self):
|
||||
"""
|
||||
This function runs every 500ms (via the global update loop).
|
||||
It updates the displayed value and toggles the alert if needed.
|
||||
"""
|
||||
input_port = self.input(0)
|
||||
value = input_port.connected_ports()[0].node().get_property('value') if input_port.connected_ports() else "0"
|
||||
self.receive_data(value)
|
||||
|
||||
# If there is a connected node, fetch its output value
|
||||
if input_port.connected_ports():
|
||||
connected_node = input_port.connected_ports()[0].node()
|
||||
if hasattr(connected_node, 'get_property'):
|
||||
value = connected_node.get_property('value')
|
||||
else:
|
||||
value = "0"
|
||||
else:
|
||||
value = "0" # Default to zero if nothing is connected
|
||||
|
||||
def receive_data(self, data, source_port_name=None):
|
||||
try:
|
||||
input_value = int(value) # Ensure we interpret input as an integer (0 or 1)
|
||||
if isinstance(data, str):
|
||||
data = float(data) if '.' in data else int(data)
|
||||
if isinstance(data, (float, int)):
|
||||
data = 1 if data > 1 else 0 if data <= 0 else int(data)
|
||||
else:
|
||||
data = 0
|
||||
except ValueError:
|
||||
input_value = 0 # Default to off if the input is not valid
|
||||
data = 0
|
||||
|
||||
# Update the value display box
|
||||
self.set_property('value', str(input_value))
|
||||
|
||||
# Check if the "Visual Alert" checkbox is enabled
|
||||
visual_alert_enabled = self.get_property('cb_2')
|
||||
|
||||
# Ensure that if "Visual Alert" is unchecked, the overlay is always hidden
|
||||
if not visual_alert_enabled:
|
||||
FlyffLowHealthAlertNode.overlay_instance.toggle_alert(0)
|
||||
else:
|
||||
FlyffLowHealthAlertNode.overlay_instance.toggle_alert(input_value)
|
||||
|
||||
# Check if "Sound Alert" is enabled and beep if necessary
|
||||
self.handle_beep(input_value)
|
||||
self.set_property('value', str(data))
|
||||
if self.get_property('cb_2'):
|
||||
FlyffLowHealthAlertNode.overlay_instance.toggle_alert(data)
|
||||
self.handle_beep(data)
|
||||
|
||||
def handle_beep(self, input_value):
|
||||
"""
|
||||
Plays a beep sound every 2 seconds when the value is `1` and "Sound Alert" is enabled.
|
||||
"""
|
||||
sound_alert_enabled = self.get_property('cb_1')
|
||||
current_time = time.time()
|
||||
# Update beep interval from the dropdown property
|
||||
interval_str = self.get_property('beep_interval')
|
||||
if interval_str.endswith("s"):
|
||||
interval_seconds = float(interval_str[:-1])
|
||||
else:
|
||||
interval_seconds = float(interval_str)
|
||||
self.BEEP_INTERVAL_SECONDS = interval_seconds
|
||||
|
||||
if input_value == 1 and sound_alert_enabled:
|
||||
if (current_time - FlyffLowHealthAlertNode.last_beep_time) >= FlyffLowHealthAlertNode.BEEP_INTERVAL_SECONDS:
|
||||
if input_value == 1 and self.get_property('cb_1'):
|
||||
current_time = time.time()
|
||||
if (current_time - FlyffLowHealthAlertNode.last_beep_time) >= self.BEEP_INTERVAL_SECONDS:
|
||||
FlyffLowHealthAlertNode.last_beep_time = current_time
|
||||
self.play_beep()
|
||||
else:
|
||||
FlyffLowHealthAlertNode.last_beep_time = 0 # Reset when health is safe
|
||||
|
||||
def play_beep(self):
|
||||
"""
|
||||
Plays a beep using `winsound.Beep` (Windows) or prints a terminal bell (`\a`).
|
||||
"""
|
||||
if HAS_WINSOUND:
|
||||
winsound.Beep(376, 100) # 376 Hz, 100ms duration
|
||||
winsound.Beep(376, 100)
|
||||
else:
|
||||
print('\a', end='') # Terminal bell for non-Windows systems
|
||||
print('\a', end='')
|
||||
|
||||
Reference in New Issue
Block a user