#include <stdio.h>
#include <stdlib.h>
#include <limits.h>
#include <cuda_runtime.h>
#include <device_launch_parameters.h>

#define BLOCK_SIZE 16
#define MATRIX_SIZE 32
#define OBSTACLE_PROB 10
#define INF UINT_MAX // Используем беззнаковый максимум

__global__ void wave_step(int* P, unsigned int* dist, int n, bool* changed) {
    int i = blockIdx.y * blockDim.y + threadIdx.y;
    int j = blockIdx.x * blockDim.x + threadIdx.x;
    int idx = i * n + j;

    if (i >= n || j >= n || P[idx] == -1) return;

    unsigned int current_dist = dist[idx];
    unsigned int min_dist = current_dist;

    // Проверка соседей с защитой от переполнения
    if (i > 0 && dist[(i-1)*n + j] != INF) 
        min_dist = min(min_dist, dist[(i-1)*n + j] + 1);
    
    if (i < n-1 && dist[(i+1)*n + j] != INF) 
        min_dist = min(min_dist, dist[(i+1)*n + j] + 1);
    
    if (j > 0 && dist[i*n + (j-1)] != INF) 
        min_dist = min(min_dist, dist[i*n + (j-1)] + 1);
    
    if (j < n-1 && dist[i*n + (j+1)] != INF) 
        min_dist = min(min_dist, dist[i*n + (j+1)] + 1);

    if (min_dist < current_dist) {
        atomicMin(&dist[idx], min_dist);
        *changed = true;
    }
}

void generate_polygon(int* P, int n) {
    srand(42);
    for (int i = 0; i < n*n; i++) {
        P[i] = (rand() % 100 < OBSTACLE_PROB) ? -1 : 0;
    }
    P[0] = 0;       // Гарантируем, что старт свободен
    P[n*n - 1] = 0; // Гарантируем, что финиш свободен
}

void print_distance_map(int* P, unsigned int* dist, int n) {
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            int idx = i * n + j;
            if (P[idx] == -1) 
                printf("████"); // Препятствие
            else if (i == 0 && j == 0) 
                printf("S   "); // Старт
            else if (i == n-1 && j == n-1) 
                printf("F   "); // Финиш
            else if (dist[idx] == INF) 
                printf(".   "); // Недостижимая область
            else 
                printf("%-4u", dist[idx]); // Вывод расстояния
        }
        printf("\n");
    }
}

int main() {
    const int n = MATRIX_SIZE;
    const int block_size = BLOCK_SIZE;

    // Инициализация полигона
    int* P = (int*)malloc(n * n * sizeof(int));
    generate_polygon(P, n);

    // Выделение памяти на GPU
    int* d_P;
    unsigned int* d_dist;
    bool* d_changed;
    cudaMalloc(&d_P, n*n*sizeof(int));
    cudaMalloc(&d_dist, n*n*sizeof(unsigned int));
    cudaMalloc(&d_changed, sizeof(bool));

    // Инициализация расстояний
    unsigned int* dist = (unsigned int*)malloc(n*n*sizeof(unsigned int));
    for (int i = 0; i < n*n; i++) dist[i] = INF;
    dist[0] = 0; // Стартовая точка

    // Копирование данных на GPU
    cudaMemcpy(d_P, P, n*n*sizeof(int), cudaMemcpyHostToDevice);
    cudaMemcpy(d_dist, dist, n*n*sizeof(unsigned int), cudaMemcpyHostToDevice);

    // Настройка запуска ядра
    dim3 grid((n + block_size - 1)/block_size, (n + block_size - 1)/block_size);
    dim3 block(block_size, block_size);

    // Замер времени
    cudaEvent_t start, stop;
    cudaEventCreate(&start);
    cudaEventCreate(&stop);
    cudaEventRecord(start);

    // Основной цикл волны
    int iterations = 0;
    bool changed;
    do {
        changed = false;
        cudaMemcpy(d_changed, &changed, sizeof(bool), cudaMemcpyHostToDevice);
        wave_step<<<grid, block>>>(d_P, d_dist, n, d_changed);
        cudaDeviceSynchronize(); // Синхронизация после ядра
        cudaMemcpy(&changed, d_changed, sizeof(bool), cudaMemcpyDeviceToHost);
        iterations++;
    } while (changed && iterations < 2*n); // Защита от бесконечного цикла

    // Финализация времени
    cudaEventRecord(stop);
    cudaEventSynchronize(stop);
    float milliseconds = 0;
    cudaEventElapsedTime(&milliseconds, start, stop);

    // Проверка результата
    cudaMemcpy(dist, d_dist, n*n*sizeof(unsigned int), cudaMemcpyDeviceToHost);
    if (dist[n*n - 1] == INF) {
        printf("Path not found!\n");
    } else {
        printf("Success! Path length: %u\n", dist[n*n - 1]);
    }

    printf("Time: %.2f ms\n", milliseconds);
    printf("Matrix: %dx%d | Blocks: %dx%d | Obstacles: %d%%\n\n", 
           n, n, block_size, block_size, OBSTACLE_PROB);

    if (MATRIX_SIZE < 100)
        print_distance_map(P, dist, MATRIX_SIZE);

    // Освобождение памяти
    free(P);
    free(dist);
    cudaFree(d_P);
    cudaFree(d_dist);
    cudaFree(d_changed);
    cudaEventDestroy(start);
    cudaEventDestroy(stop);

    return 0;
}