Function MakeComponentCoefficientFunction

fem/coefficient.cpp:4591–4617

shared_ptr<CoefficientFunction>
MakeComponentCoefficientFunction (shared_ptr<CoefficientFunction> c1, int comp)
{
  if (c1->IsZeroCF())
    return ZeroCF( Array<int>({}) );

  if (c1->GetDescription() == "unary operation ' '"  && !c1->IsVariable())
    return MakeComponentCoefficientFunction(c1->InputCoefficientFunctions()[0], comp);

  if (c1->GetDescription() == "VectorialCoefficientFunction")
  {
    int ci = 0;
    int dim = 0;
    auto coefs = c1->InputCoefficientFunctions();
    while(dim<=comp)
    {
      int dim_last = dim;
      if(dim == comp && coefs[ci]->Dimension() == 1)
        return coefs[ci];
      dim += coefs[ci]->Dimension();
      if(dim>comp)
        return MakeComponentCoefficientFunction( coefs[ci], comp-dim_last );
      ci++;
    }
  }
  return make_shared<ComponentCoefficientFunction> (c1, comp);
}


// ********************** SubTensorCoefficientFunction **********************