この記事は著者が韓国語で書いた原文の翻訳です。 原文(한국어)を読む
壁距離を O(N log N) で解く方法:Eikonal 方程式と Fast Marching
壁距離は総当たりの距離ではなく到達時間です。毎ステップ全域を行進させるかは、格子が動いたかで決まります。
1996年、カリフォルニア大学バークレー校の James Sethian 氏は、医療画像における血管境界のトラッキングアルゴリズムを改良していました。その成果である Fast Marching Method は、現在、意外な場所で使われています。航空機の翼周囲の RANS(Reynolds-Averaged Navier-Stokes)ソルバーが実行されるたびに、まさにこのアルゴリズムが各格子点の壁距離(wall distance)を生成しています。Spalart-Allmaras モデルや k-ω SST モデルは、各格子において「最も近い壁はどこか」という値を入力として必要とします。
パターンは単一円柱から始めます。速度 を上げると、同じ瞬間に黄色い帯がより遠くへ進みます。障害物をスリットに変えると、帯が出発する壁の形が変わります。
黄色い帯が現在アップデートされている Considered 集合、つまり「波面(front)」そのものです。壁と円柱の境界から同時に波面が出発し、互いに衝突する地点で距離が確定します。2つの円柱のパターンでは、2つの波面の衝突線が円柱間の中間軸(medial axis)付近に形成されます。スリット壁のパターンでは、波面が隙間を回り込んで背面に到達する様子が分かります。この「回り込み」が、単純なユークリッド距離と異なる点です。実際の航空機やタービンの形状でも、壁の背後の乱流モデル入力には、この回り込んだ距離が必要とされます。
なぜ乱流モデルに壁距離が必要なのか
境界層乱流の統計量は、壁からの距離 に強く依存します。RANS モデルはその依存性を代数式として組み込みます。Spalart-Allmaras では、破壊(destruction)項で (壁距離)を直接使用します。k-ω SST では、ブレンディング関数 の中に が入ります。LES との混合モデル(DES, DDES, IDDES)でも、壁距離がスイッチの役割を果たします。
問題は、 がメッシュの幾何量であるということです。メッシュが移動したり変形したりする場合、各タイムステップごとに再計算する必要があります。回転機械(コンプレッサー、ポンプ)のように、ローターとステーターの境界が相対運動をする場合がこれに該当します。
愚直な方法の落とし穴
すぐに思いつく方法は、「すべての壁面とすべてのセルの距離を計算して最小値を選ぶ」というものです。実際、教科書の例題ではこれが使われています。
このコストは です。航空機全体を解くとしましょう。1,000万セル、50万壁面であれば、5兆回の距離計算になります。これがタイムステップごとに実行されると、解析は終わりません。KD-tree のような空間データ構造を用いることで まで下げることができますが、複雑な形状の壁近傍クエリは依然として大きな負担です。同じ総当たり距離を粒子接触で削った記録が、セクター幅を粒子一つから決める DEM にあります。壁距離とは対象が違います。候補を幾何で減らす理由は同じです。
Eikonal 方程式へと視点を変える#
壁距離を PDE(偏微分方程式)の解として捉えてみましょう。「壁で 0 であり、すべての方向で勾配の大きさが 1」である関数こそが距離場です。
ここで、 は壁までの「到達時間」、 は伝播速度(デフォルトは 1)、 は壁の集合です。 と固定すれば、 はそのままユークリッド距離となります。
この方程式は双曲型(hyperbolic)です。情報は特性線(壁から外側へ伸びる直線)に沿って一方向にのみ流れます。この一方向性が効率的なアルゴリズムの鍵となります。「すでに値が定まった点からのみ値が伝わる」という性質を利用すれば、すべての点を一度訪問するだけで十分です。
Fast Marching:Dijkstra 法の連続版#
Sethian 氏のアイデアは、次の2ステップに要約されます。
- すべての格子点を Accepted / Considered / Far の3つの集合に分ける。
- Considered の中で到達時間が最も小さい点を Accepted として確定し、その隣接点をアップデートする。
最小値を素早く取り出すには、min-heap(最小ヒープ)が適しています。これはグラフの最短経路アルゴリズムである Dijkstra 法と同じ構造です。違いは隣接点のアップデート方法にあります。グラフでは単純な加算ですが、Eikonal 方程式ではアップウィンド(upwind)2次方程式を解いて値を求めます。
2D 構造格子におけるアップデート式は以下の通りです。
ここで、 はそれぞれ水平・垂直方向のアップウィンド隣接値(小さい方)、 は格子間隔です。2つの解のうち、 である方のみを採用します。この条件を因果律(causality)と呼びます。これは、値は「すでに確定した、より小さい」隣接点からのみ伝わらなければならないという Eikonal 方程式の一方向性を強制するものです。
一方向からのみ情報が伝わる場合(他の隣接点が Far の場合)、式は単純になります。
min-heap への挿入・削除はそれぞれ であり、各点は平均して定数回処理されるため、総コストは となります。
Python による実装#
import heapq
import numpy as np
INF = np.inf
FAR, CONSIDERED, ACCEPTED = 0, 1, 2
def solve_eikonal_neighbor(T, j, i, h, F):
"""ある点において 2次アップウィンド解を求めます。"""
ny, nx = T.shape
Tx = min(
T[j, i - 1] if i > 0 else INF,
T[j, i + 1] if i < nx - 1 else INF,
)
Ty = min(
T[j - 1, i] if j > 0 else INF,
T[j + 1, i] if j < ny - 1 else INF,
)
a, b = sorted([Tx, Ty])
inv_f = h / F
if not np.isfinite(b) or b >= a + inv_f:
return a + inv_f
diff = b - a
disc = 2 * inv_f * inv_f - diff * diff
if disc < 0:
return a + inv_f
return 0.5 * (a + b + np.sqrt(disc))
def march_wall_distance(walls, h=1.0, F=1.0):
"""2D 構造格子において Fast Marching で壁距離を計算します。"""
ny, nx = walls.shape
T = np.full_like(walls, INF, dtype=np.float64)
status = np.zeros_like(walls, dtype=np.uint8)
heap = []
T[walls] = 0.0
status[walls] = ACCEPTED
# 壁の四方向の隣接点を Considered として初期化
for j in range(ny):
for i in range(nx):
if not walls[j, i]:
continue
for dj, di in [(-1, 0), (1, 0), (0, -1), (0, 1)]:
nj, ni = j + dj, i + di
if 0 <= nj < ny and 0 <= ni < nx and status[nj, ni] != ACCEPTED:
new_t = solve_eikonal_neighbor(T, nj, ni, h, F)
if new_t < T[nj, ni]:
T[nj, ni] = new_t
status[nj, ni] = CONSIDERED
heapq.heappush(heap, (new_t, nj, ni))
while heap:
t, j, i = heapq.heappop(heap)
if status[j, i] == ACCEPTED:
continue # 古いエントリー
status[j, i] = ACCEPTED
for dj, di in [(-1, 0), (1, 0), (0, -1), (0, 1)]:
nj, ni = j + dj, i + di
if not (0 <= nj < ny and 0 <= ni < nx):
continue
if status[nj, ni] == ACCEPTED:
continue
new_t = solve_eikonal_neighbor(T, nj, ni, h, F)
if new_t < T[nj, ni]:
T[nj, ni] = new_t
status[nj, ni] = CONSIDERED
heapq.heappush(heap, (new_t, nj, ni))
return T
# 検証:平行平板チャネル
walls = np.zeros((61, 121), dtype=bool)
walls[0, :] = True
walls[-1, :] = True
d = march_wall_distance(walls, h=1.0)
# 中央線 (y=30) での壁距離は 30 である必要があります
print(f"中央壁距離 = {d[30, 60]:.4f} (正解 30.0)")
print(f"最大相対誤差 = {np.max(np.abs(d[30, :] - 30)):.4f}")チャネル流では解析解 が綺麗に得られます。上記のコードの出力は、中央線でほぼ正確に 30.0 と一致します。2D の一次アップウィンド近似であるため、コーナー付近でわずかな誤差が生じますが、実務上は十分な精度です。
片側だけ確定すると、セルごとに 1 が足される#
チャネル検証の格子は 61×121、間隔は 、速度は です。下壁のインデックスは 0、上壁は 60、中央線は 30 です。隣が片側だけ Accepted なら、更新は に縮みます。 なので、壁から一セル離れるたびに到達時間が 1 増えます。三十セルなら 30 です。上壁から下りてきても 30 です。二つの波面の最小値が、中央の 30.0 です。コードのコメントが正解として書いた数字は、この足し算から出ます。コーナーの残差は、判別式が負になって一次の枝に落ちる場所で生じます。
10M セルと 0.5M 壁面の積は 5 兆です。Fast Marching は点ごとにヒープを で触り、点は Accepted に一度だけ上がります。セルが なら は底 10 で 7、底 2 でも 24 前後です。演算が数億を大きく超えるのは難しいです。5 兆とは桁が四つ以上離れます。KD-tree も の対数をセル数に掛け直します。壁クエリの方が行進より多く残ります。
二つの隣の差 が大きく、 が負なら、解は小さい隣に を足した一次式へ戻ります。チャネルのように片側がまだ Far なら、この枝が最初から選ばれます。平面壁の距離場は、だからセル数そのものです。
非構造格子への拡張
構造格子上の公式は美しいですが、CFD の半分は非構造格子上で計算されます。Sethian & Vladimirsky (2000) はこの問題を解決しました。ある点 の値を、その点を含むシンプレックス(三角形、四面体)の他の頂点における値から求めます。
方向ベクトル を行としてまとめた行列 と、方向微分 から正規方程式を解くと、Eikonal 方程式 は に関する2次方程式となります。2つの解のうち、近似勾配がシンプレックス内部を向いている方のみを採用します。この「シンプレックス因果律(simplex causality)」が、構造格子における単純なアップウィンド条件を一般化したものです。
実務チェックリスト
- 移動メッシュ: 全体を再計算する代わりに、変形が大きい領域のみを更新するレイジーアップデートが有効です。Fast Marching は逐次的な処理であるため並列化が難しいですが、ナローバンド(narrow band)変形体は実装に適しています。
- 周期境界条件 (Periodic BC): 周期境界側から壁に向かって値が伝播するようにグラフを接続する必要があります。そうしないと、周期境界付近の壁距離が過大評価されてしまいます。
- マルチマテリアル: ローターとステーターの解析では、相対運動する両方の壁集合を に含めます。タービンのチップクリアランスのように距離が急激に変化する場所では、十分な格子解像度が必要です。
- P-Poisson 方程式による代替: () を陰的に解く方法は、Fast Marching より低速ですが、並列化が容易です。大規模な並列ソルバーでは、こちらが好まれることもあります。
- 局所的な負の誤差: 数値スキームの離散化によっては、非常に薄い境界層で が負の値になることがあります。常に でクランプしてください。
毎ステップ全域を行進させるか、P-Poisson に渡すか#
動くローター格子で Fast Marching を毎ステップ全セルに回すと、10M セルのヒープ演算が時間前進のたびに繰り返されます。変形の大きい帯だけを更新しないと、 の利得がステップ数だけ再び掛けられます。
並列が先なら 側を選びます。 は 4 から 10 の間です。反復の速さは Jacobi から BiCGSTAB へ渡る線形ソルバー が決めます。行進の因果性を捨てられるときだけ、この入れ替えをします。
参考文献
- J. A. Sethian, A. Vladimirsky, "Fast methods for the Eikonal and related Hamilton–Jacobi equations on unstructured meshes", PNAS 97(11), 5699–5703, 2000.
- P. Tucker, "Assessment of geometric multilevel convergence robustness and a wall distance method for flows with multiple internal boundaries", Applied Mathematical Modelling 22, 1998.
関連記事
役に立ったらシェアしてください。