109 lines
3.9 KiB
Python
109 lines
3.9 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Flyff Character Status Node (New Version):
|
|
- Has no inputs/outputs.
|
|
- Creates an OCR region in data_collector.
|
|
- Periodically grabs raw text from that region, parses it here in the node,
|
|
and sets data_manager's HP, MP, FP, EXP accordingly.
|
|
- Also updates its own text fields with the parsed values.
|
|
"""
|
|
|
|
import re
|
|
from OdenGraphQt import BaseNode
|
|
from PyQt5.QtWidgets import QMessageBox
|
|
from Modules import data_manager, data_collector
|
|
|
|
class FlyffCharacterStatusNode(BaseNode):
|
|
__identifier__ = 'bunny-lab.io.flyff_character_status_node'
|
|
NODE_NAME = 'Flyff - Character Status'
|
|
|
|
def __init__(self):
|
|
super(FlyffCharacterStatusNode, self).__init__()
|
|
# Prevent duplicates
|
|
if data_manager.character_status_collector_exists:
|
|
QMessageBox.critical(None, "Error", "Only one Flyff Character Status Collector node is allowed.")
|
|
raise Exception("Duplicate Character Status Node.")
|
|
data_manager.character_status_collector_exists = True
|
|
|
|
# Add text fields for display
|
|
self.add_text_input('hp', 'HP', text="HP: 0/0")
|
|
self.add_text_input('mp', 'MP', text="MP: 0/0")
|
|
self.add_text_input('fp', 'FP', text="FP: 0/0")
|
|
self.add_text_input('exp', 'EXP', text="EXP: 0%")
|
|
|
|
# Create a unique region id for this node (or just "character_status")
|
|
self.region_id = "character_status"
|
|
data_collector.create_ocr_region(self.region_id, x=250, y=50, w=180, h=130)
|
|
|
|
# Start the data_collector background thread (if not already started)
|
|
data_collector.start_collector()
|
|
|
|
# Set the node name
|
|
self.set_name("Flyff - Character Status")
|
|
|
|
def parse_character_stats(self, raw_text):
|
|
"""
|
|
Extract HP, MP, FP, EXP from the raw OCR text lines.
|
|
"""
|
|
lines = [l.strip() for l in raw_text.splitlines() if l.strip()]
|
|
hp_current, hp_total = 0, 0
|
|
mp_current, mp_total = 0, 0
|
|
fp_current, fp_total = 0, 0
|
|
exp_value = 0.0
|
|
|
|
if len(lines) >= 4:
|
|
# line 1: HP
|
|
hp_match = re.search(r"(\d+)\s*/\s*(\d+)", lines[0])
|
|
if hp_match:
|
|
hp_current = int(hp_match.group(1))
|
|
hp_total = int(hp_match.group(2))
|
|
|
|
# line 2: MP
|
|
mp_match = re.search(r"(\d+)\s*/\s*(\d+)", lines[1])
|
|
if mp_match:
|
|
mp_current = int(mp_match.group(1))
|
|
mp_total = int(mp_match.group(2))
|
|
|
|
# line 3: FP
|
|
fp_match = re.search(r"(\d+)\s*/\s*(\d+)", lines[2])
|
|
if fp_match:
|
|
fp_current = int(fp_match.group(1))
|
|
fp_total = int(fp_match.group(2))
|
|
|
|
# line 4: EXP
|
|
exp_match = re.search(r"(\d+(?:\.\d+)?)", lines[3])
|
|
if exp_match:
|
|
val = float(exp_match.group(1))
|
|
if val < 0: val = 0
|
|
if val > 100: val = 100
|
|
exp_value = val
|
|
|
|
return hp_current, hp_total, mp_current, mp_total, fp_current, fp_total, exp_value
|
|
|
|
def process_input(self):
|
|
"""
|
|
Called periodically by the global timer in your main application (borealis.py).
|
|
"""
|
|
# Grab raw text from data_collector
|
|
raw_text = data_collector.get_raw_text(self.region_id)
|
|
|
|
# Parse it
|
|
hp_c, hp_t, mp_c, mp_t, fp_c, fp_t, exp_v = self.parse_character_stats(raw_text)
|
|
|
|
# Update data_manager
|
|
data_manager.set_data_bulk({
|
|
"hp_current": hp_c,
|
|
"hp_total": hp_t,
|
|
"mp_current": mp_c,
|
|
"mp_total": mp_t,
|
|
"fp_current": fp_c,
|
|
"fp_total": fp_t,
|
|
"exp": exp_v
|
|
})
|
|
|
|
# Update the node's text fields
|
|
self.set_property('hp', f"HP: {hp_c}/{hp_t}")
|
|
self.set_property('mp', f"MP: {mp_c}/{mp_t}")
|
|
self.set_property('fp', f"FP: {fp_c}/{fp_t}")
|
|
self.set_property('exp', f"EXP: {exp_v}%")
|