4 * ***** BEGIN GPL LICENSE BLOCK *****
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version 2
9 * of the License, or (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software Foundation,
18 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
20 * The Original Code is Copyright (C) 2004 by Blender Foundation
21 * All rights reserved.
23 * The Original Code is: all of this file.
25 * Contributor(s): none yet.
27 * ***** END GPL LICENSE BLOCK *****
31 meshtools.c: no editmode (violated already :), tools operating on meshes
36 #include "MEM_guardedalloc.h"
38 #include "DNA_image_types.h"
39 #include "DNA_scene_types.h"
40 #include "DNA_screen_types.h"
41 #include "DNA_space_types.h"
42 #include "DNA_world_types.h"
44 #include "BLI_blenlib.h"
45 #include "BLI_threads.h"
47 #include "BKE_blender.h"
48 #include "BKE_context.h"
49 #include "BKE_global.h"
50 #include "BKE_image.h"
52 #include "BKE_object.h"
53 #include "BKE_utildefines.h"
54 #include "BKE_report.h"
56 #include "RE_pipeline.h"
57 #include "RE_shader_ext.h"
61 #include "IMB_imbuf_types.h"
62 #include "IMB_imbuf.h"
64 #include "GPU_draw.h" /* GPU_free_image */
70 /* ****************** render BAKING ********************** */
72 /* threaded break test */
73 static int thread_break(void *unused)
78 static ScrArea *biggest_image_area(bScreen *screen)
80 ScrArea *sa, *big= NULL;
83 for(sa= screen->areabase.first; sa; sa= sa->next) {
84 if(sa->spacetype==SPACE_IMAGE) {
85 size= sa->winx*sa->winy;
86 if(sa->winx > 10 && sa->winy > 10 && size > maxsize) {
96 typedef struct BakeRender {
110 short prev_wo_amb_occ;
111 short prev_r_raytrace;
117 /* use by exec and invoke */
118 int test_bake_internal(bContext *C, ReportList *reports)
120 Scene *scene= CTX_data_scene(C);
122 if(scene->r.renderer!=R_INTERN) {
123 BKE_report(reports, RPT_ERROR, "Bake only supported for Internal Renderer");
124 } else if((scene->r.bake_flag & R_BAKE_TO_ACTIVE) && CTX_data_active_object(C)==NULL) {
125 BKE_report(reports, RPT_ERROR, "No active object");
127 else if(scene->r.bake_mode==RE_BAKE_AO && scene->world==NULL) {
128 BKE_report(reports, RPT_ERROR, "No world set up");
137 static void init_bake_internal(BakeRender *bkr, bContext *C)
139 Scene *scene= CTX_data_scene(C);
141 bkr->sa= biggest_image_area(CTX_wm_screen(C)); /* can be NULL */
143 bkr->actob= (scene->r.bake_flag & R_BAKE_TO_ACTIVE) ? OBACT : NULL;
144 bkr->re= RE_NewRender("_Bake View_", RE_SLOT_DEFAULT);
146 if(scene->r.bake_mode==RE_BAKE_AO) {
147 /* If raytracing or AO is disabled, switch it on temporarily for baking. */
148 bkr->prev_wo_amb_occ = (scene->world->mode & WO_AMB_OCC) != 0;
149 scene->world->mode |= WO_AMB_OCC;
151 if(scene->r.bake_mode==RE_BAKE_AO || bkr->actob) {
152 bkr->prev_r_raytrace = (scene->r.mode & R_RAYTRACE) != 0;
153 scene->r.mode |= R_RAYTRACE;
157 static void finish_bake_internal(BakeRender *bkr)
159 RE_Database_Free(bkr->re);
161 /* restore raytrace and AO */
162 if(bkr->scene->r.bake_mode==RE_BAKE_AO)
163 if(bkr->prev_wo_amb_occ == 0)
164 bkr->scene->world->mode &= ~WO_AMB_OCC;
166 if(bkr->scene->r.bake_mode==RE_BAKE_AO || bkr->actob)
167 if(bkr->prev_r_raytrace == 0)
168 bkr->scene->r.mode &= ~R_RAYTRACE;
172 /* force OpenGL reload and mipmap recalc */
173 for(ima= G.main->image.first; ima; ima= ima->id.next) {
174 if(ima->ok==IMA_OK_LOADED) {
175 ImBuf *ibuf= BKE_image_get_ibuf(ima, NULL);
176 if(ibuf && (ibuf->userflags & IB_BITMAPDIRTY)) {
178 imb_freemipmapImBuf(ibuf);
185 static void *do_bake_render(void *bake_v)
187 BakeRender *bkr= bake_v;
189 bkr->tot= RE_bake_shade_all_selected(bkr->re, bkr->scene->r.bake_mode, bkr->actob, NULL);
195 static void bake_startjob(void *bkv, short *stop, short *do_update)
197 BakeRender *bkr= bkv;
198 Scene *scene= bkr->scene;
201 bkr->do_update= do_update;
203 RE_test_break_cb(bkr->re, NULL, thread_break);
204 G.afbreek= 0; /* blender_test_break uses this global */
206 RE_Database_Baking(bkr->re, scene, scene->lay, scene->r.bake_mode, bkr->actob);
208 /* baking itself is threaded, cannot use test_break in threads. we also update optional imagewindow */
209 bkr->tot= RE_bake_shade_all_selected(bkr->re, scene->r.bake_mode, bkr->actob, bkr->do_update);
212 static void bake_update(void *bkv)
214 BakeRender *bkr= bkv;
216 if(bkr->sa && bkr->sa->spacetype==SPACE_IMAGE) { /* incase the user changed while baking */
217 SpaceImage *sima= bkr->sa->spacedata.first;
219 sima->image= RE_bake_shade_get_image();
223 static void bake_freejob(void *bkv)
225 BakeRender *bkr= bkv;
226 BLI_end_threads(&bkr->threads);
227 finish_bake_internal(bkr);
229 if(bkr->tot==0) BKE_report(bkr->reports, RPT_ERROR, "No Images found to bake to");
234 static int objects_bake_render_modal(bContext *C, wmOperator *op, wmEvent *event)
236 /* no running blender, remove handler and pass through */
237 if(0==WM_jobs_test(CTX_wm_manager(C), CTX_data_scene(C)))
238 return OPERATOR_FINISHED|OPERATOR_PASS_THROUGH;
241 switch (event->type) {
243 return OPERATOR_RUNNING_MODAL;
246 return OPERATOR_PASS_THROUGH;
249 static int objects_bake_render_invoke(bContext *C, wmOperator *op, wmEvent *_event)
251 Scene *scene= CTX_data_scene(C);
253 if(test_bake_internal(C, op->reports)==0) {
254 return OPERATOR_CANCELLED;
257 BakeRender *bkr= MEM_callocN(sizeof(BakeRender), "render bake");
260 init_bake_internal(bkr, C);
261 bkr->reports= op->reports;
264 steve= WM_jobs_get(CTX_wm_manager(C), CTX_wm_window(C), scene, WM_JOB_EXCL_RENDER|WM_JOB_PRIORITY);
265 WM_jobs_customdata(steve, bkr, bake_freejob);
266 WM_jobs_timer(steve, 0.2, NC_IMAGE, 0); /* TODO - only draw bake image, can we enforce this */
267 WM_jobs_callbacks(steve, bake_startjob, NULL, bake_update);
271 WM_jobs_start(CTX_wm_manager(C), steve);
274 WM_event_add_notifier(C, NC_SCENE|ND_RENDER_RESULT, scene);
276 /* add modal handler for ESC */
277 WM_event_add_modal_handler(C, op);
280 return OPERATOR_RUNNING_MODAL;
284 static int bake_image_exec(bContext *C, wmOperator *op)
286 Scene *scene= CTX_data_scene(C);
289 if(test_bake_internal(C, op->reports)==0) {
290 return OPERATOR_CANCELLED;
296 memset(&bkr, 0, sizeof(bkr));
298 init_bake_internal(&bkr, C);
299 bkr.reports= op->reports;
301 RE_test_break_cb(bkr.re, NULL, thread_break);
302 G.afbreek= 0; /* blender_test_break uses this global */
304 RE_Database_Baking(bkr.re, scene, scene->lay, scene->r.bake_mode, (scene->r.bake_flag & R_BAKE_TO_ACTIVE)? OBACT: NULL);
306 /* baking itself is threaded, cannot use test_break in threads */
307 BLI_init_threads(&threads, do_bake_render, 1);
309 BLI_insert_thread(&threads, &bkr);
311 while(bkr.ready==0) {
316 /* used to redraw in 2.4x but this is just for exec in 2.5 */
318 blender_test_break();
320 BLI_end_threads(&threads);
322 if(bkr.tot==0) BKE_report(op->reports, RPT_ERROR, "No Images found to bake to");
324 finish_bake_internal(&bkr);
327 WM_event_add_notifier(C, NC_SCENE|ND_RENDER_RESULT, scene);
328 return OPERATOR_FINISHED;
331 void OBJECT_OT_bake_image(wmOperatorType *ot)
335 ot->description= "Bake image textures of selected objects";
336 ot->idname= "OBJECT_OT_bake_image";
339 ot->exec= bake_image_exec;
340 ot->invoke= objects_bake_render_invoke;
341 ot->modal= objects_bake_render_modal;