[Blender3.3] 日本の地形モデルメッシュ追加アドオンの開発 (5): スクリプトファイルの分割

 前回の終わりに書いた通り、アドオンのスクリプトファイル一つだと辛くなってきたため、ファイルを複数に分割することにしました。

分割する場合、特定のフォルダ配下に __init__.py 他、関係するスクリプトファイルを配置することになります。それで、これまでの terrain_model.py の名前を継承して、フォルダ名を terrain_model として、その下にファイルを置くことにしました。

なお、このterrain_modelフォルダ、以前(トレース情報の出力)トレース用に作成した trace.py と同じフォルダに置いてあります。なので、pythonからの自動読み込みできるように設定したフォルダの下です。

では、まず __init__.py から。

# -*- coding: utf-8 -*-
#
# package: terrain_model
# file: __init__.py

bl_info = {
    "name": "New Terrain Model (JP)",
    "author": "Shiki Kuraga",
    "version": (1, 0),
    "blender": (3, 3, 0),
    "location": "View3D > Add > Mesh > New Terrain Model (JP)",
    "description": "Adds a new Terrain Model (JP) Mesh Object",
    "warning": "",
    "support": "TESTING",
    "doc_url": "",
    "category": "Add Mesh",
}

__all__ = [ 'translation', 'operator', 'model' ]

from . import translation
from . import operator

import bpy
from trace import Trace

## This allows you to right click on a button and link to documentation
#def add_object_manual_map():
#    url_manual_prefix = "https://docs.blender.org/manual/en/latest/"
#    url_manual_mapping = (
#        ("bpy.ops.mesh.add_object2", "scene_layout/object/types.html"),
#    )
#    return url_manual_prefix, url_manual_mapping


def register():
    Trace.print("register()")
    bpy.app.translations.register(__name__, translation.translation_dict)
    operator.register()
#    bpy.utils.register_manual_map(add_object_manual_map)


def unregister():
    Trace.print("unregister()")
    operator.unregister()
    bpy.app.translations.unregister(__name__)
#    bpy.utils.unregister_manual_map(add_object_manual_map)


if __name__ == "__main__":
    register()

次に、翻訳辞書を入れておくための translation.py です。

# -*- coding: utf-8 -*-
#
# package: terrain_model
# file: translation.py

translation_dict = {
    "ja_JP": {
        ("*", "Latitude"):
            "緯度",
        ("*", "Longitude"):
            "経度",
        ("*", "Scope (km)"):
            "範囲長さ (km)",
        ("*", "Terrain Model (JP)"):
            "地形モデル (日本)",
        ("*", "Add Terrain Model Mesh (JP)"):
            "地形モデルの追加 (日本)"
    }
}

三つ目ですが、メニューなど操作関係の機能を実装するための operation.py になります。

# -*- coding: utf-8 -*-
#
# package: terrain_model
# file: operator.py

import bpy
import math
from bpy.types import Operator
from bpy.props import FloatProperty, IntProperty
from bpy_extras.object_utils import AddObjectHelper, object_data_add
from . import model
from trace import Trace

def T(str):
    return bpy.app.translations.pgettext(str)

class OBJECT_OT_add_object(Operator, AddObjectHelper):
    """Create a new Terrain Model Mesh Object"""
    bl_idname = "terrain_model.add_object"
    bl_label = T("Add Terrain Model Mesh (JP)")
    bl_options = {'REGISTER', 'UNDO'}

    prop_lat: FloatProperty(
        name = "Latitude",
        description = "Latitude",
        default = 35.0 * math.pi / 180.0,
        subtype = 'ANGLE',
        min = -0.495 * math.pi,
        max = 0.495 * math.pi,
        precision = 6
    )

    prop_lon: FloatProperty(
        name = "Longitude",
        description = "Longitude",
        default = 135.0 * math.pi / 180.0,
        subtype = 'ANGLE',
        min = -1 * math.pi,
        max = 1 * math.pi,
        precision = 6
    )

    prop_scope: FloatProperty(
        name = "Scope (km)",
        description = "Scope (km)",
        default = 1.0,
        step = 10, # actual value = 10/100
        min = 0.1
    )

    prop_zoom: IntProperty(
        name = "Zoom",
        description = "Zoom",
        default = 14,
        min = 3,
        max = 15
    )
    
    def __init__(self):
        Trace.print("__init__")
        
    def __del__(self):
        Trace.print("__del__")

    def execute(self, context):

        model.add_object(self, context)

        return {'FINISHED'}

    def invoke(self, context, event):
        
        wm = context.window_manager
        
        # invoke properties dialog for this object
        return wm.invoke_props_dialog(self)

# Registration

def add_object_button(self, context):
    # insert separator
    self.layout.separator()
    self.layout.operator(
        OBJECT_OT_add_object.bl_idname,
        text = T("Terrain Model (JP)"),
        icon = 'PLUGIN')


## This allows you to right click on a button and link to documentation
#def add_object_manual_map():
#    url_manual_prefix = "https://docs.blender.org/manual/en/latest/"
#    url_manual_mapping = (
#        ("bpy.ops.terrain_model.add_object", "scene_layout/object/types.html"),
#    )
#    return url_manual_prefix, url_manual_mapping


def register():
    Trace.print("operator.register()")
    bpy.utils.register_class(OBJECT_OT_add_object)
#    bpy.utils.register_manual_map(add_object_manual_map)
    bpy.types.VIEW3D_MT_mesh_add.append(add_object_button)


def unregister():
    Trace.print("operator.unregister()")
    bpy.utils.unregister_class(OBJECT_OT_add_object)
#    bpy.utils.unregister_manual_map(add_object_manual_map)
    bpy.types.VIEW3D_MT_mesh_add.remove(add_object_button)


if __name__ == "__main__":
    register()

最後に、地形モデルの管理とメッシュの作成を行う model.py です。

# -*- coding: utf-8 -*-
#
# package: terrain_model
# file: model.py

import bpy
import math
import requests
from bpy_extras.object_utils import object_data_add
from mathutils import Vector
from trace import Trace

def T(str):
    return bpy.app.translations.pgettext(str)

def add_object(self, context):
    Trace.print("add_object(lat=%f,lon=%f,scope=%f)" % (self.prop_lat, self.prop_lon, self.prop_scope))
    
    scene = context.scene
    scale_length = scene.unit_settings.scale_length

    scale_x = self.prop_scope * 1000 / scale_length #self.scale.x
    scale_y = self.prop_scope * 1000 / scale_length #self.scale.y

    zoom = self.prop_zoom
    (x, y) = TerrainTile.latlon_to_xy(self.prop_lat, self.prop_lon)
    ix = int(x * (1 << zoom))
    iy = int(y * (1 << zoom))
    
    tile = get_cyberjapandata(zoom, ix, iy)
    mesh = tile.create_mesh(-1 * scale_x, 1 * scale_y, 2 * scale_x, -2 * scale_y)
    
    # useful for development when the mesh may be invalid.
    # mesh.validate(verbose=True)
    object_data_add(context, mesh, operator = self)


def get_cyberjapandata(zoom: int, x: int, y: int, dataid = "DEM10B"):
    """Get cyberjapandata"""
# URL:https://cyberjapandata.gsi.go.jp/xyz/dem5a/{z}/{x}/{y}.txt(DEM5A テキスト形式)zoom: 1...15
# URL:https://cyberjapandata.gsi.go.jp/xyz/dem5b/{z}/{x}/{y}.txt(DEM5B テキスト形式)zoom: 1...15
# URL:https://cyberjapandata.gsi.go.jp/xyz/dem/{z}/{x}/{y}.txt(DEM10B テキスト形式) zoom: 1...14

    # check dataid
    if dataid == "DEM10B":
        (dtype, zmax) = ("dem", 14)
    elif dataid == "DEM5A":
        (dtype, zmax) = ("dem5a", 15)
    elif dataid == "DEM5B":
        (dtype, zmax) = ("dem5b", 15)
    else:
        raise ValueError("Invalid DATAID: %s" % dataid)
        
    # check zoom
    if type(zoom) is not int:
        raise TypeError("param zoom isn't int: %s" % type(zoom))
    elif zoom < 1 or zoom > zmax:
        raise ValueError("Invalid value of z(1-%d): %d" % (zmax, zoom))
        
    # check x
    tilenum = (1 << zoom)
    if type(x) is not int:
        raise TypeError("param x isn't int: %s" % type(x))
    elif x < 0 or x >= tilenum:
        Trace.print("Out of range value x(0-%d): %d" % (tilenum, x))
        return TerrainTile()

    # check y
    if type(y) is not int:
        raise TypeError("param y isn't int: %s" % type(y))
    elif y < 0 or y >= tilenum:
        Trace.print("Out of range value y(0-%d): %d" % (tilenum, y))
        return TerrainTile()
    
    # get data
    url = "https://cyberjapandata.gsi.go.jp/xyz/%s/%d/%d/%d.txt" % (dtype, zoom, x, y)
    Trace.print("get data from url: <%s>" % url)
    res = requests.get(url)
    s = res.content.decode()
    tile = TerrainTile(csvtext = s)
    Trace.print(tile)
    return tile


class TerrainTile:
    """Data Handler of Terrain Tile"""
    
    PIXEL_NUM = 256
    
    def latlon_to_xy(latitude,longitude):
        x = ((math.pi + longitude) / (2 * math.pi)) % 1
        
        siny = math.sin(latitude)
        siny = max(-0.999, min(0.999, siny))
        y = 0.5 - math.log((1 + siny) / (1 - siny)) / (4 * math.pi)
        
        return (x, y)
    
    def __init__(self, csvtext = ""):
        self.altitudes = []
        if len(csvtext) > 0:
            self.load_from_csvtext(csvtext)

    def __str__(self):
        return "" % len(self.altitudes)
        
    def load_from_csvtext(self, csvtext):
        alts = []
        if len(csvtext.strip()) == 0:
            self.altitudes = alts
            return

        PIXEL_NUM = TerrainTile.PIXEL_NUM
        lines = csvtext.split('\n')
        if len(lines) < PIXEL_NUM:
            raise ValueError("num of lines is less than PIXEL_NUM(%d): %d lines" % (PIXEL_NUM, len(lines)))
        
        for i, line in enumerate(lines[:PIXEL_NUM]):
            cols = line.split(',')
            if len(cols) != PIXEL_NUM:
                raise ValueError("num of columns at line %d is not PIXEL_NUM(%d): %d cols" % (i + 1, PIXEL_NUM, len(cols)))
            for col in cols:
                col = col.strip()
                if col == "e":
                    alts.append(None)
                else:
                    alts.append(float(col))
        self.altitudes = alts
        
    def create_mesh(self, start_x, start_y, width, height, name = "Terrain Model Mesh", adjust = True):

        offset_z = 0
        if adjust:
            max_z = max(filter(None, self.altitudes))
            min_z = min(filter(None, self.altitudes))
            Trace.print("min_z = ", min_z)
            if min_z > 0:
                offset_z = -min_z
            elif max_z < 0:
                offset_z = -max_z
            
        idmap = []
        verts = []
        for index_y in range(0, self.PIXEL_NUM):
            y = start_y + (index_y + 0.5) * height / self.PIXEL_NUM
            
            for index_x, alt in enumerate(self.altitudes[(index_y * self.PIXEL_NUM):((index_y + 1) * self.PIXEL_NUM)]):
                if alt is not None:
                    x = start_x + (index_x + 0.5) * width / self.PIXEL_NUM
                    idmap.append(len(verts))
                    verts.append(Vector((x, y, alt + offset_z)))
                else:
                    idmap.append(-1)

        faces = []
        for index_y in range(0, self.PIXEL_NUM - 1):
            iy0 = index_y * self.PIXEL_NUM
            iy1 = iy0 + self.PIXEL_NUM
            for ix0, ix1 in enumerate(range(1, self.PIXEL_NUM)):
                face = [idmap[i] for i in [ix0 + iy0, ix1 + iy0, ix1 + iy1, ix0 + iy1] if idmap[i] >= 0]
                if (len(face) >= 3):
                    faces.append(face)
        
        if len(faces) > 0:
            edges = []
            mesh = bpy.data.meshes.new(name = name)
            mesh.from_pydata(verts, edges, faces)
            return mesh
        
        return None

内容は、これまでのものと変わりません。

しかし、今まであったトレースを開始するためのコードが __init__.py に存在しません。

というのも、__init__.py に記入して呼び出すと不都合があることが分かったからです。

テキストエディタからのスクリプト実行時、__init__.pyの内容を直接実行しようとすると、from . import translation の処理が親フォルダがないというエラーになります。(正確には、"ImportError: attempted relative import with no known parent package")

これは、__init__.py の直接実行だと、__init__.pyのあるフォルダがカレントフォルダになってしまい、親フォルダがカレントフォルダ外となりエラーとなっているようです。

そのため、__init__.py にトレース開始のコードは入れられないのです。

なので、terrain_modelフォルダの親フォルダにテキストエディタ内で走らせるためのスクリプトを別に用意しました。これを、test_terrain_model.py とします。

そのファイルの内容は以下です。

# -*- coding: utf-8 -*-
#
# file: test_terrain_model.py

package_name = 'terrain_model'

# check if retry
import bpy
import sys
import importlib

if package_name in sys.modules.keys():
    # reload on retrying
    module_obj = sys.modules[package_name]
    module_obj.unregister()
    
    importlib.reload(module_obj)
    module_names = [ package_name + '.' + s for s in module_obj.__all__ ]
    modules = [ sys.modules[s] for s in sys.modules.keys() if s in module_names ]
    for module in modules:
        importlib.reload(module)

# start testing
module = importlib.import_module(package_name)
for name in module.__all__:
    importlib.import_module('.' + name, package_name)

# start testing
from trace import Trace
Trace.on()

module.register()

このファイル、再利用できるように特定のパッケージに依存しないように作成しましたので、今一つ見づらいかもしれませんが、やっていることはこれまでのterrain_model.pyでトレースを開始するためのコードです。ただ、その前にパッケージ内のモジュールの再読み込みの処理を入れてます。

これで、terrain_model内のどのファイルを更新しても、このトレース開始用のコードを走らせれば再読み込みしてくれます。(面倒なので、アドオンのインストールはしないでテストしてます)

次は、本アドオンの完成に向けた手直しを進めたいかなと思ってます。

前回(4)】日本の地形モデルメッシュ追加アドオンの開発 (5) 【次回(6)


このブログの人気の投稿

パズドラ 転生龍刀士降臨周回編成 (ベア戦)

パズドラ HUNTER☓HUNTER キメラアント周回編成 (ペア戦)

[Blender3.3] mmd_toolsはどれが最新?