Move main.py to scripts #2

gnss-lab · Aug 15, 2023 · dd71032 · dd71032
1 parent ffd89ab
commit dd71032
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 3 deletions.
diff --git a/multiprocess_sphere_animation/data_processing.py b/multiprocess_sphere_animation/data_processing.py
@@ -1,5 +1,4 @@
 import h5py
-import pandas as pd
 import numpy as np
 
 def get_all_data(file_path):

diff --git a/sphere_animation/main.py → scripts/animate_sphere.py b/sphere_animation/main.py → scripts/animate_sphere.py
@@ -1,4 +1,4 @@
-from .animation import animate_sphere
+from animate_sphere.animation import animate_sphere
 
 if __name__ == '__main__':
     file_path = 'files/...'

diff --git a/multiprocess_sphere_animation/main.py → scripts/multiprocess_animate_sphere.py b/multiprocess_sphere_animation/main.py → scripts/multiprocess_animate_sphere.py
@@ -5,7 +5,7 @@
 
     start_time = time.time()
 
-    file_path = 'files/roti_2015_254_-90_90_N_-180_180_E_3a01.h5'
+    file_path = 'files/...'
     central_longitude = 37.6156  # Долгота
     central_latitude = 55.7522  # Широта
 

diff --git a/scripts/plot_sphere.py b/scripts/plot_sphere.py
@@ -0,0 +1,15 @@
+import pandas as pd
+from matplotlib import pyplot as plt
+
+from sphere_animation.animation import plot_sphere
+from sphere_animation.utils import get_all_data
+
+if __name__ == '__main__':
+    file_path = 'files/...'
+    central_longitude = 104.280606  # Долгота
+    central_latitude = 52.289588  # Широта
+
+    data = get_all_data(file_path)
+    all_data = [(group, pd.DataFrame(dataset_data)) for group, dataset_data in data.items() if '/data/' in group]
+    plot_sphere(central_longitude, central_latitude, all_data[0][1])
+    plt.show()