pixel/BeatSaberCeiling
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

small performance improvement

Browse Source 
added milliseconds limit for frame rendering speed
removed redundant if blocks
This commit is contained in:
Elektrospy 2019-01-17 22:54:57 +01:00
parent fc4cd6353b
commit 39b848e433
1 changed files with 80 additions and 95 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

175
beatsaberceiling.py
View File

@ -1,4 +1,5 @@
#!/usr/bin/env python3
import websocket
try:
import thread
@ -29,7 +30,9 @@ except:
# get input parameter
beatsaber_host = '127.0.0.1'
try:
beatsaber_host = int(sys.argv[3])
new_host = str(sys.argv[3])
if new_host != "":
beatsaber_host = new_host
except:
pass
@ -54,10 +57,7 @@ class Layer:
self.png_image_data_flatten = self.flatten_rgb_array_from_png(self.png_image_data)
# internal event states
self.layer_current_event_value = 0
self.layer_event_triggered = False
self.animation_finished = True
self.animation_update_limit_ms = 1
self.animation_last_update_ms = self.millis()
self.animation_current_brightness = self.max_brightness
# just debug stuff
# print("created layer for " + self.path)
@ -71,18 +71,10 @@ class Layer:
self.layer_buffer_current_color[2]
)
def millis(self):
dt = datetime.now()
millis = dt.microsecond / 1000
return int(round((time_.time() * 1000) + int(millis)))
def run_next_fade_color(self):
if not self.animation_finished and self.animation_current_brightness > 0:
current_millis = self.millis()
if current_millis - self.animation_last_update_ms >= self.animation_update_limit_ms:
# fade rgb values
self.animation_current_brightness -= 10
self.animation_last_update_ms = self.millis()
# fade rgb values
self.animation_current_brightness -= 25
# set current layer color brightness to animation brightness
for i in range(0, 2):
@ -90,14 +82,9 @@ class Layer:
self.layer_buffer_current_color[i],
self.animation_current_brightness
)
elif self.animation_finished:
# if animation is finished, reset current fade brightness
self.animation_current_brightness = self.max_brightness
# print("animation finished!")
def trigger_event(self, event_value):
# print("new event " + self.path + " value: " + str(event_value))
self.layer_event_triggered = True
# get new rgb color for frame, based on event
self.layer_buffer_current_color = self.get_current_frame_rgb_color(event_value)
@ -108,45 +95,31 @@ class Layer:
# create black rgb color array
output_rgb_array = np.zeros(self.color_values, np.uint8)
if self.layer_event_triggered:
# if event_value == 0:
# print("Layer off!")
if event_value == 1 or event_value == 2:
# static blue light
output_rgb_array[2] = self.max_brightness
self.animation_finished = True
elif event_value == 3:
# fade blue light
output_rgb_array[2] = self.max_brightness
# reset current brightness of color for new animation
self.animation_current_brightness = self.max_brightness
self.animation_finished = False
# output_rgb_array[2] = self.max_brightness
elif event_value == 5 or event_value == 6:
# static red light
output_rgb_array[0] = self.max_brightness
self.animation_finished = True
elif event_value == 7:
# fade red light
output_rgb_array[0] = self.max_brightness
# reset current brightness of color for new animation
self.animation_current_brightness = self.max_brightness
self.animation_finished = False
# output_rgb_array[0] = self.max_brightness
# elif event_value != 0 and event_value > 7:
# print("unknown event effect value")
if event_value == 1 or event_value == 2:
# static blue light
output_rgb_array[2] = self.max_brightness
self.animation_finished = True
elif event_value == 3:
# fade blue light
output_rgb_array[2] = self.max_brightness
# reset current brightness of color for new animation
self.animation_current_brightness = self.max_brightness
self.animation_finished = False
output_rgb_array[2] = self.max_brightness
elif event_value == 5 or event_value == 6:
# static red light
output_rgb_array[0] = self.max_brightness
self.animation_finished = True
elif event_value == 7:
# fade red light
output_rgb_array[0] = self.max_brightness
# reset current brightness of color for new animation
self.animation_current_brightness = self.max_brightness
self.animation_finished = False
output_rgb_array[0] = self.max_brightness
return output_rgb_array
# only used for debbuging
def print_png_values(self):
output_string = ""
# get color values
for x in self.png_image_data_flatten:
output_string += str(x)
output_string += ","
# print(output_string)
def render_rgb_layer(self, r=0, g=0, b=0):
# reset layer buffer, step by number of color values (default 3, rgb)
for i in range(0, self.rgb_buffer_length):
@ -157,10 +130,10 @@ class Layer:
# if true to mapping, else ignore and keep it at zero (black)
if self.png_image_data_flatten[i] != 0:
self.layer_buffer_array[i] = self.get_new_color_value(r, self.png_image_data_flatten[i])
if self.png_image_data_flatten[i+1] != 0:
self.layer_buffer_array[i+1] = self.get_new_color_value(g, self.png_image_data_flatten[i+1])
if self.png_image_data_flatten[i+2] != 0:
self.layer_buffer_array[i+2] = self.get_new_color_value(b, self.png_image_data_flatten[i+2])
if self.png_image_data_flatten[i + 1] != 0:
self.layer_buffer_array[i + 1] = self.get_new_color_value(g, self.png_image_data_flatten[i + 1])
if self.png_image_data_flatten[i + 2] != 0:
self.layer_buffer_array[i + 2] = self.get_new_color_value(b, self.png_image_data_flatten[i + 2])
# helper method to calculate new color value
# based on current value and target mapping value
@ -184,14 +157,6 @@ class Layer:
return output_array
# create layer objects for the lights
layer_small = Layer('maps/small.png')
layer_middle = Layer('maps/middle.png')
layer_big = Layer('maps/big.png')
layer_left = Layer('maps/left.png')
layer_right = Layer('maps/right.png')
# ------------------------------------
# Websocket event part start
# ------------------------------------
@ -205,7 +170,7 @@ def parse_json(input_json):
elif current_event == "beatmapEvent":
event_beat_map(json_content["beatmapEvent"])
# else:
# print("other event: " + current_event)
# print("other event: " + current_event)
def event_note_cut(note_cut_object):
@ -251,7 +216,7 @@ def event_beat_map(event_object):
def event_beat_map_parse(beatmap_event_object):
event_type = beatmap_event_object["type"]
event_value = beatmap_event_object["value"]
if event_type >= 0 and event_type < 5:
if 0 <= event_type < 5:
if event_type == 0:
trigger_light_small(event_value)
elif event_type == 1:
@ -304,13 +269,10 @@ def on_close(ws):
def on_open(ws):
def run(*args):
while True:
time_.sleep(1)
# print("websocket connected")
thread.start_new_thread(run, ())
# ------------------------------------
# Websocket event part end
# ------------------------------------
@ -323,9 +285,6 @@ def get_higher_color_value(value1, value2):
return new_value
buffer_length = Nx * Ny * 3
def merge_layer_arrays(layer_first, layer_second):
output_array = np.zeros(buffer_length, np.uint8)
for idx in range(0, buffer_length):
@ -343,23 +302,49 @@ def get_merged_layers_array():
return output_array
# Start websocket logic thread
ws = websocket.WebSocketApp('ws://' + beatsaber_host + ':6557/socket',
on_message=on_message,
on_error=on_error,
on_close=on_close)
ws.on_open = on_open
wst = threading.Thread(target=ws.run_forever)
wst.daemon = True
wst.start()
def millis():
dt = datetime.now()
cur_millis = dt.microsecond / 1000
return int(round((time_.time() * 1000) + int(cur_millis)))
# Start layer logic loop
while True:
layer_small.run()
layer_middle.run()
layer_big.run()
layer_left.run()
layer_right.run()
output_buffer = get_merged_layers_array()
output_bytes = output_buffer.tobytes()
os.write(1, output_bytes)
if __name__ == "__main__":
buffer_length = Nx * Ny * 3
# create layer objects for the lights
layer_small = Layer('maps/small.png')
layer_middle = Layer('maps/middle.png')
layer_big = Layer('maps/big.png')
layer_left = Layer('maps/left.png')
layer_right = Layer('maps/right.png')
# Start websocket logic thread
ws = websocket.WebSocketApp('ws://' + beatsaber_host + ':6557/socket',
on_message=on_message,
on_error=on_error,
on_close=on_close,
on_open=on_open)
wst = threading.Thread(target=ws.run_forever)
wst.daemon = True
wst.start()
# frame refresh time limit, default ~16ms for about ~60fps
frame_update_limit_ms = 16
frame_last_update_ms = millis()
frame_current_millis = millis()
# Start layer logic loop
while True:
frame_current_millis = millis()
if frame_current_millis - frame_last_update_ms >= frame_update_limit_ms:
# run layer render logic
layer_small.run()
layer_middle.run()
layer_big.run()
layer_left.run()
layer_right.run()
# merge layers together
output_buffer = get_merged_layers_array()
# convert frame buffer to byte string and write to stdout
os.write(1, output_buffer.tobytes())
# refresh last frame update time
frame_last_update_ms = millis()