More work on rotate method for objects.

gprMax/cmds_geometry/add_surface_water.py

@@ -23,7 +23,7 @@ import numpy as np
 
 from ..materials import DispersiveMaterial
 from ..utilities import round_value
-from .cmds_geometry import UserObjectGeometry
+from .cmds_geometry import UserObjectGeometry, rotate_2point_object
 
 logger = logging.getLogger(__name__)
 
@@ -47,8 +47,7 @@ class AddSurfaceWater(UserObjectGeometry):
 
     def rotate(self, axis, angle, origin=None):
         pts = np.array([self.kwargs['p1'], self.kwargs['p2']])
-        rotation = UserObjectGeometry.rotate_2point_object
-        rot_pts = rotation(self, pts, axis, angle, origin)
+        rot_pts = rotate_2point_object(pts, axis, angle, origin)
         self.kwargs['p1'] = tuple(rot_pts[0, :])
         self.kwargs['p2'] = tuple(rot_pts[1, :])