RaspiQRP/main.py

150 lines
3.8 KiB
Python

#!/usr/bin/env python
import pigpio
import time
import datetime
import asyncio
from RPi import GPIO
from itertools import cycle
bands = {
"80m":[3.5, 4],
"40m":[7, 7.3],
"20m":[14, 14.35],
"10m":[28, 29.7],
"2m":[144,146],
"70cm":[430,440]
}
# Setup for encoder
band = cycle(bands)
CURRENT_BAND:str = None
def next_band():
global CURRENT_BAND
CURRENT_BAND = next(band)
FREQUENCY = bands[CURRENT_BAND][0]
print(FREQUENCY)
print(CURRENT_BAND)
return FREQUENCY
def prev_band():
global CURRENT_BAND
CURRENT_BAND = list(bands.keys())[list(bands.keys()).index(CURRENT_BAND)-1]
print(CURRENT_BAND)
FREQUENCY = bands[CURRENT_BAND][0] + 1
return FREQUENCY
FREQUENCY = next_band()
clk=17
dt=27
sw=22
GPIO.setmode(GPIO.BCM)
GPIO.setup(clk, GPIO.IN, pull_up_down=GPIO.PUD_DOWN)
clkLastState = GPIO.input(clk)
async def rotary(clk=clk, dt=dt, sw=sw, step=0.025):
GPIO.setmode(GPIO.BCM)
GPIO.setup(clk, GPIO.IN, pull_up_down=GPIO.PUD_DOWN)
GPIO.setup(dt, GPIO.IN, pull_up_down=GPIO.PUD_DOWN)
GPIO.setup(sw, GPIO.IN, pull_up_down=GPIO.PUD_DOWN)
global FREQUENCY, clkLastState
try:
while True:
clkState = GPIO.input(clk)
dtState = GPIO.input(dt)
swState = GPIO.input(sw)
if swState == GPIO.HIGH:
print("yeet")
if clkState != clkLastState:
if FREQUENCY > bands[CURRENT_BAND][1]:
FREQUENCY = next_band()
print(FREQUENCY)
if FREQUENCY < bands[CURRENT_BAND][0]:
FREQUENCY = prev_band()
print(FREQUENCY)
if dtState != clkState:
FREQUENCY += step
else:
FREQUENCY -= step
# print(round(FREQUENCY, 3))
clkLastState = clkState
await asyncio.sleep(.01)
except KeyboardInterrupt:
print("\nBye")
finally:
print("AAAAA")
GPIO.cleanup()
# 1. Hook up the screen and buttons
# * the screen should show the frequency we're at
# * read the buttons
# 2. Encoder [x]
# * Change frequency in the scope of a given band
PTT_IS_ON:bool = False
modulations = cycle(["./modulations/ssb.sh", "./modulations/fmrds.sh"])
# init GPIO
pi = pigpio.pi()
# button pin
pi.set_mode(10, pigpio.INPUT)
# modulation change pin(change this)
async def run_command(*args):
process = await asyncio.create_subprocess_exec(*args, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE)
print("Started: %s, pid=%s" % (args, process.pid), flush=True)
async def button_press(gpio:int, level):
# TODO: light an LED
global PTT_IS_ON
if(pi.read(gpio)!=0):
PTT_IS_ON = True
if(pi.read(gpio)==0):
PTT_IS_ON = False
await asyncio.sleep(.01)
# async def frequency(gpio):
# # TODO write this function after hooking up the encoder
async def main():
# TODO change after figuring out the logic
task_created = False
asyncio.create_task(rotary())
print("Ready")
cached_frequency = FREQUENCY
while True:
if FREQUENCY != cached_frequency:
cached_frequency = FREQUENCY
asyncio.create_task(run_command("./display", str(format(round(FREQUENCY,3), "7f"))))
await button_press(10, pigpio.EITHER_EDGE)
if (PTT_IS_ON and not task_created):
asyncio.create_task(run_command(next(modulations), str(FREQUENCY)))
task_created = True
elif (task_created and not PTT_IS_ON):
await asyncio.sleep(1)
asyncio.create_task(run_command("./kill.sh"))
task_created = False
await asyncio.sleep(.5)
if __name__ == "__main__":
try:
asyncio.run(main())
except KeyboardInterrupt:
print("\n")