File: Distribution2D.ispc

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 (48 lines) | stat: -rw-r--r-- 1,482 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
// Copyright 2009 Intel Corporation
// SPDX-License-Identifier: Apache-2.0

#include "Distribution1D.ih"
#include "Distribution2D.ih"

OSPRAY_BEGIN_ISPC_NAMESPACE

SYCL_EXTERNAL Sample2D Distribution2D_sample(
    const uniform Distribution2D *uniform self, const vec2f &s)
{
  // use u.y to sample a row
  const Sample1D sy = Distribution1D_sample(self->size.y, self->cdf_y, 0, s.y);

  // use u.x to sample inside the row
  const int x0 = sy.idx * self->size.x;
  const Sample1D sx = Distribution1D_sample(self->size.x, self->cdf_x, x0, s.x);

  Sample2D ret;
  ret.uv = make_vec2f((sx.idx + sx.frac) * self->rcpSize.x,
      (sy.idx + sy.frac) * self->rcpSize.y);
  ret.pdf = sx.pdf * sy.pdf;

  return ret;
}

SYCL_EXTERNAL float Distribution2D_pdf(
    const uniform Distribution2D *uniform self, const vec2f &uv)
{
  const int idy = clamp((int)(uv.y * self->size.y), 0, self->size.y - 1);
  const float befy = idy == 0 ? 0.0f : self->cdf_y[idy - 1];
  const float pdf_y = (self->cdf_y[idy] - befy) * self->size.y;

  const int idx = clamp((int)(uv.x * self->size.x), 0, self->size.x - 1);
  const int x0 = idy * self->size.x;
  const float befx = idx == 0 ? 0.0f : self->cdf_x[x0 + idx - 1];
  const float pdf_x = (self->cdf_x[x0 + idx] - befx) * self->size.x;

  return pdf_x * pdf_y;
}

SYCL_EXTERNAL void Distribution2D_init(Distribution2D *uniform self)
{
  // initialize the y distribution
  Distribution1D_create(self->size.y, self->cdf_y);
}

OSPRAY_END_ISPC_NAMESPACE