From 5174a2084174fdd8be60ada1fffffc141a22cd91 Mon Sep 17 00:00:00 2001 From: Mike Sullivan Date: Mon, 27 Jan 2025 17:45:27 +0000 Subject: [PATCH] Calculate size of stack array via utility.full_size_MB --- .../gui/windows/stack_properties_dialog/presenter.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/mantidimaging/gui/windows/stack_properties_dialog/presenter.py b/mantidimaging/gui/windows/stack_properties_dialog/presenter.py index c976d345187..41c0a0ab969 100644 --- a/mantidimaging/gui/windows/stack_properties_dialog/presenter.py +++ b/mantidimaging/gui/windows/stack_properties_dialog/presenter.py @@ -2,9 +2,9 @@ # SPDX - License - Identifier: GPL-3.0-or-later from __future__ import annotations -import os from typing import TYPE_CHECKING +from mantidimaging.core.utility.size_calculator import full_size_MB from mantidimaging.gui.mvp_base import BasePresenter if TYPE_CHECKING: @@ -37,6 +37,4 @@ def set_stack_data(self): self.view.stack_size_MB = self.get_stack_size_MB() def get_stack_size_MB(self): - return sum( - os.path.getsize(self.view.directory + f) - for f in os.listdir(self.view.directory) if os.path.isfile(self.view.directory + f)) / 1024 / 1024 + return full_size_MB(self.view.stack.data.shape, self.view.stack.data.dtype)