def __init__(self): # load centerline_pth = Path("1_centerline.json") effective_pth = Path("2_effective.json") if Path("0_skeleton.json").is_file(): self.r_skeleton = Path("0_skeleton.json").load() self.plot_segment(self.r_skeleton, Path("0_skeleton.plot.json")) # apply first pass self.r_centerline = self.pass_1(self.r_skeleton) Path("1_centerline.json").save(self.r_centerline) self.plot_segment(self.r_centerline, Path("1_centerline.plot.json")) self.plot_corridor(Path("3_corridor.plot.json")) if centerline_pth.is_file(): self.r_effective = self.pass_2(self.r_centerline) Path("2_effective.json").save(self.r_centerline) else: self.r_effective = effective_pth.load() self.plot_segment(self.r_effective, Path("2_effective.plot.json")) self.to_wsk_qnd_point(self.r_effective)
#!/usr/bin/env python3 import sys import numpy as np from cc_pathlib import Path u = np.mgrid[0:5, 0:5] u = np.random.random((1440, 5)).astype(np.float32) pth = Path("test.npy.br") pth.save(u) v = pth.load() if (u == v).all(): sys.exit(0) else: print("ERROR") sys.exit(1)