example2.py

import pyae
# from decimal import getcontext

# Example for encoding a simple text message using the PyAE module.

# Create the frequency table.
frequency_table = {"a": 2,
                   "b": 3,
                   "c": 1,
                   "d": 4}

# Create an instance of the ArithmeticEncoding class.
AE = pyae.ArithmeticEncoding(frequency_table, 
                             save_stages=True)

# Default precision is 28. Change it to do arithmetic operations with larger/smaller numbers.
# getcontext().prec = 28

original_msg = "bdab"
print("Original Message: {msg}".format(msg=original_msg))

# Encode the message
encoded_msg, encoder , interval_min_value, interval_max_value = AE.encode(msg=original_msg, 
                                                                          probability_table=AE.probability_table)
print("Encoded Message: {msg}".format(msg=encoded_msg))

# Decode the message
decoded_msg, decoder = AE.decode(encoded_msg=encoded_msg, 
                                 msg_length=len(original_msg),
                                 probability_table=AE.probability_table)
print("Decoded Message: {msg}".format(msg=decoded_msg))

decoded_msg = "".join(decoded_msg)
print("Message Decoded Successfully? {result}".format(result=original_msg == decoded_msg))