intern/cycles/kernel/osl/bsdf_refraction.cpp

   1 /*
   2  * Adapted from Open Shading Language with this license:
   3  *
   4  * Copyright (c) 2009-2010 Sony Pictures Imageworks Inc., et al.
   5  * All Rights Reserved.
   6  *
   7  * Modifications Copyright 2011, Blender Foundation.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions are
  11  * met:
  12  * * Redistributions of source code must retain the above copyright
  13  *   notice, this list of conditions and the following disclaimer.
  14  * * Redistributions in binary form must reproduce the above copyright
  15  *   notice, this list of conditions and the following disclaimer in the
  16  *   documentation and/or other materials provided with the distribution.
  17  * * Neither the name of Sony Pictures Imageworks nor the names of its
  18  *   contributors may be used to endorse or promote products derived from
  19  *   this software without specific prior written permission.
  20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  21  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  22  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  23  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  24  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  25  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  26  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  27  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  28  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  29  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  30  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31  */
  32
  33 #include <OpenImageIO/fmath.h>
  34
  35 #include <OSL/genclosure.h>
  36
  37 #include "osl_closures.h"
  38
  39 CCL_NAMESPACE_BEGIN
  40
  41 using namespace OSL;
  42
  43 class RefractionClosure : public BSDFClosure {
  44 public:
  45         Vec3 m_N;      // shading normal
  46         float m_eta;   // ratio of indices of refraction (inside / outside)
  47         RefractionClosure() : BSDFClosure(Labels::SINGULAR, Back) {}
  48
  49         void setup() {}
  50
  51         bool mergeable(const ClosurePrimitive *other) const {
  52                 const RefractionClosure *comp = (const RefractionClosure *)other;
  53                 return m_N == comp->m_N && m_eta == comp->m_eta &&
  54                        BSDFClosure::mergeable(other);
  55         }
  56
  57         size_t memsize() const { return sizeof(*this); }
  58
  59         const char *name() const { return "refraction"; }
  60
  61         void print_on(std::ostream &out) const {
  62                 out << name() << " (";
  63                 out << "(" << m_N[0] << ", " << m_N[1] << ", " << m_N[2] << "), ";
  64                 out << m_eta;
  65                 out << ")";
  66         }
  67
  68         Color3 eval_reflect(const Vec3 &omega_out, const Vec3 &omega_in, float& pdf) const
  69         {
  70                 return Color3(0, 0, 0);
  71         }
  72
  73         Color3 eval_transmit(const Vec3 &omega_out, const Vec3 &omega_in, float& pdf) const
  74         {
  75                 return Color3(0, 0, 0);
  76         }
  77
  78         float albedo(const Vec3 &omega_out) const
  79         {
  80                 return 1.0f;
  81         }
  82
  83         ustring sample(const Vec3 &Ng,
  84                        const Vec3 &omega_out, const Vec3 &domega_out_dx, const Vec3 &domega_out_dy,
  85                        float randu, float randv,
  86                        Vec3 &omega_in, Vec3 &domega_in_dx, Vec3 &domega_in_dy,
  87                        float &pdf, Color3 &eval) const
  88         {
  89                 Vec3 R, dRdx, dRdy;
  90                 Vec3 T, dTdx, dTdy;
  91                 bool inside;
  92
  93                 fresnel_dielectric(m_eta, m_N,
  94                                    omega_out, domega_out_dx, domega_out_dy,
  95                                    R, dRdx, dRdy,
  96                                    T, dTdx, dTdy,
  97                                    inside);
  98
  99                 if (!inside) {
 100                         pdf = 1;
 101                         eval.setValue(1.0f, 1.0f, 1.0f);
 102                         omega_in = T;
 103                         domega_in_dx = dTdx;
 104                         domega_in_dy = dTdy;
 105                 }
 106
 107                 return Labels::TRANSMIT;
 108         }
 109 };
 110
 111 ClosureParam bsdf_refraction_params[] = {
 112         CLOSURE_VECTOR_PARAM(RefractionClosure, m_N),
 113         CLOSURE_FLOAT_PARAM(RefractionClosure, m_eta),
 114         CLOSURE_STRING_KEYPARAM("label"),
 115         CLOSURE_FINISH_PARAM(RefractionClosure)
 116 };
 117
 118 CLOSURE_PREPARE(bsdf_refraction_prepare, RefractionClosure)
 119
 120 CCL_NAMESPACE_END
 121