Mandelbrot set

#include <iostream>
#include <fstream>
#include <vector>
#include <complex>
#include <chrono>
#include <omp.h> 
 
using namespace std;
 
typedef vector<int> Array1D;
typedef vector<Array1D> Array2D;
typedef vector<Array2D> Image;
 
// точка c пренадлежит Mandelbrot Set?
bool mandelbrot(complex<double> c, vector<int>& pixel) {
    int max_iteration = 500, iteration = 0;
    complex<double> z(0, 0);
 
    while ( abs(z) <= 4 && (iteration < max_iteration) ) {
        z = z * z + c;
        iteration++;
    }
 
    if (iteration != max_iteration) {
        // как вычислить эти значения, чтобы раскрасить выходные данные?
        
        //       (thread-id возможно?
        pixel = {255, 255, 255}; // ouside -> white 255,255,255
        return false;
    }
    
    pixel = {0, 0, 0}; // ouside -> white 255,255,255
    return true;
}
 
int main(int argc, char **argv)
{
    int width = 1200, height = 1200; 
 
    int i, j, pixels_inside=0;
 
    // Image data structure: 
    // - for each pixel we need red, green, and blue values (0-255)
    // -  используются 3 различных матрицы для соответствующих каналов
    int channels = 3; // red, green, blue
    Image image(channels, Array2D(height, Array1D(width)));
 
    vector<int> pixel = {0,0,0}; // red,green,blue (each range 0-255)
    complex<double> c; 
 
    
     auto t1 = omp_get_wtime(); //  time  OpenMP
 
    for (i = 0; i < height; i++) {
        for (j = 0; j < width; j++) {
            c = complex<double>( 2.0*((double)j/width-0.75), ((double)i/height-0.5)*2.0);
            
            if ( mandelbrot(c, pixel) ) pixels_inside++;
            
            // применить к изображению
            for (int ch = 0; ch < channels; ch++) 
                image[ch][i][j] = pixel[ch];
        }
    }
    
 
     auto t2 = omp_get_wtime(); //  time OpenMP
    
    // save image
    std::ofstream ofs("mandelbrot.ppm", std::ofstream::out);
    ofs << "P3" << std::endl;
    ofs << width << " " << height << std::endl;
    ofs << 255 << std::endl;
 
    for (int i = 0; i < height; i++) {
        for (int j = 0; j < width; j++) {
            ofs << " " << image[0][i][j] << " " << image[1][i][j] << " " << image[2][i][j] << std::endl;
        }
    }
    ofs.close();
 
    cout << "Total pixels inside: " << pixels_inside << endl;
    cout << "Execution time (without disk I/O)): " << chrono::duration<double>(t2 - t1).count() << endl;
 
    return 0;
}

#pragma omp parallel
{
    vector<int> pixel = {0,0,0}; // red,green,blue (each range 0-255)
    complex<double> c; 
 
#pragma omp for collapse(2)
    for (int i = 0; i < height; i++) 
    {
        for (int j = 0; j < width; j++) 
        {
            c = complex<double>(2.0 * ((double)j / width - 0.75), ((double)i / height - 0.5) * 2.0);
 
            if (mandelbrot(c, pixel))
            {
#pragma omp atomic
                pixels_inside++;
            }
 
            // применить к изображению
            for (int ch = 0; ch < channels; ch++)
                image[ch][i][j] = pixel[ch];
        }
    }
}

 if (iteration != max_iteration) {
        // как вычислить эти значения, чтобы раскрасить выходные данные?
        
        //       (thread-id возможно?
        pixel = {255, 255, 255}; // ouside -> white 255,255,255
        return false;
    }

#pragma omp parallel
{
   int pixel = omp_get_thread_num();
switch (pixel) {
  case 1:
   vector<int> pixel = {255,0,0};
    break;
  case 2:
     vector<int> pixel = {0,0,255};
    break;
  case 3:
     vector<int> pixel = {0,255,0};
    break;
}
    complex<double> c; 
 
#pragma omp for collapse(2)
    for (int i = 0; i < height; i++) 
    {
        for (int j = 0; j < width; j++) 
        {