本文共 1506 字,大约阅读时间需要 5 分钟。
第一次做物理建模这事,还挺有意思,由于只是demo,没有深入研究下去,所以这个目标也就比较浅显,只是:检测旋转到某1角度不要发生碰撞,不带重力
from shapely.geometry import Polygonfrom shapely import affinityimport matplotlib.pyplot as pltdef make_car(leftX, leftY, width, height): car_lines = Polygon([(leftX, leftY), (leftX+width, leftY), (leftX+width, leftY+height), (leftX, leftY+height), (leftX, leftY)]) return car_linesdef showCar(car): # x, y = car.xy x, y=car.exterior.xy plt.plot(x, y)def showCars(cars): for car in cars: showCar(car)car_width=4.7car_height=2.5car1 = make_car(1, 1, car_width, car_height)car2 = make_car(6, 1, car_width, car_height)car3 = make_car(11, 1, car_width, car_height)car4 = make_car(16, 1, car_width, car_height)car5 = make_car(1, 6, car_width, car_height)car6 = make_car(6, 6, car_width, car_height)car7 = make_car(11, 6, car_width, car_height)car8 = make_car(16, 6, car_width, car_height)plt.axis('equal')plt.ion()cars = [car1, car2, car3, car4, car5, car6, car7, car8]showCars(cars)plt.pause(1)for i in range(1, 30): plt.cla() rotated_car3 = affinity.rotate(car3, i) showCars([car1, car2, rotated_car3, car4, car5, car6, car7, car8]) #计算是否碰撞 print('=======================', i, '====================') cars = [car1, car2, car4, car5, car6, car7, car8] conflict=False for car in cars: if rotated_car3.intersects(car): conflict=True break if conflict: print('撞到了,当前旋转角度:', i) break plt.pause(0.1)plt.ioff()plt.show()
代码比较简单,用到的有几何库:shapely
转载地址:http://gfkwk.baihongyu.com/