62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
import json
|
|
import sys
|
|
import os
|
|
|
|
# 경로 설정을 위해 현재 파일의 디렉토리를 sys.path에 추가
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
sys.path.append(current_dir)
|
|
|
|
from pid_geometric_extractor import PidGeometricExtractor
|
|
from pid_topology_builder import PidTopologyBuilder, analyze_impact
|
|
|
|
def run_pipeline():
|
|
# 1. 경로 설정 (현재 디렉토리 기준 상대 경로)
|
|
input_dxf = os.path.join(current_dir, "No-10_Plant_PID.dxf")
|
|
geo_json_path = os.path.join(current_dir, "shared_geo_data.json")
|
|
graph_json_path = os.path.join(current_dir, "pid_graph_topology.json")
|
|
|
|
print("--- Phase 1: Geometric Extraction ---")
|
|
try:
|
|
extractor = PidGeometricExtractor(input_dxf)
|
|
extractor.extract_and_save(geo_json_path)
|
|
print(f"Geometric data saved to {geo_json_path}")
|
|
except Exception as e:
|
|
print(f"Phase 1 failed: {e}")
|
|
return
|
|
|
|
print("\n--- Phase 2: Topology Modeling ---")
|
|
try:
|
|
with open(geo_json_path, 'r', encoding='utf-8') as f:
|
|
geometric_data = json.load(f)
|
|
|
|
# 테스트를 위해 all_extracted_tags는 빈 리스트로 전달
|
|
# config를 None으로 전달하여 topology_config.json 설정을 사용하도록 함
|
|
builder = PidTopologyBuilder(
|
|
geometric_data=geometric_data,
|
|
all_extracted_tags=[],
|
|
config=None
|
|
)
|
|
builder.build_graph()
|
|
|
|
# 위상 검증
|
|
validation = builder.validate_topology()
|
|
print(f"Topology Validation: {validation}")
|
|
|
|
# 그래프 저장
|
|
builder.save_graph(graph_json_path)
|
|
print(f"Graph topology saved to {graph_json_path}")
|
|
|
|
# 영향도 분석 테스트 (노드가 존재하는 경우)
|
|
if validation['node_count'] > 0:
|
|
sample_node = list(builder.G.nodes())[0]
|
|
impacted = analyze_impact(builder.G, sample_node)
|
|
print(f"Impact analysis for node {sample_node}: {impacted}")
|
|
|
|
except Exception as e:
|
|
print(f"Phase 2 failed: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
|
|
if __name__ == "__main__":
|
|
run_pipeline()
|