/usr/lib/python3-escript-mpi/esys/weipa/__init__.py is in python3-escript-mpi 5.1-5.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 | ##############################################################################
#
# Copyright (c) 2003-2017 by The University of Queensland
# http://www.uq.edu.au
#
# Primary Business: Queensland, Australia
# Licensed under the Apache License, version 2.0
# http://www.apache.org/licenses/LICENSE-2.0
#
# Development until 2012 by Earth Systems Science Computational Center (ESSCC)
# Development 2012-2013 by School of Earth Sciences
# Development from 2014 by Centre for Geoscience Computing (GeoComp)
#
##############################################################################
from __future__ import print_function, division
__copyright__="""Copyright (c) 2003-2017 by The University of Queensland
http://www.uq.edu.au
Primary Business: Queensland, Australia"""
__license__="""Licensed under the Apache License, version 2.0
http://www.apache.org/licenses/LICENSE-2.0"""
__url__="https://launchpad.net/escript-finley"
from .weipacpp import visitInitialize, visitPublishData
__nodocorecursion=['weipacpp']
def interpolateEscriptData(domain, data):
"""
esys.weipa does not support the function spaces Solution and
ReducedSolution. This function interpolates Data defined on those function
spaces to compatible alternatives.
"""
from esys.escript import Solution, ReducedSolution
from esys.escript import ContinuousFunction, ReducedContinuousFunction
from esys.escript.util import interpolate
new_data={}
for n,d in sorted(list(data.items()), key=lambda x: x[0]):
if not d.isEmpty():
fs=d.getFunctionSpace()
if domain is None:
domain=fs.getDomain()
elif domain != fs.getDomain():
raise ValueError("weipa: All Data must be on the same domain!")
new_data[n]=d
try:
if fs == Solution(domain):
new_data[n]=interpolate(d, ContinuousFunction(domain))
elif domain.getDescription().startswith("speckley"):
new_data[n]=interpolate(d, ContinuousFunction(domain))
elif fs == ReducedSolution(domain):
new_data[n]=interpolate(d, ReducedContinuousFunction(domain))
except RuntimeError as e:
if str(e).startswith("FunctionSpaceException"):
pass
else:
raise e
return domain,new_data
def createDataset(domain=None, **data):
"""
Creates and returns an esys.weipa dataset consisting of a Domain and Data
objects. The returned object provides methods to access and export data.
"""
from .weipacpp import EscriptDataset
dataset=EscriptDataset()
domain,new_data=interpolateEscriptData(domain, data)
dataset.setDomain(domain)
for n,d in sorted(new_data.items()):
#TODO: data units are not supported here yet
dataset.addData(d, n, "")
return dataset
def saveSilo(filename, domain=None, write_meshdata=False, time=0., cycle=0,
**data):
"""
Writes `Data` objects and their mesh to a file using the SILO file format.
Example::
temp=Scalar(..)
v=Vector(..)
saveSilo("solution.silo", temperature=temp, velocity=v)
``temp`` and ``v`` are written to "solution.silo" where ``temp`` is named
"temperature" and ``v`` is named "velocity".
:param filename: name of the output file ('.silo' is added if required)
:type filename: ``str``
:param domain: domain of the `Data` objects. If not specified, the domain
of the given `Data` objects is used.
:type domain: `escript.Domain`
:param write_meshdata: whether to save mesh-related data such as element
identifiers, ownership etc. This is mainly useful
for debugging.
:type write_meshdata: ``bool``
:param time: the timestamp to save within the file
:type time: ``float``
:param cycle: the cycle (or timestep) of the data
:type cycle: ``int``
:keyword <name>: writes the assigned value to the Silo file using <name> as
identifier
:note: All data objects have to be defined on the same domain but they may
be defined on separate `FunctionSpace` s.
"""
dataset = createDataset(domain, **data)
dataset.setCycleAndTime(cycle, time)
dataset.setSaveMeshData(write_meshdata)
return dataset.saveSilo(filename)
def saveVTK(filename, domain=None, metadata='', metadata_schema=None,
write_meshdata=False, time=0., cycle=0, **data):
"""
Writes `Data` objects and their mesh to a file using the VTK XML file
format.
Example::
temp=Scalar(..)
v=Vector(..)
saveVTK("solution.vtu", temperature=temp, velocity=v)
``temp`` and ``v`` are written to "solution.vtu" where ``temp`` is named
"temperature" and ``v`` is named "velocity".
Meta tags, e.g. a timeStamp, can be added to the file, for instance::
tmp=Scalar(..)
v=Vector(..)
saveVTK("solution.vtu", temperature=tmp, velocity=v,
metadata="<timeStamp>1.234</timeStamp>",
metadata_schema={"gml":"http://www.opengis.net/gml"})
The argument ``metadata_schema`` allows the definition of name spaces with
a schema used in the definition of meta tags.
:param filename: name of the output file ('.vtu' is added if required)
:type filename: ``str``
:param domain: domain of the `Data` objects. If not specified, the domain
of the given `Data` objects is used.
:type domain: `escript.Domain`
:keyword <name>: writes the assigned value to the VTK file using <name> as
identifier
:param metadata: additional XML meta data which are inserted into the VTK
file. The meta data are marked by the tag ``<MetaData>``.
:type metadata: ``str``
:param metadata_schema: assigns schemas to namespaces which have been used
to define meta data.
:type metadata_schema: ``dict`` with ``metadata_schema[<namespace>]=<URI>``
to assign the scheme ``<URI>`` to the name space
``<namespace>``.
:param write_meshdata: whether to save mesh-related data such as element
identifiers, ownership etc. This is mainly useful
for debugging.
:type write_meshdata: ``bool``
:param time: the timestamp to save within the file, seperate to metadata
:type time: ``float``
:param cycle: the cycle (or timestep) of the data
:type cycle: ``int``
:note: All data objects have to be defined on the same domain. They may not
be in the same `FunctionSpace` but not all combinations of
`FunctionSpace` s can be written to a single VTK file.
Typically, data on the boundary and on the interior cannot be mixed.
"""
dataset = createDataset(domain, **data)
dataset.setCycleAndTime(cycle, time)
ss=''
ms=''
if not metadata is None:
ms=metadata
if not metadata_schema is None:
if hasattr(metadata_schema, 'items'):
for i,p in sorted(metadata_schema.items(), key=lambda x: x[0]):
ss="%s xmlns:%s=\"%s\""%(ss, i, p)
else:
ss=metadata_schema
dataset.setMetadataSchemaString(ss.strip(), ms.strip())
dataset.setSaveMeshData(write_meshdata)
return dataset.saveVTK(filename)
def saveVoxet(filename, **data):
"""
Writes `Data` objects to a file using the GOCAD Voxet file format as
separate properties on the same grid.
At the moment only Data on a `ripley` domain can be saved in this format.
Note that this function will produce one header file (ending in .vo) and
a separate property file for each `Data` object.
:param filename: name of the output file ('.vo' is added if required)
:type filename: ``str``
:note: All data objects have to be defined on the same ripley domain and
either defined on reduced Function or on a `FunctionSpace` that
allows interpolation to reduced Function.
"""
from esys.escript import ReducedFunction
from esys.escript.util import interpolate
from esys.ripley.ripleycpp import DATATYPE_FLOAT32, BYTEORDER_BIG_ENDIAN
new_data={}
domain=None
for n,d in sorted(data.items(), key=lambda x: x[0]):
if d.isEmpty():
continue
fs=d.getFunctionSpace()
if domain is None:
domain=fs.getDomain()
elif domain != fs.getDomain():
raise ValueError("saveVoxet: All Data must be on the same domain!")
try:
nd=interpolate(d, ReducedFunction(domain))
except:
raise ValueError("saveVoxet: Unable to interpolate all Data to reduced Function!")
new_data[n]=nd
if filename[-3:]=='.vo':
fileprefix=filename[:-3]+"_"
else:
fileprefix=filename+"_"
filename=filename+'.vo'
origin, spacing, NE = domain.getGridParameters()
# Voxet "origin" actually refers to centre of first cell so shift:
origin = tuple([ origin[i] + spacing[i]/2. for i in range(len(origin)) ])
# flip vertical origin
origin=origin[:-1]+(-origin[-1],)
axis_max=NE[:-1]+(-NE[-1],)
midpoint=tuple([n/2 for n in NE])
if domain.getDim() == 2:
origin=origin+(0.,)
spacing=spacing+(1.,)
NE=NE+(1,)
midpoint=midpoint+(0,)
axis_max=axis_max+(0,)
mainvar=list(new_data.keys())[0]
f=open(filename,'w')
f.write("GOCAD Voxet 1\nHEADER {\nname: escriptdata\n")
f.write("sections: 3 1 1 %d 2 1 %d 3 1 %d\n"%midpoint)
f.write("painted: on\nascii: off\n*painted*variable: %s\n}"%mainvar)
f.write("""
GOCAD_ORIGINAL_COORDINATE_SYSTEM
NAME "gocad Local"
AXIS_NAME X Y Z
AXIS_UNIT m m m
ZPOSITIVE Depth
END_ORIGINAL_COORDINATE_SYSTEM\n""")
f.write("AXIS_O %0.2f %0.2f %0.2f\n"%origin)
f.write("AXIS_U %0.2f 0 0\n"%spacing[0])
f.write("AXIS_V 0 %0.2f 0\n"%spacing[1])
f.write("AXIS_W 0 0 %0.2f\n"%spacing[2])
f.write("AXIS_MIN 0 0 0\n")
f.write("AXIS_MAX %d %d %d\n"%axis_max)
f.write("AXIS_N %d %d %d\n"%NE)
f.write("\n")
num=0
for n,d in sorted(new_data.items(), key=lambda x: x[0]):
num=num+1
propfile=fileprefix+n
domain.writeBinaryGrid(d, propfile, BYTEORDER_BIG_ENDIAN, DATATYPE_FLOAT32)
f.write("\nPROPERTY %d %s\n"%(num, n))
f.write("PROPERTY_SUBCLASS %d QUANTITY Float\n"%num)
f.write("PROP_ESIZE %d 4\n"%num)
f.write("PROP_ETYPE %d IEEE\n"%num)
f.write("PROP_FORMAT %d RAW\n"%num)
f.write("PROP_OFFSET %d 0\n"%num)
f.write("PROP_FILE %d %s\n"%(num,propfile))
f.close()
|