ingo@2798: import ogr
ingo@2798: 
ingo@2798: TABLE_NAME="floodmaps"
ingo@2798: 
ingo@2798: class UESG:
ingo@2798: 
ingo@2798:     def __init__(self, dbconn):
ingo@2798:         self.dbconn = dbconn
ingo@2798: 
ingo@2798: 
ingo@2798:     def isGeometryValid(self, geomType):
ingo@2798:         if geomType == 3 or geomType == 6:
ingo@2798:             return True
ingo@2798:         else:
ingo@2798:             return False
ingo@2798: 
ingo@2798: 
ingo@2798:     def getKind(self, path):
ingo@2798:         kind = 0
ingo@2798:         if path.find("Berechnung") > 0:
ingo@2798:             kind = kind + 100
ingo@2798: 
ingo@2798:             if path.find("Aktuell") > 0:
ingo@2798:                 kind = kind + 10
ingo@2798:             else:
ingo@2798:                 kind = kind + 20
ingo@2798: 
ingo@2798:             if path.find("Land") > 0:
ingo@2798:                 kind = kind + 2
ingo@2798:             else:
ingo@2798:                 kind = kind + 1
ingo@2798:         else:
ingo@2798:             kind = kind + 200
ingo@2798: 
ingo@2798:         return kind
ingo@2798: 
ingo@2798: 
ingo@2798:     def walkOverShapes(self, shape):
ingo@2798:         (name, path) = shape
ingo@2798:         shp = ogr.Open(shape[1])
ingo@2798:         if shp is None:
ingo@2798:             print "Shapefile '%s' could not be opened!" % path
ingo@2798:             return
ingo@2798: 
ingo@2798:         print "Opened shapefile '%s'" % path
ingo@2798:         srcLayer = shp.GetLayerByName(name)
ingo@2798: 
ingo@2798:         if srcLayer is None:
ingo@2798:             print "Layer '%s' was not found!" % name
ingo@2798:             return
ingo@2798: 
ingo@2798:         return self.shape2Database(srcLayer, name, self.getKind(path))
ingo@2798: 
ingo@2798: 
ingo@2798:     def shape2Database(self, srcLayer, name, kind):
ingo@2798:         table = ogr.Open(self.dbconn)
ingo@2798:         destLayer = table.GetLayerByName(TABLE_NAME)
ingo@2798: 
ingo@2798:         if srcLayer is None:
ingo@2798:             print "Shapefile is None!"
ingo@2798:             return -1
ingo@2798: 
ingo@2798:         if destLayer is None:
ingo@2798:             print "No destination layer given!"
ingo@2798:             return -1
ingo@2798: 
ingo@2798:         count = srcLayer.GetFeatureCount()
ingo@2798:         print "Try to add %i features to database." % count
ingo@2798: 
ingo@2798:         srcLayer.ResetReading()
ingo@2798: 
ingo@2798:         geomType = -1
ingo@2798: 
ingo@2798:         featureDef = destLayer.GetLayerDefn()
ingo@2798: 
ingo@2798:         for feat in srcLayer:
ingo@2798:             geom = feat.GetGeometryRef()
ingo@2798:             geomType = geom.GetGeometryType()
ingo@2798: 
ingo@2798:             newFeat = self.createNewFeature(featureDef, feat)
ingo@2798:             newFeat.SetField("kind", kind)
ingo@2798:             newFeat.SetField("name", name)
ingo@2798: 
ingo@2798:             if self.isGeometryValid(geomType):
ingo@2798:                 res = destLayer.CreateFeature(newFeat)
ingo@2798:                 if res is None or res > 0:
ingo@2798:                     print "Error while inserting feature: %r" % res
ingo@2798:             else:
ingo@2798:                 print "Geometry type not supported: %i" % geomType
ingo@2798: 
ingo@2798:         try:
ingo@2798:             destLayer.CommitTransaction()
ingo@2798:         except e:
ingo@2798:             print "Exception while committing transaction."
ingo@2798: 
ingo@2798:         return geomType
ingo@2798: 
ingo@2798: 
ingo@2798:     def createNewFeature(self, featureDef, feat):
ingo@2798:         newFeat = ogr.Feature(featureDef)
ingo@2798:         newFeat.SetGeometry(feat.GetGeometryRef())
ingo@2798: 
ingo@2798:         if feat.IsFieldSet("river_id"):
ingo@2798:             riverId = feat.GetField("river_id")
ingo@2798:         else:
ingo@2798:             riverId = 1
ingo@2798: 
ingo@2798:         if feat.IsFieldSet("diff"):
ingo@2798:             diff = feat.GetFieldAsDouble("diff")
ingo@2798:         else:
ingo@2798:             diff = 0
ingo@2798: 
ingo@2798:         if feat.IsFieldSet("count"):
ingo@2798:             count = feat.GetFieldAsInteger("count")
ingo@2798:         else:
ingo@2798:             count = 0
ingo@2798: 
ingo@2798:         if feat.IsFieldSet("area"):
ingo@2798:             area = feat.GetFieldAsDouble("area")
ingo@2798:         else:
ingo@2798:             area = 0
ingo@2798: 
ingo@2798:         if feat.IsFieldSet("perimeter"):
ingo@2798:             perimeter = feat.GetFieldAsDouble("perimeter")
ingo@2798:         else:
ingo@2798:             perimeter = 0
ingo@2798: 
ingo@2798:         groupId = 2
ingo@2798: 
ingo@2798:         newFeat.SetField("river_id", riverId)
ingo@2798:         newFeat.SetField("diff", diff)
ingo@2798:         newFeat.SetField("count", count)
ingo@2798:         newFeat.SetField("area", area)
ingo@2798:         newFeat.SetField("perimeter", perimeter)
ingo@2798: 
ingo@2798:         return newFeat
ingo@2798: