342 lines
9.4 KiB
C++
342 lines
9.4 KiB
C++
// SPDX-License-Identifier: LGPL-2.1-or-later
|
|
/****************************************************************************
|
|
* Copyright (c) 2018 Zheng Lei (realthunder) <realthunder.dev@gmail.com> *
|
|
* *
|
|
* This file is part of the FreeCAD CAx development system. *
|
|
* *
|
|
* This library is free software; you can redistribute it and/or *
|
|
* modify it under the terms of the GNU Library General Public *
|
|
* License as published by the Free Software Foundation; either *
|
|
* version 2 of the License, or (at your option) any later version. *
|
|
* *
|
|
* This library 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 Library General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU Library General Public *
|
|
* License along with this library; see the file COPYING.LIB. If not, *
|
|
* write to the Free Software Foundation, Inc., 59 Temple Place, *
|
|
* Suite 330, Boston, MA 02111-1307, USA *
|
|
* *
|
|
****************************************************************************/
|
|
|
|
|
|
#include <boost/algorithm/string/predicate.hpp>
|
|
#include <Inventor/elements/SoCacheElement.h>
|
|
#include <Inventor/misc/SoState.h>
|
|
|
|
|
|
#include "SoFCSelectionContext.h"
|
|
#include "SoFCUnifiedSelection.h"
|
|
|
|
|
|
using namespace Gui;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
|
|
SoFCSelectionContext::~SoFCSelectionContext()
|
|
{
|
|
if (counter) {
|
|
*counter -= 1;
|
|
}
|
|
}
|
|
|
|
bool SoFCSelectionContext::checkGlobal(SoFCSelectionContextPtr ctx)
|
|
{
|
|
bool sel = false;
|
|
bool hl = false;
|
|
SoFCSelectionRoot::checkSelection(sel, selectionColor, hl, highlightColor);
|
|
if (sel) {
|
|
selectionIndex.insert(-1);
|
|
}
|
|
else if (ctx && hl) {
|
|
selectionColor = ctx->selectionColor;
|
|
selectionIndex = ctx->selectionIndex;
|
|
}
|
|
else {
|
|
selectionIndex.clear();
|
|
}
|
|
if (hl) {
|
|
highlightAll();
|
|
}
|
|
else if (ctx && sel) {
|
|
highlightIndex = ctx->highlightIndex;
|
|
highlightColor = ctx->highlightColor;
|
|
}
|
|
else {
|
|
removeHighlight();
|
|
}
|
|
return sel || hl;
|
|
}
|
|
|
|
bool SoFCSelectionContext::removeIndex(int index)
|
|
{
|
|
auto it = selectionIndex.find(index);
|
|
if (it != selectionIndex.end()) {
|
|
selectionIndex.erase(it);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
int SoFCSelectionContext::merge(
|
|
int status,
|
|
SoFCSelectionContextBasePtr& output,
|
|
SoFCSelectionContextBasePtr input,
|
|
SoNode*
|
|
)
|
|
{
|
|
auto ctx = std::dynamic_pointer_cast<SoFCSelectionContext>(input);
|
|
if (!ctx) {
|
|
return status;
|
|
}
|
|
|
|
if (ctx->selectionIndex.empty()) {
|
|
output = ctx;
|
|
return -1;
|
|
}
|
|
|
|
auto ret = std::dynamic_pointer_cast<SoFCSelectionContext>(output);
|
|
if (!ret) {
|
|
output = ctx;
|
|
return 0;
|
|
}
|
|
|
|
if (ctx->isSelectAll()) {
|
|
return status;
|
|
}
|
|
|
|
if (ret->isSelectAll()) {
|
|
if (!status) {
|
|
output = ret->copy();
|
|
ret = std::dynamic_pointer_cast<SoFCSelectionContext>(ret);
|
|
assert(ret);
|
|
}
|
|
ret->selectionIndex = ctx->selectionIndex;
|
|
return status;
|
|
}
|
|
|
|
std::vector<int> remove;
|
|
for (auto idx : ret->selectionIndex) {
|
|
if (!ctx->selectionIndex.contains(idx)) {
|
|
remove.push_back(idx);
|
|
}
|
|
}
|
|
|
|
for (auto idx : remove) {
|
|
if (!status) {
|
|
status = 1;
|
|
output = ret->copy();
|
|
ret = std::dynamic_pointer_cast<SoFCSelectionContext>(ret);
|
|
assert(ret);
|
|
}
|
|
ret->selectionIndex.erase(idx);
|
|
if (ret->selectionIndex.empty()) {
|
|
return -1;
|
|
}
|
|
}
|
|
return status;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
bool SoFCSelectionContextEx::setColors(
|
|
const std::map<std::string, Base::Color>& colors,
|
|
const std::string& element
|
|
)
|
|
{
|
|
std::map<int, Base::Color> tmp;
|
|
auto it = colors.find("");
|
|
if (it != colors.end()) {
|
|
tmp[-1] = it->second;
|
|
}
|
|
for (auto it = colors.lower_bound(element); it != colors.end(); ++it) {
|
|
if (!boost::starts_with(it->first, element)) {
|
|
break;
|
|
}
|
|
if (it->first.size() == element.size()) {
|
|
tmp[-1] = it->second;
|
|
}
|
|
else {
|
|
int idx = std::atoi(it->first.c_str() + 4);
|
|
if (idx > 0) {
|
|
idx -= 1;
|
|
tmp[idx] = it->second;
|
|
}
|
|
}
|
|
}
|
|
if (tmp == this->colors) {
|
|
return false;
|
|
}
|
|
this->colors.swap(tmp);
|
|
return true;
|
|
}
|
|
|
|
uint32_t SoFCSelectionContextEx::packColor(const Base::Color& c, bool& hasTransparency)
|
|
{
|
|
// Convert Base::Color's alpha (opacity) to transparency for Coin3D.
|
|
float transparency = 1.0f - c.a;
|
|
|
|
// Apply any external transparency override (e.g., from picking).
|
|
float final_transparency = std::max(trans0, transparency);
|
|
|
|
if (final_transparency > 0.0f) {
|
|
hasTransparency = true;
|
|
}
|
|
|
|
return SbColor(c.r, c.g, c.b).getPackedValue(final_transparency);
|
|
}
|
|
|
|
bool SoFCSelectionContextEx::applyColor(int idx, std::vector<uint32_t>& packedColors, bool& hasTransparency)
|
|
{
|
|
if (colors.empty()) {
|
|
return false;
|
|
}
|
|
auto it = colors.find(idx);
|
|
if (it == colors.end()) {
|
|
if (colors.begin()->first >= 0) {
|
|
return false;
|
|
}
|
|
it = colors.begin();
|
|
}
|
|
packedColors.push_back(packColor(it->second, hasTransparency));
|
|
return true;
|
|
}
|
|
|
|
bool SoFCSelectionContextEx::isSingleColor(uint32_t& color, bool& hasTransparency)
|
|
{
|
|
if (!colors.empty() && colors.begin()->first < 0) {
|
|
color = packColor(colors.begin()->second, hasTransparency);
|
|
return colors.size() == 1;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
int SoFCSelectionContextEx::merge(
|
|
int status,
|
|
SoFCSelectionContextBasePtr& output,
|
|
SoFCSelectionContextBasePtr input,
|
|
SoNode* node
|
|
)
|
|
{
|
|
auto ctx = std::dynamic_pointer_cast<SoFCSelectionContextEx>(input);
|
|
SoFCSelectionRoot* selectionNode = dynamic_cast<SoFCSelectionRoot*>(node);
|
|
|
|
if (!ctx) {
|
|
if (selectionNode && selectionNode->hasColorOverride()) {
|
|
if (!status) {
|
|
status = 2;
|
|
}
|
|
else if (status == 1) {
|
|
status = 3;
|
|
}
|
|
}
|
|
return status;
|
|
}
|
|
|
|
int status_copy = status;
|
|
if (status == 2) {
|
|
status_copy = 0;
|
|
}
|
|
else if (status == 3) {
|
|
status_copy = 1;
|
|
}
|
|
status_copy = SoFCSelectionContext::merge(status_copy, output, input, node);
|
|
if (status_copy < 0) {
|
|
return status_copy;
|
|
}
|
|
|
|
if (status > 1) {
|
|
// When status>1 it means there is color override before us, all
|
|
// subsequent color override will be bypassed
|
|
if (status_copy == 1) {
|
|
status = 3;
|
|
}
|
|
else {
|
|
status = 2;
|
|
}
|
|
return status;
|
|
}
|
|
|
|
status = status_copy;
|
|
auto ret = std::dynamic_pointer_cast<SoFCSelectionContextEx>(output);
|
|
assert(ret);
|
|
for (auto& v : ctx->colors) {
|
|
if (ret->colors.contains(v.first)) {
|
|
continue;
|
|
}
|
|
if (!status) {
|
|
status = 1;
|
|
output = ret->copy();
|
|
ret = std::dynamic_pointer_cast<SoFCSelectionContextEx>(output);
|
|
assert(ret);
|
|
}
|
|
ret->colors.insert(v);
|
|
}
|
|
|
|
if (selectionNode && selectionNode->hasColorOverride()) {
|
|
if (!status) {
|
|
status = 2;
|
|
}
|
|
else if (status == 1) {
|
|
status = 3;
|
|
}
|
|
}
|
|
return status;
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////
|
|
|
|
SoFCSelectionCounter::SoFCSelectionCounter()
|
|
: counter(std::make_shared<int>(0))
|
|
{}
|
|
|
|
SoFCSelectionCounter::~SoFCSelectionCounter() = default;
|
|
|
|
|
|
bool SoFCSelectionCounter::checkRenderCache(SoState* state)
|
|
{
|
|
if (*counter || (hasSelection && Selection().hasSelection())
|
|
|| (hasPreselection && Selection().hasPreselection())) {
|
|
if (SoFCSelectionRoot::getCacheMode() != SoSeparator::OFF) {
|
|
SoCacheElement::invalidate(state);
|
|
}
|
|
return false;
|
|
}
|
|
if (!Selection().hasPreselection()) {
|
|
hasPreselection = false;
|
|
}
|
|
if (!Selection().hasSelection()) {
|
|
hasSelection = false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void SoFCSelectionCounter::checkAction(SoHighlightElementAction* hlaction)
|
|
{
|
|
if (hlaction->isHighlighted()) {
|
|
hasPreselection = true;
|
|
}
|
|
}
|
|
|
|
void SoFCSelectionCounter::checkAction(SoSelectionElementAction* selaction, SoFCSelectionContextPtr ctx)
|
|
{
|
|
switch (selaction->getType()) {
|
|
case SoSelectionElementAction::None:
|
|
return;
|
|
case SoSelectionElementAction::All:
|
|
case SoSelectionElementAction::Append:
|
|
hasSelection = true;
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
if (selaction->isSecondary()) {
|
|
if (ctx && !ctx->counter) {
|
|
*counter += 1;
|
|
ctx->counter = counter;
|
|
}
|
|
}
|
|
}
|