loic/esds
1
0
Fork 0
mirror of https://gitlab.com/manzerbredes/esds.git synced 2025-04-06 10:06:28 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Debug entry point

Browse source
This commit is contained in:
Loic Guegan 2022-09-13 10:46:44 +02:00
parent 3ae36dadc1
commit e188ca4040
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
esds/__main__.py
View file

@ -1,8 +1,5 @@
import sys, argparse, os import sys, argparse, os
import platform from .platform import YAMLPlatformFile
# Allow importlib to import file from current working directory
sys.path.insert(0, os.getcwd())
def run(arguments): def run(arguments):
parser = argparse.ArgumentParser(description='Run a simulation') parser = argparse.ArgumentParser(description='Run a simulation')
@ -10,6 +7,8 @@ def run(arguments):
args = parser.parse_args(arguments[1:]) args = parser.parse_args(arguments[1:])
if args.platform: if args.platform:
simulation=YAMLPlatformFile(args.platform) simulation=YAMLPlatformFile(args.platform)
# Allow importlib (in simulator.run()) to import file from the platform.yaml directory
sys.path.insert(0, simulation.location)
simulation.run() simulation.run()
else: else:
parser.print_help() parser.print_help()

5
esds/platform.py
View file

@ -1,7 +1,7 @@
import yaml, os, importlib import yaml, os, importlib
import numpy as np import numpy as np
import Simulator from .simulator import Simulator
class UnitsParser: class UnitsParser:
def node_range(r,limit): def node_range(r,limit):
@ -46,6 +46,7 @@ class YAMLPlatformFile:
def __init__(self, file_path): def __init__(self, file_path):
self.file_path=file_path self.file_path=file_path
self.location=os.path.dirname(os.path.abspath(file_path))
self.default={ self.default={
"breakpoints": [], "breakpoints": [],
"breakpoints_every": None, "breakpoints_every": None,
@ -152,7 +153,7 @@ class YAMLPlatformFile:
words=impl.split() words=impl.split()
r=UnitsParser.node_range(words[0],self.default["node_count"]) r=UnitsParser.node_range(words[0],self.default["node_count"])
file="".join(words[1:]) file="".join(words[1:])
if not os.path.exists(file): if not os.path.exists(os.path.join(self.location,file)):
self.parsing_error("File "+file+ " not found") self.parsing_error("File "+file+ " not found")
path, extension = os.path.splitext(file) path, extension = os.path.splitext(file)
if extension != ".py": if extension != ".py":