binary16_verify_hex.py

#!/usr/bin/env python3

#* Helper script for binary16 *#
#*  Yigit Suoglu, 20/05/2021  *#

import sys

def calculate(multi,val1,val2):
  if multi:
    print("\nMultiplying...\n")
    res = val1 * val2
  else:
    print("\nAdding...\n")
    res = val1 + val2
  print("Result: ", res)
  sign_bin = 0
  sign_res = '+'
  if res < 0:
    sign_res = '-'
    sign_bin = 1
    res*=-1
  print("Sign: ",sign_res)
  exp_res = 15
  frac_res = 0
  if res < 1:
    while res < 1:
      res*=2
      exp_res-=1
      if exp_res == 0:
        break
    if exp_res == 0:
      res/=2
    else:
      res-=1
  elif 2 < res:
    while 2 < res:
      res/=2
      exp_res+=1
      if exp_res == 31:
        print("\nOverflow!")
        return
    res-=1
  if res == 1:
    res = 0
    exp_res+=1
  else:
    res*=1024
  frac_res = int(res)
  if (res - float(frac_res)) > 0.5:
    frac_res+=1
  print("Fraction:", hex(frac_res))
  print("Exponent:", hex(exp_res))
  val = ((sign_bin&1) << 15) + ((exp_res&31) << 10) + (frac_res&1023)
  print("In Hex:", hex(val))
      

print("This script calculates the result of binary16 operation.")

if(len(sys.argv)>1):
  hex1 = sys.argv[1]
else:
  hex1 = input("\nFirst operand: ")
hex1 = int(hex1, 16)
sign1 = (hex1 >>15) & 1
frac1 = hex1&1023
frac1 = 1.0 + float(frac1/1024)
exp1 = (hex1 >>10)&31
exp1 -= 15

if exp1 == -15:
  exp1+=1
  frac1-=1

val1 = frac1 * pow(2,exp1)

if sign1:
  val1*=-1

print("\nFirst operand: ", end='')
if sign1:
  print("-", end=' ')
else:
  print("+", end=' ')
print(frac1, "* 2^", exp1, " = ", val1)

if exp1 == 16:
  if frac1 == 1.0:
    print("Operand 1 is infinite!")
  else:
    print("Operand 1 is a NaN!")
  quit()
if(len(sys.argv)>2):
  hex2 = sys.argv[2]
else:
  hex2 = input("\nSecond operand: ")
hex2 = int(hex2, 16)
sign2 = (hex2 >>15) & 1
frac2 = hex2&1023
frac2 = 1.0 + float(frac2/1024)
exp2 = (hex2 >>10)&31
exp2 -= 15

if exp2 == -15:
  exp2+=1
  frac2-=1

val2 = frac2 * pow(2,exp2)

if sign2:
  val2*=-1

print("\nSecond operand: ", end='')
if sign2:
  print("-", end=' ')
else:
  print("+", end=' ')
print(frac2, "* 2^", exp2, " = ", val2)

if exp2 == 16:
  if frac2 == 1.0:
    print("Operand 2 is infinite!")
  else:
    print("Operand 2 is a NaN!")
  quit()

calculate(False,val1,val2)
calculate(True,val1,val2)