import bpy
import os.path
import math
import sys
import os
import csv
from random import sample
#file_path = 'D:/3D-FUTURE-model/final_choosed_FUTURE_chairs_text_caption_clear.csv'
#file_list = []
#with open(file_path, 'r') as r:
# reader = csv.reader(r)
# for row in reader:
# file_name = row[0]
# file_name = file_name.split('_')[0]
# if file_name not in file_list:
# file_list.append(file_name)
C = bpy.context
D = bpy.data
scene = D.scenes['Scene']
#inters =[i for i in range(48,72)] # [i for i in range(0,24)]#+
#fixed_views = [i for i in range(40,70)]
# cameras: a list of camera positions
# a camera position is defined by two parameters: (theta, phi),
# where we fix the "r" of (r, theta, phi) in spherical coordinate system.
# 5个固定视角:前 右 后 左 上
# cameras = [(60, 0), (60, 90), (60, 180), (60, 270),(0, 0)]
# 连续视角:固定某一个角度,然后环视3D物体,环视角度间隔inter=30度
fixed_view = 60
inter = 12
cameras = [(fixed_view, i) for i in range(0, 360, inter)] # 这会生成360/12=30个视角图片
render_setting = scene.render
# 输出图像大小 (W, H)
w = 224
h = 224
render_setting.resolution_x = w*2
render_setting.resolution_y = h*2
haved_folder = os.listdir('E:\\objaverse\\imgs\\')
count = 1
'''****************************************************************'''
def main(count):
root_dir = 'E:\\objaverse\\objaverse\\'
#for i,dir in enumerate(os.listdir(root_dir)[3736:4736]):
for i,dir in enumerate(os.listdir(root_dir)):
#for i,dir in enumerate(file_list):
# inter = sample(inters, 20)
# fixed_view = sample(fixed_views, 20)