diff --git a/src/raster/r.vif/r.vif.py b/src/raster/r.vif/r.vif.py index cc73c44844..0b3e9926f4 100755 --- a/src/raster/r.vif/r.vif.py +++ b/src/raster/r.vif/r.vif.py @@ -123,6 +123,14 @@ # Functions +def prRed(skk): + print("\033[91m {}\033[00m".format(skk)) + + +def prGreen(skk): + print("\033[92m {}\033[00m".format(skk)) + + CLEAN_RAST = [] @@ -348,7 +356,13 @@ def main(options, flags): vifstat = compute_vif2(x, y) else: # Compute vif using sample - y = p[:, k] + try: + y = p[:, k] + except IndexError as e: + prRed(f"An error occurred: {str(e)}") + prGreen( + "Tip: check if all input rasters have values within the computation region." + ) x = np.delete(p, k, axis=1) vifstat = compute_vif(x, y)