/* This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0. If a copy of the MPL was not distributed with this file, You can obtain one at http://mozilla.org/MPL/2.0/. */ #include "stdafx.h" #include "application/editormode.h" #include "application/editorprojection.h" #include "application/editorgestures.h" #include "application/editormodeshared.h" #include "application/editoruilayer.h" #include "editor/editorFormat.hpp" #include "editor/editorAlignmentStore.hpp" #include "editor/editorGeometry.hpp" #include "editor/editorSwitchStore.hpp" #include "world/Track.h" #include "scene/scenelayers.h" #include "simulation/simulation.h" #include "utilities/Globals.h" #include "utilities/glmHelpers.h" #include "world/TrackSleepers.h" #include "utilities/Logs.h" #include "utilities/translation.h" #include "imgui/imgui.h" #include #include #include #include #include #include #include #include namespace { using geometry::plan_distance; ImU32 const kGradeLine{IM_COL32(90, 230, 110, 255)}; ImU32 const kTangents{IM_COL32(255, 210, 60, 130)}; ImU32 const kTrack{IM_COL32(190, 190, 190, 255)}; ImU32 const kTerrain{IM_COL32(170, 120, 70, 255)}; ImU32 const kFormation{IM_COL32(215, 185, 140, 200)}; ImU32 const kModel{IM_COL32(70, 200, 190, 230)}; // the ground of the terrain model, as measured double constexpr kNoBed{0.05}; // m, a trackbed shallower is none // a design the track lay on when it was kept, within kApplied, was applied; where the track parts from it by more than kStale since, // it was changed by other tools, and the line is read from the track there, over the stretch and kStaleMargin around it double constexpr kApplied{0.01}; double constexpr kStale{0.02}; double constexpr kStaleMargin{15.0}; ImU32 const kGrid{IM_COL32(255, 255, 255, 22)}; ImU32 const kLabel{IM_COL32(200, 200, 200, 200)}; ImU32 const kSwitch{IM_COL32(160, 100, 220, 45)}; ImU32 const kError{IM_COL32(255, 80, 70, 255)}; ImU32 const kWarning{IM_COL32(255, 200, 60, 255)}; ImU32 const kPlan{IM_COL32(120, 190, 255, 255)}; ImU32 const kSwitchLabel{IM_COL32(200, 160, 255, 220)}; ImU32 const kShared{IM_COL32(160, 100, 220, 110)}; // hatching, the colour of the switches ImU32 const kSharedEdge{IM_COL32(200, 160, 255, 220)}; ImU32 const kChip{IM_COL32(14, 15, 19, 210)}; // text on a dark plate, readable over the lines of the plot void plot_label(ImDrawList &Draw, ImVec2 const At, ImU32 const Colour, std::string const &Text) { auto const size{ImGui::CalcTextSize(Text.c_str())}; Draw.AddRectFilled(ImVec2(At.x - 3.0f, At.y - 1.0f), ImVec2(At.x + size.x + 3.0f, At.y + size.y + 1.0f), kChip, 3.0f); Draw.AddText(At, Colour, Text.c_str()); } double nice_step(double const Approximate) { auto const power{std::pow(10.0, std::floor(std::log10(std::max(Approximate, 1e-6))))}; for (auto const factor : {1.0, 2.0, 5.0, 10.0}) if (power * factor >= Approximate) return power * factor; return power * 10.0; } double route_speed(editor_track::route const &Route) { double speed{-1.0}; for (auto const &span : Route.spans) speed = std::max(speed, editor_track::velocity(*span.track)); return speed > 0.0 ? speed : 100.0; } std::string const kProfileMark{"//$p"}; profile::line reversed_line(profile::line Line, double const Length) { std::reverse(Line.points.begin(), Line.points.end()); for (auto &point : Line.points) point.chainage = Length - point.chainage; std::swap(Line.join_start, Line.join_end); return Line; } bool end_locked(profile::line const &Line, profile::context const &Context, int const Index) { if (Index == 0) return Line.join_start && Context.start_joined; if (Index + 1 == static_cast(Line.points.size())) return Line.join_end && Context.end_joined; return false; } bool has_errors(std::vector const &Issues) { return std::any_of(Issues.begin(), Issues.end(), [](profile::issue const &Issue) { return Issue.error; }); } // the chainage of the point on the route sampled, projected onto the stretch between the samples; false off the route bool sampled_chainage(std::vector const &Samples, glm::dvec3 const &Point, double &Chainage) { double best{0.3}; bool found{false}; for (std::size_t i = 0; i + 1 < Samples.size(); ++i) { glm::dvec2 const a{Samples[i].position.x, Samples[i].position.z}; glm::dvec2 const b{Samples[i + 1].position.x, Samples[i + 1].position.z}; glm::dvec2 const p{Point.x, Point.z}; auto const run{b - a}; auto const t{std::clamp(glm::dot(p - a, run) / std::max(1e-12, glm::dot(run, run)), 0.0, 1.0)}; if (auto const distance{glm::length(p - (a + run * t))}; distance < best) { best = distance; Chainage = Samples[i].chainage + (Samples[i + 1].chainage - Samples[i].chainage) * t; found = true; } } return found; } } // format of the lines, coordinates in the space of the file: // profile // speed origin join // norms = k*V^2> // point // end void editor_mode::profile_library_load() { if (m_profiles_loaded) return; m_profiles_loaded = true; m_profiles.clear(); for (auto const layer : scene::Layers.marked_layers(kProfileMark)) { auto const &context{scene::Layers.layer(layer).context_insert()}; auto const height = [&](double const Local) { return context.to_world({0.0, Local, 0.0}).y; }; std::optional current; for (auto const &text : scene::Layers.marked(layer, kProfileMark)) { std::istringstream line{text}; std::string keyword; line >> keyword; if (keyword == "profile") { stored_profile entry; entry.layer = layer; glm::dvec3 start; glm::dvec3 end; glm::dvec3 middle; line >> entry.length >> start.x >> start.y >> start.z >> end.x >> end.y >> end.z >> middle.x >> middle.y >> middle.z; if (line.fail()) { current.reset(); continue; } entry.start = context.to_world(start); entry.end = context.to_world(end); entry.middle = context.to_world(middle); current = entry; continue; } if (false == current.has_value()) continue; auto &entry{*current}; std::string word; if (keyword == "speed") { int start{1}; int end{1}; line >> entry.line.speed >> word >> entry.origin >> word >> start >> end; entry.line.join_start = (start != 0); entry.line.join_end = (end != 0); } else if (keyword == "norms") { auto &norms{entry.line.norms}; line >> norms.grade_max >> norms.grade_switch_max >> norms.radius_factor >> norms.radius_min >> norms.curve_threshold >> norms.element_min >> norms.fixed_tolerance; } else if (keyword == "departure") line >> entry.departure; else if (keyword == "point") { profile::pvi point; int automatic{0}; line >> point.chainage >> point.elevation >> point.radius >> automatic; if (line.fail()) continue; point.elevation = height(point.elevation); point.automatic = (automatic != 0); entry.line.points.push_back(point); } else if (keyword == "end") { if (entry.line.points.size() >= 2 && entry.length > 0.0) m_profiles.push_back(entry); current.reset(); } } } if (false == m_profiles.empty()) WriteLog("Editor: " + std::to_string(m_profiles.size()) + " vertical profile design(s) read from the scenery", logtype::generic); } void editor_mode::profile_library_mark(std::vector Layers) { std::sort(Layers.begin(), Layers.end()); Layers.erase(std::unique(Layers.begin(), Layers.end()), Layers.end()); for (auto const layer : Layers) { if (false == scene::Layers.valid(layer)) continue; auto const &context{scene::Layers.layer(layer).context_insert()}; auto const local = [&](glm::dvec3 const &Point) { auto const point{context.to_local(Point)}; return format("%.4f %.4f %.4f", point.x, point.y, point.z); }; std::vector lines; for (auto const &entry : m_profiles) { if (entry.layer != layer) continue; if (lines.empty()) lines.emplace_back("# vertical profiles (grade lines) designed in the track editor, the simulation skips these lines"); lines.push_back(format("profile %.4f ", entry.length) + local(entry.start) + ' ' + local(entry.end) + ' ' + local(entry.middle)); lines.push_back(format("speed %.1f origin %.4f join %d %d", entry.line.speed, entry.origin, entry.line.join_start ? 1 : 0, entry.line.join_end ? 1 : 0)); auto const &norms{entry.line.norms}; lines.push_back(format("norms %.4f %.4f %.4f %.1f %.4f %.2f %.4f", norms.grade_max, norms.grade_switch_max, norms.radius_factor, norms.radius_min, norms.curve_threshold, norms.element_min, norms.fixed_tolerance)); if (entry.departure >= 0.0) lines.push_back(format("departure %.4f", entry.departure)); for (auto const &point : entry.line.points) lines.push_back(format("point %.4f %.4f %.1f %d", point.chainage, context.to_local({0.0, point.elevation, 0.0}).y, point.radius, point.automatic ? 1 : 0)); lines.emplace_back("end"); } scene::Layers.mark(layer, kProfileMark, std::move(lines)); } } int editor_mode::profile_find(bool &Reversed) const { auto const &state{m_profile}; Reversed = false; if (state.samples.empty()) return -1; auto const start{state.samples.front().position}; auto const end{state.samples.back().position}; auto const middle{editor_track::sampled_position(state.samples, state.route.length * 0.5)}; for (std::size_t i = 0; i < m_profiles.size(); ++i) { auto const &entry{m_profiles[i]}; if (std::abs(entry.length - state.route.length) > 0.5 || plan_distance(entry.middle, middle) > 0.5) continue; if (plan_distance(entry.start, start) <= 0.1 && plan_distance(entry.end, end) <= 0.1) return static_cast(i); if (plan_distance(entry.start, end) <= 0.1 && plan_distance(entry.end, start) <= 0.1) { Reversed = true; return static_cast(i); } } return -1; } bool editor_mode::profile_restore() { profile_library_load(); auto &state{m_profile}; bool reversed; auto const index{profile_find(reversed)}; if (index < 0) return profile_restore_parts(); auto const &entry{m_profiles[index]}; state.line = reversed ? reversed_line(entry.line, entry.length) : entry.line; state.origin = entry.origin; state.selected = -1; state.changed = false; profile_check(); if (entry.departure >= 0.0 && entry.departure <= kApplied && state.departure > kStale) return profile_restore_parts(); state.status = format(STR_C("Design read from the scenery file %s, largest departure from the track %.3f m"), scene::Layers.layer(entry.layer).name.c_str(), state.departure); return true; } // the designs kept for routes lying along this one, as the lines the tracks were fitted with one by one: their points carried // over to the chainage of this route, straight between them where none lies, as over a switch none of them runs through bool editor_mode::profile_restore_parts() { auto &state{m_profile}; auto const &samples{state.samples}; if (samples.size() < 2) return false; auto const locate = [&](glm::dvec3 const &Point, double &Chainage) { return sampled_chainage(samples, Point, Chainage); }; auto const length{state.route.length}; // the direction of the route at the chainage, in the plan auto const heading = [&](double const Chainage) { auto const a{editor_track::sampled_position(samples, std::max(0.0, Chainage - 1.0))}; auto const b{editor_track::sampled_position(samples, std::min(length, Chainage + 1.0))}; return glm::dvec2{b.x - a.x, b.z - a.z}; }; std::vector points; std::size_t parts{0}; std::size_t retaken{0}; // the ends of this route inside a kept design take their heights from it std::optional start, end; // the newer designs first: where two lie over the same stretch, the one kept later is the one meant std::vector> covered; auto const taken = [&](double const Chainage) { return std::any_of(covered.begin(), covered.end(), [&](auto const &Stretch) { return Chainage > Stretch.first + 0.25 && Chainage < Stretch.second - 0.25; }); }; for (auto entries{m_profiles.rbegin()}; entries != m_profiles.rend(); ++entries) { auto const &entry{*entries}; // the design laid over this route at least in part: one of its ends on it, the route running on towards the other double a{0.0}, b{0.0}; auto const first{locate(entry.start, a)}; auto const last{locate(entry.end, b)}; if (false == first && false == last) continue; double origin, direction; double scale{1.0}; if (first && last) { // the route between the ends of the design got longer or shorter, as when a track of it was moved: the points go along scale = std::abs(b - a) / std::max(1.0, entry.length); if (scale < 0.8 || scale > 1.25) continue; direction = b >= a ? 1.0 : -1.0; origin = a; } else { auto const anchor{first ? a : b}; auto const towards{first ? entry.end - entry.start : entry.start - entry.end}; auto const along{glm::dot(heading(anchor), glm::dvec2{towards.x, towards.z})}; if (std::abs(along) < 1e-9) continue; // the chainage of the design grows along the route or against it direction = (along > 0.0) == first ? 1.0 : -1.0; origin = first ? a : b - direction * entry.length; } auto const local = [&](double const Chainage) { return (Chainage - origin) * direction / scale; }; // the stretch of this route over the paths of the design, from where it meets it: off it, as past the switch where this // route turns away, the design says nothing auto const extent_low{std::min(origin, origin + direction * entry.length * scale)}; auto const extent_high{std::max(origin, origin + direction * entry.length * scale)}; auto low{extent_low}; auto high{extent_high}; if (auto const own{profile_stored_route(entry)}) { std::set> paths; for (auto const &span : own->spans) paths.emplace(span.track, span.path); auto const &spans{state.route.spans}; auto const on = [&](std::size_t const Index) { return paths.count({spans[Index].track, spans[Index].path}) > 0; }; auto const anchor{first ? a : b}; std::optional at; for (std::size_t i = 0; i < spans.size(); ++i) if (on(i) && anchor >= spans[i].from - 0.3 && anchor <= spans[i].to + 0.3) { at = i; break; } if (false == at.has_value()) continue; auto left{*at}; auto right{*at}; while (left > 0 && on(left - 1)) --left; while (right + 1 < spans.size() && on(right + 1)) ++right; low = std::max(low, spans[left].from); high = std::min(high, spans[right].to); } auto const inside = [&](double const Chainage) { return Chainage > 0.25 && Chainage < length - 0.25 && Chainage >= low - 0.25 && Chainage <= high + 0.25; }; // where the track was changed since the design was applied to it, by the gizmo or by dragging its points, the track holds: // the design gives way there to the line read from the track std::vector> stale; if (entry.departure >= 0.0 && entry.departure <= kApplied) { for (auto const &sample : samples) { if (sample.chainage < low - 0.25 || sample.chainage > high + 0.25) continue; auto const designed{profile::elevation(entry.line, std::clamp(local(sample.chainage), 0.0, entry.length))}; if (std::abs(designed - sample.position.y) <= kStale) continue; if (false == stale.empty() && sample.chainage - stale.back().second <= kStaleMargin) stale.back().second = sample.chainage; else stale.emplace_back(sample.chainage, sample.chainage); } for (auto &stretch : stale) { stretch.first = std::max(low, stretch.first - kStaleMargin); stretch.second = std::min(high, stretch.second + kStaleMargin); } } auto const changed = [&](double const Chainage) { return std::any_of(stale.begin(), stale.end(), [&](auto const &Stretch) { return Chainage >= Stretch.first - 0.25 && Chainage <= Stretch.second + 0.25; }); }; for (auto point : entry.line.points) { point.chainage = origin + direction * point.chainage * scale; point.joint = false; if (inside(point.chainage) && false == taken(point.chainage) && false == changed(point.chainage)) points.push_back(point); } for (auto const &stretch : stale) { std::vector chainages, elevations, grades; for (auto const &sample : samples) if (sample.chainage >= stretch.first - 1e-6 && sample.chainage <= stretch.second + 1e-6) { chainages.push_back(sample.chainage); elevations.push_back(sample.position.y); grades.push_back(sample.grade); } if (chainages.size() < 2) continue; profile::recognition options; options.minimum_length = 5.0; options.detail = kStale; for (auto point : profile::recognize(chainages, elevations, grades, options).points) { point.joint = false; if (point.chainage >= 0.0 && point.chainage <= length && false == taken(point.chainage)) points.push_back(point); } ++retaken; } // where this route turns off the design, its height there: between two designs the line goes straight, over the switches. // where it runs on past the end of the design, onto track laid since, the break gets the vertical curve of the limits for (auto const edge : {low, high}) if (edge > 0.5 && edge < length - 0.5 && false == taken(edge) && false == changed(edge)) { profile::pvi point; point.chainage = edge; point.elevation = profile::elevation(entry.line, std::clamp(local(edge), 0.0, entry.length)); point.automatic = std::abs(edge - (edge == low ? extent_low : extent_high)) < 0.5; points.push_back(point); } if (false == start.has_value() && false == changed(0.0) && local(0.0) >= -0.25 && local(0.0) <= entry.length + 0.25) start = profile::elevation(entry.line, std::clamp(local(0.0), 0.0, entry.length)); if (false == end.has_value() && false == changed(length) && local(length) >= -0.25 && local(length) <= entry.length + 0.25) end = profile::elevation(entry.line, std::clamp(local(length), 0.0, entry.length)); covered.emplace_back(low, high); ++parts; } if (parts == 0) return false; std::sort(points.begin(), points.end(), [](auto const &Left, auto const &Right) { return Left.chainage < Right.chainage; }); std::vector line; for (auto point : points) { point.chainage = std::clamp(point.chainage, 0.0, length); if (false == line.empty() && point.chainage - line.back().chainage < 0.5) continue; line.push_back(point); } if (line.empty() || line.front().chainage > 0.5) line.insert(line.begin(), profile::pvi{0.0, start.value_or(samples.front().position.y)}); else line.front().chainage = 0.0; if (line.back().chainage < length - 0.5) line.push_back(profile::pvi{length, end.value_or(samples.back().position.y)}); else line.back().chainage = length; line.front().radius = 0.0; line.back().radius = 0.0; state.line.points = std::move(line); state.selected = -1; state.changed = false; profile_check(); state.status = format(STR_C("Design put together from %zu designs kept for the routes along this one, largest departure from the track %.3f m"), parts, state.departure); if (retaken > 0) state.status += format(STR_C("; %zu stretches changed on the track since read from it again"), retaken); return true; } void editor_mode::profile_store() { auto &state{m_profile}; if (state.route.spans.empty() || state.samples.empty() || state.line.points.size() < 2) return; if (scene::Layers.empty()) { state.error = STR_C("The scenery wasn't opened for editing, the design can't be kept in its files"); return; } profile_library_load(); stored_profile entry; entry.length = state.route.length; entry.start = state.samples.front().position; entry.end = state.samples.back().position; entry.middle = editor_track::sampled_position(state.samples, state.route.length * 0.5); entry.origin = state.origin; entry.line = state.line; entry.departure = state.departure; // the joints are kept as points of their own: read back over a longer route, where the end is no longer joined, the line // keeps the curve it was laid with for (auto &point : entry.line.points) if (point.joint) { point.joint = false; point.automatic = false; } // the designs of shorter routes this one runs all along are taken over by it: kept, they'd be put together with it again std::vector layers; for (auto i{m_profiles.size()}; i-- > 0;) { auto const &old{m_profiles[i]}; double a{0.0}, b{0.0}; if (old.length + 0.5 < entry.length && sampled_chainage(state.samples, old.start, a) && sampled_chainage(state.samples, old.end, b) && std::abs(std::abs(b - a) - old.length) <= 1.5) { layers.push_back(old.layer); m_profiles.erase(m_profiles.begin() + static_cast(i)); } } bool reversed; auto const index{profile_find(reversed)}; if (index >= 0) { entry.layer = m_profiles[index].layer; m_profiles[index] = entry; } else { auto layer{scene::Layers.resolve(state.route.spans.front().track->layer())}; if (false == scene::Layers.valid(layer) || false == scene::Layers.accepts(layer)) layer = scene::Layers.active(); if (false == scene::Layers.valid(layer)) layer = 1; entry.layer = layer; m_profiles.push_back(entry); } layers.push_back(entry.layer); profile_library_mark(layers); state.changed = false; } void editor_mode::profile_forget() { auto &state{m_profile}; profile_library_load(); bool reversed; auto const index{profile_find(reversed)}; if (index < 0) return; auto const layer{m_profiles[index].layer}; m_profiles.erase(m_profiles.begin() + index); profile_library_mark({layer}); state.status = format(STR_C("The design will be removed from %s on save"), scene::Layers.layer(layer).name.c_str()); } std::optional editor_mode::profile_stored_route(stored_profile const &Entry, TTrack **From, TTrack **To) const { auto const touching = [](glm::dvec3 const &Point) { std::vector result; for (auto *track : simulation::Paths.sequence()) { if (track == nullptr || track->m_editorremoved || false == editor_track::is_supported(*track)) continue; for (auto const &path : track->m_paths) for (auto const index : {segment_data::point::start, segment_data::point::end}) if (plan_distance(path.points[index], Point) <= 0.1 && std::abs(path.points[index].y - Point.y) < 3.0) { if (std::find(result.begin(), result.end(), track) == result.end()) result.push_back(track); } } return result; }; for (auto *from : touching(Entry.start)) for (auto *to : touching(Entry.end)) { editor_track::route route; std::string error; if (false == editor_track::find_route(from, to, route, error) || std::abs(route.length - Entry.length) > 0.5) continue; auto const samples{editor_track::sample_route(route, 1.0)}; if (samples.empty() || plan_distance(samples.front().position, Entry.start) > 0.1 || plan_distance(editor_track::sampled_position(samples, route.length * 0.5), Entry.middle) > 0.5) continue; if (From != nullptr) *From = from; if (To != nullptr) *To = to; return route; } return std::nullopt; } void editor_mode::profile_open_stored(std::size_t const Index) { auto &state{m_profile}; if (Index >= m_profiles.size()) return; auto const entry{m_profiles[Index]}; TTrack *from{nullptr}; TTrack *to{nullptr}; if (profile_stored_route(entry, &from, &to).has_value()) { profile_open(from, to); return; } state.open = true; state.error = STR_C("The route of the stored design isn't in the scenery any more (paths moved or removed)"); } void editor_mode::profile_edited() { m_profile.changed = true; profile_check(); first_step_done(first_step::profile_point); } void editor_mode::profile_open(TTrack *From, TTrack *To) { auto &state{m_profile}; state.open = true; state.status.clear(); editor_track::route route; if (false == editor_track::find_route(From, To, route, state.error)) return; profile_take(std::move(route), From, To); } void editor_mode::profile_open_run(TTrack &Track) { auto &state{m_profile}; state.open = true; state.status.clear(); state.error.clear(); auto route{editor_track::run_route(Track, state.run_length)}; if (route.spans.empty()) { state.error = STR_C("The path can't carry a profile"); return; } auto *from{route.spans.front().track}; auto *to{route.spans.back().track}; profile_take(std::move(route), from, to); } // the route from one end of the selected tracks to the other: of the two of them farthest apart, through the switches between void editor_mode::profile_open_selection(std::vector const &Tracks) { auto const middle = [](TTrack const &Track) { return geometry::bezier{Track.m_paths.front()}.point(0.5); }; TTrack *from{nullptr}; TTrack *to{nullptr}; double farthest{-1.0}; for (std::size_t i = 0; i < Tracks.size(); ++i) for (std::size_t j = i + 1; j < Tracks.size(); ++j) if (auto const distance{plan_distance(middle(*Tracks[i]), middle(*Tracks[j]))}; distance > farthest) { farthest = distance; from = Tracks[i]; to = Tracks[j]; } if (from == nullptr || to == nullptr) return; profile_open(from, to); auto &state{m_profile}; if (state.route.spans.empty()) return; auto const off{std::count_if(Tracks.begin(), Tracks.end(), [&](TTrack const *Track) { return std::none_of(state.route.spans.begin(), state.route.spans.end(), [&](editor_track::route_span const &Span) { return Span.track == Track; }); })}; if (off > 0) state.status = format(STR_C("%d of the selected tracks are off the route from %s to %s"), static_cast(off), from->name().c_str(), to->name().c_str()); } // the route follows the selection: one track gives its whole line unless it's on the route already, several give the route // from one end of them to the other; the changes of the design left are kept for the save void editor_mode::profile_follow_selection() { auto &state{m_profile}; auto const tracks{selected_tracks()}; if (tracks == state.followed) return; state.followed = tracks; if (tracks.empty()) return; auto const on_route{std::any_of(state.route.spans.begin(), state.route.spans.end(), [&](editor_track::route_span const &Span) { return Span.track == tracks.front(); })}; if (tracks.size() == 1 && on_route) return; if (state.changed) profile_store(); if (tracks.size() == 1) profile_open_run(*tracks.front()); else profile_open_selection(tracks); } void editor_mode::profile_take(editor_track::route Route, TTrack *From, TTrack *To) { auto &state{m_profile}; state.picked_from = From; state.from = state.picked_from; state.picked_to = To; state.to = state.picked_to; state.route = std::move(Route); state.selected = -1; state.edits = m_track_edits; first_step_done(first_step::profile); profile_resample(); if (false == profile_restore()) { profile_recognize(); if (false == state.samples.empty()) { auto const km{alignment_store::chainage_at(state.samples.front().position)}; auto const further{alignment_store::chainage_at(editor_track::sampled_position(state.samples, std::min(state.route.length, 5.0)))}; if (km.has_value() && further.has_value() && *further > *km) state.origin = *km; } } profile_fit_view(); // the camera stays where the route was picked, it follows the plot once that's moved state.view_seen = state.view_framed = {state.view_from, state.view_to}; } void editor_mode::profile_resample() { auto &state{m_profile}; state.samples = editor_track::sample_route(state.route, 1.0); auto const &samples{state.samples}; state.terrain.assign(samples.size(), std::numeric_limits::quiet_NaN()); std::vector points; points.reserve(samples.size()); for (auto const &sample : samples) points.push_back(sample.position); std::vector found; auto const heights{ground_heights(points, true, &found)}; for (std::size_t i = 0; i < samples.size(); ++i) if (found[i] != 0) state.terrain[i] = heights[i]; profile_sample_model(); state.beds.assign(samples.size(), 0.0); std::vector bedless(samples.size(), 0); std::vector own; for (std::size_t i = 0; i < samples.size(); ++i) if (samples[i].span < state.route.spans.size()) { auto const &track{*state.route.spans[samples[i].span].track}; state.beds[i] = profile_ballast(track).first; bedless[i] = profile_own_bed(track) ? 0 : 1; if (bedless[i] == 0) own.push_back(state.beds[i]); } // a track with no trackbed of its own, as the switches of the older sceneries whose bed is an include, lies in the bed of // the others if (false == own.empty()) { std::nth_element(own.begin(), own.begin() + own.size() / 2, own.end()); for (std::size_t i = 0; i < samples.size(); ++i) if (bedless[i] != 0) state.beds[i] = own[own.size() / 2]; } auto &context{state.context}; context = {}; context.length = state.route.length; if (samples.empty()) return; std::optional open; for (auto const &span : state.route.spans) { if (auto const found{switch_store::named(*span.track)}) { if (open.has_value() && open->name == found->name && span.from - context.switches.back().to <= std::max(found->a + found->b, context.switches.back().to - context.switches.back().from)) { context.switches.back().to = span.to; continue; } auto const label{switch_store::label(*found)}; context.switches.push_back({span.from, span.to, found->number != "-" ? found->number + ' ' + label : label}); open = found; } else if (span.track->eType != tt_Normal) { context.switches.push_back({span.from, span.to, span.track->name()}); open.reset(); } } // where the rails of the route lie on timbers shared with another track, at a switch, a crossover, a crossing, the grade // doesn't break nor does a vertical curve reach: the timbers carry both tracks in one plane { auto const before{context.switches.size()}; for (auto const &span : state.route.spans) { auto const length{span.track->eType == tt_Switch ? span.track->SwitchExtension->Segments[span.path]->GetLength() : span.track->Segment->GetLength()}; if (length < 1e-6) continue; auto const scale{(span.to - span.from) / length}; for (auto const &zone : track_sleepers::shared_zones(*span.track, span.path)) { auto const a{span.forward ? span.from + zone.first * scale : span.to - zone.second * scale}; auto const b{span.forward ? span.from + zone.second * scale : span.to - zone.first * scale}; // the stretches of a switch on the route are its own zone already auto const inside{std::any_of(context.switches.begin(), context.switches.begin() + static_cast(before), [&](auto const &Zone) { return a >= Zone.from - 0.5 && b <= Zone.to + 0.5; })}; if (false == inside && b - a > 0.1) context.switches.push_back({a, b, STR("shared timbers")}); } } std::sort(context.switches.begin(), context.switches.end(), [](auto const &Left, auto const &Right) { return Left.from < Right.from; }); } context.start_elevation = samples.front().position.y; context.end_elevation = samples.back().position.y; TTrack *adjoining{nullptr}; int path{0}; bool entered{false}; // a joined end on a switch inherits its height and grade from the switch, which lies in one plane auto const inherits = [&](bool const Joined, double const Elevation, double const Grade) { if (false == Joined || adjoining == nullptr || adjoining->eType != tt_Switch) return std::string{}; auto const found{switch_store::named(*adjoining)}; auto const point{entered ? "P1" : path == 0 ? "P2" : "P4"}; return format(STR_C("Grade line from the switch %s (%s): %.2f m, %+.2f per mille"), (found ? switch_store::label(*found) : adjoining->name()).c_str(), point, Elevation, Grade * 1000.0); }; context.start_joined = editor_track::adjoining_grade(state.route, false, context.start_grade, &adjoining, &path, &entered); state.start_inherits = inherits(context.start_joined, context.start_elevation, context.start_grade); adjoining = nullptr; context.end_joined = editor_track::adjoining_grade(state.route, true, context.end_grade, &adjoining, &path, &entered); state.end_inherits = inherits(context.end_joined, context.end_elevation, context.end_grade); profile_check(); } void editor_mode::profile_sample_model() { auto &state{m_profile}; auto const &samples{state.samples}; state.model.assign(samples.size(), std::numeric_limits::quiet_NaN()); state.model_waiting = false; state.model_polled = ImGui::GetTime(); if (samples.empty() || false == state.show_model) return; nmt_locate(); if (false == m_nmt.placed()) return; // the squares along the route, not all those under the rectangle around it: a long line crosses the grid std::size_t constexpr run{200}; for (std::size_t first = 0; first < samples.size(); first += run) { glm::dvec2 low{samples[first].position.x, samples[first].position.z}; auto high{low}; for (auto i = first; i < std::min(samples.size(), first + run + 1); ++i) { low = glm::min(low, glm::dvec2{samples[i].position.x, samples[i].position.z}); high = glm::max(high, glm::dvec2{samples[i].position.x, samples[i].position.z}); } if (false == m_nmt.request(low - glm::dvec2{2.0}, high + glm::dvec2{2.0})) state.model_waiting = true; } std::vector differences; for (std::size_t i = 0; i < samples.size(); ++i) if (double height; m_nmt.height(samples[i].position.x, samples[i].position.z, height)) { state.model[i] = height; differences.push_back(samples[i].position.y - height); } // a scenery put onto the model lies at its heights, or as far from them as the model was found to lie from it; one built at // heights of its own, never put onto it, is compared by the shape: the model goes as far as the route lies from it auto offset{m_nmt_state.offset}; if (false == m_nmt_state.fitted && false == differences.empty()) { std::nth_element(differences.begin(), differences.begin() + differences.size() / 2, differences.end()); if (auto const median{differences[differences.size() / 2]}; std::abs(median) > 30.0) offset = median; } for (auto &height : state.model) height += offset; } void editor_mode::profile_recognize() { auto &state{m_profile}; std::vector chainages, elevations, grades; for (auto const &sample : state.samples) { chainages.push_back(sample.chainage); elevations.push_back(sample.position.y); grades.push_back(sample.grade); } auto const norms{state.line.norms}; state.line = profile::recognize(chainages, elevations, grades, {}); state.line.norms = norms; state.line.speed = route_speed(state.route); state.selected = -1; profile_check(); state.status = format(STR_P("Grade line recognized from the track: %zu point, largest departure %.3f m", "Grade line recognized from the track: %zu points, largest departure %.3f m", state.line.points.size()), state.line.points.size(), state.departure); } void editor_mode::profile_fit_ground() { auto &state{m_profile}; std::vector chainages; for (auto const &sample : state.samples) chainages.push_back(sample.chainage); if (std::none_of(state.terrain.begin(), state.terrain.end(), [](double const Height) { return std::isfinite(Height); })) { state.error = STR_C("There's no ground under the route"); return; } auto settings{state.line}; settings.speed = route_speed(state.route); // the grade line is the rail top: it's fitted so the formation under it, not the rails, follows the ground auto ground{state.terrain}; for (std::size_t i = 0; i < ground.size() && i < state.beds.size(); ++i) ground[i] += state.beds[i]; auto line{profile::fit_ground(chainages, ground, state.context, settings, state.fit)}; if (line.points.size() < 2) return; state.line = std::move(line); state.selected = -1; state.error.clear(); profile_check(); double fill{0.0}; double cut{0.0}; for (std::size_t i = 0; i < state.samples.size(); ++i) { if (false == std::isfinite(state.terrain[i])) continue; auto const difference{profile::elevation(state.line, state.samples[i].chainage) - (i < state.beds.size() ? state.beds[i] : 0.0) - state.terrain[i]}; fill = std::max(fill, difference); cut = std::max(cut, -difference); } state.status = format(STR_P("Grade line fitted to the ground: %zu point, embankment up to %.2f m, cutting up to %.2f m", "Grade line fitted to the ground: %zu points, embankment up to %.2f m, cutting up to %.2f m", state.line.points.size()), state.line.points.size(), fill, cut); } std::pair editor_mode::profile_ballast(TTrack const &Track) { // the ballast is laid from the foot of the rails down, the rail top being the grade line auto const depth{std::max(0.0, static_cast(Track.fTexHeight1 - Track.fTexHeightOffset)) + Track.rail_height()}; auto const foot{0.5 * std::abs(Track.fTrackWidth) + std::abs(Track.fTexWidth) + std::abs(Track.fTexSlope)}; return {depth, std::max(0.5, foot)}; } bool editor_mode::profile_own_bed(TTrack const &Track) { return Track.fTexHeight1 - Track.fTexHeightOffset > kNoBed; } std::pair editor_mode::profile_generate_ground(std::vector> const &Beds, double const Reach) { auto &state{m_profile}; auto &works{state.earthworks}; auto const &samples{state.samples}; if (samples.empty() || false == ensure_terrain()) return {0, 0}; auto const size{heightmap::chunk_size}; std::set keys; for (auto const &sample : samples) { auto const &p{sample.position}; auto const low{heightmap::chunk_at(p.x - Reach, p.z - Reach)}; auto const high{heightmap::chunk_at(p.x + Reach, p.z + Reach)}; for (auto x = low.first; x <= high.first; ++x) for (auto z = low.second; z <= high.second; ++z) keys.insert({x, z}); } auto const cells{heightmap::cells_for(m_terrain_spacing)}; auto const cellsize{size / cells}; auto const side{static_cast(cells) + 1}; int made{0}; int skipped{0}; for (auto const &key : keys) { if (m_streamer.exists(key)) continue; auto const low{heightmap::chunk_corner(key)}; glm::dvec2 const middle{low + glm::dvec2{size * 0.5}}; std::vector around; for (std::size_t i = 0; i < samples.size(); ++i) if (std::abs(samples[i].position.x - middle.x) <= size * 0.5 + Reach && std::abs(samples[i].position.z - middle.y) <= size * 0.5 + Reach) around.push_back(i); if (around.empty()) continue; // where the scenery has its own ground, the chunk isn't made: the two would cover each other std::vector probes; for (int iz = 1; iz < 10; ++iz) for (int ix = 1; ix < 10; ++ix) probes.emplace_back(low.x + ix * size / 10.0, profile::elevation(state.line, samples[around.front()].chainage), low.y + iz * size / 10.0); std::vector found; ground_heights(probes, true, &found); if (std::count(found.begin(), found.end(), 1) * 20 > static_cast(probes.size())) { ++skipped; continue; } // the ground of the chunk: the ground under the nearest point of the route, or the formation there std::vector heights(side * side); for (std::size_t iz = 0; iz < side; ++iz) for (std::size_t ix = 0; ix < side; ++ix) { glm::dvec3 const point{low.x + ix * cellsize, 0.0, low.y + iz * cellsize}; auto best{around.front()}; auto distance{std::numeric_limits::max()}; for (auto const i : around) { auto const d{plan_distance(point, samples[i].position)}; if (d < distance) { distance = d; best = i; } } heights[iz * side + ix] = static_cast(best < state.terrain.size() && std::isfinite(state.terrain[best]) ? state.terrain[best] : profile::elevation(state.line, samples[best].chainage) - Beds[best].first); } auto const added{m_streamer.add_chunk(key.first, key.second, m_terrain_spacing, [&](double const X, double const Z) { auto const ix{static_cast(std::clamp(std::lround((X - low.x) / cellsize), 0, cells))}; auto const iz{static_cast(std::clamp(std::lround((Z - low.y) / cellsize), 0, cells))}; return heights[iz * side + ix]; })}; if (false == added) continue; works.generated.push_back(key); ++made; } return {made, skipped}; } void editor_mode::profile_shape_ground() { auto &state{m_profile}; auto &works{state.earthworks}; auto const &samples{state.samples}; if (samples.size() < 2 || state.line.points.size() < 2) return; if (false == ensure_terrain()) { state.error = STR_C("The heightmap terrain can't be opened"); return; } auto const slope{std::max(0.1, works.slope)}; std::vector> beds(samples.size(), {works.depth, works.half_width}); if (works.from_ballast) for (std::size_t i = 0; i < samples.size(); ++i) if (samples[i].span < state.route.spans.size()) if (auto const *track{state.route.spans[samples[i].span].track}; track != nullptr) beds[i] = profile_ballast(*track); double widest{0.0}; for (auto const &bed : beds) widest = std::max(widest, bed.second); auto const reach{widest + works.reach}; works.generated.clear(); auto const generated{works.generate ? profile_generate_ground(beds, reach) : std::pair{0, 0}}; // the shaping is a modifier of the terrain: the formation along the axis of the route, followed by the ground // around it. shaped again, the route replaces its modifier, and the terrain follows the new grade line heightmap::modifier modifier; auto const name = [](TTrack const *Track) { return Track != nullptr && false == Track->name().empty() ? Track->name() : std::string{"?"}; }; modifier.name = "route " + name(state.from) + " - " + name(state.to); modifier.slope = slope; modifier.reach = works.reach; modifier.rounding = std::max(0.0, works.rounding); std::vector points; points.reserve(samples.size()); for (std::size_t i = 0; i < samples.size(); ++i) points.push_back({{samples[i].position.x, samples[i].position.z}, profile::elevation(state.line, samples[i].chainage) - beds[i].first, beds[i].second}); // the points the line between their neighbours passes within a few millimetres of are left out double constexpr tolerance{0.005}; modifier.points.push_back(points.front()); std::size_t anchor{0}; for (std::size_t next = 2; next < points.size(); ++next) { auto const &a{points[anchor]}; auto const &b{points[next]}; auto const run{b.position - a.position}; auto const length{glm::length(run)}; bool straight{true}; for (auto i = anchor + 1; i < next && straight; ++i) { auto const t{length > 1e-9 ? glm::dot(points[i].position - a.position, run) / (length * length) : 0.0}; auto const offset{glm::length(points[i].position - (a.position + run * t))}; auto const formation{a.formation + (b.formation - a.formation) * t}; auto const width{a.half_width + (b.half_width - a.half_width) * t}; straight = offset <= tolerance && std::abs(points[i].formation - formation) <= tolerance && std::abs(points[i].half_width - width) <= tolerance; } if (false == straight) { anchor = next - 1; modifier.points.push_back(points[anchor]); } } modifier.points.push_back(points.back()); double fill{0.0}; double cut{0.0}; for (std::size_t i = 0; i < samples.size(); ++i) { if (i >= state.terrain.size() || false == std::isfinite(state.terrain[i])) continue; fill = std::max(fill, points[i].formation - state.terrain[i]); cut = std::max(cut, state.terrain[i] - points[i].formation); } if (false == works.modifier.empty() && works.modifier != modifier.name) m_streamer.remove_modifier(works.modifier); works.modifier = modifier.name; auto const count{modifier.points.size()}; m_streamer.modifier(std::move(modifier)); forget_ground(); auto status{format(STR_P("Terrain shaped by the modifier \"%s\" (%zu point): embankment up to %.2f m, cutting up to %.2f m", "Terrain shaped by the modifier \"%s\" (%zu points): embankment up to %.2f m, cutting up to %.2f m", count), works.modifier.c_str(), count, fill, cut)}; if (generated.first > 0) status += format(STR_P("; %d chunk of terrain made under the route", "; %d chunks of terrain made under the route", generated.first), generated.first); if (generated.second > 0) status += format(STR_P("; %d chunk left out: the scenery has its own ground there, convert it to shape it", "; %d chunks left out: the scenery has its own ground there, convert it to shape it", generated.second), generated.second); state.error.clear(); profile_resample(); state.status = status; WriteLog("Editor: vertical profile - " + state.status, logtype::generic); } void editor_mode::profile_restore_ground() { auto &state{m_profile}; auto &works{state.earthworks}; std::size_t restored{0}; if (false == works.modifier.empty() && m_streamer.remove_modifier(works.modifier)) ++restored; works.modifier.clear(); for (auto const &key : works.generated) { m_streamer.remove_chunk(key.first, key.second); ++restored; } works.generated.clear(); forget_ground(); profile_resample(); state.status = restored > 0 ? STR_C("The terrain is back as it was before the shaping") : STR_C("The shaping of the terrain is gone already"); } void editor_mode::render_ground_conversion() { auto &conversion{m_ground_conversion}; ImGui::PushID("ground_conversion"); if (false == conversion.open) { if (ImGui::Button(STR_C("Convert the terrain files to heightmap terrain..."))) { conversion.open = true; load_ground_files(); } if (ImGui::IsItemHovered()) ImGui::SetTooltip("%s", STR_C("Replaces the triangles of the terrain files of the scenery (terrain directive, .txtf/.btf) with heightmap terrain\n" "of the same shape, painted with their materials; the files lose these triangles on save")); if (false == conversion.status.empty()) ImGui::TextWrapped("%s", conversion.status.c_str()); ImGui::PopID(); return; } if (conversion.files.empty()) ImGui::TextWrapped("%s", STR_C("The scenery has no terrain files: no terrain directive names a .txtf or .btf file")); else ImGui::TextUnformatted(STR_C("Terrain files of the scenery:")); for (std::size_t index = 0; index < conversion.files.size(); ++index) { auto &source{conversion.files[index]}; ImGui::PushID(static_cast(index)); ImGui::BeginDisabled(false == source.good); ImGui::Checkbox(source.file.c_str(), &source.chosen); ImGui::EndDisabled(); ImGui::SameLine(); if (source.good) ImGui::TextDisabled(source.text ? STR_C("(text, %zu triangles)") : STR_C("(binary, %zu triangles)"), source.count); else ImGui::TextDisabled("(%s)", source.message.c_str()); ImGui::PopID(); } // the materials of the chosen files std::vector counts(conversion.names.size(), 0); for (auto const &source : conversion.files) if (source.good && source.chosen) for (auto index = source.first; index < source.first + source.count; ++index) ++counts[conversion.materials[index]]; if (false == conversion.names.empty()) ImGui::TextUnformatted(STR_C("Materials: tick the ones which are the ground")); for (std::size_t index = 0; index < conversion.names.size(); ++index) { if (counts[index] == 0) continue; bool chosen{conversion.chosen.count(index) != 0}; ImGui::PushID(static_cast(index) + 0x10000); if (ImGui::Checkbox(format(STR_C("%s (%zu triangles)"), conversion.names[index].c_str(), counts[index]).c_str(), &chosen)) { if (chosen) conversion.chosen.insert(index); else conversion.chosen.erase(index); } ImGui::PopID(); } ImGui::TextWrapped("%s", STR_C("The chunks are painted with the materials of the triangles, up to 8 of them per chunk, added to the palette of the terrain. " "Until saved, loading the scenery again takes it all back.")); auto const ready{std::any_of(conversion.files.begin(), conversion.files.end(), [](auto const &Source) { return Source.good && Source.chosen; }) && false == conversion.chosen.empty()}; ImGui::BeginDisabled(false == ready); if (ImGui::Button(STR_C("Convert"))) { conversion.status = convert_ground(); conversion.open = false; profile_resample(); } ImGui::EndDisabled(); ImGui::SameLine(); if (ImGui::Button(STR_C("Cancel"))) conversion.open = false; if (false == conversion.open) { // the triangles take memory, they're read again the next time conversion.triangles = {}; conversion.materials = {}; } ImGui::PopID(); } void editor_mode::render_profile_earthworks() { auto &state{m_profile}; auto &works{state.earthworks}; if (false == ImGui::TreeNode(STR_C("Shape the terrain to the grade line"))) return; ImGui::TextWrapped("%s", STR_C("Only the heightmap terrain changes: a formation and slopes are made, the models of the scenery stay as they are")); render_ground_conversion(); auto const drag = [](char const *Label, double &Value, float const Speed, char const *Format) { field_row(Label); ImGui::PushID(Label); auto const edited{ImGui::DragScalar("##value", ImGuiDataType_Double, &Value, Speed, nullptr, nullptr, Format)}; ImGui::PopID(); return edited; }; toggle_chips("earthworks", {{STR_C("Make terrain where the route has none"), &works.generate, "Chunks of heightmap terrain are made along the route where there's none, from the ground around,\nthen shaped. Where the scenery has its own ground triangles the chunk is left out: convert them instead"}, {STR_C("Formation under the ballast of the track"), &works.from_ballast, "The terrain goes as deep under the track as its ballast reaches and as wide as its foot,\nread from each path of the route; off: the values given below"}}); if (works.from_ballast) { if (false == state.route.spans.empty() && state.route.spans.front().track != nullptr) { auto const bed{profile_ballast(*state.route.spans.front().track)}; ImGui::TextDisabled(STR_C("Ballast %.2f m deep, its foot %.2f m from the axis"), bed.first, bed.second); } } if (false == works.from_ballast && begin_fields("##earthworkfields", 0.55f)) { drag("Formation below the track (m)", works.depth, 0.01f, "%.2f"); drag("Half width of the formation (m)", works.half_width, 0.05f, "%.2f"); if (ImGui::IsItemHovered()) ImGui::SetTooltip("%s", STR_C("From the axis of the route to the edge of the formation;\nfor two tracks add the distance between them")); ImGui::EndTable(); } ImGui::PushID("earthworkmore"); if (ImGui::TreeNode(STR_C("More settings"))) { if (begin_fields("##earthworkslopes", 0.55f)) { drag("Slopes 1 : n, n", works.slope, 0.05f, "%.2f"); drag("Slopes reach at most (m)", works.reach, 1.0f, "%.0f"); drag("Rounding of the edges (m)", works.rounding, 0.1f, "%.1f"); if (ImGui::IsItemHovered()) ImGui::SetTooltip("%s", STR_C("Length over which the slope bends into the formation and into the ground; 0: sharp edges")); ImGui::EndTable(); } ImGui::TreePop(); } ImGui::PopID(); works.rounding = std::clamp(works.rounding, 0.0, 20.0); works.depth = std::clamp(works.depth, 0.0, 5.0); works.half_width = std::clamp(works.half_width, 0.5, 30.0); works.slope = std::clamp(works.slope, 0.1, 10.0); works.reach = std::clamp(works.reach, 1.0, 200.0); if (has_errors(state.issues)) ImGui::TextDisabled("%s", STR_C("Correct the errors of the grade line first")); else { ImGui::PushStyleColor(ImGuiCol_Button, ImVec4(0.18f, 0.42f, 0.24f, 1.0f)); ImGui::PushStyleColor(ImGuiCol_ButtonHovered, ImVec4(0.22f, 0.52f, 0.30f, 1.0f)); if (ImGui::Button(STR_C("Shape the terrain"), ImVec2(-FLT_MIN, ImGui::GetFrameHeight() * 1.3f))) profile_shape_ground(); item_tooltip("Makes the formation and the slopes along the route in the heightmap terrain.\nThey're kept as a modifier of the terrain, which follows the grade line when you shape it again; the triangles and models of the scenery stay as they are"); ImGui::PopStyleColor(2); } if (false == works.modifier.empty() || false == works.generated.empty()) { if (ImGui::Button(STR_C("Restore the terrain"), ImVec2(-FLT_MIN, 0.0f))) profile_restore_ground(); if (ImGui::IsItemHovered()) ImGui::SetTooltip("%s", STR_C("Removes the modifier of the route, and takes away the terrain the shaping made")); } if (m_streamer.active()) { // the terrain tools take the tool options window over, the profile stays open under them if (ImGui::Button(STR_C("Sculpting..."))) terrain_workspace(true); if (ImGui::IsItemHovered()) ImGui::SetTooltip("%s", STR_C("Opens the terrain tools: raising, lowering and smoothing with a brush")); } ImGui::TreePop(); } void editor_mode::profile_check() { auto &state{m_profile}; auto &points{state.line.points}; // joint points come and go, the selection follows the point it was made on struct mark { int *index; double chainage; bool joint; }; std::vector marks; for (auto *index : {&state.selected, &state.dragging, &state.curve_grip}) if (*index >= 0 && *index < static_cast(points.size())) marks.push_back({index, points[*index].chainage, points[*index].joint}); profile::complete(state.line, state.context); for (auto const &mark : marks) { int best{-1}; double distance{std::numeric_limits::max()}; for (int i = 0; i < static_cast(points.size()); ++i) if (points[i].joint == mark.joint && std::abs(points[i].chainage - mark.chainage) < distance) { distance = std::abs(points[i].chainage - mark.chainage); best = i; } *mark.index = (mark.joint || distance < 1e-6) ? best : -1; } state.issues = profile::check(state.line, state.context); state.departure = 0.0; for (auto const &sample : state.samples) state.departure = std::max(state.departure, std::abs(profile::elevation(state.line, sample.chainage) - sample.position.y)); } void editor_mode::profile_fit_view() { auto &state{m_profile}; auto const length{std::max(1.0, state.route.length)}; state.view_from = -0.02 * length; state.view_to = length * 1.02; state.exaggeration = 0.0f; } void editor_mode::profile_follow_view() { auto &state{m_profile}; std::pair const shown{state.view_from, state.view_to}; auto const now{ImGui::GetTime()}; if (shown != state.view_seen) { state.view_seen = shown; state.view_seen_time = now; return; } if (false == state.camera_follows || shown == state.view_framed || now - state.view_seen_time < 0.25) return; state.view_framed = shown; profile_frame_view(); } void editor_mode::profile_frame_view() { auto const &state{m_profile}; std::vector points; for (auto const &sample : state.samples) if (sample.chainage >= state.view_from && sample.chainage <= state.view_to) points.push_back(sample.position); auto const &io{ImGui::GetIO()}; if (points.size() < 2 || io.DisplaySize.y <= 0.0f) return; glm::dvec3 low{points.front()}; glm::dvec3 high{points.front()}; for (auto const &point : points) { low = glm::min(low, point); high = glm::max(high, point); } auto const centre{(low + high) * 0.5}; // the room the docks and the strip of the plot leave to the view, and how far its middle lies from the middle of the screen, // in pixels glm::dvec2 const screen{io.DisplaySize.x, io.DisplaySize.y}; auto const viewlow{editor_ui::view_min()}; auto viewhigh{editor_ui::view_max()}; if (state.strip_top > viewlow.y + 100.0f) viewhigh.y = std::min(viewhigh.y, state.strip_top); auto const width{std::max(50.0, static_cast(viewhigh.x - viewlow.x))}; auto const height{std::max(50.0, static_cast(viewhigh.y - viewlow.y))}; glm::dvec2 const shift{(viewlow.x + viewhigh.x) * 0.5 - screen.x * 0.5, (viewlow.y + viewhigh.y) * 0.5 - screen.y * 0.5}; double constexpr margin{1.1}; if (Global.EditorOrtho) { auto const yaw{static_cast(Camera.Angle.y)}; double extent{5.0}; for (auto const &point : points) { auto const local{RotateY(point - centre, -yaw)}; extent = std::max({extent, std::abs(local.x) * margin * screen.y / width, std::abs(local.z) * margin * screen.y / height}); } Global.EditorOrthoExtent = std::clamp(static_cast(extent), 5.0f, 50000.0f); auto const scale{2.0 * static_cast(Global.EditorOrthoExtent) / screen.y}; auto const offset{RotateY(glm::dvec3{shift.x * scale, 0.0, shift.y * scale}, yaw)}; Camera.Pos.x = centre.x - offset.x; Camera.Pos.z = centre.z - offset.z; Camera.Velocity = {}; m_focus_active = false; Global.pCamera = Camera; return; } // seen from the side, as the plot shows it: the start on the left, from a little above glm::dvec3 chord{points.back().x - points.front().x, 0.0, points.back().z - points.front().z}; chord = glm::length(chord) > 1e-6 ? glm::normalize(chord) : glm::dvec3{1.0, 0.0, 0.0}; glm::dvec3 const up{0.0, 1.0, 0.0}; auto const look{glm::normalize(glm::cross(up, chord) + glm::dvec3{0.0, -0.4, 0.0})}; auto const right{glm::normalize(glm::cross(look, up))}; auto const above{glm::cross(right, look)}; auto const tangent{std::tan(glm::radians(static_cast(Global.FieldOfView) / std::max(0.01, static_cast(Global.ZoomFactor))) * 0.5)}; auto const across{tangent * width / screen.y}; auto const upward{tangent * height / screen.y}; double distance{10.0}; for (auto const &point : points) { auto const offset{point - centre}; auto const depth{glm::dot(offset, look)}; distance = std::max({distance, std::abs(glm::dot(offset, right)) * margin / across - depth, std::abs(glm::dot(offset, above)) * margin / upward - depth}); } distance = std::min(distance, 8000.0); auto const pixel{2.0 * distance * tangent / screen.y}; fly_to(centre - right * (shift.x * pixel) + above * (shift.y * pixel), look, distance, 0.35); } void editor_mode::profile_apply(bool const Terrain) { auto &state{m_profile}; state.status.clear(); state.error.clear(); if (state.route.spans.empty() || state.line.points.size() < 2) return; auto const edits{m_track_edits}; if (has_errors(state.issues)) { state.error = STR_C("Correct the errors to apply the grade line"); return; } std::vector tracks; for (auto const &span : state.route.spans) tracks.push_back(span.track); if (false == tracks_editable(tracks, state.error)) { state.error = STR_C("The grade line can't be applied, ") + state.error; return; } auto const line{state.line}; auto const before{state.samples}; std::vector> states; std::vector created; auto const gaps{editor_track::apply_profile( state.route, [&](double const Chainage) { return profile::elevation(line, Chainage); }, [&](double const Chainage) { return profile::grade(line, Chainage); }, profile::breaks(line), states, created)}; auto const changed{states.size()}; auto const added{created.size()}; state.created = created; // the stored design changes with the paths, an undo takes both back profile_library_load(); auto library{m_profiles}; bool const pushed{false == states.empty() || false == created.empty()}; push_track_snapshot(std::move(states), std::move(created)); state.edits = m_track_edits; first_step_done(first_step::profile_apply); if (pushed) { m_history.back().profiles = true; m_history.back().profile_library = std::move(library); } state.from = state.route.spans.front().track; state.to = state.route.spans.back().track; profile_resample(); profile_store(); state.status = format(STR_P("Grade line applied: %zu path changed, %zu added, largest departure %.3f m", "Grade line applied: %zu paths changed, %zu added, largest departure %.3f m", changed), changed, added, state.departure); if (false == scene::Layers.empty()) state.status += STR_C("\nThe paths and the design (//$p lines) go to the scenery files on save"); auto const name = [](TTrack const *Track) { return Track->name().empty() ? std::string{STR_C("(no name)")} : Track->name(); }; auto const closed{std::count_if(gaps.begin(), gaps.end(), [](auto const &Gap) { return Gap.closed; })}; if (closed > 0) state.status += STR_C("\nPaths off the route brought to its height:"); for (auto const &gap : gaps) if (gap.closed) state.status += format(STR_C("\n %s at %s: %+.3f m"), name(gap.neighbour).c_str(), name(gap.track).c_str(), -gap.gap); if (closed < static_cast(gaps.size())) state.status += STR_C("\nSwitch ends off the adjoining paths, which can't be changed here:"); for (auto const &gap : gaps) if (false == gap.closed) state.status += format("\n %s -> %s: %+.3f m", name(gap.track).c_str(), name(gap.neighbour).c_str(), gap.gap); // the tracks on the other side of the crossovers of the route go up and down with it auto const followed{follow_crossovers(before)}; if (followed > 0) state.status += format(STR_C("\nThe tracks across %zu crossover(s) followed the change"), followed); if (Terrain) profile_follow_terrain(static_cast(m_track_edits - edits)); WriteLog("Editor: vertical profile - " + state.status, logtype::generic); } void editor_mode::profile_follow_terrain(std::size_t const Edits) { auto &state{m_profile}; // the heightmap terrain the scenery has; the ground of a scenery without it stays as it is if (false == m_streamer.active() || Edits == 0) return; std::vector tracks; std::size_t found{0}; for (auto step{m_history.rbegin()}; step != m_history.rend() && found < Edits; ++step) { if (step->action != EditorSnapshot::Action::TrackEdit) continue; ++found; for (auto const &entry : step->tracks) tracks.push_back(entry.first); tracks.insert(tracks.end(), step->created.begin(), step->created.end()); tracks.insert(tracks.end(), step->removed.begin(), step->removed.end()); } std::erase(tracks, nullptr); std::sort(tracks.begin(), tracks.end()); tracks.erase(std::unique(tracks.begin(), tracks.end()), tracks.end()); if (tracks.empty()) return; // the route shaped by the earthworks of the profile keeps its modifier, shaped again to the new grade line; the tracks // shaped one by one, as the terrain model lays them, get their formations anew auto const name = [](TTrack const *Track) { return Track != nullptr && false == Track->name().empty() ? Track->name() : std::string{"?"}; }; auto const route{"route " + name(state.from) + " - " + name(state.to)}; auto const &modifiers{m_streamer.modifiers()}; auto const has = [&](std::string const &Name) { return std::any_of(modifiers.begin(), modifiers.end(), [&](heightmap::modifier const &Modifier) { return Modifier.name == Name; }); }; auto const shaped{has(route) || (false == state.earthworks.modifier.empty() && has(state.earthworks.modifier))}; auto const own{std::any_of(tracks.begin(), tracks.end(), [&](TTrack const *Track) { return false == Track->name().empty() && std::any_of(modifiers.begin(), modifiers.end(), [&](heightmap::modifier const &Modifier) { return Modifier.name == "nmt " + Track->name() || Modifier.name.starts_with("nmt " + Track->name() + " "); }); })}; auto status{state.status}; std::size_t laid{0}; if (false == shaped || own) laid = nmt_shape_tracks(&tracks); if (shaped) { profile_shape_ground(); status += "\n" + state.status; } else profile_resample(); if (laid > 0) status += format(STR_P("\nThe terrain followed the tracks: %zu formation laid anew", "\nThe terrain followed the tracks: %zu formations laid anew", laid), laid); state.status = status; // the terrain goes back with the grade line in one step auto const before{m_history.empty() ? EditorSnapshot::Action::Other : m_history.back().action}; terrain_history_step(); if (before != EditorSnapshot::Action::Terrain && false == m_history.empty() && m_history.back().action == EditorSnapshot::Action::Terrain) m_history.back().joined = true; } void editor_mode::profile_after_undo() { auto &state{m_profile}; if (state.route.spans.empty()) return; auto const removed = [](TTrack const *Track) { return Track == nullptr || Track->m_editorremoved; }; if (removed(state.from) || removed(state.to)) { state.from = state.picked_from; state.to = state.picked_to; } std::string error; if (removed(state.from) || removed(state.to) || false == editor_track::find_route(state.from, state.to, state.route, error)) { state.route = {}; state.samples.clear(); state.selected = -1; state.dragging = -1; state.curve_grip = -1; return; } profile_resample(); } // the tracks of the route changed by another tool or by an undo: the route read again, the design kept in the scenery read // again over it unless it's being changed void editor_mode::profile_follow_edits() { auto &state{m_profile}; if (state.edits == m_track_edits) return; state.edits = m_track_edits; if (state.route.spans.empty() || state.dragging >= 0) return; auto const kept{false == state.changed}; profile_after_undo(); if (kept && false == state.route.spans.empty()) profile_restore(); } void editor_mode::render_profile_body() { auto &state{m_profile}; profile_follow_selection(); step_heading(1, STR_C("Route")); render_profile_source(); if (state.route.spans.empty()) { if (false == state.error.empty()) ImGui::TextColored(ImVec4(1.0f, 0.4f, 0.3f, 1.0f), "%s", state.error.c_str()); return; } ImGui::TextWrapped(STR_P("%zu path, %.2f m: %s ... %s", "%zu paths, %.2f m: %s ... %s", state.route.spans.size()), state.route.spans.size(), state.route.length, state.route.spans.front().track->name().c_str(), state.route.spans.back().track->name().c_str()); { bool reversed; auto const stored{profile_find(reversed)}; if (stored >= 0) { ImGui::TextDisabled(STR_C("Design kept in %s%s"), scene::Layers.layer(m_profiles[stored].layer).name.c_str(), state.changed ? STR_C(", changes go there on save") : ""); ImGui::SameLine(); if (ImGui::SmallButton(STR_C("Forget the design"))) profile_forget(); } else if (state.changed) ImGui::TextDisabled(STR_C("The design goes to the scenery file of the route on save")); } step_heading(2, STR_C("Grade line")); if (render_profile_parameters()) profile_edited(); step_heading(3, STR_C("Terrain")); render_profile_earthworks(); } void editor_mode::render_profile_strip() { profile_follow_edits(); auto &state{m_profile}; auto const &io{ImGui::GetIO()}; // along the bottom of the 3d view, as wide as it; each time the profile opens, wherever it was left before auto const low{editor_ui::view_min()}; auto const high{editor_ui::view_max()}; auto const width{std::max(480.0f, high.x - low.x)}; auto const height{std::clamp((high.y - low.y) * 0.5f, 260.0f, std::max(260.0f, high.y - low.y - 80.0f))}; ImGui::SetNextWindowPos(ImVec2(low.x, high.y - height), ImGuiCond_Appearing); ImGui::SetNextWindowSize(ImVec2(width, height), ImGuiCond_Appearing); auto title{std::string{STR_C("Vertical profile")}}; if (false == state.route.spans.empty()) title += format(STR_P(" %s > %s, %zu path, %.2f m", " %s > %s, %zu paths, %.2f m", state.route.spans.size()), state.route.spans.front().track->name().c_str(), state.route.spans.back().track->name().c_str(), state.route.spans.size(), state.route.length); title += "###profilestrip"; if (ImGui::Begin(title.c_str(), nullptr, ImGuiWindowFlags_NoCollapse | ImGuiWindowFlags_NoScrollbar | ImGuiWindowFlags_NoScrollWithMouse)) { // a window left past the edge of the screen comes back into it auto const position{ImGui::GetWindowPos()}; auto const size{ImGui::GetWindowSize()}; state.strip_top = ImGui::IsWindowDocked() ? -1.0f : position.y; if (false == ImGui::IsWindowDocked() && (position.y + size.y > io.DisplaySize.y + 1.0f || position.x + 80.0f > io.DisplaySize.x || position.x + size.x < 80.0f)) ImGui::SetWindowPos(ImVec2(std::clamp(position.x, 0.0f, std::max(0.0f, io.DisplaySize.x - size.x)), std::max(0.0f, std::min(position.y, io.DisplaySize.y - size.y)))); if (state.route.spans.empty()) render_profile_empty(); else { render_profile_header(); auto const side{std::clamp(ImGui::GetContentRegionAvail().x * 0.3f, 250.0f, 400.0f)}; ImGui::BeginChild("##profileplot", ImVec2(-side, 0.0f), false, ImGuiWindowFlags_NoScrollbar | ImGuiWindowFlags_NoScrollWithMouse); render_profile_toolbar(); render_profile_canvas(ImGui::GetContentRegionAvail().y); ImGui::EndChild(); ImGui::SameLine(); ImGui::BeginChild("##profileside", ImVec2(0.0f, 0.0f), false); if (ImGui::BeginTabBar("##profiletabs")) { // a click on a point in the plot brings its tab forward, a click on the state of the grade line the remarks bool const picked{state.selected >= 0 && state.selected != state.shown}; state.shown = state.selected; if (ImGui::BeginTabItem(STR_C("Point"), nullptr, picked ? ImGuiTabItemFlags_SetSelected : 0)) { render_profile_point(); ImGui::EndTabItem(); } if (ImGui::BeginTabItem(format("%s (%zu)###profilepoints", STR_C("Points"), state.line.points.size()).c_str())) { ImGui::BeginChild("##profilepointlist"); render_profile_points(); ImGui::EndChild(); ImGui::EndTabItem(); } bool const errors{has_errors(state.issues)}; if (false == state.issues.empty()) ImGui::PushStyleColor(ImGuiCol_Text, errors ? ImVec4(1.0f, 0.45f, 0.4f, 1.0f) : ImVec4(1.0f, 0.8f, 0.3f, 1.0f)); bool const remarks{ImGui::BeginTabItem(format("%s (%zu)###profileremarks", STR_C("Remarks"), state.issues.size()).c_str(), nullptr, state.show_remarks ? ImGuiTabItemFlags_SetSelected : 0)}; state.show_remarks = false; if (false == state.issues.empty()) ImGui::PopStyleColor(); if (remarks) { render_profile_issues(); ImGui::EndTabItem(); } ImGui::EndTabBar(); } ImGui::EndChild(); } } ImGui::End(); } void editor_mode::render_profile_header() { auto &state{m_profile}; auto const errors{std::count_if(state.issues.begin(), state.issues.end(), [](profile::issue const &Issue) { return Issue.error; })}; auto const warnings{static_cast(state.issues.size()) - errors}; if (ImGui::Button(STR_C("From the track"))) { profile_recognize(); state.changed = true; } item_tooltip("The points of the grade line read from the track as it lies now"); ImGui::SameLine(); if (ImGui::Button(STR_C("To the ground"))) { profile_fit_ground(); state.changed = true; } item_tooltip("Places the breaks of the grade line along the ground under the route,\nwithin the limits and with the least earthworks; the ground itself stays as it is"); ImGui::SameLine(0.0f, 2.0f); ImGui::SetNextItemWidth(ImGui::CalcTextSize("+00.00 m").x + ImGui::GetStyle().FramePadding.x * 2.0f); ImGui::DragScalar("##fittolerance", ImGuiDataType_Double, &state.fit.tolerance, 0.05f, nullptr, nullptr, STR_C("+-%.2f m")); state.fit.tolerance = std::clamp(state.fit.tolerance, 0.05, 50.0); item_tooltip("How far the grade line may depart from the ground before it breaks;\nmore gives fewer breaks and higher embankments and deeper cuttings"); ImGui::SameLine(0.0f, ImGui::GetFontSize()); // the state of the grade line against the limits; a click lists the remarks std::string chip; ImVec4 colour; if (errors > 0) { chip = warnings > 0 ? format(STR_C("%d errors, %d warnings"), static_cast(errors), static_cast(warnings)) : format(STR_C("%d errors"), static_cast(errors)); colour = ImVec4(1.0f, 0.45f, 0.38f, 1.0f); } else if (warnings > 0) { chip = format(STR_C("%d warnings"), static_cast(warnings)); colour = ImVec4(1.0f, 0.8f, 0.3f, 1.0f); } else { chip = STR_C("Within the limits"); colour = ImVec4(0.45f, 0.9f, 0.45f, 1.0f); } if (state_chip((chip + "###profilestate").c_str(), colour)) state.show_remarks = true; item_tooltip(state.issues.empty() ? "The grade line meets the limits of the line" : "Click: the list of the remarks; a click on a remark shows its place"); // apply at the right end, as far as there's room auto const label{std::string{STR_C("Apply to the scenery")} + (state.departure > 0.0005 ? " *" : "")}; auto const button{ImGui::CalcTextSize(label.c_str()).x + ImGui::GetStyle().FramePadding.x * 2.0f}; auto const right{ImGui::GetWindowContentRegionMax().x - button}; ImGui::SameLine(0.0f, ImGui::GetFontSize()); if (ImGui::GetCursorPosX() < right) ImGui::SetCursorPosX(right); ImGui::BeginDisabled(errors > 0); ImGui::PushStyleColor(ImGuiCol_Button, ImVec4(0.2f, 0.55f, 0.25f, 1.0f)); ImGui::PushStyleColor(ImGuiCol_ButtonHovered, ImVec4(0.25f, 0.68f, 0.3f, 1.0f)); if (ImGui::Button((label + "###profileapply").c_str())) profile_apply(true); ImGui::PopStyleColor(2); ImGui::EndDisabled(); if (ImGui::IsItemHovered(ImGuiHoveredFlags_AllowWhenDisabled)) { if (errors > 0) ImGui::SetTooltip("%s", STR_C("Correct the errors first: the red remarks, a click on the state shows them")); else if (state.departure > 0.0005) ImGui::SetTooltip(STR_C("Sets the heights of the track along the route to the grade line, the track moves by up to %.3f m; Ctrl+Z takes it back"), state.departure); else ImGui::SetTooltip("%s", STR_C("The track lies on the grade line already")); } if (false == state.error.empty()) ImGui::TextColored(ImVec4(1.0f, 0.4f, 0.3f, 1.0f), "%s", state.error.c_str()); } void editor_mode::render_profile_empty() { auto &state{m_profile}; auto *selected{selected_track()}; auto const available{ImGui::GetContentRegionAvail()}; auto *draw{ImGui::GetWindowDrawList()}; auto const line{ImGui::GetTextLineHeight()}; auto const mouse{std::round(line * 2.6f)}; auto const headline{std::string{STR_C("Click a track in the view")}}; auto const detail{std::string{STR_C("The grade line of its whole line shows here; Shift+LMB on more tracks or Shift+drag a box: the route from one end of them to the other. Drag the points of the line up and down, then apply it to the scenery.")}}; auto const wrap{std::min(available.x - 40.0f, 560.0f)}; auto const top{ImGui::GetCursorScreenPos().y + std::max(0.0f, available.y * 0.5f - mouse - line * 2.0f)}; auto const left{ImGui::GetCursorScreenPos().x + std::max(0.0f, (available.x - wrap) * 0.5f)}; gesture_mouse(*draw, ImVec2(left, top), mouse, "LMB"); auto *font{ui_layer::font_bold != nullptr ? ui_layer::font_bold : ImGui::GetFont()}; auto const big{ui_layer::font_bold != nullptr ? ui_layer::font_bold->FontSize : ImGui::GetFontSize() * 1.2f}; auto const textleft{left + mouse * 0.68f + line}; draw->AddText(font, big, ImVec2(textleft, top + line * 0.2f), ImGui::GetColorU32(ImGuiCol_Text), headline.c_str()); draw->AddText(ImGui::GetFont(), ImGui::GetFontSize(), ImVec2(textleft, top + big + line * 0.6f), ImGui::GetColorU32(ImGuiCol_TextDisabled), detail.c_str(), nullptr, wrap - (textleft - left)); auto const detailsize{ImGui::CalcTextSize(detail.c_str(), nullptr, false, wrap - (textleft - left))}; ImGui::SetCursorScreenPos(ImVec2(textleft, std::max(top + mouse, top + big + line * 0.6f + detailsize.y) + line)); if (selected != nullptr) { if (ImGui::Button(format(STR_C("Along the line of %s"), selected->name().c_str()).c_str())) profile_open_run(*selected); ImGui::SameLine(); } profile_library_load(); if (false == m_profiles.empty() && ImGui::Button(format(STR_C("Designs kept in the scenery (%zu)..."), m_profiles.size()).c_str())) ImGui::OpenPopup("##profilekept"); if (ImGui::BeginPopup("##profilekept")) { for (std::size_t i = 0; i < m_profiles.size(); ++i) { auto const &entry{m_profiles[i]}; ImGui::PushID(static_cast(i)); if (ImGui::Selectable(format(STR_P("%s %.1f m, %zu point", "%s %.1f m, %zu points", entry.line.points.size()), scene::Layers.layer(entry.layer).name.c_str(), entry.length, entry.line.points.size()).c_str())) profile_open_stored(i); ImGui::PopID(); } ImGui::EndPopup(); } if (false == state.error.empty()) { ImGui::SetCursorScreenPos(ImVec2(textleft, ImGui::GetCursorScreenPos().y)); ImGui::TextColored(ImVec4(1.0f, 0.4f, 0.3f, 1.0f), "%s", state.error.c_str()); } } void editor_mode::render_profile_source() { gesture_rows({{"LMB on a track", "its whole line"}, {"Shift+LMB on more tracks", "the route between the outermost ones"}, {"Shift+drag a box", "the route over the tracks in it"}}); profile_library_load(); if (false == m_profiles.empty() && ImGui::TreeNode("##profilestored", STR_C("Designs kept in the scenery (%zu)"), m_profiles.size())) { for (std::size_t i = 0; i < m_profiles.size(); ++i) { auto const &entry{m_profiles[i]}; ImGui::PushID(static_cast(i)); if (ImGui::SmallButton(STR_C("Open"))) profile_open_stored(i); ImGui::SameLine(); ImGui::Text(STR_P("%s %.1f m, %zu point, from (%.1f, %.1f) to (%.1f, %.1f)", "%s %.1f m, %zu points, from (%.1f, %.1f) to (%.1f, %.1f)", entry.line.points.size()), scene::Layers.layer(entry.layer).name.c_str(), entry.length, entry.line.points.size(), entry.start.x, entry.start.z, entry.end.x, entry.end.z); ImGui::PopID(); } ImGui::TreePop(); } } bool editor_mode::render_profile_parameters() { auto &state{m_profile}; auto &line{state.line}; bool changed{false}; auto const drag = [](char const *Label, double &Value, float const Speed, char const *Format) { field_row(Label); ImGui::PushID(Label); auto const edited{ImGui::DragScalar("##value", ImGuiDataType_Double, &Value, Speed, nullptr, nullptr, Format)}; ImGui::PopID(); return edited; }; if (begin_fields("##profileline", 0.5f)) { drag("Chainage of the start (m)", state.origin, 1.0f, "%.2f"); changed |= drag("Design speed (km/h)", line.speed, 1.0f, "%.0f"); ImGui::EndTable(); } line.speed = std::max(1.0, line.speed); auto const join_tooltip = [] { if (ImGui::IsItemHovered()) ImGui::SetTooltip("%s", STR_C("Keeps the elevation of the joint and takes over the grade of the adjoining track,\nwith a vertical curve which starts right at the joint")); }; if (state.context.start_joined) { changed |= ImGui::Checkbox(STR_C("Join the adjoining track at the start"), &line.join_start); join_tooltip(); } if (state.context.end_joined) { changed |= ImGui::Checkbox(STR_C("Join the adjoining track at the end"), &line.join_end); join_tooltip(); } if (ImGui::TreeNode(STR_C("Norms"))) { auto &norms{line.norms}; auto const table{begin_fields("##profilelimits", 0.6f)}; changed |= table && drag("Maximum grade (per mille)", norms.grade_max, 0.1f, "%.2f"); if (table) { changed |= drag("Maximum grade over switches (per mille)", norms.grade_switch_max, 0.1f, norms.grade_switch_max > 0.0 ? "%.2f" : STR_C("as the line")); if (ImGui::IsItemHovered()) ImGui::SetTooltip("%s", STR_C("0: the limit of the line, a switch may lie on its grade.\nA lower one for the tracks of a station where the vehicles stand, e.g. 2.5")); changed |= drag("Vertical curve R >= k*V^2, k", norms.radius_factor, 0.01f, "%.2f"); item_tooltip("The required radius of a vertical curve grows with the square of the design speed: R = k * V^2, V in km/h"); changed |= drag("Vertical curve R min (m)", norms.radius_min, 10.0f, "%.0f"); item_tooltip("A vertical curve is never smaller than this, even at a low speed"); changed |= drag("Vertical curve needed above a change of (per mille)", norms.curve_threshold, 0.1f, "%.2f"); item_tooltip("A smaller change of grade may stay a sharp break, without a curve"); changed |= drag("Constant grade between curves at least (m)", norms.element_min, 1.0f, "%.0f"); item_tooltip("The shortest stretch of one grade between two vertical curves; 0: not checked"); changed |= drag("Tolerance at fixed points (m)", norms.fixed_tolerance, 0.001f, "%.3f"); item_tooltip("How far the grade line may pass from a fixed point, e.g. the joint with the adjoining track, before it's reported"); ImGui::EndTable(); } norms.grade_switch_max = std::max(0.0, norms.grade_switch_max); if (ImGui::SmallButton(STR_C("Automatic R at all points"))) { for (auto &point : line.points) point.automatic = true; changed = true; } ImGui::TextDisabled(STR_C("Required R for %.0f km/h: %.0f m"), line.speed, profile::required_radius(line)); ImGui::TreePop(); } auto const errors{std::count_if(state.issues.begin(), state.issues.end(), [](profile::issue const &Issue) { return Issue.error; })}; auto const warnings{static_cast(state.issues.size()) - errors}; ImGui::PushTextWrapPos(0.0f); auto const good{IM_COL32(100, 230, 100, 255)}; ImGui::BeginGroup(); metric_tiles({{format("%d", static_cast(errors)), STR_C("errors"), errors > 0 ? IM_COL32(255, 100, 80, 255) : good}, {format("%d", static_cast(warnings)), STR_C("warnings"), warnings > 0 ? IM_COL32(255, 200, 80, 255) : good}, {format("%.3f m", state.departure), STR_C("off the track"), ImGui::GetColorU32(ImGuiCol_Text)}}); ImGui::EndGroup(); item_tooltip("From the track, To the ground and Apply to the scenery are over the plot of the strip"); if (errors + warnings > 0) ImGui::TextDisabled("%s", STR_C("The list of them: Remarks in the strip under the view")); else ImGui::TextColored(ImVec4(0.4f, 0.9f, 0.4f, 1.0f), "%s", STR_C("The grade line meets the limits")); ImGui::PopTextWrapPos(); return changed; } void editor_mode::render_profile_issues() { auto &state{m_profile}; ImGui::BeginChild("##profileissues", ImVec2(0.0f, 0.0f), false); if (state.issues.empty()) ImGui::TextColored(ImVec4(0.4f, 0.9f, 0.4f, 1.0f), "%s", STR_C("The grade line meets the limits")); else ImGui::TextDisabled("%s", STR_C("LMB on a remark shows its place")); for (std::size_t i = 0; i < state.issues.size(); ++i) { auto const &issue{state.issues[i]}; ImGui::PushID(static_cast(i)); ImGui::PushStyleColor(ImGuiCol_Text, issue.error ? ImVec4(1.0f, 0.4f, 0.3f, 1.0f) : ImVec4(1.0f, 0.8f, 0.3f, 1.0f)); auto const text{format("%s km %s %s", issue.error ? "x" : "!", profile::format_chainage(issue.chainage + state.origin).c_str(), issue.text.c_str())}; if (ImGui::Selectable(text.c_str())) { if (issue.point >= 0) state.selected = issue.point; auto const half{(state.view_to - state.view_from) * 0.5}; state.view_from = issue.chainage - half; state.view_to = issue.chainage + half; } ImGui::PopStyleColor(); ImGui::PopID(); } if (false == state.status.empty()) ImGui::TextWrapped("%s", state.status.c_str()); if (false == state.error.empty()) ImGui::TextColored(ImVec4(1.0f, 0.4f, 0.3f, 1.0f), "%s", state.error.c_str()); ImGui::EndChild(); } void editor_mode::render_profile_point() { auto &state{m_profile}; auto &points{state.line.points}; bool changed{false}; auto const count{static_cast(points.size())}; if (state.selected >= count) state.selected = -1; auto const drag = [](char const *Label, double &Value, float const Speed, char const *Format) { field_row(Label); ImGui::PushID(Label); auto const edited{ImGui::DragScalar("##value", ImGuiDataType_Double, &Value, Speed, nullptr, nullptr, Format)}; ImGui::PopID(); return edited; }; auto const kind_of = [](double const Change) { return Change > 0.0 ? STR_C("sag") : STR_C("crest"); }; if (state.selected < 0) { ImGui::TextWrapped("%s", STR_C("No point selected. Click a point of the grade line in the plot, or double click the line to add one.")); } else if (points[state.selected].joint) { auto const index{state.selected}; auto const &point{points[index]}; auto const change{(profile::grade_after(state.line, index) - profile::grade_after(state.line, index - 1)) * 1000.0}; ImGui::Text(STR_C("Point %d: joint with the adjoining track"), index + 1); ImGui::Text(STR_C("km %s H %.3f m R %.0f m"), profile::format_chainage(point.chainage + state.origin).c_str(), point.elevation, point.radius); ImGui::Text(STR_C("Grade %+.2f -> %+.2f per mille (%s)"), profile::grade_after(state.line, index - 1) * 1000.0, profile::grade_after(state.line, index) * 1000.0, kind_of(change)); ImGui::TextWrapped("%s", STR_C("Placed automatically, moves with the neighbouring point; untick the joining in the tool options to edit the end freely")); } else { auto const index{state.selected}; auto &point{points[index]}; bool const end{index == 0 || index == count - 1}; bool const locked{end_locked(state.line, state.context, index)}; ImGui::Text(STR_C("Point %d"), index + 1); ImGui::SameLine(); ImGui::TextDisabled(STR_C("km %s%s"), profile::format_chainage(point.chainage + state.origin).c_str(), locked ? STR_C(", joined to the adjoining track") : end ? STR_C(", end of the route") : ""); if (begin_fields("##profilepoint", 0.45f)) { auto chainage{point.chainage + state.origin}; if (false == end && drag("Chainage (m)", chainage, 0.5f, "%.2f")) { point.chainage = std::clamp(chainage - state.origin, points[index - 1].chainage + 0.5, points[index + 1].chainage - 0.5); changed = true; } if (locked) { field_row("Elevation (m)"); ImGui::Text("%.3f", point.elevation); } else changed |= drag("Elevation (m)", point.elevation, 0.01f, "%.3f"); if (index > 0 && false == locked) { auto grade{profile::grade_after(state.line, index - 1) * 1000.0}; if (drag("Grade before (per mille)", grade, 0.05f, "%+.2f")) { point.elevation = points[index - 1].elevation + grade / 1000.0 * (point.chainage - points[index - 1].chainage); changed = true; } } if (index + 1 < count) { auto grade{profile::grade_after(state.line, index) * 1000.0}; if (drag("Grade after (per mille)", grade, 0.05f, "%+.2f")) { points[index + 1].elevation = point.elevation + grade / 1000.0 * (points[index + 1].chainage - point.chainage); changed = true; } } if (false == end) { field_row("Vertical curve R (m)"); auto const box{ImGui::GetFrameHeight() + ImGui::GetStyle().ItemInnerSpacing.x + ImGui::CalcTextSize(STR_C("auto")).x}; ImGui::SetNextItemWidth(std::max(40.0f, ImGui::GetContentRegionAvail().x - box - ImGui::GetStyle().ItemSpacing.x)); if (ImGui::DragScalar("##pointradius", ImGuiDataType_Double, &point.radius, 10.0f, nullptr, nullptr, "%.0f")) { point.automatic = false; changed = true; } if (ImGui::IsItemHovered()) { auto const change{(profile::grade_after(state.line, index) - profile::grade_after(state.line, index - 1)) * 1000.0}; auto const tangent{point.radius * std::abs(change) / 2000.0}; auto text{std::string{STR_C("The radius of the vertical curve which rounds off the break of the grade here")} + "\n" + format(STR_C("Change of grade %+.2f per mille, %s"), change, kind_of(change))}; if (point.radius > 0.0) text += "\n" + format(STR_C("Tangent T %.2f m, sagitta f %.3f m"), tangent, tangent * tangent / (2.0 * point.radius)); ImGui::SetTooltip("%s", text.c_str()); } point.radius = std::max(0.0, point.radius); ImGui::SameLine(); changed |= ImGui::Checkbox(STR_C("auto"), &point.automatic); if (ImGui::IsItemHovered()) ImGui::SetTooltip("%s", STR_C("R follows the limits: the required radius where the change of grade needs a curve; 0: no curve")); } ImGui::EndTable(); } if (index + 1 < count && ImGui::Button(STR_C("Add a point after"))) { profile::pvi inserted; inserted.chainage = (point.chainage + points[index + 1].chainage) * 0.5; inserted.elevation = (point.elevation + points[index + 1].elevation) * 0.5; points.insert(points.begin() + index + 1, inserted); state.selected = index + 1; changed = true; } if (false == end) { ImGui::SameLine(); if (ImGui::Button(STR_C("Remove Del"))) { points.erase(points.begin() + index); state.selected = -1; changed = true; } } } if (changed) profile_edited(); } void editor_mode::render_profile_points() { auto &state{m_profile}; auto const &points{state.line.points}; ImGui::Columns(4, "##profilepointcolumns", false); for (auto const *header : {"#", "km", "H (m)", "R (m)"}) { ImGui::TextDisabled("%s", header); ImGui::NextColumn(); } ImGui::Separator(); for (int i = 0; i < static_cast(points.size()); ++i) { auto const &point{points[i]}; if (ImGui::Selectable(format("%d##point%d", i + 1, i).c_str(), state.selected == i, ImGuiSelectableFlags_SpanAllColumns)) { state.selected = i; auto const half{(state.view_to - state.view_from) * 0.5}; if (point.chainage < state.view_from || point.chainage > state.view_to) { state.view_from = point.chainage - half; state.view_to = point.chainage + half; } } ImGui::NextColumn(); ImGui::TextUnformatted(profile::format_chainage(point.chainage + state.origin).c_str()); ImGui::NextColumn(); ImGui::Text("%.3f", point.elevation); ImGui::NextColumn(); if (point.radius > 0.0) ImGui::Text("%.0f%s", point.radius, point.joint ? STR_C(" joint") : point.automatic ? STR_C(" auto") : ""); else ImGui::TextDisabled("-"); ImGui::NextColumn(); } ImGui::Columns(1); } // screen mapping of the profile canvas, chainage across and elevation upwards struct editor_mode::profile_view { ImVec2 corner; ImVec2 size; float strip; float top; float height; double across; double vertical; double from; double centre; profile_view(profile_state const &State, ImVec2 const Corner, ImVec2 const Size) : corner{Corner}, size{Size}, strip{State.show_plan ? 44.0f : 0.0f}, top{18.0f}, height{Size.y - strip - top}, across{std::max(1.0, State.view_to - State.view_from) / Size.x}, vertical{across / State.exaggeration}, from{State.view_from}, centre{State.view_centre} { } float x_of(double const Chainage) const { return corner.x + static_cast((Chainage - from) / across); } float y_of(double const Height) const { return corner.y + top + height * 0.5f - static_cast((Height - centre) / vertical); } double chainage_of(float const X) const { return from + (X - corner.x) * across; } double height_of(float const Y) const { return centre - (Y - corner.y - top - height * 0.5f) * vertical; } float plot_top() const { return corner.y + top; } float plot_bottom() const { return corner.y + top + height; } ImVec2 bottom_right() const { return {corner.x + size.x, corner.y + size.y}; } float mouse_distance2(double const Chainage, double const Height) const { auto const &mouse{ImGui::GetIO().MousePos}; auto const dx{x_of(Chainage) - mouse.x}; auto const dy{y_of(Height) - mouse.y}; return dx * dx + dy * dy; } }; void editor_mode::render_profile_toolbar() { auto &state{m_profile}; // the next item on the same line while it fits, past the edge of the plot on the next one auto const edge{ImGui::GetWindowPos().x + ImGui::GetWindowContentRegionMax().x}; auto const next = [&](float const Gap, float const Width) { ImGui::SameLine(0.0f, Gap); if (ImGui::GetCursorScreenPos().x + Width > edge) ImGui::NewLine(); }; auto const pill = [](char const *Label) { return ImGui::CalcTextSize(Label).x + ImGui::GetFrameHeight() * 1.75f; }; ImGui::AlignTextToFramePadding(); ImGui::TextDisabled("%s", STR_C("Show:")); next(ImGui::GetStyle().ItemSpacing.x, pill(STR_C("Track"))); pill_toggle(STR_C("Track"), kTrack, state.show_track); next(4.0f, pill(STR_C("Terrain"))); pill_toggle(STR_C("Terrain"), kTerrain, state.show_terrain); if (m_nmt.placed()) { next(4.0f, pill(STR_C("Terrain model"))); if (pill_toggle(STR_C("Terrain model"), kModel, state.show_model)) profile_sample_model(); item_tooltip(state.model_waiting ? "The ground of the terrain model (NMT) as it was measured, before the earthworks; its squares along the route are still being fetched" : "The ground of the terrain model (NMT) as it was measured, before the earthworks"); } next(4.0f, pill(STR_C("Plan curvature"))); pill_toggle(STR_C("Plan curvature"), kPlan, state.show_plan); next(6.0f, ImGui::CalcTextSize("(?)").x); ImGui::TextDisabled("(?)"); if (ImGui::IsItemHovered()) { ImGui::BeginTooltip(); ImGui::TextUnformatted(STR_C("Legend")); ImGui::Separator(); auto *draw{ImGui::GetWindowDrawList()}; auto const entry = [&](ImU32 const Colour, char const *Label) { auto const at{ImGui::GetCursorScreenPos()}; auto const height{ImGui::GetTextLineHeight()}; draw->AddLine(ImVec2(at.x, at.y + height * 0.5f), ImVec2(at.x + 22.0f, at.y + height * 0.5f), Colour, 3.0f); ImGui::Dummy(ImVec2(22.0f, height)); ImGui::SameLine(0.0f, 8.0f); ImGui::TextUnformatted(STR_C(Label)); }; entry(kGradeLine, "Grade line"); entry(kTangents, "Tangents"); entry(kTrack, "Track"); entry(kTerrain, "Terrain"); entry(kFormation, "Formation"); entry(kModel, "Terrain model"); entry(kPlan, "Plan curvature"); entry(kSwitchLabel, "Switches"); entry(kSharedEdge, "Shared timbers"); entry(kError, "Errors"); entry(kWarning, "Warnings"); ImGui::EndTooltip(); } next(14.0f, 120.0f); ImGui::SetNextItemWidth(120.0f); ImGui::SliderFloat("##exaggeration", &state.exaggeration, 1.0f, 500.0f, STR_C("heights x%.0f"), ImGuiSliderFlags_Logarithmic); if (ImGui::IsItemHovered()) ImGui::SetTooltip("%s", STR_C("Exaggeration of the heights against the lengths, also Ctrl+wheel over the plot")); next(ImGui::GetStyle().ItemSpacing.x, ImGui::CalcTextSize(STR_C("Fit")).x + ImGui::GetStyle().FramePadding.x * 2.0f); if (ImGui::Button(STR_C("Fit"))) profile_fit_view(); if (ImGui::IsItemHovered()) ImGui::SetTooltip("%s", STR_C("Shows the whole route")); next(14.0f, pill(STR_C("Camera follows"))); if (pill_toggle(STR_C("Camera follows"), kGradeLine, state.camera_follows) && state.camera_follows) state.view_framed = {}; item_tooltip("The view shows the part of the route the plot shows, once the plot is zoomed or moved"); } void editor_mode::render_profile_canvas(float const Height) { auto &state{m_profile}; auto const available{ImGui::GetContentRegionAvail()}; auto const hintheight{std::round(ImGui::GetTextLineHeight() * 1.2f)}; auto const hintrow{hintheight + 8.0f}; ImVec2 const size{std::max(200.0f, available.x), std::max(100.0f, Height - hintrow)}; ImVec2 const corner{ImGui::GetCursorScreenPos()}; ImGui::InvisibleButton("##profilecanvas", size); bool const hovered{ImGui::IsItemHovered()}; if (state.exaggeration <= 0.0f) profile_fit_exaggeration(profile_view{state, corner, size}); if (state.model_waiting && ImGui::GetTime() - state.model_polled > 0.5) profile_sample_model(); profile_navigate(profile_view{state, corner, size}, hovered); profile_follow_view(); profile_view const view{state, corner, size}; if (profile_canvas_edit(view, hovered)) profile_edited(); state.hover = hovered ? view.chainage_of(ImGui::GetIO().MousePos.x) : -1.0; auto *draw{ImGui::GetWindowDrawList()}; draw->AddRectFilled(corner, view.bottom_right(), IM_COL32(22, 24, 28, 255)); draw->PushClipRect(corner, view.bottom_right(), true); draw_profile_canvas(view, *draw); if (hovered && state.hover >= 0.0 && state.hover <= state.route.length && false == state.samples.empty()) { auto const x{view.x_of(state.hover)}; draw->AddLine(ImVec2(x, view.plot_top()), ImVec2(x, view.bottom_right().y), IM_COL32(255, 255, 255, 90)); if (state.hover_point < 0 && state.hover_grip < 0 && state.curve_grip < 0) profile_canvas_tooltip(); } draw->PopClipRect(); // what the mouse does over the thing under it, along the bottom of the plot std::vector> hints{{"wheel", "zoom"}, {"Ctrl+wheel", "heights"}, {"RMB move", "pan"}, {"LMB LMB", "on the line: add a point"}}; if (state.dragging >= 0) hints = {{"Ctrl", "only the height"}, {"Shift", "keep the grade before"}}; else if (state.curve_grip >= 0 || state.hover_grip >= 0) hints = {{"LMB move", "up or down: radius of the vertical curve"}}; else if (state.hover_point >= 0) hints = {{"LMB move", "move the point"}, {"Ctrl+LMB", "only the height"}, {"Shift+LMB", "keep the grade before"}, {"Del", "remove"}}; else if (state.hover_line) hints = {{"LMB LMB", "add a point here"}}; { auto const height{hintheight}; auto x{corner.x + 6.0f}; auto const y{corner.y + size.y + 4.0f}; draw->AddRectFilled(ImVec2(corner.x, corner.y + size.y), ImVec2(corner.x + size.x, corner.y + size.y + hintrow), IM_COL32(14, 15, 19, 255)); draw->PushClipRect(ImVec2(corner.x, corner.y + size.y), ImVec2(corner.x + size.x, corner.y + size.y + hintrow), true); for (auto const &hint : hints) { auto const words{gesture_words(hint.first)}; auto const text{STR(hint.second)}; if (x + gesture_width(height, words) + 6.0f + ImGui::CalcTextSize(text.c_str()).x > corner.x + size.x - 6.0f) break; x += gesture_draw(*draw, ImVec2(x, y), height, words, kLabel) + 6.0f; draw->AddText(ImVec2(x, std::round(y + (height - ImGui::GetTextLineHeight()) * 0.5f)), IM_COL32(235, 235, 235, 255), text.c_str()); x += ImGui::CalcTextSize(text.c_str()).x + 18.0f; } draw->PopClipRect(); } ImGui::Dummy(ImVec2(size.x, hintrow)); auto const &points{state.line.points}; if (state.dragging >= 0 && state.dragging < static_cast(points.size())) { auto const index{state.dragging}; ImGui::BeginTooltip(); ImGui::Text(STR_C("Point %d"), index + 1); ImGui::Text(STR_C("km %s H %.3f m"), profile::format_chainage(points[index].chainage + state.origin).c_str(), points[index].elevation); if (index > 0) ImGui::Text(STR_C("before %+.2f per mille"), profile::grade_after(state.line, index - 1) * 1000.0); if (index + 1 < static_cast(points.size())) ImGui::Text(STR_C("after %+.2f per mille"), profile::grade_after(state.line, index) * 1000.0); ImGui::EndTooltip(); } else if (state.curve_grip > 0 && state.curve_grip < static_cast(points.size())) { ImGui::BeginTooltip(); ImGui::Text("R %.0f m", points[state.curve_grip].radius); ImGui::EndTooltip(); } } void editor_mode::profile_fit_exaggeration(profile_view const &View) { auto &state{m_profile}; double low{std::numeric_limits::max()}; double high{std::numeric_limits::lowest()}; for (auto const &sample : state.samples) { low = std::min(low, sample.position.y); high = std::max(high, sample.position.y); } for (auto const &point : state.line.points) { low = std::min(low, point.elevation); high = std::max(high, point.elevation); } if (low > high) { low = 0.0; high = 0.0; } auto const range{std::max(1.0, (high - low) * 1.3)}; state.exaggeration = static_cast(std::clamp(View.across / (range / View.height), 1.0, 500.0)); state.view_centre = (low + high) * 0.5; } void editor_mode::profile_navigate(profile_view const &View, bool const Hovered) { auto &state{m_profile}; ImGuiIO const &io{ImGui::GetIO()}; if (Hovered && io.MouseWheel != 0.0f) { if (io.KeyCtrl) state.exaggeration = std::clamp(state.exaggeration * std::pow(1.2f, io.MouseWheel), 1.0f, 500.0f); else { auto const anchor{View.chainage_of(io.MousePos.x)}; auto const factor{std::pow(0.8, static_cast(io.MouseWheel))}; state.view_from = anchor - (anchor - state.view_from) * factor; state.view_to = anchor + (state.view_to - anchor) * factor; } } if (Hovered && (ImGui::IsMouseClicked(1) || ImGui::IsMouseClicked(2))) state.panning = true; if (state.panning) { if (ImGui::IsMouseDown(1) || ImGui::IsMouseDown(2)) { state.view_from -= io.MouseDelta.x * View.across; state.view_to -= io.MouseDelta.x * View.across; state.view_centre += io.MouseDelta.y * View.vertical; } else state.panning = false; } } bool editor_mode::profile_canvas_edit(profile_view const &View, bool const Hovered) { auto &state{m_profile}; auto &line{state.line}; auto &points{line.points}; ImGuiIO const &io{ImGui::GetIO()}; ImVec2 const mouse{io.MousePos}; auto const hit_point = [&]() { int hit{-1}; float best{64.0f}; for (int i = 0; i < static_cast(points.size()); ++i) { if (points[i].joint) continue; if (auto const distance{View.mouse_distance2(points[i].chainage, points[i].elevation)}; distance < best) { best = distance; hit = i; } } return hit; }; auto const hit_grip = [&]() { for (auto const &c : profile::curves(line)) { if (points[c.point].joint) continue; auto const chainage{points[c.point].chainage}; if (View.mouse_distance2(chainage, profile::elevation(line, chainage)) < 64.0f) return static_cast(c.point); } return -1; }; bool changed{false}; bool const busy{state.dragging >= 0 || state.curve_grip >= 0 || state.panning}; state.hover_point = Hovered && false == busy ? hit_point() : -1; state.hover_grip = Hovered && false == busy && state.hover_point < 0 ? hit_grip() : -1; state.hover_line = false; if (Hovered && false == busy && state.hover_point < 0 && state.hover_grip < 0 && points.size() >= 2) { auto const chainage{View.chainage_of(mouse.x)}; state.hover_line = chainage > points.front().chainage && chainage < points.back().chainage && std::abs(View.y_of(profile::elevation(line, chainage)) - mouse.y) < 12.0f; } if (state.dragging >= 0 || state.panning) ImGui::SetMouseCursor(ImGuiMouseCursor_ResizeAll); else if (state.curve_grip >= 0 || state.hover_grip >= 0) ImGui::SetMouseCursor(ImGuiMouseCursor_ResizeNS); else if (state.hover_point >= 0 || state.hover_line) ImGui::SetMouseCursor(ImGuiMouseCursor_Hand); if (Hovered && ImGui::IsMouseClicked(0)) { auto const point{hit_point()}; auto const grip{point < 0 ? hit_grip() : -1}; if (point >= 0) { state.dragging = point; state.selected = state.dragging; state.drag_press = mouse; state.drag_grab = {points[point].chainage - View.chainage_of(mouse.x), points[point].elevation - View.height_of(mouse.y)}; state.drag_moved = false; } else if (grip >= 0) { state.curve_grip = grip; state.selected = state.curve_grip; state.drag_press = mouse; state.drag_moved = false; } else state.selected = -1; } if (Hovered && ImGui::IsMouseDoubleClicked(0) && state.dragging < 0 && state.curve_grip < 0 && points.size() >= 2) { auto const chainage{View.chainage_of(mouse.x)}; if (chainage > points.front().chainage + 0.5 && chainage < points.back().chainage - 0.5 && std::abs(View.y_of(profile::elevation(line, chainage)) - mouse.y) < 12.0f) { std::size_t index{0}; while (index + 2 < points.size() && points[index + 1].chainage < chainage) ++index; profile::pvi inserted; inserted.chainage = std::round(chainage * 100.0) / 100.0; inserted.elevation = points[index].elevation + profile::grade_after(line, index) * (inserted.chainage - points[index].chainage); points.insert(points.begin() + index + 1, inserted); state.selected = static_cast(index) + 1; changed = true; } } if (state.dragging >= 0 && state.dragging < static_cast(points.size())) { // a click selects the point, it moves only when dragged, by as much as the mouse went state.drag_moved = state.drag_moved || std::abs(mouse.x - state.drag_press.x) > 3.0f || std::abs(mouse.y - state.drag_press.y) > 3.0f; if (false == ImGui::IsMouseDown(0)) state.dragging = -1; else if (state.drag_moved) { auto const index{state.dragging}; auto &point{points[index]}; bool const end{index == 0 || index + 1 == static_cast(points.size())}; auto const incoming{index > 0 ? profile::grade_after(line, index - 1) : 0.0}; if (false == end && false == io.KeyCtrl) point.chainage = std::clamp(std::round((View.chainage_of(mouse.x) + state.drag_grab.x) * 100.0) / 100.0, points[index - 1].chainage + 0.5, points[index + 1].chainage - 0.5); if (end_locked(line, state.context, index)) ; else if (io.KeyShift && index > 0) point.elevation = points[index - 1].elevation + incoming * (point.chainage - points[index - 1].chainage); else point.elevation = std::round((View.height_of(mouse.y) + state.drag_grab.y) * 1000.0) / 1000.0; changed = true; } } else state.dragging = -1; if (state.curve_grip > 0 && state.curve_grip + 1 < static_cast(points.size())) { // a click on the grip leaves the radius as it is state.drag_moved = state.drag_moved || std::abs(mouse.y - state.drag_press.y) > 3.0f; if (false == ImGui::IsMouseDown(0)) state.curve_grip = -1; else if (state.drag_moved) { auto const sagitta{View.height_of(mouse.y) - points[state.curve_grip].elevation}; points[state.curve_grip].radius = std::max(10.0, std::round(profile::radius_for_sagitta(line, state.curve_grip, sagitta) / 10.0) * 10.0); points[state.curve_grip].automatic = false; changed = true; } } else state.curve_grip = -1; if ((Hovered || ImGui::IsWindowFocused()) && state.selected > 0 && state.selected + 1 < static_cast(points.size()) && false == points[state.selected].joint && ImGui::IsKeyPressed(ImGuiKey_Delete)) { points.erase(points.begin() + state.selected); state.selected = -1; changed = true; } return changed; } void editor_mode::draw_profile_canvas(profile_view const &View, ImDrawList &Draw) const { auto const &state{m_profile}; auto const &line{state.line}; auto const &points{line.points}; auto const &samples{state.samples}; auto const bottomright{View.bottom_right()}; auto const plottop{View.plot_top()}; auto const plotbottom{View.plot_bottom()}; auto const step{nice_step(View.across * 110.0)}; auto const firsts{std::ceil((state.view_from + state.origin) / step) * step - state.origin}; for (int i = 0; firsts + i * step <= state.view_to; ++i) { auto const s{firsts + i * step}; auto const x{View.x_of(s)}; Draw.AddLine(ImVec2(x, plottop), ImVec2(x, plotbottom), kGrid); Draw.AddText(ImVec2(x + 3.0f, View.corner.y + 2.0f), kLabel, profile::format_chainage(s + state.origin).c_str()); } auto const heightstep{nice_step(View.vertical * 45.0)}; auto const firsth{std::floor(View.height_of(plotbottom) / heightstep) * heightstep}; for (int i = 0; firsth + i * heightstep <= View.height_of(plottop); ++i) { auto const y{View.y_of(firsth + i * heightstep)}; Draw.AddLine(ImVec2(View.corner.x, y), ImVec2(bottomright.x, y), kGrid); } for (auto const &zone : state.context.switches) { auto const left{View.x_of(zone.from)}; auto const right{std::max(View.x_of(zone.to), left + 2.0f)}; if (zone.name == STR("shared timbers")) { // the timbers shared with another track, past a switch or at a crossing: drawn with their bounds, nothing // breaks the grade over them Draw.PushClipRect(ImVec2(left, plottop), ImVec2(right, plotbottom), true); auto const height{plotbottom - plottop}; for (auto x = left - height; x < right; x += 8.0f) Draw.AddLine(ImVec2(x, plotbottom), ImVec2(x + height, plottop), kShared, 1.5f); Draw.PopClipRect(); Draw.AddLine(ImVec2(left, plottop), ImVec2(left, plotbottom), kSharedEdge, 1.0f); Draw.AddLine(ImVec2(right, plottop), ImVec2(right, plotbottom), kSharedEdge, 1.0f); if (right - left > 4.0f) Draw.AddText(ImVec2(left + 2.0f, plotbottom - 32.0f), kSharedEdge, zone.name.c_str()); continue; } Draw.AddRectFilled(ImVec2(left, plottop), ImVec2(right, plotbottom), kSwitch); if (right - left > 20.0f || View.across < 0.5) Draw.AddText(ImVec2(left + 2.0f, plotbottom - 16.0f), kSwitchLabel, zone.name.c_str()); } // the ends which inherit the grade line from a switch, said where they are if (false == state.start_inherits.empty()) { auto const x{View.x_of(0.0)}; Draw.AddLine(ImVec2(x, plottop), ImVec2(x, plotbottom), kSwitchLabel, 2.0f); Draw.AddText(ImVec2(x + 4.0f, plottop + 18.0f), kSwitchLabel, ("<- " + state.start_inherits).c_str()); } if (false == state.end_inherits.empty()) { auto const x{View.x_of(state.route.length)}; auto const text{state.end_inherits + " ->"}; Draw.AddLine(ImVec2(x, plottop), ImVec2(x, plotbottom), kSwitchLabel, 2.0f); Draw.AddText(ImVec2(x - 4.0f - ImGui::CalcTextSize(text.c_str()).x, plottop + 34.0f), kSwitchLabel, text.c_str()); } auto const polyline = [&](auto const &Height, ImU32 const Colour, float const Thickness) { ImVec2 previous{}; bool open{false}; float lastx{-1e9f}; for (std::size_t i = 0; i < samples.size(); ++i) { auto const x{View.x_of(samples[i].chainage)}; auto const h{Height(i)}; if (std::isnan(h) || x < View.corner.x - 50.0f || x > bottomright.x + 50.0f) { open = false; continue; } if (open && x - lastx < 1.0f && i + 1 < samples.size()) continue; ImVec2 const next{x, View.y_of(h)}; if (open) Draw.AddLine(previous, next, Colour, Thickness); previous = next; lastx = x; open = true; } }; if (state.show_terrain) { auto const fill{(kTerrain & 0x00FFFFFFu) | 0x30000000u}; float lastx{-1e9f}; ImVec2 previous{}; bool open{false}; for (std::size_t i = 0; i < samples.size(); ++i) { auto const x{View.x_of(samples[i].chainage)}; auto const h{state.terrain[i]}; if (std::isnan(h) || x < View.corner.x - 50.0f || x > bottomright.x + 50.0f) { open = false; continue; } if (open && x - lastx < 2.0f && i + 1 < samples.size()) continue; ImVec2 const next{x, std::min(View.y_of(h), plotbottom)}; if (open) Draw.AddQuadFilled(previous, next, ImVec2(next.x, plotbottom), ImVec2(previous.x, plotbottom), fill); previous = next; lastx = x; open = true; } polyline([&](std::size_t const I) { return state.terrain[I]; }, kTerrain, 1.5f); // the formation under the grade line, which is what meets the ground polyline([&](std::size_t const I) { return profile::elevation(state.line, samples[I].chainage) - (I < state.beds.size() ? state.beds[I] : 0.0); }, kFormation, 1.0f); } if (state.show_model && state.model.size() == samples.size()) polyline([&](std::size_t const I) { return state.model[I]; }, kModel, 1.5f); if (state.show_track) polyline([&](std::size_t const I) { return samples[I].position.y; }, kTrack, 1.5f); for (std::size_t i = 0; i + 1 < points.size(); ++i) Draw.AddLine(ImVec2(View.x_of(points[i].chainage), View.y_of(points[i].elevation)), ImVec2(View.x_of(points[i + 1].chainage), View.y_of(points[i + 1].elevation)), kTangents, 1.0f); { auto const from{std::max(points.front().chainage, View.chainage_of(View.corner.x))}; auto const to{std::min(points.back().chainage, View.chainage_of(bottomright.x))}; ImVec2 previous{View.x_of(from), View.y_of(profile::elevation(line, from))}; auto const firstx{previous.x}; for (int i = 1; firstx + 2.0f * i <= View.x_of(to) + 2.0f; ++i) { auto const x{firstx + 2.0f * i}; auto const chainage{std::min(to, View.chainage_of(x))}; ImVec2 const next{View.x_of(chainage), View.y_of(profile::elevation(line, chainage))}; Draw.AddLine(previous, next, kGradeLine, 2.5f); previous = next; } } for (std::size_t i = 0; i + 1 < points.size(); ++i) { auto const middle{(points[i].chainage + points[i + 1].chainage) * 0.5}; auto const x{View.x_of(middle)}; if (View.x_of(points[i + 1].chainage) - View.x_of(points[i].chainage) < 70.0f) continue; auto const text{format("%+.2f o/oo L %.0f m", profile::grade_after(line, i) * 1000.0, points[i + 1].chainage - points[i].chainage)}; auto const size{ImGui::CalcTextSize(text.c_str())}; if (View.x_of(points[i + 1].chainage) - View.x_of(points[i].chainage) < size.x + 30.0f) continue; plot_label(Draw, ImVec2(x - size.x * 0.5f, View.y_of(profile::elevation(line, middle)) - size.y - 10.0f), kGradeLine, text); } for (auto const &c : profile::curves(line)) { for (auto const at : {c.start, c.end}) { auto const y{View.y_of(profile::elevation(line, at))}; Draw.AddLine(ImVec2(View.x_of(at), y - 7.0f), ImVec2(View.x_of(at), y + 7.0f), kGradeLine, 1.5f); } auto const chainage{points[c.point].chainage}; ImVec2 const grip{View.x_of(chainage), View.y_of(profile::elevation(line, chainage))}; bool const lit{static_cast(c.point) == state.curve_grip || static_cast(c.point) == state.hover_grip}; Draw.AddCircleFilled(grip, lit ? 6.5f : 4.5f, overlay_color::grip, 4); if (lit) Draw.AddCircle(grip, 10.0f, overlay_color::highlight, 12, 2.0f); plot_label(Draw, ImVec2(grip.x + 8.0f, grip.y + 6.0f), overlay_color::grip, format("R %.0f", points[c.point].radius)); } for (int i = 0; i < static_cast(points.size()); ++i) { ImVec2 const at{View.x_of(points[i].chainage), View.y_of(points[i].elevation)}; bool const lit{i == state.hover_point || i == state.dragging}; auto const radius{lit ? 7.0f : 5.5f}; if (i == 0 || i + 1 == static_cast(points.size())) Draw.AddRectFilled(ImVec2(at.x - radius, at.y - radius), ImVec2(at.x + radius, at.y + radius), IM_COL32(70, 140, 255, 255)); else if (points[i].joint) Draw.AddQuadFilled(ImVec2(at.x, at.y - radius), ImVec2(at.x + radius, at.y), ImVec2(at.x, at.y + radius), ImVec2(at.x - radius, at.y), IM_COL32(90, 210, 230, 255)); else Draw.AddCircleFilled(at, radius, overlay_color::marked); if (i == state.selected) Draw.AddCircle(at, 11.0f, overlay_color::highlight, 16, 2.5f); else if (lit) Draw.AddCircle(at, 11.0f, IM_COL32(255, 255, 255, 160), 16, 1.5f); // the heights only where they're asked for, the numbers of the points always auto const text{i == state.selected || lit ? format("%d H %.3f", i + 1, points[i].elevation) : std::to_string(i + 1)}; plot_label(Draw, ImVec2(at.x + 8.0f, at.y - 22.0f), overlay_color::marked, text); } // heights along the left edge, over everything else for (int i = 0; firsth + i * heightstep <= View.height_of(plottop); ++i) plot_label(Draw, ImVec2(View.corner.x + 4.0f, View.y_of(firsth + i * heightstep) - 8.0f), kLabel, format("%.1f", firsth + i * heightstep)); for (auto const &issue : state.issues) { auto const x{View.x_of(issue.chainage)}; Draw.AddTriangleFilled(ImVec2(x - 5.0f, plottop), ImVec2(x + 5.0f, plottop), ImVec2(x, plottop + 8.0f), issue.error ? kError : kWarning); } if (state.show_plan && false == samples.empty()) { auto const middle{plotbottom + View.strip * 0.5f}; Draw.AddLine(ImVec2(View.corner.x, plotbottom), ImVec2(bottomright.x, plotbottom), IM_COL32(255, 255, 255, 60)); Draw.AddLine(ImVec2(View.corner.x, middle), ImVec2(bottomright.x, middle), IM_COL32(255, 255, 255, 40)); ImVec2 previous{}; bool open{false}; float lastx{-1e9f}; for (std::size_t i = 0; i < samples.size(); ++i) { auto const x{View.x_of(samples[i].chainage)}; if (x < View.corner.x - 50.0f || x > bottomright.x + 50.0f || (open && x - lastx < 1.0f)) continue; ImVec2 const next{x, middle - std::clamp(static_cast(samples[i].curvature * 300.0 * View.strip * 0.4), -View.strip * 0.45f, View.strip * 0.45f)}; if (open) Draw.AddLine(previous, next, IM_COL32(120, 190, 255, 255), 1.5f); previous = next; lastx = x; open = true; } Draw.AddText(ImVec2(View.corner.x + 4.0f, plotbottom + 1.0f), kLabel, STR_C("plan: curvature (up: left)")); } } void editor_mode::profile_canvas_tooltip() const { auto const &state{m_profile}; auto const &samples{state.samples}; auto const designed{profile::elevation(state.line, state.hover)}; auto const existing{editor_track::sampled_elevation(samples, state.hover)}; ImGui::BeginTooltip(); ImGui::Text(STR_C("km %s"), profile::format_chainage(state.hover + state.origin).c_str()); ImGui::TextColored(ImGui::ColorConvertU32ToFloat4(kGradeLine), STR_C("Grade line %.3f m, %+.2f per mille"), designed, profile::grade(state.line, state.hover) * 1000.0); ImGui::TextColored(ImGui::ColorConvertU32ToFloat4(kTrack), STR_C("Track %.3f m (%+.3f), %+.2f per mille"), existing, existing - designed, editor_track::sampled_grade(samples, state.hover) * 1000.0); auto const next{std::lower_bound(samples.begin(), samples.end(), state.hover, [](editor_track::route_sample const &Sample, double const Value) { return Sample.chainage < Value; })}; auto const index{static_cast(std::distance(samples.begin(), std::min(next, std::prev(samples.end()))))}; if (index < state.terrain.size() && false == std::isnan(state.terrain[index])) { auto const formation{designed - (index < state.beds.size() ? state.beds[index] : 0.0)}; ImGui::TextColored(ImGui::ColorConvertU32ToFloat4(kFormation), STR_C("Formation %.3f m, %.2f m under the rail top"), formation, designed - formation); ImGui::TextColored(ImGui::ColorConvertU32ToFloat4(kTerrain), STR_C("Terrain %.3f m, %s %.2f m"), state.terrain[index], formation > state.terrain[index] ? STR_C("fill") : STR_C("cut"), std::abs(formation - state.terrain[index])); } if (state.show_model && index < state.model.size() && false == std::isnan(state.model[index])) ImGui::TextColored(ImGui::ColorConvertU32ToFloat4(kModel), STR_C("Terrain model %.3f m, the grade line %+.2f m from it"), state.model[index], designed - state.model[index]); auto const &span{state.route.spans[samples[index].span]}; ImGui::TextDisabled("%s%s", span.track->name().c_str(), span.track->eType != tt_Normal ? STR_C(" (switch, tilted as a whole with its branch)") : ""); ImGui::EndTooltip(); } void editor_mode::draw_profile_overlay() const { auto const &state{m_profile}; if (false == state.open || state.samples.size() < 2) return; screen_projection const projection; ImDrawList *drawlist{ImGui::GetBackgroundDrawList(ImGui::GetMainViewport())}; auto const &samples{state.samples}; glm::dvec3 previous{samples.front().position}; glm::dvec3 designedprevious{previous.x, profile::elevation(state.line, 0.0), previous.z}; for (std::size_t i = 1; i < samples.size(); ++i) { if (glm::distance(samples[i].position, previous) < 2.0 && i + 1 < samples.size()) continue; auto const &next{samples[i].position}; projection.line(drawlist, previous, next, IM_COL32(255, 150, 40, 200), 2.5f); glm::dvec3 const designed{next.x, profile::elevation(state.line, samples[i].chainage), next.z}; if (std::abs(designed.y - next.y) > 0.01 || std::abs(designedprevious.y - previous.y) > 0.01) projection.line(drawlist, designedprevious, designed, kGradeLine, 2.0f); previous = next; designedprevious = designed; } // the grade line told in the view: the grades of its straight stretches, the ranges of its vertical curves, its points glm::dvec3 const eye{Global.pCamera.Pos}; double constexpr labels_range{1500.0}; auto const on_line = [&](double const Chainage) { auto position{editor_track::sampled_position(samples, Chainage)}; position.y = profile::elevation(state.line, Chainage); return position; }; auto const label = [&](glm::dvec3 const &At, std::string const &Text, ImU32 const Colour) { ImVec2 screen; if (glm::distance(At, eye) > labels_range || false == projection.project(At + glm::dvec3{0.0, 1.5, 0.0}, screen)) return; auto const size{ImGui::CalcTextSize(Text.c_str())}; ImVec2 const corner{screen.x - size.x * 0.5f - 4.0f, screen.y - size.y - 4.0f}; drawlist->AddRectFilled(corner, ImVec2(corner.x + size.x + 8.0f, corner.y + size.y + 4.0f), IM_COL32(20, 20, 24, 190), 3.0f); drawlist->AddText(ImVec2(corner.x + 4.0f, corner.y + 2.0f), Colour, Text.c_str()); }; auto const &points{state.line.points}; auto const curves{profile::curves(state.line)}; // the straight stretches between the curves for (std::size_t k = 0; k + 1 < points.size(); ++k) { auto from{points[k].chainage}; auto to{points[k + 1].chainage}; for (auto const &curve : curves) { if (curve.point == k) from = std::max(from, curve.end); if (curve.point == k + 1) to = std::min(to, curve.start); } if (to - from < 5.0) continue; auto const grade{(points[k + 1].elevation - points[k].elevation) / std::max(1e-6, points[k + 1].chainage - points[k].chainage) * 1000.0}; label(on_line((from + to) * 0.5), format("%+.2f o/oo L %.0f m", grade, points[k + 1].chainage - points[k].chainage), kGradeLine); } // the vertical curves: their range along the track, the ends marked, the radius at the point for (auto const &curve : curves) { glm::dvec3 previouscurve{on_line(curve.start)}; auto const steps{std::max(2, static_cast((curve.end - curve.start) / 2.0))}; for (int s = 1; s <= steps; ++s) { auto const next{on_line(curve.start + (curve.end - curve.start) * s / steps)}; projection.line(drawlist, previouscurve, next, kTangents, 5.0f); previouscurve = next; } for (auto const end : {curve.start, curve.end}) { auto const at{on_line(end)}; auto const sample{editor_track::sampled_position(samples, std::min(state.route.length, end + 0.5)) - editor_track::sampled_position(samples, std::max(0.0, end - 0.5))}; glm::dvec2 across{-sample.z, sample.x}; across = glm::length(across) > 1e-9 ? glm::normalize(across) * 2.0 : glm::dvec2{2.0, 0.0}; projection.line(drawlist, at - glm::dvec3{across.x, 0.0, across.y}, at + glm::dvec3{across.x, 0.0, across.y}, kTangents, 2.0f); } label(on_line(points[curve.point].chainage) + glm::dvec3{0.0, 1.2, 0.0}, format("R %.0f T %.1f m", points[curve.point].radius, curve.tangent), kTangents); } for (int i = 0; i < static_cast(points.size()); ++i) { auto const position{on_line(points[i].chainage)}; ImVec2 screen; if (false == projection.project(position, screen)) continue; drawlist->AddCircleFilled(screen, i == state.selected ? 7.0f : 5.0f, overlay_color::marked); drawlist->AddText(ImVec2(screen.x + 8.0f, screen.y - 16.0f), overlay_color::marked, format("%d", i + 1).c_str()); if (glm::distance(position, eye) <= labels_range * 0.5) drawlist->AddText(ImVec2(screen.x + 8.0f, screen.y - 2.0f), overlay_color::marked, format("km %s, %.2f m", profile::format_chainage(points[i].chainage + state.origin).c_str(), points[i].elevation).c_str()); } if (state.hover >= 0.0 && state.hover <= state.route.length) { ImVec2 screen; if (projection.project(editor_track::sampled_position(samples, state.hover), screen)) drawlist->AddCircle(screen, 12.0f, IM_COL32(255, 255, 255, 230), 16, 2.5f); } }