MichaelBradley/nbody
1
0
Fork 0
Code Issues Pull requests Activity Actions

Add generation of 3d data

Browse source
This commit is contained in:
Michael Bradley 2023-10-08 12:21:51 -04:00
parent 424721ccfc
commit 22bc527716
14 changed files with 105 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
gen_data.py
View file

@ -6,6 +6,7 @@ from random import uniform, randint
class Args:
width: int
height: int
depth: int
velocity: float
mass: float
count: int
@ -20,6 +21,7 @@ if __name__ == "__main__":
parser.add_argument("-w", "--width", type=int, default=1900)
parser.add_argument("-h", "--height", type=int, default=1000)
parser.add_argument("-d", "--depth", type=int, default=0)
parser.add_argument("-v", "--velocity", type=float, default=1.)
parser.add_argument("-m", "--mass", type=float, default=1.)
parser.add_argument("-c", "--count", type=int, default=500)
@ -29,6 +31,8 @@ if __name__ == "__main__":
for _ in range(args.count):
print(f"{randint(-args.width // 2, args.width // 2)},"
f"{randint(-args.height // 2, args.height // 2)},"
f"{f'{randint(-args.depth // 2, args.depth // 2)},' if args.depth else ''}"
f"{uniform(-args.velocity, args.velocity)},"
f"{uniform(-args.velocity, args.velocity)},"
f"{f'{uniform(-args.velocity, args.velocity)},' if args.depth else ''}"
f"{uniform(1e-2, args.mass)}")