X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FSketcherPrs%2FSketcherPrs_SymbolPrs.cpp;h=2e2779c7b6963a8520656e3d55712b2a67e6cdeb;hb=63f80d9837b4dcb16e9fafc3b07fac36f6f7b00d;hp=b953501fa8b509abe938eb6182df85e9aa36e199;hpb=eb0bc337423e37ed5bfbd9cbcb3c6ec183108c3b;p=modules%2Fshaper.git diff --git a/src/SketcherPrs/SketcherPrs_SymbolPrs.cpp b/src/SketcherPrs/SketcherPrs_SymbolPrs.cpp index b953501fa..2e2779c7b 100644 --- a/src/SketcherPrs/SketcherPrs_SymbolPrs.cpp +++ b/src/SketcherPrs/SketcherPrs_SymbolPrs.cpp @@ -6,15 +6,36 @@ #include "SketcherPrs_SymbolPrs.h" #include "SketcherPrs_Tools.h" +#include "SketcherPrs_PositionMgr.h" #include +#include +#include + +#include #include +#include #include -#include #include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include #ifdef WIN32 # define FSEP "\\" @@ -22,6 +43,207 @@ # define FSEP "/" #endif +/// Step between icons +static const double MyDist = 0.02; + +/// Function to convert opengl data type +GLenum toGlDataType (const Graphic3d_TypeOfData theType, GLint& theNbComp) +{ + switch (theType) { + case Graphic3d_TOD_USHORT: + theNbComp = 1; + return GL_UNSIGNED_SHORT; + case Graphic3d_TOD_UINT: + theNbComp = 1; + return GL_UNSIGNED_INT; + case Graphic3d_TOD_VEC2: + theNbComp = 2; + return GL_FLOAT; + case Graphic3d_TOD_VEC3: + theNbComp = 3; + return GL_FLOAT; + case Graphic3d_TOD_VEC4: + theNbComp = 4; + return GL_FLOAT; + case Graphic3d_TOD_VEC4UB: + theNbComp = 4; + return GL_UNSIGNED_BYTE; + } + theNbComp = 0; + return GL_NONE; +} + + +//******************************************************************* +//! Auxiliary class for Vertex buffer with interleaved attributes. +class SketcherPrs_VertexBuffer : public OpenGl_VertexBuffer +{ + +public: + + //! Create uninitialized VBO.. + //! \param theAttribs attributes + //! \param theStride a flag + SketcherPrs_VertexBuffer (const Graphic3d_Attribute* theAttribs, + const Standard_Integer theStride) + : Stride (theStride), NbAttributes(1) + { + + memcpy (Attribs, theAttribs, sizeof(Graphic3d_Attribute) * NbAttributes); + } + + //! Create uninitialized VBO. + SketcherPrs_VertexBuffer (const Graphic3d_Buffer& theAttribs) + : Stride (theAttribs.Stride), NbAttributes(1) + { + memcpy (Attribs, theAttribs.AttributesArray(), sizeof(Graphic3d_Attribute) * NbAttributes); + } + + /// Returns True if color attribute is defined + virtual bool HasColorAttribute() const + { + for (Standard_Integer anAttribIter = 0; anAttribIter < NbAttributes; ++anAttribIter) { + const Graphic3d_Attribute& anAttrib = Attribs[anAttribIter]; + if (anAttrib.Id == Graphic3d_TOA_COLOR) { + return true; + } + } + return false; + } + + /// Returns True if normal attribute is defined + virtual bool HasNormalAttribute() const + { + for (Standard_Integer anAttribIter = 0; anAttribIter < NbAttributes; ++anAttribIter) { + const Graphic3d_Attribute& anAttrib = Attribs[anAttribIter]; + if (anAttrib.Id == Graphic3d_TOA_NORM) { + return true; + } + } + return false; + } + + /// Bind position of the attribute + /// \param theGlCtx OpenGl context + virtual void BindPositionAttribute (const Handle(OpenGl_Context)& theGlCtx) const + { + if (!OpenGl_VertexBuffer::IsValid()) { + return; + } + + OpenGl_VertexBuffer::Bind (theGlCtx); + GLint aNbComp; + const GLubyte* anOffset = OpenGl_VertexBuffer::myOffset; + for (Standard_Integer anAttribIter = 0; anAttribIter < NbAttributes; ++anAttribIter) { + const Graphic3d_Attribute& anAttrib = Attribs[anAttribIter]; + const GLenum aDataType = toGlDataType (anAttrib.DataType, aNbComp); + if (aDataType == GL_NONE) { + continue; + } else if (anAttrib.Id == Graphic3d_TOA_POS) { + OpenGl_VertexBuffer::bindAttribute(theGlCtx, Graphic3d_TOA_POS, aNbComp, + aDataType, Stride, anOffset); + break; + } + + anOffset += Graphic3d_Attribute::Stride (anAttrib.DataType); + } + } + + /// Bind all attributes + /// \param theGlCtx OpenGl context + virtual void BindAllAttributes (const Handle(OpenGl_Context)& theGlCtx) const + { + if (!OpenGl_VertexBuffer::IsValid()) + return; + + OpenGl_VertexBuffer::Bind (theGlCtx); + GLint aNbComp; + const GLubyte* anOffset = OpenGl_VertexBuffer::myOffset; + for (Standard_Integer anAttribIter = 0; anAttribIter < NbAttributes; ++anAttribIter) + { + const Graphic3d_Attribute& anAttrib = Attribs[anAttribIter]; + const GLenum aDataType = toGlDataType (anAttrib.DataType, aNbComp); + if (aDataType == GL_NONE) + continue; + + OpenGl_VertexBuffer::bindAttribute(theGlCtx, anAttrib.Id, aNbComp, + aDataType, Stride, anOffset); + anOffset += Graphic3d_Attribute::Stride (anAttrib.DataType); + } + } + + /// Unbind all attributes + /// \param theGlCtx OpenGl context + virtual void UnbindAllAttributes (const Handle(OpenGl_Context)& theGlCtx) const + { + if (!OpenGl_VertexBuffer::IsValid()) + return; + OpenGl_VertexBuffer::Unbind (theGlCtx); + + for (Standard_Integer anAttribIter = 0; anAttribIter < NbAttributes; ++anAttribIter) { + const Graphic3d_Attribute& anAttrib = Attribs[anAttribIter]; + OpenGl_VertexBuffer::unbindAttribute (theGlCtx, anAttrib.Id); + } + } + +public: + + /// Array of attributes + Graphic3d_Attribute Attribs[1]; + + /// A flag + Standard_Integer Stride; + + /// Number of attributes + Standard_Integer NbAttributes; +}; + +//************************************************************** +//! Redefinition of OpenGl_Element +class SketcherPrs_Element: public OpenGl_Element +{ +public: + /// Constructor + /// \param theObj a presentation + SketcherPrs_Element(const Handle(SketcherPrs_SymbolPrs)& theObj) : + OpenGl_Element(), myObj(theObj) {} + + /// Render the current presentation + /// \param theWorkspace OpenGL workspace + virtual void Render (const Handle(OpenGl_Workspace)& theWorkspace) const + { + if (!myObj.IsNull()) + myObj->Render(theWorkspace); + } + + /// Releases OpenGL resources + /// \param theContext OpenGL context + virtual void Release (OpenGl_Context* theContext) + { + if (!myObj.IsNull()) + myObj->Release(theContext); + } + +private: + Handle(SketcherPrs_SymbolPrs) myObj; +}; + + +//************************************************************** +//! Definition of call back +OpenGl_Element* SymbolPrsCallBack(const CALL_DEF_USERDRAW * theUserDraw) +{ + Handle(SketcherPrs_SymbolPrs) anIObj = (SketcherPrs_SymbolPrs*)theUserDraw->Data; + if (anIObj.IsNull()) { + std::cout << + "VUserDrawCallback error: null object passed, the custom scene element will not be rendered" + << std::endl; + } + return new SketcherPrs_Element(anIObj); +} + + +//***************************************************************************** IMPLEMENT_STANDARD_HANDLE(SketcherPrs_SymbolPrs, AIS_InteractiveObject); IMPLEMENT_STANDARD_RTTIEXT(SketcherPrs_SymbolPrs, AIS_InteractiveObject); @@ -29,46 +251,72 @@ IMPLEMENT_STANDARD_RTTIEXT(SketcherPrs_SymbolPrs, AIS_InteractiveObject); std::map SketcherPrs_SymbolPrs::myIconsMap; -SketcherPrs_SymbolPrs::SketcherPrs_SymbolPrs(SketchPlugin_Constraint* theConstraint, +SketcherPrs_SymbolPrs::SketcherPrs_SymbolPrs(ModelAPI_Feature* theConstraint, const std::shared_ptr& thePlane) - : AIS_InteractiveObject(), myConstraint(theConstraint), myPlane(thePlane) + : AIS_InteractiveObject(), myConstraint(theConstraint), myPlane(thePlane), myIsConflicting(false) { SetAutoHilight(Standard_False); + myPntArray = new Graphic3d_ArrayOfPoints(1); + myPntArray->AddVertex(0., 0., 0.); } +SketcherPrs_SymbolPrs::~SketcherPrs_SymbolPrs() +{ + SketcherPrs_PositionMgr* aMgr = SketcherPrs_PositionMgr::get(); + // Empty memory in position manager + aMgr->deleteConstraint(this); +} + +#ifdef _WINDOWS +#pragma warning( disable : 4996 ) +#endif Handle(Image_AlienPixMap) SketcherPrs_SymbolPrs::icon() { if (myIconsMap.count(iconName()) == 1) { return myIconsMap[iconName()]; } - TCollection_AsciiString aFile(getenv("NewGeomResources")); + // Load icon for the presentation + std::string aFile; + char* anEnv = getenv("SHAPER_ROOT_DIR"); + if (anEnv) { + aFile = std::string(anEnv) + + FSEP + "share" + FSEP + "salome" + FSEP + "resources" + FSEP + "shaper"; + } else { + anEnv = getenv("OPENPARTS_ROOT_DIR"); + if (anEnv) + aFile = std::string(anEnv) + FSEP + "resources"; + } + aFile += FSEP; aFile += iconName(); Handle(Image_AlienPixMap) aPixMap = new Image_AlienPixMap(); - if (aPixMap->Load(aFile)) { + if (aPixMap->Load(aFile.c_str())) { myIconsMap[iconName()] = aPixMap; return aPixMap; } + // The icon for constraint is not found + static const char aMsg[] = "Error! constraint images are not found"; + cout<Clear(); - } -} - void SketcherPrs_SymbolPrs::prepareAspect() { + // Create an aspect with the icon if (myAspect.IsNull()) { - myAspect = new Graphic3d_AspectMarker3d(icon()); + Handle(Image_AlienPixMap) aIcon = icon(); + if (aIcon.IsNull()) + myAspect = new Graphic3d_AspectMarker3d(); + else + myAspect = new Graphic3d_AspectMarker3d(aIcon); } } -void SketcherPrs_SymbolPrs::addLine(const Handle(Graphic3d_Group)& theGroup, std::string theAttrName) const +void SketcherPrs_SymbolPrs::addLine(const Handle(Graphic3d_Group)& theGroup, + std::string theAttrName) const { ObjectPtr aObj = SketcherPrs_Tools::getResult(myConstraint, theAttrName); std::shared_ptr aLine = SketcherPrs_Tools::getShape(aObj); @@ -79,13 +327,14 @@ void SketcherPrs_SymbolPrs::addLine(const Handle(Graphic3d_Group)& theGroup, std std::shared_ptr aPnt1 = aEdge->firstPoint(); std::shared_ptr aPnt2 = aEdge->lastPoint(); + // Draw line by two points Handle(Graphic3d_ArrayOfSegments) aLines = new Graphic3d_ArrayOfSegments(2, 1); aLines->AddVertex(aPnt1->impl()); aLines->AddVertex(aPnt2->impl()); theGroup->AddPrimitiveArray(aLines); } -void SketcherPrs_SymbolPrs::HilightSelected(const Handle(PrsMgr_PresentationManager3d)& thePM, +void SketcherPrs_SymbolPrs::HilightSelected(const Handle(PrsMgr_PresentationManager3d)& thePM, const SelectMgr_SequenceOfOwner& theOwners) { @@ -98,8 +347,10 @@ void SketcherPrs_SymbolPrs::HilightSelected(const Handle(PrsMgr_PresentationMana thePM->Highlight(this); } -void SketcherPrs_SymbolPrs::HilightOwnerWithColor(const Handle(PrsMgr_PresentationManager3d)& thePM, - const Quantity_NameOfColor theColor, const Handle(SelectMgr_EntityOwner)& theOwner) +void SketcherPrs_SymbolPrs::HilightOwnerWithColor( + const Handle(PrsMgr_PresentationManager3d)& thePM, + const Quantity_NameOfColor theColor, + const Handle(SelectMgr_EntityOwner)& theOwner) { thePM->Color(this, theColor); @@ -111,18 +362,255 @@ void SketcherPrs_SymbolPrs::HilightOwnerWithColor(const Handle(PrsMgr_Presentati thePM->AddToImmediateList(aHilightPrs); } +void SketcherPrs_SymbolPrs::Compute( + const Handle(PrsMgr_PresentationManager3d)& thePresentationManager, + const Handle(Prs3d_Presentation)& thePresentation, + const Standard_Integer theMode) +{ + // Create an icon + prepareAspect(); + + Handle(AIS_InteractiveContext) aCtx = GetContext(); + Handle(OpenGl_GraphicDriver) aDriver = + Handle(OpenGl_GraphicDriver)::DownCast(aCtx->CurrentViewer()->Driver()); + if (!aDriver.IsNull()) { + // register the custom element factory function + aDriver->UserDrawCallback() = SymbolPrsCallBack; + } + + // Update points with default shift value + // it updates array of points if the presentation is ready to display, or the array of points + // contains the previous values + + bool aReadyToDisplay = updateIfReadyToDisplay(20); + + int aNbVertex = myPntArray->VertexNumber(); + if (myOwner.IsNull()) { + myOwner = new SelectMgr_EntityOwner(this); + } + + // Create sensitive point for each symbol + mySPoints.Clear(); + for (int i = 1; i <= aNbVertex; i++) { + Handle(SketcherPrs_SensitivePoint) aSP = new SketcherPrs_SensitivePoint(myOwner, i); + mySPoints.Append(aSP); + } + + Handle(Graphic3d_Group) aGroup = Prs3d_Root::NewGroup(thePresentation); + aGroup->SetPrimitivesAspect(myAspect); + + // Recompute boundary box of the group + Graphic3d_BndBox4f& aBnd = aGroup->ChangeBoundingBox(); + gp_Pnt aVert; + aBnd.Clear(); + for (int i = 1; i <= myPntArray->ItemNumber(); i++) { + aVert = myPntArray->Vertice(i); + aBnd.Add (Graphic3d_Vec4((float)aVert.X(), (float)aVert.Y(), (float)aVert.Z(), 1.0f)); + } + + // Pint the group with custom procedure (see Render) + aGroup->UserDraw(this, true); + + // Disable frustum culling for this object by marking it as mutable + aGroup->Structure()->SetMutable(true); + //aGroup->AddPrimitiveArray(myPntArray); + + if (!aReadyToDisplay) + SketcherPrs_Tools::sendEmptyPresentationError(myConstraint, + "An empty AIS presentation: SketcherPrs_LengthDimension"); +} + void SketcherPrs_SymbolPrs::ComputeSelection(const Handle(SelectMgr_Selection)& aSelection, - const Standard_Integer aMode) + const Standard_Integer aMode) { ClearSelected(); + if ((aMode == 0) || (aMode == SketcherPrs_Tools::Sel_Constraint)) { + for (int i = 1; i <= mySPoints.Length(); i++) + aSelection->Add(mySPoints.Value(i)); + } +} + +void SketcherPrs_SymbolPrs::SetConflictingConstraint(const bool& theConflicting, + const std::vector& theColor) +{ + if (theConflicting) + { + if (!myAspect.IsNull()) + myAspect->SetColor (Quantity_Color (theColor[0] / 255., theColor[1] / 255., + theColor[2] / 255., Quantity_TOC_RGB)); + myIsConflicting = true; + } + else + { + if (!myAspect.IsNull()) + myAspect->SetColor (Quantity_Color (1.0, 1.0, 0.0, Quantity_TOC_RGB)); + myIsConflicting = false; + } +} + +void SketcherPrs_SymbolPrs::Render(const Handle(OpenGl_Workspace)& theWorkspace) const +{ + // this method is a combination of OCCT OpenGL functions. The main purpose is to have + // equal distance from the source object to symbol indpendently of zoom. + // It is base on OCCT 6.9.1 and might need changes when using later OCCT versions. + // The specific SHAPER modifications are marked by ShaperModification:start/end, + // other is OCCT code + + // do not update presentation for invalid or already removed objects: the presentation + // should be removed soon + if (!myConstraint->data().get() || !myConstraint->data()->isValid()) + return; + + const OpenGl_AspectMarker* anAspectMarker = theWorkspace->AspectMarker(Standard_True); + const Handle(OpenGl_Context)& aCtx = theWorkspace->GetGlContext(); + Handle(OpenGl_View) aView = theWorkspace->ActiveView(); + + // ShaperModification:start + double aScale = aView->Camera()->Scale(); + // Update points coordinate taking the viewer scale into account + updateIfReadyToDisplay(MyDist * aScale); + + // ShaperModification:end + + Handle(Graphic3d_Buffer) aAttribs = myPntArray->Attributes(); + + if (myVboAttribs.IsNull()) { + myVboAttribs = new SketcherPrs_VertexBuffer(*aAttribs); + } + + // Update drawing attributes + if (!myVboAttribs->init(aCtx, 0, aAttribs->NbElements, + aAttribs->Data(), GL_NONE, aAttribs->Stride)) { + myVboAttribs->Release (aCtx.operator->()); + myVboAttribs.Nullify(); + return; + } + + Handle(OpenGl_Texture) aTextureBack = theWorkspace->DisableTexture(); + + const Handle(OpenGl_PointSprite)& aSpriteNorm = anAspectMarker->SpriteRes(aCtx); + + if (!aSpriteNorm.IsNull() && !aSpriteNorm->IsDisplayList()) { + // ShaperModification:start : filling the presentation with color if there is a conflict + const bool toHilight = + (theWorkspace->NamedStatus & OPENGL_NS_HIGHLIGHT) != 0 || myIsConflicting; + // ShaperModification:end + + const Handle(OpenGl_PointSprite)& aSprite = + (toHilight && anAspectMarker->SpriteHighlightRes(aCtx)->IsValid())? + anAspectMarker->SpriteHighlightRes(aCtx) + : aSpriteNorm; + theWorkspace->EnableTexture (aSprite); + aCtx->ShaderManager()->BindProgram(anAspectMarker, aSprite, Standard_False, + Standard_False, anAspectMarker->ShaderProgramRes(aCtx)); + const TEL_COLOUR* aLineColor = &anAspectMarker->Color(); + if (theWorkspace->NamedStatus & OPENGL_NS_HIGHLIGHT) + aLineColor = theWorkspace->HighlightColor; + + // Set lighting of the symbol + if (toHilight) + aCtx->core11fwd->glDisable (GL_LIGHTING); + else + aCtx->core11fwd->glEnable (GL_LIGHTING); + + aCtx->SetColor4fv(*(const OpenGl_Vec4* )(aLineColor->rgb)); + + + myVboAttribs->BindAllAttributes(aCtx); + // Textured markers will be drawn with the point sprites + aCtx->SetPointSize (anAspectMarker->MarkerSize()); + aCtx->core11fwd->glEnable (GL_ALPHA_TEST); + aCtx->core11fwd->glAlphaFunc (GL_GEQUAL, 0.1f); + + aCtx->core11fwd->glEnable (GL_BLEND); + aCtx->core11fwd->glBlendFunc (GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA); + + aCtx->core11fwd->glDrawArrays (0, 0, myVboAttribs->GetElemsNb()); + + aCtx->core11fwd->glDisable (GL_BLEND); + aCtx->core11fwd->glDisable (GL_ALPHA_TEST); + aCtx->SetPointSize (1.0f); + } + theWorkspace->EnableTexture (aTextureBack); + aCtx->BindProgram (NULL); + + // Update selection position only if there is no selected object + // because it can corrupt selection of other objects + if ((GetContext()->NbCurrents() == 0) && (GetContext()->NbSelected() == 0)) + { + GetContext()->MainSelector()->RebuildSensitivesTree (this); + GetContext()->MainSelector()->RebuildObjectsTree (false); + } +} + + +void SketcherPrs_SymbolPrs::Release (OpenGl_Context* theContext) +{ + // Release OpenGl resources + if (!myVboAttribs.IsNull()) { + if (theContext) { + theContext->DelayedRelease (myVboAttribs); + } + myVboAttribs.Nullify(); + } +} - if (!myPntArray.IsNull()) { - Handle(SelectMgr_EntityOwner) aOwn = new SelectMgr_EntityOwner(this); - for (int i = 1; i <= myPntArray->VertexNumber(); i++) { - Handle(Select3D_SensitivePoint) aSP = new Select3D_SensitivePoint(aOwn, myPntArray->Vertice(i)); - aSelection->Add(aSP); +void SketcherPrs_SymbolPrs::drawShape(const std::shared_ptr& theShape, + const Handle(Prs3d_Presentation)& thePrs) const +{ + if (theShape->isEdge()) { + // The shape is edge + std::shared_ptr aCurve = + std::shared_ptr(new GeomAPI_Curve(theShape)); + if (aCurve->isLine()) { + // The shape is line + GeomAdaptor_Curve + aCurv(aCurve->impl(), aCurve->startParam(), aCurve->endParam()); + StdPrs_Curve::Add(thePrs, aCurv, myDrawer); + } else { + // The shape is circle or arc + GeomAdaptor_Curve + aAdaptor(aCurve->impl(), aCurve->startParam(), aCurve->endParam()); + StdPrs_DeflectionCurve::Add(thePrs,aAdaptor,myDrawer); } + } else if (theShape->isVertex()) { + // draw vertex + std::shared_ptr aVertex = + std::shared_ptr(new GeomAPI_Vertex(theShape)); + std::shared_ptr aPnt = aVertex->point(); + Handle(Geom_CartesianPoint) aPoint = new Geom_CartesianPoint(aPnt->impl()); + StdPrs_Point::Add(thePrs, aPoint, myDrawer); + } +} + +void SketcherPrs_SymbolPrs::drawListOfShapes( + const std::shared_ptr& theListAttr, + const Handle(Prs3d_Presentation)& thePrs) const +{ + int aNb = theListAttr->size(); + if (aNb == 0) + return; + int i; + ObjectPtr aObj; + for (i = 0; i < aNb; i++) { + aObj = theListAttr->object(i); + std::shared_ptr aShape = SketcherPrs_Tools::getShape(aObj); + if (aShape.get() != NULL) + drawShape(aShape, thePrs); + } +} + +void SketcherPrs_SymbolPrs::BoundingBox (Bnd_Box& theBndBox) +{ + Select3D_BndBox3d aTmpBox; + for (Select3D_EntitySequenceIter aPntIter (mySPoints); aPntIter.More(); aPntIter.Next()) { + const Handle(Select3D_SensitiveEntity)& anEnt = aPntIter.Value(); + aTmpBox.Combine (anEnt->BoundingBox()); } + + theBndBox.Update (aTmpBox.CornerMin().x(), aTmpBox.CornerMin().y(), aTmpBox.CornerMin().z(), + aTmpBox.CornerMax().x(), aTmpBox.CornerMax().y(), aTmpBox.CornerMax().z()); } +