Fixed a bunch of crap
This commit is contained in:
36
Nodes/average_node.py
Normal file
36
Nodes/average_node.py
Normal file
@ -0,0 +1,36 @@
|
||||
from NodeGraphQt import BaseNode
|
||||
|
||||
class AverageNode(BaseNode):
|
||||
"""
|
||||
Average Node:
|
||||
- Inputs: A, B, C (adjustable as needed)
|
||||
- Output: Result (the average of the inputs)
|
||||
"""
|
||||
__identifier__ = 'io.github.nicole.average'
|
||||
NODE_NAME = 'Average'
|
||||
|
||||
def __init__(self):
|
||||
super(AverageNode, self).__init__()
|
||||
self.values = {} # Ensure values is a dictionary.
|
||||
self.add_input('A')
|
||||
self.add_input('B')
|
||||
self.add_input('C')
|
||||
self.add_output('Result')
|
||||
self.value = 0
|
||||
self.set_name("Average: 0")
|
||||
|
||||
def process_input(self):
|
||||
values = []
|
||||
for port_name in ['A', 'B', 'C']:
|
||||
port = self.input(port_name)
|
||||
connected = port.connected_ports() if port is not None else []
|
||||
if connected:
|
||||
connected_port = connected[0]
|
||||
parent_node = connected_port.node()
|
||||
try:
|
||||
values.append(float(getattr(parent_node, 'value', 0)))
|
||||
except (ValueError, TypeError):
|
||||
pass
|
||||
avg = sum(values) / len(values) if values else 0
|
||||
self.value = avg
|
||||
self.set_name(f"Average: {avg}")
|
Reference in New Issue
Block a user