namespace R2C2 {
Vehicle::Vehicle(Layout &l, const VehicleType &t):
- layout(l),
+ Object(l),
type(t),
+ train(0),
next(0),
prev(0),
- direction(0),
front_sensor(0),
back_sensor(0)
{
bogies.assign(type.get_bogies().begin(), type.get_bogies().end());
rods.assign(type.get_rods().begin(), type.get_rods().end());
- layout.add_vehicle(*this);
+ layout.add(*this);
}
Vehicle::~Vehicle()
detach_back();
if(prev)
detach_front();
- layout.remove_vehicle(*this);
+ layout.remove(*this);
+}
+
+Vehicle *Vehicle::clone(Layout *to_layout) const
+{
+ Vehicle *veh = new Vehicle((to_layout ? *to_layout : layout), type);
+ veh->set_position(position);
+ veh->set_rotation(rotation);
+ return veh;
+}
+
+void Vehicle::set_train(Train *t)
+{
+ train = t;
}
void Vehicle::attach_back(Vehicle &veh)
next = &veh;
veh.prev = this;
- if(track_pos.track)
+ if(track)
propagate_backward();
}
prev = 0;
}
-void Vehicle::place(Track &t, unsigned e, float o, PlaceMode m)
+void Vehicle::place(const TrackIter &t, float o, PlaceMode m)
{
- track_pos = TrackPosition(&t, e, o);
+ if(!t)
+ throw invalid_argument("Vehicle::place");
+
+ track = TrackOffsetIter(t, o);
if(m==FRONT_AXLE)
- track_pos.advance(-type.get_front_axle_offset());
+ track = track.advance(-type.get_front_axle_offset());
else if(m==FRONT_BUFFER)
- track_pos.advance(-type.get_length()/2);
+ track = track.advance(-type.get_length()/2);
else if(m==BACK_AXLE)
- track_pos.advance(-type.get_back_axle_offset());
+ track = track.advance(-type.get_back_axle_offset());
else if(m==BACK_BUFFER)
- track_pos.advance(type.get_length()/2);
+ track = track.advance(type.get_length()/2);
update_position();
propagate_position();
void Vehicle::unplace()
{
- if(!track_pos.track)
+ if(!track)
return;
- track_pos = TrackPosition();
+ track = TrackOffsetIter();
if(prev)
prev->unplace();
void Vehicle::advance(float d)
{
- track_pos.advance(d);
+ track = track.advance(d);
update_position();
turn_axles(d);
propagate_position();
if(axles.size()>=2)
{
float wheelbase = axles.front().type->position-axles.back().type->position;
- tp = get_point(track_pos, wheelbase, -axles.back().type->position/wheelbase);
+ tp = get_point(track, wheelbase, -axles.back().type->position/wheelbase);
}
else if(bogies.size()>=2)
{
- TrackPosition front = track_pos;
- front.advance(bogies.front().type->position);
- TrackPosition back = track_pos;
- back.advance(bogies.back().type->position);
+ TrackOffsetIter front = track.advance(bogies.front().type->position);
+ TrackOffsetIter back = track.advance(bogies.back().type->position);
float bogie_spacing = bogies.front().type->position-bogies.back().type->position;
adjust_for_distance(front, back, bogie_spacing);
bogies.back().direction = back_point.dir-tp.dir;
}
else
- tp = track_pos.get_point();
+ tp = track.point();
if(!prev)
check_sensor(type.get_front_axle_offset(), front_sensor);
position = tp.pos;
position.z += layout.get_catalogue().get_rail_elevation();
- direction = tp.dir;
+ rotation = tp.dir;
+ signal_moved.emit();
}
void Vehicle::update_position_from(const Vehicle &veh)
float margin = layout.get_catalogue().get_scale();
float dist = distance(veh.position, position);
- if(!track_pos.track || dist<tdist-margin || dist>tdist+margin)
+ if(!track || dist<tdist-margin || dist>tdist+margin)
{
- track_pos = veh.track_pos;
- track_pos.advance(sign*tdist);
+ track = veh.track.advance(sign*tdist);
update_position();
dist = distance(veh.position, position);
}
- track_pos.advance(sign*(tdist-dist));
+ track = track.advance(sign*(tdist-dist));
update_position();
turn_axles(sign*(tdist-dist));
}
void Vehicle::check_sensor(float offset, unsigned &sensor)
{
- TrackPosition pos = track_pos;
- pos.advance(offset);
- unsigned s = pos.track->get_sensor_id();
+ TrackOffsetIter iter = track.advance(offset);
+ unsigned s = iter->get_sensor_id();
if(s!=sensor)
{
/* Sensor ID under axle has changed. Deduce movement direction by using
more robust would be preferable. */
unsigned old = sensor;
sensor = s;
- unsigned mid = track_pos.track->get_sensor_id();
+ unsigned mid = track->get_sensor_id();
if(s && s!=mid)
/* There's a sensor and it's different from mid. We've just entered
void Vehicle::turn_axles(float d)
{
for(vector<Axle>::iterator i=axles.begin(); i!=axles.end(); ++i)
- i->angle += d*2/i->type->wheel_dia;
+ i->angle += Angle::from_radians(d*2/i->type->wheel_dia);
for(vector<Bogie>::iterator i=bogies.begin(); i!=bogies.end(); ++i)
for(vector<Axle>::iterator j=i->axles.begin(); j!=i->axles.end(); ++j)
- j->angle += d*2/j->type->wheel_dia;
+ j->angle += Angle::from_radians(d*2/j->type->wheel_dia);
update_rods();
}
else if(i->type->pivot==VehicleType::Rod::AXLE)
{
const Axle &axle = get_fixed_axle(i->type->pivot_index);
- float c = cos(axle.angle);
- float s = sin(axle.angle);
const Vector &pp = i->type->pivot_point;
- i->position = Vector(axle.type->position+pp.x*c+pp.z*s, pp.y, axle.type->wheel_dia/2+pp.z*c-pp.x*s);
+ Transform trans = Transform::rotation(axle.angle, Vector(0, -1, 0));
+ i->position = Vector(axle.type->position, 0, axle.type->wheel_dia/2)+trans.transform(pp);
}
else if(i->type->pivot==VehicleType::Rod::ROD)
{
const Rod &prod = get_rod(i->type->pivot_index);
- float c = cos(prod.angle);
- float s = sin(prod.angle);
const Vector &pos = prod.position;
const Vector &off = i->type->pivot_point;
- i->position = Vector(pos.x+off.x*c-off.z*s, pos.y+off.y, pos.z+off.z*c+off.x*s);
+ Transform trans = Transform::rotation(prod.angle, Vector(0, 1, 0));
+ i->position = pos+trans.transform(off);
}
if(i->type->connect_index>=0)
Rod &crod = rods[i->type->connect_index];
if(i->type->limit==VehicleType::Rod::ROTATE && crod.type->limit==VehicleType::Rod::SLIDE_X)
{
- float dx = (crod.position.x+i->type->connect_offset.x)-i->position.x;
- float dz = (crod.position.z+i->type->connect_offset.z)-i->position.z;
- float cd = sqrt(i->type->connect_point.x*i->type->connect_point.x+i->type->connect_point.z*i->type->connect_point.z);
- float ca = atan2(i->type->connect_point.z, i->type->connect_point.x);
- dx = sqrt(cd*cd-dz*dz)*(dx>0 ? 1 : -1);
- i->angle = atan2(dz, dx)-ca;
- crod.position.x = i->position.x+dx-i->type->connect_offset.x;
+ Vector span = crod.position+i->type->connect_offset-i->position;
+ float cd = i->type->connect_point.norm();
+ Angle ca = Geometry::atan2(i->type->connect_point.z, i->type->connect_point.x);
+ span.x = sqrt(cd*cd-span.z*span.z)*(span.x>0 ? 1 : -1);
+ i->angle = Geometry::atan2(span.z, span.x)-ca;
+ crod.position.x = i->position.x+span.x-i->type->connect_offset.x;
}
else if(i->type->limit==VehicleType::Rod::ROTATE && crod.type->limit==VehicleType::Rod::ROTATE)
{
- float dx = crod.position.x-i->position.x;
- float dz = crod.position.z-i->position.z;
- float d = sqrt(dx*dx+dz*dz);
- float cd1 = sqrt(i->type->connect_point.x*i->type->connect_point.x+i->type->connect_point.z*i->type->connect_point.z);
- float cd2 = sqrt(i->type->connect_offset.x*i->type->connect_offset.x+i->type->connect_offset.z*i->type->connect_offset.z);
+ Vector span = crod.position-i->position;
+ float d = span.norm();
+ float cd1 = i->type->connect_point.norm();
+ float cd2 = i->type->connect_offset.norm();
float a = (d*d+cd1*cd1-cd2*cd2)/(2*d);
float b = sqrt(cd1*cd1-a*a);
- float sign = (dx*i->type->connect_point.z-dz*i->type->connect_point.x>0 ? 1 : -1);
- float cx = (dx*a-dz*b*sign)/d;
- float cz = (dz*a+dx*b*sign)/d;
- float ca1 = atan2(i->type->connect_point.z, i->type->connect_point.x);
- float ca2 = atan2(i->type->connect_offset.z, i->type->connect_offset.x);
- i->angle = atan2(cz, cx)-ca1;
- crod.angle = atan2(cz-dz, cx-dx)-ca2;
+ float sign = (cross(i->type->connect_point, span).y>0 ? 1 : -1);
+ Vector conn = Vector(span.x*a-span.z*b, 0, span.z*a+span.x*b)/(d*sign);
+ Angle ca1 = Geometry::atan2(i->type->connect_point.z, i->type->connect_point.x);
+ Angle ca2 = Geometry::atan2(i->type->connect_offset.z, i->type->connect_offset.x);
+ i->angle = Geometry::atan2(conn.z, conn.x)-ca1;
+ crod.angle = Geometry::atan2(conn.z-span.z, conn.x-span.x)-ca2;
}
}
}
}
-void Vehicle::adjust_for_distance(TrackPosition &front, TrackPosition &back, float tdist, float ratio) const
+void Vehicle::adjust_for_distance(TrackOffsetIter &front, TrackOffsetIter &back, float tdist, float ratio) const
{
float margin = 0.01*layout.get_catalogue().get_scale();
int adjust_dir = 0;
while(1)
{
- Vector front_point = front.get_point().pos;
- Vector back_point = back.get_point().pos;
+ Vector front_point = front.point().pos;
+ Vector back_point = back.point().pos;
- float dx = front_point.x-back_point.x;
- float dy = front_point.y-back_point.y;
- float dz = front_point.z-back_point.z;
- float dist = sqrt(dx*dx+dy*dy+dz*dz);
+ float dist = distance(front_point, back_point);
float diff = tdist-dist;
if(diff<-margin && adjust_dir<=0)
else
return;
- front.advance(diff*(1-ratio));
- back.advance(-diff*ratio);
+ front = front.advance(diff*(1-ratio));
+ back = back.advance(-diff*ratio);
}
}
TrackPoint Vehicle::get_point(const Vector &front, const Vector &back, float ratio) const
{
- float dx = front.x-back.x;
- float dy = front.y-back.y;
- float dz = front.z-back.z;
+ Vector span = front-back;
TrackPoint tp;
- tp.pos = Vector(back.x+dx*ratio, back.y+dy*ratio, back.z+dz*ratio);
- tp.dir = atan2(dy, dx);
+ tp.pos = back+span*ratio;
+ tp.dir = Geometry::atan2(span.y, span.x);
return tp;
}
-TrackPoint Vehicle::get_point(const TrackPosition &pos, float tdist, float ratio) const
+TrackPoint Vehicle::get_point(const TrackOffsetIter &iter, float tdist, float ratio) const
{
- TrackPosition front = pos;
- front.advance(tdist*(1-ratio));
-
- TrackPosition back = pos;
- back.advance(-tdist*ratio);
+ TrackOffsetIter front = iter.advance(tdist*(1-ratio));
+ TrackOffsetIter back = iter.advance(-tdist*ratio);
adjust_for_distance(front, back, tdist, ratio);
- return get_point(front.get_point().pos, back.get_point().pos, ratio);
+ return get_point(front.point().pos, back.point().pos, ratio);
+}
+
+unsigned Vehicle::get_n_link_slots() const
+{
+ return 2;
+}
+
+Vehicle *Vehicle::get_link(unsigned i) const
+{
+ if(i>=2)
+ throw out_of_range("Vehicle::get_link");
+
+ return (i==0 ? prev : next);
+}
+
+int Vehicle::get_link_slot(const Object &other) const
+{
+ if(&other==prev)
+ return 0;
+ else if(&other==next)
+ return 1;
+ else
+ return -1;
}
Vehicle::Axle::Axle(const VehicleType::Axle &t):
- type(&t),
- angle(0)
+ type(&t)
{ }
Vehicle::Bogie::Bogie(const VehicleType::Bogie &t):
- type(&t),
- direction(0)
+ type(&t)
{
for(VehicleType::AxleArray::const_iterator i=type->axles.begin(); i!=type->axles.end(); ++i)
axles.push_back(*i);
Vehicle::Rod::Rod(const VehicleType::Rod &t):
- type(&t),
- angle(0)
-{ }
-
-
-Vehicle::TrackPosition::TrackPosition():
- track(0),
- ep(0),
- offs(0)
-{ }
-
-Vehicle::TrackPosition::TrackPosition(Track *t, unsigned e, float o):
- track(t),
- ep(e),
- offs(o)
+ type(&t)
{ }
-void Vehicle::TrackPosition::advance(float d)
-{
- if(!track)
- return;
-
- offs += d;
- TrackIter iter(track, ep);
- while(iter)
- {
- float path_len = iter->get_type().get_path_length(iter->get_active_path());
-
- if(offs>path_len)
- {
- offs -= path_len;
- iter = iter.next();
- }
- else
- break;
- }
-
- while(iter && offs<0)
- {
- iter = iter.flip().reverse();
-
- if(iter)
- {
- float path_len = iter->get_type().get_path_length(iter->get_active_path());
- offs += path_len;
- }
- }
-
- track = iter.track();
- ep = iter.entry();
- if(!track)
- offs = 0;
-}
-
-TrackPoint Vehicle::TrackPosition::get_point() const
-{
- if(track)
- return track->get_point(ep, offs);
- else
- return TrackPoint();
-}
-
} // namespace R2C2
projects / r2c2.git / blobdiff