Aborting a level no longer resets Tux to the last finished level: Engine now updates...
[supertux.git] / src / worldmap / worldmap.cpp
1 //  $Id$
2 //
3 //  SuperTux -  A Jump'n Run
4 //  Copyright (C) 2004 Ingo Ruhnke <grumbel@gmx.de>
5 //  Copyright (C) 2006 Christoph Sommer <christoph.sommer@2006.expires.deltadevelopment.de>
6 //
7 //  This program is free software; you can redistribute it and/or
8 //  modify it under the terms of the GNU General Public License
9 //  as published by the Free Software Foundation; either version 2
10 //  of the License, or (at your option) any later version.
11 //
12 //  This program is distributed in the hope that it will be useful,
13 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
14 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 //  GNU General Public License for more details.
16 //
17 //  You should have received a copy of the GNU General Public License
18 //  along with this program; if not, write to the Free Software
19 //  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
20 #include <config.h>
21
22 #include <iostream>
23 #include <fstream>
24 #include <vector>
25 #include <cassert>
26 #include <stdexcept>
27 #include <sstream>
28 #include <unistd.h>
29 #include <physfs.h>
30
31 #include "worldmap.hpp"
32
33 #include "gettext.hpp"
34 #include "log.hpp"
35 #include "mainloop.hpp"
36 #include "shrinkfade.hpp"
37 #include "video/surface.hpp"
38 #include "video/drawing_context.hpp"
39 #include "sprite/sprite_manager.hpp"
40 #include "audio/sound_manager.hpp"
41 #include "lisp/parser.hpp"
42 #include "lisp/lisp.hpp"
43 #include "lisp/list_iterator.hpp"
44 #include "lisp/writer.hpp"
45 #include "game_session.hpp"
46 #include "sector.hpp"
47 #include "worldmap.hpp"
48 #include "resources.hpp"
49 #include "log.hpp"
50 #include "world.hpp"
51 #include "player_status.hpp"
52 #include "textscroller.hpp"
53 #include "main.hpp"
54 #include "spawn_point.hpp"
55 #include "file_system.hpp"
56 #include "gui/menu.hpp"
57 #include "gui/mousecursor.hpp"
58 #include "control/joystickkeyboardcontroller.hpp"
59 #include "object/background.hpp"
60 #include "object/tilemap.hpp"
61 #include "options_menu.hpp"
62 #include "scripting/squirrel_error.hpp"
63 #include "scripting/squirrel_util.hpp"
64 #include "worldmap/level.hpp"
65 #include "worldmap/special_tile.hpp"
66 #include "worldmap/tux.hpp"
67 #include "worldmap/sprite_change.hpp"
68
69 namespace WorldMapNS {
70
71 enum WorldMapMenuIDs {
72   MNID_RETURNWORLDMAP,
73   MNID_QUITWORLDMAP
74 };
75
76 WorldMap* WorldMap::current_ = NULL;
77
78 Direction reverse_dir(Direction direction)
79 {
80   switch(direction)
81     {
82     case D_WEST:
83       return D_EAST;
84     case D_EAST:
85       return D_WEST;
86     case D_NORTH:
87       return D_SOUTH;
88     case D_SOUTH:
89       return D_NORTH;
90     case D_NONE:
91       return D_NONE;
92     }
93   return D_NONE;
94 }
95
96 std::string
97 direction_to_string(Direction direction)
98 {
99   switch(direction)
100     {
101     case D_WEST:
102       return "west";
103     case D_EAST:
104       return "east";
105     case D_NORTH:
106       return "north";
107     case D_SOUTH:
108       return "south";
109     default:
110       return "none";
111     }
112 }
113
114 Direction
115 string_to_direction(const std::string& directory)
116 {
117   if (directory == "west")
118     return D_WEST;
119   else if (directory == "east")
120     return D_EAST;
121   else if (directory == "north")
122     return D_NORTH;
123   else if (directory == "south")
124     return D_SOUTH;
125   else if (directory == "none")
126     return D_NONE;
127   else {
128     log_warning << "unknown direction: \"" << directory << "\"" << std::endl;
129     return D_NONE;
130   }
131 }
132
133 //---------------------------------------------------------------------------
134
135 WorldMap::WorldMap(const std::string& filename, const std::string& force_spawnpoint)
136   : tux(0), solids(0), ambient_light( 1.0f, 1.0f, 1.0f, 1.0f ), force_spawnpoint(force_spawnpoint), in_level(false)
137 {
138   tile_manager.reset(new TileManager("images/worldmap.strf"));
139
140   tux = new Tux(this);
141   add_object(tux);
142
143   name = "<no title>";
144   music = "music/salcon.ogg";
145
146   total_stats.reset();
147
148   worldmap_menu.reset(new Menu());
149   worldmap_menu->add_label(_("Pause"));
150   worldmap_menu->add_hl();
151   worldmap_menu->add_entry(MNID_RETURNWORLDMAP, _("Continue"));
152   worldmap_menu->add_submenu(_("Options"), get_options_menu());
153   worldmap_menu->add_hl();
154   worldmap_menu->add_entry(MNID_QUITWORLDMAP, _("Quit World"));
155
156   load(filename);
157
158   // create a new squirrel table for the worldmap
159   using namespace Scripting;
160
161   sq_collectgarbage(global_vm);
162   sq_newtable(global_vm);
163   sq_pushroottable(global_vm);
164   if(SQ_FAILED(sq_setdelegate(global_vm, -2)))
165     throw Scripting::SquirrelError(global_vm, "Couldn't set worldmap_table delegate");
166
167   sq_resetobject(&worldmap_table);
168   if(SQ_FAILED(sq_getstackobj(global_vm, -1, &worldmap_table)))
169     throw Scripting::SquirrelError(global_vm, "Couldn't get table from stack");
170
171   sq_addref(global_vm, &worldmap_table);
172   sq_pop(global_vm, 1);
173 }
174
175 WorldMap::~WorldMap()
176 {
177   using namespace Scripting;
178
179   for(ScriptList::iterator i = scripts.begin();
180       i != scripts.end(); ++i) {
181     HSQOBJECT& object = *i;
182     sq_release(global_vm, &object);
183   }
184   sq_release(global_vm, &worldmap_table);
185
186   sq_collectgarbage(global_vm);
187
188   if(current_ == this)
189     current_ = NULL;
190
191   for(GameObjects::iterator i = game_objects.begin();
192       i != game_objects.end(); ++i) {
193     GameObject* object = *i;
194     object->unref();
195   }
196
197   for(SpawnPoints::iterator i = spawn_points.begin();
198       i != spawn_points.end(); ++i) {
199     delete *i;
200   }
201 }
202
203 void
204 WorldMap::add_object(GameObject* object)
205 {
206   TileMap* tilemap = dynamic_cast<TileMap*> (object);
207   if(tilemap != 0 && tilemap->is_solid()) {
208     solids = tilemap;
209   }
210
211   object->ref();
212   game_objects.push_back(object);
213 }
214
215 void
216 WorldMap::move_to_spawnpoint(const std::string& spawnpoint)
217 {
218   for(SpawnPoints::iterator i = spawn_points.begin(); i != spawn_points.end(); ++i) {
219     SpawnPoint* sp = *i;
220     if(sp->name == spawnpoint) {
221       Vector p = sp->pos;
222       tux->set_tile_pos(p);
223       tux->set_direction(sp->auto_dir);
224       return;
225     }
226   }
227   log_warning << "Spawnpoint '" << spawnpoint << "' not found." << std::endl;
228   if (spawnpoint != "main") {
229     move_to_spawnpoint("main");
230   }
231 }
232
233 void
234 WorldMap::change(const std::string& filename, const std::string& force_spawnpoint)
235 {
236   main_loop->exit_screen();
237   main_loop->push_screen(new WorldMap(filename, force_spawnpoint));
238 }
239
240 void
241 WorldMap::load(const std::string& filename)
242 {
243   map_filename = filename;
244   levels_path = FileSystem::dirname(map_filename);
245
246   try {
247     lisp::Parser parser;
248     std::auto_ptr<lisp::Lisp> root (parser.parse(map_filename));
249
250     const lisp::Lisp* lisp = root->get_lisp("supertux-level");
251     if(!lisp)
252       throw std::runtime_error("file isn't a supertux-level file.");
253
254     lisp->get("name", name);
255
256     const lisp::Lisp* sector = lisp->get_lisp("sector");
257     if(!sector)
258       throw std::runtime_error("No sector sepcified in worldmap file.");
259
260     lisp::ListIterator iter(sector);
261     while(iter.next()) {
262       if(iter.item() == "tilemap") {
263         add_object(new TileMap(*(iter.lisp()), tile_manager.get()));
264       } else if(iter.item() == "background") {
265         add_object(new Background(*(iter.lisp())));
266       } else if(iter.item() == "music") {
267         iter.value()->get(music);
268       } else if(iter.item() == "init-script") {
269         iter.value()->get(init_script);
270       } else if(iter.item() == "worldmap-spawnpoint") {
271         SpawnPoint* sp = new SpawnPoint(iter.lisp());
272         spawn_points.push_back(sp);
273       } else if(iter.item() == "level") {
274         LevelTile* level = new LevelTile(levels_path, iter.lisp());
275         levels.push_back(level);
276         add_object(level);
277       } else if(iter.item() == "special-tile") {
278         SpecialTile* special_tile = new SpecialTile(iter.lisp());
279         special_tiles.push_back(special_tile);
280         add_object(special_tile);
281       } else if(iter.item() == "sprite-change") {
282         SpriteChange* sprite_change = new SpriteChange(iter.lisp());
283         sprite_changes.push_back(sprite_change);
284         add_object(sprite_change);
285       } else if(iter.item() == "teleporter") {
286         Teleporter* teleporter = new Teleporter(iter.lisp());
287         teleporters.push_back(teleporter);
288         add_object(teleporter);
289       } else if(iter.item() == "ambient-light") {
290         std::vector<float> vColor;
291         sector->get_vector( "ambient-light", vColor );
292         if(vColor.size() < 3) {
293           log_warning << "(ambient-light) requires a color as argument" << std::endl;
294         } else {
295           ambient_light = Color( vColor );
296         }
297       } else if(iter.item() == "name") {
298         // skip
299       } else {
300         log_warning << "Unknown token '" << iter.item() << "' in worldmap" << std::endl;
301       }
302     }
303     if(solids == 0)
304       throw std::runtime_error("No solid tilemap specified");
305
306     move_to_spawnpoint("main");
307
308   } catch(std::exception& e) {
309     std::stringstream msg;
310     msg << "Problem when parsing worldmap '" << map_filename << "': " <<
311       e.what();
312     throw std::runtime_error(msg.str());
313   }
314 }
315
316 void
317 WorldMap::get_level_title(LevelTile& level)
318 {
319   /** get special_tile's title */
320   level.title = "<no title>";
321
322   try {
323     lisp::Parser parser;
324     std::auto_ptr<lisp::Lisp> root (parser.parse(levels_path + level.get_name()));
325
326     const lisp::Lisp* level_lisp = root->get_lisp("supertux-level");
327     if(!level_lisp)
328       return;
329
330     level_lisp->get("name", level.title);
331   } catch(std::exception& e) {
332     log_warning << "Problem when reading leveltitle: " << e.what() << std::endl;
333     return;
334   }
335 }
336
337 void WorldMap::calculate_total_stats()
338 {
339   total_stats.reset();
340   for(LevelTiles::iterator i = levels.begin(); i != levels.end(); ++i) {
341     LevelTile* level = *i;
342     if (level->solved) {
343       total_stats += level->statistics;
344     }
345   }
346 }
347
348 void
349 WorldMap::on_escape_press()
350 {
351   // Show or hide the menu
352   if(!Menu::current()) {
353     Menu::set_current(worldmap_menu.get());
354     tux->set_direction(D_NONE);  // stop tux movement when menu is called
355   } else {
356     Menu::set_current(NULL);
357   }
358 }
359
360 Vector
361 WorldMap::get_next_tile(Vector pos, Direction direction)
362 {
363   switch(direction) {
364     case D_WEST:
365       pos.x -= 1;
366       break;
367     case D_EAST:
368       pos.x += 1;
369       break;
370     case D_NORTH:
371       pos.y -= 1;
372       break;
373     case D_SOUTH:
374       pos.y += 1;
375       break;
376     case D_NONE:
377       break;
378   }
379   return pos;
380 }
381
382 bool
383 WorldMap::path_ok(Direction direction, const Vector& old_pos, Vector* new_pos)
384 {
385   *new_pos = get_next_tile(old_pos, direction);
386
387   if (!(new_pos->x >= 0 && new_pos->x < solids->get_width()
388         && new_pos->y >= 0 && new_pos->y < solids->get_height()))
389     { // New position is outsite the tilemap
390       return false;
391     }
392   else
393     { // Check if the tile allows us to go to new_pos
394       switch(direction)
395         {
396         case D_WEST:
397           return (at(old_pos)->getData() & Tile::WORLDMAP_WEST
398               && at(*new_pos)->getData() & Tile::WORLDMAP_EAST);
399
400         case D_EAST:
401           return (at(old_pos)->getData() & Tile::WORLDMAP_EAST
402               && at(*new_pos)->getData() & Tile::WORLDMAP_WEST);
403
404         case D_NORTH:
405           return (at(old_pos)->getData() & Tile::WORLDMAP_NORTH
406               && at(*new_pos)->getData() & Tile::WORLDMAP_SOUTH);
407
408         case D_SOUTH:
409           return (at(old_pos)->getData() & Tile::WORLDMAP_SOUTH
410               && at(*new_pos)->getData() & Tile::WORLDMAP_NORTH);
411
412         case D_NONE:
413           assert(!"path_ok() can't walk if direction is NONE");
414         }
415       return false;
416     }
417 }
418
419 void
420 WorldMap::finished_level(Level* gamelevel)
421 {
422   // TODO use Level* parameter here?
423   LevelTile* level = at_level();
424
425   bool old_level_state = level->solved;
426   level->solved = true;
427   level->sprite->set_action("solved");
428
429   // deal with statistics
430   level->statistics.merge(gamelevel->stats);
431   calculate_total_stats();
432
433   save_state();
434
435   if (old_level_state != level->solved && level->auto_path) {
436     // Try to detect the next direction to which we should walk
437     // FIXME: Mostly a hack
438     Direction dir = D_NONE;
439
440     const Tile* tile = at(tux->get_tile_pos());
441
442     // first, test for crossroads
443     if (tile->getData() & Tile::WORLDMAP_CNSE || tile->getData() && Tile::WORLDMAP_CNSW
444      || tile->getData() & Tile::WORLDMAP_CNEW || tile->getData() && Tile::WORLDMAP_CSEW
445      || tile->getData() & Tile::WORLDMAP_CNSEW)
446       dir = D_NONE;
447     else if (tile->getData() & Tile::WORLDMAP_NORTH
448         && tux->back_direction != D_NORTH)
449       dir = D_NORTH;
450     else if (tile->getData() & Tile::WORLDMAP_SOUTH
451         && tux->back_direction != D_SOUTH)
452       dir = D_SOUTH;
453     else if (tile->getData() & Tile::WORLDMAP_EAST
454         && tux->back_direction != D_EAST)
455       dir = D_EAST;
456     else if (tile->getData() & Tile::WORLDMAP_WEST
457         && tux->back_direction != D_WEST)
458       dir = D_WEST;
459
460     if (dir != D_NONE) {
461       tux->set_direction(dir);
462     }
463   }
464
465   if (level->extro_script != "") {
466     try {
467       std::istringstream in(level->extro_script);
468       run_script(in, "worldmap:extro_script");
469     } catch(std::exception& e) {
470       log_fatal << "Couldn't run level-extro-script: " << e.what() << std::endl;
471     }
472   }
473 }
474
475 void
476 WorldMap::update(float delta)
477 {
478   if(!in_level) {
479     Menu* menu = Menu::current();
480     if(menu != NULL) {
481       menu->update();
482
483       if(menu == worldmap_menu.get()) {
484         switch (worldmap_menu->check())
485         {
486           case MNID_RETURNWORLDMAP: // Return to game
487             Menu::set_current(0);
488             break;
489           case MNID_QUITWORLDMAP: // Quit Worldmap
490             main_loop->exit_screen();
491             break;
492         }
493       }
494
495       return;
496     }
497
498     // update GameObjects
499     for(size_t i = 0; i < game_objects.size(); ++i) {
500       GameObject* object = game_objects[i];
501       object->update(delta);
502     }
503
504     // remove old GameObjects
505     for(GameObjects::iterator i = game_objects.begin();
506         i != game_objects.end(); ) {
507       GameObject* object = *i;
508       if(!object->is_valid()) {
509         object->unref();
510         i = game_objects.erase(i);
511       } else {
512         ++i;
513       }
514     }
515
516     // position "camera"
517     Vector tux_pos = tux->get_pos();
518     camera_offset.x = tux_pos.x - SCREEN_WIDTH/2;
519     camera_offset.y = tux_pos.y - SCREEN_HEIGHT/2;
520
521     if (camera_offset.x < 0)
522       camera_offset.x = 0;
523     if (camera_offset.y < 0)
524       camera_offset.y = 0;
525
526     if (camera_offset.x > solids->get_width()*32 - SCREEN_WIDTH)
527       camera_offset.x = solids->get_width()*32 - SCREEN_WIDTH;
528     if (camera_offset.y > solids->get_height()*32 - SCREEN_HEIGHT)
529       camera_offset.y = solids->get_height()*32 - SCREEN_HEIGHT;
530
531     // handle input
532     bool enter_level = false;
533     if(main_controller->pressed(Controller::ACTION)
534         || main_controller->pressed(Controller::JUMP)
535         || main_controller->pressed(Controller::MENU_SELECT))
536       enter_level = true;
537     if(main_controller->pressed(Controller::PAUSE_MENU))
538       on_escape_press();
539
540     // check for teleporters
541     Teleporter* teleporter = at_teleporter(tux->get_tile_pos());
542     if (teleporter && (teleporter->automatic || (enter_level && (!tux->is_moving())))) {
543       enter_level = false;
544       if (teleporter->worldmap != "") {
545         change(teleporter->worldmap, teleporter->spawnpoint);
546       } else {
547         // TODO: an animation, camera scrolling or a fading would be a nice touch
548         sound_manager->play("sounds/warp.wav");
549         tux->back_direction = D_NONE;
550         move_to_spawnpoint(teleporter->spawnpoint);
551       }
552     }
553
554     // check for auto-play levels
555     LevelTile* level = at_level();
556     if (level && (level->auto_play) && (!level->solved) && (!tux->is_moving())) {
557       enter_level = true;
558       level->solved = true;
559     }
560
561     if (enter_level && !tux->is_moving())
562       {
563         /* Check level action */
564         LevelTile* level = at_level();
565         if (!level) {
566           //Respawn if player on a tile with no level and nowhere to go.
567           const Tile* tile = at(tux->get_tile_pos());
568           if(!( tile->getData() & ( Tile::WORLDMAP_NORTH |  Tile::WORLDMAP_SOUTH | Tile::WORLDMAP_WEST | Tile::WORLDMAP_EAST ))){
569             log_warning << "Player at illegal position " << tux->get_tile_pos().x << ", " << tux->get_tile_pos().y << " respawning." << std::endl;
570             move_to_spawnpoint("main");
571             return;
572           }
573           log_warning << "No level to enter at: " << tux->get_tile_pos().x << ", " << tux->get_tile_pos().y << std::endl;
574           return;
575         }
576
577         if (level->pos == tux->get_tile_pos()) {
578           try {
579             Vector shrinkpos = Vector(level->pos.x*32 + 16 - camera_offset.x,
580                                       level->pos.y*32 + 16 - camera_offset.y);
581             std::string levelfile = levels_path + level->get_name();
582
583             // update state and savegame
584             save_state();
585
586             main_loop->push_screen(new GameSession(levelfile, &level->statistics),
587                                    new ShrinkFade(shrinkpos, 0.5));
588             in_level = true;
589           } catch(std::exception& e) {
590             log_fatal << "Couldn't load level: " << e.what() << std::endl;
591           }
592         }
593       }
594     else
595       {
596   //      tux->set_direction(input_direction);
597       }
598   }
599 }
600
601 const Tile*
602 WorldMap::at(Vector p)
603 {
604   return solids->get_tile((int) p.x, (int) p.y);
605 }
606
607 LevelTile*
608 WorldMap::at_level()
609 {
610   for(LevelTiles::iterator i = levels.begin(); i != levels.end(); ++i) {
611     LevelTile* level = *i;
612     if (level->pos == tux->get_tile_pos())
613       return level;
614   }
615
616   return NULL;
617 }
618
619 SpecialTile*
620 WorldMap::at_special_tile()
621 {
622   for(SpecialTiles::iterator i = special_tiles.begin();
623       i != special_tiles.end(); ++i) {
624     SpecialTile* special_tile = *i;
625     if (special_tile->pos == tux->get_tile_pos())
626       return special_tile;
627   }
628
629   return NULL;
630 }
631
632 SpriteChange*
633 WorldMap::at_sprite_change(const Vector& pos)
634 {
635   for(SpriteChanges::iterator i = sprite_changes.begin();
636       i != sprite_changes.end(); ++i) {
637     SpriteChange* sprite_change = *i;
638     if(sprite_change->pos == pos)
639       return sprite_change;
640   }
641
642   return NULL;
643 }
644
645 Teleporter*
646 WorldMap::at_teleporter(const Vector& pos)
647 {
648   for(std::vector<Teleporter*>::iterator i = teleporters.begin(); i != teleporters.end(); ++i) {
649     Teleporter* teleporter = *i;
650     if(teleporter->pos == pos) return teleporter;
651   }
652
653   return NULL;
654 }
655
656 void
657 WorldMap::draw(DrawingContext& context)
658 {
659   context.set_ambient_color( ambient_light );
660   context.push_transform();
661   context.set_translation(camera_offset);
662
663   for(GameObjects::iterator i = game_objects.begin();
664       i != game_objects.end(); ++i) {
665     GameObject* object = *i;
666     object->draw(context);
667   }
668
669   draw_status(context);
670   context.pop_transform();
671 }
672
673 void
674 WorldMap::draw_status(DrawingContext& context)
675 {
676   context.push_transform();
677   context.set_translation(Vector(0, 0));
678
679   player_status->draw(context);
680
681   if (!tux->is_moving()) {
682     for(LevelTiles::iterator i = levels.begin(); i != levels.end(); ++i) {
683       LevelTile* level = *i;
684
685       if (level->pos == tux->get_tile_pos()) {
686         if(level->title == "")
687           get_level_title(*level);
688
689         context.draw_text(white_text, level->title,
690             Vector(SCREEN_WIDTH/2,
691               SCREEN_HEIGHT - white_text->get_height() - 30),
692             CENTER_ALLIGN, LAYER_FOREGROUND1);
693
694         // if level is solved, draw level picture behind stats
695         /*
696         if (level->solved) {
697           if (const Surface* picture = level->get_picture()) {
698             Vector pos = Vector(SCREEN_WIDTH - picture->get_width(), SCREEN_HEIGHT - picture->get_height());
699             context.push_transform();
700             context.set_alpha(0.5);
701             context.draw_surface(picture, pos, LAYER_FOREGROUND1-1);
702             context.pop_transform();
703           }
704         }
705         */
706
707         level->statistics.draw_worldmap_info(context);
708         break;
709       }
710     }
711
712     for(SpecialTiles::iterator i = special_tiles.begin();
713         i != special_tiles.end(); ++i) {
714       SpecialTile* special_tile = *i;
715
716       if (special_tile->pos == tux->get_tile_pos()) {
717         /* Display an in-map message in the map, if any as been selected */
718         if(!special_tile->map_message.empty() && !special_tile->passive_message)
719           context.draw_text(gold_text, special_tile->map_message,
720               Vector(SCREEN_WIDTH/2,
721                 SCREEN_HEIGHT - white_text->get_height() - 60),
722               CENTER_ALLIGN, LAYER_FOREGROUND1);
723         break;
724       }
725     }
726
727     // display teleporter messages
728     Teleporter* teleporter = at_teleporter(tux->get_tile_pos());
729     if (teleporter && (teleporter->message != "")) {
730       Vector pos = Vector(SCREEN_WIDTH/2, SCREEN_HEIGHT - white_text->get_height() - 30);
731       context.draw_text(white_text, teleporter->message, pos, CENTER_ALLIGN, LAYER_FOREGROUND1);
732     }
733
734   }
735
736   /* Display a passive message in the map, if needed */
737   if(passive_message_timer.started())
738     context.draw_text(gold_text, passive_message,
739             Vector(SCREEN_WIDTH/2, SCREEN_HEIGHT - white_text->get_height() - 60),
740             CENTER_ALLIGN, LAYER_FOREGROUND1);
741
742   context.pop_transform();
743 }
744
745 void
746 WorldMap::setup()
747 {
748   sound_manager->play_music(music);
749   Menu::set_current(NULL);
750
751   current_ = this;
752   load_state();
753
754   // if force_spawnpoint was set, move Tux there, then clear force_spawnpoint
755   if (force_spawnpoint != "") {
756     move_to_spawnpoint(force_spawnpoint);
757     force_spawnpoint = "";
758   }
759
760   tux->setup();
761
762   // register worldmap_table as worldmap in scripting
763   using namespace Scripting;
764
765   sq_pushroottable(global_vm);
766   sq_pushstring(global_vm, "worldmap", -1);
767   sq_pushobject(global_vm, worldmap_table);
768   if(SQ_FAILED(sq_createslot(global_vm, -3)))
769     throw SquirrelError(global_vm, "Couldn't set worldmap in roottable");
770   sq_pop(global_vm, 1);
771
772   if(init_script != "") {
773     std::istringstream in(init_script);
774     run_script(in, "WorldMap::init");
775   }
776 }
777
778 void
779 WorldMap::leave()
780 {
781   // remove worldmap_table from roottable
782   using namespace Scripting;
783
784   sq_pushroottable(global_vm);
785   sq_pushstring(global_vm, "worldmap", -1);
786   if(SQ_FAILED(sq_deleteslot(global_vm, -2, SQFalse)))
787     throw SquirrelError(global_vm, "Couldn't unset worldmap in roottable");
788   sq_pop(global_vm, 1);
789 }
790
791 static void store_float(HSQUIRRELVM vm, const char* name, float val)
792 {
793   sq_pushstring(vm, name, -1);
794   sq_pushfloat(vm, val);
795   if(SQ_FAILED(sq_createslot(vm, -3)))
796     throw Scripting::SquirrelError(vm, "Couldn't add float value to table");
797 }
798
799 /*
800 static void store_int(HSQUIRRELVM vm, const char* name, int val)
801 {
802   sq_pushstring(vm, name, -1);
803   sq_pushinteger(vm, val);
804   if(SQ_FAILED(sq_createslot(vm, -3)))
805     throw Scripting::SquirrelError(vm, "Couldn't add float value to table");
806 }
807 */
808
809 static void store_string(HSQUIRRELVM vm, const char* name, const std::string& val)
810 {
811   sq_pushstring(vm, name, -1);
812   sq_pushstring(vm, val.c_str(), val.length());
813   if(SQ_FAILED(sq_createslot(vm, -3)))
814     throw Scripting::SquirrelError(vm, "Couldn't add float value to table");
815 }
816
817 static void store_bool(HSQUIRRELVM vm, const char* name, bool val)
818 {
819   sq_pushstring(vm, name, -1);
820   sq_pushbool(vm, val ? SQTrue : SQFalse);
821   if(SQ_FAILED(sq_createslot(vm, -3)))
822     throw Scripting::SquirrelError(vm, "Couldn't add float value to table");
823 }
824
825 static float read_float(HSQUIRRELVM vm, const char* name)
826 {
827   sq_pushstring(vm, name, -1);
828   if(SQ_FAILED(sq_get(vm, -2))) {
829     std::ostringstream msg;
830     msg << "Couldn't get float value for '" << name << "' from table";
831     throw Scripting::SquirrelError(vm, msg.str());
832   }
833
834   float result;
835   if(SQ_FAILED(sq_getfloat(vm, -1, &result))) {
836     std::ostringstream msg;
837     msg << "Couldn't get float value for '" << name << "' from table";
838     throw Scripting::SquirrelError(vm, msg.str());
839   }
840   sq_pop(vm, 1);
841
842   return result;
843 }
844
845 static std::string read_string(HSQUIRRELVM vm, const char* name)
846 {
847   sq_pushstring(vm, name, -1);
848   if(SQ_FAILED(sq_get(vm, -2))) {
849     std::ostringstream msg;
850     msg << "Couldn't get string value for '" << name << "' from table";
851     throw Scripting::SquirrelError(vm, msg.str());
852   }
853
854   const char* result;
855   if(SQ_FAILED(sq_getstring(vm, -1, &result))) {
856     std::ostringstream msg;
857     msg << "Couldn't get string value for '" << name << "' from table";
858     throw Scripting::SquirrelError(vm, msg.str());
859   }
860   sq_pop(vm, 1);
861
862   return std::string(result);
863 }
864
865 static bool read_bool(HSQUIRRELVM vm, const char* name)
866 {
867   sq_pushstring(vm, name, -1);
868   if(SQ_FAILED(sq_get(vm, -2))) {
869     std::ostringstream msg;
870     msg << "Couldn't get bool value for '" << name << "' from table";
871     throw Scripting::SquirrelError(vm, msg.str());
872   }
873
874   SQBool result;
875   if(SQ_FAILED(sq_getbool(vm, -1, &result))) {
876     std::ostringstream msg;
877     msg << "Couldn't get bool value for '" << name << "' from table";
878     throw Scripting::SquirrelError(vm, msg.str());
879   }
880   sq_pop(vm, 1);
881
882   return result == SQTrue;
883 }
884
885 void
886 WorldMap::save_state()
887 {
888   using namespace Scripting;
889
890   HSQUIRRELVM vm = global_vm;
891   int oldtop = sq_gettop(vm);
892
893   try {
894     // get state table
895     sq_pushroottable(vm);
896     sq_pushstring(vm, "state", -1);
897     if(SQ_FAILED(sq_get(vm, -2)))
898       throw Scripting::SquirrelError(vm, "Couldn't get state table");
899
900     // get or create worlds table
901     sq_pushstring(vm, "worlds", -1);
902     if(SQ_FAILED(sq_get(vm, -2))) {
903       sq_pushstring(vm, "worlds", -1);
904       sq_newtable(vm);
905       if(SQ_FAILED(sq_createslot(vm, -3)))
906         throw Scripting::SquirrelError(vm, "Couldn't create state.worlds");
907
908       sq_pushstring(vm, "worlds", -1);
909       if(SQ_FAILED(sq_get(vm, -2)))
910         throw Scripting::SquirrelError(vm, "Couldn't create.get state.worlds");
911     }
912
913     sq_pushstring(vm, map_filename.c_str(), map_filename.length());
914     if(SQ_FAILED(sq_deleteslot(vm, -2, SQFalse)))
915       sq_pop(vm, 1);
916
917     // construct new table for this worldmap
918     sq_pushstring(vm, map_filename.c_str(), map_filename.length());
919     sq_newtable(vm);
920
921     // store tux
922     sq_pushstring(vm, "tux", -1);
923     sq_newtable(vm);
924
925     store_float(vm, "x", tux->get_tile_pos().x);
926     store_float(vm, "y", tux->get_tile_pos().y);
927     store_string(vm, "back", direction_to_string(tux->back_direction));
928
929     sq_createslot(vm, -3);
930
931     // levels...
932     sq_pushstring(vm, "levels", -1);
933     sq_newtable(vm);
934
935     for(LevelTiles::iterator i = levels.begin(); i != levels.end(); ++i) {
936       LevelTile* level = *i;
937
938       if (level->solved) {
939         sq_pushstring(vm, level->get_name().c_str(), -1);
940         sq_newtable(vm);
941
942         store_bool(vm, "solved", true);
943         // TODO write statistics
944         // i->statistics.write(writer);
945
946         sq_createslot(vm, -3);
947       }
948     }
949
950     sq_createslot(vm, -3);
951
952     // push world into worlds table
953     sq_createslot(vm, -3);
954   } catch(std::exception& e) {
955     sq_settop(vm, oldtop);
956   }
957
958   sq_settop(vm, oldtop);
959   
960   if(World::current() != NULL)
961     World::current()->save_state();
962 }
963
964 void
965 WorldMap::load_state()
966 {
967   using namespace Scripting;
968
969   HSQUIRRELVM vm = global_vm;
970   int oldtop = sq_gettop(vm);
971
972   try {
973     // get state table
974     sq_pushroottable(vm);
975     sq_pushstring(vm, "state", -1);
976     if(SQ_FAILED(sq_get(vm, -2)))
977       throw Scripting::SquirrelError(vm, "Couldn't get state table");
978
979     // get worlds table
980     sq_pushstring(vm, "worlds", -1);
981     if(SQ_FAILED(sq_get(vm, -2)))
982       throw Scripting::SquirrelError(vm, "Couldn't get state.worlds");
983
984     // get table for our world
985     sq_pushstring(vm, map_filename.c_str(), map_filename.length());
986     if(SQ_FAILED(sq_get(vm, -2)))
987       throw Scripting::SquirrelError(vm, "Couldn't get state.world.mapfilename");
988
989     // load tux
990     sq_pushstring(vm, "tux", -1);
991     if(SQ_FAILED(sq_get(vm, -2)))
992       throw Scripting::SquirrelError(vm, "Couldn't get tux");
993
994     Vector p;
995     p.x = read_float(vm, "x");
996     p.y = read_float(vm, "y");
997     std::string back_str = read_string(vm, "back");
998     tux->back_direction = string_to_direction(back_str);
999     tux->set_tile_pos(p);
1000
1001     sq_pop(vm, 1);
1002
1003     // load levels
1004     sq_pushstring(vm, "levels", -1);
1005     if(SQ_FAILED(sq_get(vm, -2)))
1006       throw Scripting::SquirrelError(vm, "Couldn't get levels");
1007
1008     for(LevelTiles::iterator i = levels.begin(); i != levels.end(); ++i) {
1009       LevelTile* level = *i;
1010       sq_pushstring(vm, level->get_name().c_str(), -1);
1011       if(SQ_SUCCEEDED(sq_get(vm, -2))) {
1012         level->solved = read_bool(vm, "solved");
1013         level->sprite->set_action(level->solved ? "solved" : "default");
1014         // i->statistics.parse(*level);
1015         sq_pop(vm, 1);
1016       }
1017     }
1018     sq_pop(vm, 1);
1019
1020   } catch(std::exception& e) {
1021     log_debug << "Not loading worldmap state: " << e.what() << std::endl;
1022   }
1023   sq_settop(vm, oldtop);
1024
1025   in_level = false;
1026 }
1027
1028 size_t
1029 WorldMap::level_count()
1030 {
1031   return levels.size();
1032 }
1033
1034 size_t
1035 WorldMap::solved_level_count()
1036 {
1037   size_t count = 0;
1038   for(LevelTiles::iterator i = levels.begin(); i != levels.end(); ++i) {
1039     LevelTile* level = *i;
1040
1041     if(level->solved)
1042       count++;
1043   }
1044
1045   return count;
1046 }
1047
1048 HSQUIRRELVM
1049 WorldMap::run_script(std::istream& in, const std::string& sourcename)
1050 {
1051   using namespace Scripting;
1052
1053   // garbage collect thread list
1054   for(ScriptList::iterator i = scripts.begin();
1055       i != scripts.end(); ) {
1056     HSQOBJECT& object = *i;
1057     HSQUIRRELVM vm = object_to_vm(object);
1058
1059     if(sq_getvmstate(vm) != SQ_VMSTATE_SUSPENDED) {
1060       sq_release(global_vm, &object);
1061       i = scripts.erase(i);
1062       continue;
1063     }
1064
1065     ++i;
1066   }
1067
1068   HSQOBJECT object = create_thread(global_vm);
1069   scripts.push_back(object);
1070
1071   HSQUIRRELVM vm = object_to_vm(object);
1072
1073   // set worldmap_table as roottable for the thread
1074   sq_pushobject(vm, worldmap_table);
1075   sq_setroottable(vm);
1076
1077   compile_and_run(vm, in, sourcename);
1078
1079   return vm;
1080 }
1081
1082 } // namespace WorldMapNS