49 template <
typename ordinal_type,
typename value_type,
typename node_type>
58 sz(this->basis->size()),
60 quad_points(quad->getQuadPoints()),
61 quad_weights(quad->getQuadWeights()),
62 quad_values(quad->getBasisAtQuadPoints()),
63 norms(this->basis->norm_squared()),
64 nqp(quad_points.size()),
73 qv[qp*sz+i] = quad_values[qp][i];
74 sqv[qp*sz+i] = quad_values[qp][i]/norms[i];
78 if (params == Teuchos::null)
80 use_quad_for_times = params->
get(
"Use Quadrature for Times",
false);
81 use_quad_for_division = params->
get(
"Use Quadrature for Division",
true);
84 template <
typename ordinal_type,
typename value_type,
typename node_type>
85 template <
typename FuncT>
107 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
118 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
125 fvals[qp] = func(avals[qp])*quad_weights[qp];
133 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
144 template <
typename ordinal_type,
typename value_type,
typename node_type>
145 template <
typename FuncT>
156 if (pa == 1 && pb == 1)
164 c[0] = func(a[0], b[0]);
169 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
182 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
189 fvals[qp] = func(avals[qp], bvals[qp])*quad_weights[qp];
196 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
207 template <
typename ordinal_type,
typename value_type,
typename node_type>
208 template <
typename FuncT>
226 c[0] = func(a, b[0]);
231 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
242 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
249 fvals[qp] = func(a, bvals[qp])*quad_weights[qp];
256 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
267 template <
typename ordinal_type,
typename value_type,
typename node_type>
268 template <
typename FuncT>
286 c[0] = func(a[0], b);
291 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
302 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
309 fvals[qp] = func(avals[qp], b)*quad_weights[qp];
316 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
327 template <
typename ordinal_type,
typename value_type,
typename node_type>
328 template <
typename FuncT>
335 const int N = FuncT::N;
337 for (
int i=0; i<N; i++) {
338 if (na[i]->size() > 1) {
353 for (
int i=0; i<N; i++)
354 val[i] = (*na[i])[0];
359 if (N >= navals.size())
361 if (navals[N].size() != N) {
363 for (
int i=0; i<N; i++)
364 navals[N][i].resize(nqp);
368 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
373 for (
int i=0; i<N; i++) {
376 blas.GEMV(
Teuchos::TRANS, pa, nqp, 1.0, &qv[0], sz, na[i]->coeff(), 1, 0.0,
377 &navals[N][i][0], 1);
383 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
391 for (
int i=0; i<N; i++)
392 val[i] = navals[N][i][qp];
393 fvals[qp] = func(
val)*quad_weights[qp];
402 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
413 template <
typename ordinal_type,
typename value_type,
typename node_type>
423 template <
typename ordinal_type,
typename value_type,
typename node_type>
433 template <
typename ordinal_type,
typename value_type,
typename node_type>
440 if (use_quad_for_times)
441 binary_op(times_quad_func(), c, c,
x);
446 template <
typename ordinal_type,
typename value_type,
typename node_type>
453 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
464 if (use_quad_for_division)
465 binary_op(div_quad_func(), c, c,
x);
471 template <
typename ordinal_type,
typename value_type,
typename node_type>
478 if (use_quad_for_times)
479 binary_op(times_quad_func(), c, a, b);
484 template <
typename ordinal_type,
typename value_type,
typename node_type>
494 template <
typename ordinal_type,
typename value_type,
typename node_type>
504 template <
typename ordinal_type,
typename value_type,
typename node_type>
511 #ifdef STOKHOS_TEUCHOS_TIME_MONITOR
527 if (use_quad_for_division)
528 binary_op(div_quad_func(), c, a, b);
534 template <
typename ordinal_type,
typename value_type,
typename node_type>
541 if (use_quad_for_division)
542 binary_op(div_quad_func(), c, a, b);
547 template <
typename ordinal_type,
typename value_type,
typename node_type>
557 template <
typename ordinal_type,
typename value_type,
typename node_type>
563 unary_op(exp_quad_func(), c, a);
566 template <
typename ordinal_type,
typename value_type,
typename node_type>
572 unary_op(log_quad_func(), c, a);
575 template <
typename ordinal_type,
typename value_type,
typename node_type>
581 unary_op(log10_quad_func(), c, a);
584 template <
typename ordinal_type,
typename value_type,
typename node_type>
590 unary_op(sqrt_quad_func(), c, a);
593 template <
typename ordinal_type,
typename value_type,
typename node_type>
599 unary_op(cbrt_quad_func(), c, a);
602 template <
typename ordinal_type,
typename value_type,
typename node_type>
609 binary_op(pow_quad_func(), c, a, b);
612 template <
typename ordinal_type,
typename value_type,
typename node_type>
619 binary_op(pow_quad_func(), c, a, b);
622 template <
typename ordinal_type,
typename value_type,
typename node_type>
629 binary_op(pow_quad_func(), c, a, b);
632 template <
typename ordinal_type,
typename value_type,
typename node_type>
638 unary_op(sin_quad_func(), s, a);
641 template <
typename ordinal_type,
typename value_type,
typename node_type>
647 unary_op(cos_quad_func(), c, a);
650 template <
typename ordinal_type,
typename value_type,
typename node_type>
656 unary_op(tan_quad_func(), t, a);
659 template <
typename ordinal_type,
typename value_type,
typename node_type>
665 unary_op(sinh_quad_func(), s, a);
668 template <
typename ordinal_type,
typename value_type,
typename node_type>
674 unary_op(cosh_quad_func(), c, a);
677 template <
typename ordinal_type,
typename value_type,
typename node_type>
683 unary_op(tanh_quad_func(), t, a);
686 template <
typename ordinal_type,
typename value_type,
typename node_type>
692 unary_op(acos_quad_func(), c, a);
695 template <
typename ordinal_type,
typename value_type,
typename node_type>
701 unary_op(asin_quad_func(), c, a);
704 template <
typename ordinal_type,
typename value_type,
typename node_type>
710 unary_op(atan_quad_func(), c, a);
713 template <
typename ordinal_type,
typename value_type,
typename node_type>
720 binary_op(atan2_quad_func(), c, a, b);
723 template <
typename ordinal_type,
typename value_type,
typename node_type>
730 binary_op(atan2_quad_func(), c, a, b);
733 template <
typename ordinal_type,
typename value_type,
typename node_type>
740 binary_op(atan2_quad_func(), c, a, b);
743 template <
typename ordinal_type,
typename value_type,
typename node_type>
749 unary_op(acosh_quad_func(), c, a);
752 template <
typename ordinal_type,
typename value_type,
typename node_type>
758 unary_op(asinh_quad_func(), c, a);
761 template <
typename ordinal_type,
typename value_type,
typename node_type>
767 unary_op(atanh_quad_func(), c, a);
770 template <
typename ordinal_type,
typename value_type,
typename node_type>
771 template <
typename ExprT1,
typename ExprT2>
779 if (pa > 1 && pb > 1) {
789 "Stokhos::QuadOrthogPolyExpansion::compute_times_coeff()"
790 <<
": Index " << i <<
" is out of range [0,"
791 << this->Cijk->num_i() <<
")!");
797 k_it != this->Cijk->k_end(i_it); ++k_it) {
804 aa = a.higher_order_coeff(k);
810 aa = b.higher_order_coeff(k);
813 j_it != this->Cijk->j_end(k_it); ++j_it) {
821 bb = b.higher_order_coeff(
j);
827 bb = a.higher_order_coeff(
j);
834 return cc / norms[i];
837 return a.val() * b.val();
839 return a.higher_order_coeff(i)*b.val();
842 return a.val()*b.higher_order_coeff(i);
846 template <
typename ordinal_type,
typename value_type,
typename node_type>
847 template <
typename ExprT1,
typename ExprT2>
855 "Stokhos::QuadOrthogPolyExpansion::fast_ompute_times_coeff()"
856 <<
": Index " << i <<
" is out of range [0,"
857 << this->Cijk->num_i() <<
")!");
863 k_it != this->Cijk->k_end(i_it); ++k_it) {
868 aa = a.higher_order_coeff(k);
870 j_it != this->Cijk->j_end(k_it); ++j_it) {
876 bb = b.higher_order_coeff(
j);
881 return cc / norms[i];