def cellsBuild(imIn, imInOut, grid=mamba.DEFAULT_GRID): """ Geodesic reconstruction of the cells of the partition image 'imIn' which are marked by the image 'imInOut'. The marked cells take the value of their corresponding marker. Note that the background cells (labelled by 0) are also modified if they are marked. The result is stored in 'imInOut'. The images can be 8-bit or 32-bit images. 'grid' can be set to HEXAGONAL or SQUARE. """ imWrk1 = mamba.imageMb(imIn) imWrk2 = mamba.imageMb(imIn) imWrk3 = mamba.imageMb(imIn, 1) vol = 0 prec_vol = -1 dirs = mamba.getDirections(grid)[1:] while (prec_vol!=vol): prec_vol = vol for d in dirs: ed = 1<<d mamba.copy(imIn, imWrk1) mamba.copy(imIn, imWrk2) mamba.supNeighbor(imWrk1, imWrk1, ed, grid=grid) mamba.infNeighbor(imWrk2, imWrk2, ed, grid=grid) mamba.generateSupMask(imWrk2, imWrk1, imWrk3, False) mamba.convertByMask(imWrk3, imWrk1, 0, mamba.computeMaxRange(imIn)[1]) mamba.linearDilate(imInOut, imWrk2, d, 1, grid=grid) mamba.logic(imWrk2, imWrk1, imWrk2, "inf") v = mamba.buildNeighbor(imWrk1, imWrk2, d, grid=grid) mamba.logic(imWrk2, imInOut, imInOut, "sup") vol = mamba.computeVolume(imInOut)
def nonEqualNeighbors(imIn, imOut, nb, grid=mamba.DEFAULT_GRID, edge=mamba.FILLED): """ This operator compares the value of each pixel of image 'imIn' with the value of its neighbors encoded in 'nb'. If all the neighbor values are different, the pixel is unchanged. Otherwise, it takes value 0. This operator works on hexagonal or square 'grid' and 'edge' is set to FILLED by default. This operator works for 8-bit and 32-bit images. """ imWrk1 = mamba.imageMb(imIn) imWrk2 = mamba.imageMb(imIn) imWrk3 = mamba.imageMb(imIn, 1) imWrk4 = mamba.imageMb(imIn, 1) for d in mamba.getDirections(grid): ed = 1<<d if (nb & ed): mamba.copy(imIn, imWrk1) mamba.copy(imIn, imWrk2) mamba.supNeighbor(imWrk1, imWrk1, ed, grid=grid, edge=edge) mamba.infNeighbor(imWrk2, imWrk2, ed, grid=grid, edge=edge) mamba.generateSupMask(imWrk2, imWrk1, imWrk3, False) mamba.logic(imWrk4, imWrk3, imWrk4, "or") mamba.convertByMask(imWrk4, imWrk1, mamba.computeMaxRange(imIn)[1], 0) mamba.logic(imIn, imWrk1, imOut, "inf")
def nonEqualNeighbors(imIn, imOut, nb, grid=mamba.DEFAULT_GRID, edge=mamba.FILLED): """ This operator compares the value of each pixel of image 'imIn' with the value of its neighbors encoded in 'nb'. If all the neighbor values are different, the pixel is unchanged. Otherwise, it takes value 0. This operator works on hexagonal or square 'grid' and 'edge' is set to FILLED by default. This operator works for 8-bit and 32-bit images. """ imWrk1 = mamba.imageMb(imIn) imWrk2 = mamba.imageMb(imIn) imWrk3 = mamba.imageMb(imIn, 1) imWrk4 = mamba.imageMb(imIn, 1) for d in mamba.getDirections(grid): ed = 1 << d if (nb & ed): mamba.copy(imIn, imWrk1) mamba.copy(imIn, imWrk2) mamba.supNeighbor(imWrk1, imWrk1, ed, grid=grid, edge=edge) mamba.infNeighbor(imWrk2, imWrk2, ed, grid=grid, edge=edge) mamba.generateSupMask(imWrk2, imWrk1, imWrk3, False) mamba.logic(imWrk4, imWrk3, imWrk4, "or") mamba.convertByMask(imWrk4, imWrk1, mamba.computeMaxRange(imIn)[1], 0) mamba.logic(imIn, imWrk1, imOut, "inf")
def cellsBuild(imIn, imInOut, grid=mamba.DEFAULT_GRID): """ Geodesic reconstruction of the cells of the partition image 'imIn' which are marked by the image 'imInOut'. The marked cells take the value of their corresponding marker. Note that the background cells (labelled by 0) are also modified if they are marked. The result is stored in 'imInOut'. The images can be 8-bit or 32-bit images. 'grid' can be set to HEXAGONAL or SQUARE. """ imWrk1 = mamba.imageMb(imIn) imWrk2 = mamba.imageMb(imIn) imWrk3 = mamba.imageMb(imIn, 1) vol = 0 prec_vol = -1 dirs = mamba.getDirections(grid)[1:] while (prec_vol != vol): prec_vol = vol for d in dirs: ed = 1 << d mamba.copy(imIn, imWrk1) mamba.copy(imIn, imWrk2) mamba.supNeighbor(imWrk1, imWrk1, ed, grid=grid) mamba.infNeighbor(imWrk2, imWrk2, ed, grid=grid) mamba.generateSupMask(imWrk2, imWrk1, imWrk3, False) mamba.convertByMask(imWrk3, imWrk1, 0, mamba.computeMaxRange(imIn)[1]) mamba.linearDilate(imInOut, imWrk2, d, 1, grid=grid) mamba.logic(imWrk2, imWrk1, imWrk2, "inf") v = mamba.buildNeighbor(imWrk1, imWrk2, d, grid=grid) mamba.logic(imWrk2, imInOut, imInOut, "sup") vol = mamba.computeVolume(imInOut)
def equalNeighbors(imIn, imOut, nb, grid=mamba.DEFAULT_GRID, edge=mamba.FILLED): """ This operator compares the value of each pixel of image 'imIn' with the value of its neighbors encoded in 'nb'. If all the neighbor values are equal, the pixel is unchanged. Otherwise, it takes value 0. This operator works on hexagonal or square 'grid' and 'edge' is set to FILLED by default. This operator works for 8-bit and 32-bit images. """ imWrk1 = mamba.imageMb(imIn) imWrk2 = mamba.imageMb(imIn, 1) mamba.copy(imIn, imWrk1) mamba.copy(imIn, imOut) mamba.supNeighbor(imIn, imWrk1, nb, grid=grid, edge=edge) mamba.infNeighbor(imOut, imOut, nb, grid=grid, edge=edge) mamba.generateSupMask(imOut, imWrk1, imWrk2, False) mamba.convertByMask(imWrk2, imWrk1, 0, mamba.computeMaxRange(imIn)[1]) mamba.logic(imOut, imWrk1, imOut, "inf")
def buildNeighbor3D(imMask, imInOut, d, grid=m3D.DEFAULT_GRID3D): """ Builds 3D image 'imInout' in direction 'd' according to 'grid' using 'imMask' as a mask (the propagation is performed only in 'd' direction). The function also returns the volume of the image 'imInout' after the build operation. 'grid' value can be any 3D grid. """ (width, height, length) = imInOut.getSize() if length != len(imMask): mamba.raiseExceptionOnError(core.MB_ERR_BAD_SIZE) grid2D = grid.get2DGrid() scan = grid.convertFromDir(d, 0)[0] volume = 0 if scan == 0: for i in range(length): vol = mamba.buildNeighbor(imMask[i], imInOut[i], d, grid2D) volume += vol else: if scan == 1: startPlane, endPlane = 0, length - 1 else: startPlane, endPlane = length - 1, 0 for i in range(startPlane, endPlane, scan): mamba.logic(imInOut[i], imMask[i], imInOut[i], "inf") vol = mamba.computeVolume(imInOut[i]) volume += vol td = grid.getTranDir(d) dh = grid.convertFromDir(td, i + scan)[1] mamba.supNeighbor(imInOut[i], imInOut[i + scan], 1 << dh, grid2D) mamba.logic(imInOut[endPlane], imMask[endPlane], imInOut[endPlane], "inf") vol = mamba.computeVolume(imInOut[endPlane]) volume += vol return volume
def buildNeighbor3D(imMask, imInOut, d, grid=m3D.DEFAULT_GRID3D): """ Builds 3D image 'imInout' in direction 'd' according to 'grid' using 'imMask' as a mask (the propagation is performed only in 'd' direction). The function also returns the volume of the image 'imInout' after the build operation. 'grid' value can be any 3D grid. """ (width, height, length) = imInOut.getSize() if length!=len(imMask): mamba.raiseExceptionOnError(core.MB_ERR_BAD_SIZE) grid2D = grid.get2DGrid() scan = grid.convertFromDir(d,0)[0] volume = 0 if scan == 0: for i in range(length): vol = mamba.buildNeighbor(imMask[i], imInOut[i], d, grid2D) volume += vol else: if scan == 1: startPlane, endPlane = 0, length - 1 else: startPlane, endPlane = length - 1, 0 for i in range(startPlane, endPlane, scan): mamba.logic(imInOut[i], imMask[i], imInOut[i], "inf") vol = mamba.computeVolume(imInOut[i]) volume += vol td = grid.getTranDir(d) dh = grid.convertFromDir(td,i+scan)[1] mamba.supNeighbor(imInOut[i], imInOut[i+scan], 1<<dh, grid2D) mamba.logic(imInOut[endPlane], imMask[endPlane], imInOut[endPlane], "inf") vol = mamba.computeVolume(imInOut[endPlane]) volume += vol return volume
def dilate3D(imIn, imOut, n=1, se=CUBOCTAHEDRON1, edge=mamba.EMPTY): """ This operator performs a dilation, using the structuring element 'se' (set by default as CUBOCTAHEDRON1), of 3D image 'imIn' and puts the result in 'imOut'. The operation is repeated 'n' times (default is 1). This operator assumes an 'EMPTY' edge by default. This operator always considers that the origin of the structuring element in use is at position 0 even if this point does not belong to it. """ (width,height,length) = imIn.getSize() depth = imIn.getDepth() if length!=len(imOut): mamba.raiseExceptionOnError(core.MB_ERR_BAD_SIZE) zext = se.grid.getZExtension() imWrk = m3D.image3DMb(width, height, length+zext*2, depth) if edge==mamba.EMPTY: for i in range(zext): imWrk[i].reset() imWrk[length+zext*2-1-i].reset() else: value = mamba.computeMaxRange(imIn[0])[1] for i in range(zext): imWrk[i].fill(value) imWrk[length+zext*2-1-i].fill(value) m3D.copy3D(imIn, imOut) dirs = se.getDirections(withoutZero=True) for size in range(n): m3D.copy3D(imOut, imWrk, 0, 1) if not se.hasZero(): imOut.reset() for i in range(length): dirs_enc = se.grid.getEncodedDirs(dirs,i) mamba.supNeighbor(imWrk[i], imOut[i], dirs_enc[-1], grid=se.grid.get2DGrid(), edge=edge) mamba.supNeighbor(imWrk[i+1], imOut[i], dirs_enc[0], grid=se.grid.get2DGrid(), edge=edge) mamba.supNeighbor(imWrk[i+2], imOut[i], dirs_enc[1], grid=se.grid.get2DGrid(), edge=edge)