1 #include <msp/core/maputils.h>
2 #include <msp/geometry/box.h>
3 #include <msp/geometry/transformedshape.h>
4 #include <msp/strings/format.h>
5 #include "vehicletype.h"
12 VehicleType::VehicleType(const ArticleNumber &an):
15 swap_direction(false),
22 unsigned VehicleType::get_max_function() const
26 return (--functions.end())->first;
29 const VehicleType::Axle &VehicleType::get_axle(unsigned i) const
32 throw out_of_range("VehicleType::get_axle");
36 const VehicleType::Axle &VehicleType::get_fixed_axle(unsigned i) const
38 if(i>=fixed_axles.size())
39 throw out_of_range("VehicleType::get_fixed_axle");
40 return *fixed_axles[i];
43 const VehicleType::Bogie &VehicleType::get_bogie(unsigned i) const
46 throw out_of_range("VehicleType::get_bogie");
50 const VehicleType::Axle &VehicleType::get_bogie_axle(unsigned i, unsigned j) const
53 throw out_of_range("VehicleType::get_bogie_axle");
54 if(j>=bogies[i].axles.size())
55 throw out_of_range("VehicleType::get_bogie_axle");
56 return *bogies[i].axles[j];
59 const VehicleType::Rod &VehicleType::get_rod(unsigned i) const
62 throw out_of_range("VehicleType::get_rod");
66 float VehicleType::get_front_axle_offset() const
69 return axles.front().position;
73 float VehicleType::get_back_axle_offset() const
76 return axles.back().position;
81 VehicleType::Axle::Axle():
91 VehicleType::Bogie::Bogie():
97 VehicleType::Rod::Rod():
102 VehicleType::Loader::Loader(VehicleType &vt):
103 DataFile::DerivedObjectLoader<VehicleType, ObjectType::Loader>(vt)
105 add("axle", &Loader::axle);
106 add("bogie", &Loader::bogie);
107 add("function", &Loader::function);
108 add("height", &Loader::height);
109 add("length", &Loader::length);
110 add("locomotive", &VehicleType::locomotive);
111 add("mirror_rods", &Loader::mirror_rods);
112 add("object", &VehicleType::object);
113 add("rod", &Loader::rod);
114 add("rotate_object", &VehicleType::rotate_object);
115 add("swap_direction", &VehicleType::swap_direction);
116 add("width", &Loader::width);
119 void VehicleType::Loader::finish()
121 for(unsigned i=0; i<obj.bogies.size(); ++i)
123 obj.bogies[i].index = i;
124 for(unsigned j=0; j<obj.bogies[i].axles.size(); ++j)
126 obj.bogies[i].axles[j] = &obj.axles[obj.bogies[i].first_axle+j];
127 obj.bogies[i].axles[j]->bogie = &obj.bogies[i];
128 obj.bogies[i].axles[j]->position += obj.bogies[i].position;
132 for(unsigned i=0; i<obj.axles.size(); ++i)
134 obj.axles[i].index = i;
135 if(!obj.axles[i].bogie)
136 obj.fixed_axles.push_back(&obj.axles[i]);
139 for(TagMap::const_iterator i=rod_tags.begin(); i!=rod_tags.end(); ++i)
140 if(i->second>=0x10000)
141 throw runtime_error(format("unresolved reference to %s\n", i->first));
143 obj.shape = new Geometry::TransformedShape<float, 3>(
144 Geometry::Box<float>(obj.length, obj.width, obj.height),
145 Transform::translation(Vector(0, 0, obj.height/2)));
148 void VehicleType::Loader::axle()
152 obj.axles.push_back(axl);
155 void VehicleType::Loader::bogie()
158 Bogie::Loader ldr(obj, bog);
160 obj.bogies.push_back(bog);
163 void VehicleType::Loader::function(unsigned i, const string &f)
165 obj.functions[i] = f;
168 void VehicleType::Loader::height(float h)
173 void VehicleType::Loader::length(float l)
178 void VehicleType::Loader::mirror_rods()
180 MirrorParametersLoader params;
181 load_sub_with(params);
183 Transform axle_trans = Transform::rotation(params.phase_offset, Vector(0, 1, 0));
185 unsigned index_offset = obj.rods.size();
186 for(unsigned i=0; i<index_offset; ++i)
188 Rod mr = obj.rods[i];
189 mr.initial_position.y = -mr.initial_position.y;
190 mr.mirror_object = !mr.mirror_object;
191 for(vector<RodConstraint>::iterator j=mr.constraints.begin(); j!=mr.constraints.end(); ++j)
193 j->target_position.y = -j->target_position.y;
194 j->local_position.y = -j->local_position.y;
195 j->axis.y = -j->axis.y;
196 if(j->target==RodConstraint::ROD)
197 j->target_index += index_offset;
198 else if(j->target==RodConstraint::AXLE)
199 j->target_position = axle_trans.transform(j->target_position);
202 obj.rods.push_back(mr);
206 void VehicleType::Loader::rod(const string &t)
209 Rod::Loader ldr(rd, rod_tags);
212 unsigned n = obj.rods.size();
213 if(rod_tags.count(t))
215 unsigned p = rod_tags[t];
216 for(vector<Rod>::iterator i=obj.rods.begin(); i!=obj.rods.end(); ++i)
217 for(vector<RodConstraint>::iterator j=i->constraints.begin(); j!=i->constraints.end(); ++j)
218 if(j->target_index==p)
222 obj.rods.push_back(rd);
225 void VehicleType::Loader::width(float w)
231 VehicleType::Axle::Loader::Loader(Axle &a):
232 DataFile::ObjectLoader<Axle>(a)
234 add("object", &Axle::object);
235 add("position", &Loader::position);
236 add("powered", &Axle::powered);
237 add("wheel_diameter", &Loader::wheel_diameter);
240 void VehicleType::Axle::Loader::position(float p)
242 obj.local_position = p/1000;
243 obj.position = obj.local_position;
246 void VehicleType::Axle::Loader::wheel_diameter(float d)
248 obj.wheel_dia = d/1000;
252 VehicleType::Bogie::Loader::Loader(VehicleType &t, Bogie &b):
253 DataFile::ObjectLoader<Bogie>(b),
256 add("axle", &Loader::axle);
257 add("object", &Bogie::object);
258 add("position", &Loader::position);
259 add("rotate_object", &Bogie::rotate_object);
262 void VehicleType::Bogie::Loader::axle()
266 if(obj.axles.empty())
267 obj.first_axle = parent.axles.size();
268 parent.axles.push_back(axl);
269 // Actual pointers will be filled after everything is loaded
270 obj.axles.push_back(0);
273 void VehicleType::Bogie::Loader::position(float p)
275 obj.position = p/1000;
279 VehicleType::RodConstraint::RodConstraint():
286 VehicleType::RodConstraint::Loader::Loader(RodConstraint &c, TagMap &t):
287 DataFile::ObjectLoader<RodConstraint>(c),
290 add("axis", &Loader::axis);
291 add("local_position", &Loader::local_position);
292 add("target_axle", &Loader::target_axle);
293 add("target_position", &Loader::target_position);
294 add("target_rod", &Loader::target_rod);
297 void VehicleType::RodConstraint::Loader::axis(float x, float y, float z)
299 obj.axis = Vector(x, y, z);
300 obj.axis.normalize();
303 void VehicleType::RodConstraint::Loader::local_position(float x, float y, float z)
305 obj.local_position = Vector(x/1000, y/1000, z/1000);
308 void VehicleType::RodConstraint::Loader::target_axle(unsigned i)
311 obj.target_index = i;
315 void VehicleType::RodConstraint::Loader::target_position(float x, float y, float z)
317 obj.target_position = Vector(x/1000, y/1000, z/1000);
320 void VehicleType::RodConstraint::Loader::target_rod(const string &n)
323 TagMap::iterator i = tags.find(n);
325 obj.target_index = i->second;
328 obj.target_index = 0x10000+tags.size();
329 tags[n] = obj.target_index;
334 VehicleType::Rod::Loader::Loader(Rod &r, TagMap &t):
335 DataFile::ObjectLoader<Rod>(r),
338 add("initial_position", &Loader::initial_position);
339 add("mirror_object", &Rod::mirror_object);
340 add("move", &Loader::constraint<RodConstraint::MOVE>);
341 add("object", &Rod::object);
342 add("rotate", &Loader::constraint<RodConstraint::ROTATE>);
343 add("slide", &Loader::constraint<RodConstraint::SLIDE>);
346 template<VehicleType::RodConstraint::Type t>
347 void VehicleType::Rod::Loader::constraint()
352 obj.constraints.push_back(cns);
355 void VehicleType::Rod::Loader::initial_position(float x, float y, float z)
357 obj.initial_position = Vector(x/1000, y/1000, z/1000);
361 VehicleType::MirrorParametersLoader::MirrorParametersLoader()
363 add("phase_offset", &MirrorParametersLoader::phase_offs);
366 void VehicleType::MirrorParametersLoader::phase_offs(float o)
368 phase_offset = Angle::from_degrees(o);