Clean up for git storage
Add shebang dos2unix - This is primary a Windows tool, so oops
This commit is contained in:
parent
a5463094e2
commit
9ac2bf04dc
1 changed files with 224 additions and 222 deletions
446
main.pyw
Normal file → Executable file
446
main.pyw
Normal file → Executable file
|
|
@ -1,222 +1,224 @@
|
||||||
import json
|
#!/usr/bin/env python
|
||||||
import os
|
|
||||||
import sys
|
import json
|
||||||
import pygame
|
import os
|
||||||
import time
|
import sys
|
||||||
|
import pygame
|
||||||
|
import time
|
||||||
class Visualizer:
|
|
||||||
def __init__(self, profile):
|
|
||||||
# Change to script dir to allow loading of configs relative to script
|
class Visualizer:
|
||||||
if getattr(sys, 'frozen', False):
|
def __init__(self, profile):
|
||||||
# The application is frozen, figure out the path of the EXE
|
# Change to script dir to allow loading of configs relative to script
|
||||||
pathname = os.path.dirname(sys.executable)
|
if getattr(sys, 'frozen', False):
|
||||||
else:
|
# The application is frozen, figure out the path of the EXE
|
||||||
# The application is not frozen, get PWD
|
pathname = os.path.dirname(sys.executable)
|
||||||
pathname = os.path.dirname(__file__)
|
else:
|
||||||
if pathname:
|
# The application is not frozen, get PWD
|
||||||
os.chdir(pathname)
|
pathname = os.path.dirname(__file__)
|
||||||
|
if pathname:
|
||||||
# Load config
|
os.chdir(pathname)
|
||||||
self.config = self.load_config(profile)["profile"]
|
|
||||||
|
# Load config
|
||||||
# initialize the screen
|
self.config = self.load_config(profile)["profile"]
|
||||||
self.win_size = self.config["size"]
|
|
||||||
pygame.init()
|
# initialize the screen
|
||||||
self.screen = pygame.display.set_mode(self.config["size"], pygame.RESIZABLE)
|
self.win_size = self.config["size"]
|
||||||
pygame.display.set_caption("SRTK Gamepad Visualizer: %s" % profile)
|
pygame.init()
|
||||||
|
self.screen = pygame.display.set_mode(self.config["size"], pygame.RESIZABLE)
|
||||||
# Load and prep the background
|
pygame.display.set_caption("SRTK Gamepad Visualizer: %s" % profile)
|
||||||
background_file = os.path.join("profiles", profile, self.config["background_image"])
|
|
||||||
self.background = pygame.image.load(background_file).convert_alpha()
|
# Load and prep the background
|
||||||
|
background_file = os.path.join("profiles", profile, self.config["background_image"])
|
||||||
# Create the buffer image to draw on, set to the size of the background image
|
self.background = pygame.image.load(background_file).convert_alpha()
|
||||||
self.buffer = pygame.surface.Surface(self.background.get_size())
|
|
||||||
|
# Create the buffer image to draw on, set to the size of the background image
|
||||||
# Create the font
|
self.buffer = pygame.surface.Surface(self.background.get_size())
|
||||||
self.text_font = pygame.font.SysFont(self.config["font_face"], self.config["font_size"])
|
|
||||||
self.font_color = self.config["font_color"]
|
# Create the font
|
||||||
self.text_pos = self.config["text_pos"]
|
self.text_font = pygame.font.SysFont(self.config["font_face"], self.config["font_size"])
|
||||||
|
self.font_color = self.config["font_color"]
|
||||||
# APS counter
|
self.text_pos = self.config["text_pos"]
|
||||||
self.actions = []
|
|
||||||
self.aps_min = self.config["aps_minimum"]
|
# APS counter
|
||||||
self.aps = 0
|
self.actions = []
|
||||||
self.aps_throttle = 3
|
self.aps_min = self.config["aps_minimum"]
|
||||||
|
self.aps = 0
|
||||||
# Set up our inputs
|
self.aps_throttle = 3
|
||||||
# Inputs are defined as key = (light position, light surface object)
|
|
||||||
self.active_inputs = set()
|
# Set up our inputs
|
||||||
self.last_inputs = set()
|
# Inputs are defined as key = (light position, light surface object)
|
||||||
self.inputs = {}
|
self.active_inputs = set()
|
||||||
|
self.last_inputs = set()
|
||||||
inputs = self.config["inputs"]
|
self.inputs = {}
|
||||||
for i in inputs:
|
|
||||||
if "enabled" in inputs[i] and not inputs[i]["enabled"]:
|
inputs = self.config["inputs"]
|
||||||
continue
|
for i in inputs:
|
||||||
|
if "enabled" in inputs[i] and not inputs[i]["enabled"]:
|
||||||
pos = inputs[i]["pos"]
|
continue
|
||||||
size = inputs[i]["size"]
|
|
||||||
color = inputs[i]["color"] if "color" in inputs[i] else self.config["default_color"]
|
pos = inputs[i]["pos"]
|
||||||
|
size = inputs[i]["size"]
|
||||||
img = pygame.Surface(size)
|
color = inputs[i]["color"] if "color" in inputs[i] else self.config["default_color"]
|
||||||
img.fill(color)
|
|
||||||
|
img = pygame.Surface(size)
|
||||||
try:
|
img.fill(color)
|
||||||
i = int(i)
|
|
||||||
except:
|
try:
|
||||||
pass
|
i = int(i)
|
||||||
|
except:
|
||||||
self.inputs[i] = (pos, img)
|
pass
|
||||||
|
|
||||||
def load_config(self, profile):
|
self.inputs[i] = (pos, img)
|
||||||
infile_name = os.path.join("profiles", profile, "config.txt")
|
|
||||||
infile = open(infile_name, "r")
|
def load_config(self, profile):
|
||||||
config = json.load(infile)
|
infile_name = os.path.join("profiles", profile, "config.txt")
|
||||||
|
infile = open(infile_name, "r")
|
||||||
return config
|
config = json.load(infile)
|
||||||
|
|
||||||
def button_down(self, button_id):
|
return config
|
||||||
self.active_inputs.add(button_id)
|
|
||||||
self.actions.append(time.time())
|
def button_down(self, button_id):
|
||||||
|
self.active_inputs.add(button_id)
|
||||||
def button_up(self, button_id):
|
self.actions.append(time.time())
|
||||||
try:
|
|
||||||
self.active_inputs.remove(button_id)
|
def button_up(self, button_id):
|
||||||
except:
|
try:
|
||||||
pass
|
self.active_inputs.remove(button_id)
|
||||||
|
except:
|
||||||
def axis_change(self, axis, value):
|
pass
|
||||||
if value > 0.1:
|
|
||||||
self.active_inputs.add("A%dP" % axis)
|
def axis_change(self, axis, value):
|
||||||
elif value < -0.1:
|
if value > 0.1:
|
||||||
self.active_inputs.add("A%dM" % axis)
|
self.active_inputs.add("A%dP" % axis)
|
||||||
else:
|
elif value < -0.1:
|
||||||
for a in ["A%dM" % axis, "A%dP" % axis]:
|
self.active_inputs.add("A%dM" % axis)
|
||||||
try:
|
else:
|
||||||
self.active_inputs.remove(a)
|
for a in ["A%dM" % axis, "A%dP" % axis]:
|
||||||
except:
|
try:
|
||||||
pass
|
self.active_inputs.remove(a)
|
||||||
|
except:
|
||||||
def hat_change(self, hat, value):
|
pass
|
||||||
# H0U, H0D, H0L, H0R
|
|
||||||
if value[0] < 0:
|
def hat_change(self, hat, value):
|
||||||
self.active_inputs.add("H%dL" % hat)
|
# H0U, H0D, H0L, H0R
|
||||||
elif value[0] > 0:
|
if value[0] < 0:
|
||||||
self.active_inputs.add("H%dR" % hat)
|
self.active_inputs.add("H%dL" % hat)
|
||||||
else:
|
elif value[0] > 0:
|
||||||
for a in ["H%dL" % hat, "H%dR" % hat]:
|
self.active_inputs.add("H%dR" % hat)
|
||||||
try:
|
else:
|
||||||
self.active_inputs.remove(a)
|
for a in ["H%dL" % hat, "H%dR" % hat]:
|
||||||
except:
|
try:
|
||||||
pass
|
self.active_inputs.remove(a)
|
||||||
|
except:
|
||||||
if value[1] < 0:
|
pass
|
||||||
self.active_inputs.add("H%dD" % hat)
|
|
||||||
elif value[1] > 0:
|
if value[1] < 0:
|
||||||
self.active_inputs.add("H%dU" % hat)
|
self.active_inputs.add("H%dD" % hat)
|
||||||
else:
|
elif value[1] > 0:
|
||||||
for a in ["H%dU" % hat, "H%dD" % hat]:
|
self.active_inputs.add("H%dU" % hat)
|
||||||
try:
|
else:
|
||||||
self.active_inputs.remove(a)
|
for a in ["H%dU" % hat, "H%dD" % hat]:
|
||||||
except:
|
try:
|
||||||
pass
|
self.active_inputs.remove(a)
|
||||||
|
except:
|
||||||
def update(self):
|
pass
|
||||||
"""
|
|
||||||
Handles screen drawing. Blits images to a buffer image, resizes the
|
def update(self):
|
||||||
buffer to the size of the screen, then blits the buffer to the screen.
|
"""
|
||||||
"""
|
Handles screen drawing. Blits images to a buffer image, resizes the
|
||||||
if self.active_inputs == self.last_inputs:
|
buffer to the size of the screen, then blits the buffer to the screen.
|
||||||
return None
|
"""
|
||||||
|
if self.active_inputs == self.last_inputs:
|
||||||
self.last_inputs = set(self.active_inputs)
|
return None
|
||||||
|
|
||||||
buf = self.buffer
|
self.last_inputs = set(self.active_inputs)
|
||||||
screen = self.screen
|
|
||||||
|
buf = self.buffer
|
||||||
# Blank the buffer, draw the button hilights
|
screen = self.screen
|
||||||
buf.fill(self.config["background_color"])
|
|
||||||
|
# Blank the buffer, draw the button hilights
|
||||||
for i in self.active_inputs:
|
buf.fill(self.config["background_color"])
|
||||||
try:
|
|
||||||
loc, img = self.inputs[i]
|
for i in self.active_inputs:
|
||||||
buf.blit(img, loc)
|
try:
|
||||||
except KeyError:
|
loc, img = self.inputs[i]
|
||||||
pass
|
buf.blit(img, loc)
|
||||||
|
except KeyError:
|
||||||
# Draw the background overtop as a mask
|
pass
|
||||||
buf.blit(self.background, (0, 0))
|
|
||||||
|
# Draw the background overtop as a mask
|
||||||
# Draw APS
|
buf.blit(self.background, (0, 0))
|
||||||
# Throttle keeps us from "Vibrating number syndrome"
|
|
||||||
self.aps_throttle -= 1
|
# Draw APS
|
||||||
if self.aps_throttle < 1:
|
# Throttle keeps us from "Vibrating number syndrome"
|
||||||
self.aps_throttle = 3
|
self.aps_throttle -= 1
|
||||||
|
if self.aps_throttle < 1:
|
||||||
# Only score APS off the last 30 actions and last 5s
|
self.aps_throttle = 3
|
||||||
while self.actions and self.actions[0] < (time.time() - 5):
|
|
||||||
self.actions.pop(0)
|
# Only score APS off the last 30 actions and last 5s
|
||||||
if len(self.actions) > 30:
|
while self.actions and self.actions[0] < (time.time() - 5):
|
||||||
self.actions = self.actions[-30:]
|
self.actions.pop(0)
|
||||||
|
if len(self.actions) > 30:
|
||||||
# Time from last action to current time, to get APS
|
self.actions = self.actions[-30:]
|
||||||
# We want at least 5 actions before we consider drawing APS
|
|
||||||
if len(self.actions) > 5:
|
# Time from last action to current time, to get APS
|
||||||
aps_len = time.time() - self.actions[0] + 0.01
|
# We want at least 5 actions before we consider drawing APS
|
||||||
self.aps = len(self.actions) / aps_len
|
if len(self.actions) > 5:
|
||||||
else:
|
aps_len = time.time() - self.actions[0] + 0.01
|
||||||
self.aps = 0
|
self.aps = len(self.actions) / aps_len
|
||||||
|
else:
|
||||||
# We only care about APS if we're mashing, so only draw it if it's over aps_min
|
self.aps = 0
|
||||||
if self.aps > self.aps_min:
|
|
||||||
aps_surf = self.text_font.render("%0.1f" % self.aps, 1, self.font_color)
|
# We only care about APS if we're mashing, so only draw it if it's over aps_min
|
||||||
buf.blit(aps_surf, self.config["text_pos"])
|
if self.aps > self.aps_min:
|
||||||
|
aps_surf = self.text_font.render("%0.1f" % self.aps, 1, self.font_color)
|
||||||
# Scale the buffer to the screen size if necessary
|
buf.blit(aps_surf, self.config["text_pos"])
|
||||||
if buf.get_size() != screen.get_size():
|
|
||||||
buf = pygame.transform.scale(buf, screen.get_size())
|
# Scale the buffer to the screen size if necessary
|
||||||
|
if buf.get_size() != screen.get_size():
|
||||||
# Blit buffer to screen
|
buf = pygame.transform.scale(buf, screen.get_size())
|
||||||
screen.blit(buf, (0, 0))
|
|
||||||
pygame.display.flip()
|
# Blit buffer to screen
|
||||||
|
screen.blit(buf, (0, 0))
|
||||||
if __name__ == "__main__":
|
pygame.display.flip()
|
||||||
# Read in a profile name from commandline
|
|
||||||
if len(sys.argv) == 2:
|
if __name__ == "__main__":
|
||||||
profile = sys.argv[1]
|
# Read in a profile name from commandline
|
||||||
else:
|
if len(sys.argv) == 2:
|
||||||
profile = "universal"
|
profile = sys.argv[1]
|
||||||
|
else:
|
||||||
# Initialize all our joysticks so we get the one we want
|
profile = "universal"
|
||||||
# Since this is a one-player app, this should be fine...
|
|
||||||
pygame.joystick.init()
|
# Initialize all our joysticks so we get the one we want
|
||||||
joysticks = [pygame.joystick.Joystick(x) for x in range(pygame.joystick.get_count())]
|
# Since this is a one-player app, this should be fine...
|
||||||
for joy in joysticks:
|
pygame.joystick.init()
|
||||||
joy.init()
|
joysticks = [pygame.joystick.Joystick(x) for x in range(pygame.joystick.get_count())]
|
||||||
|
for joy in joysticks:
|
||||||
# Initialize our visualizer context
|
joy.init()
|
||||||
vis = Visualizer(profile)
|
|
||||||
|
# Initialize our visualizer context
|
||||||
quit = False
|
vis = Visualizer(profile)
|
||||||
|
|
||||||
while not quit:
|
quit = False
|
||||||
|
|
||||||
# Note that we wait for pygame events before doing any redraws
|
while not quit:
|
||||||
# This limits redraws to when button states change, normally
|
|
||||||
event = pygame.event.wait()
|
# Note that we wait for pygame events before doing any redraws
|
||||||
if event.type == pygame.VIDEORESIZE:
|
# This limits redraws to when button states change, normally
|
||||||
vis.win_size = (event.w, event.h)
|
event = pygame.event.wait()
|
||||||
vis.screen = pygame.display.set_mode(vis.win_size, pygame.RESIZABLE)
|
if event.type == pygame.VIDEORESIZE:
|
||||||
elif event.type == pygame.QUIT:
|
vis.win_size = (event.w, event.h)
|
||||||
quit = True
|
vis.screen = pygame.display.set_mode(vis.win_size, pygame.RESIZABLE)
|
||||||
elif event.type == pygame.JOYBUTTONDOWN:
|
elif event.type == pygame.QUIT:
|
||||||
vis.button_down(event.button)
|
quit = True
|
||||||
elif event.type == pygame.JOYBUTTONUP:
|
elif event.type == pygame.JOYBUTTONDOWN:
|
||||||
vis.button_up(event.button)
|
vis.button_down(event.button)
|
||||||
elif event.type == pygame.JOYAXISMOTION:
|
elif event.type == pygame.JOYBUTTONUP:
|
||||||
vis.axis_change(event.axis, event.value)
|
vis.button_up(event.button)
|
||||||
elif event.type == pygame.JOYHATMOTION:
|
elif event.type == pygame.JOYAXISMOTION:
|
||||||
vis.hat_change(event.hat, event.value)
|
vis.axis_change(event.axis, event.value)
|
||||||
|
elif event.type == pygame.JOYHATMOTION:
|
||||||
vis.update()
|
vis.hat_change(event.hat, event.value)
|
||||||
|
|
||||||
|
vis.update()
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue