Add Block::has_track and Route::has_track methods

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

diff --git a/source/libmarklin/block.h b/source/libmarklin/block.h
index df3ed4a525444271c931d4dd02a6c4e70dd5d63b..23a48c0ffbc40344ee3c98a04c5e6833a3173613 100644 (file)
--- a/source/libmarklin/block.h
+++ b/source/libmarklin/block.h
@@ -1,5 +1,12 @@
-#ifndef MARKLIN_3D_SECTION_H_
-#define MARKLIN_3D_SECTION_H_
+/* $Id$
+
+This file is part of the MSP Märklin suite
+Copyright © 2006-2010  Mikkosoft Productions, Mikko Rasa
+Distributed under the GPL
+*/
+
+#ifndef LIBMARKLIN_BLOCK_H_
+#define LIBMARKLIN_BLOCK_H_
 
 #include <list>
 #include <set>
@@ -7,8 +14,9 @@
 
 namespace Marklin {
 
+class Layout;
+class Route;
 class Train;
-class TrafficManager;
 
 class Block
 {
@@ -16,35 +24,45 @@ public:
        struct Endpoint
        {
                Track *track;
-               const Track::Endpoint *track_ep;
-               Block  *link;
-               unsigned routes;
+               unsigned track_ep;
+               Block *link;
+               unsigned paths;
 
-               Endpoint(Track *, const Track::Endpoint *);
+               Endpoint(Track *, unsigned);
        };
-       typedef std::list<Endpoint> EndpointSeq;
 
-       Block(TrafficManager &, Track *);
+private:
+       Layout &layout;
+       unsigned id;
+       unsigned sensor_id;
+       unsigned turnout_id;
+       std::set<Track *> tracks;
+       std::vector<Endpoint> endpoints;
+       Train *train;
+
+public:
+       Block(Layout &, Track &);
+       ~Block();
+
+       unsigned get_id() const { return id; }
        unsigned get_sensor_id() const { return sensor_id; }
-       const TrackSet &get_tracks() const { return tracks; }
-       const Endpoint *get_endpoint_by_link(const Block *) const;
-       const Endpoint *traverse(const Endpoint *) const;
+       unsigned get_turnout_id() const { return turnout_id; }
+       const std::set<Track *> &get_tracks() const { return tracks; }
+       bool has_track(Track &) const;
+       const std::vector<Endpoint> &get_endpoints() const { return endpoints; }
+       int get_endpoint_by_link(Block &) const;
+       unsigned traverse(unsigned, float * =0) const;
+       unsigned traverse(unsigned, const Route *, float * =0) const;
        void check_link(Block &);
-       bool reserve(const Train *);
+       void break_link(Block &);
+       Block *get_link(unsigned) const;
+       bool reserve(Train *);
+       Train *get_train() const { return train; }
        void print_debug();
 private:
-       TrafficManager &trfc_mgr;
-       unsigned       id;
-       unsigned       sensor_id;
-       TrackSet       tracks;
-       EndpointSeq    endpoints;
-       const Train    *train;
-
-       void find_routes(Track *, const Track::Endpoint *, unsigned, TrackSet &);
-
-       static unsigned next_id;
+       void find_paths(Track &, unsigned, unsigned, std::set<Track *> &);
+       void determine_id();
 };
-typedef std::list<Block *> BlockSeq;
 
 } // namespace Marklin