From b6df8433d498ab6b9f98488161da205eb9b5770c Mon Sep 17 00:00:00 2001 From: 1F616EMO Date: Tue, 29 Aug 2023 06:48:17 +0800 Subject: [PATCH] Done json plotting --- mods/rp_export_json/init.lua | 26 +++- utils/plot_json_export/.gitignore | 160 +++++++++++++++++++++ utils/plot_json_export/plot_json_export.py | 30 ++++ 3 files changed, 210 insertions(+), 6 deletions(-) create mode 100644 utils/plot_json_export/.gitignore create mode 100644 utils/plot_json_export/plot_json_export.py diff --git a/mods/rp_export_json/init.lua b/mods/rp_export_json/init.lua index 5ecd725..30621b6 100644 --- a/mods/rp_export_json/init.lua +++ b/mods/rp_export_json/init.lua @@ -19,9 +19,10 @@ USA ]] +local S = minetest.get_translator("rp_export_json") local color_map = {} for hex, _ in pairs(rp_nodes.colors) do - color_map[minetest.get_content_id("rp_nodes:color_" .. hex)] = hex + color_map[minetest.get_content_id("rp_nodes:color_" .. hex)] = tonumber(hex, 16) end local function save(callback) @@ -40,12 +41,14 @@ local function save(callback) json_data.z_axis = (area[2][2] - area[1][2] + 1) json_data.map = {} local VA = VoxelArea(minp, maxp) - for z = area[2][1], area[2][2] do - for x = area[1][1], area[1][2] do + for z = area[1][2], area[2][2] do + local x_data = {} + for x = area[1][1], area[2][1] do local i = VA:index(x,1,z) local id = data[i] - json_data.map[#json_data.map+1] = color_map[id] or "" + x_data[#x_data+1] = color_map[id] or 0 end + json_data.map[#json_data.map+1] = x_data end local json = minetest.write_json(json_data) @@ -53,7 +56,9 @@ local function save(callback) minetest.safe_file_write(WP .. "/r_place.json", json) end, function(...) minetest.log("action","[rp_export_json] Done saving to JSON") - callback(...) + if callback then + callback(...) + end end, color_map, rp_core.area, data, minp, maxp) end @@ -63,4 +68,13 @@ local function loop() end) end -minetest.after(1,loop) \ No newline at end of file +minetest.after(1,loop) + +minetest.register_chatcommand("json_force_export", { + description = S("Forcely start export to JSON job"), + privs = {server = true}, + func = function(name, param) + save() + return true, minetest.colorize("orange", S("Job started.")) + end +}) \ No newline at end of file diff --git a/utils/plot_json_export/.gitignore b/utils/plot_json_export/.gitignore new file mode 100644 index 0000000..68bc17f --- /dev/null +++ b/utils/plot_json_export/.gitignore @@ -0,0 +1,160 @@ +# 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/#use-with-ide +.pdm.toml + +# 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/ diff --git a/utils/plot_json_export/plot_json_export.py b/utils/plot_json_export/plot_json_export.py new file mode 100644 index 0000000..0144314 --- /dev/null +++ b/utils/plot_json_export/plot_json_export.py @@ -0,0 +1,30 @@ +#!/usr/bin/env python + +import json, numpy as np, matplotlib.pyplot as plt + +def color_int2tuple(colorint): + # https://stackoverflow.com/a/2262152, License https://creativecommons.org/licenses/by-sa/2.5/ + return (colorint & 255, (colorint >> 8) & 255, (colorint >> 16) & 255) + +def main(argv): + if len(argv) < 2: + print("Please supply JSON filename in command line.") + return 1 + filename = argv[1] + try: + file = open(filename, "r") + except IOError: + print("Error opening file.") + raise + else: + with file: + data = json.load(file) + assert "map" in data, "Invalid JSON savefile format" + plts = np.array(tuple(tuple(color_int2tuple(int(c)) for c in r) for r in data["map"]), dtype=np.uint8) + plt.imshow(plts) + plt.show() + return 0 + +if __name__ == "__main__": + from sys import argv, exit + exit(main(argv)) \ No newline at end of file