-// Copyright (C) 2014-2019 CEA/DEN, EDF R&D
+// Copyright (C) 2014-2023 CEA, EDF
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
#include "PartSet_FieldStepPrs.h"
+#include <ModuleBase_Preferences.h>
+
#include <CollectionPlugin_Field.h>
#include <ModelAPI_AttributeSelectionList.h>
#include <ModelAPI_AttributeIntArray.h>
#include <ModelAPI_AttributeStringArray.h>
-#include <AIS_ColorScale.hxx>
+#include <SUIT_ResourceMgr.h>
+#include <AIS_ColorScale.hxx>
+#include <Prs3d_Drawer.hxx>
+#include <Prs3d_PointAspect.hxx>
+#include <BRep_Tool.hxx>
+#include <BRepAdaptor_Surface.hxx>
+#include <TopoDS.hxx>
+#include <TopoDS_Vertex.hxx>
+#include <TopExp_Explorer.hxx>
+#include <Graphic3d_Text.hxx>
+#include <Graphic3d_AspectText3d.hxx>
IMPLEMENT_STANDARD_RTTIEXT(PartSet_FieldStepPrs, ViewerData_AISShape);
+#define POINT_SIZE 8
-void emptyDeleter(ModelAPI_ResultField* theF)
+
+void emptyDeleter(ModelAPI_ResultField* /*theF*/)
{
// Do nothing
}
// Make shared_ptr which will not delete original pointer after exit
std::shared_ptr<ModelAPI_ResultField> aFieldPtr(aField, emptyDeleter);
myFeature = ModelAPI_Feature::feature(aFieldPtr);
+
+ if (dataType() != ModelAPI_AttributeTables::STRING) {
+ Handle(Prs3d_Drawer) aDrawer = Attributes();
+ if (aDrawer->HasOwnPointAspect()) {
+ aDrawer->PointAspect()->SetTypeOfMarker(Aspect_TOM_POINT);
+ aDrawer->PointAspect()->SetScale(POINT_SIZE);
+ }
+ else
+ aDrawer->SetPointAspect(
+ new Prs3d_PointAspect(Aspect_TOM_POINT, Quantity_NOC_YELLOW, POINT_SIZE));
+ }
+ SetMaterial(Graphic3d_NOM_PLASTIC);
}
{
ModelAPI_AttributeTables::ValueType aType = dataType();
DataPtr aData = myFeature->data();
+ AttributeSelectionListPtr aSelList = aData->selectionList(CollectionPlugin_Field::SELECTED_ID());
+ std::string aTypeStr = aSelList->selectionType();
+
int aStep = myStep->id();
AttributeTablesPtr aTablesAttr = aData->tables(CollectionPlugin_Field::VALUES_ID());
int aRows = aTablesAttr->rows();
int aCols = aTablesAttr->columns();
QList<double> aFieldStepData;
- for (int j = 0; j < aCols; j++) {
- for (int k = 1; k < aRows; k++) { // Do not use default values
+ int aStart = (aTypeStr == "part")? 0:1;
+ for (int k = aStart; k < aRows; k++) { // Do not use default values
+ for (int j = 0; j < aCols; j++) {
ModelAPI_AttributeTables::Value aVal = aTablesAttr->value(k, j, aStep);
switch (aType) {
case ModelAPI_AttributeTables::DOUBLE:
case ModelAPI_AttributeTables::INTEGER:
aFieldStepData << aVal.myInt;
break;
+ default: // [to avoid compilation warning]
+ break;
}
}
}
QList<double> aShapeData;
double aRangeMin = aFieldStepData.first(), aRangeMax = aFieldStepData.last();
- for (int aRow = 0; aRow < aRows - 1; aRow++) {
- double aNorm = 0;
- int aBaseIndex = aRow * aCols;
- for (int aCol = 0; aCol < aCols; aCol++) {
- int anIndex = aCol + aBaseIndex;
- double aValue = aFieldStepData.at(anIndex);
- aNorm += aValue * aValue;
+ if (aCols == 1) {
+ for (int aRow = 0; aRow < aRows - aStart; aRow++) {
+ double aValue = aFieldStepData.at(aRow);
+ aRangeMin = Min(aRangeMin, aValue);
+ aRangeMax = Max(aRangeMax, aValue);
+ aShapeData << aValue;
+ }
+ }
+ else {
+ for (int aRow = 0; aRow < aRows - aStart; aRow++) {
+ double aNorm = 0;
+ int aBaseIndex = aRow * aCols;
+ for (int aCol = 0; aCol < aCols; aCol++) {
+ int anIndex = aCol + aBaseIndex;
+ double aValue = aFieldStepData.at(anIndex);
+ aNorm += aValue * aValue;
+ }
+ aNorm = pow(aNorm, 0.5);
+ aRangeMin = Min(aRangeMin, aNorm);
+ aRangeMax = Max(aRangeMax, aNorm);
+ aShapeData << aNorm;
}
- aNorm = pow(aNorm, 0.5);
- aRangeMin = Min(aRangeMin, aNorm);
- aRangeMax = Max(aRangeMax, aNorm);
- aShapeData << aNorm;
}
theMin = aRangeMin;
theMax = aRangeMax;
void PartSet_FieldStepPrs::Compute(const Handle(PrsMgr_PresentationManager3d)& thePrsMgr,
const Handle(Prs3d_Presentation)& thePrs, const Standard_Integer theMode)
{
+ SUIT_ResourceMgr* aResMgr = ModuleBase_Preferences::resourceMgr();
+ QColor aQColor = aResMgr->colorValue("Viewer", "scalar_bar_text_color", Qt::black);
+ Quantity_Color aLabelColor = Quantity_Color(aQColor.redF(), aQColor.greenF(), aQColor.blueF(),
+ Quantity_TOC_RGB);
+
ModelAPI_AttributeTables::ValueType aType = dataType();
- if ((aType == ModelAPI_AttributeTables::DOUBLE) ||
- (aType == ModelAPI_AttributeTables::INTEGER)) {
+ DataPtr aData = myFeature->data();
+ switch (aType) {
+ case ModelAPI_AttributeTables::DOUBLE:
+ case ModelAPI_AttributeTables::INTEGER:
+ {
double aMin, aMax;
QList<double> aShapeData = range(aMin, aMax);
- int aNbIntertvals = 20;
- DataPtr aData = myFeature->data();
- AttributeSelectionListPtr aSelList = aData->selectionList(CollectionPlugin_Field::SELECTED_ID());
+ AttributeSelectionListPtr aSelList =
+ aData->selectionList(CollectionPlugin_Field::SELECTED_ID());
+ std::string aTypeStr = aSelList->selectionType();
+ if (aTypeStr == "part") {
+ Quantity_Color aColor;
+ if (AIS_ColorScale::FindColor(aMin, aMin, aMax, 1, aColor)) {
+ SetColor(aColor);
+ }
+ }
+ else {
+ int aNbIntertvals = aResMgr->integerValue("Viewer", "scalar_bar_nb_intervals", 20);
+ for (int i = 0; i < aSelList->size(); i++) {
+ AttributeSelectionPtr aSelection = aSelList->value(i);
+ GeomShapePtr aShapePtr = aSelection->value();
+ TopoDS_Shape aShape = aShapePtr->impl<TopoDS_Shape>();
+ double aValue = aShapeData.at(i);
+ Quantity_Color aColor;
+ if (AIS_ColorScale::FindColor(aValue, aMin, aMax, aNbIntertvals, aColor))
+ SetCustomColor(aShape, aColor);
+ }
+ }
+ }
+ break;
+ case ModelAPI_AttributeTables::BOOLEAN:
+ {
+ QList<double> aShapeData = booleanValues();
+
+ AttributeSelectionListPtr aSelList =
+ aData->selectionList(CollectionPlugin_Field::SELECTED_ID());
for (int i = 0; i < aSelList->size(); i++) {
AttributeSelectionPtr aSelection = aSelList->value(i);
GeomShapePtr aShapePtr = aSelection->value();
TopoDS_Shape aShape = aShapePtr->impl<TopoDS_Shape>();
double aValue = aShapeData.at(i);
Quantity_Color aColor;
- if (AIS_ColorScale::FindColor(aValue, aMin, aMax, aNbIntertvals, aColor))
+ if (AIS_ColorScale::FindColor(aValue, 0., 1., 2, aColor))
SetCustomColor(aShape, aColor);
}
}
+ break;
+ case ModelAPI_AttributeTables::STRING:
+ {
+ QStringList aValues = strings();
+ AttributeSelectionListPtr aSelList =
+ aData->selectionList(CollectionPlugin_Field::SELECTED_ID());
+ Handle(Graphic3d_Group) aGroup = thePrs->NewGroup();
+ for (int i = 0; i < aSelList->size(); i++) {
+ AttributeSelectionPtr aSelection = aSelList->value(i);
+ GeomShapePtr aShapePtr = aSelection->value();
+ TopoDS_Shape aShape = aShapePtr->impl<TopoDS_Shape>();
+ gp_Pnt aCenter;
+ if (computeMassCenter(aShape, aCenter)) {
+ Handle(Graphic3d_AspectText3d) anAspectText3d = new Graphic3d_AspectText3d();
+ anAspectText3d->SetStyle(Aspect_TOST_ANNOTATION);
+ anAspectText3d->SetColor(aLabelColor);
+ aGroup->SetPrimitivesAspect(anAspectText3d);
+
+ int aT = aResMgr->integerValue("Viewer", "scalar_bar_text_height", 14);
+ QString aString = aValues.at(i);
+ Handle(Graphic3d_Text) aText = new Graphic3d_Text(aT);
+ aText->SetText(aString.toUtf8().constData());
+ aText->SetPosition(aCenter);
+ aGroup->AddText(aText);
+ }
+ }
+ }
+ break;
+ }
ViewerData_AISShape::Compute(thePrsMgr, thePrs, theMode);
}
+
+QList<double> PartSet_FieldStepPrs::booleanValues() const
+{
+ DataPtr aData = myFeature->data();
+ int aStep = myStep->id();
+ AttributeTablesPtr aTablesAttr = aData->tables(CollectionPlugin_Field::VALUES_ID());
+ int aRows = aTablesAttr->rows();
+ int aCols = aTablesAttr->columns();
+ QList<int> aFieldStepData;
+ for (int k = 1; k < aRows; k++) { // Do not use default values
+ for (int j = 0; j < aCols; j++) {
+ ModelAPI_AttributeTables::Value aVal = aTablesAttr->value(k, j, aStep);
+ aFieldStepData << (aVal.myBool ? 1 : 0);
+ }
+ }
+ QList<double> aShapeData;
+ for (int aRow = 0; aRow < aRows - 1; aRow++) {
+ double aNorm = 0;
+ int aBaseIndex = aRow * aCols;
+ for (int aCol = 0; aCol < aCols; aCol++) {
+ aNorm += aFieldStepData.at(aCol + aBaseIndex);
+ }
+ aNorm /= aCols;
+ aShapeData << aNorm;
+ }
+ return aShapeData;
+}
+
+QStringList PartSet_FieldStepPrs::strings() const
+{
+ DataPtr aData = myFeature->data();
+ int aStep = myStep->id();
+ AttributeTablesPtr aTablesAttr = aData->tables(CollectionPlugin_Field::VALUES_ID());
+ int aRows = aTablesAttr->rows();
+ int aCols = aTablesAttr->columns();
+ QStringList aFieldStepData;
+ for (int k = 1; k < aRows; k++) { // Do not use default values
+ for (int j = 0; j < aCols; j++) {
+ ModelAPI_AttributeTables::Value aVal = aTablesAttr->value(k, j, aStep);
+ aFieldStepData << aVal.myStr.c_str();
+ }
+ }
+ QStringList aShapeData;
+ for (int aRow = 0; aRow < aRows - 1; aRow++) {
+ QStringList aRowStrings;
+ int aBaseIndex = aRow * aCols;
+ for (int aCol = 0; aCol < aCols; aCol++) {
+ aRowStrings << aFieldStepData.at(aCol + aBaseIndex);
+ }
+ aRowStrings.join('\n');
+ aShapeData << aRowStrings;
+ }
+ return aShapeData;
+}
+
+bool PartSet_FieldStepPrs::computeMassCenter(const TopoDS_Shape& theShape, gp_Pnt& theCenter)
+{
+ theCenter.SetCoord(0, 0, 0);
+ int aNbPoints = 0;
+
+ if (theShape.ShapeType() == TopAbs_EDGE) {
+ double f, l;
+ Handle(Geom_Curve) curve = BRep_Tool::Curve(TopoDS::Edge(theShape), f, l);
+ if (!curve.IsNull()) {
+ theCenter = curve->Value(0.5 * (f + l));
+ aNbPoints = 1;
+ }
+ }
+ else if (theShape.ShapeType() == TopAbs_FACE) {
+ const TopoDS_Face& F = TopoDS::Face(theShape);
+ BRepAdaptor_Surface surface(F);
+
+ TopLoc_Location L;
+ Handle(Poly_Triangulation) triangulation = BRep_Tool::Triangulation(F, L);
+ if (!triangulation.IsNull() && triangulation->HasUVNodes()) {
+ gp_XY C(0, 0);
+ double A = 0;
+ int n1, n2, n3;
+ for (int iT = 1; iT <= triangulation->NbTriangles(); ++iT) {
+ triangulation->Triangle(iT).Get(n1, n2, n3);
+ const gp_Pnt2d& uv1 = triangulation->UVNode(n1);
+ const gp_Pnt2d& uv2 = triangulation->UVNode(n2);
+ const gp_Pnt2d& uv3 = triangulation->UVNode(n3);
+ double a = 0.5 * sqrt((uv1.X() - uv3.X()) * (uv2.Y() - uv1.Y()) -
+ (uv1.X() - uv2.X()) * (uv3.Y() - uv1.Y()));
+ C += (uv1.XY() + uv2.XY() + uv3.XY()) / 3. * a;
+ A += a;
+ }
+ if (A > std::numeric_limits<double>::min()) {
+ C /= A;
+ theCenter = surface.Value(C.X(), C.Y());
+ aNbPoints = 1;
+ }
+ }
+ if (aNbPoints == 0) {
+ theCenter = surface.Value(0.5 * (surface.FirstUParameter() + surface.LastUParameter()),
+ 0.5 * (surface.FirstVParameter() + surface.LastVParameter()));
+ }
+ aNbPoints = 1;
+ }
+
+ if (aNbPoints == 0) {
+ TopExp_Explorer anExp;
+ for (anExp.Init(theShape, TopAbs_VERTEX); anExp.More(); anExp.Next()) {
+ TopoDS_Vertex aVertex = TopoDS::Vertex(anExp.Current());
+ if (!aVertex.IsNull()) {
+ gp_Pnt aPnt = BRep_Tool::Pnt(aVertex);
+ theCenter.ChangeCoord() += aPnt.XYZ();
+ aNbPoints++;
+ }
+ }
+ }
+
+ if (aNbPoints > 0)
+ theCenter.ChangeCoord() /= (double)aNbPoints;
+
+ return aNbPoints;
+}
