17
0
mirror of https://github.com/MaSzyna-EU07/maszyna.git synced 2026-10-26 22:54:54 +01:00
Files
maszyna/application/editormodeprofile.cpp
maj00r 7fb9d2d458 fix(editor): the whole-scenery grade fit takes spans a few metres apart for one bridge, leaves the culverts on the embankment and ends a bridge at a joint by its abutment
Bridged stretches less than 5 m apart are merged into one bridge on its piers, and those shorter than 3 m, a ditch or a
culvert under the embankment, are no bridge. A bridge end within 1 m of a joint of the route or a break of the line moves to
it, so the cut leaves no sliver of a path by the abutment. Against the OpenStreetMap railway bridges the marked share stays
the same (lk91 60 of 66, TG 90 of 104); the places with nothing under them in the map fall from 6 to 4 on lk91 and from 18
to 12 on TG, and the bridge paths shorter than 1 m from 9 to 0 and from 11 to 1.
2026-10-22 05:56:05 +02:00

2888 lines
121 KiB
C++

/*
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 <algorithm>
#include <cmath>
#include <cstdio>
#include <limits>
#include <map>
#include <optional>
#include <set>
#include <sstream>
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;
}
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<int>(Line.points.size()))
return Line.join_end && Context.end_joined;
return false;
}
bool has_errors(std::vector<profile::issue> 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<editor_track::route_sample> 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 <route length> <start x y z> <end x y z> <middle x y z>
// speed <km/h> origin <chainage of the start> join <start 0/1> <end 0/1>
// norms <grade max> <grade over switches max> <k of R >= k*V^2> <R min> <change needing a curve> <constant grade min> <fixed tolerance>
// point <chainage> <elevation> <vertical curve R> <automatic R 0/1>
// 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<stored_profile> 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<scene::layer_handle> 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<std::string> 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<int>(i);
if (plan_distance(entry.start, end) <= 0.1 && plan_distance(entry.end, start) <= 0.1)
{
Reversed = true;
return static_cast<int>(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<profile::pvi> 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<double> start, end;
// the newer designs first: where two lie over the same stretch, the one kept later is the one meant
std::vector<std::pair<double, double>> 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<std::pair<TTrack const *, int>> 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<std::size_t> 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<std::pair<double, double>> 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<double> 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<profile::pvi> 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<scene::layer_handle> 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<std::ptrdiff_t>(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_track::route> editor_mode::profile_stored_route(stored_profile const &Entry, TTrack **From, TTrack **To) const
{
auto const touching = [](glm::dvec3 const &Point) {
std::vector<TTrack *> 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_remove_point(int const Index)
{
auto &state{m_profile};
auto &points{state.line.points};
if (Index <= 0 || Index + 1 >= static_cast<int>(points.size()))
return;
points.erase(points.begin() + Index);
for (auto const neighbour : {Index - 1, Index})
if (neighbour > 0 && neighbour + 1 < static_cast<int>(points.size()) && false == points[neighbour].joint)
points[neighbour].automatic = true;
state.selected = -1;
profile_edited();
}
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<TTrack *> 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<int>(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<double>::quiet_NaN());
std::vector<glm::dvec3> points;
points.reserve(samples.size());
for (auto const &sample : samples)
points.push_back(sample.position);
std::vector<char> 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<char> bedless(samples.size(), 0);
std::vector<double> 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<switch_store::record> 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<std::ptrdiff_t>(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<double>::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<double> 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<double> 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 = editor_track::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<double> 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 = editor_track::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<double, double> 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<double>(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<int, int> editor_mode::profile_generate_ground(std::vector<std::pair<double, double>> 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<heightmap::chunk_key> 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<std::size_t>(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<std::size_t> 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<glm::dvec3> 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<char> found;
ground_heights(probes, true, &found);
if (std::count(found.begin(), found.end(), 1) * 20 > static_cast<long>(probes.size()))
{
++skipped;
continue;
}
// the ground of the chunk: the ground under the nearest point of the route, or the formation there
std::vector<float> 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<double>::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<float>(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::size_t>(std::clamp<long>(std::lround((X - low.x) / cellsize), 0, cells))};
auto const iz{static_cast<std::size_t>(std::clamp<long>(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<std::pair<double, double>> 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<int, int>{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<heightmap::corridor_point> 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<int>(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<std::size_t> 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<int>(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<mark> marks;
for (auto *index : {&state.selected, &state.dragging, &state.curve_grip})
if (*index >= 0 && *index < static_cast<int>(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<double>::max()};
for (int i = 0; i < static_cast<int>(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<double, double> 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<glm::dvec3> 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<double>(viewhigh.x - viewlow.x))};
auto const height{std::max(50.0, static_cast<double>(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<double>(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<float>(extent), 5.0f, 50000.0f);
auto const scale{2.0 * static_cast<double>(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<double>(Global.FieldOfView) / std::max(0.01, static_cast<double>(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, std::vector<profile::departure> const &Bridges)
{
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<TTrack *> 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<std::pair<TTrack *, editor_track::state>> states;
std::vector<TTrack *> created;
auto cuts{profile::breaks(line)};
// a bridge ends at a joint or a break of the line by its abutment rather than leave a sliver of a path there
double constexpr kBridgeSnap{1.0}; // m
auto joints{cuts};
for (auto const &span : state.route.spans)
joints.insert(joints.end(), {span.from, span.to});
auto bridges{Bridges};
for (auto &bridge : bridges)
for (auto *end : {&bridge.from, &bridge.to})
if (auto const nearest{std::min_element(joints.begin(), joints.end(), [&](double const A, double const B) { return std::abs(A - *end) < std::abs(B - *end); })};
nearest != joints.end() && std::abs(*nearest - *end) <= kBridgeSnap)
*end = *nearest;
for (auto const &bridge : bridges)
cuts.insert(cuts.end(), {bridge.from, bridge.to});
std::sort(cuts.begin(), cuts.end());
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); }, cuts, states, created)};
// the paths over a bridge sound and light as on one, and get no formation; within what a cut too near an end leaves on
double constexpr kBridgeEnd{0.5}; // m
std::size_t bridged{0};
for (auto const &span : state.route.spans)
if (span.track->eEnvironment != e_bridge && span.track->eEnvironment != e_tunnel
&& std::any_of(bridges.begin(), bridges.end(), [&](profile::departure const &Bridge) { return span.from >= Bridge.from - kBridgeEnd && span.to <= Bridge.to + kBridgeEnd; }))
{
span.track->eEnvironment = e_bridge;
++bridged;
}
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");
if (bridged > 0)
state.status += format(STR_P("\nOver a bridge or a viaduct: %zu path", "\nOver a bridge or a viaduct: %zu paths", bridged), bridged);
auto const name = [](TTrack const *Track) { return name_of(*Track); };
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<long>(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<std::size_t>(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<TTrack *> 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<std::ptrdiff_t>(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<int>(errors), static_cast<int>(warnings)) : format(STR_C("%d errors"), static_cast<int>(errors));
colour = ImVec4(1.0f, 0.45f, 0.38f, 1.0f);
}
else if (warnings > 0)
{
chip = format(STR_C("%d warnings"), static_cast<int>(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<int>(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<int>(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);
norms.radius_factor = std::max(0.0, norms.radius_factor);
norms.radius_min = std::max(10.0, norms.radius_min);
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<std::ptrdiff_t>(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<int>(errors)), STR_C("errors"), errors > 0 ? IM_COL32(255, 100, 80, 255) : good},
{format("%d", static_cast<int>(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<int>(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<int>(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")))
{
profile_remove_point(index);
return;
}
}
}
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<int>(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<float>((Chainage - from) / across); }
float y_of(double const Height) const { return corner.y + top + height * 0.5f - static_cast<float>((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<std::pair<char const *, char const *>> 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<int>(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<int>(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<int>(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<double>::max()};
double high{std::numeric_limits<double>::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<float>(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<double>(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<int>(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<int>(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<int>(index) + 1;
changed = true;
}
}
if (state.dragging >= 0 && state.dragging < static_cast<int>(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<int>(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<int>(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<int>(points.size()) && false == points[state.selected].joint && ImGui::IsKeyPressed(ImGuiKey_Delete))
profile_remove_point(state.selected);
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<int>(c.point) == state.curve_grip || static_cast<int>(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<int>(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<int>(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<float>(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::size_t>(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<int>((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<int>(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);
}
}