Fix #34700: cycles depth of field now works with orthographic cameras too.
[blender.git] / intern / cycles / kernel / kernel_camera.h
1 /*
2  * Copyright 2011, Blender Foundation.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  */
18
19 CCL_NAMESPACE_BEGIN
20
21 /* Perspective Camera */
22
23 __device float2 camera_sample_aperture(KernelGlobals *kg, float u, float v)
24 {
25         float blades = kernel_data.cam.blades;
26
27         if(blades == 0.0f) {
28                 /* sample disk */
29                 return concentric_sample_disk(u, v);
30         }
31         else {
32                 /* sample polygon */
33                 float rotation = kernel_data.cam.bladesrotation;
34                 return regular_polygon_sample(blades, rotation, u, v);
35         }
36 }
37
38 __device void camera_sample_perspective(KernelGlobals *kg, float raster_x, float raster_y, float lens_u, float lens_v, Ray *ray)
39 {
40         /* create ray form raster position */
41         Transform rastertocamera = kernel_data.cam.rastertocamera;
42         float3 Pcamera = transform_perspective(&rastertocamera, make_float3(raster_x, raster_y, 0.0f));
43
44         ray->P = make_float3(0.0f, 0.0f, 0.0f);
45         ray->D = Pcamera;
46
47         /* modify ray for depth of field */
48         float aperturesize = kernel_data.cam.aperturesize;
49
50         if(aperturesize > 0.0f) {
51                 /* sample point on aperture */
52                 float2 lensuv = camera_sample_aperture(kg, lens_u, lens_v)*aperturesize;
53
54                 /* compute point on plane of focus */
55                 float ft = kernel_data.cam.focaldistance/ray->D.z;
56                 float3 Pfocus = ray->P + ray->D*ft;
57
58                 /* update ray for effect of lens */
59                 ray->P = make_float3(lensuv.x, lensuv.y, 0.0f);
60                 ray->D = normalize(Pfocus - ray->P);
61         }
62
63         /* transform ray from camera to world */
64         Transform cameratoworld = kernel_data.cam.cameratoworld;
65
66 #ifdef __CAMERA_MOTION__
67         if(kernel_data.cam.have_motion)
68                 transform_motion_interpolate(&cameratoworld, (const DecompMotionTransform*)&kernel_data.cam.motion, ray->time);
69 #endif
70
71         ray->P = transform_point(&cameratoworld, ray->P);
72         ray->D = transform_direction(&cameratoworld, ray->D);
73         ray->D = normalize(ray->D);
74
75 #ifdef __RAY_DIFFERENTIALS__
76         /* ray differential */
77         float3 Ddiff = transform_direction(&cameratoworld, Pcamera);
78
79         ray->dP.dx = make_float3(0.0f, 0.0f, 0.0f);
80         ray->dP.dy = make_float3(0.0f, 0.0f, 0.0f);
81
82         ray->dD.dx = normalize(Ddiff + float4_to_float3(kernel_data.cam.dx)) - normalize(Ddiff);
83         ray->dD.dy = normalize(Ddiff + float4_to_float3(kernel_data.cam.dy)) - normalize(Ddiff);
84 #endif
85
86 #ifdef __CAMERA_CLIPPING__
87         /* clipping */
88         ray->P += kernel_data.cam.nearclip*ray->D;
89         ray->t = kernel_data.cam.cliplength;
90 #else
91         ray->t = FLT_MAX;
92 #endif
93 }
94
95 /* Orthographic Camera */
96
97 __device void camera_sample_orthographic(KernelGlobals *kg, float raster_x, float raster_y, float lens_u, float lens_v, Ray *ray)
98 {
99         /* create ray form raster position */
100         Transform rastertocamera = kernel_data.cam.rastertocamera;
101         float3 Pcamera = transform_perspective(&rastertocamera, make_float3(raster_x, raster_y, 0.0f));
102
103         ray->P = Pcamera;
104         ray->D = make_float3(0.0f, 0.0f, 1.0f);
105
106         /* modify ray for depth of field */
107         float aperturesize = kernel_data.cam.aperturesize;
108
109         if(aperturesize > 0.0f) {
110                 /* sample point on aperture */
111                 float2 lensuv = camera_sample_aperture(kg, lens_u, lens_v)*aperturesize;
112
113                 /* compute point on plane of focus */
114                 float ft = kernel_data.cam.focaldistance/ray->D.z;
115                 float3 Pfocus = ray->P + ray->D*ft;
116
117                 /* update ray for effect of lens */
118                 ray->P = make_float3(lensuv.x, lensuv.y, 0.0f);
119                 ray->D = normalize(Pfocus - ray->P);
120         }
121
122         /* transform ray from camera to world */
123         Transform cameratoworld = kernel_data.cam.cameratoworld;
124
125 #ifdef __CAMERA_MOTION__
126         if(kernel_data.cam.have_motion)
127                 transform_motion_interpolate(&cameratoworld, (const DecompMotionTransform*)&kernel_data.cam.motion, ray->time);
128 #endif
129
130         ray->P = transform_point(&cameratoworld, ray->P);
131         ray->D = transform_direction(&cameratoworld, ray->D);
132         ray->D = normalize(ray->D);
133
134 #ifdef __RAY_DIFFERENTIALS__
135         /* ray differential */
136         ray->dP.dx = float4_to_float3(kernel_data.cam.dx);
137         ray->dP.dy = float4_to_float3(kernel_data.cam.dy);
138
139         ray->dD.dx = make_float3(0.0f, 0.0f, 0.0f);
140         ray->dD.dy = make_float3(0.0f, 0.0f, 0.0f);
141 #endif
142
143 #ifdef __CAMERA_CLIPPING__
144         /* clipping */
145         ray->t = kernel_data.cam.cliplength;
146 #else
147         ray->t = FLT_MAX;
148 #endif
149 }
150
151 /* Panorama Camera */
152
153 __device void camera_sample_panorama(KernelGlobals *kg, float raster_x, float raster_y, float lens_u, float lens_v, Ray *ray)
154 {
155         Transform rastertocamera = kernel_data.cam.rastertocamera;
156         float3 Pcamera = transform_perspective(&rastertocamera, make_float3(raster_x, raster_y, 0.0f));
157
158         /* create ray form raster position */
159         ray->P = make_float3(0.0f, 0.0f, 0.0f);
160
161 #ifdef __CAMERA_CLIPPING__
162         /* clipping */
163         ray->t = kernel_data.cam.cliplength;
164 #else
165         ray->t = FLT_MAX;
166 #endif
167
168         ray->D = panorama_to_direction(kg, Pcamera.x, Pcamera.y);
169
170         /* modify ray for depth of field */
171         float aperturesize = kernel_data.cam.aperturesize;
172
173         if(aperturesize > 0.0f) {
174                 /* sample point on aperture */
175                 float2 lensuv = camera_sample_aperture(kg, lens_u, lens_v)*aperturesize;
176
177                 /* compute point on plane of focus */
178                 float3 D = normalize(ray->D);
179                 float3 Pfocus = D * kernel_data.cam.focaldistance;
180
181                 /* calculate orthonormal coordinates perpendicular to D */
182                 float3 U, V;
183                 make_orthonormals(D, &U, &V);
184
185                 /* update ray for effect of lens */
186                 ray->P = U * lensuv.x + V * lensuv.y;
187                 ray->D = normalize(Pfocus - ray->P);
188         }
189
190         /* indicates ray should not receive any light, outside of the lens */
191         if(len_squared(ray->D) == 0.0f) {
192                 ray->t = 0.0f;
193                 return;
194         }
195
196         /* transform ray from camera to world */
197         Transform cameratoworld = kernel_data.cam.cameratoworld;
198
199 #ifdef __CAMERA_MOTION__
200         if(kernel_data.cam.have_motion)
201                 transform_motion_interpolate(&cameratoworld, (const DecompMotionTransform*)&kernel_data.cam.motion, ray->time);
202 #endif
203
204         ray->P = transform_point(&cameratoworld, ray->P);
205         ray->D = transform_direction(&cameratoworld, ray->D);
206         ray->D = normalize(ray->D);
207
208 #ifdef __RAY_DIFFERENTIALS__
209         /* ray differential */
210         ray->dP.dx = make_float3(0.0f, 0.0f, 0.0f);
211         ray->dP.dy = make_float3(0.0f, 0.0f, 0.0f);
212
213         Pcamera = transform_perspective(&rastertocamera, make_float3(raster_x + 1.0f, raster_y, 0.0f));
214         ray->dD.dx = normalize(transform_direction(&cameratoworld, panorama_to_direction(kg, Pcamera.x, Pcamera.y))) - ray->D;
215
216         Pcamera = transform_perspective(&rastertocamera, make_float3(raster_x, raster_y + 1.0f, 0.0f));
217         ray->dD.dy = normalize(transform_direction(&cameratoworld, panorama_to_direction(kg, Pcamera.x, Pcamera.y))) - ray->D;
218 #endif
219 }
220
221 /* Common */
222
223 __device void camera_sample(KernelGlobals *kg, int x, int y, float filter_u, float filter_v,
224         float lens_u, float lens_v, float time, Ray *ray)
225 {
226         /* pixel filter */
227         float raster_x = x + kernel_tex_interp(__filter_table, filter_u, FILTER_TABLE_SIZE);
228         float raster_y = y + kernel_tex_interp(__filter_table, filter_v, FILTER_TABLE_SIZE);
229
230 #ifdef __CAMERA_MOTION__
231         /* motion blur */
232         if(kernel_data.cam.shuttertime == -1.0f)
233                 ray->time = TIME_INVALID;
234         else
235                 ray->time = 0.5f + 0.5f*(time - 0.5f)*kernel_data.cam.shuttertime;
236 #endif
237
238         /* sample */
239         if(kernel_data.cam.type == CAMERA_PERSPECTIVE)
240                 camera_sample_perspective(kg, raster_x, raster_y, lens_u, lens_v, ray);
241         else if(kernel_data.cam.type == CAMERA_ORTHOGRAPHIC)
242                 camera_sample_orthographic(kg, raster_x, raster_y, lens_u, lens_v, ray);
243         else
244                 camera_sample_panorama(kg, raster_x, raster_y, lens_u, lens_v, ray);
245 }
246
247 /* Utilities */
248
249 __device_inline float camera_distance(KernelGlobals *kg, float3 P)
250 {
251         Transform cameratoworld = kernel_data.cam.cameratoworld;
252         float3 camP = make_float3(cameratoworld.x.w, cameratoworld.y.w, cameratoworld.z.w);
253
254         if(kernel_data.cam.type == CAMERA_ORTHOGRAPHIC) {
255                 float3 camD = make_float3(cameratoworld.x.z, cameratoworld.y.z, cameratoworld.z.z);
256                 return fabsf(dot((P - camP), camD));
257         }
258         else
259                 return len(P - camP);
260 }
261
262 CCL_NAMESPACE_END
263