initial commit

pyctr/util.py

@@ -0,0 +1,41 @@
+# This file is a part of ninfs.
+#
+# Copyright (c) 2017-2019 Ian Burgwin
+# This file is licensed under The MIT License (MIT).
+# You can find the full license text in LICENSE.md in the root of this project.
+
+import os
+from math import ceil
+from sys import platform
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from typing import List
+
+__all__ = ['windows', 'macos', 'readle', 'readbe', 'roundup', 'config_dirs']
+
+windows = platform in {'win32', 'cygwin'}
+macos = platform == 'darwin'
+
+
+def readle(b: bytes) -> int:
+    """Convert little-endian bytes to an int."""
+    return int.from_bytes(b, 'little')
+
+
+def readbe(b: bytes) -> int:
+    """Convert big-endian bytes to an int."""
+    return int.from_bytes(b, 'big')
+
+
+def roundup(offset: int, alignment: int) -> int:
+    """Round up a number to a provided alignment."""
+    return int(ceil(offset / alignment) * alignment)
+
+
+_home = os.path.expanduser('~')
+config_dirs: 'List[str]' = [os.path.join(_home, '.3ds'), os.path.join(_home, '3ds')]
+if windows:
+    config_dirs.insert(0, os.path.join(os.environ.get('APPDATA'), '3ds'))
+elif macos:
+    config_dirs.insert(0, os.path.join(_home, 'Library', 'Application Support', '3ds'))