【C++】Mandelbrot集绘制(生成ppm文件)

时间:2024-01-16 22:43:14

曼德勃罗特集是人类有史以来做出的最奇异,最瑰丽的几何图形.曾被称为“上帝的指纹”。 这个点集均出自公式:Zn+1=(Zn)^2+C。(此处Z、C均为复数)所有使得该公式无限迭代后的结果能保持有限数值的复数C的集合,构成曼德勃罗集。

曼德勃罗集:

【C++】Mandelbrot集绘制(生成ppm文件)

看起来十分美丽和神秘,接下来就让我们用程序来绘制它。


在编写代码之前,我们先要了解这个图片中不同颜色所代表的含义。

首先注意到的自然是占面积最大的*黑色。黑色代表在迭代次数达到上限后其模仍小于某固定值(这里为2)的所有点。

周围的渐变色则是由点的发散速度决定的,即经过了多少次迭代,其模值开始大于2。

程序主体分为以下四个方面

1. 计算迭代次数,也就是它的发散速度。迭代次数越少,其发散速度越快。

int k=0;
for (; k < 1000; k++)
{
if (z.get_mod2() > 4.0) break;
z = z * z + c;
}

2. 对坐标系统进行一定的缩放来与显示区域的坐标系统相匹配.

假设显示高度为disp_height,宽度为disp_width.而点在显示区域中的位置为(x, y).如果显

示复数平面的这个窗口具有最小值(real_min, imag_min)和最大值(real_max, imag_max),则每

个点需用以下系数加以缩放.

c.real = real_min + x * (real_max - real_min) / disp_width;
c.imag = imag_min + y * (imag_max - imag_min) / disp_height;

3. 配色

为了使得绘制出的图形更漂亮,一个好的配色方案是少不了的。这里我使用的是以下函数生成的配色:

void InitColor()
{
// 使用 HSL 颜色模式产生角度 h1 到 h2 的渐变色
int h1 = 240, h2 = 30;
for (int i = 0; i<MAXCOLOR / 2; i++)
{
Color[i] = HSLtoRGB((float)h1, 1.0f, i * 2.0f / MAXCOLOR);
Color[MAXCOLOR - 1 - i] = HSLtoRGB((float)h2, 1.0f, i * 2.0f / MAXCOLOR);
}
}

4. ppm文件的格式

文件头部分

image << "P6\n"  //PPM magic number
<< width << ' ' //width, in ASCII decimal
<< height << '\n' //height, in ASCII decimal
<< 255<<'\n'; //maximum color value, in ASCII decimal

第一行是P2/P3/P6

第二行是图像的大小,先是行像素数,后是列像素数,中间有一个空格

第三行是一个介于1和65535之间的整数,而且必须是文本的,用来表示每一个像素的一个分量用几个比特表示。

图象数据部分

P6格式,图象数据以字节格式存储,每个色彩成分(R,G,B)一个字节,3个字节为一个像素点。

参考代码:

#include<iostream>
#include<string>
#include<fstream>
using namespace std; struct pixel //单个像素
{
char r, g, b;
}; class Complex
{
public:
Complex(double real = 0, double imag = 0) :real(real), imag(imag){}
Complex(Complex& com) :real(com.real), imag(com.imag){}
double& operator[](int i){ return i == 0 ? real : imag; }
double get_mod2(){ return pow(real, 2) + pow(imag, 2); }
Complex& operator=(Complex& com){ real = com.real; imag = com.imag; return *this; }
Complex operator+(Complex& com){ return Complex(real + com.real, imag + com.imag); }
Complex operator*(Complex& com){ return Complex(real*com.real - imag*com.imag, real*com.imag + imag*com.real); } private:
double real, imag;
}; class Mandelbrot
{
public:
Mandelbrot(int _width=640, int _height=480)
:path(path), width(_width), height(_height){}
~Mandelbrot(){}
public:
void set_path(string path)
{
this->path = path;
}
void draw(double from_x,double to_x,double from_y,double to_y) //图片保存路径
{
create_image();
write_head(); for (int y = 0; y < height; y++)
{
Complex c(0, from_y + (to_y - from_y) * y / (double)height);
for (int x = 0; x < width; x++)
{
c[0] = from_x + (to_x - from_x) * x / (double)width;
Complex z(0, 0);
int k;
for (k = 0; k < 1000; k++)
{
if (z.get_mod2() > 4.0) break;
z = z * z + c;
}
pixel p = { Color[k % 64].b, Color[k % 64].g,
Color[k % 64].r
};
pixel q = { 0, 0, 0 };
if (k == 1000) //收敛
{
image.write((char*)&q,sizeof(q));
}
else //发散
{
image.write((char*)&p,sizeof(p));
}
}
}
image.close();
}
void get_color(string path)
{
color.open(path,ios::binary);
for (int i = 0; i < 64; ++i)
{
color.read((char*)&Color[i], sizeof(pixel));
}
color.close();
}
int get_fileSize(){ return height*width*sizeof(pixel) / 1024; } //KB
private:
ofstream image;
ifstream color;
void create_image()
{
image.open(path,ios::binary);
}
void write_head() //文件头
{
image << "P6\n"
<< "#Mandelbrot Set\n"
<< width << ' '
<< height << '\n'
<< 255<<'\n';
} int width, height; //图片尺寸
pixel Color[64];
string path;
}; int main()
{ Mandelbrot test;
test.set_path("Mandelbrot.ppm");
test.get_color("color.dat"); //从文件中读取颜色数据
cout << test.get_fileSize()<<"KB";
test.draw(-2.1, 1.1, -1.2, 1.2); return 0;
}

---参考自:http://codebus.easyx.cn/yw80/post/zoomable-mandelbrot-set