17
0
mirror of https://github.com/MaSzyna-EU07/maszyna.git synced 2026-10-17 08:39:19 +02:00
Files
maszyna/application/editormodetrackbed.cpp

133 lines
4.5 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 "editor/editorGeometry.hpp"
#include "editor/editorGroundFollow.hpp"
#include "utilities/Globals.h"
#include "world/Track.h"
#include <algorithm>
#include <cmath>
#include <limits>
namespace
{
using geometry::plan_distance;
using geometry::plan_of;
} // namespace
glm::dvec3 editor_mode::cursor_terrain(double const Height)
{
auto point{cursor_level(Height)};
for (int i = 0; i < 5 && std::isfinite(point.x) && std::isfinite(point.z); ++i)
{
std::vector<char> found;
auto ground{ground_heights({glm::dvec3{point.x, std::max(point.y, static_cast<double>(Global.pCamera.Pos.y)), point.z}}, true, &found).front()};
if (found[0] == 0)
ground = nearby_ground(point);
if (std::isnan(ground))
return point;
auto const next{cursor_level(ground)};
auto const settled{plan_distance(next, point) < 0.01};
point = {next.x, ground, next.z};
if (settled)
break;
}
return point;
}
double editor_mode::nearby_ground(glm::dvec3 const &Point) const
{
auto const ceiling{std::max(Point.y, static_cast<double>(Global.pCamera.Pos.y))};
for (double const radius : {10.0, 25.0, 50.0, 100.0, 200.0, 400.0})
{
std::vector<glm::dvec3> probes;
for (int i = 0; i < 12; ++i)
{
auto const angle{glm::two_pi<double>() * i / 12.0};
probes.push_back({Point.x + radius * std::cos(angle), ceiling, Point.z + radius * std::sin(angle)});
}
std::vector<char> found;
auto const heights{ground_heights(probes, true, &found)};
double sum{0.0};
int count{0};
for (std::size_t i = 0; i < probes.size(); ++i)
if (found[i] != 0)
{
sum += heights[i];
++count;
}
if (count > 0)
return sum / count;
}
return std::numeric_limits<double>::quiet_NaN();
}
std::pair<double, double> editor_mode::trackbed(TTrack const *Style)
{
return Style != nullptr ? profile_ballast(*Style) : std::pair<double, double>{0.2, 0.5 * 1.435 + 0.5 + 1.1};
}
std::vector<double> editor_mode::bed_heights(std::vector<glm::dvec3> const &Points, std::vector<glm::dvec2> const &Directions, TTrack const *Style) const
{
auto const [depth, half]{trackbed(Style)};
std::vector<glm::dvec3> probes;
probes.reserve(Points.size() * 3);
for (std::size_t i = 0; i < Points.size(); ++i)
{
auto const &direction{Directions[i]};
auto const side{glm::length(direction) > 1e-6 ? glm::normalize(glm::dvec2{-direction.y, direction.x}) * half : glm::dvec2{0.0}};
probes.push_back(Points[i]);
probes.push_back(Points[i] + glm::dvec3{side.x, 0.0, side.y});
probes.push_back(Points[i] - glm::dvec3{side.x, 0.0, side.y});
}
std::vector<char> found;
auto const heights{ground_heights(probes, true, &found)};
std::vector<double> result(Points.size(), std::numeric_limits<double>::quiet_NaN());
for (std::size_t i = 0; i < Points.size(); ++i)
{
auto const axis{found[3 * i] != 0 ? heights[3 * i] : std::numeric_limits<double>::quiet_NaN()};
auto toe{std::numeric_limits<double>::infinity()};
for (std::size_t j = 3 * i + 1; j < 3 * i + 3; ++j)
if (found[j] != 0)
toe = std::min(toe, heights[j]);
if (std::isfinite(toe))
result[i] = std::isnan(axis) ? toe + depth : std::max(toe + depth, axis);
else if (false == std::isnan(axis))
result[i] = axis + depth;
}
return result;
}
double editor_mode::bed_height(glm::dvec3 const &Point, glm::dvec2 const &Direction, TTrack const *Style) const
{
auto const height{bed_heights({Point}, {Direction}, Style).front()};
if (false == std::isnan(height))
return height;
auto const nearby{nearby_ground(Point)};
return std::isnan(nearby) ? Point.y : nearby + trackbed(Style).first;
}
void editor_mode::follow_ground(std::vector<segment_data> &Pieces, double const Height0, double const Grade0, double const Height1, double const Grade1, TTrack const *Style) const
{
ground_follow::follow(Pieces, {Height0, Grade0, Height1, Grade1}, [&](std::vector<glm::dvec3> const &Points, std::vector<glm::dvec2> const &Directions) { return bed_heights(Points, Directions, Style); });
}
glm::dvec3 editor_mode::lay_bed(glm::dvec3 const &Point)
{
auto const &tool{m_lay};
TTrack const *style{tool.start.track != nullptr ? tool.start.track : tool.copy_selected ? selected_track() : nullptr};
return {Point.x, bed_height(Point, tool.points.empty() ? glm::dvec2{0.0} : plan_of(Point - tool.points.back()), style), Point.z};
}