alu/tra-analysis
1
0
Fork 0
mirror of https://github.com/titanscouting/tra-analysis.git synced 2024-12-27 18:09:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct indentation

Browse Source 
Signed-off-by: Dev Singh <dev@devksingh.com>
This commit is contained in:
Dev Singh 2020-09-23 14:31:29 +00:00
parent 10586e752b
commit 26e6223ff4
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
analysis-master/tra_analysis/fits.py
View File

@ -27,12 +27,12 @@ class CircleFit:
# For more information on the LSC method, see: # For more information on the LSC method, see:
# http://www.dtcenter.org/sites/default/files/community-code/met/docs/write-ups/circle_fit.pdf # http://www.dtcenter.org/sites/default/files/community-code/met/docs/write-ups/circle_fit.pdf
def __init__(self, x, y, xy=None): def __init__(self, x, y, xy=None):
if type(x) == list: if type(x) == list:
x = np.array(x) x = np.array(x)
if type(y) == list: if type(y) == list:
y = np.array(y) y = np.array(y)
if type(xy) == list: if type(xy) == list:
xy = np.array(xy) xy = np.array(xy)
if data != None: if data != None:
self.coords = data self.coords = data
self.ournp = np #todo: implement cupy correctly self.ournp = np #todo: implement cupy correctly
@ -43,9 +43,9 @@ class CircleFit:
"""Returns distance between center and point""" """Returns distance between center and point"""
return self.ournp.sqrt((x-xc)**2 + (y-yc)**2) return self.ournp.sqrt((x-xc)**2 + (y-yc)**2)
def f(c, x, y): def f(c, x, y):
"""Returns distance between point and circle at c""" """Returns distance between point and circle at c"""
Ri = calc_R(x, y, *c) Ri = calc_R(x, y, *c)
return Ri - Ri.mean() return Ri - Ri.mean()
def LSC(self): def LSC(self):
"""Fits given data to a circle and returns the center, radius, and variance""" """Fits given data to a circle and returns the center, radius, and variance"""
x = coords[:, 0] x = coords[:, 0]