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(const ArticleNumber &an):
16 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("height", &Loader::height);
111 add("length", &Loader::length);
112 add("locomotive", &VehicleType::locomotive);
113 add("maximum_speed", &VehicleType::max_speed);
114 add("mirror_rods", &Loader::mirror_rods);
115 add("object", &VehicleType::object);
116 add("rod", &Loader::rod);
117 add("rotate_object", &VehicleType::rotate_object);
118 add("swap_direction", &VehicleType::swap_direction);
119 add("width", &Loader::width);
122 void VehicleType::Loader::finish()
124 for(unsigned i=0; i<obj.bogies.size(); ++i)
126 obj.bogies[i].index = i;
127 for(unsigned j=0; j<obj.bogies[i].axles.size(); ++j)
129 obj.bogies[i].axles[j] = &obj.axles[obj.bogies[i].first_axle+j];
130 obj.bogies[i].axles[j]->bogie = &obj.bogies[i];
131 obj.bogies[i].axles[j]->position += obj.bogies[i].position;
135 for(unsigned i=0; i<obj.axles.size(); ++i)
137 obj.axles[i].index = i;
138 if(!obj.axles[i].bogie)
139 obj.fixed_axles.push_back(&obj.axles[i]);
142 for(TagMap::const_iterator i=rod_tags.begin(); i!=rod_tags.end(); ++i)
143 if(i->second>=0x10000)
144 throw runtime_error(format("unresolved reference to %s\n", i->first));
146 obj.shape = new Geometry::TransformedShape<float, 3>(
147 Geometry::Box<float>(obj.length, obj.width, obj.height),
148 Transform::translation(Vector(0, 0, obj.height/2)));
151 void VehicleType::Loader::axle()
155 obj.axles.push_back(axl);
158 void VehicleType::Loader::bogie()
161 Bogie::Loader ldr(obj, bog);
163 obj.bogies.push_back(bog);
166 void VehicleType::Loader::function(unsigned i, const string &f)
168 obj.functions[i] = f;
171 void VehicleType::Loader::height(float h)
176 void VehicleType::Loader::length(float l)
181 void VehicleType::Loader::mirror_rods()
183 MirrorParametersLoader params;
184 load_sub_with(params);
185 Regex r_filter(params.filter);
187 vector<unsigned> mirror_indices(obj.rods.size(), 0);
188 for(TagMap::const_iterator i=rod_tags.begin(); i!=rod_tags.end(); ++i)
189 if(i->second<0x10000 && r_filter.match(i->first))
190 mirror_indices[i->second] = 1;
192 for(unsigned i=0, j=obj.rods.size(); i<mirror_indices.size(); ++i)
193 if(mirror_indices[i])
194 mirror_indices[i] = j++;
196 Transform axle_trans = Transform::rotation(params.phase_offset, Vector(0, 1, 0));
198 for(unsigned i=0; i<mirror_indices.size(); ++i)
200 if(!mirror_indices[i])
203 Rod mr = obj.rods[i];
204 mr.initial_position.y = -mr.initial_position.y;
205 mr.mirror_object = !mr.mirror_object;
206 for(vector<RodConstraint>::iterator j=mr.constraints.begin(); j!=mr.constraints.end(); ++j)
208 j->target_position.y = -j->target_position.y;
209 j->local_position.y = -j->local_position.y;
210 j->axis.y = -j->axis.y;
211 if(j->target==RodConstraint::ROD && mirror_indices[j->target_index])
212 j->target_index = mirror_indices[j->target_index];
213 else if(j->target==RodConstraint::AXLE)
214 j->target_position = axle_trans.transform(j->target_position);
217 obj.rods.push_back(mr);
221 void VehicleType::Loader::rod(const string &t)
224 Rod::Loader ldr(rd, rod_tags);
227 unsigned n = obj.rods.size();
228 if(rod_tags.count(t))
230 unsigned p = rod_tags[t];
231 for(vector<Rod>::iterator i=obj.rods.begin(); i!=obj.rods.end(); ++i)
232 for(vector<RodConstraint>::iterator j=i->constraints.begin(); j!=i->constraints.end(); ++j)
233 if(j->target_index==p)
237 obj.rods.push_back(rd);
240 void VehicleType::Loader::width(float w)
246 VehicleType::Axle::Loader::Loader(Axle &a):
247 DataFile::ObjectLoader<Axle>(a)
249 add("object", &Axle::object);
250 add("position", &Loader::position);
251 add("powered", &Axle::powered);
252 add("wheel_diameter", &Loader::wheel_diameter);
255 void VehicleType::Axle::Loader::position(float p)
257 obj.local_position = p/1000;
258 obj.position = obj.local_position;
261 void VehicleType::Axle::Loader::wheel_diameter(float d)
263 obj.wheel_dia = d/1000;
267 VehicleType::Bogie::Loader::Loader(VehicleType &t, Bogie &b):
268 DataFile::ObjectLoader<Bogie>(b),
271 add("axle", &Loader::axle);
272 add("object", &Bogie::object);
273 add("position", &Loader::position);
274 add("rotate_object", &Bogie::rotate_object);
277 void VehicleType::Bogie::Loader::axle()
281 if(obj.axles.empty())
282 obj.first_axle = parent.axles.size();
283 parent.axles.push_back(axl);
284 // Actual pointers will be filled after everything is loaded
285 obj.axles.push_back(0);
288 void VehicleType::Bogie::Loader::position(float p)
290 obj.position = p/1000;
294 VehicleType::RodConstraint::RodConstraint():
301 VehicleType::RodConstraint::Loader::Loader(RodConstraint &c, TagMap &t):
302 DataFile::ObjectLoader<RodConstraint>(c),
305 add("axis", &Loader::axis);
306 add("local_position", &Loader::local_position);
307 add("target_axle", &Loader::target_axle);
308 add("target_position", &Loader::target_position);
309 add("target_rod", &Loader::target_rod);
312 void VehicleType::RodConstraint::Loader::axis(float x, float y, float z)
314 obj.axis = Vector(x, y, z);
315 obj.axis.normalize();
318 void VehicleType::RodConstraint::Loader::local_position(float x, float y, float z)
320 obj.local_position = Vector(x/1000, y/1000, z/1000);
323 void VehicleType::RodConstraint::Loader::target_axle(unsigned i)
326 obj.target_index = i;
330 void VehicleType::RodConstraint::Loader::target_position(float x, float y, float z)
332 obj.target_position = Vector(x/1000, y/1000, z/1000);
335 void VehicleType::RodConstraint::Loader::target_rod(const string &n)
338 TagMap::iterator i = tags.find(n);
340 obj.target_index = i->second;
343 obj.target_index = 0x10000+tags.size();
344 tags[n] = obj.target_index;
349 VehicleType::Rod::Loader::Loader(Rod &r, TagMap &t):
350 DataFile::ObjectLoader<Rod>(r),
353 add("initial_position", &Loader::initial_position);
354 add("mirror_object", &Rod::mirror_object);
355 add("move", &Loader::constraint<RodConstraint::MOVE>);
356 add("object", &Rod::object);
357 add("rotate", &Loader::constraint<RodConstraint::ROTATE>);
358 add("slide", &Loader::constraint<RodConstraint::SLIDE>);
361 template<VehicleType::RodConstraint::Type t>
362 void VehicleType::Rod::Loader::constraint()
367 obj.constraints.push_back(cns);
370 void VehicleType::Rod::Loader::initial_position(float x, float y, float z)
372 obj.initial_position = Vector(x/1000, y/1000, z/1000);
376 VehicleType::MirrorParametersLoader::MirrorParametersLoader()
378 add("filter", &MirrorParametersLoader::filt);
379 add("phase_offset", &MirrorParametersLoader::phase_offs);
382 void VehicleType::MirrorParametersLoader::filt(const string &f)
387 void VehicleType::MirrorParametersLoader::phase_offs(float o)
389 phase_offset = Angle::from_degrees(o);