[pycharm] PY-23411 Debugger: add PNG support (except from hover)

GitOrigin-RevId: af0bd8bf0be60b06676644941aecb31fec89e25d
This commit is contained in:
ekaterina.itsenko
2025-02-27 03:07:17 +00:00
committed by intellij-monorepo-bot
parent 3d7b600a60
commit 35147ce462
2 changed files with 17 additions and 2 deletions
@@ -18,6 +18,7 @@ DEFAULT_IMAGE_FORMAT = 'PNG'
DEFAULT_ENCODING = 'utf-8'
GRAYSCALE_MODE = 'L'
RGB_MODE = 'RGB'
RGBA_MODE = 'RGBA'
def get_bytes(arr):
@@ -50,7 +51,14 @@ def get_bytes(arr):
else:
arr_to_convert = ((arr_to_convert - arr_min) / (arr_max - arr_min) * 255).astype(np.uint8)
mode = GRAYSCALE_MODE if arr_to_convert.ndim == 2 else RGB_MODE
arr_to_convert_ndim = arr_to_convert.ndim
if arr_to_convert_ndim == 2:
mode = GRAYSCALE_MODE
elif arr_to_convert_ndim == 3 and arr_to_convert.shape[2] == 4:
mode = RGBA_MODE
else:
mode = RGB_MODE
bytes_buffer = io.BytesIO()
image = Image.fromarray(arr_to_convert, mode=mode)
image.save(bytes_buffer, format=DEFAULT_IMAGE_FORMAT)
@@ -8,6 +8,7 @@ DEFAULT_IMAGE_FORMAT = 'PNG'
DEFAULT_ENCODING = 'utf-8'
GRAYSCALE_MODE = 'L'
RGB_MODE = 'RGB'
RGBA_MODE = 'RGBA'
def get_bytes(arr):
@@ -31,7 +32,13 @@ def get_bytes(arr):
else:
arr_to_convert = ((arr_to_convert - arr_min) / (arr_max - arr_min) * 255).astype(np.uint8)
mode = GRAYSCALE_MODE if arr_to_convert.ndim == 2 else RGB_MODE
arr_to_convert_ndim = arr_to_convert.ndim
if arr_to_convert_ndim == 2:
mode = GRAYSCALE_MODE
elif arr_to_convert_ndim == 3 and arr_to_convert.shape[2] == 4:
mode = RGBA_MODE
else:
mode = RGB_MODE
bytes_buffer = io.BytesIO()
image = Image.fromarray(arr_to_convert, mode=mode)
image.save(bytes_buffer, format=DEFAULT_IMAGE_FORMAT)