def write_silo_datfile(fname,mname,cycle=0, time=0, nodefields=[], edgefields=[]):
from pyvisfile.silo import SiloFile, IntVector, DB_ZONETYPE_BEAM,\
DB_NODECENT, DB_ZONECENT, DBOPT_CYCLE, DBOPT_DTIME, DBOPT_TIME, DB_CLOBBER
from pyvisfile.silo import DBObjectType as DBOBjectType
import numpy as np
silo = SiloFile(fname, mode=DB_CLOBBER)
silo.put_multimesh('foo', [(mname+":foo",DBOBjectType.DB_UCDMESH)])
def putvar(n,fo,LOC):
if len(f.shape)==1 or f.shape[1]==1:
silo.put_ucdvar1("node_"+n,"foo",
np.asarray(f.T,order="C",dtype=np.double),
LOC, {DBOPT_CYCLE:cycle,DBOPT_DTIME:float(time),DBOPT_TIME:float(time)})
elif f.shape[1]==2:
silo.put_ucdvar("node_"+n,"foo", [n+"x",n+"y"],
np.asarray(f.T,order="C",dtype=np.double),
LOC, {DBOPT_CYCLE:cycle,DBOPT_DTIME:float(time),DBOPT_TIME:float(time)})
else:
silo.put_ucdvar("node_"+n,"foo", [n+"x",n+"y",n+"z"],
np.asarray(f.T,order="C",dtype=np.double),
LOC, {DBOPT_CYCLE:cycle,DBOPT_DTIME:float(time),DBOPT_TIME:float(time)})
for n,f in nodefields:
putvar(n,f,DB_NODECENT)
for n,f in edgefields:
putvar(n,f,DB_ZONECENT)
silo.close()
def silo_write_meshfile(fname, X, elems):
try:
from pyvisfile.silo import SiloFile, IntVector, DB_ZONETYPE_BEAM, DB_ZONETYPE_HEX, \
DB_NODECENT, DB_ZONECENT, DBOPT_CYCLE, DBOPT_DTIME, DBOPT_TIME, DB_CLOBBER
except ImportError:
print("Need PyVisFile to write silo files. Use vtk for now.")
raise
import numpy as np
silo = SiloFile(fname, mode=DB_CLOBBER)
# TODO: Check to see if it should be connections or bricks
zonelist_name = "foo_zonelist"
nodelist = IntVector()
nodelist.extend(int(i) for i in elems[:, :].flat)
shapetypes = IntVector()
if elems.shape[1] == 2:
shapetypes.append(DB_ZONETYPE_BEAM)
elif elems.shape[1] == 8:
shapetypes.append(DB_ZONETYPE_HEX)
shapesizes = IntVector()
shapesizes.append(elems.shape[1])
shapecounts = IntVector()
shapecounts.append(len(elems))
silo.put_zonelist_2(zonelist_name, len(elems), 2, nodelist, 0, 0,
shapetypes, shapesizes, shapecounts)
silo.put_ucdmesh("foo", [], np.asarray(X.T, order="C"), len(elems),
zonelist_name, None)
silo.close()
print("Wrote file " + fname)
def main():
path = tempfile.mkdtemp()
try:
silo = SiloFile(os.path.join(path, 'file.silo'))
add_to_silo_file(silo, Mesh())
silo.close()
finally:
shutil.rmtree(path)
def main():
path = tempfile.mkdtemp()
try:
silo = SiloFile(os.path.join(path, 'file.silo'))
add_to_silo_file(silo, Mesh())
silo.close()
finally:
shutil.rmtree(path)
def silo_write_datafile(fname,
mname,
cycle=0,
time=0,
nodefields={},
cellfields={}):
try:
from pyvisfile.silo import SiloFile, IntVector, DB_ZONETYPE_BEAM,\
DB_NODECENT, DB_ZONECENT, DBOPT_CYCLE, DBOPT_DTIME, DBOPT_TIME, DB_CLOBBER
from pyvisfile.silo import DBObjectType as DBOBjectType
except ImportError:
print("Need PyVisFile to write silo files. Use vtk for now.")
raise
import numpy as np
silo = SiloFile(fname, mode=DB_CLOBBER)
silo.put_multimesh('foo', [(mname + ":foo", DBOBjectType.DB_UCDMESH)])
def putvar(n, fo, LOC):
if len(f.shape) == 1 or f.shape[1] == 1:
silo.put_ucdvar1(
"node_" + n, "foo", np.asarray(f.T, order="C",
dtype=np.double), LOC, {
DBOPT_CYCLE: cycle,
DBOPT_DTIME: float(time),
DBOPT_TIME: float(time)
})
elif f.shape[1] == 2:
silo.put_ucdvar(
"node_" + n, "foo", [n + "x", n + "y"],
np.asarray(f.T, order="C", dtype=np.double), LOC, {
DBOPT_CYCLE: cycle,
DBOPT_DTIME: float(time),
DBOPT_TIME: float(time)
})
else:
silo.put_ucdvar(
"node_" + n, "foo", [n + "x", n + "y", n + "z"],
np.asarray(f.T, order="C", dtype=np.double), LOC, {
DBOPT_CYCLE: cycle,
DBOPT_DTIME: float(time),
DBOPT_TIME: float(time)
})
for n, f in nodefields.iteritems():
putvar(n, f, DB_NODECENT)
for n, f in cellfields.iteritems():
putvar(n, f, DB_ZONECENT)
silo.close()
print("Wrote file " + fname)
def write_silo(fname, H,X,cycle=0, time=0,nodefields=[], edgefields=[], PUTMESH=True, PUTCONN=True):
from pyvisfile.silo import SiloFile, IntVector, DB_ZONETYPE_BEAM, DB_NODECENT, DB_ZONECENT, DBOPT_CYCLE, DBOPT_DTIME, DBOPT_TIME, DB_CLOBBER
import numpy as np
silo = SiloFile(fname, mode=DB_CLOBBER)
pair_edges = H.view()[0]
zonelist_name = "foo_zonelist"
nodelist = IntVector()
nodelist.extend( int(i) for i in pair_edges[:,0:2].flat)
shapetypes = IntVector()
shapetypes.append(DB_ZONETYPE_BEAM)
shapesizes = IntVector()
shapesizes.append(2)
shapecounts = IntVector()
shapecounts.append(len(pair_edges))
#from IPython import embed
#embed()
if PUTCONN:
silo.put_zonelist_2(zonelist_name, len(pair_edges), 2, nodelist,
0,0, shapetypes, shapesizes, shapecounts)
if PUTMESH:
silo.put_ucdmesh("foo", [],
np.asarray(X.T,order="C"), len(pair_edges),
zonelist_name, None)
def putvar(n,fo,LOC):
if len(f.shape)==1 or f.shape[1]==1:
silo.put_ucdvar1("node_"+n,"foo",
np.asarray(f,order="C",dtype=np.double),
LOC, {DBOPT_CYCLE:cycle,DBOPT_DTIME:float(time),DBOPT_TIME:float(time)})
elif f.shape[1]==2:
silo.put_ucdvar("node_"+n,"foo", [n+"x",n+"y"],
np.asarray(f.T,order="C",dtype=np.double),
LOC, {DBOPT_CYCLE:cycle,DBOPT_DTIME:float(time),DBOPT_TIME:float(time)})
else:
silo.put_ucdvar("node_"+n,"foo", [n+"x",n+"y",n+"z"],
np.asarray(f.T,order="C",dtype=np.double),
LOC, {DBOPT_CYCLE:cycle,DBOPT_DTIME:float(time),DBOPT_TIME:float(time)})
for n,f in nodefields:
putvar(n,f,DB_NODECENT)
for n,f in edgefields:
putvar(n,f,DB_ZONECENT)
silo.close()
def write_silo_meshfile(fname, H,X):
from pyvisfile.silo import SiloFile, IntVector, DB_ZONETYPE_BEAM, DB_NODECENT, DB_ZONECENT, DBOPT_CYCLE, DBOPT_DTIME, DBOPT_TIME, DB_CLOBBER
import numpy as np
silo = SiloFile(fname, mode=DB_CLOBBER)
pair_edges = H.view()[0]
zonelist_name = "foo_zonelist"
nodelist = IntVector()
nodelist.extend( int(i) for i in pair_edges[:,0:2].flat)
shapetypes = IntVector()
shapetypes.append(DB_ZONETYPE_BEAM)
shapesizes = IntVector()
shapesizes.append(2)
shapecounts = IntVector()
shapecounts.append(len(pair_edges))
silo.put_zonelist_2(zonelist_name, len(pair_edges), 2, nodelist,
0,0, shapetypes, shapesizes, shapecounts)
silo.put_ucdmesh("foo", [],
np.asarray(X.T,order="C"), len(pair_edges),
zonelist_name, None)
silo.close()
def main():
from pyvisfile.silo import SiloFile
silo = SiloFile("out.silo")
add_to_silo_file(silo, Mesh())
silo.close()
def main():
from pyvisfile.silo import SiloFile
silo = SiloFile("out.silo")
add_to_silo_file(silo, Mesh())
silo.close()
def test_silo_unstructured():
from pyvisfile.silo import SiloFile
silo = SiloFile("out.silo")
add_to_silo_file(silo, Mesh())
silo.close()