379 строки
11 KiB
Python
Исполняемый файл
379 строки
11 KiB
Python
Исполняемый файл
#!/usr/bin/env pytest
|
|
# -*- coding: utf-8 -*-
|
|
###############################################################################
|
|
# $Id$
|
|
#
|
|
# Project: GDAL/OGR Test Suite
|
|
# Purpose: Test OGR VFK driver functionality.
|
|
# Author: Martin Landa <landa.martin gmail.com>
|
|
#
|
|
###############################################################################
|
|
# Copyright (c) 2009-2019 Martin Landa <landa.martin gmail.com>
|
|
# Copyright (c) 2010-2012, Even Rouault <even dot rouault at spatialys.com>
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a
|
|
# copy of this software and associated documentation files (the "Software"),
|
|
# to deal in the Software without restriction, including without limitation
|
|
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
# and/or sell copies of the Software, and to permit persons to whom the
|
|
# Software is furnished to do so, subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included
|
|
# in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
|
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
# DEALINGS IN THE SOFTWARE.
|
|
###############################################################################
|
|
|
|
import os
|
|
|
|
import gdaltest
|
|
import pytest
|
|
|
|
from osgeo import gdal, ogr
|
|
|
|
###############################################################################
|
|
# Open file, check number of layers, get first layer,
|
|
# check number of fields and features
|
|
|
|
|
|
def test_ogr_vfk_1():
|
|
|
|
gdaltest.vfk_drv = ogr.GetDriverByName("VFK")
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
with gdal.config_option("OGR_VFK_DB_OVERWRITE", "YES"):
|
|
gdaltest.vfk_ds = ogr.Open("data/vfk/bylany.vfk")
|
|
|
|
assert gdaltest.vfk_ds is not None
|
|
|
|
assert gdaltest.vfk_ds.GetLayerCount() == 61, "expected exactly 61 layers!"
|
|
|
|
gdaltest.vfk_layer_par = gdaltest.vfk_ds.GetLayer(0)
|
|
|
|
assert gdaltest.vfk_layer_par is not None, "cannot get first layer"
|
|
|
|
assert (
|
|
gdaltest.vfk_layer_par.GetName() == "PAR"
|
|
), 'did not get expected layer name "PAR"'
|
|
|
|
defn = gdaltest.vfk_layer_par.GetLayerDefn()
|
|
assert defn.GetFieldCount() == 28, (
|
|
"did not get expected number of fields, got %d" % defn.GetFieldCount()
|
|
)
|
|
|
|
fc = gdaltest.vfk_layer_par.GetFeatureCount()
|
|
assert fc == 1, "did not get expected feature count, got %d" % fc
|
|
|
|
|
|
###############################################################################
|
|
# Read the first feature from layer 'PAR', check envelope
|
|
|
|
|
|
def test_ogr_vfk_2():
|
|
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
gdaltest.vfk_layer_par.ResetReading()
|
|
|
|
feat = gdaltest.vfk_layer_par.GetNextFeature()
|
|
|
|
assert feat.GetFID() == 1, "did not get expected fid for feature 1"
|
|
|
|
geom = feat.GetGeometryRef()
|
|
assert (
|
|
geom.GetGeometryType() == ogr.wkbPolygon
|
|
), "did not get expected geometry type."
|
|
|
|
envelope = geom.GetEnvelope()
|
|
area = (envelope[1] - envelope[0]) * (envelope[3] - envelope[2])
|
|
exp_area = 2010.5
|
|
|
|
assert area >= exp_area - 0.5 and area <= exp_area + 0.5, (
|
|
"envelope area not as expected, got %g." % area
|
|
)
|
|
|
|
|
|
###############################################################################
|
|
# Read features from layer 'SOBR', test attribute query
|
|
|
|
|
|
def test_ogr_vfk_3():
|
|
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
gdaltest.vfk_layer_sobr = gdaltest.vfk_ds.GetLayer(43)
|
|
|
|
assert (
|
|
gdaltest.vfk_layer_sobr.GetName() == "SOBR"
|
|
), 'did not get expected layer name "SOBR"'
|
|
|
|
gdaltest.vfk_layer_sobr.SetAttributeFilter("CISLO_BODU = '55'")
|
|
|
|
gdaltest.vfk_layer_sobr.ResetReading()
|
|
|
|
feat = gdaltest.vfk_layer_sobr.GetNextFeature()
|
|
count = 0
|
|
while feat:
|
|
feat = gdaltest.vfk_layer_sobr.GetNextFeature()
|
|
count += 1
|
|
|
|
assert count == 1, "did not get expected number of features, got %d" % count
|
|
|
|
|
|
###############################################################################
|
|
# Read features from layer 'SBP', test random access, check length
|
|
|
|
|
|
def test_ogr_vfk_4():
|
|
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
gdaltest.vfk_layer_sbp = gdaltest.vfk_ds.GetLayerByName("SBP")
|
|
|
|
assert gdaltest.vfk_layer_sbp, 'did not get expected layer name "SBP"'
|
|
|
|
feat = gdaltest.vfk_layer_sbp.GetFeature(5)
|
|
length = int(feat.geometry().Length())
|
|
|
|
assert length == 10, "did not get expected length, got %d" % length
|
|
|
|
|
|
###############################################################################
|
|
# Read features from layer 'HP', check geometry type
|
|
|
|
|
|
def test_ogr_vfk_5():
|
|
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
gdaltest.vfk_layer_hp = gdaltest.vfk_ds.GetLayerByName("HP")
|
|
|
|
assert gdaltest.vfk_layer_hp != "HP", 'did not get expected layer name "HP"'
|
|
|
|
geom_type = gdaltest.vfk_layer_hp.GetGeomType()
|
|
|
|
assert geom_type == ogr.wkbLineString, (
|
|
"did not get expected geometry type, got %d" % geom_type
|
|
)
|
|
|
|
|
|
###############################################################################
|
|
# Re-Open file (test .db persistence)
|
|
|
|
|
|
def test_ogr_vfk_6():
|
|
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
gdaltest.vfk_layer_par = None
|
|
gdaltest.vfk_layer_sobr = None
|
|
gdaltest.vfk_ds = None
|
|
gdaltest.vfk_ds = ogr.Open("data/vfk/bylany.vfk")
|
|
|
|
assert gdaltest.vfk_ds is not None
|
|
|
|
assert gdaltest.vfk_ds.GetLayerCount() == 61, "expected exactly 61 layers!"
|
|
|
|
gdaltest.vfk_layer_par = gdaltest.vfk_ds.GetLayer(0)
|
|
|
|
assert gdaltest.vfk_layer_par is not None, "cannot get first layer"
|
|
|
|
assert (
|
|
gdaltest.vfk_layer_par.GetName() == "PAR"
|
|
), 'did not get expected layer name "PAR"'
|
|
|
|
defn = gdaltest.vfk_layer_par.GetLayerDefn()
|
|
assert defn.GetFieldCount() == 28, (
|
|
"did not get expected number of fields, got %d" % defn.GetFieldCount()
|
|
)
|
|
|
|
fc = gdaltest.vfk_layer_par.GetFeatureCount()
|
|
assert fc == 1, "did not get expected feature count, got %d" % fc
|
|
|
|
|
|
###############################################################################
|
|
# Read PAR layer, check data types (Integer64 new in GDAL 2.2)
|
|
|
|
|
|
def test_ogr_vfk_7():
|
|
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
defn = gdaltest.vfk_layer_par.GetLayerDefn()
|
|
|
|
for idx, name, ctype in (
|
|
(0, "ID", ogr.OFTInteger64),
|
|
(1, "STAV_DAT", ogr.OFTInteger),
|
|
(2, "DATUM_VZNIKU", ogr.OFTString),
|
|
(22, "CENA_NEMOVITOSTI", ogr.OFTReal),
|
|
):
|
|
col = defn.GetFieldDefn(idx)
|
|
assert (
|
|
col.GetName() == name and col.GetType() == ctype
|
|
), "PAR: '{}' column name/type mismatch".format(name)
|
|
|
|
|
|
###############################################################################
|
|
# Open DB file as datasource (new in GDAL 2.2)
|
|
|
|
|
|
def test_ogr_vfk_8():
|
|
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
# open by SQLite driver first
|
|
vfk_ds_db = ogr.Open("data/vfk/bylany.db")
|
|
count1 = vfk_ds_db.GetLayerCount()
|
|
vfk_ds_db = None
|
|
|
|
# then open by VFK driver
|
|
os.environ["OGR_VFK_DB_READ"] = "YES"
|
|
vfk_ds_db = ogr.Open("data/vfk/bylany.db")
|
|
count2 = vfk_ds_db.GetLayerCount()
|
|
vfk_ds_db = None
|
|
|
|
assert (
|
|
count1 == count2
|
|
), "layer count differs when opening DB by SQLite and VFK drivers"
|
|
|
|
del os.environ["OGR_VFK_DB_READ"]
|
|
|
|
|
|
###############################################################################
|
|
# Open datasource with SUPPRESS_GEOMETRY open option (new in GDAL 2.3)
|
|
|
|
|
|
def test_ogr_vfk_9():
|
|
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
# open with suppressing geometry
|
|
vfk_ds = None
|
|
vfk_ds = gdal.OpenEx("data/vfk/bylany.vfk", open_options=["SUPPRESS_GEOMETRY=YES"])
|
|
|
|
vfk_layer_par = vfk_ds.GetLayerByName("PAR")
|
|
|
|
assert vfk_layer_par != "PAR", 'did not get expected layer name "PAR"'
|
|
|
|
geom_type = vfk_layer_par.GetGeomType()
|
|
vfk_layer_par = None
|
|
vfk_ds = None
|
|
|
|
assert geom_type == ogr.wkbNone, (
|
|
"did not get expected geometry type, got %d" % geom_type
|
|
)
|
|
|
|
|
|
###############################################################################
|
|
# Open datasource with FILE_FIELD open option (new in GDAL 2.4)
|
|
|
|
|
|
def test_ogr_vfk_10():
|
|
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
# open with suppressing geometry
|
|
vfk_ds = None
|
|
vfk_ds = gdal.OpenEx("data/vfk/bylany.vfk", open_options=["FILE_FIELD=YES"])
|
|
|
|
vfk_layer_par = vfk_ds.GetLayerByName("PAR")
|
|
|
|
assert vfk_layer_par != "PAR", 'did not get expected layer name "PAR"'
|
|
|
|
vfk_layer_par.ResetReading()
|
|
feat = vfk_layer_par.GetNextFeature()
|
|
file_field = feat.GetField("VFK_FILENAME")
|
|
vfk_layer_par = None
|
|
vfk_ds = None
|
|
|
|
assert file_field == "bylany.vfk", "did not get expected file field value"
|
|
|
|
|
|
###############################################################################
|
|
# Read PAR layer, check sequential feature access consistency
|
|
|
|
|
|
def test_ogr_vfk_11():
|
|
def count_features():
|
|
gdaltest.vfk_layer_par.ResetReading()
|
|
count = 0
|
|
while True:
|
|
feat = gdaltest.vfk_layer_par.GetNextFeature()
|
|
if not feat:
|
|
break
|
|
count += 1
|
|
|
|
return count
|
|
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
count = gdaltest.vfk_layer_par.GetFeatureCount()
|
|
for i in range(2): # perform check twice, mix with random access
|
|
if count != count_features():
|
|
feat = gdaltest.vfk_layer_par.GetFeature(i)
|
|
feat.DumpReadable()
|
|
pytest.fail("did not get expected number of features")
|
|
|
|
|
|
###############################################################################
|
|
# Read SBP layer, check curved geometry
|
|
|
|
|
|
def test_ogr_vfk_12():
|
|
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
gdaltest.vfk_layer_sbp = gdaltest.vfk_ds.GetLayerByName("SBP")
|
|
|
|
gdaltest.vfk_layer_sbp.SetAttributeFilter("PARAMETRY_SPOJENI = '16'")
|
|
feat = gdaltest.vfk_layer_sbp.GetNextFeature()
|
|
geom = feat.GetGeometryRef()
|
|
|
|
assert (
|
|
geom.GetGeometryType() == ogr.wkbLineString
|
|
), "did not get expected geometry type."
|
|
|
|
assert geom.GetPointCount() == 92, "did not get expected number of points."
|
|
|
|
|
|
###############################################################################
|
|
# cleanup
|
|
|
|
|
|
def test_ogr_vfk_cleanup():
|
|
|
|
if gdaltest.vfk_drv is None:
|
|
pytest.skip()
|
|
|
|
gdaltest.vfk_layer_par = None
|
|
gdaltest.vfk_layer_hp = None
|
|
gdaltest.vfk_layer_sobr = None
|
|
gdaltest.vfk_ds = None
|
|
|
|
try:
|
|
os.remove("data/vfk/bylany.db")
|
|
except OSError:
|
|
pass
|
|
|
|
|
|
###############################################################################
|
|
#
|