File: Texture2D.cpp

package info (click to toggle)
ospray 3.2.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 10,048 kB
  • sloc: cpp: 80,569; ansic: 951; sh: 805; makefile: 170; python: 69
file content (148 lines) | stat: -rw-r--r-- 4,746 bytes parent folder | download | duplicates (2)
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
// Copyright 2009 Intel Corporation
// SPDX-License-Identifier: Apache-2.0

#include "Texture2D.h"
#ifndef OSPRAY_TARGET_SYCL
#include "texture/Texture2D_ispc.h"
#endif
#include "texture/MipMapGeneration_ispc.h"

namespace ispc {

void Texture2D::set(const rkcommon::math::vec2i &aSize,
    void **aData,
    int aMaxLevel,
    OSPTextureFormat aFormat,
    OSPTextureFilter aFilter,
    const rkcommon::math::vec2ui &aWrapMode)
{
  size = aSize;
  maxLevel = aMaxLevel;
  format = aFormat;
  filter = aFilter;
  wrapMode = aWrapMode;

  for (int l = 0; l <= aMaxLevel; l++)
    data[l] = aData[l];

  super.hasAlpha = aFormat == OSP_TEXTURE_RGBA8 || aFormat == OSP_TEXTURE_SRGBA
      || aFormat == OSP_TEXTURE_RA8 || aFormat == OSP_TEXTURE_LA8
      || aFormat == OSP_TEXTURE_RGBA32F || aFormat == OSP_TEXTURE_RGBA16
      || aFormat == OSP_TEXTURE_RA16 || aFormat == OSP_TEXTURE_RA16F
      || aFormat == OSP_TEXTURE_RA32F || aFormat == OSP_TEXTURE_RGBA16F;
#ifndef OSPRAY_TARGET_SYCL
  super.get =
      reinterpret_cast<ispc::Texture_get>(ispc::Texture2D_get_addr(aFormat));
  super.getNormal =
      reinterpret_cast<ispc::Texture_getN>(ispc::Texture2D_getN_addr(aFormat));
#endif
}

} // namespace ispc

namespace ospray {

Texture2D::~Texture2D()
{
  // If no one else is referencing the MIP map buffer (just this object and the
  // cache map), we need to remove it from the MIP map cache so the buffer will
  // be deleted as well
  if (mipMapData && mipMapData.use_count() == 2)
    getISPCDevice().getMipMapCache().remove(texData->data());
}

std::string Texture2D::toString() const
{
  return "ospray::Texture2D";
}

void Texture2D::commit()
{
  texData = getParamObject<Data>("data");

  if (!texData || texData->numItems.z > 1) {
    throw std::runtime_error(toString()
        + " must have 2D 'data' array using the first two dimensions.");
  }

  if (texData->size() > std::numeric_limits<std::uint32_t>::max())
    throw std::runtime_error(toString()
        + " too large (over 4B texels, indexing is limited to 32bit");

  const vec2i size = vec2i(texData->numItems.x, texData->numItems.y);
  if (!texData->compact()) {
    postStatusMsg(OSP_LOG_INFO)
        << toString()
        << " does currently not support strides, copying texture data.";

    auto data = new Data(getISPCDevice(), texData->type, texData->numItems);
    data->copy(*texData, vec3ui(0));
    texData = data;
    data->refDec();
  }

  format = static_cast<OSPTextureFormat>(
      getParam<uint32_t>("format", OSP_TEXTURE_FORMAT_INVALID));
  filter = static_cast<OSPTextureFilter>(
      getParam<uint32_t>("filter", OSP_TEXTURE_FILTER_LINEAR));
  wrapMode = getParam<vec2ui>("wrapMode",
      vec2ui(getParam<uint32_t>("wrapMode", OSP_TEXTURE_WRAP_REPEAT)));

  if (format == OSP_TEXTURE_FORMAT_INVALID)
    throw std::runtime_error(toString() + ": invalid 'format'");

  if (sizeOf(format) != sizeOf(texData->type))
    throw std::runtime_error(toString() + ": 'format'='" + stringFor(format)
        + "' does not match type of 'data'='" + stringFor(texData->type)
        + "'!");

  std::vector<void *> dataPtr;
  dataPtr.push_back(texData->data());

  if (getISPCDevice().disableMipMapGeneration)
    mipMapData = nullptr;
  else {
    // Check if MIP maps were generated already
    bool generateMipMaps = false;
    MipMapCache &mmc = getISPCDevice().getMipMapCache();
    mipMapData = mmc.find(texData->data());
    if (!mipMapData) {
      // conservatively estimate needed space for MIP maps and allocate
      const auto maxMipTexels = (size.product() + 2) / 3 // for square textures
          + (size.x + size.y - 1)
              / std::min(size.x, size.y); // remaining 1D case
      mipMapData = std::make_shared<devicert::BufferShared<char>>(
          getISPCDevice().getDRTDevice(), maxMipTexels * sizeOf(format));

      // Add generated MIP maps to the cache
      mmc.add(texData->data(), mipMapData);
      generateMipMaps = true;
    }

    // generate MIP map levels
    vec2i srcSize = size;
    void *srcData = texData->data();
    char *dstData = (char *)mipMapData->data();
    while (srcSize != vec2i(1)) {
      assert(dataPtr.size() <= MAX_MIPMAP_LEVEL); // max tex size is 2^32
      vec2i dstSize(max(srcSize / 2, vec2i(1)));
      if (generateMipMaps)
        ispc::MipMap_generate(srcData,
            (ispc::vec2i &)srcSize,
            dstData,
            (ispc::vec2i &)dstSize,
            format);

      dataPtr.push_back(dstData);
      srcSize = dstSize;
      srcData = dstData;
      dstData += dstSize.product() * ospray::sizeOf(format);
    }
  }

  // Initialize ispc shared structure
  getSh()->set(
      size, dataPtr.data(), dataPtr.size() - 1, format, filter, wrapMode);
}

} // namespace ospray