Use Dim and Dir enums from utilities.mpi

gprMax/grid/mpi_grid.py

@@ -19,7 +19,6 @@
 
 import itertools
 import logging
-from enum import IntEnum, unique
 from typing import List, Optional, Tuple, TypeVar, Union
 
 import numpy as np
@@ -35,25 +34,13 @@ from gprMax.grid.fdtd_grid import FDTDGrid
 from gprMax.pml import MPIPML, PML
 from gprMax.receivers import Rx
 from gprMax.sources import Source
+from gprMax.utilities.mpi import Dim, Dir
 
 logger = logging.getLogger(__name__)
 
 CoordType = TypeVar("CoordType", bound=Union[Rx, Source])
 
 
-@unique
-class Dim(IntEnum):
-    X = 0
-    Y = 1
-    Z = 2
-
-
-@unique
-class Dir(IntEnum):
-    NEG = 0
-    POS = 1
-
-
 class MPIGrid(FDTDGrid):
     HALO_SIZE = 1
     COORDINATOR_RANK = 0

gprMax/snapshots.py

@@ -19,7 +19,6 @@
 
 import logging
 import sys
-from enum import IntEnum, unique
 from pathlib import Path
 from typing import Dict, Generic, List
 
@@ -32,6 +31,7 @@ from tqdm import tqdm
 import gprMax.config as config
 from gprMax.geometry_outputs.grid_view import GridType, GridView, MPIGridView
 from gprMax.grid.mpi_grid import MPIGrid
+from gprMax.utilities.mpi import Dim, Dir
 
 from ._version import __version__
 from .cython.snapshots import calculate_snapshot_fields
@@ -305,19 +305,6 @@ class Snapshot(Generic[GridType]):
         f.close()
 
 
-@unique
-class Dim(IntEnum):
-    X = 0
-    Y = 1
-    Z = 2
-
-
-@unique
-class Dir(IntEnum):
-    NEG = 0
-    POS = 1
-
-
 class MPISnapshot(Snapshot[MPIGrid]):
     H_TAG = 0
     EX_TAG = 1