1 #include <msp/core/maputils.h>
2 #include <msp/geometry/box.h>
3 #include <msp/geometry/transformedshape.h>
4 #include <msp/strings/regex.h>
5 #include <msp/strings/format.h>
6 #include "vehicletype.h"
13 VehicleType::VehicleType():
15 swap_direction(false),
24 unsigned VehicleType::get_max_function() const
28 return (--functions.end())->first;
31 const VehicleType::Axle &VehicleType::get_axle(unsigned i) const
34 throw out_of_range("VehicleType::get_axle");
38 const VehicleType::Axle &VehicleType::get_fixed_axle(unsigned i) const
40 if(i>=fixed_axles.size())
41 throw out_of_range("VehicleType::get_fixed_axle");
42 return *fixed_axles[i];
45 const VehicleType::Bogie &VehicleType::get_bogie(unsigned i) const
48 throw out_of_range("VehicleType::get_bogie");
52 const VehicleType::Axle &VehicleType::get_bogie_axle(unsigned i, unsigned j) const
55 throw out_of_range("VehicleType::get_bogie_axle");
56 if(j>=bogies[i].axles.size())
57 throw out_of_range("VehicleType::get_bogie_axle");
58 return *bogies[i].axles[j];
61 const VehicleType::Rod &VehicleType::get_rod(unsigned i) const
64 throw out_of_range("VehicleType::get_rod");
68 float VehicleType::get_front_axle_offset() const
71 return axles.front().position;
75 float VehicleType::get_back_axle_offset() const
78 return axles.back().position;
83 VehicleType::Axle::Axle():
93 VehicleType::Bogie::Bogie():
99 VehicleType::Rod::Rod():
104 VehicleType::Loader::Loader(VehicleType &vt):
105 DataFile::DerivedObjectLoader<VehicleType, ObjectType::Loader>(vt)
107 add("axle", &Loader::axle);
108 add("bogie", &Loader::bogie);
109 add("function", &Loader::function);
110 add("gauge", &Loader::gauge);
111 add("height", &Loader::height);
112 add("length", &Loader::length);
113 add("locomotive", &VehicleType::locomotive);
114 add("maximum_speed", &VehicleType::max_speed);
115 add("mirror_rods", &Loader::mirror_rods);
116 add("object", &VehicleType::object);
117 add("rod", &Loader::rod);
118 add("rotate_object", &VehicleType::rotate_object);
119 add("swap_direction", &VehicleType::swap_direction);
120 add("width", &Loader::width);
123 void VehicleType::Loader::finish()
125 for(unsigned i=0; i<obj.bogies.size(); ++i)
127 obj.bogies[i].index = i;
128 for(unsigned j=0; j<obj.bogies[i].axles.size(); ++j)
130 obj.bogies[i].axles[j] = &obj.axles[obj.bogies[i].first_axle+j];
131 obj.bogies[i].axles[j]->bogie = &obj.bogies[i];
132 obj.bogies[i].axles[j]->position += obj.bogies[i].position;
136 for(unsigned i=0; i<obj.axles.size(); ++i)
138 obj.axles[i].index = i;
139 if(!obj.axles[i].bogie)
140 obj.fixed_axles.push_back(&obj.axles[i]);
143 for(TagMap::const_iterator i=rod_tags.begin(); i!=rod_tags.end(); ++i)
144 if(i->second>=0x10000)
145 throw runtime_error(format("unresolved reference to %s\n", i->first));
147 obj.shape = new Geometry::TransformedShape<float, 3>(
148 Geometry::Box<float>(obj.length, obj.width, obj.height),
149 Transform::translation(Vector(0, 0, obj.height/2)));
152 void VehicleType::Loader::axle()
156 obj.axles.push_back(axl);
159 void VehicleType::Loader::bogie()
162 Bogie::Loader ldr(obj, bog);
164 obj.bogies.push_back(bog);
167 void VehicleType::Loader::function(unsigned i, const string &f)
169 obj.functions[i] = f;
172 void VehicleType::Loader::gauge(float g)
177 void VehicleType::Loader::height(float h)
182 void VehicleType::Loader::length(float l)
187 void VehicleType::Loader::mirror_rods()
189 MirrorParametersLoader params;
190 load_sub_with(params);
191 Regex r_filter(params.filter);
193 vector<unsigned> mirror_indices(obj.rods.size(), 0);
194 for(TagMap::const_iterator i=rod_tags.begin(); i!=rod_tags.end(); ++i)
195 if(i->second<0x10000 && r_filter.match(i->first))
196 mirror_indices[i->second] = 1;
198 for(unsigned i=0, j=obj.rods.size(); i<mirror_indices.size(); ++i)
199 if(mirror_indices[i])
200 mirror_indices[i] = j++;
202 Transform axle_trans = Transform::rotation(params.phase_offset, Vector(0, 1, 0));
204 for(unsigned i=0; i<mirror_indices.size(); ++i)
206 if(!mirror_indices[i])
209 Rod mr = obj.rods[i];
210 mr.initial_position.y = -mr.initial_position.y;
211 mr.mirror_object = !mr.mirror_object;
212 for(vector<RodConstraint>::iterator j=mr.constraints.begin(); j!=mr.constraints.end(); ++j)
214 j->target_position.y = -j->target_position.y;
215 j->local_position.y = -j->local_position.y;
216 j->axis.y = -j->axis.y;
217 if(j->target==RodConstraint::ROD && mirror_indices[j->target_index])
218 j->target_index = mirror_indices[j->target_index];
219 else if(j->target==RodConstraint::AXLE)
220 j->target_position = axle_trans.transform(j->target_position);
223 obj.rods.push_back(mr);
227 void VehicleType::Loader::rod(const string &t)
230 Rod::Loader ldr(rd, rod_tags);
233 unsigned n = obj.rods.size();
234 if(rod_tags.count(t))
236 unsigned p = rod_tags[t];
237 for(vector<Rod>::iterator i=obj.rods.begin(); i!=obj.rods.end(); ++i)
238 for(vector<RodConstraint>::iterator j=i->constraints.begin(); j!=i->constraints.end(); ++j)
239 if(j->target_index==p)
243 obj.rods.push_back(rd);
246 void VehicleType::Loader::width(float w)
252 VehicleType::Axle::Loader::Loader(Axle &a):
253 DataFile::ObjectLoader<Axle>(a)
255 add("object", &Axle::object);
256 add("position", &Loader::position);
257 add("powered", &Axle::powered);
258 add("wheel_diameter", &Loader::wheel_diameter);
261 void VehicleType::Axle::Loader::position(float p)
263 obj.local_position = p/1000;
264 obj.position = obj.local_position;
267 void VehicleType::Axle::Loader::wheel_diameter(float d)
269 obj.wheel_dia = d/1000;
273 VehicleType::Bogie::Loader::Loader(VehicleType &t, Bogie &b):
274 DataFile::ObjectLoader<Bogie>(b),
277 add("axle", &Loader::axle);
278 add("object", &Bogie::object);
279 add("position", &Loader::position);
280 add("rotate_object", &Bogie::rotate_object);
283 void VehicleType::Bogie::Loader::axle()
287 if(obj.axles.empty())
288 obj.first_axle = parent.axles.size();
289 parent.axles.push_back(axl);
290 // Actual pointers will be filled after everything is loaded
291 obj.axles.push_back(0);
294 void VehicleType::Bogie::Loader::position(float p)
296 obj.position = p/1000;
300 VehicleType::RodConstraint::RodConstraint():
307 VehicleType::RodConstraint::Loader::Loader(RodConstraint &c, TagMap &t):
308 DataFile::ObjectLoader<RodConstraint>(c),
311 add("axis", &Loader::axis);
312 add("local_position", &Loader::local_position);
313 add("target_axle", &Loader::target_axle);
314 add("target_position", &Loader::target_position);
315 add("target_rod", &Loader::target_rod);
318 void VehicleType::RodConstraint::Loader::axis(float x, float y, float z)
320 obj.axis = Vector(x, y, z);
321 obj.axis.normalize();
324 void VehicleType::RodConstraint::Loader::local_position(float x, float y, float z)
326 obj.local_position = Vector(x/1000, y/1000, z/1000);
329 void VehicleType::RodConstraint::Loader::target_axle(unsigned i)
332 obj.target_index = i;
336 void VehicleType::RodConstraint::Loader::target_position(float x, float y, float z)
338 obj.target_position = Vector(x/1000, y/1000, z/1000);
341 void VehicleType::RodConstraint::Loader::target_rod(const string &n)
344 TagMap::iterator i = tags.find(n);
346 obj.target_index = i->second;
349 obj.target_index = 0x10000+tags.size();
350 tags[n] = obj.target_index;
355 VehicleType::Rod::Loader::Loader(Rod &r, TagMap &t):
356 DataFile::ObjectLoader<Rod>(r),
359 add("initial_position", &Loader::initial_position);
360 add("mirror_object", &Rod::mirror_object);
361 add("move", &Loader::constraint<RodConstraint::MOVE>);
362 add("object", &Rod::object);
363 add("rotate", &Loader::constraint<RodConstraint::ROTATE>);
364 add("slide", &Loader::constraint<RodConstraint::SLIDE>);
367 template<VehicleType::RodConstraint::Type t>
368 void VehicleType::Rod::Loader::constraint()
373 obj.constraints.push_back(cns);
376 void VehicleType::Rod::Loader::initial_position(float x, float y, float z)
378 obj.initial_position = Vector(x/1000, y/1000, z/1000);
382 VehicleType::MirrorParametersLoader::MirrorParametersLoader()
384 add("filter", &MirrorParametersLoader::filt);
385 add("phase_offset", &MirrorParametersLoader::phase_offs);
388 void VehicleType::MirrorParametersLoader::filt(const string &f)
393 void VehicleType::MirrorParametersLoader::phase_offs(float o)
395 phase_offset = Angle::from_degrees(o);