#!/usr/bin/env python3
"""
RSK World
Founder: Molla Samser
Designer & Tester: Rima Khatun
Email: help@rskworld.in
Phone: +91 93305 39277
Address: Nutanhat, Mongolkote, Purba Burdwan, West Bengal, India, 713147
Website: https://rskworld.in/

Example script for player tracking using the Sports Analysis Dataset
"""

import cv2
import json
import numpy as np
from pathlib import Path

def load_tracking_data(json_path):
    """
    Load player tracking data from JSON file
    
    Args:
        json_path: Path to tracking JSON file
        
    Returns:
        Dictionary containing tracking data
    """
    with open(json_path, 'r') as f:
        return json.load(f)

def track_players(video_path, tracking_data):
    """
    Visualize player tracking on video
    
    Args:
        video_path: Path to video file
        tracking_data: Dictionary with tracking information
    """
    cap = cv2.VideoCapture(video_path)
    
    if not cap.isOpened():
        print(f"Error: Could not open video {video_path}")
        return
    
    frame_count = 0
    
    while True:
        ret, frame = cap.read()
        if not ret:
            break
        
        # Get tracking data for current frame
        current_time = frame_count / 30.0  # Assuming 30 FPS
        
        # Draw player positions
        for player in tracking_data.get('players', []):
            if abs(player['timestamp'] - current_time) < 0.1:
                x, y = player['position']['x'], player['position']['y']
                cv2.circle(frame, (int(x), int(y)), 10, (0, 255, 0), -1)
                cv2.putText(frame, f"P{player['player_id']}", 
                           (int(x)+15, int(y)), 
                           cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 2)
        
        cv2.imshow('Player Tracking', frame)
        
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
        
        frame_count += 1
    
    cap.release()
    cv2.destroyAllWindows()

def main():
    """
    Main function to run player tracking example
    """
    # Example paths (adjust based on your dataset structure)
    video_path = '../data/videos/game_001.mp4'
    tracking_path = '../data/annotations/tracking_001.json'
    
    print("Loading tracking data...")
    tracking_data = load_tracking_data(tracking_path)
    
    print("Starting player tracking visualization...")
    track_players(video_path, tracking_data)
    
    print("Tracking complete!")

if __name__ == '__main__':
    main()