from shutil import copyfile
from PIL import Image, ImageFont, ImageDraw
from pysstv import color

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)
    copyfile('photos/TEST_PATTERN.jpg', 'working/working.jpg')

    # Software to take the photo should be here
    copyfile('photos/camera_latest.jpg', 'working/working.jpg')

def mark_photo():
    raw_img = Image.open("working/working.jpg") # Open the current working image
    img = raw_img.resize((320, 240), Image.ANTIALIAS) # resize it for the radio

    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)

    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__":
    # Take photograph. 
    take_photo() # Saves a photograph to the working/working.jpg location

    # Draw neccicary text on photo
    radio_photo = mark_photo() # draws text on working/working.jpg and returns a PIL image

    slowscan = color.Robot36(radio_photo, 48000, 16) # Image, rate, bits
    slowscan.write_wav('working/working.wav')

    #sstv('working/working.png', 'working/radio.wav', mode='Robot36')