]> git.xonotic.org Git - xonotic/netradiant.git/blob - libs/modulesystem/modulesmap.h
fix int overflow in "loof"
[xonotic/netradiant.git] / libs / modulesystem / modulesmap.h
1 /*
2 Copyright (C) 2001-2006, William Joseph.
3 All Rights Reserved.
4
5 This file is part of GtkRadiant.
6
7 GtkRadiant is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2 of the License, or
10 (at your option) any later version.
11
12 GtkRadiant 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 GtkRadiant; if not, write to the Free Software
19 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
20 */
21
22 #if !defined(INCLUDED_MODULESYSTEM_MODULESMAP_H)
23 #define INCLUDED_MODULESYSTEM_MODULESMAP_H
24
25 #include "modulesystem.h"
26 #include "string/string.h"
27 #include <map>
28 #include <set>
29
30 template<typename Type>
31 class ModulesMap : public Modules<Type>
32 {
33   typedef std::map<CopiedString, Module*> modules_t;
34   modules_t m_modules;
35 public:
36   ~ModulesMap()
37   {
38     for(modules_t::iterator i = m_modules.begin(); i != m_modules.end(); ++i) 
39     {
40       (*i).second->release();
41     }
42   }
43
44   typedef modules_t::const_iterator iterator;
45
46   iterator begin() const
47   {
48     return m_modules.begin();
49   }
50   iterator end() const
51   {
52     return m_modules.end();
53   }
54
55   void insert(const char* name, Module& module)
56   {
57     module.capture();
58     if(globalModuleServer().getError())
59     {
60       module.release();
61       globalModuleServer().setError(false);
62     }
63     else
64     {
65       m_modules.insert(modules_t::value_type(name, &module));
66     }
67   }
68
69   Type* find(const char* name)
70   {
71     modules_t::iterator i = m_modules.find(name);
72     if(i != m_modules.end())
73     {
74       return static_cast<Type*>(Module_getTable(*(*i).second));
75     }
76     return 0;
77   }
78
79   Type* findModule(const char* name)
80   {
81     return find(name);
82   }
83   void foreachModule(const typename Modules<Type>::Visitor& visitor)
84   {
85     for(modules_t::iterator i = m_modules.begin(); i != m_modules.end(); ++i)
86     {
87       visitor.visit((*i).first.c_str(), *static_cast<const Type*>(Module_getTable(*(*i).second)));
88     }
89   }
90 };
91
92 template<typename Type>
93 class InsertModules : public ModuleServer::Visitor
94 {
95   ModulesMap<Type>& m_modules;
96 public:
97   InsertModules(ModulesMap<Type>& modules)
98     : m_modules(modules)
99   {
100   }
101   void visit(const char* name, Module& module) const
102   {
103     m_modules.insert(name, module);
104   }
105 };
106
107 template<typename Type>
108 class ModulesRef
109 {
110   ModulesMap<Type> m_modules;
111 public:
112   ModulesRef(const char* names)
113   {
114     if(!globalModuleServer().getError())
115     {
116       if(string_equal(names, "*"))
117       {
118         InsertModules<Type> visitor(m_modules);
119         globalModuleServer().foreachModule(typename Type::Name(), typename Type::Version(), visitor);
120       }
121       else
122       {
123         StringTokeniser tokeniser(names);
124         for(;;)
125         {
126           const char* name = tokeniser.getToken();
127           if(string_empty(name))
128           {
129             break;
130           }
131           Module* module = globalModuleServer().findModule(typename Type::Name(), typename Type::Version(), name);
132           if(module == 0)
133           {
134             globalModuleServer().setError(true);
135             globalErrorStream() << "ModulesRef::initialise: type=" << makeQuoted(typename Type::Name()) << " version=" << makeQuoted(typename Type::Version()) << " name=" << makeQuoted(name) << " - not found\n";
136             break;
137           }
138           else
139           {
140             m_modules.insert(name, *module);
141           }
142         }
143       }
144     }
145   }
146   ModulesMap<Type>& get()
147   {
148     return m_modules;
149   }
150 };
151
152 #endif