subset

This module contains functionality for subsetting a large segmented image into a smaller one for checking etc. See subsetImage().

exception pyshepseg.subset.PyShepSegSubsetError
pyshepseg.subset.copyColumns(inRat, outRat)

Copy column structure from inRat to outRat. Note that this just creates the empty columns in outRat, it does not copy any data.

Parameters
inRat, outRatgdal.RasterAttributeTable

Columns found in inRat are created on outRat

Returns
numIntColsint

Number of integer columns found

numFloatColsint

Number of float columns found

pyshepseg.subset.copySubsettedSegmentsToNew(inPage, outPagedRat, recodeDict)

Using the recodeDict, copy across the rows inPage to outPage.

inPage is processed and (taking into account of inPage.startSegId) the original input row found. This value is then looked up in recodeDict to find the row in the output RAT to copy the row from the input to.

Parameters
inPagetiling.RatPage

A page of RAT from the input file

outPagedRatnumba.typed.Dict

In-memory pages of the output RAT, as created by createPagedRat(). This is modified in-place, creating new pages as required.

recodeDictnumba.typed.Dict

Keyed by original segment ID, values are the corresponding segment IDs in the subset

pyshepseg.subset.processSubsetTile(tile, recodeDict, histogramDict, maskData)

Process a tile of the subset area. Returns a new tile with the new codes. Fills in the recodeDict as it goes and also updates histogramDict.

Parameters
tileshepseg.SegIdType ndarray (tileNrows, tileNcols)

Input tile of segment IDs

recodeDictnumba.typed.Dict

Keyed by original segment ID, values are the corresponding segment IDs in the subset

histogramDictnumba.typed.Dict

Histogram counts in the subset, keyed by new segment ID

maskDataNone or int ndarray (tileNrows, tileNcols)

If not None, then is a raster mask. Only pixels which are non-zero in the mask will be included in the subset

Returns
outDatashepseg.SegIdType ndarray (tileNrows, tileNcols)

Recoded copy of the input tile.

pyshepseg.subset.readColDataIntoPage(page, data, idx, colType, minVal)

Numba function to quickly read a column returned by rat.ReadAsArray() info a RatPage.

pyshepseg.subset.readRATIntoPage(rat, numIntCols, numFloatCols, minVal, maxVal)

Create a new RatPage() object that represents the section of the RAT for a tile of an image. The part of the RAT between minVal and maxVal is read in and a RatPage() instance returned with the startSegId param set to minVal.

pyshepseg.subset.setHistogramFromDictionary(dictn, histArray)

Given a dictionary of pixel counts keyed on index, write these values to the array.

pyshepseg.subset.setSubsetRecodeFromDictionary(dictn, array)

Given the recodeDict write the original values to the array at the new indices.

pyshepseg.subset.subsetImage(inname, outname, tlx, tly, newXsize, newYsize, outformat, creationOptions=[], origSegIdColName=None, maskImage=None)

Subset an image and “compress” the RAT so only values that are in the new image are in the RAT. Note that the image values will be recoded in the process.

gdal_translate seems to have a problem with files that have large RAT’s so while that gets fixed do the subsetting in this function.

Parameters
innamestr

Filename of input raster

outnamestr

Filename of output raster

tlx, tlyint

The x & y pixel coordinates (i.e. col & row, respectively) of the top left pixel of the image subset to extract.

newXsize, newYsizeint

The x & y size (in pixels) of the image subset to extract

outformatstr

The GDAL short name of the format driver to use for the output raster file

creationOptionslist of str

The GDAL creation options for the output file

origSegIdColNamestr or None

The name of a RAT column. If not None, this will be created in the output file, and will contain the original segment ID numbers so the new segment IDs can be linked back to the old

maskImagestr or None

If not None, then the filename of a mask raster. Only pixels which are non-zero in this mask image will be included in the subset. This image is assumed to match the shape and position of the output subset

pyshepseg.subset.writeCompletedPagesForSubset(inRAT, outRAT, outPagedRat)

For the subset operation. Writes out any completed pages to outRAT using the inRAT as a template.

Parameters
inRAT, outRATgdal.RasterAttributeTable

The input and output raster attribute tables.

outPagedRatnumba.typed.Dict

The paged RAT in memory, as created by createPagedRat()