lp_final.py

from neopixel import *
import time
import json
import RPi.GPIO as GPIO
import os, fnmatch
import numpy as np

#Button-config:
GPIO.setmode(GPIO.BCM)
GPIO.setup(25, GPIO.IN, pull_up_down=GPIO.PUD_UP)

# LED strip configuration:
LED_COUNT	= 144	  # Number of LED pixels.
LED_PIN		= 18	  # GPIO pin connected to the pixels (18 uses PWM!).
LED_FREQ_HZ	= 800000  # LED signal frequency in hertz (usually 800khz)
LED_DMA		= 5	   # DMA channel to use for generating signal (try 5)
LED_BRIGHTNESS 	= 50	 # Set to 0 for darkest and 255 for brightest
LED_INVERT	= False   # True to invert the signal (when using NPN transistor level shift)
LED_CHANNEL	= 0	   # set to '1' for GPIOs 13, 19, 41, 45 or 53
LED_STRIP	= ws.WS2811_STRIP_GRB   # Strip type and colour ordering
strip = Adafruit_NeoPixel(LED_COUNT, LED_PIN, LED_FREQ_HZ, LED_DMA, LED_INVERT, LED_BRIGHTNESS, LED_CHANNEL, LED_STRIP)
strip.begin()


def convert_image(image_path):
	with open(image_path,"r") as f:
		new_image = []
		f.readline()
		f.readline()
		dims=f.readline()
		file_len=int(dims[:dims.index(" ")])*int(dims[dims.index(" "):])
		new_image.append(int(dims[:dims.index(" ")]))
		f.readline()
		for i in range(file_len):
			p1 = int(f.readline()[:-1])
			p2= int(f.readline()[:-1])
			p3= int(f.readline()[:-1])
			pixel = [p1,p2,p3]
			new_image.append(pixel)
		f.close()
	return new_image


def read_image(image_path):
	converted_image = convert_image(image_path) 								#will be a list with 1 LE / 1ine
	image_width = converted_image[0]
	converted_image = converted_image[1:]
	inverted_list = [[] for i in range(image_width)]
	list_iteration_counter = 0

	for i in range(144):
		for j in range(image_width):
			inverted_list[j].append(converted_image[list_iteration_counter])
			list_iteration_counter+=1
	with open("images/new_format/"+image_path[image_path.rfind("/"):image_path.index(".ppm")]+".json","w") as f:
		json.dump({"pixel":inverted_list,"dims_width":image_width},f)
		f.close()
	del inverted_list
	width_of_each_image.append(image_width)
	image_as_json_list.append("images/new_format/"+image_path[image_path.rfind("/"):image_path.index(".ppm")]+".json")


def show_picture(json_path):
	with open(json_path,"r") as f:
		image_dict = json.load(f)
		f.close()
	pixel = image_dict["pixel"]
	for _ in range(len(pixel)):
		for __ in range(len(pixel[_])):
			pixel_temp = pixel[_][__]
			#print(pixel_temp[0],pixel_temp[1],pixel_temp[2])
			strip.setPixelColorRGB(LED_COUNT-1-__,pixel_temp[0],pixel_temp[1],pixel_temp[2])
		strip.show()
		time.sleep(0.08)


def clear_strip():
	for i in range(144):
		strip.setPixelColorRGB(i,0,0,0)
	strip.show()


################################################################################
#Init. vars:
image_old_list = []
image_as_json_list = []
image_already_parsed = []
width_of_each_image = []
button_counter = 0
while_counter = 0


#show program start (not ready):
for _ in range(10):
	 strip.setPixelColorRGB(_,200,0,0)
strip.show()


list_all = os.listdir('images')
list_already_parsed = os.listdir('images/new_format')
pattern = "*.ppm"
print(list_all,list_already_parsed)
for entry in list_all:
	if fnmatch.fnmatch(entry, pattern):
		 image_old_list.append(entry)
         
pattern = "*.json"
for entry in list_already_parsed:
	if fnmatch.fnmatch(entry, pattern):
		 image_already_parsed.append(entry)
         
for old_img in image_old_list:
    print(old_img[:old_img.index(".ppm")] + ".json")
    if (old_img[:old_img.index(".ppm")] + ".json") not in image_already_parsed:
        read_image("images/"+old_img)
    else:
	image_as_json_list = image_already_parsed

clear_strip()
#show that ready:
for _ in range(10):
	strip.setPixelColorRGB(_,0,200,0)
strip.show()
time.sleep(2)

print(image_as_json_list)
while True:
	if while_counter == 0:
		print(list_all[button_counter] + " is next on the list.")
	while_counter += 1
	input_state = GPIO.input(25)
	if input_state == False:
		time.sleep(0.3)
		while_counter = 0
		start_time = time.time()
		print("PRESS NOW! GOGOGOGO")
		delta_t = True
		while delta_t:
			print(button_counter)
			input_state = GPIO.input(25)
			if input_state == False:
				delta_t = False
				print("Button pressed twice")

			if time.time() - start_time >= 0.5:
				print("Too slow noob...")
				show_picture(image_as_json_list[button_counter])
				delta_t = False
				
		button_counter+=1
	clear_strip()
	time.sleep(0.2)
	if button_counter == len(image_as_json_list):
		button_counter = 0