用 Python 實現了導彈自動追蹤!

數據分析那些事
11 min readMar 16, 2022

--

自動追蹤演算法,在我們製作射擊類遊戲時經常會用到。這個聽起來很高大上的東西,其實並不是軍事學的專利,從數學上來說就是解微分方程。

這個沒有點數學基礎是很難算出來的。但是我們有了計算機就不一樣了,依靠計算機極快速的運算速度,我們利用微分的思想,加上一點簡單的三角學知識,就可以實現它。

好,話不多說,我們來看看它的演算法原理,看圖:

由於待會要用pygame演示,它的座標系是y軸向下,所以這裡我們也用y向下的座標系。

演算法總的思想就是根據上圖,把時間t分割成足夠小的片段(比如1/1000,這個時間片越小越精確),每一個片段分別構造如上三角形,計算出導彈下一個時間片走的方向(即∠a)和走的路程(即vt=|AC|),這時候目標再在第二個時間片移動了位置,這時剛才計算的C點又變成了第二個時間片的初始點,這時再在第二個時間片上在C點和新的目標點構造三角形計算新的vt,然後進入第三個時間片,如此反覆即可。

假定導彈和目標的初始狀態下座標分別是(x1,y1),(x,y),構造出直角三角形ABE,這個三角形用來求∠a的正弦和餘弦值,因為vt是自己設定的,我們需要計算A到C點x和y座標分別移動了多少,移動的值就是AD和CD的長度,這兩個分別用vt乘cos(a)和sin(a)即可。

計算sin(a)和cos(a),正弦對比斜,餘弦鄰比斜,斜邊可以利用兩點距離公式計算出,即:

於是

AC的長度就是導彈的速度乘以時間即 |AC|=vt,然後即可計算出AD和CD的長度,於是這一個時間片過去後,導彈應該出現在新的位置C點,他的座標就是老的點A的x增加AD和y減去CD。

於是,新的C點座標就是:

只要一直反覆迴圈執行這個操作即可,好吧,為了更形象,把第一個時間片和第二個時間片放在一起看看:

第一個是時間片構造出的三角形是ABE,經過一個時間片後,目標從B點走到了D點,導彈此時在C點,於是構造新的三角形CDF,重複剛才的計算過程即可,圖中的角∠b就是導彈需要旋轉的角度,現實中只需要每個時間片修正導彈的方向就可以了,具體怎麼讓導彈改變方向,這就不是我們需要研究的問題了

好,由於最近在用Python的pygame庫製作小遊戲玩,接下來我們就用pygame來演示一下這個效果,效果如下圖:

很簡單的程式碼如下:

import pygame,sys
from math import *
pygame.init()
screen=pygame.display.set_mode((800,700),0,32)
missile=pygame.image.load('element/red_pointer.png').convert_alpha()
x1,y1=100,600 #导弹的初始发射位置
velocity=800 #导弹速度
time=1/1000 #每个时间片的长度
clock=pygame.time.Clock()
old_angle=0
while True:
for event in pygame.event.get():
if event.type==pygame.QUIT:
sys.exit()
clock.tick(300)
x,y=pygame.mouse.get_pos() #獲取滑鼠位置,滑鼠就是需要打擊的目標
distance=sqrt(pow(x1-x,2)+pow(y1-y,2)) #兩點距離公式
section=velocity*time #每個時間片需要移動的距離
sina=(y1-y)/distance
cosa=(x-x1)/distance
angle=atan2(y-y1,x-x1) #兩點線段的弧度值
x1,y1=(x1+section*cosa,y1-section*sina)
d_angle = degrees(angle) #弧度轉角度
screen.blit(missile, (x1-missile.get_width(), y1-missile.get_height()/2))
dis_angle=d_angle-old_angle #dis_angle就是到下一個位置需要改變的角度
old_angle=d_angle #更新初始角度
pygame.display.update()

如果僅把導彈考慮為一個質點的話,那麼以上演算法就已經足矣,我沒有做導彈的旋轉,因為一個質點也不分頭尾不需要旋轉,當然這前提得是你載入的導彈圖片很小的時候不旋轉看起來也沒什麼問題。但是在pygame裡面做旋轉並不是一件容易的事情,我們先把圖片替換成一張矩形的,再加入旋轉函式看看效果如何

missiled = pygame.transform.rotate(missile, -(d_angle))
screen.blit(missiled, (x1-missile.get_width(), y1-
missile.get_height()/2))

因為圖片的座標點是它的左上角的點,所以如果我們想讓圖片的座標固定在箭頭尖點,那麼把圖片實際列印位置x減少圖片長度,y減少一半寬度就行。但是實際執行效果並不好:

大致方向相同,但是圖片箭頭的尖點並沒有一直跟隨滑鼠,這是為什麼呢。經過一番研究,我發現原來是這個圖旋轉的機制問題,我們看看旋轉後的圖片變成什麼樣了:

旋轉後的圖片變成了藍色的那個範圍,根據旋轉角度的不同,所變成的圖片大小也不一樣,我們看旋轉90的情況

我們發現,旋轉後的圖片不僅面積變大了,導彈頭的位置也變了。那應該怎麼解決這個問題呢?思路是,每一次旋轉圖片以後,求出旋轉圖的頭位置(圖中的綠色箭頭點),然後把綠圖的列印位置移動一下,下,x,y分別移動兩個頭的距離,就可以讓旋轉後的導彈頭對準實際我們參與運算的那個導彈頭的位置,移動後應該是這樣的:

這樣,兩個導彈頭的點就一致了。接下來我們分析求旋轉後的導彈頭的演算法。根據旋轉角度的不同,旋轉角在不同象限引數不一樣,所以我們分為這四種情況

1,2象限

3,4象限,它的旋轉只有正負0–180,所以3,4象限就是負角

顯示圖片的時候我們將他移動

screen.blit(missiled, (x1-width+(x1-C[0]),y1-height/2+(y1-C[1])))

這裡的 (x1-width, y1-height/2) 其實才是上圖中的 (x1, y1)

所以最後我們加入相關演算法程式碼,效果就比較完美了

大功告成,最後附上全部的演算法程式碼

import pygame,sys
from math import *
pygame.init()
font1=pygame.font.SysFont(‘microsoftyaheimicrosoftyaheiui’,23)
textc=font1.render(‘*’,True,(250,0,0))
screen=pygame.display.set_mode((800,700),0,32)
missile=pygame.image.load(‘element/rect1.png’).convert_alpha()
height=missile.get_height()
width=missile.get_width()
pygame.mouse.set_visible(0)
x1,y1=100,600 #導彈的初始發射位置
velocity=800 #導彈速度
time=1/1000 #每個時間片的長度
clock=pygame.time.Clock()
A=()
B=()
C=()
while True:
for event in pygame.event.get():
if event.type==pygame.QUIT:
sys.exit()
clock.tick(300)
x,y=pygame.mouse.get_pos() #獲取滑鼠位置,滑鼠就是需要打擊的目標
distance=sqrt(pow(x1-x,2)+pow(y1-y,2)) #兩點距離公式
section=velocity*time #每個時間片需要移動的距離
sina=(y1-y)/distance
cosa=(x-x1)/distance
angle=atan2(y-y1,x-x1) #兩點間線段的弧度值
fangle=degrees(angle) #弧度轉角度
x1,y1=(x1+section*cosa,y1-section*sina)
missiled=pygame.transform.rotate(missile,-(fangle))
if 0<=-fangle<=90:
A=(width*cosa+x1-width,y1-height/2)
B=(A[0]+height*sina,A[1]+height*cosa)
if 90<-fangle<=180:
A = (x1 — width, y1 — height/2+height*(-cosa))
B = (x1 — width+height*sina, y1 — height/2)
if -90<=-fangle<0:
A = (x1 — width+missiled.get_width(), y1 — height/2+missiled.get_height()-height*cosa)
B = (A[0]+height*sina, y1 — height/2+missiled.get_height())
if -180<-fangle<-90:
A = (x1-width-height*sina, y1 — height/2+missiled.get_height())
B = (x1 — width,A[1]+height*cosa )
C = ((A[0] + B[0]) / 2, (A[1] + B[1]) / 2)screen.fill((0,0,0))
screen.blit(missiled, (x1-width+(x1-C[0]),y1-height/2+(y1-C[1])))
screen.blit(textc, (x,y)) #滑鼠用一個紅色*代替
pygame.display.update()

以上便是用Python模擬導彈自動追蹤的程式碼例項。

※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※

我是「數據分析那些事」。常年分享數據分析乾貨,不定期分享好用的職場技能工具。各位也可以關注我的Facebook,按讚我的臉書並私訊「10」,送你十週入門數據分析電子書唷!期待你與我互動起來~

文章推薦

Excel圖表配色原理

麥肯錫:優秀資料科學家的5個特徵!

《 Python 進階 》中文版(附下載)

8個酷炫視覺化圖表,快速寫出老闆愛看的視覺化分析報告

--

--

數據分析那些事
數據分析那些事

Written by 數據分析那些事

這是一個專注於數據分析職場的內容部落格,聚焦一批數據分析愛好者,在這裡,我會分享數據分析相關知識點推送、(工具/書籍)等推薦、職場心得、熱點資訊剖析以及資源大盤點,希望同樣熱愛數據的我們一同進步! 臉書會有更多互動喔:https://www.facebook.com/shujvfenxi/

No responses yet