# -*- coding: utf-8 -*-
#
# This file is subject to the terms and conditions defined in
# file 'LICENSE.txt', which is part of this source code package.
#
import numpy as np
import os
from BasicTools.IO.WriterBase import WriterBase as WriterBase
import BasicTools.IO.GeofReader as GR
import BasicTools.IO.UtReader as UR
import BasicTools.Containers.ElementNames as EN
from BasicTools.FE.IntegrationsRules import LagrangeIsoParam
from BasicTools.Helpers.TextFormatHelper import TFormat
[docs]class UtMerger(WriterBase):
"This class can generate monolithic .ut, .goef, .ctnod, .node, .integ from files obtained from a parallel Z-set computation"
def __init__(self):
super(UtMerger,self).__init__()
self.timeSteps = "all"
def __str__(self):
res = 'UtMerge : \n'
res += ' Name : '+str(self.name)+'\n'
return res
[docs] def SetName(self,name):
self.name = name
[docs] def SetdataFolder(self,folder):
self.dataFolder = folder
from os import listdir
from os.path import isfile, join
temp = [f.split(".") for f in listdir(self.dataFolder) if isfile(join(self.dataFolder, f))]
temp2 = []
count = 0
for f in temp:
temp2.append([])
for fi in f:
splitted = fi.split("-")
for fil in splitted:
try:
temp2[count].append(int(fil))
except ValueError:
temp2[count].append(fil)
count += 1
subdomains = []
for f in temp2:
if 'ut' in f and self.name in f:
for fi in f:
if type(fi) == int:
subdomains.append(fi)
self.nbsd = np.max(subdomains)
print("Number of found subdomains:", self.nbsd)
[docs] def SetOutputFolder(self,outputFolder):
self.outputFolder = outputFolder
[docs] def SetTimeSteps(self,iterator):
self.timeSteps = iterator
[docs] def Merge(self):
localDataInteg = []
localDataNode = []
localIdstotreat = []
localoriginalIDNodes = []
from BasicTools.Helpers.ProgressBar import printProgressBar
printProgressBar(0, self.nbsd, prefix = 'Reading Local solutions:', suffix = 'Complete', length = 50)
#Read each subdomain computation
for sd in range(1,self.nbsd+1):
sdString = "-" + str(sd).zfill(3)
reader = UR.UtReader()
reader.SetFileName(self.dataFolder + self.name + sdString + ".ut")
reader.ReadMetaData()
nbeTimeSteps = reader.time.shape[0]
if self.timeSteps != "all":
reader.time = reader.time[self.timeSteps,:]
if len(reader.time.shape) == 1:
reader.time.shape = (1,nbeTimeSteps)
localMesh = GR.ReadGeof(fileName = self.dataFolder+reader.meshfile,readElset=False,readFaset=False,printNotRead=False)
Tag3D(localMesh)
idstotreat, metaDataMesh3D = Return3DElements(localMesh)
originalIDNodes = np.array(localMesh.originalIDNodes-1, dtype=int)
reader.atIntegrationPoints = True
dataInteg = {}
for din in reader.integ:
dataInteg[din] = np.empty((reader.meshMetadata['nbIntegrationPoints'],nbeTimeSteps))
for timeStep in range(nbeTimeSteps):
dataInteg[din][:,timeStep] = reader.ReadField(fieldname=din, timeIndex=int(reader.time[timeStep,0])-1)
reader.atIntegrationPoints = False
dataNode = {}
for din in reader.node:
dataNode[din] = np.empty((reader.meshMetadata['nbNodes'],nbeTimeSteps))
for timeStep in range(nbeTimeSteps):
dataNode[din][:,timeStep] = reader.ReadField(fieldname=din, timeIndex=int(reader.time[timeStep,0])-1)
localDataInteg.append(dataInteg)
localDataNode.append(dataNode)
localIdstotreat.append(idstotreat)
localoriginalIDNodes.append(originalIDNodes)
printProgressBar(sd, self.nbsd, prefix = 'Reading Local solutions:', suffix = 'Complete', length = 50)
print("Local solutions have been read")
cutGeof = GeofFromCut(self.dataFolder, self.name)
globalMesh = GR.ReadGeof(fileName = self.dataFolder + cutGeof,readElset=False,readFaset=False,printNotRead=False)
Tag3D(globalMesh)
globalIdstotreat, metaDataMesh3D = Return3DElements(globalMesh)
#globalMesh.originalIDNodes = np.array(globalMesh.originalIDNodes-1, dtype=int)
globaldataInteg = {}
for din in reader.integ:
globaldataInteg[din] = np.empty((metaDataMesh3D.NGauss,nbeTimeSteps))
globaldataNode = {}
for din in reader.node:
globaldataNode[din] = np.empty((metaDataMesh3D.Nodes,nbeTimeSteps))
#write .integ
data_integ = np.empty(len(reader.integ)*metaDataMesh3D.NGauss*(nbeTimeSteps))
nGpE = metaDataMesh3D.NGaussperEl
count0 = 0
printProgressBar(0, nbeTimeSteps, prefix = 'Writing global .integ:', suffix = 'Complete', length = 50)
for timeStep in range(nbeTimeSteps):
field = np.empty((len(reader.integ),metaDataMesh3D.NGauss))
count = 0
for sd in range(self.nbsd):
for k in range(len(reader.integ)):
count = 0
for el in localIdstotreat[sd]:
field[k,el*nGpE:(el+1)*nGpE] = localDataInteg[sd][reader.integ[k]][count:count+nGpE,timeStep]
count += nGpE
for m in range(len(globalIdstotreat)):
for k in range(len(reader.integ)):
data_integ[count0:count0+nGpE] = field[k,nGpE*m:nGpE*(m+1)]
count0 += nGpE
printProgressBar(timeStep+1, nbeTimeSteps, prefix = 'Writing global .integ:', suffix = 'Complete', length = 50)
data_integ.astype(np.float32).byteswap().tofile(self.outputFolder + self.name + ".integ")
print("Global .integ has been written")
#write .node
data_node = np.zeros(len(reader.node)*metaDataMesh3D.Nodes*(nbeTimeSteps))
printProgressBar(0, nbeTimeSteps, prefix = 'Writing global .node:', suffix = 'Complete', length = 50)
for timeStep in range(nbeTimeSteps):
count = 0
for sd in range(self.nbsd):
for k in range(len(reader.node)):
indices = list(map(lambda x: x + len(reader.node)*metaDataMesh3D.Nodes*timeStep+k*metaDataMesh3D.Nodes, localoriginalIDNodes[sd]))
data_node[indices] = localDataNode[sd][reader.node[k]][:,timeStep]
printProgressBar(timeStep+1, nbeTimeSteps, prefix = 'Writing global .node:', suffix = 'Complete', length = 50)
data_node.astype(np.float32).byteswap().tofile(self.outputFolder + self.name + ".node")
print("Global .node has been written")
#write .ut
try:
relativePath = os.path.relpath(self.dataFolder, self.outputFolder)
except:
print("Error using relative path. Using absolute path (files generated are not relocatable)")
relativePath = self.dataFolder
__string = "**meshfile " + relativePath + os.sep + cutGeof+"\n"
with open(self.dataFolder + self.name + "-001.ut", 'r') as inFile:
inFile.readline()
for i in range(3):
__string += inFile.readline()
with open(self.outputFolder + self.name + ".ut", "w") as outFile:
outFile.write(__string)
for timeStep in range(nbeTimeSteps):
line = ""
for i in range(4):
line += str(int(reader.time[timeStep,i]))+" "
line += str(reader.time[timeStep,4])+"\n"
outFile.write(line)
[docs]def GeofFromCut(dataFolder, cutName):
cutFile = open(dataFolder+cutName+".cut", 'r')
strings = cutFile.readlines()
return strings[1].split()[1]
[docs]def Tag3D(mesh):
for name, data in mesh.elements.items():
if EN.dimension[name] == 3:
mesh.GetElementsOfType(name).tags.CreateTag('3D').SetIds(mesh.GetElementsOfType(name).originalIds-1)
[docs]def Return3DElements(mesh):
class metaDataMesh3DClass():
pass
for name, data in mesh.elements.items():
if '3D' in data.tags:
idstotreat = data.tags['3D'].GetIds()
metaDataMesh3D = metaDataMesh3DClass()
metaDataMesh3D.NnodeperEl = EN.numberOfNodes[name]
metaDataMesh3D.p, metaDataMesh3D.w = LagrangeIsoParam[name]
metaDataMesh3D.NGaussperEl = len(metaDataMesh3D.w)
metaDataMesh3D.NGauss = data.GetNumberOfElements()*metaDataMesh3D.NGaussperEl
metaDataMesh3D.nbElements = data.GetNumberOfElements()
metaDataMesh3D.Nodes = mesh.GetNumberOfNodes()
return idstotreat, metaDataMesh3D
raise('no 3D tag detected')
[docs]def CheckIntegrity():
from BasicTools.Helpers.Tests import TestTempDir
tempdir = TestTempDir.GetTempPath()
import BasicTools.TestData as BasicToolsTestData
##################################
# EXEMPLE SYNTAXE DU MERGER
import BasicTools.IO.Parallel.UtMerger as UM
merger = UM.UtMerger()
merger.SetName("cube")
merger.SetdataFolder(BasicToolsTestData.GetTestDataPath() + "UtParExample"+os.sep)
merger.SetOutputFolder(tempdir)
merger.Merge()
##################################
import filecmp
print(TFormat.InRed("node files equals ? "+ str(filecmp.cmp(tempdir + "cube.node", BasicToolsTestData.GetTestDataPath() + "UtParExample"+os.sep+"cube.node", shallow=False))))
print(TFormat.InRed("integ files equals ? "+ str(filecmp.cmp(tempdir + "cube.integ", BasicToolsTestData.GetTestDataPath() + "UtParExample"+os.sep+"cube.integ", shallow=False))))
print(tempdir)
return "ok"
if __name__ == '__main__':
print((CheckIntegrity()))# pragma: no cover