Implement train priority system

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

diff --git a/source/libmarklin/catalogue.h b/source/libmarklin/catalogue.h
index e531941b40fb6e1c072491f90094b2ae8164ce1a..524d3f2b56a6593cb82c859af044dfbf027838d9 100644 (file)
--- a/source/libmarklin/catalogue.h
+++ b/source/libmarklin/catalogue.h
@@ -17,6 +17,7 @@ namespace Marklin {
 
 class LocoType;
 class TrackType;
+class VehicleType;
 
 class Catalogue
 {
@@ -33,15 +34,20 @@ public:
                void rail_profile();
                void scale(float, float);
                void track(unsigned);
+               void vehicle(unsigned);
        };
 
+       sigc::signal<void, const TrackType &> signal_track_added;
+       sigc::signal<void, const VehicleType &> signal_vehicle_added;
+
 private:
        float scale;
        float gauge;
        Profile rail_profile;
        Profile ballast_profile;
+       Profile path_profile;
        std::map<unsigned, TrackType *> tracks;
-       std::map<unsigned, LocoType *> locos;
+       std::map<unsigned, VehicleType *> vehicles;
        Layout layout;
 
 public:
@@ -52,10 +58,17 @@ public:
        float get_gauge() const { return gauge; }
        const Profile &get_rail_profile() const { return rail_profile; }
        const Profile &get_ballast_profile() const { return ballast_profile; }
-       TrackType &get_track(unsigned) const;
+       const Profile &get_path_profile() const { return path_profile; }
+
+       void add_track(TrackType &);
+       const TrackType &get_track(unsigned) const;
        const std::map<unsigned, TrackType *> &get_tracks() const { return tracks; }
-       LocoType &get_locomotive(unsigned) const;
-       const std::map<unsigned, LocoType *> &get_locomotives() const { return locos; }
+
+       void add_vehicle(VehicleType &);
+       const VehicleType &get_vehicle(unsigned) const;
+       const LocoType &get_locomotive(unsigned) const;
+       const std::map<unsigned, VehicleType *> &get_vehicles() const { return vehicles; }
+
        Layout &get_layout() { return layout; }
 };