VulkanCompute/main.c at main · pingpongcat/VulkanCompute · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#include <stdio.h>
#include <time.h>
#include "compute.h"
#include "instance.h"
#include "device.h"
#include "pipeline.h"
#include "memory.h"

 #define STB_IMAGE_WRITE_IMPLEMENTATION
 #include "stb_image_write.h"

 #define COLOR_CHANNELS 4

uint32_t InputData[1000];
float OutputData[1000][1000];

void generate(void)
{
    for (uint32_t row = 0; row < 1000; row++)
    {
        for (uint32_t column = 0; column < 1000; column++)
        {
            float r = (float)row / 500.0 - 1.0;
            float i = (float)column / 500.0 - 1.0;

            uint32_t cnt = 0;
            while ((r * r + i * i < 4.0) && (cnt < 63))
            {
                float temp = r*r - i * i + 0.17;
                i = 2 * r * i + 0.57;
                r = temp;
                cnt++;
            }
            OutputData[row][column] = (cnt << 10) | 0xff000000;

        }

    }

}

int main(int argc, char const *argv[])
{
    CreateInstance();
    GetPhysicalDevice();
    CreateDeviceAndComuteQueue();
    CreatePipeline();
    CreateDescriptorSet();
    CreateBuffers(sizeof(InputData), sizeof(OutputData));
    CreateCommandPool();
    PrepareCommandBuffer();

    CoppyToInputBuffer(InputData, sizeof(InputData));

    double time_spent = 0.0;
    clock_t begin = clock();

    generate();

    clock_t end = clock();
    time_spent = (double)(end - begin) / CLOCKS_PER_SEC;
    printf("Time of CPU generated fractal: %f s.\n", time_spent);

    stbi_write_bmp("fractactal_cpu.bmp", 1000, 1000, COLOR_CHANNELS, OutputData);

    begin = clock();

    Compute();
    CoppyFromOutputBuffer(OutputData, sizeof(OutputData));

    end = clock();
    time_spent = (double)(end - begin) / CLOCKS_PER_SEC;
    printf("Time of GPU generated fractal: %f s.\n", time_spent);


    stbi_write_bmp("fractactal_gpu.bmp", 1000, 1000, COLOR_CHANNELS, OutputData);

    DestroyPipeline();
    DestroyShaderModule();
    DestroyCommandPoolAndLogicalDevice();

    return 0;
}