Fix memory leaks and other bad stuff

[r2c2.git] / source / libmarklin / locotype.h

diff --git a/source/libmarklin/locotype.h b/source/libmarklin/locotype.h
index bce5c57249ff8f8601e5e78289f8672203240f15..5fd34d8c32f20d56d043075e7fa5dfd9d72ed985 100644 (file)
--- a/source/libmarklin/locotype.h
+++ b/source/libmarklin/locotype.h
@@ -1,3 +1,10 @@
+/* $Id$
+
+This file is part of the MSP Märklin suite
+Copyright © 2006-2009  Mikkosoft Productions, Mikko Rasa
+Distributed under the GPL
+*/
+
 #ifndef LIBMARKLIN_LOCOTYPE_H_
 #define LIBMARKLIN_LOCOTYPE_H_
 
@@ -8,14 +15,10 @@ namespace Marklin {
 class LocoType
 {
 public:
-       class Loader: public Msp::DataFile::Loader
+       class Loader: public Msp::DataFile::BasicLoader<LocoType>
        {
-       private:
-               LocoType &ltype;
-
        public:
                Loader(LocoType &);
-               LocoType &get_object() const { return ltype; }
        private:
                void function(unsigned, const std::string &);
        };
@@ -29,6 +32,7 @@ public:
        LocoType(unsigned);
        unsigned get_article_number() const { return art_nr; }
        const std::string &get_name() const { return name; }
+       unsigned get_max_function() const;
        const std::map<unsigned, std::string> &get_functions() const { return funcs; }
 };