356 lines
9.1 KiB
Python
356 lines
9.1 KiB
Python
|
# Part of VCStudio
|
|||
|
|
# GPLv3 or later
|
|||
|
|
|
|||
|
|
# This file will visualize the rendering process
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import json
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
if len(sys.argv) == 1:
|
|||
|
|
project = input("Project: ")
|
|||
|
|
else:
|
|||
|
|
project = sys.argv[1]
|
|||
|
|
|
|||
|
|
def get_runtime_data(project):
|
|||
|
|
|
|||
|
|
# This will read a specific .json file from the system.
|
|||
|
|
|
|||
|
|
template = {"to_render":True,
|
|||
|
|
"current_progress":"Fra: 69 | Mem: 420 |Sample: 69/420"}
|
|||
|
|
try:
|
|||
|
|
with open(project+"/render_runtime.json") as json_file:
|
|||
|
|
data = json.load(json_file)
|
|||
|
|
except:
|
|||
|
|
data = template.copy()
|
|||
|
|
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
def save_runtime_data(data, project):
|
|||
|
|
with open(project+"/render_runtime.json", 'w') as fp:
|
|||
|
|
json.dump(data, fp, indent=4)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# Colors are used to make the
|
|||
|
|
clr = {
|
|||
|
|
"norm":"\033[00m", # Reset to normal
|
|||
|
|
"bold":"\033[01m", # Bold Text
|
|||
|
|
"ital":"\033[03m", # Italic Text
|
|||
|
|
"undr":"\033[04m", # Underlined
|
|||
|
|
"blnk":"\033[05m", # Blinking
|
|||
|
|
|
|||
|
|
# Text
|
|||
|
|
"tdbl":"\033[30m", # Dark Black
|
|||
|
|
"tdrd":"\033[31m", # Dark Red
|
|||
|
|
"tdgr":"\033[32m", # Dark Green
|
|||
|
|
"tdyl":"\033[33m", # Dark Yellow
|
|||
|
|
"tdbu":"\033[34m", # Dark Blue
|
|||
|
|
"tdma":"\033[35m", # Dark Magenta
|
|||
|
|
"tdcy":"\033[36m", # Dark Cyan
|
|||
|
|
"tdwh":"\033[37m", # Dark White
|
|||
|
|
|
|||
|
|
"tbbl":"\033[90m", # Bright Black
|
|||
|
|
"tbrd":"\033[91m", # Bright Red
|
|||
|
|
"tbgr":"\033[92m", # Bright Green
|
|||
|
|
"tbyl":"\033[93m", # Bright Yellow
|
|||
|
|
"tbbu":"\033[94m", # Bright Blue
|
|||
|
|
"tbma":"\033[95m", # Bright Magenta
|
|||
|
|
"tbcy":"\033[96m", # Bright Cyan
|
|||
|
|
"tbwh":"\033[97m", # Bright White
|
|||
|
|
# Background
|
|||
|
|
"bdbl":"\033[40m", # Dark Black
|
|||
|
|
"bdrd":"\033[41m", # Dark Red
|
|||
|
|
"bdgr":"\033[42m", # Dark Green
|
|||
|
|
"bdyl":"\033[43m", # Dark Yellow
|
|||
|
|
"bdbu":"\033[44m", # Dark Blue
|
|||
|
|
"bdma":"\033[45m", # Dark Magenta
|
|||
|
|
"bdcy":"\033[46m", # Dark Cyan
|
|||
|
|
"bdwh":"\033[47m", # Dark White
|
|||
|
|
|
|||
|
|
"bbbl":"\033[100m", # Bright Black
|
|||
|
|
"bbrd":"\033[101m", # Bright Red
|
|||
|
|
"bbgr":"\033[102m", # Bright Green
|
|||
|
|
"bbyl":"\033[103m", # Bright Yellow
|
|||
|
|
"bbbu":"\033[104m", # Bright Blue
|
|||
|
|
"bbma":"\033[105m", # Bright Magenta
|
|||
|
|
"bbcy":"\033[106m", # Bright Cyan
|
|||
|
|
"bbwh":"\033[108m" # Bright White
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# A function that insures a specific width of the printed part
|
|||
|
|
def wdth(x, n):
|
|||
|
|
|
|||
|
|
# Convert Data to String
|
|||
|
|
mode = "normal"
|
|||
|
|
if type(x) == bool and x == True:
|
|||
|
|
x = "V"
|
|||
|
|
mode = "bdgr"
|
|||
|
|
elif type(x) == bool and x == False:
|
|||
|
|
x = "X"
|
|||
|
|
mode = "bdrd"
|
|||
|
|
else:
|
|||
|
|
x = str(x)
|
|||
|
|
|
|||
|
|
# Turn emogis
|
|||
|
|
#x = emote(x)
|
|||
|
|
|
|||
|
|
# Some characters are too wide. They do not obey the
|
|||
|
|
# monospace of the terminal, thus making it not pretty.
|
|||
|
|
|
|||
|
|
# This is the string of characters which are checked to
|
|||
|
|
good = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!\"#$%&'’()*+,-./:;<=>?@[\]^_`{|}~ йцукенгшщзхъфывапролджэячсмитьбюЙЦУКЕНГШЩЗХЪФЫВАПРОЛДЖЭЯЧСМИТЬБЮёЁ"
|
|||
|
|
|
|||
|
|
# let's filter the string
|
|||
|
|
y = x
|
|||
|
|
|
|||
|
|
|
|||
|
|
x = ""
|
|||
|
|
for i in y:
|
|||
|
|
if i in good:
|
|||
|
|
x = x + i
|
|||
|
|
else:
|
|||
|
|
x = x + "▓"
|
|||
|
|
|
|||
|
|
# Now let's print what we've got.
|
|||
|
|
if len(y) < n:
|
|||
|
|
fac = n-len(y)
|
|||
|
|
fac1 = int(round(fac/2))
|
|||
|
|
fac2 = fac1
|
|||
|
|
while fac1 + fac2 > fac:
|
|||
|
|
fac2 -=1
|
|||
|
|
while fac1 + fac2 < fac:
|
|||
|
|
fac2 +=1
|
|||
|
|
x = (" "*fac1)+x+(" "*fac2)
|
|||
|
|
elif len(y) > n:
|
|||
|
|
if n > 10:
|
|||
|
|
x = x[:n-3]+"..."
|
|||
|
|
else:
|
|||
|
|
x = x[:n]
|
|||
|
|
if mode == "normal":
|
|||
|
|
return x
|
|||
|
|
else:
|
|||
|
|
return clr[mode]+clr["bold"]+x+clr["norm"]
|
|||
|
|
|
|||
|
|
def tsize():
|
|||
|
|
|
|||
|
|
# This funtion will get the size of the terminal and
|
|||
|
|
# return it to the variables provided width, height
|
|||
|
|
|
|||
|
|
# On some systems this may not work. So there is a
|
|||
|
|
# try function.
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# Getting the size of the terminal
|
|||
|
|
import os
|
|||
|
|
w, h = os.get_terminal_size()
|
|||
|
|
|
|||
|
|
# Sometimes when the terminal width is either
|
|||
|
|
# even or odd. It breaks code for some other
|
|||
|
|
# thing written differenly. For example:
|
|||
|
|
|
|||
|
|
# You may have an even width ( like 84 ) when
|
|||
|
|
# writing a function. And then it works on different
|
|||
|
|
# widths as well like 62 and 80 and 48. All of them
|
|||
|
|
# are still even. Then you scale the terminal to be
|
|||
|
|
# something off like 63 and the function breaks. You
|
|||
|
|
# have one character too much or one character too little.
|
|||
|
|
|
|||
|
|
# This is why I do not want to have a difference. And
|
|||
|
|
# force width to be one less, if it's not divisible by 2.
|
|||
|
|
|
|||
|
|
if not w % 2:
|
|||
|
|
w = w - 1
|
|||
|
|
|
|||
|
|
return w, h
|
|||
|
|
|
|||
|
|
except:
|
|||
|
|
|
|||
|
|
# If, by any reason the terminal can't get it's size.
|
|||
|
|
# We want to return some size regardless.
|
|||
|
|
|
|||
|
|
w = 60
|
|||
|
|
h = 20
|
|||
|
|
|
|||
|
|
return w, h
|
|||
|
|
|
|||
|
|
|
|||
|
|
def table(data, number=True):
|
|||
|
|
|
|||
|
|
# This function will present data in a pretty table thing.
|
|||
|
|
|
|||
|
|
# So let's first of all get the size of the terminal
|
|||
|
|
w, h = tsize()
|
|||
|
|
|
|||
|
|
if number:
|
|||
|
|
w = w - 4
|
|||
|
|
|
|||
|
|
# Then let's draw the categories for this we need to extract
|
|||
|
|
# it's sizes. If there is no 'size' variable the sizes of
|
|||
|
|
# each category will be spread equally.
|
|||
|
|
|
|||
|
|
size = [] # Here the size will go as pure character value.
|
|||
|
|
|
|||
|
|
if "size" in data:
|
|||
|
|
for i in data["size"]:
|
|||
|
|
size.append(int(( w - 10 ) / sum(data["size"]) * i))
|
|||
|
|
|
|||
|
|
while sum(size) < w - 10:
|
|||
|
|
size[-1] += 1
|
|||
|
|
|
|||
|
|
# printing categories
|
|||
|
|
nb = ""
|
|||
|
|
if number:
|
|||
|
|
nb = " "
|
|||
|
|
s = " "+clr["bdma"]+" "+clr["tbwh"]+nb
|
|||
|
|
|
|||
|
|
for n, item in enumerate(data["categories"]):
|
|||
|
|
s = s + wdth(item.upper(), size[n])
|
|||
|
|
print(s+clr["bdma"]+" "+clr["norm"])
|
|||
|
|
|
|||
|
|
size[-1] += 1
|
|||
|
|
|
|||
|
|
# printing items
|
|||
|
|
for b, i in enumerate(data["data"]):
|
|||
|
|
|
|||
|
|
# dark bright sequence thingy
|
|||
|
|
if b % 2:
|
|||
|
|
d = "b"
|
|||
|
|
else:
|
|||
|
|
d = "d"
|
|||
|
|
nb = ""
|
|||
|
|
if number:
|
|||
|
|
nb = clr["tbwh"]+wdth(b,4)
|
|||
|
|
s = " "+clr["bdma"]+" "+nb+clr["norm"]+clr["b"+d+"bu"]#+clr["tbwh"]
|
|||
|
|
for n, item in enumerate(i):
|
|||
|
|
s = s +clr["b"+d+"bu"]+ clr["tbwh"]+wdth(item, size[n]-1)+clr["bdma"]+" "
|
|||
|
|
print(s+clr["norm"])
|
|||
|
|
|
|||
|
|
def center(line, c="bdma", blink=False):
|
|||
|
|
|
|||
|
|
# This funtiocn will bring a given string of text
|
|||
|
|
# in the center of the terminal with a nice backgroud
|
|||
|
|
# around it.
|
|||
|
|
|
|||
|
|
|
|||
|
|
w, h = tsize()
|
|||
|
|
|
|||
|
|
if blink:
|
|||
|
|
blink = clr["blnk"]
|
|||
|
|
else:
|
|||
|
|
blink = ""
|
|||
|
|
|
|||
|
|
if len(line) % 2:
|
|||
|
|
line = line + " "
|
|||
|
|
|
|||
|
|
if len(line) < w - 11:
|
|||
|
|
print(" "+clr[c],
|
|||
|
|
wdth(" ", int((w-10)/2 - (len(line)/2))),
|
|||
|
|
clr["bold"]+clr["tbwh"]+blink+line,
|
|||
|
|
wdth(" ", int((w-10)/2 - (len(line)/2))-1),
|
|||
|
|
clr["norm"])
|
|||
|
|
else:
|
|||
|
|
print(" "+clr[c],
|
|||
|
|
clr["bold"]+clr["tbwh"]+blink+wdth(line,w-10),
|
|||
|
|
clr["norm"])
|
|||
|
|
|
|||
|
|
running_for = 0
|
|||
|
|
lastframe = 0
|
|||
|
|
maxtime = 1000000
|
|||
|
|
|
|||
|
|
while True:
|
|||
|
|
|
|||
|
|
running_for = running_for + 1
|
|||
|
|
|
|||
|
|
runtime = get_runtime_data(project)
|
|||
|
|
if not runtime.get("to_render"):
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
x,y = tsize()
|
|||
|
|
|
|||
|
|
if y > 8:
|
|||
|
|
print("\n\n")
|
|||
|
|
|
|||
|
|
process = runtime.get("current_progress", "").split(" | ")
|
|||
|
|
try:
|
|||
|
|
del runtime["current_progress"]
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
data_print = {"categories":list(runtime.keys()),
|
|||
|
|
"size":[1]*(len(runtime)),
|
|||
|
|
"data":[list(runtime.values())]}
|
|||
|
|
|
|||
|
|
table(data_print, False)
|
|||
|
|
|
|||
|
|
data_print = {"categories":[""]*(len(process)),
|
|||
|
|
"size":[1]*(len(process)),
|
|||
|
|
"data":[process]}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
table(data_print, False)
|
|||
|
|
center("")
|
|||
|
|
|
|||
|
|
|
|||
|
|
# Latest frames data
|
|||
|
|
|
|||
|
|
filename = runtime.get("current_file", "")
|
|||
|
|
folder = filename[:filename.rfind("/")]+"/extra"
|
|||
|
|
savefile = folder+filename[filename.rfind("/"):]+".json"
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
with open(project+savefile) as json_file:
|
|||
|
|
data = json.load(json_file)
|
|||
|
|
except:
|
|||
|
|
data = {}
|
|||
|
|
|
|||
|
|
listoftimesx = data.get("analytics", {}).get(data.get("save_folder", ""), [])
|
|||
|
|
listoftimes = {}
|
|||
|
|
minframe = data.get("start_frame", 0)
|
|||
|
|
for i in listoftimesx:
|
|||
|
|
try:
|
|||
|
|
listoftimes[int(i)] = listoftimesx[i]
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
try:
|
|||
|
|
maxframe = max(list(listoftimes.keys()))
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
if maxframe > lastframe:
|
|||
|
|
lastframe = maxframe
|
|||
|
|
running_for = 0
|
|||
|
|
maxtime = max(list(listoftimes.values()))
|
|||
|
|
maxtime = max(maxtime, running_for*1000000)
|
|||
|
|
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
|
|||
|
|
for i in range(max(minframe,maxframe-y+16), maxframe+1):
|
|||
|
|
|
|||
|
|
amount = int((x-14)/(maxtime-minframe)*(listoftimes[i]-minframe))
|
|||
|
|
|
|||
|
|
|
|||
|
|
print(" ",wdth(i, 4), "▒"*amount)
|
|||
|
|
except Exception as e:
|
|||
|
|
#print(e)
|
|||
|
|
maxframe = 0
|
|||
|
|
#maxtime = 1
|
|||
|
|
lastframe = maxframe
|
|||
|
|
running_for = 0
|
|||
|
|
|
|||
|
|
amount = int((x-14)/maxtime*running_for*1000000)
|
|||
|
|
|
|||
|
|
print(" " , wdth(maxframe+1, 4), "█"*amount)
|
|||
|
|
|
|||
|
|
|
|||
|
|
amount = int((x-14)/(data.get("end_frame", maxframe+1)-minframe)*(maxframe+1-minframe))
|
|||
|
|
|
|||
|
|
print()
|
|||
|
|
print(" "+clr["tbma"]+wdth(data.get("end_frame", maxframe+1), 4)+" "+("█"*amount)+clr["norm"])
|
|||
|
|
|
|||
|
|
time.sleep(1)
|
|||
|
|
os.system("clear")
|