eevee_sampling.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2021 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */

 
#pragma once
 
#include "BLI_vector.hh"
#include "DNA_scene_types.h"
 
#include "eevee_sampling_shared.hh"
#include "eevee_uniform_shared.hh"
 
namespace blender::eevee {
 
class Instance;
 
using SamplingDataBuf = draw::StorageBuffer<SamplingData>;
 
class Sampling {
 private:
  Instance &inst_;
 
  /* Number of samples in the first ring of jittered depth of field. */
  static constexpr uint64_t dof_web_density_ = 6;
  /* High number of sample for viewport infinite rendering. */
  static constexpr uint64_t infinite_sample_count_ = 0xFFFFFFu;
  /* During interactive rendering, loop over the first few samples. */
  static constexpr uint64_t interactive_sample_aa_ = 8;
  static constexpr uint64_t interactive_sample_raytrace_ = 32;
  static constexpr uint64_t interactive_sample_volume_ = 32;
  static constexpr uint64_t interactive_sample_max_ = interactive_sample_aa_ *
                                                      interactive_sample_raytrace_ *
                                                      interactive_sample_volume_;

  uint64_t sample_ = 0;
  uint64_t sample_count_ = 64;
  uint64_t dof_ring_count_ = 0;
  uint64_t dof_sample_count_ = 1;
  uint64_t motion_blur_steps_ = 1;
  int64_t viewport_sample_ = 0;
  bool reset_ = false;
  bool interactive_mode_ = false;
  static constexpr int interactive_mode_threshold = 3;
 
  SamplingDataBuf data_ = {"SamplingDataBuf"};
 
  ClampData &clamp_data_;
 
 public:
  Sampling(Instance &inst, ClampData &clamp_data) : inst_(inst), clamp_data_(clamp_data) {};
  ~Sampling() {};
 
  void init(const Scene *scene);
  void init(const Object &probe_object);
  void end_sync();
  void step();
 
  /* Viewport Only: Function to call to notify something in the scene changed.
   * This will reset accumulation. Do not call after end_sync() or during sample rendering. */
  void reset();
 
  /* Viewport Only: true if an update happened in the scene and accumulation needs reset. */
  bool is_reset() const;
 
  template<typename PassType> void bind_resources(PassType &pass)
  {
    pass.bind_ssbo(SAMPLING_BUF_SLOT, &data_);
  }
 
  /* Returns a pseudo random number in [0..1] range. Each dimension are de-correlated.
   * WARNING: Don't use during init or sync,
   * results are only valid during render, after step() has been called. */
  float rng_get(eSamplingDimension dimension) const
  {
    return data_.dimensions[dimension];
  }
 
  /* Returns a pseudo random number in [0..1] range. Each dimension are de-correlated.
   * WARNING: Don't use during init or sync,
   * results are only valid during render, after step() has been called. */
  float2 rng_2d_get(eSamplingDimension starting_dimension) const
  {
    return *reinterpret_cast<const float2 *>(&data_.dimensions[starting_dimension]);
  }
 
  /* Returns a pseudo random number in [0..1] range. Each dimension are de-correlated.
   * WARNING: Don't use during init or sync,
   * results are only valid during render, after step() has been called. */
  float3 rng_3d_get(eSamplingDimension starting_dimension) const
  {
    return *reinterpret_cast<const float3 *>(&data_.dimensions[starting_dimension]);
  }
 
  /* Returns true if rendering has finished. */
  bool finished() const
  {
    return (sample_ >= sample_count_);
  }
 
  /* Returns true if viewport smoothing and sampling has finished. */
  bool finished_viewport() const
  {
    return (viewport_sample_ >= sample_count_) && !interactive_mode_;
  }
 
  /* Returns true if viewport renderer is in interactive mode and should use TAA. */
  bool interactive_mode() const
  {
    return interactive_mode_;
  }
 
  /* Target sample count. */
  uint64_t sample_count() const
  {
    return sample_count_;
  }
 
  /* 0 based current sample. Might not increase sequentially in viewport. */
  uint64_t sample_index() const
  {
    return sample_;
  }
 
  bool use_clamp_direct() const
  {
    return clamp_data_.surface_direct != 0.0f;
  }
 
  bool use_clamp_indirect() const
  {
    return clamp_data_.surface_indirect != 0.0f;
  }
 
  /* Return true if we are starting a new motion blur step. We need to run sync again since
   * depsgraph was updated by MotionBlur::step(). */
  bool do_render_sync() const
  {
    return ((sample_ % (sample_count_ / motion_blur_steps_)) == 0);
  }

  static float3 sample_ball(const float3 &rand);

  static float2 sample_disk(const float2 &rand);

  static float3 sample_hemisphere(const float2 &rand);

  static float3 sample_sphere(const float2 &rand);

  static float2 sample_spiral(const float2 &rand);

  void dof_disk_sample_get(float *r_radius, float *r_theta) const;

  uint64_t dof_ring_count_get() const
  {
    return dof_ring_count_;
  }

  uint64_t dof_sample_count_get() const
  {
    return dof_sample_count_;
  }
 
  /* Cumulative Distribution Function Utils. */

  static void cdf_from_curvemapping(const CurveMapping &curve, Vector<float> &cdf);
  static void cdf_invert(Vector<float> &cdf, Vector<float> &inverted_cdf);
};
 
}  // namespace blender::eevee