#1274 - pre-proto - showing stimPlan mesh

ApplicationCode/ModelVisualization/RivFaultGeometryGenerator.h

@@ -58,9 +58,10 @@ public:
     cvf::ref<cvf::DrawableGeo>    createMeshDrawable();
     cvf::ref<cvf::DrawableGeo>    createOutlineMeshDrawable(double creaseAngle);
 
-private:
     static cvf::ref<cvf::UIntArray> lineIndicesFromQuadVertexArray(const cvf::Vec3fArray* vertexArray);
 
+private:
+
     void computeArrays();
 
 private:

ApplicationCode/ModelVisualization/RivWellFracturePartMgr.cpp

@@ -41,6 +41,7 @@
 #include "cvfPrimitiveSetIndexedUInt.h"
 #include "cvfScalarMapperContinuousLinear.h"
 #include "cvfRenderStateDepth.h"
+#include "RivFaultGeometryGenerator.h"
 
 
 //--------------------------------------------------------------------------------------------------
@@ -166,6 +167,7 @@ void RivWellFracturePartMgr::updatePartGeometryTexture(caf::DisplayCoordTransfor
         m_part->setDrawable(geo.p());
 
         generateFractureOutlinePolygonPart(displayCoordTransform);
+        generateStimPlanMeshPart(displayCoordTransform);
 
         float opacityLevel = activeView->stimPlanColors->opacityLevel();
         if (legendConfig)
@@ -244,6 +246,80 @@ void RivWellFracturePartMgr::generateFractureOutlinePolygonPart(caf::DisplayCoor
     m_polygonPart->setEffect(eff.p());
 }
 
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+void RivWellFracturePartMgr::generateStimPlanMeshPart(caf::DisplayCoordTransform* displayCoordTransform)
+{
+
+    if (!m_rimFracture->attachedFractureDefinition()) return;
+
+    RimStimPlanFractureTemplate* stimPlanFracTemplate = dynamic_cast<RimStimPlanFractureTemplate*>(m_rimFracture->attachedFractureDefinition());
+    if (!stimPlanFracTemplate) return;
+
+    cvf::ref<cvf::DrawableGeo> stimPlanMeshGeo = createStimPlanMeshDrawable(stimPlanFracTemplate, displayCoordTransform);
+
+
+    //  From cvf::ref<cvf::DrawableGeo> RivFaultGeometryGenerator::createMeshDrawable()
+
+
+
+    m_StimPlanMeshPart = new cvf::Part;
+    m_StimPlanMeshPart->setDrawable(stimPlanMeshGeo.p());
+
+    m_StimPlanMeshPart->updateBoundingBox();
+    m_StimPlanMeshPart->setPriority(RivPartPriority::PartType::TransparentMeshLines);
+
+    caf::MeshEffectGenerator lineEffGen(cvf::Color3::BLACK);
+    lineEffGen.setLineWidth(1.0f);
+    cvf::ref<cvf::Effect> eff = lineEffGen.generateCachedEffect();
+
+    m_StimPlanMeshPart->setEffect(eff.p());
+
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+cvf::ref<cvf::DrawableGeo> RivWellFracturePartMgr::createStimPlanMeshDrawable(RimStimPlanFractureTemplate* stimPlanFracTemplate, caf::DisplayCoordTransform* displayCoordTransform)
+{
+    std::vector<double> depthCoords = stimPlanFracTemplate->adjustedDepthCoordsAroundWellPathPosition();
+    std::vector<double> xCoords = stimPlanFracTemplate->getNegAndPosXcoords();
+
+    std::vector<cvf::Vec3f> stimPlanMeshVertices;
+    for (int i = 0; i < xCoords.size() - 1; i++)
+    {
+        for (int j = 0; j < depthCoords.size() - 1; j++)
+        {
+            if (stimPlanCellTouchesPolygon(xCoords[i], xCoords[i + 1], depthCoords[j], depthCoords[j + 1]))
+            {
+                stimPlanMeshVertices.push_back(cvf::Vec3f(static_cast<float>(xCoords[i]), static_cast<float>(depthCoords[j]), 0.0f));
+                stimPlanMeshVertices.push_back(cvf::Vec3f(static_cast<float>(xCoords[i + 1]), static_cast<float>(depthCoords[j]), 0.0f));
+                stimPlanMeshVertices.push_back(cvf::Vec3f(static_cast<float>(xCoords[i + 1]), static_cast<float>(depthCoords[j + 1]), 0.0f));
+                stimPlanMeshVertices.push_back(cvf::Vec3f(static_cast<float>(xCoords[i]), static_cast<float>(depthCoords[j + 1]), 0.0f));
+            }
+        }
+    }
+
+    cvf::Mat4f m = m_rimFracture->transformMatrix();
+    std::vector<cvf::Vec3f> stimPlanMeshVerticesDisplayCoords = transfromToFractureDisplayCoords(stimPlanMeshVertices, m, displayCoordTransform);
+
+    cvf::Vec3fArray* stimPlanMeshVertexList;
+    stimPlanMeshVertexList = new cvf::Vec3fArray;
+    stimPlanMeshVertexList->assign(stimPlanMeshVerticesDisplayCoords);
+
+    cvf::ref<cvf::DrawableGeo> stimPlanMeshGeo = new cvf::DrawableGeo;
+    stimPlanMeshGeo->setVertexArray(stimPlanMeshVertexList);
+    cvf::ref<cvf::UIntArray> indices = RivFaultGeometryGenerator::lineIndicesFromQuadVertexArray(stimPlanMeshVertexList);
+    cvf::ref<cvf::PrimitiveSetIndexedUInt> prim = new cvf::PrimitiveSetIndexedUInt(cvf::PT_LINES);
+    prim->setIndices(indices.p());
+
+    stimPlanMeshGeo->addPrimitiveSet(prim.p());
+
+
+    return stimPlanMeshGeo;
+}
+
 //--------------------------------------------------------------------------------------------------
 /// 
 //--------------------------------------------------------------------------------------------------
@@ -252,25 +328,17 @@ cvf::ref<cvf::DrawableGeo> RivWellFracturePartMgr::createPolygonDrawable(caf::Di
 
     std::vector<cvf::Vec3f> polygon = m_rimFracture->attachedFractureDefinition()->fracturePolygon(m_rimFracture->fractureUnit);
 
-    //Transform to model coordinates and then display coords
-    std::vector<cvf::Vec3f> polygonInDisplayCoords;
     cvf::Mat4f m = m_rimFracture->transformMatrix();
-    for (cvf::Vec3f v : polygon)
-    {
-        v.transformPoint(m);
-        cvf::Vec3d nodeCoordsDouble = static_cast<cvf::Vec3d>(v);
-        cvf::Vec3d displayCoordsDouble = displayCoordTransform->transformToDisplayCoord(nodeCoordsDouble);
-        polygonInDisplayCoords.push_back(static_cast<cvf::Vec3f>(displayCoordsDouble));
+    std::vector<cvf::Vec3f> polygonDisplayCoords = transfromToFractureDisplayCoords(polygon, m, displayCoordTransform);
 
-    }
 
     std::vector<cvf::uint> lineIndices;
     std::vector<cvf::Vec3f> vertices;
 
-    for (size_t i = 0; i < polygonInDisplayCoords.size(); ++i)
+    for (size_t i = 0; i < polygonDisplayCoords.size(); ++i)
     {
-        vertices.push_back(cvf::Vec3f(polygonInDisplayCoords[i])); 
-        if (i < polygonInDisplayCoords.size() - 1)
+        vertices.push_back(cvf::Vec3f(polygonDisplayCoords[i]));
+        if (i < polygonDisplayCoords.size() - 1)
         {
             lineIndices.push_back(static_cast<cvf::uint>(i));
             lineIndices.push_back(static_cast<cvf::uint>(i + 1));
@@ -294,6 +362,23 @@ cvf::ref<cvf::DrawableGeo> RivWellFracturePartMgr::createPolygonDrawable(caf::Di
     return polygonGeo;
 }
 
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+std::vector<cvf::Vec3f> RivWellFracturePartMgr::transfromToFractureDisplayCoords(std::vector<cvf::Vec3f> coordinatesVector, cvf::Mat4f m, caf::DisplayCoordTransform* displayCoordTransform)
+{
+    std::vector<cvf::Vec3f> polygonInDisplayCoords;
+    for (cvf::Vec3f v : coordinatesVector)
+    {
+        v.transformPoint(m);
+        cvf::Vec3d nodeCoordsDouble = static_cast<cvf::Vec3d>(v);
+        cvf::Vec3d displayCoordsDouble = displayCoordTransform->transformToDisplayCoord(nodeCoordsDouble);
+        polygonInDisplayCoords.push_back(static_cast<cvf::Vec3f>(displayCoordsDouble));
+
+    }
+    return polygonInDisplayCoords;
+}
+
 //--------------------------------------------------------------------------------------------------
 /// 
 //--------------------------------------------------------------------------------------------------
@@ -329,6 +414,14 @@ void RivWellFracturePartMgr::appendGeometryPartsToModel(cvf::ModelBasicList* mod
             if (dynamic_cast<RimStimPlanFractureTemplate*>(m_rimFracture->attachedFractureDefinition()))
             {
                 updatePartGeometryTexture(displayCoordTransform);
+
+                RimStimPlanFractureTemplate* stimPlanFracTemplate = dynamic_cast<RimStimPlanFractureTemplate*>(m_rimFracture->attachedFractureDefinition());
+                if (stimPlanFracTemplate->showStimPlanMesh() && m_StimPlanMeshPart.notNull())
+                {
+                    model->addPart(m_StimPlanMeshPart.p());
+                }
+
+
             }
             else
             {
@@ -372,3 +465,24 @@ cvf::ref<cvf::DrawableGeo> RivWellFracturePartMgr::createGeo(const std::vector<c
     return geo;
 }
 
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+bool RivWellFracturePartMgr::stimPlanCellTouchesPolygon(double xMin, double xMax, double yMin, double yMax)
+{
+    std::vector<cvf::Vec3f> polygon = m_rimFracture->attachedFractureDefinition()->fracturePolygon(m_rimFracture->fractureUnit);
+
+    for (cvf::Vec3f v : polygon)
+    {
+        if (v.x() > xMin && v.x() < xMax)
+        {
+            if (v.y() > yMin && v.y() < yMax)
+            {
+                return true;
+            }
+        }
+    }
+
+    return false;
+}
+

ApplicationCode/ModelVisualization/RivWellFracturePartMgr.h

@@ -19,6 +19,7 @@
 #pragma once
 
 #include "cvfBase.h"
+#include "cvfMatrix4.h"
 #include "cvfObject.h"
 #include "cvfVector3.h"
 
@@ -38,9 +39,8 @@ namespace caf
     class DisplayCoordTransform;
 }
 
-
 class RimFracture;
-
+class RimStimPlanFractureTemplate;
 
 //--------------------------------------------------------------------------------------------------
 /// 
@@ -59,14 +59,22 @@ private:
     void updatePartGeometryTexture(caf::DisplayCoordTransform* displayCoordTransform);
 
     void generateFractureOutlinePolygonPart(caf::DisplayCoordTransform* displayCoordTransform);
+    void generateStimPlanMeshPart(caf::DisplayCoordTransform* displayCoordTransform);
 
     cvf::ref<cvf::DrawableGeo>          createPolygonDrawable(caf::DisplayCoordTransform* displayCoordTransform);
+    cvf::ref<cvf::DrawableGeo>          createStimPlanMeshDrawable(RimStimPlanFractureTemplate* stimPlanFracTemplate, caf::DisplayCoordTransform* displayCoordTransform);
+    std::vector<cvf::Vec3f>             transfromToFractureDisplayCoords(std::vector<cvf::Vec3f> polygon, cvf::Mat4f m, caf::DisplayCoordTransform* displayCoordTransform);
+    bool                                stimPlanCellTouchesPolygon(double xMin, double xMax, double yMin, double yMax);
+
     std::vector<double>                 mirrorDataAtSingleDepth(std::vector<double> depthData);
     static cvf::ref<cvf::DrawableGeo>   createGeo(const std::vector<cvf::uint>& triangleIndices, const std::vector<cvf::Vec3f>& nodeCoords);
-        
+     
+
 private:
     caf::PdmPointer<RimFracture> m_rimFracture;
     cvf::ref<cvf::Part>          m_part;
     cvf::ref<cvf::Part>          m_polygonPart;
 
+   cvf::ref<cvf::Part>           m_StimPlanMeshPart;
+
 };

ApplicationCode/ProjectDataModel/RimStimPlanFractureTemplate.cpp

@@ -59,6 +59,7 @@ RimStimPlanFractureTemplate::RimStimPlanFractureTemplate(void)
 
     CAF_PDM_InitField(&parameterForPolygon, "parameterForPolyton", QString(""), "Parameter", "", "", "");
     CAF_PDM_InitField(&timestepForPolygon, "timestepForPolygon", 0, "TimeStep", "", "", "");
+    CAF_PDM_InitField(&showStimPlanMesh, "showStimPlanMesh", true, "Show StimPlan Mesh", "", "", "")
 
 }
 
@@ -83,7 +84,7 @@ void RimStimPlanFractureTemplate::fieldChangedByUi(const caf::PdmFieldHandle* ch
         setDefaultsBasedOnXMLfile();
     }
 
-    if (&wellPathDepthAtFracture == changedField || &parameterForPolygon == changedField || &timestepForPolygon == changedField)
+    if (&wellPathDepthAtFracture == changedField || &parameterForPolygon == changedField || &timestepForPolygon == changedField || &showStimPlanMesh == changedField)
     {
         RimProject* proj;
         this->firstAncestorOrThisOfType(proj);
@@ -838,6 +839,7 @@ void RimStimPlanFractureTemplate::defineUiOrdering(QString uiConfigName, caf::Pd
     RimFractureTemplate::defineUiOrdering(uiConfigName, uiOrdering);
 
     uiOrdering.add(&name);
+    uiOrdering.add(&showStimPlanMesh);
 
     caf::PdmUiGroup* fileGroup = uiOrdering.addNewGroup("File");
     fileGroup->add(&m_stimPlanFileName);

ApplicationCode/ProjectDataModel/RimStimPlanFractureTemplate.h

@@ -52,6 +52,7 @@ public:
 
     caf::PdmField<QString>                  parameterForPolygon;
     caf::PdmField<int>                      timestepForPolygon;
+    caf::PdmField<bool>                     showStimPlanMesh;
     
     virtual void                            fieldChangedByUi(const caf::PdmFieldHandle* changedField, const QVariant& oldValue, const QVariant& newValue) override;