Opencv+python 零基础入门作业（二）

题目：在作业1的基础上画一个小球（通过键盘输入小球半径和颜色），让其沿着空心圆的路径运动一周，窗口显示小球运动的动态效果，并将小球运动的过程录成一个视频文件保存到硬盘

备注：作业一：读入一幅图像，在上面画一个空心圆（通过键盘输入圆心坐半径和颜色值），窗口显示画圆前后的效果，并保存画圆之后的图像到硬盘上为一个新的图像文件。

 

一开始的思路是制作的小球绕这个圆的路径分为四分，奈何自己数学功底不够，弄不出来，还是选择了bresenham算法，把小球的圆心当做是上一题要画的点，那么半径就是这两个圆的半径之和

分成8份开始运动，注意他们的编码格式，用了8个循环画出来，代码就显得很长，最后制作视频的时候再循环读取，我发现图片格式与视频格式不一致时制作的视频是打不开的，不知道能不能解决

#测试用例：圆心坐标(100,100) 圆半径：50，圆颜色(255,0,0)
#小球半径25,小球颜色(0,255,0)

import numpy as np
import time
import cv2

img = cv2.imread('1.jpg')
(center_x,center_y) = eval(input("请输入圆心坐标(输入格式为：(x,y))："))
r = eval(input("请输入大圆的半径："))
color1 = eval(input("请输入大圆颜色(用三原色格式)："))
cv2.circle(img,(center_x,center_y),1,color1,-1)   #画圆心
(x,y)=(r,0)  #起始点
d=3-2*r

while x>=y:    #在y=x这条直线上绘画结束
    cv2.circle(img,(center_x+x,center_y+y),1,color1,1) 
    cv2.circle(img,(center_x+x,center_y-y),1,color1,1)   
    cv2.circle(img,(center_x-x,center_y+y),1,color1,1)    
    cv2.circle(img,(center_x-x,center_y-y),1,color1,1)    
    cv2.circle(img,(center_y+y,center_x+x),1,color1,1)    
    cv2.circle(img,(center_y+y,center_x-x),1,color1,1)    
    cv2.circle(img,(center_y-y,center_x+x),1,color1,1)
    cv2.circle(img,(center_y-y,center_x-x),1,color1,1)
    
    if d>=0:         #朝西北方向移动
        d=d+4*y-4*x+10
        x=x-1       
    else:           #朝正北方向移动
        d=d+4*y+6  
    y=y+1            #像素点朝北移动

cv2.imwrite('2.jpg',img)  #保存画完圆的图


#作业一与作业二分界线

r2 = eval(input("请输入小圆的半径："))
color2 = eval(input("请输入小圆颜色(用三原色格式)："))
r=r+r2  #实际上是小圆的圆心是大圆+小圆的半径组成，他的路径是这个圆的弧
(x,y)=(r,0)  #起始点
d=3-2*r
picnum = 0

while x>=y:    
    img2 = cv2.imread('2.jpg') 
    cv2.circle(img2,(center_x+x,center_y+y),r2,color2,-1)
    picnum=picnum+1 
    filename='D:/pythontest/'+str(picnum)+'.jpg'  #图片保存路径
    cv2.imwrite(filename,img2)
    if d>=0:         
        d=d+4*y-4*x+10
        x=x-1       
    else:           
        d=d+4*y+6  
    y=y+1

total=picnum

(x,y)=(r,0)  
d=3-2*r
picnum = total*2 +1 
    
while x>=y:    
    img2 = cv2.imread('2.jpg') 
    cv2.circle(img2,(center_y+y,center_x+x),r2,color2,-1)
    picnum=picnum-1
    filename='D:/pythontest/'+str(picnum)+'.jpg'  #图片保存路径
    cv2.imwrite(filename,img2)
    if d>=0:         
        d=d+4*y-4*x+10
        x=x-1       
    else:           
        d=d+4*y+6  
    y=y+1

(x,y)=(r,0)  
d=3-2*r
picnum = total*2 +1
while x>=y:    
    img2 = cv2.imread('2.jpg') 
    cv2.circle(img2,(center_y-y,center_x+x),r2,color2,-1)
    picnum=picnum+1
    filename='D:/pythontest/'+str(picnum)+'.jpg'  #图片保存路径
    cv2.imwrite(filename,img2)
    if d>=0:         
        d=d+4*y-4*x+10
        x=x-1       
    else:           
        d=d+4*y+6  
    y=y+1


(x,y)=(r,0)  
d=3-2*r
picnum = total*4 +3
while x>=y:    
    img2 = cv2.imread('2.jpg') 
    cv2.circle(img2,(center_x-x,center_y+y),r2,color2,-1)
    picnum=picnum-1
    filename='D:/pythontest/'+str(picnum)+'.jpg'  #图片保存路径
    cv2.imwrite(filename,img2)
    if d>=0:         
        d=d+4*y-4*x+10
        x=x-1       
    else:           
        d=d+4*y+6  
    y=y+1

(x,y)=(r,0)  
d=3-2*r
picnum = total*4 +4
while x>=y:    
    img2 = cv2.imread('2.jpg') 
    cv2.circle(img2,(center_x-x,center_y-y),r2,color2,-1)
    picnum=picnum+1
    filename='D:/pythontest/'+str(picnum)+'.jpg'  #图片保存路径
    cv2.imwrite(filename,img2)
    if d>=0:         
        d=d+4*y-4*x+10
        x=x-1       
    else:           
        d=d+4*y+6  
    y=y+1
    
(x,y)=(r,0)  
d=3-2*r
picnum = total*6 +5
while x>=y:    
    img2 = cv2.imread('2.jpg') 
    cv2.circle(img2,(center_y-y,center_x-x),r2,color2,-1)
    picnum=picnum-1
    filename='D:/pythontest/'+str(picnum)+'.jpg'  #图片保存路径
    cv2.imwrite(filename,img2)
    if d>=0:         
        d=d+4*y-4*x+10
        x=x-1       
    else:           
        d=d+4*y+6  
    y=y+1

(x,y)=(r,0)  
d=3-2*r
picnum = total*6 +6
while x>=y:    
    img2 = cv2.imread('2.jpg') 
    cv2.circle(img2,(center_y+y,center_x-x),r2,color2,-1)
    picnum=picnum+1
    filename='D:/pythontest/'+str(picnum)+'.jpg'  #图片保存路径
    cv2.imwrite(filename,img2)
    if d>=0:         
        d=d+4*y-4*x+10
        x=x-1       
    else:           
        d=d+4*y+6  
    y=y+1


(x,y)=(r,0)  
d=3-2*r
picnum = total*8 +7
while x>=y:    
    img2 = cv2.imread('2.jpg') 
    cv2.circle(img2,(center_x+x,center_y-y),r2,color2,-1)
    picnum=picnum-1
    filename='D:/pythontest/'+str(picnum)+'.jpg'  #图片保存路径
    cv2.imwrite(filename,img2)
    if d>=0:         
        d=d+4*y-4*x+10
        x=x-1       
    else:           
        d=d+4*y+6  
    y=y+1
    
print("视频素材制作完成")




fourcc = cv2.VideoWriter_fourcc(*'XVID') 

fps = 30  
size = (1152,648)  # 需要转为视频的图片的尺寸，我发现图片尺寸与这个不一致制作的视频不能播放，不知道怎么觉得
# 可以使用cv2.resize()进行修改

video = cv2.VideoWriter('C:/Users/aa/Desktop/test.avi', fourcc,int(fps), size)  #保存到桌面



picnum=8*total+7

while picnum>=1:
    path = 'D:/pythontest/'+str(picnum)+'.jpg'
    img = cv2.imread(path)
    video.write(img)
    picnum=picnum-1
video.release()
print("视频制作完毕")



cap = cv2.VideoCapture("test.avi")
while cap.isOpened():
    ret, frame = cap.read()
    if ret:
        cv2.imshow("frame", frame)
    else:
        print("视频播放完成！")
        break
    key = cv2.waitKey(25)
    if key == 27:  # 按键esc
        break
cap.release()
cv2.destroyAllWindows()