Add fft module

brainunit/__init__.py

@@ -20,6 +20,7 @@
 from . import math
 from . import linalg
 from . import autograd
+from . import fft
 from ._base import *
 from ._base import __all__ as _base_all
 from ._celsius import *
@@ -31,5 +32,5 @@
 from ._unit_shortcuts import *
 from ._unit_shortcuts import __all__ as _std_units_all
 
-__all__ = ['math', 'linalg', 'autograd'] + _common_all + _std_units_all + _constants_all + _base_all + _celsius_all
+__all__ = ['math', 'linalg', 'autograd', 'fft'] + _common_all + _std_units_all + _constants_all + _base_all + _celsius_all
 del _common_all, _std_units_all, _constants_all, _base_all, _celsius_all, _matplotlib_compat

brainunit/_base.py

@@ -1506,6 +1506,9 @@ def __deepcopy__(self, memodict):
             is_fullname=self.is_fullname,
         )
 
+    def __hash__(self):
+        return hash((self.dim, self.base, self.scale, self.name, self.dispname, self.iscompound, self.is_fullname))
+
     def has_same_scale(self, other: 'Unit') -> bool:
         """
         Whether this Unit has the same ``scale`` as another Unit.

brainunit/fft/__init__.py

@@ -0,0 +1,25 @@
+# Copyright 2024 BDP Ecosystem Limited. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+
+from ._fft_change_unit import *
+from ._fft_change_unit import __all__ as _fft_change_unit_all
+from ._fft_keep_unit import *
+from ._fft_keep_unit import __all__ as _fft_keep_unit_all
+
+__all__ = (_fft_change_unit_all +
+           _fft_keep_unit_all)
+
+del (_fft_change_unit_all,
+     _fft_keep_unit_all,)