115 lines
3.8 KiB
C
115 lines
3.8 KiB
C
#include "tricorn_simd.h"
|
|
#include <immintrin.h>
|
|
|
|
static unsigned converge(float c_rea, float c_imag, unsigned n);
|
|
|
|
void tricorn_simd(float complex start, size_t width, ssize_t height, float res, unsigned n, _Bool color, unsigned char* img){
|
|
int bytesPerColor = color ? 3 : 1;
|
|
|
|
// Absolute height for iteration purposes
|
|
ssize_t actual_height = height < 0 ? (ssize_t)(-height) : height;
|
|
ssize_t imag_step = height < 0 ? -1:1;
|
|
size_t simd_width = (width / 4) * 4;
|
|
|
|
float c_start_real = creal(start);
|
|
float c_start_imag = cimag(start);
|
|
|
|
__m128 four = _mm_set1_ps(4.0f);
|
|
__m128 minus_two = _mm_set1_ps(-2.0f);
|
|
|
|
for(ssize_t y = 0; y < actual_height;y++) {
|
|
float c_imag = c_start_imag - (y*imag_step) * res;
|
|
__m128 c_imag_vector = _mm_set1_ps(c_imag);
|
|
|
|
size_t x = 0;
|
|
for(; x < simd_width;x += 4) { //SIMD Loop
|
|
float c_real = c_start_real + (x*res);
|
|
__m128 c_real_vector = _mm_setr_ps(c_real,c_real+1.0f*res,c_real+2.0f*res,c_real+3.0f*res);
|
|
|
|
__m128 real = c_real_vector;
|
|
__m128 imag = c_imag_vector;
|
|
__m128i iterations = _mm_setzero_si128();
|
|
__m128i active_mask = _mm_set1_epi32(-1);
|
|
|
|
for(unsigned i = 0; i < n;i++) {
|
|
__m128 real2 = _mm_mul_ps(real,real);
|
|
__m128 imag2 = _mm_mul_ps(imag,imag);
|
|
|
|
__m128 unbounded = _mm_cmple_ps(_mm_add_ps(real2,imag2),four);
|
|
__m128i unbounded_i = _mm_castps_si128(unbounded);
|
|
|
|
active_mask = _mm_and_si128(active_mask,unbounded_i);
|
|
|
|
if(_mm_test_all_zeros(active_mask,active_mask)) {
|
|
break;
|
|
}
|
|
|
|
iterations = _mm_sub_epi32(iterations,active_mask);
|
|
|
|
imag = _mm_add_ps(c_imag_vector,_mm_mul_ps(minus_two, _mm_mul_ps(real,imag)));
|
|
real = _mm_add_ps(_mm_sub_ps(real2,imag2),c_real_vector);
|
|
}
|
|
|
|
|
|
__m128i mod64 = _mm_and_si128(iterations, _mm_set1_epi32(63));
|
|
__m128i times4 = _mm_slli_epi32(mod64, 2);
|
|
__m128i values = _mm_sub_epi32(_mm_set1_epi32(255), times4);
|
|
|
|
__m128i max_iter = _mm_set1_epi32(n);
|
|
__m128i is_max = _mm_cmpeq_epi32(iterations, max_iter);
|
|
values = _mm_andnot_si128(is_max, values);
|
|
|
|
unsigned int value_array[4];
|
|
_mm_storeu_si128((__m128i*)value_array,values);
|
|
|
|
for(unsigned i = 0 ; i < 4;i++) {
|
|
unsigned int value = value_array[i];
|
|
size_t position = (y*width + (x+i)) * bytesPerColor;
|
|
|
|
if(!color) {
|
|
img[position] = value;
|
|
}
|
|
else {
|
|
img[position] = ((value*3) % 255);
|
|
img[position + 1] = ((value*2) % 255);
|
|
img[position + 2] = ((value*5) % 255);
|
|
}
|
|
}
|
|
}
|
|
|
|
for(;x < width; x++) {
|
|
float c_real = c_start_real + (x*res);
|
|
unsigned int value = converge(c_real,c_imag,n);
|
|
|
|
size_t position = (y*width + x) * bytesPerColor;
|
|
|
|
if(!color) {
|
|
img[position] = value;
|
|
}
|
|
else {
|
|
img[position] = ((value*3) % 255);
|
|
img[position + 1] = ((value*2) % 255);
|
|
img[position + 2] = ((value*5) % 255);
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
static unsigned converge(float c_real, float c_imag, unsigned n) {
|
|
float real = 0.0f;
|
|
float imag = 0.0f;
|
|
|
|
for (unsigned i = 0; i < n; i++) {
|
|
float real2 = real * real;
|
|
float imag2 = imag * imag;
|
|
|
|
if (real2 + imag2 > 4.0f) {
|
|
return 255 - 4 * (i % 64);
|
|
}
|
|
|
|
imag = -2.0f * real * imag + c_imag;
|
|
real = real2 - imag2 + c_real;
|
|
}
|
|
return 0;
|
|
} |