示例#1
0
 def test_writer(self):
     """`WriteTableToGSLib`: check data integrity across I/O"""
     writer = WriteTableToGSLib()
     filename = os.path.join(self.test_dir, 'test-writer.dat')
     writer.SetFileName(filename)
     writer.Write(self.TABLE)
     # Now read that data and compare
     reader = GSLibReader()
     read = reader.apply(filename)
     # Compare data
     truedata = self.TABLE.GetRowData()
     testdata = read.GetRowData()
     self.assertEqual(truedata.GetNumberOfArrays(),
                      testdata.GetNumberOfArrays())
     #self.assertEqual(truedata.GetNumberOfRows(), testdata.GetNumberOfRows())
     #self.assertEqual(truedata.GetNumberOfColumns(), testdata.GetNumberOfColumns())
     wtbl = dsa.WrapDataObject(self.TABLE)
     wrd = dsa.WrapDataObject(read)
     for i in range(truedata.GetNumberOfArrays()):
         self.assertIsNotNone(wtbl.RowData[i])
         self.assertIsNotNone(wrd.RowData[i])
         self.assertTrue(np.allclose(wtbl.RowData[i], wrd.RowData[i]))
示例#2
0
Read any GSLib file as a table :class:`pyvist.Table`

"""
import pyvista as pv
from pyvista import examples
from PVGeo.gslib import GSLibReader
from PVGeo.grids import TableToTimeGrid

###############################################################################

# points_url = 'http://www.trainingimages.org/uploads/3/4/7/0/34703305/sundarbans.zip'
filename, _ = examples.downloads._download_file('sundarbans.SGEMS.zip')

reader = GSLibReader()
table = reader.apply(filename)

# Print the file header
print(reader.get_file_header())

###############################################################################
print(table)

###############################################################################
# From inspecting the header, we realize that this dataset os gridded, so let's
# use the :class:`PVGeo.grid.TableToTimeGrid` filter to create a
# :class:`pyvista.UniformGrid` of that dataset.

# 1200 x, 1750 y, 1 z, 1 t
grid = TableToTimeGrid(extent=(1200, 1750, 1, 1)).apply(table)
print(grid)