mirror of
https://github.com/electronicarts/CNC_TS_and_RA2_Mission_Editor.git
synced 2025-07-17 19:56:38 -04:00
TS_RA2_Mission_Editor_sourcecode_23.02.2023_1
TS_RA2_Mission_Editor_sourcecode_23.02.2023_1
This commit is contained in:
brbarnesEA
commit
a547a31a37
377 changed files with 90136 additions and 0 deletions
251
MissionEditor/TubeTool.cpp
Normal file
251
MissionEditor/TubeTool.cpp
Normal file
|
|
@ -0,0 +1,251 @@
|
|||
/*
|
||||
FinalSun/FinalAlert 2 Mission Editor
|
||||
|
||||
Copyright (C) 1999-2024 Electronic Arts, Inc.
|
||||
Authored by Matthias Wagner
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include "StdAfx.h"
|
||||
#include "TubeTool.h"
|
||||
#include "MapData.h"
|
||||
#include "IsoView.h"
|
||||
#include "Structs.h"
|
||||
#include "inlines.h"
|
||||
#include <memory>
|
||||
#include <algorithm>
|
||||
|
||||
std::unique_ptr<CTube> findTubeEndAt(const MapCoords& mapCoords3d, const ProjectedCoords& projCoords, const CMapData& map)
|
||||
{
|
||||
const auto& tubes = map.GetTubes();
|
||||
auto it = std::find_if(tubes.begin(), tubes.end(), [mapCoords3d](const auto& tube) { return tube->getEndCoords() == mapCoords3d; });
|
||||
if (it != tubes.end())
|
||||
return std::make_unique<CTube>(**it);
|
||||
// if not found, also try with z - 4 (e.g. behind cliffs)
|
||||
auto backMapCoords = map.ToMapCoords3d(projCoords, map.GetHeightAt(mapCoords3d) - 4);
|
||||
it = std::find_if(tubes.begin(), tubes.end(), [backMapCoords](const auto& tube) { return tube->getEndCoords() == backMapCoords; });
|
||||
if (it != tubes.end())
|
||||
return std::make_unique<CTube>(**it);
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
std::vector<std::uint16_t> findTubesAt(const ProjectedCoords& projCoords, const CMapData& map)
|
||||
{
|
||||
std::vector<std::uint16_t> foundTubes;
|
||||
for (auto& t : map.GetTubes())
|
||||
{
|
||||
auto coords = map.ToMapCoords3d(projCoords, map.GetHeightAt(t->getStartCoords()));
|
||||
if (t->touches(coords))
|
||||
foundTubes.push_back(t->getId());
|
||||
}
|
||||
return foundTubes;
|
||||
}
|
||||
|
||||
AddTubeTool::AddTubeTool(CMapData& map, CIsoView& view, bool bidirectional): MapTool(map, view),
|
||||
m_bidirectional(bidirectional)
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
void AddTubeTool::finish()
|
||||
{
|
||||
if (m_tube) {
|
||||
if (m_tube->isValid())
|
||||
{
|
||||
getMap().SetTube(m_tube.get());
|
||||
if (m_bidirectional)
|
||||
{
|
||||
auto t2 = m_tube->reverse();
|
||||
getMap().SetTube(&t2);
|
||||
}
|
||||
}
|
||||
getView().RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);
|
||||
}
|
||||
m_mm_tube.reset();
|
||||
m_tube.reset();
|
||||
m_modified_tubes.clear();
|
||||
}
|
||||
|
||||
bool AddTubeTool::onRButtonUp(const ProjectedCoords& projCoords, const MapCoords& mapCoords3d, MapToolMouseFlags flags)
|
||||
{
|
||||
if (!m_modified_tubes.empty())
|
||||
{
|
||||
// reset modified tubes to original state
|
||||
for (auto& tube : m_modified_tubes)
|
||||
{
|
||||
getMap().SetTube(tube.get());
|
||||
}
|
||||
}
|
||||
bool ret = m_tube != nullptr;
|
||||
m_mm_tube.reset();
|
||||
m_tube.reset();
|
||||
m_modified_tubes.clear();
|
||||
getView().RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);
|
||||
return ret; // if we were adding or modifying a tube, a right click should just cancel this current tube, not the whole tool
|
||||
}
|
||||
|
||||
void AddTubeTool::onLButtonDblClick(const ProjectedCoords& projCoords, const MapCoords& mapCoords3d, MapToolMouseFlags flags)
|
||||
{
|
||||
}
|
||||
|
||||
void AddTubeTool::onLButtonUp(const ProjectedCoords& projCoords, const MapCoords& mapCoords3d, MapToolMouseFlags flags)
|
||||
{
|
||||
if (!m_tube) {
|
||||
// first check if there is already a tube at the given location and take ownership if existing
|
||||
auto tube = getTubeToModify(mapCoords3d, projCoords, flags);
|
||||
if (tube) {
|
||||
m_tube = std::move(tube);
|
||||
|
||||
// also find reverse direction
|
||||
if (m_bidirectional)
|
||||
{
|
||||
const auto& tubes = getMap().GetTubes();
|
||||
auto reversed = m_tube->reverse();
|
||||
auto reverseIt = std::find_if(tubes.begin(), tubes.end(), [&reversed](const auto& tube) { return tube->isEqual(reversed, true); });
|
||||
if (reverseIt != tubes.end())
|
||||
{
|
||||
m_modified_tubes.push_back(std::make_unique<CTube>(**reverseIt));
|
||||
getMap().DeleteTube((*reverseIt)->getId());
|
||||
}
|
||||
}
|
||||
m_modified_tubes.push_back(std::make_unique<CTube>(*m_tube));
|
||||
getMap().DeleteTube(m_tube->getId());
|
||||
}
|
||||
|
||||
if (!m_tube) {
|
||||
// create a new tube, first click
|
||||
m_tube = std::move(createNewTube(mapCoords3d));
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
int z = getMap().GetHeightAt(m_tube->getStartCoords());
|
||||
auto mc = getMap().ToMapCoords3d(projCoords, z);
|
||||
if (m_tube->getEndCoords() == MapCoords(mc.x, mc.y))
|
||||
{
|
||||
finish();
|
||||
}
|
||||
else
|
||||
{
|
||||
m_tube->append(mc.x, mc.y, m_tube->GetTubeParts().empty() ? 2 : 0);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
getView().RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);
|
||||
}
|
||||
|
||||
std::unique_ptr<CTube> AddTubeTool::createNewTube(const MapCoords& mapCoords3d) const
|
||||
{
|
||||
return std::make_unique<CTube>(mapCoords3d.x, mapCoords3d.y, ETubeDirection::Undefined, mapCoords3d.x, mapCoords3d.y, std::vector<ETubeDirection>());
|
||||
}
|
||||
|
||||
std::unique_ptr<CTube> AddTubeTool::getTubeToModify(const MapCoords& mapCoords3d, const ProjectedCoords& projCoords, MapToolMouseFlags flags) const
|
||||
{
|
||||
if ((flags & MapToolMouseFlags::SHIFT) == MapToolMouseFlags::SHIFT)
|
||||
return findTubeEndAt(mapCoords3d, projCoords, getMap());
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
void AddTubeTool::onMouseMove(const ProjectedCoords& projCoords, const MapCoords& mapCoords3d, MapToolMouseFlags flags)
|
||||
{
|
||||
if (m_tube) {
|
||||
m_mm_tube.reset(new CTube(*m_tube));
|
||||
int z = getMap().GetHeightAt(m_tube->getStartCoords());
|
||||
auto mc = getMap().ToMapCoords3d(projCoords, z);
|
||||
m_mm_tube->append(mc.x, mc.y, m_tube->GetTubeParts().empty() ? 2 : 0);
|
||||
getView().RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);
|
||||
}
|
||||
else
|
||||
{
|
||||
auto tube = getTubeToModify(mapCoords3d, projCoords, flags);
|
||||
if (tube)
|
||||
{
|
||||
m_hover_tube = std::move(tube);
|
||||
getView().RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);
|
||||
}
|
||||
else if (m_hover_tube)
|
||||
{
|
||||
m_hover_tube.reset();
|
||||
getView().RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void AddTubeTool::render()
|
||||
{
|
||||
if (m_mm_tube)
|
||||
{
|
||||
// mouse-move tube
|
||||
getView().DrawTube(*m_mm_tube);
|
||||
}
|
||||
else if (m_tube) {
|
||||
getView().DrawTube(*m_tube);
|
||||
}
|
||||
else if (m_hover_tube)
|
||||
{
|
||||
COLORREF col = RGB(0, 255, 0);
|
||||
getView().DrawTube(*m_hover_tube, nullptr, &col);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
RemoveTubeTool::RemoveTubeTool(CMapData& map, CIsoView& view) : MapTool(map, view)
|
||||
{
|
||||
}
|
||||
|
||||
void RemoveTubeTool::onMouseMove(const ProjectedCoords& projCoords, const MapCoords& mapCoords3d, MapToolMouseFlags flags)
|
||||
{
|
||||
auto& m = getMap();
|
||||
std::vector<std::uint16_t> tubes = findTubesAt(projCoords, getMap());
|
||||
|
||||
if ((flags & MapToolMouseFlags::LBUTTON) == MapToolMouseFlags::LBUTTON)
|
||||
{
|
||||
for (auto& id : tubes)
|
||||
{
|
||||
m.DeleteTube(id);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
m_hover_tubes.reserve(tubes.size());
|
||||
std::transform(tubes.begin(), tubes.end(), std::back_inserter(m_hover_tubes), [this](std::uint16_t tubeId) { return std::make_unique<CTube>(*getMap().GetTube(tubeId)); });
|
||||
}
|
||||
getView().RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);
|
||||
m_hover_tubes.clear();
|
||||
}
|
||||
|
||||
void RemoveTubeTool::render()
|
||||
{
|
||||
for(auto& tube: m_hover_tubes)
|
||||
{
|
||||
COLORREF col = RGB(0, 255, 0);
|
||||
getView().DrawTube(*tube, nullptr, &col);
|
||||
}
|
||||
}
|
||||
|
||||
ModifyTubeTool::ModifyTubeTool(CMapData& map, CIsoView& view, bool bidirectional): AddTubeTool(map, view, bidirectional)
|
||||
{
|
||||
}
|
||||
|
||||
std::unique_ptr<CTube> ModifyTubeTool::getTubeToModify(const MapCoords& mapCoords3d, const ProjectedCoords& projCoords, MapToolMouseFlags flags) const
|
||||
{
|
||||
return findTubeEndAt(mapCoords3d, projCoords, getMap());
|
||||
}
|
||||
|
||||
std::unique_ptr<CTube> ModifyTubeTool::createNewTube(const MapCoords& mapCoords3d) const
|
||||
{
|
||||
return nullptr;
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue