RF2_rfl_rfc/rf2.py at master · FactionFiles/RF2_rfl_rfc · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#!/usr/bin/python3


import struct

class xyz:
	x = 0.
	y = 0.
	z = 0.

	def __init__(self, _x = 0., _y = 0., _z = 0.):
		self.x = _x
		self.y = _y
		self.z = _z

	def __add__(self, o):
		return xyz(self.x + o.x, self.y + o.y, self.z + o.z)

	def __str__(self):
		return "{:f} {:f} {:f}".format(self.x, self.y, self.z)

	def TransformBy(self, mtx):
		if (mtx != None):
			return xyz(mtx[0] * self.x + mtx[1] * self.y + mtx[2] * self.z ,
				       mtx[3] * self.x + mtx[4] * self.y + mtx[5] * self.z ,
					   mtx[6] * self.x + mtx[7] * self.y + mtx[8] * self.z )
		else:
			return xyz(self.x, self.y, self.z)


def read4u(hndl):
	return int.from_bytes(hndl.read(4), byteorder="little")

def read4S(hndl):
	return int.from_bytes(hndl.read(4), byteorder="little", signed=True)

def read2u(hndl):
	return int.from_bytes(hndl.read(2), byteorder="little")

def read1u(hndl):
	return int.from_bytes(hndl.read(1), byteorder="little")

def read1S(hndl):
	return int.from_bytes(hndl.read(1), byteorder="little", signed=True)

def readNstr(hndl):
	num = read2u(hndl)
	tmp = hndl.read(num)
	st = ""
	for c in tmp:
		if (c == 0):
			break
		st += chr(c)
	return st

def readCstr(hndl):
	tmp = ""
	while True:
		a = read1u(hndl)
		if (a == 0):
			break

		tmp += chr(a)

	return tmp

def readFixNstr(hndl, num):
	tmp = hndl.read(num)
	st = ""
	for c in tmp:
		if (c == 0):
			break
		st += chr(c)
	return st

def readflt(hndl):
	return struct.unpack("<f", hndl.read(4))[0]

def readXYZ(hndl):
	x = readflt(hndl)
	y = readflt(hndl)
	z = readflt(hndl)
	return xyz(x, y, z)

def readMt33(hndl):
	mtx =  [readflt(hndl), readflt(hndl), readflt(hndl),
			readflt(hndl), readflt(hndl), readflt(hndl),
			readflt(hndl), readflt(hndl), readflt(hndl)]
	return [mtx[3], mtx[6], mtx[0],
	        mtx[4], mtx[7], mtx[1],
			mtx[5], mtx[8], mtx[2]]