spot_rotate.py

import json
import os
import sys
import psutil
import shutil
from easygui import *
# Progress bar, and other utilities

#from py_clui import gauge
from progress.bar import IncrementalBar

def clear_console():
    os.system('cls' if os.name=='nt' else 'clear')
# Command Line Wrapper
from sultan.api import Sultan
s = Sultan()

# why is this still here????
from colorama import init
init()
from colorama import Fore, Back, Style
def inform(text):
    print(Fore.CYAN + text + Style.RESET_ALL)
def log(text):
    print(Fore.BLACK + Style.BRIGHT + text + Style.RESET_ALL)
def ask(text):
    print(Fore.MAGENTA + text + Style.RESET_ALL)
    return input()
def warn(text):
    print(Back.RED + text + Style.RESET_ALL)

# Wrapper for the wrapper???
def run_and_return(command):
    with Sultan.load() as s:
        result = s.java('-jar ChunkyLauncher.jar -' + command).run()
        log(str(result.stdout))
        return result.stderr
def chunky_run(command):
    with Sultan.load() as s:
        result = s.java('-jar ChunkyLauncher.jar -' + command).run()
def execute_spot_rotate():
    msgbox('This is the Spot Rotation script.\n\nIt is assumed that the current camera position is the location to rotate at, and will start rotating 360deg from 0deg (ie south).')
    rotation_type = buttonbox('What direction would you like to rotate?', choices=['Clockwise', 'Anti-Clockwise'])
    log('Rotating ' + rotation_type)

    frames = None
    while frames == None:
        frames = integerbox('How many frames would you like the rotation to last?\n\n 60 Frames = 1s at 60fps or 2s at 30fps; etc')
    rotation_increment = 360 / int(frames)

    if rotation_type == 'Clockwise':
        log('Rotating clockwise at ' + str(rotation_increment) + ' per frame')
    else:
        rotation_increment = rotation_increment * -1
        log('Rotating Anti-Clockwise at ' + str(rotation_increment) + ' per frame')

    target = None
    while target == None:
        target = integerbox('How many SPP should each frame render to?')
        log('Rendering to ' + str(target) + ' SPP per frame')
        target = int(target)
    msgbox('Writing scene JSON files, click OK and check console...')
    # Get info

    with open('info.json') as json_file:
        data = json.load(json_file)
        scene_name = data['details'][0]['scene_name']
        chunky_directory = data['details'][0]['chunky_directory']
        json_file.close()

    scene_file = chunky_directory + '/scenes/' + scene_name + '.json'

    # Setting the camera in the template
    count = 0

    with open(scene_file, 'r') as original_scene:
        data = json.load(original_scene)
        data['camera']['orientation']['yaw'] = 0.0
        data['sppTarget'] = target
        data['name'] = scene_name + str(0)
        original_scene.close()

    with open(scene_file, "w") as new_scene:
        json.dump(data, new_scene)
        new_scene.close()


        with IncrementalBar('Copying JSON...', max=frames, suffix='%(percent).1f%% - %(eta)ds') as bar:
            for count in range(frames):
                shutil.copyfile(scene_file, chunky_directory + '/scenes/' + scene_name + str(count) + '.json')
                with open(scene_file, 'r') as original_scene:
                    data = json.load(original_scene)
                    data['camera']['orientation']['yaw'] += rotation_increment * count
                    data['sppTarget'] = target
                    data['name'] = scene_name + str(count)
                    original_scene.close()

                with open(chunky_directory + '/scenes/' + scene_name + str(count) + '.json', "w") as new_scene:
                    json.dump(data, new_scene)
                    new_scene.close()
            bar.next()
    log('Scene JSON created.')
    msgbox('Scene JSON created. Handing over to the Render script.')

    
    # Handover below. Must be included at the end of every prep script
    # Write number of frames to frames.txt and SPP target to target.txt
    # It's incredibly hacky, fight me.
    log('Writing number of frames')
    file = open('frames.txt', 'w')
    file.write(str(frames))
    file.close()
    log('Writing SPP target..')
    file = open('target.txt', 'w')
    file.write(str(target))
    file.close()
    log('Handing over to rendering.py ...')