Path-Tracer/buffer.cpp at test · avin1cius/Path-Tracer · 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
#include "buffer.h"

// Creates a image buffer with default size (512 x 512 pixels)
Buffer::Buffer( void )
{
    // Allocate memory space for the image buffer
    buffer_data_.resize( h_resolution_, std::vector< glm::vec3 >( v_resolution_ ) );

    // Fills the image buffer with zeroes (black)
    for ( std::size_t y = 0; y < v_resolution_; y++ )
        for ( std::size_t x = 0; x < h_resolution_; x++ )
            buffer_data_[x][y] = glm::vec3{ 0.0f, 0.0f, 0.0f };
}

 // Creates a image buffer with a custom size (h_resolution x v_resolution pixels)
Buffer::Buffer( unsigned int h_resolution,
                unsigned int v_resolution ) :
        h_resolution_{ h_resolution },
        v_resolution_{ v_resolution }
{
    // Allocate memory space for the image buffer
    buffer_data_.resize( h_resolution_, std::vector< glm::vec3 >( v_resolution_ ) );

    // Fills the image buffer with zeroes (black)
    for ( std::size_t y = 0; y < v_resolution_; y++ )
        for ( std::size_t x = 0; x < h_resolution_; x++ )
            buffer_data_[x][y] = glm::vec3{ 0.0f, 0.0f, 0.0f };
}

Buffer::~Buffer( void )
{}

void Buffer::save( const std::string &filename ) const
{
    std::clog << "Buffer saving started... ";

    std::ofstream rendering_file;

    rendering_file.open( filename.c_str() );

    // Header of the PPM file. More on this filetype in:
    //     http://netpbm.sourceforge.net/doc/ppm.html
    rendering_file << "P3"
                   << std::endl
                   << h_resolution_
                   << " "
                   << v_resolution_
                   << std::endl
                   << 255
                   << std::endl;

    // Writes color data to the output PPM file
    for ( unsigned int y = 0; y < v_resolution_; y++ )
    {
        for ( unsigned int x = 0; x < h_resolution_; x++ )
        {
            // convert from real to [0.0, 1.0] with clamp, then
            //         from [0.0, 1.0] to [0.0, 255.0], then
            //         from [0.0, 255.0] to [0.5, 255.5], then
            //         from [0.5, 255.5] to [0, 255] with round to nearest.

            /*rendering_file << static_cast< int >( clamp( buffer_data_[x][y][0] ) * 255.0f + 0.5f ) << " ";
            rendering_file << static_cast< int >( clamp( buffer_data_[x][y][1] ) * 255.0f + 0.5f ) << " ";
            rendering_file << static_cast< int >( clamp( buffer_data_[x][y][2] ) * 255.0f + 0.5f ) << " ";*/

            //gama compression
            rendering_file << static_cast< int >( pow( clamp( buffer_data_[x][y][0] ), 1.0f/2.2f ) * 255 + 0.5f ) << " ";
            rendering_file << static_cast< int >( pow( clamp( buffer_data_[x][y][1] ), 1.0f/2.2f ) * 255 + 0.5f ) << " ";
            rendering_file << static_cast< int >( pow( clamp( buffer_data_[x][y][2] ), 1.0f/2.2f ) * 255 + 0.5f ) << " ";
        }
    }

    rendering_file.close();

    std::clog << "finished!\n";
}