from video import Video
from sim import Sim
if __name__ == '__main__':
# video_table = VideoTable("test_data/check0.png")
# print(video_table.field_corners)
video_file = "test_data/game1/clip1.mp4"
# video_file = "test_data/game1/clip2.mp4"
# video_file = "test_data/game1/clip3.mp4"
# video_file = "test_data/game2/clip4.mp4"
my_video1 = Video(video_file, resize=True)
my_video1.real_time_tracking()
my_video1.draw_simple_trajectory()
# print(my_video1.ball_tracking_rec_complete)
my_sim = Sim()
H = projection.create_homography(my_video1.table.table_corners,
my_sim.sim_table.table_corners)
print(H)
sim_tracking_ball_dic = projection.gen_tracking_ball_dic_from_video(
my_video1.ball_tracking_rec_complete, H)
# print(sim_tracking_ball_dic)
my_sim.set_tracking_ball_dic(sim_tracking_ball_dic)
my_sim.draw_simple_trajectory()
my_sim.animation()
