X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibr2c2%2Fvehicle.cpp;h=1c4a5b0f441ec9d5e6b40f711cc54afd176cbf08;hb=54392d65e2053d1eacb4cfcc435f1013993f2973;hp=e0a5c9695f210e11938ab2d4c22b4d69d1d26d10;hpb=1ff06c5bc46a677fa389ef86c6b26664368f1653;p=r2c2.git diff --git a/source/libr2c2/vehicle.cpp b/source/libr2c2/vehicle.cpp index e0a5c96..1c4a5b0 100644 --- a/source/libr2c2/vehicle.cpp +++ b/source/libr2c2/vehicle.cpp @@ -1,10 +1,3 @@ -/* $Id$ - -This file is part of R²C² -Copyright © 2010 Mikkosoft Productions, Mikko Rasa -Distributed under the GPL -*/ - #include #include "catalogue.h" #include "driver.h" @@ -21,15 +14,17 @@ using namespace Msp; namespace R2C2 { Vehicle::Vehicle(Layout &l, const VehicleType &t): - layout(l), + Object(l), type(t), next(0), prev(0), - direction(0), - bogie_dirs(type.get_bogies().size()), front_sensor(0), back_sensor(0) { + axles.assign(type.get_fixed_axles().begin(), type.get_fixed_axles().end()); + bogies.assign(type.get_bogies().begin(), type.get_bogies().end()); + rods.assign(type.get_rods().begin(), type.get_rods().end()); + layout.add_vehicle(*this); } @@ -42,10 +37,18 @@ Vehicle::~Vehicle() layout.remove_vehicle(*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::attach_back(Vehicle &veh) { if(next || veh.prev) - throw InvalidState("Already attached"); + throw attachment_error("already attached"); next = &veh; veh.prev = this; @@ -57,7 +60,7 @@ void Vehicle::attach_back(Vehicle &veh) void Vehicle::attach_front(Vehicle &veh) { if(prev || veh.next) - throw InvalidState("Already attached"); + throw attachment_error("already attached"); prev = &veh; veh.next = this; @@ -69,7 +72,7 @@ void Vehicle::attach_front(Vehicle &veh) void Vehicle::detach_back() { if(!next) - throw InvalidState("Not attached"); + throw attachment_error("not attached"); next->prev = 0; next = 0; @@ -78,15 +81,18 @@ void Vehicle::detach_back() void Vehicle::detach_front() { if(!prev) - throw InvalidState("Not attached"); + throw attachment_error("not attached"); prev->next = 0; 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_pos = TrackPosition(t, o); if(m==FRONT_AXLE) track_pos.advance(-type.get_front_axle_offset()); @@ -118,48 +124,70 @@ void Vehicle::advance(float d) { track_pos.advance(d); update_position(); + turn_axles(d); propagate_position(); } -float Vehicle::get_bogie_direction(unsigned i) const +const Vehicle::Axle &Vehicle::get_fixed_axle(unsigned i) const { - if(i>=bogie_dirs.size()) - throw InvalidParameterValue("Bogie index out of range"); - return bogie_dirs[i]; + if(i>=axles.size()) + throw out_of_range("Vehicle::get_fixed_axle"); + return axles[i]; +} + +const Vehicle::Bogie &Vehicle::get_bogie(unsigned i) const +{ + if(i>=bogies.size()) + throw out_of_range("Vehicle::get_bogie"); + return bogies[i]; +} + +const Vehicle::Axle &Vehicle::get_bogie_axle(unsigned i, unsigned j) const +{ + if(i>=bogies.size()) + throw out_of_range("Vehicle::get_bogie_axle"); + if(j>=bogies[i].axles.size()) + throw out_of_range("Vehicle::get_bogie_axle"); + return bogies[i].axles[j]; +} + +const Vehicle::Rod &Vehicle::get_rod(unsigned i) const +{ + if(i>=rods.size()) + throw out_of_range("Vehicle::get_rod"); + return rods[i]; } void Vehicle::update_position() { TrackPoint tp; - const vector &axles = type.get_axles(); - const vector &bogies = type.get_bogies(); if(axles.size()>=2) { - float wheelbase = axles.front().position-axles.back().position; - tp = get_point(track_pos, wheelbase, -axles.back().position/wheelbase); + float wheelbase = axles.front().type->position-axles.back().type->position; + tp = get_point(track_pos, wheelbase, -axles.back().type->position/wheelbase); } else if(bogies.size()>=2) { TrackPosition front = track_pos; - front.advance(bogies.front().position); + front.advance(bogies.front().type->position); TrackPosition back = track_pos; - back.advance(bogies.back().position); - float bogie_spacing = bogies.front().position-bogies.back().position; + back.advance(bogies.back().type->position); + float bogie_spacing = bogies.front().type->position-bogies.back().type->position; adjust_for_distance(front, back, bogie_spacing); - const vector &front_axles = bogies.front().axles; - float wheelbase = front_axles.front().position-front_axles.back().position; - TrackPoint front_point = get_point(front, wheelbase, -front_axles.back().position/wheelbase); + const vector &front_axles = bogies.front().axles; + float wheelbase = front_axles.front().type->position-front_axles.back().type->position; + TrackPoint front_point = get_point(front, wheelbase, -front_axles.back().type->position/wheelbase); - const vector &back_axles = bogies.back().axles; - wheelbase = back_axles.front().position-back_axles.back().position; - TrackPoint back_point = get_point(back, wheelbase, -back_axles.back().position/wheelbase); + const vector &back_axles = bogies.back().axles; + wheelbase = back_axles.front().type->position-back_axles.back().type->position; + TrackPoint back_point = get_point(back, wheelbase, -back_axles.back().type->position/wheelbase); - tp = get_point(front_point.pos, back_point.pos, -bogies.back().position/bogie_spacing); + tp = get_point(front_point.pos, back_point.pos, -bogies.back().type->position/bogie_spacing); - bogie_dirs.front() = front_point.dir-tp.dir; - bogie_dirs.back() = back_point.dir-tp.dir; + bogies.front().direction = front_point.dir-tp.dir; + bogies.back().direction = back_point.dir-tp.dir; } else tp = track_pos.get_point(); @@ -171,7 +199,7 @@ void Vehicle::update_position() position = tp.pos; position.z += layout.get_catalogue().get_rail_elevation(); - direction = tp.dir; + rotation = tp.dir; } void Vehicle::update_position_from(const Vehicle &veh) @@ -182,7 +210,7 @@ void Vehicle::update_position_from(const Vehicle &veh) float margin = layout.get_catalogue().get_scale(); float dist = distance(veh.position, position); - if(disttdist+margin) + if(!track_pos.track || disttdist+margin) { track_pos = veh.track_pos; track_pos.advance(sign*tdist); @@ -193,6 +221,7 @@ void Vehicle::update_position_from(const Vehicle &veh) track_pos.advance(sign*(tdist-dist)); update_position(); + turn_axles(sign*(tdist-dist)); } void Vehicle::propagate_position() @@ -245,19 +274,80 @@ void Vehicle::check_sensor(float offset, unsigned &sensor) } } +void Vehicle::turn_axles(float d) +{ + for(vector::iterator i=axles.begin(); i!=axles.end(); ++i) + i->angle += Angle::from_radians(d*2/i->type->wheel_dia); + for(vector::iterator i=bogies.begin(); i!=bogies.end(); ++i) + for(vector::iterator j=i->axles.begin(); j!=i->axles.end(); ++j) + j->angle += Angle::from_radians(d*2/j->type->wheel_dia); + + update_rods(); +} + +void Vehicle::update_rods() +{ + for(vector::iterator i=rods.begin(); i!=rods.end(); ++i) + { + if(i->type->pivot==VehicleType::Rod::BODY) + i->position = i->type->pivot_point; + else if(i->type->pivot==VehicleType::Rod::AXLE) + { + const Axle &axle = get_fixed_axle(i->type->pivot_index); + const Vector &pp = i->type->pivot_point; + 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); + const Vector &pos = prod.position; + const Vector &off = i->type->pivot_point; + 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) + { + 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) + { + 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 = (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 { float margin = 0.01*layout.get_catalogue().get_scale(); int adjust_dir = 0; while(1) { - Point front_point = front.get_point().pos; - Point back_point = back.get_point().pos; + Vector front_point = front.get_point().pos; + Vector back_point = back.get_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) @@ -278,15 +368,13 @@ void Vehicle::adjust_for_distance(TrackPosition &front, TrackPosition &back, flo } } -TrackPoint Vehicle::get_point(const Point &front, const Point &back, float ratio) const +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 = Point(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; } @@ -303,16 +391,54 @@ TrackPoint Vehicle::get_point(const TrackPosition &pos, float tdist, float ratio return get_point(front.get_point().pos, back.get_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) +{ } + + +Vehicle::Bogie::Bogie(const VehicleType::Bogie &t): + 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) +{ } + Vehicle::TrackPosition::TrackPosition(): - track(0), - ep(0), offs(0) { } -Vehicle::TrackPosition::TrackPosition(Track *t, unsigned e, float o): +Vehicle::TrackPosition::TrackPosition(const TrackIter &t, float o): track(t), - ep(e), offs(o) { } @@ -322,33 +448,30 @@ void Vehicle::TrackPosition::advance(float d) return; offs += d; - TrackIter iter(track, ep); - while(iter) + while(track) { - float path_len = iter->get_type().get_path_length(iter->get_active_path()); + float path_len = track->get_type().get_path_length(track->get_active_path()); if(offs>path_len) { offs -= path_len; - iter = iter.next(); + track = track.next(); } else break; } - while(iter && offs<0) + while(track && offs<0) { - iter = iter.flip().reverse(); + track = track.flip().reverse(); - if(iter) + if(track) { - float path_len = iter->get_type().get_path_length(iter->get_active_path()); + float path_len = track->get_type().get_path_length(track->get_active_path()); offs += path_len; } } - track = iter.track(); - ep = iter.entry(); if(!track) offs = 0; } @@ -356,7 +479,7 @@ void Vehicle::TrackPosition::advance(float d) TrackPoint Vehicle::TrackPosition::get_point() const { if(track) - return track->get_point(ep, offs); + return track->get_point(track.entry(), offs); else return TrackPoint(); }