import os 
import xlrd
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc

## Données : 


RT =  6356

## On charge les fichiers tex et xls


fname=r'C:\Users\bapt-\Mon Drive\PCSI_24\Mecanique\7_forces_centrales\Simulations\UCS-Satellite-Database-1-1-2022.xls' # a changer 



plt.ioff()
xl_workbook = xlrd.open_workbook(fname) # on ouvre le fichier xls
xl_sheet = xl_workbook.sheet_by_index(0) # on utilise la première feuille de calcul


row = xl_sheet.col(11)  

rm = [] 

for idx, cell_obj in enumerate(row):
    rm.append(float(cell_obj.value))


row = xl_sheet.col(12)  

rM = [] 

for idx, cell_obj in enumerate(row):
    rM.append(float(cell_obj.value))


row = xl_sheet.col(15)  

T=[] 

for idx, cell_obj in enumerate(row):
    T.append(float(cell_obj.value))
    


rm = np.array(rm)
rM = np.array(rM)
T = np.array(T)

T = T*60

a = (2*RT+rm+rM)/2*1000


rapport_attendu = 4*np.pi**2/(6.674*10**-11 * 5.927*10**24)
rapport = T**2/a**3/rapport_attendu






rapport = rapport[rapport<1.5]
plt.close('all')
plt.figure('', figsize = [10, 10] )
plt.axes([0.1, 0.2, 0.85, 0.75])
plt.hist(rapport, bins = 400)
plt.xlim([0.9,1.1 ])
plt.xlabel(r'$\frac{T^2}{a^3}\frac{\mathcal{G}M_T}{4\pi^2}~{\rm (s^2.m^{-3})}$', fontsize = 20)
plt.ylabel(r'nombre de satellites', fontsize = 20)
plt.show()

rapport_moyen = np.mean(T**2/a**3)
ecart_type = np.std(T**2/a**3)/np.sqrt(len(rapport))

EN = np.abs(rapport_moyen-rapport_attendu)/ecart_type