mirror/collision-analyisis-and-interpolation
1
0
Fork 0
mirror of https://github.com/Findus23/collision-analyisis-and-interpolation.git synced 2024-09-19 15:13:50 +02:00
Code Releases Activity

allow specifying a file name to load SimulationList from

Browse source
This commit is contained in:
Lukas Winkler 2021-02-03 22:16:58 +01:00
parent 5e55a18b99
commit e2e0ae067b
Signed by: lukas
GPG key ID: 54DE4D798D244853
1 changed files with 9 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
simulation_list.py
View file

@ -1,6 +1,6 @@
import json
import os
import pickle
from pathlib import Path
from typing import List
import numpy as np
@ -21,30 +21,25 @@ class SimulationList:
def append(self, value: Simulation):
self.simlist.append(value)
def save_path(self, extension):
script_dir = os.path.dirname(__file__)
rel_path = "save" + extension
return os.path.join(script_dir, rel_path)
def pickle_save(self):
with open(self.save_path(".pickle"), 'wb') as file:
def pickle_save(self, pickle_file: Path):
with pickle_file.open("wb") as file:
pickle.dump(self.simlist, file)
@classmethod
def pickle_load(cls):
def pickle_load(cls, pickle_file: Path):
tmp = cls()
with open(cls.save_path(tmp, ".pickle"), 'rb') as file:
with pickle_file.open("rb") as file:
return cls(pickle.load(file))
def jsonlines_save(self):
with open(self.save_path(".jsonl"), 'w') as file:
def jsonlines_save(self, jsonl_file: Path):
with jsonl_file.open("w") as file:
for sim in self.simlist:
file.write(json.dumps(vars(sim)) + "\n")
@classmethod
def jsonlines_load(cls):
def jsonlines_load(cls, jsonl_file: Path):
simlist = cls()
with open(cls.save_path(simlist, ".jsonl"), 'r') as file:
with jsonl_file.open() as file:
for line in file:
sim = Simulation.from_dict(json.loads(line))
simlist.append(sim)