import argparse
from EndoCalibrator import EndoCalibrator

parser = argparse.ArgumentParser()
parser.add_argument("--mode", type=str, required=True, choices=["calibrate"], help="処理モード")
parser.add_argument("--imgs_dir", type=str, required=True, help="画像が入ってるディレクトリ")
parser.add_argument("--output_dir", type=str, default="./temp", help="出力先")

# カメラキャリブレーション用
parser.add_argument("--square_size", type=float, default=20.0, help='正方形の一辺のサイズ[mm]')
parser.add_argument("--pattern_size_col", type=int, default=9, help="チェスの列数")
parser.add_argument("--pattern_size_row", type=int, default=6, help="チェスの行数")
parser.add_argument("--chess_img_extend", type=str, default=".JPG", help="チェス画像の拡張子")

args = parser.parse_args()

if __name__ == '__main__':
    mode = args.mode

    if mode == 'calibrate':
        editer = EndoCalibrator(args)
        editer.calibrate()