render_batch_views.py

#!/usr/bin/env python
import os 
import sys
import subprocess
from multiprocessing import Pool

'''
 Usage: 
 ./render_batch_views <input_directory> <output_directory> <sample_type>
                <num_samples> <width> <height>
  input_directory - root input path to the models to render [.OFF]
  output_directory - root output path directory for rendered views
  sample_type - type of sampling to use:
              - 0 : sample the vertices of an icosahedron
              - 1 : sample the vertices of a cylinder with <num_samples>
                    samples.
              - 2 : sample a sphere uniformly at random using <num_samples>
                    samples.
 num_samples - number of samples to use, only useful for sample_type = 1,2.
 width - output image width
 height - output image height

 If a model file is at 
        <input_directory>/path/to/model.off, then the output
 will be placed in 
        <output_directory>/path/to/png/out.png
 '''

root_input_dir = os.path.abspath(sys.argv[1])
root_output_dir = sys.argv[2]
sample_type = sys.argv[3]
num_samples = sys.argv[4]
width = sys.argv[5]
height = sys.argv[6]

if (not os.path.isabs(root_output_dir)):
    root_output_dir = os.getcwd() + '/' + root_output_dir

bin_path = os.getcwd() + '/build/bin/render_views'

root_output_dir = os.path.abspath(root_output_dir)

print('Root input dir: %s' % root_input_dir)
print('Root output dir: %s' % root_output_dir)

def render_views(arg):
    input_path, output_path, image_width, image_height = arg
    subprocess.call([bin_path, input_path, output_path, sample_type,\
        num_samples, str(image_width),str(image_height)])

if __name__ == '__main__':
    pool = Pool(processes=1)  
    worklist = []
    for dir_name, dir_list, file_list in os.walk(root_input_dir):
        dir_name=os.path.abspath(dir_name)
        print('Found dir: %s' % dir_name)
        for file_name in file_list:
            name, extension = os.path.splitext(file_name)
            if extension in ['.off', '.obj', '.ply']:
                print('\tname=%s ext=%s' % (name, extension))
                input_path = dir_name + '/' + file_name
                output_path=dir_name[len(root_input_dir)+1:]
                full_output_path = root_output_dir + '/' + output_path
                if not os.path.exists(full_output_path):
                    os.makedirs(full_output_path + '/png/')
                    os.makedirs(full_output_path + '/cfg')
                print('\toutput_path=%s/%s' % (root_output_dir, output_path))
                worklist.append((input_path, full_output_path, width, height))
    pool.map(render_views, worklist)