Beispiel #1
0
    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
Beispiel #2
0
    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