src/openvic-simulation/utility/BMP.hpp


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

#pragma once

#include <cstdio>
#include <vector>

#include "openvic-simulation/types/Colour.hpp"

namespace OpenVic {
   class BMP {
#pragma pack(push)
#pragma pack(1)
      struct header_t {
         uint16_t signature;        // Signature: 0x4d42 (or 'B' 'M')
         uint32_t file_size;        // File size in bytes
         uint16_t reserved1;        // Not used
         uint16_t reserved2;        // Not used
         uint32_t offset;        // Offset to image data in bytes from beginning of file
         uint32_t dib_header_size;  // DIB header size in bytes
         int32_t width_px;       // Width of the image
         int32_t height_px;         // Height of image
         uint16_t num_planes;    // Number of colour planes
         uint16_t bits_per_pixel;   // Bits per pixel
         uint32_t compression;      // Compression type
         uint32_t image_size_bytes; // Image size in bytes
         int32_t x_resolution_ppm;  // Pixels per meter
         int32_t y_resolution_ppm;  // Pixels per meter
         uint32_t num_colours;      // Number of colours
         uint32_t important_colours; // Important colours
      } header;
#pragma pack(pop)

      FILE* file = nullptr;
      bool header_validated = false;
      uint32_t palette_size = 0;
      std::vector<colour_t> palette;

   public:
      static constexpr uint32_t PALETTE_COLOUR_SIZE = sizeof(colour_t);

      BMP() = default;
      ~BMP();

      bool open(char const* filepath);
      bool read_header();
      bool read_palette();
      void close();
      void reset();

      std::vector<colour_t> const& get_palette() const;
   };
}