Switch to tabs
This commit is contained in:
parent
054b54c291
commit
8bb6ba19d5
5 changed files with 157 additions and 157 deletions
60
main.py
60
main.py
|
@ -7,40 +7,40 @@ import physics
|
|||
|
||||
|
||||
class Args:
|
||||
filename: str
|
||||
gravity: float
|
||||
dimensions: typing.Literal[2, 3]
|
||||
filename: str
|
||||
gravity: float
|
||||
dimensions: typing.Literal[2, 3]
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser(
|
||||
prog="n-body simulation",
|
||||
description="Simulating gravitational effects"
|
||||
)
|
||||
parser = argparse.ArgumentParser(
|
||||
prog="n-body simulation",
|
||||
description="Simulating gravitational effects"
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"-f",
|
||||
"--filename",
|
||||
default="data/2d/simple.csv"
|
||||
)
|
||||
parser.add_argument(
|
||||
"-g",
|
||||
"--gravity",
|
||||
type=float,
|
||||
default=1.
|
||||
)
|
||||
parser.add_argument(
|
||||
"-d",
|
||||
"--dimensions",
|
||||
type=int,
|
||||
choices=[2, 3],
|
||||
default=2
|
||||
)
|
||||
parser.add_argument(
|
||||
"-f",
|
||||
"--filename",
|
||||
default="data/2d/simple.csv"
|
||||
)
|
||||
parser.add_argument(
|
||||
"-g",
|
||||
"--gravity",
|
||||
type=float,
|
||||
default=1.
|
||||
)
|
||||
parser.add_argument(
|
||||
"-d",
|
||||
"--dimensions",
|
||||
type=int,
|
||||
choices=[2, 3],
|
||||
default=2
|
||||
)
|
||||
|
||||
args: Args = parser.parse_args()
|
||||
args: Args = parser.parse_args()
|
||||
|
||||
physics.G = args.gravity
|
||||
physics.G = args.gravity
|
||||
|
||||
objects = data.parse_csv(args.filename, dimensions=args.dimensions)
|
||||
a = data.Animator(*objects)
|
||||
a.show()
|
||||
objects = data.parse_csv(args.filename, dimensions=args.dimensions)
|
||||
a = data.Animator(*objects)
|
||||
a.show()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue