Просмотр кода страницы Работа с векторными данными при помощи OGR и Python

=== Получение информации и работа с объектами ===

Количество объектов в слое можно узнать при помощи <tt>GetFeatureCount()</tt>, вызов <tt>GetSpatialRef()</tt> покажет информацию о используемой проекции и системе координат. Охват слоя можно получить при помощи <tt>GetExtent()</tt>, а количество полей — <tt>GetFieldCount()</tt>. При получении охвата следует помнить, что в зависимости от используемого драйвера значение охвата может зависеть или не зависить от наличия пространственного фильтра, поэтому перед вызовом <tt>GetExtent()</tt> рекомендуется деактивировать прострaнственный фильтр.

<syntaxhighlight lang="python">
print "Feature count", layer.GetFeatureCount()
print "Layer SRC", layer.GetSpatialRef()
print "Layer extent", layer.GetExtent()
print "Field count", layer.GetFieldCount()
</syntaxhighlight>

После указания рабочего слоя можно приступать к чтению объектов. Для доступа к объектам удобно использовать цикл и функцию <tt>GetNextFeature()</tt>, которая ввозвращает следующий объект набора или None если достигнут конец набора. Перед получением объектов можно задать пространственный фильтр или фильтр по атрибутам используя <tt>SetSpatialFilter()</tt>/<tt>SetSpatialFilterRect()</tt> и <tt>SetAttributeFilter()</tt> соответственно.

<syntaxhighlight lang="python">
feat = layer.GetNextFeature()
while feat is not None:
  # обрабатываем объект
</syntaxhighlight>

Для получения списка полей объекта и их значений используется вот такая конструкция:

<syntaxhighlight lang="python">
feat = layer.GetNextFeature()
featDef = layer.GetLayerDefn() # схема (таблица атрибутов) слоя
while feat is not None:
  for i in range( featDef.GetFieldCount() ): # проходим по всем полям
    fieldDef = featDef.GetFieldDefn( i ) # получаем i-тое поле
    print "Field name", fieldDef.GetNameRef() # и выводим информацию
    print "Field type", fieldDef.GetType()
    print "Field value", feat.GetFieldAsString(i)
  feat = layer.GetNextFeature() # переходим к следующему объекту
</syntaxhighlight>

Поля могут иметь разный тип, для каждого типа данных есть соответсвующая функция доступа. Например, для типа <tt>OFTInteger</tt> — <tt>GetFieldAsInteger()</tt>, для <tt>OFTDateTime</tt> — <tt>GetFieldAsDateTime()</tt> и т.д. Кроме того, адекватное представление значения любого поля можно получить при помощи <tt>GetFieldAsString()</tt>.

<tt>GetGeometryRef()</tt> вернет геометрию объекта:

<syntaxhighlight lang="python">
geom = feat.GetGeometryRef()
if geom is None:
  print "Invalid geometry"
if geom.GetGeometryType() == ogr.wkbPoint:
  print "%.3f, %.3f" % ( geom.GetX(), geom.GetY() )
else:
  print "Non point geometry"
</syntaxhighlight>

Используя OGR полученную геометрию можно экспортировать в различные форматы при помощи функций <tt>ExportToGML()</tt>, <tt>ExportToJson()</tt>, <tt>ExportToKML()</tt>, <tt>ExportToWkb()</tt>, <tt>ExportToWkt()</tt> а также выполнять различные действия над ней: <tt>ConvexHull()</tt>, <tt>Buffer()</tt>, <tt>Difference()</tt>...

После окончания работы нужно освободить ресурсы и закрыть набор данных

<syntaxhighlight lang="python">
ogrData.Destroy()
</syntaxhighlight>