Function ExportVisFunctions

solve/python_solve.cpp:450–955

}

void ExportVisFunctions(py::module &m) {
    m.def("_GetVisualizationData", [] (shared_ptr<ngcomp::MeshAccess> ma, map<ngfem::ELEMENT_TYPE, IntegrationRule> irs) {
            struct ElementInformation {
                ElementInformation( ngfem::ELEMENT_TYPE type_, bool curved_=false)
                  : type(type_), curved(curved_), nelements(0) { }
                Array<int> data; // the data that will go into the gpu texture buffer
                ngfem::ELEMENT_TYPE type;
                bool curved;
                int nelements;
            };

            auto getVB = [](int codim) {
                switch(codim) {
                  case 0: return VOL;
                  case 1: return BND;
                  case 2: return BBND;
                  case 3: return BBBND;
                  default: throw Exception("invalid codim");
                }
            };

            Vector<> min(3);
            min = std::numeric_limits<double>::max();
            Vector<> max(3);
            max = std::numeric_limits<double>::lowest();

            ngstd::Array<float> vertices;
            vertices.SetAllocSize(ma->GetNV()*3);
            for ( auto vi : Range(ma->GetNV()) ) {
                auto v = ma->GetPoint<3>(vi);
                for (auto i : Range(3)) {
                  vertices.Append(v[i]);
                  min[i] = min2(min[i], v[i]);
                  max[i] = max2(max[i], v[i]);
                }
            }

            LocalHeap lh(1000000, "GetMeshData");
            auto toDict = [] (ElementInformation &ei) {
              py::dict res;
              res["data"] = py::cast(ei.data);
              res["type"] = py::cast(ei.type);
              res["curved"] = py::cast(ei.curved);
              res["nelements"] = py::cast(ei.nelements);
              return res;
            };

            const int dim = ma->GetDimension();

            std::map<VorB, py::list> element_data;
            ElementInformation points(ET_POINT);
            points.nelements = ma->GetNE(getVB(dim));
            points.data.SetAllocSize(points.nelements);
            for(auto el : ma->Elements(getVB(dim)))
              points.data.Append(el.vertices[0]);
            element_data[getVB(dim)].append(toDict(points));

            ElementInformation edges(ET_SEGM);