@Override
 public void setDOF(DOF trialDOF, DOF testDOF) {
   this.trialDOF = trialDOF;
   this.testDOF = testDOF;
   if (trialDOF != null) {
     this.u = trialDOF.getSSF();
     this.uDOFLocalIndex = trialDOF.getLocalIndex();
   }
   if (testDOF != null) {
     this.v = testDOF.getSSF();
     this.vDOFLocalIndex = testDOF.getLocalIndex();
   }
 }
Пример #2
0
 public void imposeDirichletCondition(SparseBlockMatrix BM, SparseBlockVector BV, Function diri) {
   ElementList eList = mesh.getElementList();
   int nNode = mesh.getNodeList().size();
   for (int i = 1; i <= eList.size(); i++) {
     Element e = eList.at(i);
     DOFList DOFs = e.getAllDOFList(DOFOrder.NEFV);
     for (int j = 1; j <= DOFs.size(); j++) {
       DOF dof = DOFs.at(j);
       GeoEntity ge = dof.getOwner();
       if (ge instanceof Node) {
         Node n = (Node) ge;
         if (n.getNodeType() == NodeType.Dirichlet) {
           Variable v = Variable.createFrom(diri, n, 0);
           setDirichlet(BM, BV, dof.getGlobalIndex(), diri.value(v));
           setDirichlet(BM, BV, nNode + dof.getGlobalIndex(), diri.value(v));
           setDirichlet(BM, BV, nNode * 2 + dof.getGlobalIndex(), diri.value(v));
         }
       }
     }
   }
 }