Gael-de-Sailly
/
python-mtschem
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added ability to create blank schems

master
Gael-de-Sailly 2018-11-17 18:47:27 +01:00
parent a9662c1fb7
commit 639fd250d3
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mtschem.py
View File

@ -50,7 +50,17 @@ from io import BytesIO
bulk_dtype = np.dtype([("node", ">u2"), ("prob", "u1"), ("force", "?"), ("param2", "u1")])
class Schem:
def __init__(self, filename):
def __init__(self, *args):
if isinstance(args[0], str):
self.load(args[0])
else:
shape = tuple(args[:3])
self.version = 4
self.yprobs = np.zeros(shape[1], dtype="u1")
self.nodes = ["air"]
self.data = np.zeros(shape, dtype=bulk_dtype)
def load(self, filename):
f = open(filename, "rb")
if f.read(4) != b"MTSM":