linarphy/naroo_reader
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

optimize consecutive & clean unused var

Browse source
This commit is contained in:
linarphy 2023-06-16 15:06:36 +02:00
parent a6781588bb
commit 9b2d879dab
No known key found for this signature in database
GPG key ID: 3D4AAAC3AD16E79C
1 changed files with 1 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
utils.py
View file

@ -2,8 +2,6 @@ from astropy.io.fits import open
import cv2
import numpy as np
global_i = 0
def load( filename ):
"""
retrieve data from naroo fits
@ -96,8 +94,6 @@ def fill( data , point , tolerance , limit = 100000 ):
"""
give the coordinate of all points that fill the area with the given tolerance
"""
global global_i
global_i += 1
if not isinstance( data , np.ndarray ) and not isinstance( data , list ):
raise ValueError( 'data must be a list, ' + type( data ) + ' given' )
if not isinstance( point , np.ndarray ) and not isinstance( point , tuple ) and not isinstance( point , list ):
@ -185,7 +181,7 @@ def consecutive( list_ ):
index = last_consecutive( list_ )
if index == len( list_ ) - 1:
return [ list_ ]
return consecutive( list_[ : index + 1 ] ) + consecutive( list_[ index + 1 : ] ) # happy recursion \o/
return [ list_[ : index + 1 ] ] + consecutive( list_[ index + 1 : ] ) # happy recursion \o/
def last_consecutive( list_ ):
"""
return the last index of the first consecutive list