3pp-mirror/Meshroom
1
0
Fork 0
mirror of https://github.com/alicevision/Meshroom.git synced 2025-04-28 17:57:16 +02:00
Code Issues Projects Releases Packages Wiki Activity

[processGraph] statistics in a separate file (no more inside status file)

Browse source
This commit is contained in:
Fabien Castan 2017-09-21 02:13:10 +02:00
parent 3f001d46d8
commit b7d906adb7
1 changed files with 32 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
meshroom/processGraph/graph.py
View file

@ -160,17 +160,15 @@ class StatusData:
self.status = Status.NONE
self.nodeName = nodeName
self.nodeType = nodeType
self.statistics = Statistics()
self.graph = ''
def toDict(self):
return {k: (v.toDict() if getattr(v, "toDict", None) else v) for k, v in self.__dict__.items()}
return self.__dict__
def fromDict(self, d):
self.status = Status._member_map_[d['status']]
self.nodeName = d['nodeName']
self.nodeType = d['nodeType']
self.statistics.fromDict(d['statistics'])
self.graph = d['graph']
@ -182,7 +180,7 @@ class StatisticsThread(threading.Thread):
threading.Thread.__init__(self)
self.node = node
self.running = True
self.statistics = self.node.status.statistics
self.statistics = self.node.statistics
self.initStats()
def initStats(self):
@ -205,7 +203,7 @@ class StatisticsThread(threading.Thread):
self.statistics.ramUsage.append(psutil.virtual_memory().percent)
self.statistics.swapUsage.append(psutil.swap_memory().percent)
self.statistics.vramUsage.append(0)
self.node.saveStatusFile()
self.node.saveStatistics()
def run(self):
while self.running:
@ -228,6 +226,7 @@ class Node:
for k, v in kwargs.items():
self.attributes[k]._value = v
self.status = StatusData(self.name, self.nodeType())
self.statistics = Statistics()
def __getattr__(self, k):
try:
@ -320,6 +319,9 @@ class Node:
def statusFile(self):
return os.path.join(pg.cacheFolder, self.internalFolder(), 'status')
def statisticsFile(self):
return os.path.join(pg.cacheFolder, self.internalFolder(), 'statistics')
def logFile(self):
return os.path.join(pg.cacheFolder, self.internalFolder(), 'log')
@ -349,6 +351,27 @@ class Node:
json.dump(data, jsonFile, indent=4)
shutil.move(statusFilepathWriting, statusFilepath)
def updateStatisticsFromCache(self):
"""
"""
statisticsFile = self.statisticsFile()
if not os.path.exists(statisticsFile):
return
with open(statisticsFile, 'r') as jsonFile:
statisticsData = json.load(jsonFile)
self.statistics.fromDict(statisticsData)
def saveStatistics(self):
data = self.statistics.toDict()
statisticsFilepath = self.statisticsFile()
folder = os.path.dirname(statisticsFilepath)
if not os.path.exists(folder):
os.makedirs(folder)
statisticsFilepathWriting = statisticsFilepath + '.writing.' + str(uuid.uuid4())
with open(statisticsFilepathWriting, 'w') as jsonFile:
json.dump(data, jsonFile, indent=4)
shutil.move(statisticsFilepathWriting, statisticsFilepath)
def upgradeStatusTo(self, newStatus):
if int(newStatus.value) <= int(self.status.status.value):
print('WARNING: downgrade status on node "{}" from {} to {}'.format(self.name, self.status.status.name,
@ -591,6 +614,10 @@ class Graph:
for node in self.nodes.values():
node.updateStatusFromCache()
def updateStatisticsFromCache(self):
for node in self.nodes.values():
node.updateStatisticsFromCache()
def update(self):
self.updateInternals()
self.updateStatusFromCache()