- committed patch from Richard Smith
[supertux.git] / src / level_subset.cpp
1 //  $Id$
2 // 
3 //  SuperTux
4 //  Copyright (C) 2004 SuperTux Development Team, see AUTHORS for details
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
19 //  02111-1307, USA.
20
21 #include <assert.h>
22 #include <unistd.h>
23 #include "app/setup.h"
24 #include "level.h"
25 #include "app/globals.h"
26 #include "video/surface.h"
27 #include "level_subset.h"
28
29 using namespace SuperTux;
30
31 static bool has_suffix(const std::string& data, const std::string& suffix)
32 {
33   if (data.length() >= suffix.length())
34     return data.compare(data.length() - suffix.length(), suffix.length(), suffix) == 0;
35   else
36     return false;
37 }
38
39 LevelSubset::LevelSubset()
40   : levels(0)
41 {
42 }
43
44 LevelSubset::~LevelSubset()
45 {
46 }
47
48 void LevelSubset::create(const std::string& subset_name)
49 {
50   Level new_lev;
51   LevelSubset new_subset;
52   new_subset.name = subset_name;
53   new_subset.title = "Unknown Title";
54   new_subset.description = "No description so far.";
55   new_subset.hide_from_contribs = false;
56   new_subset.save();
57 }
58
59 void LevelSubset::read_info_file(const std::string& info_file)
60 {
61   lisp_object_t* root_obj = lisp_read_from_file(info_file);
62   if (root_obj == NULL)
63     return;
64   lisp_object_t* cur = lisp_car(root_obj);
65
66   if (lisp_symbol_p(cur) && strcmp(lisp_symbol(cur), "supertux-level-subset") == 0)
67     {
68       LispReader reader(lisp_cdr(root_obj));
69
70       reader.read_string("title", title, true);
71       reader.read_string("description", description, true);
72       reader.read_string_vector("levels", levels);
73       hide_from_contribs = false;
74       reader.read_bool("hide-from-contribs", hide_from_contribs);
75     }
76   else
77     {
78       std::cout << "LevelSubset: parse error in info file: " << info_file << std::endl;
79     }
80
81   lisp_free(root_obj);
82 }
83
84 void LevelSubset::load(const std::string& subset)
85 {
86   name = subset;
87   
88   // Check in which directory our subset is located (ie. ~/.supertux/
89   // or SUPERTUX_DATADIR)
90   std::string filename;
91   filename = st_dir + "/levels/" + subset + "/info";
92   if (access(filename.c_str(), R_OK) != 0)
93     {
94       filename = datadir + "/levels/" + subset + "/info";
95       if (access(filename.c_str(), R_OK) != 0)
96         std::cout << "Error: LevelSubset: couldn't find subset: " << subset << std::endl;
97     }
98   
99   read_info_file(filename);
100
101   if (levels.empty())
102     { // Level info file doesn't define any levels, so read the
103       // directory to see what we can find
104       std::set<std::string> files;
105   
106       filename = datadir + "/levels/" + subset + "/";
107       files = FileSystem::read_directory(filename);
108
109       filename = st_dir + "/levels/" + subset + "/";
110       std::set<std::string> user_files = FileSystem::read_directory(filename);
111       files.insert(user_files.begin(), user_files.end());
112   
113       for(std::set<std::string>::iterator i = files.begin(); i != files.end(); ++i)
114         {
115           if (has_suffix(*i, ".stl"))
116             levels.push_back(subset+ "/" + *i);
117         }
118     }
119 }
120
121 void
122 LevelSubset::save()
123 {
124   FILE* fi;
125   std::string filename;
126
127   /* Save data file: */
128   filename = "/levels/" + name + "/";
129
130   FileSystem::fcreatedir(filename.c_str());
131   filename = std::string(st_dir) + "/levels/" + name + "/info";
132   if(!FileSystem::fwriteable(filename.c_str()))
133     filename = datadir + "/levels/" + name + "/info";
134   if(FileSystem::fwriteable(filename.c_str()))
135     {
136       fi = fopen(filename.c_str(), "w");
137       if (fi == NULL)
138         {
139           perror(filename.c_str());
140         }
141
142       /* Write header: */
143       fprintf(fi,";; SuperTux-Level-Subset\n");
144       fprintf(fi,"(supertux-level-subset\n");
145
146       /* Save title info: */
147       fprintf(fi,"  (title \"%s\")\n", title.c_str());
148
149       /* Save the description: */
150       fprintf(fi,"  (description \"%s\")\n", description.c_str());
151
152       /* Save the hide from Contrbis menu boolean: */
153       fprintf(fi,"  (hide-from-contribs \"%s\")\n", hide_from_contribs ? "#t" : "#f");
154
155       fprintf( fi,")");
156       fclose(fi);
157     }
158 }
159
160 void
161 LevelSubset::add_level(const std::string& name)
162 {
163   levels.push_back(name);
164 }
165
166 std::string
167 LevelSubset::get_level_filename(unsigned int num)
168 {
169   assert(num < levels.size());
170   return levels[num];
171 }
172
173 int
174 LevelSubset::get_num_levels() const
175 {
176   return levels.size();
177 }
178
179 /* EOF */