71 lines
2.0 KiB
C++
71 lines
2.0 KiB
C++
#include <iostream>
|
|
#include <map>
|
|
|
|
#define OLC_PGE_APPLICATION
|
|
#include "olcPixelGameEngine.h"
|
|
|
|
class Mandelbrot : public olc::PixelGameEngine {
|
|
public:
|
|
Mandelbrot() {
|
|
sAppName = "Mandelbrot set";
|
|
}
|
|
|
|
bool OnUserCreate() override {
|
|
return true;
|
|
}
|
|
|
|
bool OnUserUpdate(float fElapsedTime) override {
|
|
#pragma omp parallel for schedule(dynamic)
|
|
for (int pX = 0; pX < 1000; pX++) {
|
|
float mX = ((float)pX - 1000.0f / 2.0f) / 100 * m_scale - 1 + m_x;
|
|
#pragma omp parallel for schedule(dynamic)
|
|
for (int pY = 0; pY < 600; pY++) {
|
|
float mY = (600.0f / 2.0f - (float)pY) / 100 * m_scale - 1 + m_y;
|
|
|
|
float x = 0.0f; float y = 0.0f;
|
|
int iteration = 0;
|
|
while ((x * x) + (y * y) <= 2 * 2 && iteration < m_maxIterations) {
|
|
float tX = (x * x) - (y * y) + mX;
|
|
y = (2 * x * y) + mY;
|
|
x = tX;
|
|
iteration++;
|
|
}
|
|
olc::Pixel colour = olc::Pixel(iteration * 2, iteration * 2, iteration * 2);
|
|
DrawRect(pX, pY, 1, 1, colour);
|
|
}
|
|
}
|
|
// m_scale -= 0.01f;
|
|
|
|
if (GetKey(olc::Key::W).bHeld)
|
|
m_y *= 1.01f;
|
|
if (GetKey(olc::Key::S).bHeld)
|
|
m_y /= 1.01f;
|
|
if (GetKey(olc::Key::D).bHeld)
|
|
m_x *= 1.01f;
|
|
if (GetKey(olc::Key::A).bHeld)
|
|
m_x /= 1.01f;
|
|
if (GetKey(olc::Key::DOWN).bHeld)
|
|
m_scale *= 1.01f;
|
|
if (GetKey(olc::Key::UP).bHeld)
|
|
m_scale /= 1.01f;
|
|
|
|
std::cout << "Render time: " << fElapsedTime << std::endl;
|
|
return true;
|
|
}
|
|
private:
|
|
float m_x = 1.0f;
|
|
float m_y = 1.0f;
|
|
float m_scale = 0.7f;
|
|
int m_maxIterations = 100;
|
|
std::map<int, olc::Pixel> m_palete;
|
|
};
|
|
|
|
int main(int argc, char** argv) {
|
|
Mandelbrot app;
|
|
|
|
app.Construct(1000, 600, 1, 1);
|
|
app.Start();
|
|
|
|
return 0;
|
|
}
|