聚合国内IT技术精华文章,分享IT技术精华,帮助IT从业人士成长

60几行代码绘制丘比特爱情之箭!

2022-08-04 17:20 浏览: 2309232 次 我要评论(0 条) 字号:

【CSDN 编者按】七夕是中国民间的传统节日,不同时代、不同地域的人们给这个节日赋予了不同的含义。在漫长的演变过程中,七夕成了牛郎织女相会的日子。正因为这个美丽的爱情传说,七夕被视为中国最具浪漫色彩的、象征爱情的节日。在西风东渐的背景下,七夕又成了中国的“情人节”。

作者 | 许向武       责编 | 张红月
出品 | CSDN(ID:CSDNnews)

既然是情人节,浪漫的程序员总喜欢搞点花样出来,为紧张且平淡的生活添点色彩。今天给各位带来的是3D动画版的丘比特爱情之箭:一只金色的丘比特爱情之箭以慢镜头方式缓缓刺中一个脉动的红心,红心的主人将会无法控制地爱上TA所看到第一个人。

我猜,当初牛郎和织女一定是被丘比特的爱情之箭射中以后才相爱的,毕竟,他们都生活的天上——西方文化中,木星(Jupiter)就是小爱神丘比特,金星(Venus)则是大爱神维纳斯。

在丘比特的出场爱情之箭前,先用matplotlib绘制一个2D的红心热热身。

import numpy as npimport matplotlib.pyplot as plt
x = np.linspace(-np.pi/2, np.pi/2, 1000)y1 = np.power(np.cos(x),0.5) * np.cos(200*x) + np.power(np.absolute(x),0.5) - 0.7y2 = np.power(4-np.power(x, 2), 0.01)
plt.plot(x, y1*y2, c='r')plt.show()

上下波动的曲线形成的心形

放大一点看的话,就会发现上面这个红心其实是一条连续上下波动的曲线。下面的红心,是一个真正封闭的心形,只不过是由四条曲线首尾相连组成的。

import numpy as npimport matplotlib.pyplot as plt
x1 = np.linspace(0, 2, 300)x2 = np.linspace(0, -2, 300)
y11 = (np.power(x1, 2/3) + np.power(np.power(x1, 4/3)-4*np.power(x1, 2)+4, 0.5))/2 - 0.12y12 = (np.power(x1, 2/3) - np.power(np.power(x1, 4/3)-4*np.power(x1, 2)+4, 0.5))/2y21 = (np.power(-x2, 2/3) + np.power(np.power(-x2, 4/3)-4*np.power(-x2, 2)+4, 0.5))/2 - 0.12y22 = (np.power(-x2, 2/3) - np.power(np.power(-x2, 4/3)-4*np.power(-x2, 2)+4, 0.5))/2
plt.plot(x1, y11, c='r')plt.plot(x1, y12, c='r')plt.plot(x2, y21, c='r')plt.plot(x2, y22, c='r')plt.show()

四条曲线首尾相连组成的心形

那么,可用用一笔画一个心形吗?当然可以,极坐标方程 r = Arccos(sinθ)就是一条心形线,只是形状不够完美而已。

import numpy as npimport matplotlib.pyplot as plt
theta = np.linspace(0, 2*np.pi, 1000)x = np.arccos(np.sin(theta)) * np.cos(theta)y = np.arccos(np.sin(theta)) * np.sin(theta)
plt.plot(x, y, c='r')plt.show()

心形线

是时候展示3D版的红心了。运行下面的代码,除了需要numpy模块,还需要安装wxgl模块——基于PyOpenGL的三维数据绘图工具包。

pip install wxgl

关于wxgl模块的更多详情,请点击十分钟玩转3D绘图:WxGL完全手册,这里是中文文档

下面是3D版的红心的绘制代码:

import numpy as npimport wxgl.glplot as glt
a = np.linspace(0, 2*np.pi, 500)b = np.linspace(0.5*np.pi, -0.5*np.pi, 500)lons, lats = np.meshgrid(a, b)w = np.sqrt(np.abs(a - np.pi)) * 2x = 2 * np.cos(lats) * np.sin(lons) * wy = -2 * np.cos(lats) * np.cos(lons) * wz = 2 * np.sin(lats)
glt.mesh(x, y, z, color='crimson') # crimson - 绯红glt.show()

是不是很简单?wxgl的使用风格几乎和matplotlib完全一致。拖动鼠标,就可以看到这个3D版的红心的前后左右了。

3D红心

接下来,这段代码将红心数据的生成封装了一个函数,然后调用自定义的模型变换函数,这颗红心就跳动起来了。

import numpy as npimport wxgl.glplot as glt
def red_heart(r, slices=100, thick=2.0, shift=(0,0,0)): """返回红心顶点数据""" a = np.linspace(0, 2*np.pi, slices) b = np.linspace(0.5*np.pi, -0.5*np.pi, slices) lons, lats = np.meshgrid(a, b) w = np.sqrt(np.abs(a - np.pi)) * thick x = r * np.cos(lats) * np.sin(lons) * w + shift[0] y = -r * np.cos(lats) * np.cos(lons) * w + shift[1] z = r * np.sin(lats) + shift[2] return x, y, z
def heart_beat(t): """心跳函数""" t %= 1000 if t < 300: scale = 1 + t/3000 elif t > 700: scale = 1 + (1000-t)/3000 else: scale = 1.1 return (scale,)
glt.mesh(*red_heart(2), color='crimson', transform=heart_beat)glt.show()

点击左侧工具条上的播放按钮,这颗红心就会按照60次/秒的心率跳动起来。熟悉了wxgl的使用,不难读懂下面这个完整的丘比特爱情之箭的代码。

import numpy as npimport wxglimport wxgl.glplot as glt
def red_heart(r, slices=100, thick=2.0, shift=(0,0,0)): """返回红心顶点数据""" a = np.linspace(0, 2*np.pi, slices) b = np.linspace(0.5*np.pi, -0.5*np.pi, slices) lons, lats = np.meshgrid(a, b) w = np.sqrt(np.abs(a - np.pi)) * thick x = r * np.cos(lats) * np.sin(lons) * w + shift[0] y = -r * np.cos(lats) * np.cos(lons) * w + shift[1] z = r * np.sin(lats) + shift[2] return x, y, z
def heart_beat(t): """心跳函数""" t %= 1000 if t < 300: scale = 1 + t/3000 elif t > 700: scale = 1 + (1000-t)/3000 else: scale = 1.1 return (scale,)
def arrow_fly(t): """丘比特之箭飞行函数""" t %= 4000 if t > 2000: return ((0,-2,40-t/100),) else: return ((0,-2,0),)
def heart_fly(t): """丘比特之箭飞行函数""" t %= 4000 if t > 2000: return ((0,0,1,90), (0,-2,40-t/100)) else: return ((0,0,1,90), (0,-2,0))
glt.figure(azim=50, elev=16, style='gray') # 设置初始方位角、高度角、背景色glt.mesh(*red_heart(2), color='crimson', transform=heart_beat) # 绘制跳动的红心
x, y, z = red_heart(0.2, thick=3.0, shift=(0,-8,0)) # 生成心形箭头顶点数据glt.mesh(x, -z, y, color='crimson', transform=heart_fly) # 绘制心形箭头
light = wxgl.SunLight(roughness=0, metalness=0, shininess=0.5) # 箭杆灯光glt.cylinder((0,0,-8), (0,0,9), 0.1, color='goldenrod', transform=arrow_fly, light=light) # 绘制箭杆
vs = [(-1,1,11), (1,-1,11), (1,-1,6), (-1,1,6),(-1,-1,11), (1,1,11), (1,1,6), (-1,-1,6)] # 箭尾顶点数据texture = wxgl.Texture(r'resfeather.png') # 箭尾羽毛纹理texcoord = [(0,1), (0,0), (1,0), (1,1),(0,1), (0,0), (1,0), (1,1)] # 箭尾纹理坐标light = wxgl.BaseLight() # 箭尾灯光glt.quad(vs, texture=texture, texcoord=texcoord, transform=arrow_fly, light=light) # 绘制箭尾
glt.show()

wxgl的工具条提供了录制gif和MP4文件的功能,下面的gif就是使用wxgl自带的录制功能生成的。

代码中用到了羽毛的纹理,请下载这个图片,并保存到代码指定的位置。

— 推荐阅读 —
☞奖金池高达 20 万,RTE 2022 创新编程挑战赛正式开启
☞越来越火的图数据库到底能做什么?
☞“我用 400 行 Swift 代码给破旧的自行车加了一个动感单车计步器!”

— 活动推荐 —

不止七夕,让《新程序员》陪伴你每个朝夕!
凡在七夕当日订阅《新程序员》的朋友
都将获得CSDN赠送的定制马克杯一个!
扫描下方二维码立即订阅



网友评论已有0条评论, 我也要评论

发表评论

*

* (保密)

Ctrl+Enter 快捷回复