File size: 1,717 Bytes
50762c1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | """procmap：CPU 上的求交 / 地面纹理小测（真实 clip 的整套生成见 procmap.py --default4 的 report.json）。"""
import numpy as np
import pytest
torch = pytest.importorskip("torch")
import procmap as PM
def _prims():
return [PM.box([0, 0], 200, 100, 150, 0.3, 0, 3), PM.ell([300, 0, 50], [80, 60, 40], 0.7, 5),
PM.cyl([0, 300], 40, 0, 220, 6), PM.gable([0, -300], 400, 200, 0.2, 300, 100, 30, 3),
PM.prism(PM._ngon([600, 600], 100, 8), 0, 250, 4, taper=0.4), PM.log(type("W", (), dict(
rng=np.random.default_rng(0), z0=0.0))(), np.array([-400.0, 400.0]))[0][0]]
def test_cast_matches_sdf_marching():
prims = _prims()
rng = np.random.default_rng(0)
ro = rng.uniform(-700, 700, (300, 3)); ro[:, 2] = 900
rd = rng.normal(0, 1, (300, 3)); rd[:, 2] = -np.abs(rd[:, 2]) - 0.5
rd /= np.linalg.norm(rd, axis=1, keepdims=True)
t, k = PM.Caster(prims, "cpu").cast(ro, rd)
ts = np.linspace(0, 3000, 30001)
for i in range(len(ro)):
P = ro[i] + ts[:, None] * rd[i]
s = np.min([PM.sdf_lower(p, P) for p in prims], 0)
tr = ts[np.argmax(s <= 0)] if (s <= 0).any() else np.inf
assert (np.isinf(tr) and k[i] == -1) or abs(tr - t[i]) < 0.2, (i, tr, t[i])
def test_ground_layers_paint_in_order():
g = PM.Ground()
g.layers.append(("seg", PM.PAVING, np.array([[0, 0, 1000, 0.0]]), np.array([100.0])))
g.layers.append(("disk", PM.TERRAIN, np.array([500, 0, 50.0])))
g.layers.append(("poly", PM.PAVING, PM._rect([0, 500], 200, 200, 0.0)))
xy = np.array([[200, 50], [200, 150], [500, 0], [0, 550], [0, 650]], float)
assert PM.Ground.classify(g, xy, "cpu").tolist() == [2, 1, 1, 2, 1]
|