"""
Script to render IBL renders
Usage: blender <blend file> -P hdr_blender.py -b
"""
import os
import random
from math import radians
from os import listdir
from os.path import isfile, join

import bpy
import cycles
from mathutils import Euler, Matrix, Vector


RENDER_OUT = '/render/'
path = './'

data_dir = '/Users/momo/Desktop/Pandora-HDR_NeRF/final_results/meeting_room/'
# method_directories = [x[0] for x in os.walk(data_dir)]
method_directories = os.listdir(data_dir) 
method_directories = list(set(method_directories) - set([data_dir, '.DS_Store', 'capture_settings.json']))

inp_dirs = []
out_dirs = []
for method_dir in method_directories:
    input_dir = data_dir + method_dir + '/pano/'
    inp_dirs.append(input_dir)
    output_dir = data_dir + method_dir + RENDER_OUT
    out_dirs.append(output_dir)


pairs = list(zip(inp_dirs,out_dirs))
scn = bpy.context.scene
scn.render.engine = 'CYCLES'
scn.render.image_settings.file_format = 'OPEN_EXR'
scn.world.use_nodes = True
# Enabf_le GPU rendering.
# bpy.context.preferences.addons['cycles'].preferences.compute_device_type = 'CUDA'
bpy.context.preferences.addons['cycles'].preferences.get_devices()
for device in bpy.context.preferences.addons['cycles'].preferences.devices:
    if "TITAN V" == device.name:
        device.use = True
    else:
        device.use = False
# select world node tree
wd = scn.world
nt = bpy.data.worlds[wd.name].node_tree

for pair in pairs:
    inp,out = pair    
    inp = inp    
    out = os.path.join(path,out)
    os.makedirs(out,exist_ok=True)
    S = bpy.context.scene
    envFiles = [f for f in listdir(inp) if isfile(join(inp, f)) and f.endswith('.exr')]
    print(envFiles)
    i = 0
    for env_map in envFiles:
        env_file = join(inp, env_map)
        bpy.ops.image.open(filepath=env_file, directory=inp, files=[
                           {"name": env_map, "name": env_map}], show_multiview=False)
        output_name = env_map.split('.')[0] +'.exr'
        
        backNode = nt.nodes['Environment Texture']
        backNode.image = bpy.data.images.load(env_file)

        bpy.context.scene.render.filepath = os.path.join(out, output_name)
        print("Output",os.path.join(out, output_name))
        bpy.ops.render.render(write_still=True)