-
Notifications
You must be signed in to change notification settings - Fork 1
/
create_charpacks_eye.py
141 lines (106 loc) · 5.16 KB
/
create_charpacks_eye.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
from furhat_remote_api import FurhatRemoteAPI
import json
import shutil
import os
from zipfile import ZipFile
################ GLOBAL VARIABLES ################
BASE_CHAR_NAME = "Isabel"
NEW_CHAR_NAME = "Eyes"
JSON_SRC = "json_src"
TEXTURE_SRC = "plot_src/"+NEW_CHAR_NAME
TEXTURE_TYPE = "facial-hair"
TEXTURE_NAME = "eyes"
# OVERLAY_NAMES = ['scars', 'skin-texture', 'eyebrows', 'eye-shadow', 'eye-liner', 'eye-effects',
# 'freckles', 'spots', 'skin-blush', 'nose', 'cyborg-top', 'cyborg-bottom',
# 'facial-hair', 'lips', 'cyborg-lips', 'tattoos', 'piercings']
OVERLAY_NAMES = ['scars', 'eyebrows', 'freckles', 'spots', 'facial-hair', 'tattoos', 'piercings']
################# TO EXPAND THE SCALE AND SCOPE OF CONCIOUSNESS #################
# 1. Choose the total number of frames (ie number of faces / json files to generate)
# 2. Have a list of dictionaries, each dict corresponds to the configuration of the face in each frame
# 3. Loop through all frames, for each frame, read the config dict and update json accordingly
################ UTIL FUNCTIONS ################
def list_files_in_directory(directory):
files = os.listdir(directory)
sorted_files = sorted(files)
return sorted_files
def remove_folder(folder_path):
try:
shutil.rmtree(folder_path)
print(f"Folder '{folder_path}' has been removed.")
except FileNotFoundError:
print(f"Folder '{folder_path}' not found.")
except Exception as e:
print(f"An error occurred: {e}")
def remove_file(file_path):
try:
if os.path.exists(file_path):
os.remove(file_path)
print(f"File '{file_path}' has been removed.")
else:
print(f"File '{file_path}' does not exist.")
except Exception as e:
print(f"An error occurred: {e}")
def create_new_character_copy(source_folder, new_folder):
# Copy the entire folder and its contents
shutil.copytree(source_folder, new_folder)
print(f"Folder '{source_folder}' copied to '{new_folder}'")
################ Renames and moves an image to a designated folder for textures ################
def move_and_rename_image(image_path, new_folder_name, new_image_name):
print(image_path)
print(new_folder_name)
print(new_image_name)
# Create a new subfolder if it doesn't exist
if not os.path.exists(new_folder_name):
print(new_folder_name)
# os.mkdir(new_folder_name) ### somehow doesn't work for my Windows computer :(
os.makedirs(new_folder_name)
# Form the new path for the image in the new folder with the new name
new_path = os.path.join(new_folder_name, new_image_name)
# Move and rename the image
shutil.copy(image_path, new_path)
################ Updates the character profile and points to the new texture ################
def update_json(jsonfile, new_json_filename, eye_index):
# Read the JSON file
with open(jsonfile, 'r') as file:
data = json.load(file)
# Modify the value of "facial-hair" overlay
# new_facial_hair_value = "new_facial_hair_value"
texture_sets_dict = data["TextureController"]["TextureSets"]
# texture_sets_dict['facial-hair'] = "29_adult_male_middle_east_realistic"
texture_sets_dict['eyes'] = "eye"+str(eye_index)
# overlay_list = data["TextureController"]["Overlays"]
# facial_hair_dict = {
# "CAT": "facial-hair",
# "VAL": new_texture_folder,
# "COLOR": "#FFFFFFFF"
# }
# overlay_list.append(facial_hair_dict)
# Save the updated JSON data to a new file
with open(new_json_filename, 'w') as outfile:
json.dump(data, outfile, separators=(',', ':'))
print(f"Updated JSON data saved to '{new_json_filename}'")
###################### ACTUAL STEPS ######################
### Clean previous build
remove_folder("models")
remove_file(".charpacks/"+NEW_CHAR_NAME+".charpack")
### Copy data to create a new character pack into the models folder based on the models src
create_new_character_copy("models_src", "models")
# for each png image in the plot_src folder, created a new character profile linking to that png as a texture
# counter = 1
# files_in_directory = list_files_in_directory("./models_src/adult/textures/eye")
# for file in files_in_directory:
# print(file)
# numbered_txtr_name = TEXTURE_NAME+""+str(counter)
# move_and_rename_image(TEXTURE_SRC+"/"+file, "./models/adult/textures/"+TEXTURE_TYPE+"/"+numbered_txtr_name, new_image_name="albedo.png")
# # update_json('models_src/adult/profiles/characters/Victor.json', numbered_txtr_name, 'models/adult/profiles/characters/'+NEW_CHAR_NAME+str(counter)+".json")
# update_json(JSON_SRC+'/'+BASE_CHAR_NAME+'.json', numbered_txtr_name, 'models/adult/profiles/characters/'+NEW_CHAR_NAME+str(counter)+".json")
# counter+=1
for ei in range(1, 24):
update_json(JSON_SRC+'/'+BASE_CHAR_NAME+'.json', 'models/adult/profiles/characters/'+NEW_CHAR_NAME+str(ei)+".json", ei)
### zip all the characters into a charpack
zf = ZipFile("./charpacks/"+NEW_CHAR_NAME+".charpack", "w")
for dirname, subdirs, files in os.walk("models"):
zf.write(dirname)
for filename in files:
zf.write(os.path.join(dirname, filename))
zf.close()