e518f3675c8e7f973683b209240ad6c1221e4796
[blender.git] / intern / cycles / kernel / closure / bsdf_phong_ramp.h
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 2012, 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 #ifndef __BSDF_PHONG_RAMP_H__
34 #define __BSDF_PHONG_RAMP_H__
35
36 CCL_NAMESPACE_BEGIN
37
38 __device float3 bsdf_phong_ramp_get_color(const ShaderClosure *sc, const float3 colors[8], float pos)
39 {
40         int MAXCOLORS = 8;
41         
42         float npos = pos * (float)(MAXCOLORS - 1);
43         int ipos = (int)npos;
44         if (ipos >= (MAXCOLORS - 1))
45                 return colors[MAXCOLORS - 1];
46         float offset = npos - (float)ipos;
47         return colors[ipos] * (1.0f - offset) + colors[ipos+1] * offset;
48 }
49
50 __device int bsdf_phong_ramp_setup(ShaderClosure *sc)
51 {
52         sc->type = CLOSURE_BSDF_PHONG_RAMP_ID;
53         return SD_BSDF | SD_BSDF_HAS_EVAL | SD_BSDF_GLOSSY;
54 }
55
56 __device void bsdf_phong_ramp_blur(ShaderClosure *sc, float roughness)
57 {
58 }
59
60 __device float3 bsdf_phong_ramp_eval_reflect(const ShaderClosure *sc, const float3 colors[8], const float3 I, const float3 omega_in, float *pdf)
61 {
62         float m_exponent = sc->data0;
63         float cosNI = dot(sc->N, omega_in);
64         float cosNO = dot(sc->N, I);
65         
66         if (cosNI > 0 && cosNO > 0) {
67                 // reflect the view vector
68                 float3 R = (2 * cosNO) * sc->N - I;
69                 float cosRI = dot(R, omega_in);
70                 if (cosRI > 0) {
71                         float cosp = powf(cosRI, m_exponent);
72                         float common = 0.5f * M_1_PI_F * cosp;
73                         float out = cosNI * (m_exponent + 2) * common;
74                         *pdf = (m_exponent + 1) * common;
75                         return bsdf_phong_ramp_get_color(sc, colors, cosp) * out;
76                 }
77         }
78         
79         return make_float3(0.0f, 0.0f, 0.0f);
80 }
81
82 __device float3 bsdf_phong_ramp_eval_transmit(const ShaderClosure *sc, const float3 colors[8], const float3 I, const float3 omega_in, float *pdf)
83 {
84         return make_float3(0.0f, 0.0f, 0.0f);
85 }
86
87 __device int bsdf_phong_ramp_sample(const ShaderClosure *sc, const float3 colors[8], float3 Ng, float3 I, float3 dIdx, float3 dIdy, float randu, float randv, float3 *eval, float3 *omega_in, float3 *domega_in_dx, float3 *domega_in_dy, float *pdf)
88 {
89         float cosNO = dot(sc->N, I);
90         float m_exponent = sc->data0;
91         
92         if (cosNO > 0) {
93                 // reflect the view vector
94                 float3 R = (2 * cosNO) * sc->N - I;
95
96 #ifdef __RAY_DIFFERENTIALS__
97                 *domega_in_dx = (2 * dot(sc->N, dIdx)) * sc->N - dIdx;
98                 *domega_in_dy = (2 * dot(sc->N, dIdy)) * sc->N - dIdy;
99 #endif
100                 
101                 float3 T, B;
102                 make_orthonormals (R, &T, &B);
103                 float phi = M_2PI_F * randu;
104                 float cosTheta = powf(randv, 1 / (m_exponent + 1));
105                 float sinTheta2 = 1 - cosTheta * cosTheta;
106                 float sinTheta = sinTheta2 > 0 ? sqrtf(sinTheta2) : 0;
107                 *omega_in = (cosf(phi) * sinTheta) * T +
108                                    (sinf(phi) * sinTheta) * B +
109                                    (            cosTheta) * R;
110                 if (dot(Ng, *omega_in) > 0.0f)
111                 {
112                         // common terms for pdf and eval
113                         float cosNI = dot(sc->N, *omega_in);
114                         // make sure the direction we chose is still in the right hemisphere
115                         if (cosNI > 0)
116                         {
117                                 float cosp = powf(cosTheta, m_exponent);
118                                 float common = 0.5f * M_1_PI_F * cosp;
119                                 *pdf = (m_exponent + 1) * common;
120                                 float out = cosNI * (m_exponent + 2) * common;
121                                 *eval = bsdf_phong_ramp_get_color(sc, colors, cosp) * out;
122                         }
123                 }
124         }
125         return LABEL_REFLECT|LABEL_GLOSSY;
126 }
127
128
129 CCL_NAMESPACE_END
130
131 #endif /* __BSDF_PHONG_RAMP_H__ */