X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibmarklin%2Ftracktype.h;h=d13ea28c98354785075e08177ae514b985b92bf5;hb=9b05c573a38639827697fe393d55b7c76f5bde45;hp=99989be664ae15e7ba3f14dd64be865b8da186e5;hpb=f409cd39a687900724d987c9db629cbff0dd49be;p=r2c2.git

diff --git a/source/libmarklin/tracktype.h b/source/libmarklin/tracktype.h
index 99989be..d13ea28 100644
--- a/source/libmarklin/tracktype.h
+++ b/source/libmarklin/tracktype.h
@@ -9,7 +9,8 @@ Distributed under the GPL
 #define LIBMARKLIN_TRACKTYPE_H_
 
 #include <msp/datafile/loader.h>
-#include "endpoint.h"
+#include "articlenumber.h"
+#include "geometry.h"
 #include "trackpart.h"
 
 namespace Marklin {
@@ -17,6 +18,15 @@ namespace Marklin {
 class TrackType
 {
 public:
+	struct Endpoint
+	{
+		Point pos;
+		float dir;  // Direction outwards from the endpoint
+		unsigned paths;
+
+		Endpoint(float, float, float, unsigned);
+	};
+
 	class Loader: public Msp::DataFile::BasicLoader<TrackType>
 	{
 	public:
@@ -28,16 +38,17 @@ public:
 	};
 
 private:
-	unsigned art_nr;
+	ArticleNumber art_nr;
 	std::string description;
 	std::vector<TrackPart> parts;
 	std::vector<Endpoint> endpoints;
 	bool double_address;
+	unsigned autofit_preference;
 
 public:
-	TrackType(unsigned);
+	TrackType(const ArticleNumber &);
 
-	unsigned get_article_number() const { return art_nr; }
+	const ArticleNumber &get_article_number() const { return art_nr; }
 	const std::string &get_description() const { return description; }
 	float get_total_length() const;
 	float get_path_length(int) const;
@@ -46,8 +57,10 @@ public:
 	bool is_turnout() const;
 	bool is_dead_end() const;
 	bool is_double_address() const { return double_address; }
+	unsigned get_autofit_preference() const { return autofit_preference; }
 	const std::vector<TrackPart> &get_parts() const { return parts; }
 	const std::vector<Endpoint> &get_endpoints() const { return endpoints; }
+	const Endpoint &get_endpoint(unsigned) const;
 	TrackPoint get_point(unsigned, unsigned, float) const;
 
 private: