131 lines
5.5 KiB
C++
131 lines
5.5 KiB
C++
/******************************************************************************
|
|
* Copyright (c) 2012 Konstantinos Poulios <logari81@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_ReferenceSelection_H
|
|
#define GUI_ReferenceSelection_H
|
|
|
|
#include <Gui/Selection/SelectionFilter.h>
|
|
#include <Mod/PartDesign/Gui/EnumFlags.h>
|
|
|
|
namespace App
|
|
{
|
|
class OriginGroupExtension;
|
|
}
|
|
namespace PartDesign
|
|
{
|
|
class Body;
|
|
}
|
|
namespace PartDesignGui
|
|
{
|
|
|
|
class ReferenceSelection: public Gui::SelectionFilterGate
|
|
{
|
|
const App::DocumentObject* support;
|
|
AllowSelectionFlags type;
|
|
|
|
public:
|
|
ReferenceSelection(const App::DocumentObject* support_, AllowSelectionFlags type)
|
|
: Gui::SelectionFilterGate(nullPointer())
|
|
, support(support_)
|
|
, type(type)
|
|
{}
|
|
/**
|
|
* Allow the user to pick only edges or faces (or both) from the defined support
|
|
* Optionally restrict the selection to planar edges/faces
|
|
*/
|
|
bool allow(App::Document* pDoc, App::DocumentObject* pObj, const char* sSubName) override;
|
|
|
|
private:
|
|
PartDesign::Body* getBody() const;
|
|
App::OriginGroupExtension* getOriginGroupExtension(PartDesign::Body* body) const;
|
|
bool allowOrigin(
|
|
PartDesign::Body* body,
|
|
App::OriginGroupExtension* originGroup,
|
|
App::DocumentObject* pObj
|
|
) const;
|
|
bool allowDatum(PartDesign::Body* body, App::DocumentObject* pObj) const;
|
|
bool allowPartFeature(App::DocumentObject* pObj, const char* sSubName) const;
|
|
bool isEdge(App::DocumentObject* pObj, const char* sSubName) const;
|
|
bool isFace(App::DocumentObject* pObj, const char* sSubName) const;
|
|
bool isCircle(App::DocumentObject* pObj, const char* sSubName) const;
|
|
};
|
|
|
|
class NoDependentsSelection: public Gui::SelectionFilterGate
|
|
{
|
|
const App::DocumentObject* support;
|
|
|
|
public:
|
|
NoDependentsSelection(const App::DocumentObject* support_)
|
|
: Gui::SelectionFilterGate(nullPointer())
|
|
, support(support_)
|
|
{}
|
|
/**
|
|
* Allow the user to pick only objects which are not in objs getDependencyList
|
|
*/
|
|
bool allow(App::Document* pDoc, App::DocumentObject* pObj, const char* sSubName) override;
|
|
};
|
|
|
|
class CombineSelectionFilterGates: public Gui::SelectionFilterGate
|
|
{
|
|
std::unique_ptr<Gui::SelectionFilterGate> filter1;
|
|
std::unique_ptr<Gui::SelectionFilterGate> filter2;
|
|
|
|
public:
|
|
CombineSelectionFilterGates(
|
|
std::unique_ptr<Gui::SelectionFilterGate>& filter1_,
|
|
std::unique_ptr<Gui::SelectionFilterGate>& filter2_
|
|
)
|
|
: Gui::SelectionFilterGate(nullPointer())
|
|
, filter1(std::move(filter1_))
|
|
, filter2(std::move(filter2_))
|
|
{}
|
|
bool allow(App::Document* pDoc, App::DocumentObject* pObj, const char* sSubName) override;
|
|
};
|
|
// Convenience methods
|
|
/// Extract reference from Selection
|
|
bool getReferencedSelection(
|
|
const App::DocumentObject* thisObj,
|
|
const Gui::SelectionChanges& msg,
|
|
App::DocumentObject*& selObj,
|
|
std::vector<std::string>& selSub
|
|
);
|
|
/// Return reference as string for UI elements (format <obj>:<subelement>
|
|
QString getRefStr(const App::DocumentObject* obj, const std::vector<std::string>& sub);
|
|
/// Return reference as string for python in the format (<obj> ["sub1", "sub2", ...])
|
|
std::string buildLinkSubPythonStr(const App::DocumentObject* obj, const std::vector<std::string>& subs);
|
|
/// Return reference as string for python in the format (<obj> ["sub"?])
|
|
std::string buildLinkSingleSubPythonStr(
|
|
const App::DocumentObject* obj,
|
|
const std::vector<std::string>& subs
|
|
);
|
|
/// Return reference as string for python in the format [obj1, obj2, ...,]
|
|
std::string buildLinkListPythonStr(const std::vector<App::DocumentObject*>& objs);
|
|
/// Returns sub reference list as a python string in the format [(obj1,"sub1"),(obj2,"sub2"),...]
|
|
std::string buildLinkSubListPythonStr(
|
|
const std::vector<App::DocumentObject*>& objs,
|
|
const std::vector<std::string>& subs
|
|
);
|
|
} // namespace PartDesignGui
|
|
|
|
#endif // GUI_ReferenceSelection_H
|