使用python和pygame绘制繁花曲线的方法


Posted in Python onFebruary 24, 2018

前段时间看了一期《最强大脑》,里面各种繁花曲线组合成了非常美丽的图形,一时心血来潮,想尝试自己用代码绘制繁花曲线,想怎么组合就怎么组合。

真实的繁花曲线使用一种称为繁花曲线规的小玩意绘制,繁花曲线规由相互契合大小两个圆组成,用笔插在小圆上的一个孔中,紧贴大圆的内壁滚动,就可以绘制出漂亮的图案。这个过程可以做一个抽象:有两个半径不相等的圆,大圆位置固定,小圆在大圆内部,小圆紧贴着大圆内壁滚动,求小圆上的某一点走过的轨迹。

进一步分析,小圆的运动可以分解为两个部分:小圆圆心绕大圆圆心公转、小圆绕自身圆心自转。设大圆圆心为A,半径为Ra,小圆圆心为B,半径为Rb,轨迹点为C,半径为Rc(BC距离),设小圆公转的弧度为θ [0,∞),如图:

使用python和pygame绘制繁花曲线的方法

因为大圆的圆心坐标是固定的,要求得小圆上的某点的轨迹,需要先求出小圆当前时刻的圆心坐标,再求出小圆自转的弧度,最后求出小圆上某点的坐标。

第一步:求小圆圆心坐标

小圆圆心的公转轨迹是一个半径为 RA- RB 的圆,求小圆圆心坐标,相当于是求半径为 RA- RB 的圆上θ 弧度对应的点的坐标。

圆上的点的坐标公式为:

x = r * cos(θ), y = r * sin(θ)

小圆圆心坐标为:( xa+ (Ra - Rb) * cos(θ), ya + (Ra - Rb) * sin(θ) )

第二步:求小圆自转弧度

设小圆自转弧度为α,小圆紧贴大圆运动,两者走过的路程相同,因此有:

Ra *θ = Rb *α

小圆自转弧度α = (Ra / Rb) *θ

第三步:求点C坐标

点C相对小圆圆心B的公转轨迹是一个半径为 Rc 的圆,类似第一步,有:

轨迹点C的坐标为:( xa+ Rc* cos(θ), ya+ Rc* sin(θ))

按照以上算法分析,用python代码实现如下:

# -*- coding: utf-8 -*-

import math

'''
功能:
  已知圆的圆心和半径,获取某弧度对应的圆上点的坐标
入参:
  center:圆心
  radius:半径
  radian:弧度
'''
def get_point_in_circle(center, radius, radian):
  return (center[0] + radius * math.cos(radian), center[1] - radius * math.sin(radian))

'''
功能:
  内外圆A和B,内圆A沿着外圆B的内圈滚动,已知外圆圆心、半径,已知内圆半径,已知公转弧度和绕点半径,计算绕点坐标
入参:
  center_A:外圆圆心
  radius_A:外圆半径
  radius_B:内圆半径
  radius_C:绕点半径
  radian:公转弧度
'''
def get_point_in_child_circle(center_A, radius_A, radius_B, radius_C, radian):
  # 计算内圆圆心坐标
  center_B = get_point_in_circle(center_A, radius_A - radius_B, radian)
  # 计算绕点弧度(公转为逆时针,则自转为顺时针)
  radian_C = 2.0*math.pi - ((radius_A / radius_B * radian) % (2.0*math.pi))
  # 计算绕点坐标
  return get_point_in_circle(center_B, radius_C, radian_C)

有两点需要注意:

(1)屏幕坐标系左上角为原点,垂直向下为Y正轴,与数学坐标系Y轴方向相反,所以第14行Y坐标为减法;

(2)默认公转为逆时针,则自转为顺时针,所以第30行求自转弧度时,使用了2π - α%(2π);

坐标已经计算出来,接下来使用pygame绘制。思想是以0.01弧度为一个步长,不断计算出新的坐标,把一系列坐标连起来就会形成轨迹图。

为了能够形成一个封闭图形,还需要知道绘制点什么时候会重新回到起点。想了一个办法,以X轴正半轴为基准线,每次绘制点到达基准线,计算此时绘制点与起点的距离,达到一定精度认为已经回到起点,形成封闭图形。

''' 计算两点距离(平方和) '''
def get_instance(p1, p2):
  return (p1[0] - p2[0]) * (p1[0] - p2[0]) + (p1[1] - p2[1]) * (p1[1] - p2[1])
  
'''
功能:
  获取绕点路径的所有点的坐标
入参:
  center:外圆圆心
  radius_A:外圆半径
  radius_B:内圆半径
  radius_C:绕点半径
  shift_radian:每次偏移的弧度,默认0.01,值越小,精度越高,计算量越大
'''
def get_points(center, radius_A, radius_B, radius_C, shift_radian=0.01):
  # 转为实数
  radius_A *= 1.0
  radius_B *= 1.0
  radius_C *= 1.0
  
  P2 = 2*math.pi # 一圈的弧度为 2PI
  R_PER_ROUND = int(P2/shift_radian/4) + 1 # 一圈需要走多少步(弧度偏移多少次)
  
  # 第一圈的起点坐标
  start_point = get_point_in_child_circle(center, radius_A, radius_B, radius_C, 0)
  points = [start_point]
  # 第一圈的路径坐标
  for r in range(1, R_PER_ROUND):
    points.append(get_point_in_child_circle(center, radius_A, radius_B, radius_C, shift_radian*r))
  
  # 以圈为单位,每圈的起始弧度为 2PI*round,某圈的起点坐标与第一圈的起点坐标距离在一定范围内,认为路径结束
  for round in range(1, 100):
    s_radian = round*P2
    s_point = get_point_in_child_circle(center, radius_A, radius_B, radius_C, s_radian)
    if get_instance(s_point, start_point) < 0.1:
      break
    points.append(s_point)
    for r in range(1, R_PER_ROUND):
      points.append(get_point_in_child_circle(center, radius_A, radius_B, radius_C, s_radian + shift_radian*r))
    
  return points

再加上绘制代码,完整代码如下:

# -*- coding: utf-8 -*-
import math
import random

'''
功能:
  已知圆的圆心和半径,获取某弧度对应的圆上点的坐标
入参:
  center:圆心
  radius:半径
  radian:弧度
'''
def get_point_in_circle(center, radius, radian):
  return (center[0] + radius * math.cos(radian), center[1] - radius * math.sin(radian))

'''
功能:
  内外圆A和B,内圆A沿着外圆B的内圈滚动,已知外圆圆心、半径,已知内圆半径、公转弧度,已知绕点半径,计算绕点坐标
入参:
  center_A:外圆圆心
  radius_A:外圆半径
  radius_B:内圆半径
  radius_C:绕点半径
  radian:公转弧度
'''
def get_point_in_child_circle(center_A, radius_A, radius_B, radius_C, radian):
  # 计算内圆圆心坐标
  center_B = get_point_in_circle(center_A, radius_A - radius_B, radian)
  # 计算绕点弧度(公转为逆时针,则自转为顺时针)
  radian_C = 2.0*math.pi - ((radius_A / radius_B * radian) % (2.0*math.pi))
  # 计算绕点坐标
  center_C = get_point_in_circle(center_B, radius_C, radian_C)
  center_B_Int = (int(center_B[0]), int(center_B[1]))
  return center_B_Int, center_C

''' 计算两点距离(平方和) '''
def get_instance(p1, p2):
  return (p1[0] - p2[0]) * (p1[0] - p2[0]) + (p1[1] - p2[1]) * (p1[1] - p2[1])
  
'''
功能:
  获取绕点路径的所有点的坐标
入参:
  center:外圆圆心
  radius_A:外圆半径
  radius_B:内圆半径
  radius_C:绕点半径
  shift_radian:每次偏移的弧度,默认0.01,值越小,精度越高,计算量越大
'''
def get_points(center_A, radius_A, radius_B, radius_C, shift_radian=0.01):
  # 转为实数
  radius_A *= 1.0
  radius_B *= 1.0
  radius_C *= 1.0
  
  P2 = 2*math.pi # 一圈的弧度为 2PI
  R_PER_ROUND = int(P2/shift_radian) + 1 # 一圈需要走多少步(弧度偏移多少次)
  
  # 第一圈的起点坐标
  start_center, start_point = get_point_in_child_circle(center_A, radius_A, radius_B, radius_C, 0)
  points = [start_point]
  centers = [start_center]
  # 第一圈的路径坐标
  for r in range(1, R_PER_ROUND):
    center, point = get_point_in_child_circle(center_A, radius_A, radius_B, radius_C, shift_radian*r)
    points.append(point)
    centers.append(center)
  
  # 以圈为单位,每圈的起始弧度为 2PI*round,某圈的起点坐标与第一圈的起点坐标距离在一定范围内,认为路径结束
  for round in range(1, 100):
    s_radian = round*P2
    s_center, s_point = get_point_in_child_circle(center_A, radius_A, radius_B, radius_C, s_radian)
    if get_instance(s_point, start_point) < 0.1:
      break
    points.append(s_point)
    centers.append(s_center)
    for r in range(1, R_PER_ROUND):
      center, point = get_point_in_child_circle(center_A, radius_A, radius_B, radius_C, s_radian + shift_radian*r)
      points.append(point)
      centers.append(center)  
  print(len(points)/R_PER_ROUND)    
  return centers, points

import pygame
from pygame.locals import *

pygame.init()
screen = pygame.display.set_mode((600, 400))
clock = pygame.time.Clock()

color_black = (0, 0, 0)
color_white = (255, 255, 255)
color_red = (255, 0, 0)
color_yello = (255, 255, 0)

center = (300, 200)
radius_A = 150
radius_B = 110
radius_C = 50

test_centers, test_points = get_points(center, radius_A, radius_B, radius_C)
test_idx = 2
draw_point_num_per_tti = 5

while True:
  for event in pygame.event.get():
    if event.type==pygame.QUIT:
      pygame.quit() 
      exit(0)
  
  screen.fill(color_white)
  
  pygame.draw.circle(screen, color_black, center, int(radius_A), 2)
  
  if test_idx <= len(test_points):
    pygame.draw.aalines(screen, (0, 0, 255), False, test_points[:test_idx], 1)
    if test_idx < len(test_centers):
      pygame.draw.circle(screen, color_black, test_centers[test_idx], int(radius_B), 1)
      pygame.draw.aaline(screen, color_black, test_centers[test_idx], test_points[test_idx], 1)
    test_idx = min(test_idx + draw_point_num_per_tti, len(test_points))
  
  clock.tick(50)
  pygame.display.flip()

效果:

使用python和pygame绘制繁花曲线的方法

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python的urllib模块显示下载进度示例
Jan 17 Python
Python实现二叉搜索树
Feb 03 Python
Python实现一个转存纯真IP数据库的脚本分享
May 21 Python
解决pandas 作图无法显示中文的问题
May 24 Python
用python3教你任意Html主内容提取功能
Nov 05 Python
Python之列表实现栈的工作功能
Jan 28 Python
Python之NumPy(axis=0 与axis=1)区分详解
May 27 Python
python五子棋游戏的设计与实现
Jun 18 Python
Python将文字转成语音并读出来的实例详解
Jul 15 Python
解决Django后台ManyToManyField显示成Object的问题
Aug 09 Python
Django连接数据库并实现读写分离过程解析
Nov 13 Python
Python如何使用bokeh包和geojson数据绘制地图
Mar 21 Python
python3操作微信itchat实现发送图片
Feb 24 #Python
python自动12306抢票软件实现代码
Feb 24 #Python
浅谈Python中的zip()与*zip()函数详解
Feb 24 #Python
python模仿网页版微信发送消息功能
Feb 24 #Python
python2.7读取文件夹下所有文件名称及内容的方法
Feb 24 #Python
python opencv之分水岭算法示例
Feb 24 #Python
python3爬取各类天气信息
Feb 24 #Python
You might like
用PHP 4.2书写安全的脚本
2006/10/09 PHP
MySql中正则表达式的使用方法描述
2008/07/30 PHP
PHP 杂谈《重构-改善既有代码的设计》之三 重新组织数据
2012/04/09 PHP
php利用腾讯ip分享计划获取地理位置示例分享
2014/01/20 PHP
PHP 断点续传实例详解
2017/11/11 PHP
PHP yield关键字功能与用法分析
2019/01/03 PHP
PHP PDOStatement::fetchColumn讲解
2019/01/31 PHP
tp5框架基于ajax实现异步删除图片的方法示例
2020/02/10 PHP
php 使用html5 XHR2实现上传文件与进度显示功能示例
2020/03/03 PHP
js中的string.format函数代码
2020/08/11 Javascript
js实现页面跳转的几种方法小结
2016/05/16 Javascript
jQuery实现的右下角广告窗体跟随效果示例
2016/09/16 Javascript
Move.js入门
2017/02/08 Javascript
使用Bootstrap + Vue.js实现添加删除数据示例
2017/02/27 Javascript
Vue.js点击切换按钮改变内容的实例讲解
2018/08/22 Javascript
vue-cli脚手架的安装教程图解
2018/09/02 Javascript
推荐15个最好用的JavaScript代码压缩工具
2019/02/13 Javascript
vue两组件间值传递 $router.push实现方法
2019/05/15 Javascript
Layui选项卡制作历史浏览记录的方法
2019/09/28 Javascript
react实现移动端下拉菜单的示例代码
2020/01/16 Javascript
Vue中keep-alive组件作用详解
2020/02/04 Javascript
详解vue-template-admin三级路由无法缓存的解决方案
2020/03/10 Javascript
python抓取京东价格分析京东商品价格走势
2014/01/09 Python
用Python进行基础的函数式编程的教程
2015/03/31 Python
用Python实现换行符转换的脚本的教程
2015/04/16 Python
python 获取当天凌晨零点的时间戳方法
2018/05/22 Python
python爬取Ajax动态加载网页过程解析
2019/09/05 Python
pycharm实现在虚拟环境中引入别人的项目
2020/03/09 Python
python集合能干吗
2020/07/19 Python
Python Selenium库的基本使用教程
2021/01/04 Python
图书室管理制度
2014/01/19 职场文书
党的群众路线教育实践活动整改方案
2014/10/28 职场文书
审计局2014法制宣传日活动总结
2014/11/01 职场文书
同乡会致辞
2015/07/30 职场文书
Golang 实现获取当前函数名称和文件行号等操作
2021/05/08 Golang
Golang gRPC HTTP协议转换示例
2022/06/16 Golang