Das deutsche Python-Forum

Code: Alles auswählen

Holzmenge = 40
r = 1
xindex = 2
yindex = 2
xmin = xindex - r
xmax = xindex + r 
ymin = (yindex - r)# + 1
ymax = (yindex + r)# - 1
minilist_zeile = []
minilist_spalte = []
minilist_total = []
actualsum = 0

#zentralpixel addieren, geht nur, wenn man annimmt das zentralpixel < Holzmenge
minilist_zeile.append(array[xindex,yindex])

while actualsum <= Holzmenge:
    #zeile + randpixel jeder zeile
    for xmin in range (xmin, xmax):
        minilist_zeile.append(array[xindex-r, xmin])
        xmin = xmin + 1
      #spalte ohne randpixel jeder zeile
        for ymin in range (ymin, ymax+1):
            minilist_spalte.append(array[xindex, ymin])
            ymin = ymin + 1
    r = r + 1
    minilist_total = minilist_zeile + minilist_spalte
    actualsum = sum(minilist_total)
    
print "akutelle Summe: " + str(actualsum)
print "zeilen: "+str(minilist_zeile)
print "spalten: "+str(minilist_spalte)
#print minilist
print "Der Radius um den Holzvorrat bereitzustellen beträgt: " + str(r-1)

Code: Alles auswählen

def Holzernter(ZuErntendeMenge, xindex, yindex):
    summe = 0
    summe = summe + array[xindex, yindex]
    r = 1
    
    while summe <= ZuErntendeMenge:
        print "r = " + str(r)
        print "summe " + str(summe) + " zem: " + str(ZuErntendeMenge) + "\n"
        xmin = xindex - r
        xmax = xindex + r 
        ymin = yindex - r
        ymax = yindex + r  
        #zeile min, zeile max, zeile min usw.
        if ((xmin >= 0) and (xmax >= 0)) or ((ymin >= 0) and (ymax >=0)):
            for i in range (xmin, xmax+1, 1):
                summe = summe + array[ymax, i] + array[ymin, i]
                print "y = " + str(i) + " value " + str(array[ymax, i])
                print "y = " + str(i) + " value " + str(array[ymin, i])
                #spalten
            for j in range (ymin+1, ymax, 1):
                print "x = " + str(i) + " value " + str(array[j, xmin])
                print "x = " + str(i) + " value " + str(array[j, xmax])
                summe = summe + array[j, xmin] + array[j, xmax]
            r = r + 1
            print summe
        else:
            print "HIER!"
            summe = summe + 0
            r = r + 1
            print summe
    radius = [r-1]
    return radius

Code: Alles auswählen

def createarrayfromgrid(gridname):
    # Importieren des Datensatzes
    dataset = gdal.Open(gridname, GA_ReadOnly )

    if dataset is None:
        print "Es wurde kein file gefunden, das importiert werden könnte"
    # was sagt mir dieser output
    print dataset

    #einige informationen über das importfile, welches oben spezifiziert wurde
    print 'Driver: ', dataset.GetDriver().ShortName,'/', \
              dataset.GetDriver().LongName
    print 'Size is ',dataset.RasterXSize,'x',dataset.RasterYSize, \
    'x',dataset.RasterCount
    print 'Projection is ',dataset.GetProjection()
        
    geotransform = dataset.GetGeoTransform()
    if not geotransform is None:
        print 'Origin = (',geotransform[0], ',',geotransform[3],')'
        print 'Pixel Size = (',geotransform[1], ',',geotransform[5],')'


    #ein GetRasterBand Objekt wird erzeugt
    band = dataset.GetRasterBand(1)

    print 'Band Type=',gdal.GetDataTypeName(band.DataType)

    #minimum und maximum aus dem file lesen
    min = band.GetMinimum()
    max = band.GetMaximum()
    if min is None or max is None:
        (min,max) = band.ComputeRasterMinMax(1)
        print 'Min=%.3f, Max=%.3f' % (min,max)
    if band.GetOverviewCount() > 0:
        print 'Band has ', band.GetOverviewCount(), ' overviews.'
    if not band.GetRasterColorTable() is None:
        print 'Band has a color table with ', \
        band.GetRasterColorTable().GetCount(), ' entries.'
    array = ()
    array = dataset.ReadAsArray()
    
    return array, dataset.RasterXSize, dataset.RasterYSize, geotransform[1]

Code: Alles auswählen

def get_value(array, x, y):
    """Returns value at x, y or 0 if x or y is not within the array."""
    width, height = array.shape
    return array[x, y] if 0 <= x < width and 0 <= y < height else 0


def wood_radius(array, needed_amount, start_x, start_y):
    radius = 0
    total = get_value(array, start_x, start_y)
    max_radius = max(array.shape)
    while total < needed_amount and radius <= max_radius:
        radius += 1
        min_x = start_x - radius
        max_x = start_x + radius
        min_y = start_y - radius
        max_y = start_y + radius
        for i in xrange(min_x, max_x + 1):
            total += get_value(array, i, min_y) + get_value(array, i, max_y)
        for i in xrange(min_y + 1, max_y):
            total += get_value(array, min_x, i) + get_value(array, max_x, i)
    if total < needed_amount:
        raise ValueError('not enough wood')
    return radius