Merging r50625 through r51896 from trunk into soc-2011-tomato
[blender.git] / intern / cycles / kernel / shaders / oslutil.h
1 /////////////////////////////////////////////////////////////////////////////
2 // Copyright (c) 2009-2010 Sony Pictures Imageworks Inc., et al.  All Rights Reserved.
3 //
4 // Redistribution and use in source and binary forms, with or without
5 // modification, are permitted provided that the following conditions are
6 // met:
7 // * Redistributions of source code must retain the above copyright
8 //   notice, this list of conditions and the following disclaimer.
9 // * Redistributions in binary form must reproduce the above copyright
10 //   notice, this list of conditions and the following disclaimer in the
11 //   documentation and/or other materials provided with the distribution.
12 // * Neither the name of Sony Pictures Imageworks nor the names of its
13 //   contributors may be used to endorse or promote products derived from
14 //   this software without specific prior written permission.
15 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
18 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
19 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
21 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
25 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 /////////////////////////////////////////////////////////////////////////////
27
28
29 #ifndef CCL_OSLUTIL_H
30 #define CCL_OSLUTIL_H
31
32 /* NB: must match the value in kernel_types.h */
33 #define RAMP_TABLE_SIZE 256
34
35 // Return wireframe opacity factor [0, 1] given a geometry type in
36 // ("triangles", "polygons" or "patches"), and a line_width in raster
37 // or world space depending on the last (raster) boolean argument.
38 //
39 float wireframe(string edge_type, float line_width, int raster)
40 {
41    // ray differentials are so big in diffuse context that this function would always return "wire"
42    if (raytype("path:diffuse")) return 0.0;
43
44    int np = 0;
45    point p[64];
46    float pixelWidth = 1;
47
48    if (edge_type == "triangles")
49    {
50       np = 3;
51       if (!getattribute("geom:trianglevertices", p))
52          return 0.0;
53    }
54    else if (edge_type == "polygons" || edge_type == "patches")
55    {
56       getattribute("geom:numpolyvertices", np);
57       if (np < 3 || !getattribute("geom:polyvertices", p))
58          return 0.0;
59    }
60
61    if (raster)
62    {
63       // Project the derivatives of P to the viewing plane defined
64       // by I so we have a measure of how big is a pixel at this point
65       float pixelWidthX = length(Dx(P) - dot(Dx(P), I) * I);
66       float pixelWidthY = length(Dy(P) - dot(Dy(P), I) * I);
67       // Take the average of both axis' length
68       pixelWidth = (pixelWidthX + pixelWidthY) / 2;
69    }
70
71    // Use half the width as the neighbor face will render the
72    // other half. And take the square for fast comparison
73    pixelWidth *= 0.5 * line_width;
74    pixelWidth *= pixelWidth;
75    for (int i = 0; i < np; i++)
76    {
77       int i2 = i ? i - 1 : np - 1;
78       vector dir = P - p[i];
79       vector edge = p[i] - p[i2];
80       vector crs = cross(edge, dir);
81       // At this point dot(crs, crs) / dot(edge, edge) is
82       // the square of area / length(edge) == square of the
83       // distance to the edge.
84       if (dot(crs, crs) < (dot(edge, edge) * pixelWidth))
85          return 1;
86    }
87    return 0;
88 }
89
90 float wireframe(string edge_type, float line_width) { return wireframe(edge_type, line_width, 1); }
91 float wireframe(string edge_type) { return wireframe(edge_type, 1.0, 1); }
92 float wireframe() { return wireframe("polygons", 1.0, 1); }
93
94 #endif /* CCL_OSLUTIL_H */