'''

Code for rendering the groundtruths of Doc3D dataset

https://www3.cs.stonybrook.edu/~cvl/projects/dewarpnet/storage/paper.pdf (ICCV 2019)

This code renders the gts needed for the DewarpNet training (image, uv, 3D coordinates)

and saves the .blend files. The .blend files can be later used

to render other gts (normal, depth, checkerboard, albedo).

Each .blend file takes ~2.5MB set the save_blend_file flag to False if you don't need.

Written by: Sagnik Das and Ke Ma

Stony Brook University, New York

December 2018

'''

import sys

import csv

import bpy

import bmesh

import random

import math

from mathutils import Vector, Euler

import os

import string

from bpy_extras.object_utils import world_to_camera_view

rridx = 1

save_blend_file=True

def reset_blend():

bpy_data_iter.remove(id_data, do_unlink=True)

def isVisible(mesh, cam):

return True

def select_object(ob):

bpy.context.view_layer.objects.active=ob

def prepare_scene():

bpy.data.scenes['Scene'].view_settings.view_transform='Standard'

def prepare_rendersettings():

bpy.data.scenes['Scene'].render.resolution_percentage=100

def position_object(mesh_name):

return mesh

def add_lighting():

# wlinks=world.node_tree.links

def reset_camera(mesh):

return vid

def page_texturing(mesh, texpath):

links.new(texture_node.inputs[0],texturecoord_node.outputs[2])

# def get_image(objpath, texpath):

# bpy.context.scene.use_nodes = True

# tree = bpy.context.scene.node_tree

# links = tree.links

# # clear default nodes

# for n in tree.nodes:

# tree.nodes.remove(n)

# # create input render layer node

# render_layers = tree.nodes.new('CompositorNodeRLayers')

# file_output_node_0 = tree.nodes.new("CompositorNodeOutputFile")

# file_output_node_0.base_path = path_to_output_images

# # change output image name to obj file name + texture name + random three

# # characters (upper lower alphabet and digits)

# id_name = objpath.split('/')[-1][:-4] + '-' + texpath.split('/')[-1][:-4] + '-' + \

# ''.join(random.sample(string.ascii_letters + string.digits, 3))

# file_output_node_0.file_slots[0].path = id_name

# links.new(render_layers.outputs[0], file_output_node_0.inputs[0])

# return id_name

def color_wc_material(obj,mat_name):

links.new(em_node.outputs[0],mat_node.inputs[0])

def get_worldcoord_img(img_name):

links.new(render_layers.outputs[0], file_output_node_0.inputs[0])

def prepare_no_env_render():

scene.view_settings.view_transform='Default'

def render_pass(obj, objpath, texpath):

return fn

def render_img(objpath, texpath):

fn = render_pass(mesh, objpath, texpath)

print('sys.argv ',sys.argv)

id1 = int(sys.argv[-2])

id2 = int(sys.argv[-1])

rridx = int(sys.argv[-3])

print('id1 ',id1)

print('id2 ',id2)

print('rridx ',rridx)

# id1 = int(sys.argv[0])

# id2 = int(sys.argv[0])

# rridx = int(sys.argv[0])

path_to_output_images='./img/{}/'.format(rridx)

path_to_output_uv = './uv/{}/'.format(rridx)

path_to_output_wc = './wc/{}/'.format(rridx)

if save_blend_file:

path_to_output_blends='./bld/{}/'.format(rridx)

for fd in [path_to_output_images, path_to_output_uv, path_to_output_wc, path_to_output_blends]:

if not os.path.exists(fd):

os.makedirs(fd)

env_list = './envs.csv'

tex_list = './tex.csv'

obj_list = './objs.csv'

envlist = []

with open(env_list, 'r') as f:

envlist = list(csv.reader(f))

with open(tex_list, 'r') as t, open(obj_list, 'r') as m:

texlist = list(csv.reader(t))

objlist = list(csv.reader(m))

print(objlist)

for k in range(id1, id2):

#print(k)

objpath = objlist[k][0]

idx = random.randint(0, len(texlist))

texpath=texlist[idx][0]

print(objpath)

print(texpath)

render_img(objpath, texpath)