def RenderFur(self, params_dict, img_path, exportCSV=True): ## assign fur parameters Fur.SetFurDescription(self.fur_desc, params_dict) Fur.CopyFurBaseColor2Material(self.fur_desc, self.material) t_render_start = datetime.now() ## export single frame as image cmds.setAttr("defaultRenderGlobals.imageFilePrefix", img_path, type="string") mel.eval('renderWindowRenderCamera "render" renderView ' + "{0}".format(self.camera) + ";") t_render_end = datetime.now() t_render_elapsed = t_render_end - t_render_start t_imageio_start = datetime.now() ## export parameter as CSV file if needed if exportCSV: csv_file = img_path + ".csv" FurParam.dict2csv(params_dict, csv_file) ## read rendered image for further processing img_file = img_path + "_tmp." + self.imgFileExt img_cv2 = cv2.imread(img_file) t_imageio_end = datetime.now() t_imageio_elapsed = t_imageio_end - t_imageio_start return img_cv2, t_render_elapsed, t_imageio_elapsed
def write_csv(kv_id1, kv_id2): global count params01_dict[kv_id1[0]] = kv_id1[1] params01_dict[kv_id2[0]] = kv_id2[1] export_path = "{0}/Space{1:04}.csv".format(folder_path, count) FurParam.dict2csv(FurParam.ConvertFurParams(params01_dict, True), export_path) count = count + 1 return None