nurbs.hpp

Go to the documentation of this file.

// Copyright (c) 2010, Lawrence Livermore National Security, LLC. Produced at
// the Lawrence Livermore National Laboratory. LLNL-CODE-443211. All Rights
// reserved. See file COPYRIGHT for details.
//
// This file is part of the MFEM library. For more information and source code
// availability see http://mfem.googlecode.com.
//
// MFEM is free software; you can redistribute it and/or modify it under the
// terms of the GNU Lesser General Public License (as published by the Free
// Software Foundation) version 2.1 dated February 1999.

#ifndef MFEM_NURBS
#define MFEM_NURBS

#include "../config/config.hpp"
#include "../general/table.hpp"
#include "../general/communication.hpp"
#include "../linalg/vector.hpp"
#include "element.hpp"
#include "mesh.hpp"
#include <iostream>

namespace mfem
{

class GridFunction;


class KnotVector
{
protected:
   static const int MaxOrder;

   Vector knot;
   int Order, NumOfControlPoints, NumOfElements;

public:
   KnotVector() { }
   KnotVector(std::istream &input);
   KnotVector(int Order_, int NCP);
   KnotVector(const KnotVector &kv){ (*this) = kv; }

   KnotVector &operator=(const KnotVector &kv);

   int GetNE()    const { return NumOfElements; }
   int GetNKS()   const { return NumOfControlPoints - Order; }
   int GetNCP()   const { return NumOfControlPoints; }
   int GetOrder() const { return Order; }
   int Size()     const { return knot.Size(); }

   void GetElements();

   bool isElement(int i) const { return (knot(Order+i) != knot(Order+i+1)); }

   double getKnotLocation(double xi, int ni) const
   { return (xi*knot(ni+1) + (1. - xi)*knot(ni)); }

   int findKnotSpan(double u) const;

   void CalcShape (Vector &shape, int i, double xi);
   void CalcDShape(Vector &grad,  int i, double xi);

   void Difference(const KnotVector &kv, Vector &diff) const;
   void UniformRefinement(Vector &newknots) const;
   KnotVector *DegreeElevate(int t) const;

   void Flip();

   void Print(std::ostream &out) const;

   ~KnotVector() { }

   double &operator[](int i) { return knot(i); }
   const double &operator[](int i) const { return knot(i); }
};


class NURBSPatch
{
protected:
   int     ni, nj, nk, Dim;
   double *data;

   Array<KnotVector *> kv;

   int sd, nd;

   void swap(NURBSPatch *np);

   // Special B-NET acces functions
   int SetLoopDirection(int dir);
   inline       double &operator()(int i, int j);
   inline const double &operator()(int i, int j) const;

   void init(int dim_);

   NURBSPatch(NURBSPatch *parent, int dir, int Order, int NCP);

public:
   NURBSPatch(std::istream &input);
   NURBSPatch(KnotVector *kv0, KnotVector *kv1, int dim_);
   NURBSPatch(KnotVector *kv0, KnotVector *kv1, KnotVector *kv2, int dim_);
   NURBSPatch(Array<KnotVector *> &kv, int dim_);

   ~NURBSPatch();

   void Print(std::ostream &out);

   void DegreeElevate(int dir, int t);
   void KnotInsert   (int dir, const KnotVector &knot);
   void KnotInsert   (int dir, const Vector     &knot);

   void KnotInsert(Array<KnotVector *> &knot);
   void DegreeElevate(int t);
   void UniformRefinement();

   KnotVector *GetKV(int i) { return kv[i]; }

   // Standard B-NET acces functions
   inline       double &operator()(int i, int j, int l);
   inline const double &operator()(int i, int j, int l) const;

   inline       double &operator()(int i, int j, int k, int l);
   inline const double &operator()(int i, int j, int k, int l) const;

   static void Get3DRotationMatrix(double n[], double angle, double r,
                                   DenseMatrix &T);
   void FlipDirection(int dir);
   void SwapDirections(int dir1, int dir2);
   void Rotate3D(double normal[], double angle);
   int MakeUniformDegree();
   friend NURBSPatch *Interpolate(NURBSPatch &p1, NURBSPatch &p2);
   friend NURBSPatch *Revolve3D(NURBSPatch &patch, double n[], double ang,
                                int times);
};


#ifdef MFEM_USE_MPI
class ParNURBSExtension;
#endif

class NURBSPatchMap;


class NURBSExtension
{
#ifdef MFEM_USE_MPI
   friend class ParNURBSExtension;
#endif
   friend class NURBSPatchMap;

protected:
   int Order;
   int NumOfKnotVectors;
   // global entity counts
   int NumOfVertices, NumOfElements, NumOfBdrElements, NumOfDofs;
   // local entity counts
   int NumOfActiveVertices, NumOfActiveElems, NumOfActiveBdrElems;
   int NumOfActiveDofs;

   Array<int>  activeVert; // activeVert[glob_vert] = loc_vert or -1
   Array<bool> activeElem;
   Array<bool> activeBdrElem;
   Array<int>  activeDof; // activeDof[glob_dof] = loc_dof + 1 or 0

   Mesh *patchTopo;
   int own_topo;
   Array<int> edge_to_knot;
   Array<KnotVector *> knotVectors;
   Vector weights;

   // global offsets, meshOffsets == meshVertexOffsets
   Array<int> v_meshOffsets;
   Array<int> e_meshOffsets;
   Array<int> f_meshOffsets;
   Array<int> p_meshOffsets;

   // global offsets, spaceOffsets == dofOffsets
   Array<int> v_spaceOffsets;
   Array<int> e_spaceOffsets;
   Array<int> f_spaceOffsets;
   Array<int> p_spaceOffsets;

   Table *el_dof, *bel_dof;

   Array<int> el_to_patch;
   Array<int> bel_to_patch;
   Array2D<int> el_to_IJK;  // IJK are "knot-span" indices!
   Array2D<int> bel_to_IJK; // they are NOT element indices!

   Array<NURBSPatch *> patches;

   inline int         KnotInd(int edge);
   inline KnotVector *KnotVec(int edge);
   inline KnotVector *KnotVec(int edge, int oedge, int *okv);

   void CheckPatches();
   void CheckBdrPatches();

   void GetPatchKnotVectors   (int p, Array<KnotVector *> &kv);
   void GetBdrPatchKnotVectors(int p, Array<KnotVector *> &kv);

   // also count the global NumOfVertices and the global NumOfDofs
   void GenerateOffsets();
   // count the global NumOfElements
   void CountElements();
   // count the global NumOfBdrElements
   void CountBdrElements();

   // generate the mesh elements
   void Get2DElementTopo(Array<Element *> &elements);
   void Get3DElementTopo(Array<Element *> &elements);

   // generate the boundary mesh elements
   void Get2DBdrElementTopo(Array<Element *> &boundary);
   void Get3DBdrElementTopo(Array<Element *> &boundary);


   // FE space generation functions

   // based on activeElem, count NumOfActiveDofs, generate el_dof,
   // el_to_patch, el_to_IJK, activeDof map (global-to-local)
   void GenerateElementDofTable();

   // generate elem_to_global-dof table for the active elements
   // define el_to_patch, el_to_IJK, activeDof (as bool)
   void Generate2DElementDofTable();
   void Generate3DElementDofTable();

   // call after GenerateElementDofTable
   void GenerateBdrElementDofTable();

   // generate the bdr-elem_to_global-dof table for the active bdr. elements
   // define bel_to_patch, bel_to_IJK
   void Generate2DBdrElementDofTable();
   void Generate3DBdrElementDofTable();

   // Patch <--> FE translation functions
   void GetPatchNets  (const Vector &Nodes);
   void Get2DPatchNets(const Vector &Nodes);
   void Get3DPatchNets(const Vector &Nodes);

   void SetSolutionVector  (Vector &Nodes);
   void Set2DSolutionVector(Vector &Nodes);
   void Set3DSolutionVector(Vector &Nodes);

   // determine activeVert, NumOfActiveVertices from the activeElem array
   void GenerateActiveVertices();

   // determine activeBdrElem, NumOfActiveBdrElems
   void GenerateActiveBdrElems();

   void MergeWeights(Mesh *mesh_array[], int num_pieces);

   // to be used by ParNURBSExtension constructor(s)
   NURBSExtension() { }

public:
   NURBSExtension(std::istream &input);
   NURBSExtension(NURBSExtension *parent, int Order);
   NURBSExtension(Mesh *mesh_array[], int num_pieces);

   void MergeGridFunctions(GridFunction *gf_array[], int num_pieces,
                           GridFunction &merged);

   virtual ~NURBSExtension();

   // Print functions
   void Print(std::ostream &out) const;
   void PrintCharacteristics(std::ostream &out);

   // Meta data functions
   int Dimension() { return patchTopo->Dimension(); }
   int GetNP()     { return patchTopo->GetNE(); }
   int GetNBP()    { return patchTopo->GetNBE(); }
   int GetOrder()  { return Order; }
   int GetNKV()    { return NumOfKnotVectors; }

   int GetGNV()  { return NumOfVertices; }
   int GetNV()   { return NumOfActiveVertices; }
   int GetGNE()  { return NumOfElements; }
   int GetNE()   { return NumOfActiveElems; }
   int GetGNBE() { return NumOfBdrElements; }
   int GetNBE()  { return NumOfActiveBdrElems; }

   int GetNTotalDof() { return NumOfDofs; }
   int GetNDof()      { return NumOfActiveDofs; }

   // Knotvector acces function
   const KnotVector *GetKnotVector(int i) const { return knotVectors[i]; }

   // Mesh generation functions
   void GetElementTopo   (Array<Element *> &elements);
   void GetBdrElementTopo(Array<Element *> &boundary);

   bool HavePatches() { return (patches.Size() != 0); }

   Table *GetElementDofTable() { return el_dof; }
   Table *GetBdrElementDofTable() { return bel_dof; }

   void GetVertexLocalToGlobal(Array<int> &lvert_vert);
   void GetElementLocalToGlobal(Array<int> &lelem_elem);

   // Load functions
   void LoadFE(int i, const FiniteElement *FE);
   void LoadBE(int i, const FiniteElement *BE);

   const Vector &GetWeights() const { return  weights; }
   Vector       &GetWeights()       { return  weights; }

   // Translation functions: from FE coordinates into to IJK patch
   // format and vice versa
   void ConvertToPatches(const Vector &Nodes);
   void SetKnotsFromPatches();
   void SetCoordsFromPatches(Vector &Nodes);

   // Refinement methods
   void DegreeElevate(int t);
   void UniformRefinement();
   void KnotInsert(Array<KnotVector *> &kv);
};


#ifdef MFEM_USE_MPI
class ParNURBSExtension : public NURBSExtension
{
private:
   int *partitioning;

   Table *GetGlobalElementDofTable();
   Table *Get2DGlobalElementDofTable();
   Table *Get3DGlobalElementDofTable();

   void SetActive(int *partitioning, const Array<bool> &active_bel);
   void BuildGroups(int *partitioning, const Table &elem_dof);

public:
   GroupTopology gtopo;

   Array<int> ldof_group;

   ParNURBSExtension(MPI_Comm comm, NURBSExtension *parent, int *partitioning,
                     const Array<bool> &active_bel);

   // create a parallel version of 'parent' with partitioning as in
   // 'par_parent'; the 'parent' object is destroyed
   ParNURBSExtension(NURBSExtension *parent, ParNURBSExtension *par_parent);

   virtual ~ParNURBSExtension() { delete [] partitioning; }
};
#endif


class NURBSPatchMap
{
private:
   NURBSExtension *Ext;

   int I, J, K, pOffset, opatch;
   Array<int> verts, edges, faces, oedge, oface;

   inline static int F(const int n, const int N)
   { return (n < 0) ? 0 : ((n >= N) ? 2 : 1); }

   inline static int Or1D(const int n, const int N, const int Or)
   { return (Or > 0) ? n : (N - 1 - n); }

   inline static int Or2D(const int n1, const int n2,
                          const int N1, const int N2, const int Or);

   // also set verts, edges, faces, orientations etc
   void GetPatchKnotVectors   (int p, KnotVector *kv[]);
   void GetBdrPatchKnotVectors(int p, KnotVector *kv[], int *okv);

public:
   NURBSPatchMap(NURBSExtension *ext) { Ext = ext; }

   int nx() { return I + 1; }
   int ny() { return J + 1; }
   int nz() { return K + 1; }

   void SetPatchVertexMap(int p, KnotVector *kv[]);
   void SetPatchDofMap   (int p, KnotVector *kv[]);

   void SetBdrPatchVertexMap(int p, KnotVector *kv[], int *okv);
   void SetBdrPatchDofMap   (int p, KnotVector *kv[], int *okv);

   inline int operator()(const int i) const;
   inline int operator[](const int i) const { return (*this)(i); }

   inline int operator()(const int i, const int j) const;

   inline int operator()(const int i, const int j, const int k) const;
};


// Inline function implementations

inline double &NURBSPatch::operator()(int i, int j)
{
   return data[j%sd + sd*(i + (j/sd)*nd)];
}

inline const double &NURBSPatch::operator()(int i, int j) const
{
   return data[j%sd + sd*(i + (j/sd)*nd)];
}

inline double &NURBSPatch::operator()(int i, int j, int l)
{
#ifdef MFEM_DEBUG
   if (data == 0 || i < 0 || i >= ni || j < 0 || j >= nj || nk > 0 ||
       l < 0 || l >= Dim)
      mfem_error("NURBSPatch::operator() 2D");
#endif

   return data[(i+j*ni)*Dim+l];
}

inline const double &NURBSPatch::operator()(int i, int j, int l) const
{
#ifdef MFEM_DEBUG
   if (data == 0 || i < 0 || i >= ni || j < 0 || j >= nj || nk > 0 ||
       l < 0 || l >= Dim)
      mfem_error("NURBSPatch::operator() const 2D");
#endif

   return data[(i+j*ni)*Dim+l];
}

inline double &NURBSPatch::operator()(int i, int j, int k, int l)
{
#ifdef MFEM_DEBUG
   if (data == 0 || i < 0 || i >= ni || j < 0 || j >= nj || k < 0 ||
       k >= nk || l < 0 || l >= Dim)
      mfem_error("NURBSPatch::operator() 3D");
#endif

   return data[(i+(j+k*nj)*ni)*Dim+l];
}

inline const double &NURBSPatch::operator()(int i, int j, int k, int l) const
{
#ifdef MFEM_DEBUG
   if (data == 0 || i < 0 || i >= ni || j < 0 || j >= nj || k < 0 ||
       k >= nk ||  l < 0 || l >= Dim)
      mfem_error("NURBSPatch::operator() const 3D");
#endif

   return data[(i+(j+k*nj)*ni)*Dim+l];
}


inline int NURBSExtension::KnotInd(int edge)
{
   int kv = edge_to_knot[edge];
   return (kv >= 0) ? kv : (-1-kv);
}

inline KnotVector *NURBSExtension::KnotVec(int edge)
{
   return knotVectors[KnotInd(edge)];
}

inline KnotVector *NURBSExtension::KnotVec(int edge, int oedge, int *okv)
{
   int kv = edge_to_knot[edge];
   if (kv >= 0)
   {
      *okv = oedge;
      return knotVectors[kv];
   }
   else
   {
      *okv = -oedge;
      return knotVectors[-1-kv];
   }
}


inline int NURBSPatchMap::Or2D(const int n1, const int n2,
                               const int N1, const int N2, const int Or)
{
   // Needs testing
   switch (Or)
   {
   case 0: return n1 + n2*N1;
   case 1: return n2 + n1*N2;
   case 2: return n2 + (N1 - 1 - n1)*N2;
   case 3: return (N1 - 1 - n1) + n2*N1;
   case 4: return (N1 - 1 - n1) + (N2 - 1 - n2)*N1;
   case 5: return (N2 - 1 - n2) + (N1 - 1 - n1)*N2;
   case 6: return (N2 - 1 - n2) + n1*N2;
   case 7: return n1 + (N2 - 1 - n2)*N1;
   }
#ifdef MFEM_DEBUG
   mfem_error("NURBSPatchMap::Or2D");
#endif
   return -1;
}

inline int NURBSPatchMap::operator()(const int i) const
{
   int i1 = i - 1;
   switch (F(i1, I))
   {
   case 0: return verts[0];
   case 1: return pOffset + Or1D(i1, I, opatch);
   case 2: return verts[1];
   }
#ifdef MFEM_DEBUG
   mfem_error("NURBSPatchMap::operator() const 1D");
#endif
   return -1;
}

inline int NURBSPatchMap::operator()(const int i, const int j) const
{
   int i1 = i - 1, j1 = j - 1;
   switch (3*F(j1, J) + F(i1, I))
   {
   case 0: return verts[0];
   case 1: return edges[0] + Or1D(i1, I, oedge[0]);
   case 2: return verts[1];
   case 3: return edges[3] + Or1D(j1, J, -oedge[3]);
   case 4: return pOffset + Or2D(i1, j1, I, J, opatch);
   case 5: return edges[1] + Or1D(j1, J, oedge[1]);
   case 6: return verts[3];
   case 7: return edges[2] + Or1D(i1, I, -oedge[2]);
   case 8: return verts[2];
   }
#ifdef MFEM_DEBUG
   mfem_error("NURBSPatchMap::operator() const 2D");
#endif
   return -1;
}

inline int NURBSPatchMap::operator()(const int i, const int j, const int k)
   const
{
   // Needs testing
   int i1 = i - 1, j1 = j - 1, k1 = k - 1;
   switch (3*(3*F(k1, K) + F(j1, J)) + F(i1, I))
   {
   case  0: return verts[0];
   case  1: return edges[0] + Or1D(i1, I, oedge[0]);
   case  2: return verts[1];
   case  3: return edges[3] + Or1D(j1, J, oedge[3]);
   case  4: return faces[0] + Or2D(i1, J - 1 - j1, I, J, oface[0]);
   case  5: return edges[1] + Or1D(j1, J, oedge[1]);
   case  6: return verts[3];
   case  7: return edges[2] + Or1D(i1, I, oedge[2]);
   case  8: return verts[2];
   case  9: return edges[8] + Or1D(k1, K, oedge[8]);
   case 10: return faces[1] + Or2D(i1, k1, I, K, oface[1]);
   case 11: return edges[9] + Or1D(k1, K, oedge[9]);
   case 12: return faces[4] + Or2D(J - 1 - j1, k1, J, K, oface[4]);
   case 13: return pOffset + I*(J*k1 + j1) + i1;
   case 14: return faces[2] + Or2D(j1, k1, J, K, oface[2]);
   case 15: return edges[11] + Or1D(k1, K, oedge[11]);
   case 16: return faces[3] + Or2D(I - 1 - i1, k1, I, K, oface[3]);
   case 17: return edges[10] + Or1D(k1, K, oedge[10]);
   case 18: return verts[4];
   case 19: return edges[4] + Or1D(i1, I, oedge[4]);
   case 20: return verts[5];
   case 21: return edges[7] + Or1D(j1, J, oedge[7]);
   case 22: return faces[5] + Or2D(i1, j1, I, J, oface[5]);
   case 23: return edges[5] + Or1D(j1, J, oedge[5]);
   case 24: return verts[7];
   case 25: return edges[6] + Or1D(i1, I, oedge[6]);
   case 26: return verts[6];
   }
#ifdef MFEM_DEBUG
   mfem_error("NURBSPatchMap::operator() const 3D");
#endif
   return -1;
}

}

#endif