Made lightmap as big as screen to prevent rounding errors.
[supertux.git] / src / video / drawing_context.cpp
1 //  $Id$
2 //
3 //  SuperTux
4 //  Copyright (C) 2006 Matthias Braun <matze@braunis.de>
5 //
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.
10 //
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.
15 //
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
18 //  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
19
20 #include <config.h>
21
22 #include <algorithm>
23 #include <cassert>
24 #include <iostream>
25 #include <SDL_image.h>
26 #include <GL/gl.h>
27
28 #include "drawing_context.hpp"
29 #include "surface.hpp"
30 #include "font.hpp"
31 #include "main.hpp"
32 #include "gameconfig.hpp"
33 #include "glutil.hpp"
34 #include "texture.hpp"
35 #include "texture_manager.hpp"
36 #define LIGHTMAP_DIV 1
37
38 static inline int next_po2(int val)
39 {
40   int result = 1;
41   while(result < val)
42     result *= 2;
43
44   return result;
45 }
46
47 DrawingContext::DrawingContext()
48   : ambient_color( 1.0f, 1.0f, 1.0f, 1.0f )
49 {
50   screen = SDL_GetVideoSurface();
51
52   lightmap_width = screen->w / LIGHTMAP_DIV;
53   lightmap_height = screen->h / LIGHTMAP_DIV;
54   unsigned int width = next_po2(lightmap_width);
55   unsigned int height = next_po2(lightmap_height);
56
57   lightmap = new Texture(width, height, GL_RGB);
58
59   lightmap_uv_right = static_cast<float>(lightmap_width) / static_cast<float>(width);
60   lightmap_uv_bottom = static_cast<float>(lightmap_height) / static_cast<float>(height);
61   texture_manager->register_texture(lightmap);
62
63   requests = &drawing_requests;
64 }
65
66 DrawingContext::~DrawingContext()
67 {
68   texture_manager->remove_texture(lightmap);
69   delete lightmap;
70 }
71
72 void
73 DrawingContext::draw_surface(const Surface* surface, const Vector& position,
74                              float angle, const Color& color, const Blend& blend,
75                              int layer)
76 {
77   assert(surface != 0);
78
79   DrawingRequest request;
80
81   request.type = SURFACE;
82   request.pos = transform.apply(position);
83
84   if(request.pos.x >= SCREEN_WIDTH || request.pos.y >= SCREEN_HEIGHT
85       || request.pos.x + surface->get_width() < 0
86       || request.pos.y + surface->get_height() < 0)
87     return;
88
89   request.layer = layer;
90   request.drawing_effect = transform.drawing_effect;
91   request.alpha = transform.alpha;
92   request.angle = angle;
93   request.color = color;
94   request.blend = blend;
95
96   request.request_data = const_cast<Surface*> (surface);
97
98   requests->push_back(request);
99 }
100
101 void
102 DrawingContext::draw_surface(const Surface* surface, const Vector& position,
103     int layer)
104 {
105   draw_surface(surface, position, 0.0f, Color(1.0f, 1.0f, 1.0f), Blend(), layer);
106 }
107
108 void
109 DrawingContext::draw_surface_part(const Surface* surface, const Vector& source,
110     const Vector& size, const Vector& dest, int layer)
111 {
112   assert(surface != 0);
113
114   DrawingRequest request;
115
116   request.type = SURFACE_PART;
117   request.pos = transform.apply(dest);
118   request.layer = layer;
119   request.drawing_effect = transform.drawing_effect;
120   request.alpha = transform.alpha;
121
122   SurfacePartRequest* surfacepartrequest = new SurfacePartRequest();
123   surfacepartrequest->size = size;
124   surfacepartrequest->source = source;
125   surfacepartrequest->surface = surface;
126
127   // clip on screen borders
128   if(request.pos.x < 0) {
129     surfacepartrequest->size.x += request.pos.x;
130     if(surfacepartrequest->size.x <= 0)
131       return;
132     surfacepartrequest->source.x -= request.pos.x;
133     request.pos.x = 0;
134   }
135   if(request.pos.y < 0) {
136     surfacepartrequest->size.y += request.pos.y;
137     if(surfacepartrequest->size.y <= 0)
138       return;
139     surfacepartrequest->source.y -= request.pos.y;
140     request.pos.y = 0;
141   }
142   request.request_data = surfacepartrequest;
143
144   requests->push_back(request);
145 }
146
147 void
148 DrawingContext::draw_text(const Font* font, const std::string& text,
149     const Vector& position, FontAlignment alignment, int layer)
150 {
151   DrawingRequest request;
152
153   request.type = TEXT;
154   request.pos = transform.apply(position);
155   request.layer = layer;
156   request.drawing_effect = transform.drawing_effect;
157   request.alpha = transform.alpha;
158
159   TextRequest* textrequest = new TextRequest;
160   textrequest->font = font;
161   textrequest->text = text;
162   textrequest->alignment = alignment;
163   request.request_data = textrequest;
164
165   requests->push_back(request);
166 }
167
168 void
169 DrawingContext::draw_center_text(const Font* font, const std::string& text,
170     const Vector& position, int layer)
171 {
172   draw_text(font, text, Vector(position.x + SCREEN_WIDTH/2, position.y),
173       CENTER_ALLIGN, layer);
174 }
175
176 void
177 DrawingContext::draw_gradient(const Color& top, const Color& bottom, int layer)
178 {
179   DrawingRequest request;
180
181   request.type = GRADIENT;
182   request.pos = Vector(0,0);
183   request.layer = layer;
184
185   request.drawing_effect = transform.drawing_effect;
186   request.alpha = transform.alpha;
187
188   GradientRequest* gradientrequest = new GradientRequest;
189   gradientrequest->top = top;
190   gradientrequest->bottom = bottom;
191   request.request_data = gradientrequest;
192
193   requests->push_back(request);
194 }
195
196 void
197 DrawingContext::draw_filled_rect(const Vector& topleft, const Vector& size,
198                                  const Color& color, int layer)
199 {
200   DrawingRequest request;
201
202   request.type = FILLRECT;
203   request.pos = transform.apply(topleft);
204   request.layer = layer;
205
206   request.drawing_effect = transform.drawing_effect;
207   request.alpha = transform.alpha;
208
209   FillRectRequest* fillrectrequest = new FillRectRequest;
210   fillrectrequest->size = size;
211   fillrectrequest->color = color;
212   fillrectrequest->color.alpha = color.alpha * transform.alpha;
213   request.request_data = fillrectrequest;
214
215   requests->push_back(request);
216 }
217
218 void
219 DrawingContext::draw_filled_rect(const Rect& rect, const Color& color,
220                                  int layer)
221 {
222   DrawingRequest request;
223
224   request.type = FILLRECT;
225   request.pos = transform.apply(rect.p1);
226   request.layer = layer;
227
228   request.drawing_effect = transform.drawing_effect;
229   request.alpha = transform.alpha;
230
231   FillRectRequest* fillrectrequest = new FillRectRequest;
232   fillrectrequest->size = Vector(rect.get_width(), rect.get_height());
233   fillrectrequest->color = color;
234   fillrectrequest->color.alpha = color.alpha * transform.alpha;
235   request.request_data = fillrectrequest;
236
237   requests->push_back(request);
238 }
239
240 void
241 DrawingContext::get_light(const Vector& position, Color* color)
242 {
243   if( ambient_color.red == 1.0f && ambient_color.green == 1.0f
244       && ambient_color.blue  == 1.0f ) {
245     *color = Color( 1.0f, 1.0f, 1.0f);
246     return;
247   }
248   DrawingRequest request;
249   request.type = GETLIGHT;
250   request.pos = transform.apply(position);
251   request.layer = LAYER_GUI; //make sure all get_light requests are handled last.
252
253   GetLightRequest* getlightrequest = new GetLightRequest;
254   getlightrequest->color_ptr = color;
255   request.request_data = getlightrequest;
256   lightmap_requests.push_back(request);
257 }
258
259 void
260 DrawingContext::get_light(DrawingRequest& request)
261 {
262   GetLightRequest* getlightrequest = (GetLightRequest*) request.request_data;
263
264   float pixels[3];
265   for( int i = 0; i<3; i++)
266     pixels[i] = 0.0f; //set to black
267
268   float posX = request.pos.x /LIGHTMAP_DIV;
269   float posY = SCREEN_HEIGHT - request.pos.y / LIGHTMAP_DIV;
270   glReadPixels((GLint) posX, (GLint) posY , 1, 1, GL_RGB, GL_FLOAT, pixels);
271     *(getlightrequest->color_ptr) = Color( pixels[0], pixels[1], pixels[2]);  
272   //draw_filled_rect( Vector(posX, posY), Vector(1,1), Color( 1.0f, 1.0f, 1.0f) ,LAYER_GUI);
273   //printf("get_light %f/%f r%f g%f b%f\n", request.pos.x, request.pos.y, pixels[0], pixels[1], pixels[2]);
274
275   delete getlightrequest;
276 }
277
278 void
279 DrawingContext::draw_surface_part(DrawingRequest& request)
280 {
281   SurfacePartRequest* surfacepartrequest
282     = (SurfacePartRequest*) request.request_data;
283
284   surfacepartrequest->surface->draw_part(
285       surfacepartrequest->source.x, surfacepartrequest->source.y,
286       request.pos.x, request.pos.y,
287       surfacepartrequest->size.x, surfacepartrequest->size.y,
288       request.alpha, request.drawing_effect);
289
290   delete surfacepartrequest;
291 }
292
293 void
294 DrawingContext::draw_gradient(DrawingRequest& request)
295 {
296   GradientRequest* gradientrequest = (GradientRequest*) request.request_data;
297   const Color& top = gradientrequest->top;
298   const Color& bottom = gradientrequest->bottom;
299
300   glDisable(GL_TEXTURE_2D);
301   glBegin(GL_QUADS);
302   glColor4f(top.red, top.green, top.blue, top.alpha);
303   glVertex2f(0, 0);
304   glVertex2f(SCREEN_WIDTH, 0);
305   glColor4f(bottom.red, bottom.green, bottom.blue, bottom.alpha);
306   glVertex2f(SCREEN_WIDTH, SCREEN_HEIGHT);
307   glVertex2f(0, SCREEN_HEIGHT);
308   glEnd();
309   glEnable(GL_TEXTURE_2D);
310
311   delete gradientrequest;
312 }
313
314 void
315 DrawingContext::draw_text(DrawingRequest& request)
316 {
317   TextRequest* textrequest = (TextRequest*) request.request_data;
318
319   textrequest->font->draw(textrequest->text, request.pos,
320       textrequest->alignment, request.drawing_effect, request.alpha);
321
322   delete textrequest;
323 }
324
325 void
326 DrawingContext::draw_filled_rect(DrawingRequest& request)
327 {
328   FillRectRequest* fillrectrequest = (FillRectRequest*) request.request_data;
329
330   float x = request.pos.x;
331   float y = request.pos.y;
332   float w = fillrectrequest->size.x;
333   float h = fillrectrequest->size.y;
334
335   glDisable(GL_TEXTURE_2D);
336   glColor4f(fillrectrequest->color.red, fillrectrequest->color.green,
337             fillrectrequest->color.blue, fillrectrequest->color.alpha);
338
339   glBegin(GL_QUADS);
340   glVertex2f(x, y);
341   glVertex2f(x+w, y);
342   glVertex2f(x+w, y+h);
343   glVertex2f(x, y+h);
344   glEnd();
345   glEnable(GL_TEXTURE_2D);
346
347   delete fillrectrequest;
348 }
349
350 void
351 DrawingContext::draw_lightmap(DrawingRequest& request)
352 {
353   const Texture* texture = reinterpret_cast<Texture*> (request.request_data);
354
355   // multiple the lightmap with the framebuffer
356   glBlendFunc(GL_DST_COLOR, GL_ZERO);
357
358   glBindTexture(GL_TEXTURE_2D, texture->get_handle());
359   glBegin(GL_QUADS);
360
361   glTexCoord2f(0, lightmap_uv_bottom);
362   glVertex2f(0, 0);
363
364   glTexCoord2f(lightmap_uv_right, lightmap_uv_bottom);
365   glVertex2f(SCREEN_WIDTH, 0);
366
367   glTexCoord2f(lightmap_uv_right, 0);
368   glVertex2f(SCREEN_WIDTH, SCREEN_HEIGHT);
369
370   glTexCoord2f(0, 0);
371   glVertex2f(0, SCREEN_HEIGHT);
372
373   glEnd();
374
375   glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
376 }
377
378 void
379 DrawingContext::do_drawing()
380 {
381 #ifdef DEBUG
382   assert(transformstack.empty());
383   assert(target_stack.empty());
384 #endif
385   transformstack.clear();
386   target_stack.clear();
387
388   //Use Lightmap if ambient color is not white.
389   bool use_lightmap = ( ambient_color.red != 1.0f   || ambient_color.green != 1.0f ||
390                         ambient_color.blue  != 1.0f );
391
392   // PART1: create lightmap
393   if(use_lightmap) {
394     glViewport(0, screen->h - lightmap_height, lightmap_width, lightmap_height);
395     glMatrixMode(GL_PROJECTION);
396     glLoadIdentity();
397     glOrtho(0, SCREEN_WIDTH, SCREEN_HEIGHT, 0, -1.0, 1.0);
398     glMatrixMode(GL_MODELVIEW);
399     glLoadIdentity();
400
401     glClearColor( ambient_color.red, ambient_color.green, ambient_color.blue, 1 );
402     glClear(GL_COLOR_BUFFER_BIT);
403     handle_drawing_requests(lightmap_requests);
404     lightmap_requests.clear();
405
406     glDisable(GL_BLEND);
407     glBindTexture(GL_TEXTURE_2D, lightmap->get_handle());
408     glCopyTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, 0, screen->h - lightmap_height, lightmap_width, lightmap_height);
409
410     glViewport(0, 0, screen->w, screen->h);
411     glMatrixMode(GL_PROJECTION);
412     glLoadIdentity();
413     glOrtho(0, SCREEN_WIDTH, SCREEN_HEIGHT, 0, -1.0, 1.0);
414     glMatrixMode(GL_MODELVIEW);
415     glLoadIdentity();
416     glEnable(GL_BLEND);
417
418     // add a lightmap drawing request into the queue
419     DrawingRequest request;
420     request.type = LIGHTMAPREQUEST;
421     request.layer = LAYER_HUD - 1;
422     request.request_data = lightmap;
423     requests->push_back(request);
424   }
425
426   //glClear(GL_COLOR_BUFFER_BIT);
427   handle_drawing_requests(drawing_requests);
428   drawing_requests.clear();
429   assert_gl("drawing");
430
431   SDL_GL_SwapBuffers();
432 }
433
434 void
435 DrawingContext::handle_drawing_requests(DrawingRequests& requests)
436 {
437   std::stable_sort(requests.begin(), requests.end());
438
439   for(DrawingRequests::iterator i = requests.begin();
440       i != requests.end(); ++i) {
441     switch(i->type) {
442       case SURFACE:
443       {
444         const Surface* surface = (const Surface*) i->request_data;
445         if (i->angle == 0.0f && 
446             i->color.red == 1.0f && i->color.green == 1.0f  &&
447             i->color.blue == 1.0f &&  i->color.alpha == 1.0f  )
448           surface->draw(i->pos.x, i->pos.y, i->alpha, i->drawing_effect);
449         else
450           surface->draw(i->pos.x, i->pos.y, i->alpha, i->angle, i->color, i->blend, i->drawing_effect);
451         break;
452       }
453       case SURFACE_PART:
454         draw_surface_part(*i);
455         break;
456       case GRADIENT:
457         draw_gradient(*i);
458         break;
459       case TEXT:
460         draw_text(*i);
461         break;
462       case FILLRECT:
463         draw_filled_rect(*i);
464         break;
465       case LIGHTMAPREQUEST:
466         draw_lightmap(*i);
467         break;
468       case GETLIGHT:
469         get_light(*i);
470         break;
471     }
472   }
473 }
474
475 void
476 DrawingContext::push_transform()
477 {
478   transformstack.push_back(transform);
479 }
480
481 void
482 DrawingContext::pop_transform()
483 {
484   assert(!transformstack.empty());
485
486   transform = transformstack.back();
487   transformstack.pop_back();
488 }
489
490 void
491 DrawingContext::set_drawing_effect(DrawingEffect effect)
492 {
493   transform.drawing_effect = effect;
494 }
495
496 DrawingEffect
497 DrawingContext::get_drawing_effect() const
498 {
499   return transform.drawing_effect;
500 }
501
502 void
503 DrawingContext::set_alpha(float alpha)
504 {
505   transform.alpha = alpha;
506 }
507
508 float
509 DrawingContext::get_alpha() const
510 {
511   return transform.alpha;
512 }
513
514 void
515 DrawingContext::push_target()
516 {
517   target_stack.push_back(target);
518 }
519
520 void
521 DrawingContext::pop_target()
522 {
523   set_target(target_stack.back());
524   target_stack.pop_back();
525 }
526
527 void
528 DrawingContext::set_target(Target target)
529 {
530   this->target = target;
531   if(target == LIGHTMAP)
532     requests = &lightmap_requests;
533   else
534     requests = &drawing_requests;
535 }
536
537 void
538 DrawingContext::set_ambient_color( Color new_color )
539 {
540   ambient_color = new_color;
541 }