mirror of
https://github.com/cyberboy666/r_e_c_u_r.git
synced 2025-12-12 03:10:17 +01:00
inital commit
(rough sketch of idea at this point)
This commit is contained in:
275
data_centre.py
Normal file
275
data_centre.py
Normal file
@@ -0,0 +1,275 @@
|
||||
import json
|
||||
|
||||
import logging
|
||||
from collections import OrderedDict
|
||||
|
||||
import os
|
||||
from random import randint
|
||||
|
||||
#TODO : standise the paths to things, use constants for file names, standise the naming convention of files
|
||||
import time
|
||||
|
||||
PATH_TO_BROWSER = 'C:/TestFolderStucture'
|
||||
|
||||
PATH_TO_DATA_OBJECTS = 'C:/Users/Tim/PycharmProjects/videoLooper/'
|
||||
|
||||
EMPTY_BANK = dict(name='',location='',length=-1,start=-1,end=-1)
|
||||
|
||||
logger = logging.getLogger('myapp')
|
||||
hdlr = logging.FileHandler('C:/Users/Tim/PycharmProjects/videoLooper/myapp.log')
|
||||
formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
|
||||
hdlr.setFormatter(formatter)
|
||||
logger.addHandler(hdlr)
|
||||
logger.setLevel(logging.INFO)
|
||||
|
||||
class data(object):
|
||||
def __init__(self):
|
||||
|
||||
self._open_folders = []
|
||||
self._browser_list = []
|
||||
|
||||
def rewrite_browser_list(self):
|
||||
self._browser_list = generate_browser_list(PATH_TO_BROWSER, 0, self._open_folders)
|
||||
|
||||
def get_browser_data_for_display(self):
|
||||
if not self._browser_list:
|
||||
self.rewrite_browser_list()
|
||||
|
||||
browser_list_for_display = []
|
||||
for index , dir in enumerate(self._browser_list):
|
||||
browser_list_for_display.append(([dir['name'],dir['bank']],index))
|
||||
#logger.info(browser_list_for_display)
|
||||
return browser_list_for_display
|
||||
|
||||
def update_open_folders(self, folder_name):
|
||||
if folder_name not in self._open_folders:
|
||||
self._open_folders.append(folder_name)
|
||||
else:
|
||||
self._open_folders.remove(folder_name)
|
||||
|
||||
|
||||
def get_all_looper_data_for_display():
|
||||
|
||||
memory_bank = read_json('DisplayData.json')
|
||||
loop_data = []
|
||||
for index, bank in enumerate(memory_bank):
|
||||
length = convert_int_to_string_for_display(bank["length"])
|
||||
start = convert_int_to_string_for_display(bank["start"])
|
||||
end = convert_int_to_string_for_display(bank["end"])
|
||||
loop_data.append(([str(index),bank["name"],length,start,end],index))
|
||||
|
||||
return loop_data
|
||||
|
||||
def is_file_in_memory_bank(file_name, memory_bank=[]):
|
||||
if not memory_bank:
|
||||
memory_bank = read_json('DisplayData.json')
|
||||
for index, bank in enumerate(memory_bank):
|
||||
if(file_name == bank['name']):
|
||||
return True , index
|
||||
return False, ''
|
||||
|
||||
def generate_browser_list(current_path, current_level, open_folder_list):
|
||||
|
||||
global results
|
||||
results = []
|
||||
|
||||
add_folder_to_browser_list(current_path, current_level,open_folder_list)
|
||||
|
||||
memory_bank = read_json('DisplayData.json')
|
||||
|
||||
for browser_line in results:
|
||||
is_file, file_name = extract_file_type_and_name_from_browser_format(browser_line['name'])
|
||||
if is_file:
|
||||
is_banked, bank_number = is_file_in_memory_bank(file_name, memory_bank)
|
||||
if is_banked:
|
||||
browser_line['bank'] = str(bank_number)
|
||||
|
||||
return results
|
||||
|
||||
def add_folder_to_browser_list(current_path, current_level,open_folder_list):
|
||||
root, dirs, files = next(os.walk(current_path))
|
||||
|
||||
indent = ' ' * 4 * (current_level)
|
||||
for folder in dirs:
|
||||
is_open, char = check_folder_state(folder,open_folder_list)
|
||||
#print('{}{}{}'.format(indent, folder, char))
|
||||
results.append(dict(name='{}{}{}'.format(indent, folder, char), bank='x'))
|
||||
if (is_open):
|
||||
next_path = '{}/{}'.format(root, folder)
|
||||
next_level = current_level + 1
|
||||
add_folder_to_browser_list(next_path, next_level,open_folder_list)
|
||||
|
||||
for f in files:
|
||||
if (os.path.splitext(f)[1] in ['.mp4']):
|
||||
#print('{}{}'.format(indent, f))
|
||||
results.append(dict(name='{}{}'.format(indent, f), bank='-'))
|
||||
|
||||
def check_folder_state(folder_name,open_folder_list):
|
||||
if (folder_name in open_folder_list):
|
||||
return True, '/'
|
||||
else:
|
||||
return False, '|'
|
||||
|
||||
def extract_file_type_and_name_from_browser_format(dir_name):
|
||||
#logger.info('the name we got was {}'.format(dir_name))
|
||||
if(dir_name.endswith('|') or dir_name.endswith('/')):
|
||||
return False , dir_name.lstrip()[:-1]
|
||||
else:
|
||||
return True , dir_name.lstrip()
|
||||
|
||||
def get_length_for_file(location):
|
||||
#TODO: will have omx.player get length of file probs..
|
||||
pass
|
||||
|
||||
def create_new_bank_mapping_in_first_open(file_name):
|
||||
memory_bank = read_json('DisplayData.json')
|
||||
for index , bank in enumerate(memory_bank):
|
||||
if(not bank['name']):
|
||||
create_new_bank_mapping(index,file_name,memory_bank)
|
||||
return True
|
||||
return False
|
||||
|
||||
def create_new_bank_mapping(bank_number,file_name,memory_bank=[]):
|
||||
has_location , location = get_path_for_file(file_name)
|
||||
length = get_length_for_file(location)
|
||||
new_bank = dict(name=file_name, location=location, length=-1, start=-1, end=-1)
|
||||
update_a_banks_data(bank_number, new_bank, memory_bank)
|
||||
|
||||
def get_path_for_file(file_name):
|
||||
for root, dirs, files in os.walk(PATH_TO_BROWSER):
|
||||
if file_name in files:
|
||||
print root
|
||||
return True, '{}/{}'.format(root,file_name)
|
||||
else:
|
||||
return False, ''
|
||||
|
||||
def update_a_banks_data(bank_number, bank_info, memory_bank=[]):
|
||||
if not memory_bank:
|
||||
memory_bank = read_json('DisplayData.json')
|
||||
memory_bank[bank_number] = bank_info
|
||||
update_json('DisplayData.json', memory_bank)
|
||||
|
||||
def clear_all_banks():
|
||||
memory_bank = read_json('DisplayData.json')
|
||||
for index , bank in enumerate(memory_bank):
|
||||
memory_bank[index] = EMPTY_BANK
|
||||
update_json('DisplayData.json', memory_bank)
|
||||
|
||||
def read_json(file_name):
|
||||
|
||||
with open(PATH_TO_DATA_OBJECTS + file_name) as data_file:
|
||||
data = json.load(data_file)
|
||||
|
||||
return data
|
||||
|
||||
def update_json(file_name,data):
|
||||
|
||||
with open('{}{}'.format(PATH_TO_DATA_OBJECTS, file_name), 'w') as data_file:
|
||||
json.dump(data, data_file)
|
||||
|
||||
def get_all_settings_data_for_display():
|
||||
settings = read_json('Settings.json')
|
||||
display_settings = []
|
||||
for index, setting in enumerate(settings):
|
||||
display_settings.append(([setting['name'],setting['value']],index))
|
||||
return display_settings
|
||||
|
||||
def get_a_banks_data(bank_number):
|
||||
memory_bank = read_json('DisplayData.json')
|
||||
return memory_bank[bank_number]
|
||||
|
||||
def switch_settings(setting_name):
|
||||
settings = read_json('Settings.json')
|
||||
|
||||
for index, setting in enumerate(settings):
|
||||
if setting['name'] == setting_name:
|
||||
setting = cycle_setting_value(setting)
|
||||
|
||||
update_json('Settings.json',settings)
|
||||
|
||||
def cycle_setting_value(setting):
|
||||
if setting['name'] == 'PLAYBACK_MODE':
|
||||
if setting['value'] == 'LOOPER':
|
||||
setting['value'] = 'PLAYLIST'
|
||||
elif setting['value'] == 'PLAYLIST':
|
||||
setting['value'] = 'RANDOM'
|
||||
else:
|
||||
setting['value'] = 'LOOPER'
|
||||
elif setting['name'] == 'SYNC_LENGTHS':
|
||||
if setting['value'] == 'ON':
|
||||
setting['value'] = 'OFF'
|
||||
else:
|
||||
setting['value'] = 'ON'
|
||||
elif setting['name'] == 'RAND_START':
|
||||
if setting['value'] == 'ON':
|
||||
setting['value'] = 'OFF'
|
||||
else:
|
||||
setting['value'] = 'ON'
|
||||
elif setting['name'] == 'VIDEO_OUTPUT':
|
||||
if setting['value'] == 'HDMI':
|
||||
setting['value'] = 'COMPOSITE'
|
||||
else:
|
||||
setting['value'] = 'HDMI'
|
||||
|
||||
return setting
|
||||
|
||||
|
||||
def get_next_context():
|
||||
next_bank_number = read_json('next_bank_number.json')
|
||||
memory_bank = read_json('DisplayData.json')
|
||||
next_bank_details = memory_bank[next_bank_number]
|
||||
start_value = next_bank_details['start']
|
||||
end_value = next_bank_details['end']
|
||||
length = next_bank_details['length']
|
||||
|
||||
use_rand_start, use_sync_length, sync_length, playback_mode = get_context_options_from_settings()
|
||||
set_next_bank_number_from_playback_mode(playback_mode,next_bank_number)
|
||||
|
||||
if use_rand_start and use_sync_length:
|
||||
start_value = randint(0, length - sync_length)
|
||||
end_value = start_value + sync_length
|
||||
elif use_rand_start and not use_sync_length:
|
||||
start_value = randint(0, end_value)
|
||||
elif not use_rand_start and use_sync_length:
|
||||
end_value = min(length, start_value + sync_length)
|
||||
|
||||
context = dict(location=next_bank_details['location'],start=start_value,end=end_value, bank_number=next_bank_number)
|
||||
|
||||
def get_context_options_from_settings():
|
||||
settings = read_json('Settings.json')
|
||||
use_sync_length = False
|
||||
sync_length = 0
|
||||
use_rand_start = False
|
||||
playback_mode = ''
|
||||
|
||||
for index, setting in enumerate(settings):
|
||||
if setting['name'] == 'SYNC_LENGTHS' and setting['value'] == 'ON':
|
||||
use_sync_length = True
|
||||
elif setting['name'] == 'SYNC_LENGTHS_TO':
|
||||
sync_length = setting['value']
|
||||
elif setting['name'] == 'RAND_START' and setting['value'] == 'ON':
|
||||
use_rand_start = True
|
||||
elif setting['name'] == 'PLAYBACK_MODE':
|
||||
playback_mode = setting['value']
|
||||
|
||||
return use_rand_start , use_sync_length , sync_length , playback_mode
|
||||
|
||||
def set_next_bank_number_from_playback_mode(playback_mode, current_bank_number):
|
||||
next_bank_number = 0
|
||||
if playback_mode == 'LOOPER':
|
||||
next_bank_number = current_bank_number
|
||||
elif playback_mode == 'RANDOM':
|
||||
#TODO: actually find which banks have value and only use those
|
||||
next_bank_number = randint(0,14)
|
||||
elif playback_mode == 'PLAYLIST':
|
||||
#TODO: implement some playlist objects and logic at some point
|
||||
next_bank_number = current_bank_number
|
||||
update_json('next_bank_number.json',next_bank_number)
|
||||
|
||||
def convert_int_to_string_for_display(time_in_seconds):
|
||||
if time_in_seconds < 0:
|
||||
return ''
|
||||
elif time_in_seconds >= 6000:
|
||||
return '99:99'
|
||||
else:
|
||||
return time.strftime("%M:%S", time.gmtime(time_in_seconds))
|
||||
Reference in New Issue
Block a user