import numpy as np

from ellipsoid_fit import ellipsoid_fit as ellipsoid_fit, data_regularize

if __name__ == '__main__':

data = np.loadtxt("mag_out.txt")

# data2 = data_regularize(data)

center, evecs, radii, v = ellipsoid_fit(data)

a, b, c = radii

r = (a * b * c) ** (1. / 3.)

D = np.array([[r/a, 0., 0.], [0., r/b, 0.], [0., 0., r/c]])

transformation = evecs.dot(D).dot(evecs.T)

print('')

print('center: ', center)

print('radii: ', radii)

print('evecs: ', evecs)

print('transformation:')

print(transformation)

print('Coefficients:')

print(v)

np.savetxt('magcal_ellipsoid.txt', np.vstack((center.T, transformation)))