creyD/entro.py
1
0
Fork 0
mirror of https://github.com/creyD/entro.py.git synced 2026-04-12 20:00:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

Added maximum entropy calculator

Browse Source 
- Optimised some of the code
This commit is contained in:
Conrad
2019-07-18 13:58:45 +02:00
parent 3d3f6a44ac
commit f85c3120ee
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
entro.py
View File

@@ -11,29 +11,31 @@ import argparse
# Calculates the entropy of a given string (as described in the docstring) # Calculates the entropy of a given string (as described in the docstring)
def calculateEntropy(input_string): def calculateEntropy(input_string):
alphabet, alphabet_size, entropy = {}, 0, 0 alphabet, alphabet_size, entropy = {}, len(input_string), 0
for char in input_string: for char in input_string:
if char in alphabet: if char in alphabet:
alphabet[char] += 1 alphabet[char] += 1
else: else:
alphabet[char] = 1 alphabet[char] = 1
alphabet_size += 1
for char in alphabet: for char in alphabet:
alphabet[char] = alphabet[char] / alphabet_size alphabet[char] = alphabet[char] / alphabet_size
entropy -= alphabet[char] * math.log(alphabet[char], 2) entropy -= alphabet[char] * math.log(alphabet[char], 2)
return entropy, alphabet max_entropy = - len(alphabet) * (1/len(alphabet) * math.log(1/len(alphabet), 2))
return entropy, alphabet, max_entropy
# Outputs a given entropy including the original text and the alphabet with probabilities # Outputs a given entropy including the original text and the alphabet with probabilities
def printEntropy(original_string, entropy_value, alphabet_dict, simple_bool): def printEntropy(original_string, entropy_value, alphabet_dict, simple_bool, max_value):
print('---') print('---')
if simple_bool == False: if simple_bool == False:
print('Content: ' + original_string) print('Content: ' + original_string)
print('Probabilities: ' + str(alphabet_dict)) print('Probabilities: ' + str(alphabet_dict))
print('Entropy: ' + str(entropy_value) + ' bits') print('Entropy: ' + str(entropy_value) + ' bits')
if max_value:
print('Maximum Entropy: ' + str(max_value) + ' bits')
print('---') print('---')
@@ -46,7 +48,7 @@ def readEntropyFile(path_string):
# List of the arguments one can use to influence the behavior of the program # List of the arguments one can use to influence the behavior of the program
parser = argparse.ArgumentParser(description='Calculate the information entropy of some strings.') parser = argparse.ArgumentParser(description='Calculate the information entropy of alphabets.')
# INPUT ARGUMENTS # INPUT ARGUMENTS
parser.add_argument('strings', nargs='*', default='', type=str, help='Strings to calculate the entropy of.') parser.add_argument('strings', nargs='*', default='', type=str, help='Strings to calculate the entropy of.')
@@ -54,6 +56,7 @@ parser.add_argument('--files', nargs='*', type=str, default='', help='Provide fi
# OUTPUT OPTIONS # OUTPUT OPTIONS
parser.add_argument('--simple', nargs='?', type=bool, default=False, help='Determines the explicitness of the output. (True = only entropy shown)') parser.add_argument('--simple', nargs='?', type=bool, default=False, help='Determines the explicitness of the output. (True = only entropy shown)')
parser.add_argument('--max', nargs='?', type=bool, default=False, help='Includes the maximum entropy.')
# CONVERT OPTIONS # CONVERT OPTIONS
parser.add_argument('--lower', nargs='?', type=bool, default=False, help='Converts given strings or textfiles to lowercase before calculating.') parser.add_argument('--lower', nargs='?', type=bool, default=False, help='Converts given strings or textfiles to lowercase before calculating.')
@@ -83,5 +86,5 @@ for string in queue:
if args.squash != False: if args.squash != False:
string = string.replace(" ", "") string = string.replace(" ", "")
a, b = calculateEntropy(string) a, b, c = calculateEntropy(string)
printEntropy(string, a, b, args.simple) printEntropy(string, a, b, args.simple, (False if args.max == False else c))