Сравнение областей пикселей - PullRequest
0 голосов
/ 20 апреля 2019

Я пытаюсь написать скрипт Python для GIMP, цель которого - нарезать изображение на набор плиток (идентифицировать каждую уникальную плитку размером 16x16 на рисунке).

Пока я могучитать плитки (фактически область 16x16 пикселей) и записывать их где-нибудь.Но все мои попытки сравнивать плитки потерпели неудачу.

Я что-то упустил?Мой сценарий выглядит следующим образом:

#!/usr/bin/env python

from gimpfu import *

    # compare 2 tiles,
    # return 1 if identical, false otherwise
def tileCompare(tile1, tile2):
    if(tile1 == tile2):
        return 1
    return 0

     # return tile at (x, y) coordinates
def readTile(layer, x, y):
    pr = layer.get_pixel_rgn(x,y,16,16)
    return pr[x:x+16, y:y+16]

    # write tile at (x, y) coordinates on given layer
def writeTile(layer, x, y, tile):    
    pr = layer.get_pixel_rgn(x,y,16,16)
    pr[x:x+16, y:y+16] = tile

def TilesSlicer(sourceLayer, targetLayer):
        # Actual plug-in code will go here

        # iterate tiles (result in tileSource)
    for x in range(0, sourceLayer.width, 16):
        for y in range(0, sourceLayer.height, 16):
            tileSource = readTile(sourceLayer, x, y)
            found = 0
                # iterate tiles again (result in tileIterator)
            for a in range(0, sourceLayer.width, 16):
                for b in range(0, sourceLayer.height, 16):
                    tileIterator = readTile(sourceLayer, x, y)
                        # compare tiles
                        # if identical and not yet found
                        # write it in the target layer
                    if (tileCompare(tileSource, tileIterator) == 1):
                        if(found == 0):
                            writeTile(tileIterator, a, b, tileSource)
                        found = 1

register(
    "TilesSlicer",
    "Tiles slicer",
    "Slice a picture into tiles",
    "Fabrice Lambert",
    "Fabrice Lambert",
    "April 2019",
    "Tiles slicer...",
    "RGB*",
    [
    (PF_DRAWABLE, "sourceLayer", "Source Layer: ", None),
    (PF_DRAWABLE, "targetLayer", "Target Layer: ", None),
    ],
    [],
    TilesSlicer,
        menu="<Image>/Filters/My Scripts")

main()

Спасибо за ваши предложения.

Ответы [ 2 ]

0 голосов
/ 21 апреля 2019

Хорошо,

После небольшого уточнения скрипт выглядит следующим образом:
- Добавлены ширина и высота плиток для обработки любого размера плиток.
- Удален параметр целевого слоя, теперь скрипт создаетit.
- Добавлен дисплей в реальном времени для обратной связи с пользователем (к сожалению, индикатор выполнения не работает).
- Улучшена скорость.

#!/usr/bin/env python

from gimpfu import *

    # compare 2 tiles,
    # return 1 if identical, 0 otherwise
def tileCompare(tile1, tile2):
    if(tile1 == tile2):
        return 1
    return 0

     # return tile at (x, y) coordinates
def readTile(layer, x, y, width, height):
    pr = layer.get_pixel_rgn(x, y, width, height)
    return pr[x:x+width, y:y+height]

    # write tile at (x, y) coordinates on given layer
def writeTile(layer, x, y, width, height, tile):    
    pr = layer.get_pixel_rgn(x, y, width, height)
    pr[x:x+width, y:y+height] = tile
    layer.update(x, y, width, height)
    gimp.displays_flush()

def TilesSlicer(sourceLayer, tileWidth, tileHeight):
        # Actual plug-in code will go here
    if((sourceLayer.width % tileWidth) != 0):
        gimp.message("The layer width is not multiple of " + str(tileWidth))
        gimp.quit()
    if((sourceLayer.height % tileWidth) != 0):
        gimp.message("The layer height is not multiple of " + str(tileHeight))
        gimp.quit()
    totalTiles = (sourceLayer.width / tileWidth) * (sourceLayer.height / tileHeight)
    tilesProcessed = 0

    gimp.progress_init("Processing...")
    gimp.progress_update(0.0)

    sourceImage = sourceLayer.image
    targetLayer = pdb.gimp_layer_new(sourceImage, sourceLayer.width, sourceLayer.height, sourceImage.base_type, "Target", 100.0, sourceLayer.mode)
    targetLayer.add_alpha()
    targetLayer.fill(TRANSPARENT_FILL)
    sourceImage.add_layer(targetLayer, 0)
        # iterate tiles (result in tileSource)
    for x in range(0, sourceLayer.width, tileWidth):
        for y in range(0, sourceLayer.height, tileHeight):
            tileSource = readTile(sourceLayer, x, y, tileWidth, tileHeight)
            found = 0
                # iterate tiles again (result in tileIterator)
            for a in range(0, sourceLayer.width, tileWidth):
                for b in range(0, sourceLayer.height, tileHeight):
                    tileIterator = readTile(sourceLayer, a, b, tileWidth, tileHeight)
                        # compare tiles
                        # if identical and not yet found
                        # write it in the target layer
                        # and abort iteration (for speed purpose)
                    if (tileCompare(tileSource, tileIterator) == 1):
                        if(found == 0):
                            writeTile(targetLayer, a, b, tileWidth, tileHeight, tileIterator)
                        found = 1
                        break
                if(found == 1):
                    break
            tilesProcessed = tilesProcessed + 1
            gimp.progress_update(tilesProcessed / totalTiles)
            gimp.displays_flush()

register(
    "TilesSlicer",
    "Tiles slicer",
    "Slice a picture into tiles",
    "Fabrice Lambert",
    "Fabrice Lambert",
    "April 2019",
    "Tiles slicer...",
    "RGB*",
    [
    (PF_DRAWABLE, "sourceLayer", "Source Layer: ", None),
    (PF_INT8, "tileWidth", "Tile width: ", 16),
    (PF_INT8, "tileHeight", "Tile height: ", 16),
    ],
    [],
    TilesSlicer, 
        menu="<Image>/Filters/My Scripts")

main()

Возможно, этоуточняется лучше, и если у кого-то есть что-то, что касается индикатора выполнения, дайте мне знать.
Я открыт для предложений.

0 голосов
/ 20 апреля 2019

Nvm,

Я обнаружил проблему:
tileIterator = readTile (sourceLayer, a, b)
вместо:
tileIterator = readTile (sourceLayer, x, y)

...