FreeCAD / src /Gui /Selection /SoFCSelectionContext.cpp

Upload folder using huggingface_hub

985c397 verified 10 months ago

9.67 kB

// 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;
        }
    }
}

	// 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;
	}
	}
	}