88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
from shutil import copyfile
|
|
from PIL import Image, ImageFont, ImageDraw
|
|
from pysstv import color
|
|
|
|
import logging as log
|
|
|
|
try:
|
|
from picamera import PiCamera
|
|
except ModuleNotFoundError:
|
|
log.info("Running in simulator mode")
|
|
|
|
|
|
def take_photo():
|
|
# This def is meant to take a photograph from the robot,
|
|
# it should include all steps and error checking to raise the mast
|
|
# Take the photo, and put the mast down.
|
|
|
|
# Copy in the test pattern png (if photo process errors out, this will be used instead)
|
|
log.debug('Copying test pattern.')
|
|
copyfile('photos/TEST_PATTERN.jpg', 'working/working.jpg')
|
|
|
|
# Software to take the photo should be here
|
|
#copyfile('photos/camera_latest.jpg', 'working/working.jpg')
|
|
log.debug('Initalizing camera.')
|
|
try:
|
|
camera = PiCamera()
|
|
log.info('Saving photo.')
|
|
camera.capture('working/working.jpg')
|
|
except NameError:
|
|
log.info("Running in simulator mode, not replacing test pattern")
|
|
|
|
def mark_photo():
|
|
log.info('Opening photo for viewing.')
|
|
raw_img = Image.open("working/working.jpg") # Open the current working image
|
|
log.info('Resizing image.')
|
|
img = raw_img.resize((320, 240), Image.ANTIALIAS) # resize it for the radio
|
|
|
|
log.info('Drawing on image.')
|
|
if False:
|
|
TINT_COLOR = (255, 255, 255) # White text bg
|
|
TEXT_COLOR = (0,0,0)
|
|
else:
|
|
TINT_COLOR = (0, 0, 0) # Black text bg
|
|
TEXT_COLOR = (255,255,255)
|
|
TRANSPARENCY = .25 # Degree of transparency, 0-100%
|
|
OPACITY = int(255 * TRANSPARENCY)
|
|
overlay = Image.new('RGBA', img.size, TINT_COLOR+(0,))
|
|
|
|
draw = ImageDraw.Draw(overlay)
|
|
#bigfont = ImageFont.truetype(r'C:\Users\System-Pc\Desktop\arial.ttf', 20)
|
|
#smallfont = ImageFont.truetype(r'C:\Users\System-Pc\Desktop\arial.ttf', 17)
|
|
|
|
draw.rectangle(((0, 0), (90, 20)), fill=TINT_COLOR+(OPACITY,))
|
|
#draw.text((0, 0),"KW1FOX",TEXT_COLOR,font=bigfont) # Draw KW1FOX in the top left
|
|
|
|
draw.rectangle(((0, 40), (83, 80)), fill=TINT_COLOR+(OPACITY,))
|
|
#draw.text((0, 40),"day: 25.2",TEXT_COLOR,font=smallfont)
|
|
#draw.text((0, 60),"Volt: 13.8",TEXT_COLOR,font=smallfont)
|
|
#draw.text((0, 80),"Miles: 1.02",TEXT_COLOR,font=smallfont)
|
|
|
|
log.info('Converting image color.')
|
|
img = img.convert("RGBA")
|
|
img = Image.alpha_composite(img, overlay)
|
|
img = img.convert("RGB")
|
|
img.save('working/working.jpg') # Save the working image
|
|
return img
|
|
|
|
if __name__ == "__main__":
|
|
log.basicConfig(level=log.DEBUG)
|
|
|
|
# Take photograph.
|
|
log.info('Taking Photograph')
|
|
take_photo() # Saves a photograph to the working/working.jpg location
|
|
log.info('Done taking photo.')
|
|
|
|
# Draw neccicary text on photo
|
|
log.info('Drawing on photo.')
|
|
radio_photo = mark_photo() # draws text on working/working.jpg and returns a PIL image
|
|
log.info('Done drawing on photo.')
|
|
|
|
log.info('Creating slowscan.')
|
|
slowscan = color.Robot36(radio_photo, 48000, 16) # Image, rate, bits
|
|
#slowscan = color.MartinM1(radio_photo, 48000, 16) # Image, rate, bits
|
|
log.info('Saving out slowscan.')
|
|
slowscan.write_wav('working/working.wav')
|
|
|
|
#sstv('working/working.png', 'working/radio.wav', mode='Robot36')
|