kindred/create
1
1
Fork 0
Code Issues 37 Pull Requests Actions Packages Projects 9 Releases 2 Wiki Activity

Part: support of vertexes in ReferenceHighlighter

Browse Source
This commit is contained in:
wmayer
2022-01-18 19:47:06 +01:00
parent af868ed079
commit 74fe1fa4db
2 changed files with 102 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
src/Mod/Part/Gui/ReferenceHighlighter.cpp
View File

@@ -39,11 +39,101 @@ ReferenceHighlighter::ReferenceHighlighter(const TopoDS_Shape& shape, const App:
, elementColor(1.0f,0.0f,1.0f) // magenta
, objectColor(0.6f,0.0f,1.0f) // purple
{
TopExp::MapShapes(shape, TopAbs_VERTEX, vMap);
TopExp::MapShapes(shape, TopAbs_EDGE, eMap);
TopExp::MapShapes(shape, TopAbs_WIRE, wMap);
TopExp::MapShapes(shape, TopAbs_FACE, fMap);
}
void ReferenceHighlighter::getVertexColor(const std::string& element, std::vector<App::Color>& colors) const
{
int idx = std::stoi(element.substr(6)) - 1;
assert ( idx >= 0 );
std::size_t pos = std::size_t(idx);
if (pos < colors.size())
colors[pos] = elementColor;
}
void ReferenceHighlighter::getVertexColorsOfEdge(const std::string& element, std::vector<App::Color>& colors) const
{
int idx = std::stoi(element.substr(4));
assert ( idx > 0 );
// get the vertexes of the edge
TopoDS_Shape edge = eMap.FindKey(idx);
for (TopExp_Explorer xp(edge, TopAbs_VERTEX); xp.More(); xp.Next()) {
int vertexIndex = vMap.FindIndex(xp.Current());
// Vertex found?
if (vertexIndex > 0) {
std::size_t pos = std::size_t(vertexIndex - 1);
if (pos < colors.size())
colors[pos] = elementColor;
}
}
}
void ReferenceHighlighter::getVertexColorsOfWire(const std::string& element, std::vector<App::Color>& colors) const
{
int idx = std::stoi(element.substr(4));
assert ( idx > 0 );
// get the vertexes of the wire
TopoDS_Shape wire = wMap.FindKey(idx);
for (TopExp_Explorer xp(wire, TopAbs_VERTEX); xp.More(); xp.Next()) {
int vertexIndex = vMap.FindIndex(xp.Current());
// Vertex found?
if (vertexIndex > 0) {
std::size_t pos = std::size_t(vertexIndex - 1);
if (pos < colors.size())
colors[pos] = elementColor;
}
}
}
void ReferenceHighlighter::getVertexColorsOfFace(const std::string& element, std::vector<App::Color>& colors) const
{
int idx = std::stoi(element.substr(4));
assert ( idx > 0 );
// get the vertexes of the face
TopoDS_Shape face = fMap.FindKey(idx);
for (TopExp_Explorer xp(face, TopAbs_VERTEX); xp.More(); xp.Next()) {
int vertexIndex = vMap.FindIndex(xp.Current());
// Vertex found?
if (vertexIndex > 0) {
std::size_t pos = std::size_t(vertexIndex - 1);
if (pos < colors.size())
colors[pos] = elementColor;
}
}
}
void ReferenceHighlighter::getVertexColors(const std::vector<std::string>& elements,
std::vector<App::Color>& colors) const
{
colors.resize(vMap.Extent(), defaultColor);
if (!elements.empty()) {
for (std::string e : elements) {
if (boost::starts_with(e, "Vertex")) {
getVertexColor(e, colors);
}
else if (boost::starts_with(e, "Edge")) {
getVertexColorsOfEdge(e, colors);
}
else if (boost::starts_with(e, "Wire")) {
getVertexColorsOfWire(e, colors);
}
else if (boost::starts_with(e, "Face")) {
getVertexColorsOfFace(e, colors);
}
}
}
else {
std::fill(colors.begin(), colors.end(), objectColor);
}
}
void ReferenceHighlighter::getEdgeColor(const std::string& element, std::vector<App::Color>& colors) const
{
int idx = std::stoi(element.substr(4)) - 1;

12
src/Mod/Part/Gui/ReferenceHighlighter.h
View File

@@ -56,6 +56,13 @@ public:
objectColor = c;
}
/*!
* \brief getVertexColors
* \param elements The sub-element names. If this list is empty \a colors will be filled with the default color.
* \param colors The size of the \a colors array is equal to the number of vertexes of the shape
*/
void getVertexColors(const std::vector<std::string>& elements,
std::vector<App::Color>& colors) const;
/*!
* \brief getEdgeColors
* \param elements The sub-element names. If this list is empty \a colors will be filled with the default color.
@@ -72,6 +79,10 @@ public:
std::vector<App::Color>& colors) const;
private:
void getVertexColor(const std::string& element, std::vector<App::Color>& colors) const;
void getVertexColorsOfEdge(const std::string& element, std::vector<App::Color>& colors) const;
void getVertexColorsOfWire(const std::string& element, std::vector<App::Color>& colors) const;
void getVertexColorsOfFace(const std::string& element, std::vector<App::Color>& colors) const;
void getEdgeColor(const std::string& element, std::vector<App::Color>& colors) const;
void getEdgeColorsOfWire(const std::string& element, std::vector<App::Color>& colors) const;
void getEdgeColorsOfFace(const std::string& element, std::vector<App::Color>& colors) const;
@@ -81,6 +92,7 @@ private:
App::Color defaultColor;
App::Color elementColor;
App::Color objectColor;
TopTools_IndexedMapOfShape vMap;
TopTools_IndexedMapOfShape eMap;
TopTools_IndexedMapOfShape wMap;
TopTools_IndexedMapOfShape fMap;