148 lines
5.0 KiB
Python
148 lines
5.0 KiB
Python
from vectorscope import Vectorscope
|
|
from dds import DDS
|
|
|
|
import gc
|
|
from vos_debug import debug_print as debug
|
|
import vectoros
|
|
import vos_state
|
|
import vos_debug
|
|
import keyboardcb
|
|
import keyleds
|
|
import keyboardio
|
|
import asyncio
|
|
|
|
_abort=False
|
|
## map waveform types to LEDs
|
|
_waves_lookup = {0:"sine", 3:"sawtooth", 1:"square", 2:"triangle"}
|
|
_waves_reverse_lookup = {"sine":0, "sawtooth":3, "square":1, "triangle":2}
|
|
|
|
lissajous_state = {
|
|
"selected_axis":0,
|
|
"selected_waveform":0,
|
|
"waves_leds":[0,0]
|
|
|
|
}
|
|
|
|
async def do_dds_loop(d):
|
|
while not _abort:
|
|
for i in range(50):
|
|
d.do_dds()
|
|
d.populate_buffer()
|
|
await asyncio.sleep(0)
|
|
|
|
def do_abort(key):
|
|
global _abort
|
|
_abort=True
|
|
|
|
|
|
async def vos_main():
|
|
|
|
vectoros.get_screen().idle()
|
|
gc.collect()
|
|
vos_state.gc_suspend=True
|
|
# await asyncio.sleep(1)
|
|
|
|
keyboardio.KeyboardIO.leds = (1<<7)
|
|
keyboardio.KeyboardIO.leds |= (1<<5) ## sine wave
|
|
keyboardio.KeyboardIO.scan()
|
|
|
|
v = Vectorscope()
|
|
d = DDS(v)
|
|
d.increment = [1500, 1200]
|
|
d.amplitude=[0.5, 0.5]
|
|
d.recalculate_waveforms()
|
|
|
|
def toggle_xy(key):
|
|
## not right yet
|
|
global lissajous_state
|
|
global d
|
|
if lissajous_state["selected_axis"] == 0:
|
|
lissajous_state["selected_axis"] = 1
|
|
keyboardio.KeyboardIO.leds |= (1<<6)
|
|
keyboardio.KeyboardIO.leds &= ~(1<<7)
|
|
else:
|
|
lissajous_state["selected_axis"] = 0
|
|
keyboardio.KeyboardIO.leds |= (1<<7)
|
|
keyboardio.KeyboardIO.leds &= ~(1<<6)
|
|
|
|
## Update leds to reflect switch
|
|
keyboardio.KeyboardIO.leds &= (0b11000011)
|
|
which_led = lissajous_state["waves_leds"][lissajous_state["selected_axis"]]
|
|
keyboardio.KeyboardIO.leds |= (1<<(5-which_led))
|
|
|
|
def toggle_waveform(key):
|
|
global lissajous_state
|
|
which_led = lissajous_state["waves_leds"][lissajous_state["selected_axis"]]
|
|
## clear leds
|
|
keyboardio.KeyboardIO.leds &= (0b11000011)
|
|
## update led
|
|
which_led = ( which_led + 1) % 4
|
|
keyboardio.KeyboardIO.leds |= (1<<(5-which_led ))
|
|
## update storage
|
|
lissajous_state["waves_leds"][lissajous_state["selected_axis"]] = which_led
|
|
## update waveform
|
|
d.waveform[lissajous_state["selected_axis"]] = _waves_lookup[which_led]
|
|
d.recalculate_waveforms()
|
|
|
|
def handle_joystick_up(key):
|
|
current_keys = keyboardcb.KeyboardCB.current_keys
|
|
if len(current_keys) == 1: ## just joystick
|
|
d.increment[1] = int(d.increment[1] * 1.1)
|
|
if keyleds.KEY_RANGE in current_keys:
|
|
d.amplitude[1] = d.amplitude[1] * 1.1
|
|
d.recalculate_waveforms()
|
|
## increase amplitude Y
|
|
if keyleds.KEY_LEVEL in current_keys:
|
|
d.phase_increment[1] = d.phase_increment[1] + 1
|
|
## increase phase Y
|
|
|
|
def handle_joystick_down(key):
|
|
current_keys = keyboardcb.KeyboardCB.current_keys
|
|
if len(current_keys) == 1: ## just joystick
|
|
d.increment[1] = int(d.increment[1] * 0.91)
|
|
if keyleds.KEY_RANGE in current_keys:
|
|
d.amplitude[1] = d.amplitude[1] * 0.91
|
|
d.recalculate_waveforms()
|
|
## increase amplitude Y
|
|
if keyleds.KEY_LEVEL in current_keys:
|
|
d.phase_increment[1] = d.phase_increment[1] - 1
|
|
## increase phase Y
|
|
def handle_joystick_right(key):
|
|
current_keys = keyboardcb.KeyboardCB.current_keys
|
|
if len(current_keys) == 1: ## just joystick
|
|
d.increment[0] = int(d.increment[0] * 1.1)
|
|
if keyleds.KEY_RANGE in current_keys:
|
|
d.amplitude[0] = d.amplitude[0] * 1.1
|
|
d.recalculate_waveforms()
|
|
## increase amplitude Y
|
|
if keyleds.KEY_LEVEL in current_keys:
|
|
d.phase_increment[0] = d.phase_increment[0] + 1
|
|
## increase phase Y
|
|
|
|
def handle_joystick_left(key):
|
|
current_keys = keyboardcb.KeyboardCB.current_keys
|
|
if len(current_keys) == 1: ## just joystick
|
|
d.increment[0] = int(d.increment[0] * 0.91)
|
|
if keyleds.KEY_RANGE in current_keys:
|
|
d.amplitude[0] = d.amplitude[0] * 0.91
|
|
d.recalculate_waveforms()
|
|
## increase amplitude X
|
|
if keyleds.KEY_LEVEL in current_keys:
|
|
d.phase_increment[0] = d.phase_increment[0] - 1
|
|
## increase phase X
|
|
|
|
mykeys=keyboardcb.KeyboardCB({keyleds.KEY_MENU: do_abort,
|
|
keyleds.KEY_XY:toggle_xy,
|
|
keyleds.KEY_WAVE: toggle_waveform,
|
|
keyleds.JOY_UP: handle_joystick_up,
|
|
keyleds.JOY_DN: handle_joystick_down,
|
|
keyleds.JOY_RT: handle_joystick_right,
|
|
keyleds.JOY_LT: handle_joystick_left
|
|
})
|
|
await do_dds_loop(d)
|
|
|
|
vectoros.reset()
|
|
|
|
|
|
|