Skip to content
Snippets Groups Projects
Commit 9f5366a2 authored by hui's avatar hui
Browse files

Merge branch 'main' into combinedSimulation

parents 5f39cc00 e8f750df
No related branches found
No related tags found
1 merge request!14add a auto-documentation
...@@ -8,3 +8,4 @@ bike/__pycache__/* ...@@ -8,3 +8,4 @@ bike/__pycache__/*
debug.log debug.log
data/* data/*
images/popu*
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import json
import os
def save_bike_image(points_position, counter): def save_bike_image(points_position, counter):
"""This function save an image of the bike. """This function save an image of the bike.
...@@ -11,8 +12,8 @@ def save_bike_image(points_position, counter): ...@@ -11,8 +12,8 @@ def save_bike_image(points_position, counter):
fig, ax = plt.subplots() fig, ax = plt.subplots()
ax.set_title(f"Bike {counter}") ax.set_title(f"Bike {counter}")
ax.axis('equal') ax.axis('equal')
ax.set_xlim(0, 35) ax.set_xlim(0, 15)
ax.set_ylim(-5, 20) ax.set_ylim(-5, 10)
num_point=4 num_point=4
wheel_radius = 1.2 wheel_radius = 1.2
...@@ -41,4 +42,18 @@ def save_bike_image(points_position, counter): ...@@ -41,4 +42,18 @@ def save_bike_image(points_position, counter):
#num_point=4 #num_point=4
#points_position = np.random.rand(num_point, 2)*8+3 #points_position = np.random.rand(num_point, 2)*8+3
#save_bike_image(points_position, 2) #save_bike_image(points_position, 2)
\ No newline at end of file
list_files = os.listdir('../data')
for index,item in enumerate(list_files):
os.makedirs(f'population_{index}', exist_ok=True)
os.chdir(f'population_{index}')
with open(f'../../data/{item}', 'r') as f:
bikes = json.load(f)
print(len(bikes))
f.close()
for bike in range(len(bikes)):
print(bike)
save_bike_image(np.array(bikes[bike]),bike)
os.chdir('../')
#print(type(bikes[0][0]))
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment