FreeCAD / src /Mod /PartDesign /Gui /TaskDraftParameters.cpp

Upload folder using huggingface_hub

985c397 verified 9 months ago

12.6 kB

/***************************************************************************
 *   Copyright (c) 2012 Jan Rheinländer                                    *
 *                                   <jrheinlaender@users.sourceforge.net> *
 *                                                                         *
 *   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 <QAction>
#include <QKeyEvent>
#include <QListWidget>
#include <QMessageBox>


#include <Base/Interpreter.h>
#include <App/Document.h>
#include <App/DocumentObject.h>
#include <Gui/Command.h>
#include <Gui/Selection/Selection.h>
#include <Gui/ViewProvider.h>
#include <Gui/Inventor/Draggers/Gizmo.h>
#include <Mod/PartDesign/App/FeatureDraft.h>
#include <Mod/PartDesign/Gui/ReferenceSelection.h>
#include <Mod/Part/App/GizmoHelper.h>

#include "ui_TaskDraftParameters.h"
#include "TaskDraftParameters.h"

using namespace PartDesignGui;
using namespace Gui;

/* TRANSLATOR PartDesignGui::TaskDraftParameters */

TaskDraftParameters::TaskDraftParameters(ViewProviderDressUp* DressUpView, QWidget* parent)
    : TaskDressUpParameters(DressUpView, false, true, parent)
    , ui(new Ui_TaskDraftParameters)
{
    // we need a separate container widget to add all controls to
    proxy = new QWidget(this);
    ui->setupUi(proxy);

    this->groupLayout()->addWidget(proxy);

    PartDesign::Draft* pcDraft = DressUpView->getObject<PartDesign::Draft>();
    double a = pcDraft->Angle.getValue();

    ui->draftAngle->setMinimum(pcDraft->Angle.getMinimum());
    ui->draftAngle->setMaximum(pcDraft->Angle.getMaximum());
    ui->draftAngle->setValue(a);
    ui->draftAngle->selectAll();
    QMetaObject::invokeMethod(ui->draftAngle, "setFocus", Qt::QueuedConnection);

    // Bind input fields to properties
    ui->draftAngle->bind(pcDraft->Angle);

    bool r = pcDraft->Reversed.getValue();
    ui->checkReverse->setChecked(r);

    std::vector<std::string> strings = pcDraft->Base.getSubValues();
    for (const auto& string : strings) {
        ui->listWidgetReferences->addItem(QString::fromStdString(string));
    }

    QMetaObject::connectSlotsByName(this);

    // clang-format off
    connect(ui->draftAngle, qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
            this, &TaskDraftParameters::onAngleChanged);
    connect(ui->checkReverse, &QCheckBox::toggled,
            this, &TaskDraftParameters::onReversedChanged);
    connect(ui->buttonRefSel, &QToolButton::toggled,
            this, &TaskDraftParameters::onButtonRefSel);
    connect(ui->buttonPlane, &QToolButton::toggled,
            this, &TaskDraftParameters::onButtonPlane);
    connect(ui->buttonLine, &QToolButton::toggled,
            this, &TaskDraftParameters::onButtonLine);

    // Create context menu
    createDeleteAction(ui->listWidgetReferences);
    connect(deleteAction, &QAction::triggered, this, &TaskDraftParameters::onRefDeleted);

    connect(ui->listWidgetReferences, &QListWidget::currentItemChanged,
            this, &TaskDraftParameters::setSelection);
    connect(ui->listWidgetReferences, &QListWidget::itemClicked,
            this, &TaskDraftParameters::setSelection);
    connect(ui->listWidgetReferences, &QListWidget::itemDoubleClicked,
            this, &TaskDraftParameters::doubleClicked);
    // clang-format on

    App::DocumentObject* ref = pcDraft->NeutralPlane.getValue();
    strings = pcDraft->NeutralPlane.getSubValues();
    ui->linePlane->setText(getRefStr(ref, strings));

    ref = pcDraft->PullDirection.getValue();
    strings = pcDraft->PullDirection.getSubValues();
    ui->lineLine->setText(getRefStr(ref, strings));

    if (strings.size() == 0) {
        setSelectionMode(refSel);
    }
    else {
        hideOnError();
    }
}

void TaskDraftParameters::onSelectionChanged(const Gui::SelectionChanges& msg)
{
    // executed when the user selected something in the CAD object
    // adds/deletes the selection accordingly

    if (msg.Type == Gui::SelectionChanges::AddSelection) {
        if (selectionMode == refSel) {
            referenceSelected(msg, ui->listWidgetReferences);
        }
        else if (selectionMode == plane) {
            auto pcDraft = getObject<PartDesign::Draft>();
            std::vector<std::string> planes;
            App::DocumentObject* selObj {};
            getReferencedSelection(pcDraft, msg, selObj, planes);
            if (!selObj) {
                return;
            }
            setupTransaction();
            pcDraft->NeutralPlane.setValue(selObj, planes);
            ui->linePlane->setText(getRefStr(selObj, planes));

            pcDraft->getDocument()->recomputeFeature(pcDraft);
            // highlight existing references for possible further selections
            getDressUpView()->highlightReferences(true);
            // hide the draft if there was a computation error
            hideOnError();
        }
        else if (selectionMode == line) {
            auto pcDraft = getObject<PartDesign::Draft>();
            std::vector<std::string> edges;
            App::DocumentObject* selObj;
            getReferencedSelection(pcDraft, msg, selObj, edges);
            if (!selObj) {
                return;
            }
            setupTransaction();
            pcDraft->PullDirection.setValue(selObj, edges);
            ui->lineLine->setText(getRefStr(selObj, edges));

            pcDraft->getDocument()->recomputeFeature(pcDraft);
            // highlight existing references for possible further selections
            getDressUpView()->highlightReferences(true);
            // hide the draft if there was a computation error
            hideOnError();
        }
    }
}

void TaskDraftParameters::setButtons(const selectionModes mode)
{
    ui->buttonRefSel->setText(mode == refSel ? stopSelectionLabel() : startSelectionLabel());
    ui->buttonRefSel->setChecked(mode == refSel);
    ui->buttonLine->setChecked(mode == line);
    ui->buttonPlane->setChecked(mode == plane);
}

void TaskDraftParameters::onButtonPlane(bool checked)
{
    if (checked) {
        setButtons(plane);
        getViewObject()->showPreviousFeature(true);
        selectionMode = plane;
        Gui::Selection().clearSelection();
        Gui::Selection().addSelectionGate(new ReferenceSelection(
            this->getBase(),
            AllowSelection::EDGE | AllowSelection::FACE | AllowSelection::PLANAR
        ));
    }
}

void TaskDraftParameters::onButtonLine(bool checked)
{
    if (checked) {
        setButtons(line);
        getViewObject()->showPreviousFeature(true);
        selectionMode = line;
        Gui::Selection().clearSelection();
        Gui::Selection().addSelectionGate(
            new ReferenceSelection(this->getBase(), AllowSelection::EDGE | AllowSelection::PLANAR)
        );
    }
}

void TaskDraftParameters::onRefDeleted()
{
    TaskDressUpParameters::deleteRef(ui->listWidgetReferences);
}

void TaskDraftParameters::getPlane(App::DocumentObject*& obj, std::vector<std::string>& sub) const
{
    sub = std::vector<std::string>(1, "");
    QStringList parts = ui->linePlane->text().split(QChar::fromLatin1(':'));
    obj = getObject()->getDocument()->getObject(parts[0].toStdString().c_str());
    if (parts.size() > 1) {
        sub[0] = parts[1].toStdString();
    }
}

void TaskDraftParameters::getLine(App::DocumentObject*& obj, std::vector<std::string>& sub) const
{
    sub = std::vector<std::string>(1, "");
    QStringList parts = ui->lineLine->text().split(QChar::fromLatin1(':'));
    obj = getObject()->getDocument()->getObject(parts[0].toStdString().c_str());
    if (parts.size() > 1) {
        sub[0] = parts[1].toStdString();
    }
}

void TaskDraftParameters::onAngleChanged(double angle)
{
    if (auto draft = getObject<PartDesign::Draft>()) {
        setButtons(none);
        setupTransaction();
        draft->Angle.setValue(angle);
        draft->recomputeFeature();
        // hide the draft if there was a computation error
        hideOnError();
    }
}

double TaskDraftParameters::getAngle() const
{
    return ui->draftAngle->value().getValue();
}

void TaskDraftParameters::onReversedChanged(const bool on)
{
    if (auto draft = getObject<PartDesign::Draft>()) {
        setButtons(none);
        setupTransaction();
        draft->Reversed.setValue(on);
        draft->recomputeFeature();
        // hide the draft if there was a computation error
        hideOnError();
    }
}

bool TaskDraftParameters::getReversed() const
{
    return ui->checkReverse->isChecked();
}

TaskDraftParameters::~TaskDraftParameters()
{
    try {
        Gui::Selection().clearSelection();
        Gui::Selection().rmvSelectionGate();
    }
    catch (const Py::Exception&) {
        Base::PyException e;  // extract the Python error text
        e.reportException();
    }
}

void TaskDraftParameters::changeEvent(QEvent* e)
{
    TaskBox::changeEvent(e);
    if (e->type() == QEvent::LanguageChange) {
        ui->retranslateUi(proxy);
    }
}

void TaskDraftParameters::apply()
{
    // Alert user if he created an empty feature
    if (ui->listWidgetReferences->count() == 0) {
        Base::Console().warning(tr("Empty draft created!\n").toStdString().c_str());
    }

    TaskDressUpParameters::apply();
}

//**************************************************************************
//**************************************************************************
// TaskDialog
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

TaskDlgDraftParameters::TaskDlgDraftParameters(ViewProviderDraft* DressUpView)
    : TaskDlgDressUpParameters(DressUpView)
{
    parameter = new TaskDraftParameters(DressUpView);

    Content.push_back(parameter);
    Content.push_back(preview);
}

TaskDlgDraftParameters::~TaskDlgDraftParameters() = default;

//==== calls from the TaskView ===============================================================

bool TaskDlgDraftParameters::accept()
{
    auto tobj = getObject();
    if (!tobj->isError()) {
        getViewObject()->showPreviousFeature(false);
    }

    parameter->apply();

    std::vector<std::string> strings;
    App::DocumentObject* obj;
    TaskDraftParameters* draftparameter = static_cast<TaskDraftParameters*>(parameter);

    draftparameter->getPlane(obj, strings);
    std::string neutralPlane = buildLinkSingleSubPythonStr(obj, strings);

    draftparameter->getLine(obj, strings);
    std::string pullDirection = buildLinkSingleSubPythonStr(obj, strings);

    FCMD_OBJ_CMD(tobj, "Angle = " << draftparameter->getAngle());
    FCMD_OBJ_CMD(tobj, "Reversed = " << draftparameter->getReversed());
    if (neutralPlane.empty()) {
        neutralPlane = "None";
    }
    FCMD_OBJ_CMD(tobj, "NeutralPlane = " << neutralPlane);
    if (pullDirection.empty()) {
        pullDirection = "None";
    }
    FCMD_OBJ_CMD(tobj, "PullDirection = " << pullDirection);

    return TaskDlgDressUpParameters::accept();
}

#include "moc_TaskDraftParameters.cpp"

	/***************************************************************************
	* Copyright (c) 2012 Jan Rheinländer *
	* <jrheinlaender@users.sourceforge.net> *
	* *
	* 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 <QAction>
	#include <QKeyEvent>
	#include <QListWidget>
	#include <QMessageBox>


	#include <Base/Interpreter.h>
	#include <App/Document.h>
	#include <App/DocumentObject.h>
	#include <Gui/Command.h>
	#include <Gui/Selection/Selection.h>
	#include <Gui/ViewProvider.h>
	#include <Gui/Inventor/Draggers/Gizmo.h>
	#include <Mod/PartDesign/App/FeatureDraft.h>
	#include <Mod/PartDesign/Gui/ReferenceSelection.h>
	#include <Mod/Part/App/GizmoHelper.h>

	#include "ui_TaskDraftParameters.h"
	#include "TaskDraftParameters.h"

	using namespace PartDesignGui;
	using namespace Gui;

	/* TRANSLATOR PartDesignGui::TaskDraftParameters */

	TaskDraftParameters::TaskDraftParameters(ViewProviderDressUp* DressUpView, QWidget* parent)
	: TaskDressUpParameters(DressUpView, false, true, parent)
	, ui(new Ui_TaskDraftParameters)
	{
	// we need a separate container widget to add all controls to
	proxy = new QWidget(this);
	ui->setupUi(proxy);

	this->groupLayout()->addWidget(proxy);

	PartDesign::Draft* pcDraft = DressUpView->getObject<PartDesign::Draft>();
	double a = pcDraft->Angle.getValue();

	ui->draftAngle->setMinimum(pcDraft->Angle.getMinimum());
	ui->draftAngle->setMaximum(pcDraft->Angle.getMaximum());
	ui->draftAngle->setValue(a);
	ui->draftAngle->selectAll();
	QMetaObject::invokeMethod(ui->draftAngle, "setFocus", Qt::QueuedConnection);

	// Bind input fields to properties
	ui->draftAngle->bind(pcDraft->Angle);

	bool r = pcDraft->Reversed.getValue();
	ui->checkReverse->setChecked(r);

	std::vector<std::string> strings = pcDraft->Base.getSubValues();
	for (const auto& string : strings) {
	ui->listWidgetReferences->addItem(QString::fromStdString(string));
	}

	QMetaObject::connectSlotsByName(this);

	// clang-format off
	connect(ui->draftAngle, qOverload<double>(&Gui::QuantitySpinBox::valueChanged),
	this, &TaskDraftParameters::onAngleChanged);
	connect(ui->checkReverse, &QCheckBox::toggled,
	this, &TaskDraftParameters::onReversedChanged);
	connect(ui->buttonRefSel, &QToolButton::toggled,
	this, &TaskDraftParameters::onButtonRefSel);
	connect(ui->buttonPlane, &QToolButton::toggled,
	this, &TaskDraftParameters::onButtonPlane);
	connect(ui->buttonLine, &QToolButton::toggled,
	this, &TaskDraftParameters::onButtonLine);

	// Create context menu
	createDeleteAction(ui->listWidgetReferences);
	connect(deleteAction, &QAction::triggered, this, &TaskDraftParameters::onRefDeleted);

	connect(ui->listWidgetReferences, &QListWidget::currentItemChanged,
	this, &TaskDraftParameters::setSelection);
	connect(ui->listWidgetReferences, &QListWidget::itemClicked,
	this, &TaskDraftParameters::setSelection);
	connect(ui->listWidgetReferences, &QListWidget::itemDoubleClicked,
	this, &TaskDraftParameters::doubleClicked);
	// clang-format on

	App::DocumentObject* ref = pcDraft->NeutralPlane.getValue();
	strings = pcDraft->NeutralPlane.getSubValues();
	ui->linePlane->setText(getRefStr(ref, strings));

	ref = pcDraft->PullDirection.getValue();
	strings = pcDraft->PullDirection.getSubValues();
	ui->lineLine->setText(getRefStr(ref, strings));

	if (strings.size() == 0) {
	setSelectionMode(refSel);
	}
	else {
	hideOnError();
	}
	}

	void TaskDraftParameters::onSelectionChanged(const Gui::SelectionChanges& msg)
	{
	// executed when the user selected something in the CAD object
	// adds/deletes the selection accordingly

	if (msg.Type == Gui::SelectionChanges::AddSelection) {
	if (selectionMode == refSel) {
	referenceSelected(msg, ui->listWidgetReferences);
	}
	else if (selectionMode == plane) {
	auto pcDraft = getObject<PartDesign::Draft>();
	std::vector<std::string> planes;
	App::DocumentObject* selObj {};
	getReferencedSelection(pcDraft, msg, selObj, planes);
	if (!selObj) {
	return;
	}
	setupTransaction();
	pcDraft->NeutralPlane.setValue(selObj, planes);
	ui->linePlane->setText(getRefStr(selObj, planes));

	pcDraft->getDocument()->recomputeFeature(pcDraft);
	// highlight existing references for possible further selections
	getDressUpView()->highlightReferences(true);
	// hide the draft if there was a computation error
	hideOnError();
	}
	else if (selectionMode == line) {
	auto pcDraft = getObject<PartDesign::Draft>();
	std::vector<std::string> edges;
	App::DocumentObject* selObj;
	getReferencedSelection(pcDraft, msg, selObj, edges);
	if (!selObj) {
	return;
	}
	setupTransaction();
	pcDraft->PullDirection.setValue(selObj, edges);
	ui->lineLine->setText(getRefStr(selObj, edges));

	pcDraft->getDocument()->recomputeFeature(pcDraft);
	// highlight existing references for possible further selections
	getDressUpView()->highlightReferences(true);
	// hide the draft if there was a computation error
	hideOnError();
	}
	}
	}

	void TaskDraftParameters::setButtons(const selectionModes mode)
	{
	ui->buttonRefSel->setText(mode == refSel ? stopSelectionLabel() : startSelectionLabel());
	ui->buttonRefSel->setChecked(mode == refSel);
	ui->buttonLine->setChecked(mode == line);
	ui->buttonPlane->setChecked(mode == plane);
	}

	void TaskDraftParameters::onButtonPlane(bool checked)
	{
	if (checked) {
	setButtons(plane);
	getViewObject()->showPreviousFeature(true);
	selectionMode = plane;
	Gui::Selection().clearSelection();
	Gui::Selection().addSelectionGate(new ReferenceSelection(
	this->getBase(),
	AllowSelection::EDGE \| AllowSelection::FACE \| AllowSelection::PLANAR
	));
	}
	}

	void TaskDraftParameters::onButtonLine(bool checked)
	{
	if (checked) {
	setButtons(line);
	getViewObject()->showPreviousFeature(true);
	selectionMode = line;
	Gui::Selection().clearSelection();
	Gui::Selection().addSelectionGate(
	new ReferenceSelection(this->getBase(), AllowSelection::EDGE \| AllowSelection::PLANAR)
	);
	}
	}

	void TaskDraftParameters::onRefDeleted()
	{
	TaskDressUpParameters::deleteRef(ui->listWidgetReferences);
	}

	void TaskDraftParameters::getPlane(App::DocumentObject*& obj, std::vector<std::string>& sub) const
	{
	sub = std::vector<std::string>(1, "");
	QStringList parts = ui->linePlane->text().split(QChar::fromLatin1(':'));
	obj = getObject()->getDocument()->getObject(parts[0].toStdString().c_str());
	if (parts.size() > 1) {
	sub[0] = parts[1].toStdString();
	}
	}

	void TaskDraftParameters::getLine(App::DocumentObject*& obj, std::vector<std::string>& sub) const
	{
	sub = std::vector<std::string>(1, "");
	QStringList parts = ui->lineLine->text().split(QChar::fromLatin1(':'));
	obj = getObject()->getDocument()->getObject(parts[0].toStdString().c_str());
	if (parts.size() > 1) {
	sub[0] = parts[1].toStdString();
	}
	}

	void TaskDraftParameters::onAngleChanged(double angle)
	{
	if (auto draft = getObject<PartDesign::Draft>()) {
	setButtons(none);
	setupTransaction();
	draft->Angle.setValue(angle);
	draft->recomputeFeature();
	// hide the draft if there was a computation error
	hideOnError();
	}
	}

	double TaskDraftParameters::getAngle() const
	{
	return ui->draftAngle->value().getValue();
	}

	void TaskDraftParameters::onReversedChanged(const bool on)
	{
	if (auto draft = getObject<PartDesign::Draft>()) {
	setButtons(none);
	setupTransaction();
	draft->Reversed.setValue(on);
	draft->recomputeFeature();
	// hide the draft if there was a computation error
	hideOnError();
	}
	}

	bool TaskDraftParameters::getReversed() const
	{
	return ui->checkReverse->isChecked();
	}

	TaskDraftParameters::~TaskDraftParameters()
	{
	try {
	Gui::Selection().clearSelection();
	Gui::Selection().rmvSelectionGate();
	}
	catch (const Py::Exception&) {
	Base::PyException e; // extract the Python error text
	e.reportException();
	}
	}

	void TaskDraftParameters::changeEvent(QEvent* e)
	{
	TaskBox::changeEvent(e);
	if (e->type() == QEvent::LanguageChange) {
	ui->retranslateUi(proxy);
	}
	}

	void TaskDraftParameters::apply()
	{
	// Alert user if he created an empty feature
	if (ui->listWidgetReferences->count() == 0) {
	Base::Console().warning(tr("Empty draft created!\n").toStdString().c_str());
	}

	TaskDressUpParameters::apply();
	}

	//**************************************************************************
	//**************************************************************************
	// TaskDialog
	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

	TaskDlgDraftParameters::TaskDlgDraftParameters(ViewProviderDraft* DressUpView)
	: TaskDlgDressUpParameters(DressUpView)
	{
	parameter = new TaskDraftParameters(DressUpView);

	Content.push_back(parameter);
	Content.push_back(preview);
	}

	TaskDlgDraftParameters::~TaskDlgDraftParameters() = default;

	//==== calls from the TaskView ===============================================================

	bool TaskDlgDraftParameters::accept()
	{
	auto tobj = getObject();
	if (!tobj->isError()) {
	getViewObject()->showPreviousFeature(false);
	}

	parameter->apply();

	std::vector<std::string> strings;
	App::DocumentObject* obj;
	TaskDraftParameters* draftparameter = static_cast<TaskDraftParameters*>(parameter);

	draftparameter->getPlane(obj, strings);
	std::string neutralPlane = buildLinkSingleSubPythonStr(obj, strings);

	draftparameter->getLine(obj, strings);
	std::string pullDirection = buildLinkSingleSubPythonStr(obj, strings);

	FCMD_OBJ_CMD(tobj, "Angle = " << draftparameter->getAngle());
	FCMD_OBJ_CMD(tobj, "Reversed = " << draftparameter->getReversed());
	if (neutralPlane.empty()) {
	neutralPlane = "None";
	}
	FCMD_OBJ_CMD(tobj, "NeutralPlane = " << neutralPlane);
	if (pullDirection.empty()) {
	pullDirection = "None";
	}
	FCMD_OBJ_CMD(tobj, "PullDirection = " << pullDirection);

	return TaskDlgDressUpParameters::accept();
	}

	#include "moc_TaskDraftParameters.cpp"