indymedia
/
epicyon
mirror of https://gitlab.com/bashrc2/epicyon
7
4
Fork 1
Code Issues 14 Releases Wiki Activity

Static analysis on all source files

main
Bob Mottram 2020-04-05 18:22:37 +01:00
parent a8a0465c01
commit 51db9230ae
2 changed files with 46 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
blurhash.py
View File

@ -8,8 +8,8 @@ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions: furnished to do so, subject to the following conditions:
* The above copyright notice and this permission notice shall be included in all * The above copyright notice and this permission notice shall be included
copies or substantial portions of the Software. in all copies or substantial portions of the Software.
* You and any organization you work for may not promote white supremacy, hate * You and any organization you work for may not promote white supremacy, hate
speech and homo- or transphobia - this license is void if you do. speech and homo- or transphobia - this license is void if you do.
@ -33,7 +33,9 @@ import math
# Alphabet for base 83 # Alphabet for base 83
alphabet = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz#$%*+,-.:;=?@[]^_{|}~" alphabet = \
"0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ" + \
"abcdefghijklmnopqrstuvwxyz#$%*+,-.:;=?@[]^_{|}~"
alphabet_values = dict(zip(alphabet, range(len(alphabet)))) alphabet_values = dict(zip(alphabet, range(len(alphabet))))
@ -55,10 +57,11 @@ def base83_encode(value, length):
if the specified length is too short. if the specified length is too short.
""" """
if int(value) // (83 ** (length)) != 0: if int(value) // (83 ** (length)) != 0:
raise ValueError("Specified length is too short to encode given value.") raise ValueError("Specified length is too short to " +
"encode given value.")
result = "" result = ""
for i in range(1, length+1): for i in range(1, length + 1):
digit = int(value) // (83 ** (length - i)) % 83 digit = int(value) // (83 ** (length - i)) % 83
result += alphabet[int(digit)] result += alphabet[int(digit)]
return result return result
@ -130,7 +133,7 @@ def blurhash_decode(blurhash, width, height, punch=1.0, linear=False):
size_x = (size_info % 9) + 1 size_x = (size_info % 9) + 1
quant_max_value = base83_decode(blurhash[1]) quant_max_value = base83_decode(blurhash[1])
real_max_value = (float(quant_max_value+1) / 166.0) * punch real_max_value = (float(quant_max_value + 1) / 166.0) * punch
# Make sure we at least have the right number of characters # Make sure we at least have the right number of characters
if len(blurhash) != 4 + 2 * size_x * size_y: if len(blurhash) != 4 + 2 * size_x * size_y:
@ -148,9 +151,12 @@ def blurhash_decode(blurhash, width, height, punch=1.0, linear=False):
for component in range(1, size_x * size_y): for component in range(1, size_x * size_y):
ac_value = base83_decode(blurhash[4+component*2:4+(component+1)*2]) ac_value = base83_decode(blurhash[4+component*2:4+(component+1)*2])
colours.append(( colours.append((
sign_pow((float(int(ac_value / (19 * 19))) - 9.0) / 9.0, 2.0) * real_max_value, sign_pow((float(int(ac_value / (19 * 19))) - 9.0)
sign_pow((float(int(ac_value / 19) % 19) - 9.0) / 9.0, 2.0) * real_max_value, / 9.0, 2.0) * real_max_value,
sign_pow((float(ac_value % 19) - 9.0) / 9.0, 2.0) * real_max_value sign_pow((float(int(ac_value / 19) % 19) - 9.0)
/ 9.0, 2.0) * real_max_value,
sign_pow((float(ac_value % 19) - 9.0)
/ 9.0, 2.0) * real_max_value
)) ))
# Return image RGB values, as a list of lists of lists, # Return image RGB values, as a list of lists of lists,
@ -163,13 +169,15 @@ def blurhash_decode(blurhash, width, height, punch=1.0, linear=False):
for j in range(size_y): for j in range(size_y):
for i in range(size_x): for i in range(size_x):
basis = math.cos(math.pi * float(x) * float(i) / float(width)) * \ basis = \
math.cos(math.pi * float(x) * float(i) /
float(width)) * \
math.cos(math.pi * float(y) * float(j) / float(height)) math.cos(math.pi * float(y) * float(j) / float(height))
colour = colours[i + j * size_x] colour = colours[i + j * size_x]
pixel[0] += colour[0] * basis pixel[0] += colour[0] * basis
pixel[1] += colour[1] * basis pixel[1] += colour[1] * basis
pixel[2] += colour[2] * basis pixel[2] += colour[2] * basis
if linear == False: if linear is False:
pixel_row.append([ pixel_row.append([
linear_to_srgb(pixel[0]), linear_to_srgb(pixel[0]),
linear_to_srgb(pixel[1]), linear_to_srgb(pixel[1]),
@ -183,15 +191,18 @@ def blurhash_decode(blurhash, width, height, punch=1.0, linear=False):
def blurhash_encode(image, components_x=4, components_y=4, linear=False): def blurhash_encode(image, components_x=4, components_y=4, linear=False):
""" """
Calculates the blurhash for an image using the given x and y component counts. Calculates the blurhash for an image using the given x and y
component counts.
Image should be a 3-dimensional array, with the first dimension being y, the second Image should be a 3-dimensional array, with the first dimension
being x, and the third being the three rgb components that are assumed to be 0-255 being y, the second being x, and the third being the three rgb
srgb integers (incidentally, this is the format you will get from a PIL RGB image). components that are assumed to be 0-255 srgb integers
(incidentally, this is the format you will get from a PIL RGB image).
You can also pass in already linear data - to do this, set linear to True. This is You can also pass in already linear data - to do this, set linear
useful if you want to encode a version of your image resized to a smaller size (which to True. This is useful if you want to encode a version of your
you should ideally do in linear colour). image resized to a smaller size (which you should ideally do in
linear colour).
""" """
if components_x < 1 or components_x > 9 or \ if components_x < 1 or components_x > 9 or \
components_y < 1 or components_y > 9: components_y < 1 or components_y > 9:
@ -202,7 +213,7 @@ def blurhash_encode(image, components_x=4, components_y=4, linear=False):
# Convert to linear if neeeded # Convert to linear if neeeded
image_linear = [] image_linear = []
if linear == False: if linear is False:
for y in range(int(height)): for y in range(int(height)):
image_linear_line = [] image_linear_line = []
for x in range(int(width)): for x in range(int(width)):
@ -224,7 +235,9 @@ def blurhash_encode(image, components_x=4, components_y=4, linear=False):
component = [0.0, 0.0, 0.0] component = [0.0, 0.0, 0.0]
for y in range(int(height)): for y in range(int(height)):
for x in range(int(width)): for x in range(int(width)):
basis = norm_factor * math.cos(math.pi * float(i) * float(x) / width) * \ basis = \
norm_factor * \
math.cos(math.pi * float(i) * float(x) / width) * \
math.cos(math.pi * float(j) * float(y) / height) math.cos(math.pi * float(j) * float(y) / height)
component[0] += basis * image_linear[y][x][0] component[0] += basis * image_linear[y][x][0]
component[1] += basis * image_linear[y][x][1] component[1] += basis * image_linear[y][x][1]
@ -235,9 +248,10 @@ def blurhash_encode(image, components_x=4, components_y=4, linear=False):
component[2] /= (width * height) component[2] /= (width * height)
components.append(component) components.append(component)
if not (i==0 and j==0): if not (i == 0 and j == 0):
max_ac_component = max(max_ac_component,abs(component[0]), max_ac_component = \
abs(component[1]),abs(component[2])) max(max_ac_component, abs(component[0]),
abs(component[1]), abs(component[2]))
# Encode components # Encode components
dc_value = (linear_to_srgb(components[0][0]) << 16) + \ dc_value = (linear_to_srgb(components[0][0]) << 16) + \
@ -251,10 +265,16 @@ def blurhash_encode(image, components_x=4, components_y=4, linear=False):
ac_values = [] ac_values = []
for r, g, b in components[1:]: for r, g, b in components[1:]:
r2 = r / ac_component_norm_factor
g2 = g / ac_component_norm_factor
b2 = b / ac_component_norm_factor
r3 = math.floor(sign_pow(r2, 0.5) * 9.0 + 9.5)
g3 = math.floor(sign_pow(g2, 0.5) * 9.0 + 9.5)
b3 = math.floor(sign_pow(b2, 0.5) * 9.0 + 9.5)
ac_values.append( ac_values.append(
int(max(0.0,min(18.0,math.floor(sign_pow(r / ac_component_norm_factor, 0.5) * 9.0 + 9.5)))) * 19 * 19 + \ int(max(0.0, min(18.0, r3))) * 19 * 19 +
int(max(0.0,min(18.0, math.floor(sign_pow(g / ac_component_norm_factor, 0.5) * 9.0 + 9.5)))) * 19 + \ int(max(0.0, min(18.0, g3))) * 19 +
int(max(0.0,min(18.0, math.floor(sign_pow(b / ac_component_norm_factor, 0.5) * 9.0 + 9.5)))) int(max(0.0, min(18.0, b3)))
) )
# Build final blurhash # Build final blurhash

3
static_analysis
View File

@ -10,9 +10,6 @@ echo "Starting static analysis"
for sourceFile in *.py for sourceFile in *.py
do do
if [[ "$sourceFile" == 'blurhash.py' ]]; then
continue
fi
result=$($cmd "$sourceFile") result=$($cmd "$sourceFile")
if [ "$result" ]; then if [ "$result" ]; then
echo '' echo ''