103 lines
2.7 KiB
C++
103 lines
2.7 KiB
C++
#include "include/constants.hpp"
|
|
#include "include/mandelbrot.hpp"
|
|
#include <iostream>
|
|
#include <raylib.h>
|
|
|
|
Color LerpColor(Color a, Color b, float t) {
|
|
return {(unsigned char)(a.r + (b.r - a.r) * t),
|
|
(unsigned char)(a.g + (b.g - a.g) * t),
|
|
(unsigned char)(a.b + (b.b - a.b) * t), 255};
|
|
}
|
|
|
|
void drawMandelbrot(double maxReal = 1.0, double minReal = -2.5,
|
|
double maxImag = 1.25, double minImag = -1.25) {
|
|
using namespace Constants;
|
|
|
|
double xNorm{};
|
|
double yNorm{};
|
|
double cReal{};
|
|
double cImag{};
|
|
|
|
for (int x{}; x <= windowWidth; ++x) {
|
|
for (int y{}; y <= windowHeight; ++y) {
|
|
xNorm = x / static_cast<double>(Constants::windowWidth);
|
|
yNorm = y / static_cast<double>(Constants::windowHeight);
|
|
|
|
cReal = minReal + xNorm * (maxReal - minReal);
|
|
cImag = minImag + yNorm * (maxImag - minImag);
|
|
|
|
int iterationsMandelbrot{calculateMandelbrot(cReal, cImag)};
|
|
|
|
if (iterationsMandelbrot - 1 == maxIterations) {
|
|
DrawPixel(x, y, BLACK);
|
|
} else {
|
|
float t = static_cast<float>(iterationsMandelbrot) / maxIterations;
|
|
Color color = LerpColor(BLUE, RED, t);
|
|
DrawPixel(x, y, color);
|
|
}
|
|
|
|
// std::cout << "Iterations: " << iterationsMandelbrot << '\n';
|
|
}
|
|
}
|
|
}
|
|
|
|
int main() {
|
|
double maxReal{1.0};
|
|
double minReal{-2.5};
|
|
double maxImag{1.25};
|
|
double minImag{-1.25};
|
|
InitWindow(Constants::windowWidth, Constants::windowHeight,
|
|
"Mandelbrot Plot");
|
|
SetTargetFPS(60);
|
|
|
|
RenderTexture2D target =
|
|
LoadRenderTexture(Constants::windowWidth, Constants::windowHeight);
|
|
bool needsRedraw = true;
|
|
|
|
while (!WindowShouldClose()) {
|
|
if (IsKeyPressed(KEY_UP)) {
|
|
maxImag += 0.1;
|
|
minImag += 0.1;
|
|
needsRedraw = true;
|
|
std::cout << "Runter!" << '\n';
|
|
}
|
|
if (IsKeyPressed(KEY_DOWN)) {
|
|
maxImag -= 0.1;
|
|
minImag -= 0.1;
|
|
needsRedraw = true;
|
|
std::cout << "Hoch!" << '\n';
|
|
}
|
|
if (IsKeyPressed(KEY_LEFT)) {
|
|
maxReal += 0.1;
|
|
minReal += 0.1;
|
|
needsRedraw = true;
|
|
std::cout << "Links!" << '\n';
|
|
}
|
|
if (IsKeyPressed(KEY_RIGHT)) {
|
|
maxReal -= 0.1;
|
|
minReal -= 0.1;
|
|
needsRedraw = true;
|
|
std::cout << "Rechts!" << '\n';
|
|
}
|
|
|
|
if (needsRedraw) {
|
|
BeginTextureMode(target);
|
|
ClearBackground(RAYWHITE);
|
|
drawMandelbrot(maxReal, minReal, maxImag, minImag);
|
|
EndTextureMode();
|
|
needsRedraw = false;
|
|
}
|
|
|
|
BeginDrawing();
|
|
DrawTextureRec(
|
|
target.texture,
|
|
{0, 0, (float)target.texture.width, -(float)target.texture.height},
|
|
{0, 0}, WHITE);
|
|
EndDrawing();
|
|
}
|
|
|
|
UnloadRenderTexture(target);
|
|
CloseWindow();
|
|
return 0;
|
|
}
|