mirror of
https://github.com/Andreabont/OpenMandelbrot.git
synced 2024-09-19 14:03:05 +00:00
75 lines
2.3 KiB
C++
75 lines
2.3 KiB
C++
#ifndef LIBFUNCTIONS_H
|
|
#define LIBFUNCTIONS_H
|
|
|
|
#include <complex>
|
|
#include <SFML/Graphics.hpp>
|
|
|
|
typedef std::complex<double> (*fractal_function)(std::complex<double>, std::complex<double>);
|
|
|
|
template <fractal_function F>
|
|
int fractal_function_template(const std::complex<double>& point, const int& max_iterations) {
|
|
|
|
std::complex<double> z(0);
|
|
int iter = 0;
|
|
|
|
while (abs(z) < 2.0 && iter < max_iterations) {
|
|
z = F(z, point);
|
|
iter++;
|
|
}
|
|
|
|
return iter;
|
|
|
|
}
|
|
|
|
inline std::complex<double> fractal_mandelbrot(std::complex<double> z, std::complex<double> c) {
|
|
return (z * z) + c;
|
|
}
|
|
|
|
inline std::complex<double> fractal_triple_mandelbrot(std::complex<double> z, std::complex<double> c) {
|
|
return (z * z * z) + c;
|
|
}
|
|
|
|
inline std::complex<double> fractal_quadruple_mandelbrot(std::complex<double> z, std::complex<double> c) {
|
|
return (z * z * z * z) + c;
|
|
}
|
|
|
|
inline std::complex<double> fractal_quintuple_mandelbrot(std::complex<double> z, std::complex<double> c) {
|
|
return (z * z * z * z * z) + c;
|
|
}
|
|
|
|
inline sf::Color render_linear(int iteration_number, int max_iterations) {
|
|
int N = 256; // colors per element
|
|
int N3 = N * N * N;
|
|
double t = (double)iteration_number/(double)max_iterations;
|
|
// expand n on the 0 .. 256^3 interval (integers)
|
|
iteration_number = (int)(t * (double) N3);
|
|
|
|
int b = iteration_number/(N * N);
|
|
int nn = iteration_number - b * N * N;
|
|
int r = nn/N;
|
|
int g = nn - r * N;
|
|
return sf::Color(r, g, b);
|
|
}
|
|
|
|
inline sf::Color render_smooth(int iteration_number, int max_iterations) {
|
|
double t = (double)iteration_number/(double)max_iterations;
|
|
// Use smooth polynomials for r, g, b
|
|
int r = (int)(9*(1-t)*t*t*t*255);
|
|
int g = (int)(15*(1-t)*(1-t)*t*t*255);
|
|
int b = (int)(8.5*(1-t)*(1-t)*(1-t)*t*255);
|
|
return sf::Color(r, g, b);
|
|
}
|
|
|
|
inline sf::Color render_black_and_white(int iteration_number, int max_iterations) {
|
|
double t = (double)iteration_number/(double)max_iterations;
|
|
return (t != 1)? sf::Color(0, 0, 0) : sf::Color(255, 255, 255);
|
|
}
|
|
|
|
inline sf::Color render_green_gradient(int iteration_number, int max_iterations) {
|
|
double t = (double)iteration_number/(double)max_iterations;
|
|
if(t == 1) return sf::Color(0, 0, 0);
|
|
return (t > 0.5)? sf::Color(t*255, 255, t*255) : sf::Color(0, t*255, 0);
|
|
}
|
|
|
|
#endif //LIBFUNCTIONS_H
|