kago-utils 0.1.0__tar.gz

kago_utils-0.1.0/.github/workflows/publish-to-pypi.yml

@@ -0,0 +1,33 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types:
+      - released
+
+jobs:
+  release:
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v4
+
+    - name: Setup Python
+      uses: actions/setup-python@v5
+      with:
+        python-version-file: .python-version
+
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        python -m pip install build twine hatch-vcs
+
+    - name: Build the package
+      run: python -m build
+
+    - name: Publish to PyPI
+      env:
+        TWINE_USERNAME: __token__
+        TWINE_PASSWORD: ${{ secrets.PYPI_API_TOKEN }}
+      run: python -m twine upload dist/*
+

kago_utils-0.1.0/.github/workflows/publish-to-test-pypi.yml

@@ -0,0 +1,33 @@
+name: Publish to Test PyPI
+
+on:
+  release:
+    types:
+      - prereleased
+
+jobs:
+  release:
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v4
+
+    - name: Setup Python
+      uses: actions/setup-python@v5
+      with:
+        python-version-file: .python-version
+
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        python -m pip install build twine hatch-vcs
+
+    - name: Build the package
+      run: python -m build
+
+    - name: Publish to Test PyPI
+      env:
+        TWINE_USERNAME: __token__
+        TWINE_PASSWORD: ${{ secrets.TEST_PYPI_API_TOKEN }}
+      run: python -m twine upload --repository testpypi dist/*
+

kago_utils-0.1.0/.github/workflows/test.yml

@@ -0,0 +1,33 @@
+name: Test
+
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - '**'
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Setup Python
+        uses: actions/setup-python@v5
+        with:
+          python-version-file: .python-version
+          cache: 'pipenv'
+
+      - name: Install pipenv
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install pipenv
+
+      - name: Install dependencies
+        run: python -m pipenv install --dev
+
+      - name: Run tests
+        run: python -m pipenv run test

kago_utils-0.1.0/.gitignore

@@ -0,0 +1,165 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/latest/usage/project/#working-with-version-control
+.pdm.toml
+.pdm-python
+.pdm-build/
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+
+# MacOS
+.DS_Store

kago_utils-0.1.0/.python-version

@@ -0,0 +1 @@
+3.10

kago_utils-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 zurukumo
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

kago_utils-0.1.0/PKG-INFO

@@ -0,0 +1,17 @@
+Metadata-Version: 2.3
+Name: kago-utils
+Version: 0.1.0
+Summary: Mahjong Library for Python
+Project-URL: Homepage, https://github.com/zurukumo/kago-utils
+Project-URL: Issues, https://github.com/zurukumo/kago-utils/issues
+Author-email: zurukumo <zurukumo@gmail.com>
+License-File: LICENSE
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+
+# kago-utils

kago_utils-0.1.0/Pipfile

@@ -0,0 +1,16 @@
+[[source]]
+url = "https://pypi.org/simple"
+verify_ssl = true
+name = "pypi"
+
+[packages]
+
+[dev-packages]
+numpy = "*"
+shanten-tools = "*"
+
+[requires]
+python_version = "3.10"
+
+[scripts]
+test = "python -m unittest discover -v tests"

kago_utils-0.1.0/Pipfile.lock

@@ -0,0 +1,97 @@
+{
+    "_meta": {
+        "hash": {
+            "sha256": "abc924915a23d4cdd24cd8b32e7fc7831428a27f2e8e6bd1f5fd47d9b7705df9"
+        },
+        "pipfile-spec": 6,
+        "requires": {
+            "python_version": "3.10"
+        },
+        "sources": [
+            {
+                "name": "pypi",
+                "url": "https://pypi.org/simple",
+                "verify_ssl": true
+            }
+        ]
+    },
+    "default": {},
+    "develop": {
+        "numpy": {
+            "hashes": [
+                "sha256:08458fbf403bff5e2b45f08eda195d4b0c9b35682311da5a5a0a0925b11b9bd8",
+                "sha256:0fbb536eac80e27a2793ffd787895242b7f18ef792563d742c2d673bfcb75134",
+                "sha256:12f5d865d60fb9734e60a60f1d5afa6d962d8d4467c120a1c0cda6eb2964437d",
+                "sha256:15eb4eca47d36ec3f78cde0a3a2ee24cf05ca7396ef808dda2c0ddad7c2bde67",
+                "sha256:173a00b9995f73b79eb0191129f2455f1e34c203f559dd118636858cc452a1bf",
+                "sha256:1b902ce0e0a5bb7704556a217c4f63a7974f8f43e090aff03fcf262e0b135e02",
+                "sha256:1f682ea61a88479d9498bf2091fdcd722b090724b08b31d63e022adc063bad59",
+                "sha256:1f87fec1f9bc1efd23f4227becff04bd0e979e23ca50cc92ec88b38489db3b55",
+                "sha256:24a0e1befbfa14615b49ba9659d3d8818a0f4d8a1c5822af8696706fbda7310c",
+                "sha256:2c3a346ae20cfd80b6cfd3e60dc179963ef2ea58da5ec074fd3d9e7a1e7ba97f",
+                "sha256:36d3a9405fd7c511804dc56fc32974fa5533bdeb3cd1604d6b8ff1d292b819c4",
+                "sha256:3fdabe3e2a52bc4eff8dc7a5044342f8bd9f11ef0934fcd3289a788c0eb10018",
+                "sha256:4127d4303b9ac9f94ca0441138acead39928938660ca58329fe156f84b9f3015",
+                "sha256:4658c398d65d1b25e1760de3157011a80375da861709abd7cef3bad65d6543f9",
+                "sha256:485b87235796410c3519a699cfe1faab097e509e90ebb05dcd098db2ae87e7b3",
+                "sha256:529af13c5f4b7a932fb0e1911d3a75da204eff023ee5e0e79c1751564221a5c8",
+                "sha256:5a3d94942c331dd4e0e1147f7a8699a4aa47dffc11bf8a1523c12af8b2e91bbe",
+                "sha256:5daab361be6ddeb299a918a7c0864fa8618af66019138263247af405018b04e1",
+                "sha256:61728fba1e464f789b11deb78a57805c70b2ed02343560456190d0501ba37b0f",
+                "sha256:6790654cb13eab303d8402354fabd47472b24635700f631f041bd0b65e37298a",
+                "sha256:69ff563d43c69b1baba77af455dd0a839df8d25e8590e79c90fcbe1499ebde42",
+                "sha256:6bf4e6f4a2a2e26655717a1983ef6324f2664d7011f6ef7482e8c0b3d51e82ac",
+                "sha256:6e4eeb6eb2fced786e32e6d8df9e755ce5be920d17f7ce00bc38fcde8ccdbf9e",
+                "sha256:72dc22e9ec8f6eaa206deb1b1355eb2e253899d7347f5e2fae5f0af613741d06",
+                "sha256:75b4e316c5902d8163ef9d423b1c3f2f6252226d1aa5cd8a0a03a7d01ffc6268",
+                "sha256:7b9853803278db3bdcc6cd5beca37815b133e9e77ff3d4733c247414e78eb8d1",
+                "sha256:7d6fddc5fe258d3328cd8e3d7d3e02234c5d70e01ebe377a6ab92adb14039cb4",
+                "sha256:81b0893a39bc5b865b8bf89e9ad7807e16717f19868e9d234bdaf9b1f1393868",
+                "sha256:8efc84f01c1cd7e34b3fb310183e72fcdf55293ee736d679b6d35b35d80bba26",
+                "sha256:8fae4ebbf95a179c1156fab0b142b74e4ba4204c87bde8d3d8b6f9c34c5825ef",
+                "sha256:99d0d92a5e3613c33a5f01db206a33f8fdf3d71f2912b0de1739894668b7a93b",
+                "sha256:9adbd9bb520c866e1bfd7e10e1880a1f7749f1f6e5017686a5fbb9b72cf69f82",
+                "sha256:a1e01dcaab205fbece13c1410253a9eea1b1c9b61d237b6fa59bcc46e8e89343",
+                "sha256:a8fc2de81ad835d999113ddf87d1ea2b0f4704cbd947c948d2f5513deafe5a7b",
+                "sha256:b83e16a5511d1b1f8a88cbabb1a6f6a499f82c062a4251892d9ad5d609863fb7",
+                "sha256:bb2124fdc6e62baae159ebcfa368708867eb56806804d005860b6007388df171",
+                "sha256:bfc085b28d62ff4009364e7ca34b80a9a080cbd97c2c0630bb5f7f770dae9414",
+                "sha256:cbab9fc9c391700e3e1287666dfd82d8666d10e69a6c4a09ab97574c0b7ee0a7",
+                "sha256:e5eeca8067ad04bc8a2a8731183d51d7cbaac66d86085d5f4766ee6bf19c7f87",
+                "sha256:e9e81fa9017eaa416c056e5d9e71be93d05e2c3c2ab308d23307a8bc4443c368",
+                "sha256:ea2326a4dca88e4a274ba3a4405eb6c6467d3ffbd8c7d38632502eaae3820587",
+                "sha256:eacf3291e263d5a67d8c1a581a8ebbcfd6447204ef58828caf69a5e3e8c75990",
+                "sha256:ec87f5f8aca726117a1c9b7083e7656a9d0d606eec7299cc067bb83d26f16e0c",
+                "sha256:f1659887361a7151f89e79b276ed8dff3d75877df906328f14d8bb40bb4f5101",
+                "sha256:f9cf5ea551aec449206954b075db819f52adc1638d46a6738253a712d553c7b4"
+            ],
+            "index": "pypi",
+            "markers": "python_version >= '3.9'",
+            "version": "==2.0.1"
+        },
+        "shanten-tools": {
+            "hashes": [
+                "sha256:0530752f8dc1b5b0a73c7b2724c91404d9123d8777cbcebc892f5203baeca9ba",
+                "sha256:1b9e2ace7377a4e05f7a76fe794aaabc4717612427ef6648d41714fe1b212454",
+                "sha256:1e471ba0f50a52f9f1108656f96644ea6977d73373d4fcefcbf2eae194e38593",
+                "sha256:1e7301736942b02ccd40079ad19ba34b3bcc8bc6c31ff92634fa541bf1a4daaa",
+                "sha256:21a95c312a28b41084092410e0f2705cf48d563f4a63bb15788adebfe9af28db",
+                "sha256:2caa1d86a223aa5613003da778362c293c3bd7d3158039ddef3a902957a12b9c",
+                "sha256:54dcdfbb726e13e952d19cb2fcecb0f8a5c01455d112d529fba674add6ad17f5",
+                "sha256:64cc3fa25dcc9eefcff8440397917d4b19b31d21f0eb8fe80202bb07c18fd375",
+                "sha256:756cf523c6dff1707e1d6ba4cfb0d6e7eeb96fc4aea6179f8593973a3dfc7f85",
+                "sha256:76d0bd55aa36a9e7ec11787bc5799bba8ec10e5c6b1dbbe8d1fc6a4ef7cc760b",
+                "sha256:ba5fef7e0d6ae9a57b8914e77794512e50d39fbb219b421845d7b0c40c25a65b",
+                "sha256:d0ac4442734dd589d762035319c1fc2da8e0eb7d88cf64c2332df98b2ff19c65",
+                "sha256:d49e5e0e7cdd627d42930c676281375d9736f940d06788c4cc4921c8e2d59e3c",
+                "sha256:dea9bac83c6a00ce5a2a30333fa7dd8bfe47627471dcd1fcc525c97e5e7aa059",
+                "sha256:e592fed3e2187aff4da9262c264a6eb9ab117ec96744210bf5ae310ff1725296",
+                "sha256:e646bafe00da80a926dbc3c6c9d45bb8c6f3c36a7578da429b919bfcc4bc908b",
+                "sha256:e8a820493589833fd687f61f2f0d2a3a3e0a598133fc4339d3c2ea7811978e0d"
+            ],
+            "index": "pypi",
+            "markers": "python_version >= '3.7'",
+            "version": "==0.1.8"
+        }
+    }
+}

kago_utils-0.1.0/README.md

@@ -0,0 +1 @@
+# kago-utils

kago_utils-0.1.0/kago_utils/__init__.py

@@ -0,0 +1,6 @@
+from importlib.metadata import PackageNotFoundError, version
+
+try:
+    __version__ = version(__package__)
+except PackageNotFoundError:
+    __version__ = "unknown"

kago_utils-0.1.0/kago_utils/generate_shanten_table.py

@@ -0,0 +1,205 @@
+import pickle
+from collections import deque
+from itertools import product
+
+
+def extract_shuntsu(tehai):
+    N = len(tehai)
+
+    # queue format: [tehai, n_shuntsu, start]
+    queue = deque([])
+    queue.append([tehai.copy(), 0, 0])
+
+    while queue:
+        tehai, n_shuntsu, start = queue.popleft()
+        yield tehai.copy(), n_shuntsu
+
+        for i in range(start, N):
+            if i + 2 < N and i <= 6 and tehai[i] >= 1 and tehai[i+1] >= 1 and tehai[i+2] >= 1:
+                tehai[i] -= 1
+                tehai[i+1] -= 1
+                tehai[i+2] -= 1
+                queue.append([tehai.copy(), n_shuntsu + 1, i])
+                tehai[i] += 1
+                tehai[i+1] += 1
+                tehai[i+2] += 1
+
+
+def extract_kootsu(tehai):
+    N = len(tehai)
+
+    # queue format: [tehai, n_kootsu, start]
+    queue = deque([])
+    queue.append([tehai.copy(), 0, 0])
+
+    while queue:
+        tehai, n_kootsu, start = queue.popleft()
+        yield tehai.copy(), n_kootsu
+
+        for i in range(start, N):
+            if tehai[i] >= 3:
+                tehai[i] -= 3
+                queue.append([tehai.copy(), n_kootsu + 1, i])
+                tehai[i] += 3
+
+
+def extract_toitsu(tehai):
+    N = len(tehai)
+
+    # queue format: [tehai, n_toitsu, start]
+    queue = deque([])
+    queue.append([tehai.copy(), 0, 0])
+
+    while queue:
+        tehai, n_toitsu, start = queue.popleft()
+        yield tehai.copy(), n_toitsu
+
+        for i in range(start, N):
+            if tehai[i] >= 2:
+                tehai[i] -= 2
+                queue.append([tehai.copy(), n_toitsu + 1, i])
+                tehai[i] += 2
+
+
+def extract_taatsu(tehai):
+    N = len(tehai)
+
+    # queue format: [tehai, n_taatsu, start]
+    queue = deque([])
+    queue.append([tehai.copy(), 0, 0])
+
+    while queue:
+        tehai, n_taatsu, start = queue.popleft()
+        yield tehai.copy(), n_taatsu
+
+        for i in range(start, N):
+            if i + 1 < N and tehai[i] >= 1 and tehai[i+1] >= 1:
+                tehai[i] -= 1
+                tehai[i+1] -= 1
+                queue.append([tehai.copy(), n_taatsu + 1, i])
+                tehai[i] += 1
+                tehai[i+1] += 1
+
+            if i + 2 < N and tehai[i] >= 1 and tehai[i+2] >= 1:
+                tehai[i] -= 1
+                tehai[i+2] -= 1
+                queue.append([tehai.copy(), n_taatsu + 1, i])
+                tehai[i] += 1
+                tehai[i+2] += 1
+
+
+def list_suuhai_patterns(tehai):
+    tehai = list(tehai)
+    for rest_tehai, n_kootsu in extract_kootsu(tehai):
+        for rest_tehai, n_shuntsu in extract_shuntsu(rest_tehai):
+            for rest_tehai, n_toitsu in extract_toitsu(rest_tehai):
+                for rest_tehai, n_taatsu in extract_taatsu(rest_tehai):
+                    yield n_kootsu, n_shuntsu, n_toitsu, n_taatsu
+
+
+def list_zihai_patterns(tehai):
+    tehai = list(tehai)
+    for rest_tehai, n_kootsu in extract_kootsu(tehai):
+        for rest_tehai, n_toitsu in extract_toitsu(rest_tehai):
+            yield n_kootsu, 0, n_toitsu, 0
+
+
+def select_better_pattern1(max1, current):
+    max_score = max1[0] * 2 + max1[1]
+    current_score = current[0] * 2 + current[1]
+    if current_score > max_score or (current_score == max_score and current[2]):
+        return current
+    else:
+        return max1
+
+
+def select_better_pattern2(max2, current):
+    max_score = max2[0] * 8 + max2[1]
+    current_score = current[0] * 8 + current[1]
+    if current_score > max_score or (current_score == max_score and current[2]):
+        return current
+    else:
+        return max2
+
+
+def select_better_pattern3(max3, current):
+    max_score = max3[0] * 2 + max3[1] + (100 if max3[2] else 0)
+    current_score = current[0] * 2 + current[1] + (100 if current[2] else 0)
+    if current_score > max_score or (current_score == max_score and current[2]):
+        return current
+    else:
+        return max3
+
+
+def select_better_pattern4(max4, current):
+    max_score = max4[0] * 8 + max4[1] + (100 if max4[2] else 0)
+    current_score = current[0] * 8 + current[1] + (100 if current[2] else 0)
+    if current_score > max_score or (current_score == max_score and current[2]):
+        return current
+    else:
+        return max4
+
+
+# generate patterns
+def generate_patterns():
+    suuhai_patterns = dict()
+    zihai_patterns = dict()
+
+    # generate suuhai patterns
+    for tehai in product([0, 1, 2, 3, 4], repeat=9):
+        if sum(tehai) > 14:
+            continue
+
+        print(tehai)
+
+        # pattern format: [n_mentsu, n_pre_mentsu, has_toitsu]
+        max_pattern1 = 0, 0, False  # maximize n_mentsu * 2 + n_pre_mentsu
+        max_pattern2 = 0, 0, False  # maximize n_mentsu * 8 + n_pre_mentsu
+        max_pattern3 = 0, 0, False  # maximize n_mentsu * 2 + n_pre_mentsu + 100 if has_toitsu
+        max_pattern4 = 0, 0, False  # maximize n_mentsu * 8 + n_pre_mentsu + 100 if has_toitsu
+        for n_kootsu, n_shuntsu, n_toitsu, n_taatsu in list_suuhai_patterns(tehai):
+            n_mentsu = n_kootsu + n_shuntsu
+            n_pre_mentsu = n_toitsu + n_taatsu
+            has_toitsu = n_toitsu > 0
+            current_pattern = n_mentsu, n_pre_mentsu, has_toitsu
+            max_pattern1 = select_better_pattern1(max_pattern1, current_pattern)
+            max_pattern2 = select_better_pattern2(max_pattern2, current_pattern)
+            max_pattern3 = select_better_pattern3(max_pattern3, current_pattern)
+            max_pattern4 = select_better_pattern4(max_pattern4, current_pattern)
+
+        # when max1 is same as max2, only store max1 because of memory efficiency
+        suuhai_patterns[tehai] = list(set([max_pattern1, max_pattern2, max_pattern3, max_pattern4]))
+
+    # generate zihai patterns
+    for tehai in product([0, 1, 2, 3, 4], repeat=7):
+        if sum(tehai) > 14:
+            continue
+
+        print(tehai)
+
+        # pattern format: [n_mentsu, n_pre_mentsu, has_toitsu]
+        max_pattern1 = 0, 0, False  # maximize n_mentsu * 2 + n_pre_mentsu
+        max_pattern2 = 0, 0, False  # maximize n_mentsu * 8 + n_pre_mentsu
+        max_pattern3 = 0, 0, False  # maximize n_mentsu * 2 + n_pre_mentsu + (100 if has_toitsu)
+        max_pattern4 = 0, 0, False  # maximize n_mentsu * 8 + n_pre_mentsu + (100 if has_toitsu)
+        for n_kootsu, n_shuntsu, n_toitsu, n_taatsu in list_zihai_patterns(tehai):
+            n_mentsu = n_kootsu + n_shuntsu
+            n_pre_mentsu = n_toitsu + n_taatsu
+            has_toitsu = n_toitsu > 0
+            current_pattern = n_mentsu, n_pre_mentsu, has_toitsu
+            max_pattern1 = select_better_pattern1(max_pattern1, current_pattern)
+            max_pattern2 = select_better_pattern2(max_pattern2, current_pattern)
+            max_pattern3 = select_better_pattern3(max_pattern3, current_pattern)
+            max_pattern4 = select_better_pattern4(max_pattern4, current_pattern)
+
+        # when max1 is same as max2, only store max1 because of memory efficiency
+        zihai_patterns[tehai] = list(set([max_pattern1, max_pattern2, max_pattern3, max_pattern4]))
+
+    with open('suuhai_patterns.pickle', 'wb') as f:
+        pickle.dump(suuhai_patterns, f)
+    with open('zihai_patterns.pickle', 'wb') as f:
+        pickle.dump(zihai_patterns, f)
+
+
+if __name__ == '__main__':
+    generate_patterns()