make_graph.py

from pathlib import Path
import cv2
import pandas as pd

# 画像を読み込む
base_path = Path.home() / "network" / "src" / "tsukuba_network.png"
csv_path = Path.home() / "network" / "src" / "node_positions.csv"  # 保存先

node_number = 0  # ノード番号のカウンタ

if base_path.exists():
    print("画像ファイルは存在します。")
else:
    print("Error: 指定したパスに画像ファイルが存在しません。")

image = cv2.imread(str(base_path), cv2.IMREAD_COLOR)
print("画像を読み込みました")

# クリックした座標を保存するリスト
points = []

def get_coordinates(event, x, y, flags, param):
    global node_number  # 関数内で変更するため global 宣言
    if event == cv2.EVENT_LBUTTONDOWN:  # 左クリック時
        node_number += 1
        print(f"({node_number}, {x}, {y})")
        points.append((node_number, x, y))  # ノード番号も記録

# ウィンドウを開き、マウスイベントを設定
cv2.namedWindow("Image")
cv2.setMouseCallback("Image", get_coordinates)

while True:
    cv2.imshow("Image", image)
    key = cv2.waitKey(1) & 0xFF
    if key == 27 or key == ord('q'):  # ESC または 'q' で終了
        break

cv2.destroyAllWindows()

# pandas で CSV に保存
df = pd.DataFrame(points, columns=["Node", "X", "Y"])
df.to_csv(csv_path, index=False, encoding="utf-8")

print(f"座標データを {csv_path} に保存しました。")