class RenderManager:
def __init__(self):
pass
#self.executor = ProcessPoolExecutor()
#self.tasks = []
# self.scene = Scene()
# if scene.init_scene('../res/refltest.json') == True:
# scene.render('../outputs/refltest')
def init(self, cfgpath, outputPath):
if os.path.exists(cfgpath) == False:
print("配置文件不存在：%s"%cfgpath)
return False
try:
file = open(cfgpath)
cfgjson = json.load(file)
file.close()
#print("加载摄像机")
#camPamras = cfgjson["Camera"]
#self.camera = create_camera(camPamras["params"])
cfg = cfgjson["Result"]
#print("初始化渲染配置")
self.tex = Texture(cfg["width"], cfg["height"])
#self.camera.set_render_target(self.tex)
#print("等待队列初始化完成")
#self.camera.render_for_task(queue, log_progress)
#print("队列初始化完成")
#taskResults = []
#for i in range(0, self.executor._max_workers):
# taskResults.append(self.executor.submit(RenderManager.render_task, cfgpath, i))
#for result in taskResults:
# result.result()
#tasks = []
taskqueue = Queue()
resultqueue = Queue()
tasks = []
#colorqueues = []
# scene = Scene()
# if scene.init_scene(cfgpath) == False:
# print("Scene Init Faild")
# return
bx = 0
by = 0
#pool = Pool(8)
#results = []
while by < self.tex.height():
while bx < self.tex.width():
bw = self.tex.width() - bx
bh = self.tex.height() - by
if bw > 32:
bw = 32
if bh > 32:
bh = 32
#print("Area Prepare: X:%d,Y:%d,W:%d,H:%d"%(bx,by,bw,bh))
#pw = Process(target = RenderManager.render_task, args=(cfgpath, bx, by, bw, bh, colorqueue))
#queue.put((bx, by, bw, bh, self.tex.width(), self.tex.height()))
taskqueue.put((bx, by, bw, bh, self.tex.width(), self.tex.height()))
#tasks.append(pw)
#results.append(pool.apply_async(RenderManager.render_task, args = (cfgpath, (bx, by, bw, bh, self.tex.width(), self.tex.height()))))
bx += 32
by += 32
bx = 0
#pool.close()
#pool.join()
# for result in results:
# if result != None:
# #print(dir(result))
# r = result.get()
# if r != None:
# for c in r:
# px = c[0]
# py = c[1]
# c = Color(c[2], c[3], c[4], c[5])
# self.tex.set_pixel(px, py, c)
# # for c in result:
# # px = c[0]
# # py = c[1]
# # c = Color(c[2], c[3], c[4], c[5])
# # self.tex.set_pixel(px, py, c)
#with Manager() as manager:
#mycolors = []
for i in range(0, 7):
#mycolorlist = manager.list()
#colorqueue = Queue()
pw = Process(target = RenderManager.render_task, args=(cfgpath, i, taskqueue, resultqueue))
#colorqueues.append(colorqueue)
tasks.append(pw)
#mycolors.append(mycolorlist)
for task in tasks:
print("Task Start...")
task.start()
#task.join()
# for result in tasks:
# print(type(result))
# for cq in colorqueues:
# while cq.empty() == False:
# dt = cq.get(True)
# self.tex.set_pixel(dt[0], dt[1], dt[2])
# for cols in return_dict.values():
# for c in cols:
# self.tex.set_pixel(c[0], c[1], c[2])
currentpixel = 0
pixelcount = self.tex.width() * self.tex.height()
currentprogress = -1
while currentpixel < pixelcount:
if resultqueue.empty() == False:
pixel = resultqueue.get_nowait()
px = pixel[0]
py = pixel[1]
c = Color(pixel[2], pixel[3], pixel[4], pixel[5])
progress = currentpixel/pixelcount*100
progress = int(progress)
if currentprogress != progress:
print("渲染进度：%d"%(progress))
currentprogress = progress
#print("设置像素颜色:(%d,%d)"%(px,py))
self.tex.set_pixel(px, py, c)
currentpixel += 1
self.tex.save(outputPath)
except Exception as e:
print('渲染失败，请检查文件：%s'%cfgpath)
print(e.args)
return False
return True
@staticmethod
def render_task(cfgpath, pid, taskqueue, resultqueue):
#print('pid:%d'%(pid))
# print("SceneInit:X:%d,Y:%d,W:%d,H:%d"%(beginx,beginy,width,height))
# scene = Scene()
# if scene.init_scene(cfgpath) == False:
# print("Scene Init Faild")
# return
# print("Scene Render Begin")
# scene.render_range(beginx, beginy, width, height, pixelqueue)
# print("Range Finish")
#colors = []
#colordict[pid] = []
#if colordict.has_key(pid) == False:
#print("Key:%d"%(pid))
#colordict[7] = 5
scene = Scene()
if scene.init_scene(cfgpath) == False:
print("Scene Init Faild")
return None
# print("Scene Render Begin:PID:,X:%d,Y:%d,W:%d,H:%d"%(data[0],data[1],data[2],data[3]))
# scene.render_range(data[0],data[1],data[2],data[3],data[4],data[5], colors)
# print("Range Finish")
# return colors
while taskqueue.empty() == False:
dt = taskqueue.get(True)
print("Scene Render Begin:PID:%d,X:%d,Y:%d,W:%d,H:%d"%(pid,dt[0],dt[1],dt[2],dt[3]))
scene.render_range(dt[0],dt[1],dt[2],dt[3],dt[4],dt[5], resultqueue)
print("Range Finish")
