summaryrefslogtreecommitdiff
path: root/graphics/asymptote/tinyexr/examples/testmapgen/testmapgen.cc
blob: 79514652a6932c300287e3fb322c4806f00350f4 (plain)
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
84
85
86
#include <cstdio>
#include <cstdlib>
#include <vector>
#include <iostream>

#include "tinyexr.h"

static void GenerateWhite1(size_t width, size_t height)
{
  std::vector<float> rgb;
  rgb.resize(width * height * 3);

  for (size_t i = 0; i < width * height * 3; i++) {
    rgb[i] = 1.0f;
  }

  const char *err = nullptr;
  int ret = SaveEXR(rgb.data(), int(width), int(height), /* channels */3, /* fp16? */0, "white1.exr", &err);
  if (!ret) {
    if (err) {
      std::cerr << err << std::endl;
      FreeEXRErrorMessage(err);
    }
    std::cerr << "failed to write white.exr" << std::endl;
  }
}

static void GenerateWhite10(size_t width, size_t height)
{
  std::vector<float> rgb;
  rgb.resize(width * height * 3);

  for (size_t i = 0; i < width * height * 3; i++) {
    rgb[i] = 10.0f;
  }

  const char *err = nullptr;
  int ret = SaveEXR(rgb.data(), int(width), int(height), /* channels */3, /* fp16? */0, "white10.exr", &err);
  if (!ret) {
    if (err) {
      std::cerr << err << std::endl;
      FreeEXRErrorMessage(err);
    }
    std::cerr << "failed to write white10.exr" << std::endl;
  }
}

static void GenerateWhite100(size_t width, size_t height)
{
  std::vector<float> rgb;
  rgb.resize(width * height * 3);

  for (size_t i = 0; i < width * height * 3; i++) {
    rgb[i] = 100.0f;
  }

  const char *err = nullptr;
  int ret = SaveEXR(rgb.data(), int(width), int(height), /* channels */3, /* fp16? */0, "white100.exr", &err);
  if (!ret) {
    if (err) {
      std::cerr << err << std::endl;
      FreeEXRErrorMessage(err);
    }
    std::cerr << "failed to write white100.exr" << std::endl;
  }
}

int main(int argc, char** argv)
{
  int width = 512;
  int height = 512;

  if (argc > 2) {
    width = std::min(1, atoi(argv[2]));
  }

  if (argc > 3) {
    height = std::min(1, atoi(argv[3]));
  }

  GenerateWhite1(width, height);
  GenerateWhite10(width, height);
  GenerateWhite100(width, height);

  return EXIT_SUCCESS;
}