openstarlab-spaceEval 0.0.1__tar.gz

openstarlab_spaceeval-0.0.1/LICENSE.md

@@ -0,0 +1,23 @@
+MIT License
+
+Copyright (c) 2021 Friends-of-Tracking-Data-FoTD
+Copyright (c) 2022 Masakiyo Teranishi and Keisuke Fujii
+Copyright (c) 2025 Titouan Jeannot, Keisuke Fujii, OpenSTARLab
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

openstarlab_spaceeval-0.0.1/PKG-INFO

@@ -0,0 +1,41 @@
+Metadata-Version: 2.1
+Name: openstarlab-spaceEval
+Version: 0.0.1
+Summary: openstarlab space eval package
+Author-email: Calvin Yeung <yeung.chikwong@g.sp.m.is.nagoya-u.ac.jp>
+License: Apache License 2.0
+Project-URL: Homepage, https://github.com/open-starlab/
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE.md
+Requires-Dist: pandas
+Requires-Dist: numpy
+Requires-Dist: matplotlib
+Requires-Dist: scipy
+Requires-Dist: tqdm
+Requires-Dist: opencv-python
+
+# OpenSTARLab SpaceEval package
+
+## Introduction
+The OpenSTARLab SpaceEval package is designed to provide simple and efficient way to visualize and estimate space evaluation for different sports. This package supports the data preprocessed by the OpenSTARLab PreProcessing package.
+
+
+This package is continuously evolving to support future OpenSTARLab projects. If you have any suggestions or encounter any bugs, please feel free to open an issue.
+
+
+## Installation
+
+- To install this package via PyPI
+```
+pip install openstarlab-spaceEval
+```
+- To install manually
+```
+git clone git@github.com:open-starlab/spaceEval.git
+cd ./spaceEval
+pip install -e .
+```

openstarlab_spaceeval-0.0.1/README.md

@@ -0,0 +1,21 @@
+# OpenSTARLab SpaceEval package
+
+## Introduction
+The OpenSTARLab SpaceEval package is designed to provide simple and efficient way to visualize and estimate space evaluation for different sports. This package supports the data preprocessed by the OpenSTARLab PreProcessing package.
+
+
+This package is continuously evolving to support future OpenSTARLab projects. If you have any suggestions or encounter any bugs, please feel free to open an issue.
+
+
+## Installation
+
+- To install this package via PyPI
+```
+pip install openstarlab-spaceEval
+```
+- To install manually
+```
+git clone git@github.com:open-starlab/spaceEval.git
+cd ./spaceEval
+pip install -e .
+```

openstarlab_spaceeval-0.0.1/openstarlab_spaceEval.egg-info/PKG-INFO

@@ -0,0 +1,41 @@
+Metadata-Version: 2.1
+Name: openstarlab-spaceEval
+Version: 0.0.1
+Summary: openstarlab space eval package
+Author-email: Calvin Yeung <yeung.chikwong@g.sp.m.is.nagoya-u.ac.jp>
+License: Apache License 2.0
+Project-URL: Homepage, https://github.com/open-starlab/
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE.md
+Requires-Dist: pandas
+Requires-Dist: numpy
+Requires-Dist: matplotlib
+Requires-Dist: scipy
+Requires-Dist: tqdm
+Requires-Dist: opencv-python
+
+# OpenSTARLab SpaceEval package
+
+## Introduction
+The OpenSTARLab SpaceEval package is designed to provide simple and efficient way to visualize and estimate space evaluation for different sports. This package supports the data preprocessed by the OpenSTARLab PreProcessing package.
+
+
+This package is continuously evolving to support future OpenSTARLab projects. If you have any suggestions or encounter any bugs, please feel free to open an issue.
+
+
+## Installation
+
+- To install this package via PyPI
+```
+pip install openstarlab-spaceEval
+```
+- To install manually
+```
+git clone git@github.com:open-starlab/spaceEval.git
+cd ./spaceEval
+pip install -e .
+```

openstarlab_spaceeval-0.0.1/openstarlab_spaceEval.egg-info/SOURCES.txt

@@ -0,0 +1,23 @@
+LICENSE.md
+README.md
+pyproject.toml
+openstarlab_spaceEval.egg-info/PKG-INFO
+openstarlab_spaceEval.egg-info/SOURCES.txt
+openstarlab_spaceEval.egg-info/dependency_links.txt
+openstarlab_spaceEval.egg-info/requires.txt
+openstarlab_spaceEval.egg-info/top_level.txt
+spaceeval/__init__.py
+spaceeval/sports/__init__.py
+spaceeval/sports/main_class.py
+spaceeval/sports/basketball/__init__.py
+spaceeval/sports/basketball/application/__init__.py
+spaceeval/sports/basketball/application/heatmap.py
+spaceeval/sports/basketball/main_class_basketball/__init__.py
+spaceeval/sports/basketball/main_class_basketball/main.py
+spaceeval/sports/basketball/model_parameter/__init__.py
+spaceeval/sports/basketball/models/BIMOS.py
+spaceeval/sports/basketball/models/BMOS.py
+spaceeval/sports/basketball/models/__init__.py
+spaceeval/sports/basketball/utils/SportVU_IO.py
+spaceeval/sports/basketball/utils/__init__.py
+spaceeval/sports/basketball/utils/get_residual_param.py

openstarlab_spaceeval-0.0.1/openstarlab_spaceEval.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

openstarlab_spaceeval-0.0.1/openstarlab_spaceEval.egg-info/requires.txt

@@ -0,0 +1,6 @@
+pandas
+numpy
+matplotlib
+scipy
+tqdm
+opencv-python

openstarlab_spaceeval-0.0.1/openstarlab_spaceEval.egg-info/top_level.txt

@@ -0,0 +1,2 @@
+dist
+spaceeval

openstarlab_spaceeval-0.0.1/pyproject.toml

@@ -0,0 +1,39 @@
+# event/pyproject.toml
+
+[build-system]
+requires = ["setuptools>=42", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "openstarlab-spaceEval"
+version = "0.0.1"
+description = "openstarlab space eval package"
+readme = "README.md"
+requires-python = ">=3.8"
+authors = [
+    { name="Calvin Yeung", email="yeung.chikwong@g.sp.m.is.nagoya-u.ac.jp" },
+]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: Apache Software License",
+    "Operating System :: OS Independent",
+]
+dependencies = [
+    "pandas",
+    "numpy",
+    "matplotlib",
+    "scipy",
+    "tqdm",
+    "opencv-python"
+]
+
+license = { text = "Apache License 2.0" }
+
+[tool.setuptools]
+include-package-data = true
+
+[tool.setuptools.packages.find]
+where = ["."]
+
+[project.urls]
+Homepage = "https://github.com/open-starlab/"

openstarlab_spaceeval-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

openstarlab_spaceeval-0.0.1/spaceeval/__init__.py

@@ -0,0 +1,4 @@
+from .sports.main_class import Space_Model
+
+__all__ = ['Space_Model']
+#oohh j'ai changé le nom

openstarlab_spaceeval-0.0.1/spaceeval/sports/basketball/application/heatmap.py

@@ -0,0 +1,334 @@
+import os
+import numpy as np
+from tqdm import tqdm
+import matplotlib.pyplot as plt
+import importlib.resources as pkg_resources
+from ...basketball import utils
+from PIL import Image
+from ..models.BIMOS import BIMOS
+from ..models.BMOS import BMOS
+import datetime
+from ..utils import SportVU_IO as sio
+import glob
+import cv2
+
+COURT_SIZE = (28, 15)  # Court dimensions in meters
+EVENT_LABEL = 1
+EVENT_LABELS = {
+    'nonevent': 0,
+    'pass': 1,
+    'catch and pass': 2,
+    'handoff catch and pass': 3,
+    'catch': 4,
+    'handoff pass': 5,
+    'handoff catch and handoff pass': 6,
+    'catch and handoff pass': 7,
+    'handoff catch': 8,
+    '2 point shot': 9,
+    '3 point shot': 10,
+    'turnover': 11
+}
+
+def plotCourt():
+    with pkg_resources.open_binary(utils, 'nba_court.png') as file:
+        img = Image.open(file)
+        img = img.copy()
+
+    plt.imshow(img, extent=[0, COURT_SIZE[0], 0, COURT_SIZE[1]], zorder=0)
+    plt.xlim(0, COURT_SIZE[0])
+    plt.ylim(0, COURT_SIZE[1])
+
+def plot_heat_map_frame(save_path_folder, attValue, data,
+                  include_player_velocities = True, BID=True, colorbar = True, title=True
+                  ,field_dimen = (COURT_SIZE[0],COURT_SIZE[1]), colormap = "Reds"):
+
+    att_x = []
+    att_y = []
+    dim_att = []
+    for a in range(5):  # 5 attaquants
+        x = data[f'x_att{a}'].values[0]
+        y = data[f'y_att{a}'].values[0]
+        att_x.append(x)
+        att_y.append(y)
+    
+    dim_att = [np.array(att_x), np.array(att_y)]
+
+    def_x = []
+    def_y = []
+    dim_def = []
+    for b in range(5):
+        x = data[f'x_def{b}'].values[0]
+        y = data[f'y_def{b}'].values[0]
+        def_x.append(x)
+        def_y.append(y)
+
+    dim_def = [np.array(def_x), np.array(def_y)]
+    
+    dim_ball = []
+    x = data['x_ball'].values[0]
+    y = data['y_ball'].values[0]
+    dim_ball = np.array([x, y])
+    
+    fig, ax = plt.subplots(figsize = (6,5))
+    fig.subplots_adjust(left = 0.01, bottom = 0.08, right = 0.99, top=0.95)
+    plt.text(14,-2,'[m]', ha = 'center')
+
+    ax.imshow(attValue, cmap = colormap, extent =(0, field_dimen[0], 0, field_dimen[1]), alpha=0.9)
+    if colorbar:
+        plt.colorbar(ax.imshow(attValue, cmap=colormap, extent=(0, field_dimen[0], 0, field_dimen[1]), alpha=0.9))
+
+    plotCourt()
+
+    ax.scatter(*dim_att, s = 100, edgecolor = 'r', c = 'white')
+    ax.scatter(*dim_def, s = 100, edgecolor = 'b', c = 'white')
+    ax.scatter(*dim_ball, s = 100, edgecolor = 'black', c = 'white')
+    ax.scatter(*dim_ball, s = 30, c = 'black')
+
+    if BID and data['ball_holder'].values[0] > 0 and data['ball_holder'].values[0] < 9:
+        bid_idx = int(data['ball_holder'].values[0] - 1)
+        ax.scatter(
+            data[f'x_att{bid_idx}'].values[0], 
+            data[f'y_att{bid_idx}'].values[0], 
+            s=17, facecolors='none', edgecolors='black'
+            )
+    
+    if include_player_velocities:
+        for c in range(5):
+            plt.quiver(
+                data[f'x_att{c}'].values[0], 
+                data[f'y_att{c}'].values[0], 
+                data[f'vx_att{c}'].values[0],
+                data[f'vy_att{c}'].values[0], 
+                angles='xy', scale_units='xy', scale=1, color='black'
+                )
+            
+        for d in range(5):
+            plt.quiver(
+                data[f'x_def{d}'].values[0], 
+                data[f'y_def{d}'].values[0], 
+                data[f'vx_def{d}'].values[0],
+                data[f'vy_def{d}'].values[0], 
+                angles='xy', scale_units='xy', scale=1, color='black'
+                )
+    
+    if title:
+        plt.title(f"Game {data['gameID'].values[0]} - Attack {data['attackid'].values[0]} - Frame {data['f_id'].values[0]}")
+
+    
+    if 'gameID' in data.columns:
+        game_id = data['gameID'].values[0]
+    else:
+        game_id = 'unknown'
+    
+    if 'attackid' in data.columns:
+        attackid = data['attackid'].values[0]
+    else:
+        attackid = 'unknown'
+    
+    if 'f_id' in data.columns:
+        frame_id = data['f_id'].values[0]
+    else:
+        frame_id = 'unknown'
+    
+    filename = f"heatmap_game_{game_id}_attack_{attackid}_frame_{frame_id}.png"
+    save_path = f"{save_path_folder}/{filename}"
+    fig.savefig(save_path, dpi=300, bbox_inches='tight')
+
+    plt.close(fig)
+
+
+def movie_from_images(image_files, output_file, frames_with_heatmap, fps=8, heatmap_duration_multiplier=10):
+    frame = cv2.imread(image_files[0])
+    height, width, _ = frame.shape
+
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    out = cv2.VideoWriter(output_file, fourcc, fps, (width, height))
+
+    frame_indices = {}
+    for i, file in enumerate(image_files):
+        frame_number = int(file.split('_')[-1].split('.')[0])
+        frame_indices[file] = frame_number
+
+    for file in image_files:
+        img = cv2.imread(file)
+        frame_number = frame_indices[file]
+        
+        if frame_number in frames_with_heatmap:
+            for _ in range(heatmap_duration_multiplier):
+                out.write(img)
+        else:
+            out.write(img)
+
+    out.release()
+
+    # Clean up by deleting image files after creating the movie
+    print("Video created at:", output_file)
+
+    if os.path.exists("figures"):
+        for file in glob.glob("figures/*"):
+            os.remove(file)
+        os.rmdir("figures")
+        print("Figures directory removed")
+
+
+def plot_heat_map_sequence(model, data, save_path_folder, heatmap=True, 
+    EVENT=True, JERSEY=True, BID=False, axis=False, title=True, field_dimen=(COURT_SIZE[0],COURT_SIZE[1])):
+    """
+    Plots animation for a specific scene.
+    """
+    
+    def extract_date_info(gamename):
+        parts = gamename.split('_')
+
+        day = int(parts[1])
+        month = int(parts[0])
+        year = int(parts[2])
+        
+        suffix = 'th' if 11 <= day <= 13 else {1: 'st', 2: 'nd', 3: 'rd'}.get(day % 10, 'th')
+        day_formatted = f"{day}{suffix}"
+        
+        month_name = datetime.date(1900, month, 1).strftime('%b')
+        
+        return day_formatted, month_name, str(year)
+    
+    game_id = data['gameID'].values[0]
+    s_id = data['attackid'].values[0]
+
+    # Obtain date info
+    gamename = data['gamename'].values[0]
+    date, month, year = extract_date_info(gamename)
+
+    # Obtain team info
+    team_id_O = data['team_O'].values[0]
+    team_id_D = data['team_D'].values[0]
+    team_name_O = sio.load_team_name(team_id_O)
+    team_name_D = sio.load_team_name(team_id_D)
+
+    frames_with_heatmap = []
+    for i in tqdm(range(len(data))):
+        row_cumput = data.iloc[[i]].copy()
+
+        att_x = []
+        att_y = []
+        dim_att = []
+        for j in range(5):  # 5 attaquants
+            x = row_cumput[f'x_att{j}'].values[0]
+            y = row_cumput[f'y_att{j}'].values[0]
+            att_x.append(x)
+            att_y.append(y)
+
+        dim_att = [np.array(att_x), np.array(att_y)]
+
+        def_x = []
+        def_y = []
+        dim_def = []
+        for j in range(5):
+            x = row_cumput[f'x_def{j}'].values[0]
+            y = row_cumput[f'y_def{j}'].values[0]
+            def_x.append(x)
+            def_y.append(y)
+
+        dim_def = [np.array(def_x), np.array(def_y)]
+        
+        dim_ball = []
+        x = row_cumput['x_ball'].values[0]
+        y = row_cumput['y_ball'].values[0]
+        dim_ball = np.array([x, y])
+
+        frame_info = {
+            'dim_att': dim_att,
+            'dim_def': dim_def,
+            'dim_ball': dim_ball
+        }
+
+        if row_cumput['event_label'].values[0] not in [0, 4, 8]:
+            frames_with_heatmap.append(i)
+            if model == "BIMOS":
+                frame_info['attValue'] = BIMOS(row_cumput).values
+            
+            else:
+                frame_info['attValue'] = BMOS(row_cumput).values
+        else:
+            frame_info['attValue'] = [[0] * 28 for _ in range(15)]
+
+        if EVENT:
+            frame_info['event_label'] = row_cumput['event_label'].values[0]
+
+        if JERSEY:
+            frame_info['jersey_number'] = row_cumput.filter(regex='^jersey_', axis=1).iloc[0].tolist()
+
+        # Create the figure and set up basic layout
+        fig, ax = plt.subplots(figsize=(6, 5))
+        fig.subplots_adjust(left=0.01, bottom=0.08, right=0.99, top=0.95)
+        plt.text(14, -2, '[m]', ha='center')
+        
+        # Plot heatmap if enabled
+        if heatmap:
+            ax.imshow(frame_info['attValue'], cmap='Reds', vmin=0., vmax=1., 
+                 extent=(0, field_dimen[0], 0, field_dimen[1]), alpha=0.9)
+        
+        # Plot players and ball
+        ax.scatter(*frame_info['dim_att'], s=100, edgecolor='r', c="white")
+        ax.scatter(*frame_info['dim_def'], s=100, edgecolor='b', c="white")
+        ax.scatter(*frame_info['dim_ball'], s=30, c="black")
+
+        # Highlight ball possessor if enabled
+        if BID and row_cumput['ball_holder'].values[0] > 0 and row_cumput['ball_holder'].values[0] < 9:
+            bid_idx = int(row_cumput['ball_holder'].values[0] - 1)
+            ax.scatter(
+            row_cumput[f'x_att{bid_idx}'].values[0], 
+            row_cumput[f'y_att{bid_idx}'].values[0],
+            s=60, facecolors='none', edgecolors='black'
+            )
+
+        # Add event labels if enabled
+        if EVENT and 'event_label' in frame_info:
+            ax.text(*frame_info['dim_ball'], f"{frame_info['event_label']}", fontsize=8)
+
+        # Add jersey numbers if enabled
+        if JERSEY and 'jersey_number' in frame_info:
+            for j in range(10):
+                jersey_number = int(frame_info["jersey_number"][j])
+                if j < 5:  # Attacking team
+                    x, y = frame_info['dim_att'][0][j], frame_info['dim_att'][1][j]
+                else:  # Defending team
+                    x, y = frame_info['dim_def'][0][j-5], frame_info['dim_def'][1][j-5]
+                
+                ax.text(x, y, f'{jersey_number}', 
+                       fontsize=8, ha='center', va='center')
+
+    
+        fid_str = str(i).zfill(3)
+        # Add title based on configuration
+        if title:
+            # Detailed title with team names and date
+            ax.set_title('')
+            ax.text(0.08, 1.025, team_name_O, color='red', fontsize=12, ha='center', transform=ax.transAxes)
+            ax.text(0.17, 1.025, 'vs.', color='black', fontsize=12, ha='center', transform=ax.transAxes)
+            ax.text(0.25, 1.025, team_name_D, color='blue', fontsize=12, ha='center', transform=ax.transAxes)
+            ax.text(0.65, 1.025, f"{date} {month}. {year} - Frame {fid_str}", 
+               color='black', fontsize=12, ha='center', transform=ax.transAxes)
+        else:
+            # Simple title with game info and event type
+            ax.set_title(f'Game {game_id} - Attack {s_id} - Frame {fid_str}')
+
+        # Hide axis if not needed
+        if not axis:
+            plt.xticks([])
+            plt.yticks([])
+
+        # Save the frame and close the plot
+# Add the court background
+        plotCourt()
+
+        if not os.path.exists("figures"):
+            os.makedirs("figures")
+        plt.savefig(f"figures/frame_{game_id}_{s_id}_{fid_str}.png")
+        plt.close()
+
+    image_files = sorted(glob.glob(f"figures/frame_{game_id}_{s_id}_*.png"))
+    print(image_files)
+    video_name = f"{save_path_folder}/space_evaluation_{game_id}_{s_id}.mp4"
+    movie_from_images(image_files=image_files, output_file=video_name, frames_with_heatmap = frames_with_heatmap)
+
+    

openstarlab_spaceeval-0.0.1/spaceeval/sports/basketball/main_class_basketball/main.py

@@ -0,0 +1,40 @@
+from ..models.BIMOS import BIMOS
+from ..models.BMOS import BMOS
+import os
+import gdown
+import zipfile
+
+from ..application.heatmap import plot_heat_map_frame, plot_heat_map_sequence
+
+class space_model_basketball:
+    def __init__(self, model_name):
+        """
+        Initializes the class with model name and ensures the required file is downloaded.
+
+        :param model_name: str, the name of the model (e.g., "BIMOS", "BMOS")
+        :param file_id: str, the Google Drive file ID
+        :param dest_path: str, the local path where the file will be saved
+        """
+        self.model_name = model_name
+
+    def plot_heat_map_frame(self, save_path_folder, data, *args, **kwargs):
+
+        if self.model_name == "BIMOS":
+            attValues = BIMOS(data).values
+        
+        if self.model_name == "BMOS":
+            attValues = BMOS(data).values
+
+        plot_heat_map_frame(save_path_folder, attValues, data, *args, **kwargs)
+        
+
+    def plot_heat_map_sequence(self, data, save_path_folder,*args, **kwargs):
+
+        if self.model_name == "BIMOS":
+            model = "BIMOS"
+        
+        if self.model_name == "BMOS":
+            model = "BMOS"
+
+        plot_heat_map_sequence(model, data, save_path_folder,*args, **kwargs)
+