historical/toontown-classic.git/panda/direct/directdevices/DirectRadamec.py
2024-01-16 11:20:27 -06:00

83 lines
2.9 KiB
Python

""" Class used to create and control radamec device """
from math import *
from direct.showbase.DirectObject import DirectObject
from .DirectDeviceManager import *
from direct.directnotify import DirectNotifyGlobal
"""
TODO:
Handle interaction between widget, followSelectedTask and updateTask
"""
# ANALOGS
RAD_PAN = 0
RAD_TILT = 1
RAD_ZOOM = 2
RAD_FOCUS = 3
class DirectRadamec(DirectObject):
radamecCount = 0
notify = DirectNotifyGlobal.directNotify.newCategory('DirectRadamec')
def __init__(self, device = 'Analog0', nodePath = None):
# See if device manager has been initialized
if base.direct.deviceManager == None:
base.direct.deviceManager = DirectDeviceManager()
# Set name
self.name = 'Radamec-' + repr(DirectRadamec.radamecCount)
DirectRadamec.radamecCount += 1
# Get analogs
self.device = device
self.analogs = base.direct.deviceManager.createAnalogs(self.device)
self.numAnalogs = len(self.analogs)
self.aList = [0, 0, 0, 0, 0, 0, 0, 0]
# Radamec device max/mins
# Note: These values change quite often, i.e. everytime
# you unplug the radamec cords, or jostle them too
# much. For best results, re-record these values often.
self.minRange = [-180.0, -90, 522517.0, 494762.0]
self.maxRange = [180.0, 90, 547074.0, 533984.0]
# Spawn update task
self.enable()
def enable(self):
# Kill existing task
self.disable()
# Update task
taskMgr.add(self.updateTask, self.name + '-updateTask')
def disable(self):
taskMgr.remove(self.name + '-updateTask')
def destroy(self):
self.disable()
def updateTask(self, state):
# Update analogs
for i in range(len(self.analogs)):
self.aList[i] = self.analogs.getControlState(i)
return Task.cont
def radamecDebug(self):
panVal = self.normalizeChannel(RAD_PAN, -180, 180)
tiltVal = self.normalizeChannel(RAD_TILT, -90, 90)
self.notify.debug("PAN = %s" % self.aList[RAD_PAN])
self.notify.debug("TILT = %s" % self.aList[RAD_TILT])
self.notify.debug("ZOOM = %s" % self.aList[RAD_ZOOM])
self.notify.debug("FOCUS = %s" % self.aList[RAD_FOCUS])
self.notify.debug("Normalized: panVal: %s tiltVal: %s" % (panVal, tiltVal))
# Normalize to the range [-minVal, maxVal] based on some hard-coded
# max/min numbers of the Radamec device
def normalizeChannel(self, chan, minVal = -1, maxVal = 1):
try:
maxRange = self.maxRange[chan]
minRange = self.minRange[chan]
except IndexError:
raise RuntimeError("can't normalize this channel (channel %d)" % chan)
range = maxRange - minRange
clampedVal = CLAMP(self.aList[chan], minRange, maxRange)
return ((maxVal - minVal) * (clampedVal - minRange) / range) + minVal