/***************************************************************************
 *   Copyright (c) 2022 Ajinkya Dahale <dahale.a.p@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                                *
 *                                                                         *
 ***************************************************************************/

#ifndef GUI_TASKVIEW_TaskFemConstraintRigidBody_H
#define GUI_TASKVIEW_TaskFemConstraintRigidBody_H

#include <QObject>

#include "TaskFemConstraintOnBoundary.h"
#include "ViewProviderFemConstraintRigidBody.h"


class Ui_TaskFemConstraintRigidBody;

namespace FemGui
{
class TaskFemConstraintRigidBody: public TaskFemConstraintOnBoundary
{
    Q_OBJECT

public:
    explicit TaskFemConstraintRigidBody(ViewProviderFemConstraintRigidBody* ConstraintView,
                                        QWidget* parent = nullptr);
    ~TaskFemConstraintRigidBody() override;

    const std::string getReferences() const override;
    Base::Vector3d getReferenceNode() const;
    Base::Vector3d getDisplacement() const;
    Base::Rotation getRotation() const;
    std::vector<std::string> getForce() const;
    std::vector<std::string> getMoment() const;
    std::vector<std::string> getTranslationalMode() const;
    std::vector<std::string> getRotationalMode() const;

private Q_SLOTS:
    void onReferenceDeleted();
    void addToSelection() override;
    void removeFromSelection() override;
    void onTransModeXChanged(int);
    void onTransModeYChanged(int);
    void onTransModeZChanged(int);
    void onRotModeXChanged(int);
    void onRotModeYChanged(int);
    void onRotModeZChanged(int);
    void onRefNodeXChanged(double);
    void onRefNodeYChanged(double);
    void onRefNodeZChanged(double);

protected:
    void changeEvent(QEvent* e) override;
    void clearButtons(const SelectionChangeModes notThis) override;

private:
    void updateUI();
    Ui_TaskFemConstraintRigidBody* ui;
};

class TaskDlgFemConstraintRigidBody: public TaskDlgFemConstraint
{
    Q_OBJECT

public:
    explicit TaskDlgFemConstraintRigidBody(ViewProviderFemConstraintRigidBody* ConstraintView);
    bool accept() override;
};

}  // namespace FemGui

#endif  // GUI_TASKVIEW_TaskFemConstraintRigidBody_H
