PythonForAivia/PythonEnvForAivia/Recipes/ProcessImages/Dilate_2D_Labeled_Mask.py at master · AiviaCommunity/PythonForAivia · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
import subprocess
import os.path
import numpy as np
from skimage.io import imread, imsave
from skimage.segmentation import expand_labels

"""
Dilate 2D labeled masks.

Requirements
------------
numpy (comes with Aivia installer)
scikit-image (comes with Aivia installer)
ctypes

Parameters
----------
Input channel:
    Input channel to be dilated. Prefer a binary mask.

Returns
-------
Channel in Aivia
"""


# [INPUT Name:inputImagePath Type:string DisplayName:'Labeled Mask']
# [INPUT Name:dilation Type:int DisplayName:'Dilate distance (pixels)' Default:5 Min:0 Max:65535]
# [OUTPUT Name:resultPath Type:string DisplayName:'Dilated Labeled Mask']
def run(params):
    image_location = params['inputImagePath']
    result_location = params['resultPath']
    zCount = int(params['ZCount'])
    tCount = int(params['TCount'])
    dilation = int(params['dilation'])

    if not os.path.exists(image_location):
        print(f"Error: {image_location} does not exist")
        return

    if zCount > 1:
        Mbox('Error', 'This recipes currently only supports 2D images.', 0)
        return

    lbl_data = imread(image_location)
    dims = lbl_data.shape
    print('-- Input dimensions (expected (T,) (Z,) Y, X): ', np.asarray(dims), ' --')

    if tCount > 1:
        output_data = np.empty(lbl_data.shape, dtype=lbl_data.dtype)
        for t in range(0, dims[0]):
            output_data[t, :, :] = expand_labels(lbl_data[t, :, :], dilation)
    else:
        output_data = expand_labels(lbl_data, dilation)

    imsave(result_location, output_data)


def Mbox(title, text, style):
    style_tags = ["OkOnly", "OkCancel", "YesNo", "YesNoCancel"]
    cmd = [
        "powershell",
        "-Command",
        "Add-Type -AssemblyName Microsoft.VisualBasic; "
        f"$x=[Microsoft.VisualBasic.Interaction]::MsgBox('{text}', "
        f"[Microsoft.VisualBasic.MsgBoxStyle]::{style_tags[style]}, '{title}');"
        "Write-Output $x"
    ]

    result = subprocess.run(cmd, capture_output=True, text=True)
    try:
        return result.stdout.strip()        # return = "Ok" or "Cancel"
    except:
        return None


if __name__ == '__main__':
    params = {'inputImagePath': r'D:\PythonCode\_tests\3D-image.aivia.tif',
              'resultPath': r'D:\PythonCode\_tests\test.tif',
              'ZCount': 51, 'TCount': 1,
              'threshold': 1, 'dilation': 3}

    run(params)

# CHANGELOG
#   v1_00: - Comes from Dilate_3D_v1_00.py
#   v1_10: - Mbox with VB for Aivia 16
#   v1_20: - Previous version was dilating XYT as it would be XYZ. Fixed.