56 lines
No EOL
1.7 KiB
C++
56 lines
No EOL
1.7 KiB
C++
#include <cstdio>
|
||
#include <cstdlib>
|
||
#include <cmath>
|
||
#include "complex.h"
|
||
using namespace std;
|
||
|
||
// В этой программе мы рисуем в картинку комплексную функцию,
|
||
// которая задаётся в функции func
|
||
|
||
struct Color {
|
||
unsigned char r, g, b;
|
||
};
|
||
|
||
Complex func(Complex z) {
|
||
Complex f = 100/(z - 1)*exp(z);
|
||
f.re = fabs(f.re);
|
||
f.im = fabs(f.im);
|
||
if (f.re > 255)
|
||
f.re = 255;
|
||
if (f.im > 255)
|
||
f.im = 255;
|
||
return f;
|
||
}
|
||
|
||
|
||
int main() {
|
||
int width = 800, height = 800;
|
||
float x0 = -2.0f, x1 = 2.0f;
|
||
float y0 = -2.0f, y1 = 2.0f;
|
||
|
||
// Выделяем память под пиксели
|
||
Color* data = (Color*)malloc(sizeof(Color) * width * height);
|
||
|
||
// data - это массив цветов размером width * height
|
||
// Задаём значения этого массива так, чтобы
|
||
// реальная часть функции func соответствовала зелёному цвету,
|
||
// а мнимая часть -- синей компоненте цвета
|
||
for (int j = 0; j < height; j++) {
|
||
for (int i = 0; i < width; i++) {
|
||
Complex z = {x0 + (x1-x0) / width * i, y0 + (y1-y0) / width * j};
|
||
Complex f = func(z);
|
||
data[i + width * j].r = 0;
|
||
data[i + width * j].g = f.re;
|
||
data[i + width * j].b = f.im;
|
||
}
|
||
}
|
||
|
||
// Сохраняем массив цветов data как картинку в формате .ppm
|
||
FILE* file = fopen("complex_image.ppm", "wb");
|
||
fprintf(file, "P6\n%d %d\n255\n", width, height);
|
||
fwrite(data, sizeof(Color), height * width, file);
|
||
fclose(file);
|
||
|
||
// Освобождаем память
|
||
free(data);
|
||
} |