普通浮點數轉半精度工具實現

工作中遇到的小工具制作，直接壓成exe就可以用了

功能：輸入文件，將文件中的一列浮點轉為一列半精度浮點(半精度浮點)或者兩列浮點轉為一列復數型半精度浮點，輸出為同級目錄源文件名+Out.dat，輸入0退出

case1:文件中是一行數據0.5，那么輸出就是0x3c00

case2:文件中一行數據是 0.5 0或者0 0.5；輸出就是0x3C000000和0x00003C00

#! /usr/bin/env python
# encoding:utf-8
# ! /usr/bin/env python
# encoding:utf-8
import math
import re


def Real2HalfFloat(data):
	MINNUM = -65536
	MAXNUM = 65535
	FloatVal = 0
	if data:
		if data < MINNUM:
			data = MINNUM
		if data > MAXNUM:
			data = MAXNUM
		
		sign = 0
		if data < 0:
			sign = 1
			data = -data
		
		exp = math.floor((math.log2(data)))
		expout = exp + 16
		
		Mantial = round(data / pow(2, exp - 10)) - 1024
		
		if expout <= 0:
			FloatVal = 0
		else:
			FloatVal = sign * 32768 + expout * 1024 + Mantial
	return FloatVal


def ReadCfloatData(sourcefile):
	input = []
	with open(sourcfile, 'r') as f:
		for line in f.readlines():
			line = line.strip()
			line = re.sub('\s+',' ',line) #兩個數字間多個空格
			input.append(line.split(' '))
	destfile = sourcefile.replace('.dat', '')
	destfile = destfile.replace('.txt', '')
	destfile += 'Out.dat'
	with open(destfile, 'w') as fw:
		for i in range(len(input)):
			if len(input[i]) == 2:
				real = Real2HalfFloat(float(input[i][0]))
				imag = Real2HalfFloat(float(input[i][1]))
				result = real * 65536 + imag
				if imag and not real:
					fw.write('0x0000' + "%X" % result + '\n')
				elif not imag and not real:
					fw.write('0x00000000' + '\n')
				else:
					fw.write('0x' + "%X" % result + '\n')
			elif len(input[i]) == 1:
				result = Real2HalfFloat(float(input[i][0]))
				if result:
					fw.write('0x' + "%X" % result + '\n')
				else:
					fw.write('0x0000' + '\n')


if __name__ == '__main__':
	print('Tips: Input number 0 if you want to exit!\n')
	while True:
		sourcfile = input("input source file:\n")
		if sourcfile is '0':
			break
		ReadCfloatData(sourcfile)
		print('Transfer Success!')