C++ (Cpp) Expr Examples

Example #1

File: Sacado_ELRCacheFad_SFadImp.hpp Project: gitter-badger/quinoa

inline Sacado::ELRCacheFad::Expr< Sacado::ELRCacheFad::SFadExprTag<T,Num> >& 
Sacado::ELRCacheFad::Expr< Sacado::ELRCacheFad::SFadExprTag<T,Num> >::
operator /= (const Sacado::ELRCacheFad::Expr<S>& x)
{
  x.cache();

  T xval = x.val();

#ifdef SACADO_DEBUG
  if (x.size() != Num)
    throw "ELRCacheFad::operator/=() Error:  Attempt to assign with incompatible sizes";
#endif

  // Number of arguments
  const int N = Expr<S>::num_args;

  // Compute partials
  LocalAccumOp< Expr<S> > op(x);

  T xval2 = xval*xval;
    
  // Compute each tangent direction
  for(int i=0; i<Num; ++i) {
    op.t = T(0.);
    op.i = i;
    
    // Automatically unrolled loop that computes
    // for (int j=0; j<N; j++)
    //   op.t += op.partials[j] * x.getTangent<j>(i);
    Sacado::mpl::for_each< mpl::range_c< int, 0, N > > f(op);
    
    dx_[i] = (dx_[i] * xval - val_ * op.t) / xval2;
  }

  // Compute value
  update_val_ = x.updateValue();
  if (update_val_)
    val_ /= xval;

  return *this;
}

Example #2

File: Sacado_ELRCacheFad_GeneralFadImp.hpp Project: jgoldfar/trilinos

inline Sacado::ELRCacheFad::GeneralFad<T,Storage>& 
Sacado::ELRCacheFad::GeneralFad<T,Storage>::
operator /= (const Sacado::ELRCacheFad::Expr<S>& x)
{
  x.cache();

  int xsz = x.size(), sz = this->size();
  update_val_ = x.updateValue();
  T xval = x.val();
  T v = this->val();

#ifdef SACADO_DEBUG
  if ((xsz != sz) && (xsz != 0) && (sz != 0))
    throw "Fad Error:  Attempt to assign with incompatible sizes";
#endif

  if (Expr<S>::is_linear) {
    if (xsz) {
      if (sz) {
	if (x.hasFastAccess())
	  for(int i=0; i<sz; ++i)
	    this->fastAccessDx(i) = ( this->fastAccessDx(i)*xval - v*x.fastAccessDx(i) )/ (xval*xval);
	else
	  for (int i=0; i<sz; ++i)
	    this->fastAccessDx(i) = ( this->fastAccessDx(i)*xval - v*x.dx(i) )/ (xval*xval);
      }
      else {
	this->resize(xsz);
	if (x.hasFastAccess())
	  for(int i=0; i<xsz; ++i)
	    this->fastAccessDx(i) = - v*x.fastAccessDx(i) / (xval*xval);
	else
	  for (int i=0; i<xsz; ++i)
	    this->fastAccessDx(i) = -v*x.dx(i) / (xval*xval);
      }
    }
    else {
      if (sz) {
	for (int i=0; i<sz; ++i)
	  this->fastAccessDx(i) /= xval;
      }
    }
  }
  else {

    // Number of arguments
    const int N = Expr<S>::num_args;

    if (xsz) {

      T xval2 = xval*xval;
      
      if (sz) {

	if (x.hasFastAccess()) {
	  // Compute partials
	  FastLocalAccumOp< Expr<S> > op(x);
	  
	  // Compute each tangent direction
	  for(op.i=0; op.i<xsz; ++op.i) {
	    op.t = T(0.);
	    
	    // Automatically unrolled loop that computes
	    // for (int j=0; j<N; j++)
	    //   op.t += op.partials[j] * x.getTangent<j>(i);
	    Sacado::mpl::for_each< mpl::range_c< int, 0, N > > f(op);
	  
	    this->fastAccessDx(op.i) = 
	      (this->fastAccessDx(op.i) * xval - v * op.t) / xval2;
	  }
	}
	else {
	  // Compute partials
	  SlowLocalAccumOp< Expr<S> > op(x);
	  
	  // Compute each tangent direction
	  for(op.i=0; op.i<xsz; ++op.i) {
	    op.t = T(0.);
	    
	    // Automatically unrolled loop that computes
	    // for (int j=0; j<N; j++)
	    //   op.t += op.partials[j] * x.getTangent<j>(i);
	    Sacado::mpl::for_each< mpl::range_c< int, 0, N > > f(op);
	  
	    this->fastAccessDx(op.i) = 
	      (this->fastAccessDx(op.i) * xval - v * op.t) / xval2;
	  }
	}
	
      }
      
      else {
	
	this->resize(xsz);
	
	if (x.hasFastAccess()) {
	  // Compute partials
	  FastLocalAccumOp< Expr<S> > op(x);
	  
	  // Compute each tangent direction
	  for(op.i=0; op.i<xsz; ++op.i) {
	    op.t = T(0.);
	    
	    // Automatically unrolled loop that computes
	    // for (int j=0; j<N; j++)
	    //   op.t += op.partials[j] * x.getTangent<j>(i);
	    Sacado::mpl::for_each< mpl::range_c< int, 0, N > > f(op);
	  
	    this->fastAccessDx(op.i) = (-v * op.t) / xval2;
	  }
	}
	else {
	  // Compute partials
	  SlowLocalAccumOp< Expr<S> > op(x);
	  
	  // Compute each tangent direction
	  for(op.i=0; op.i<xsz; ++op.i) {
	    op.t = T(0.);
	    
	    // Automatically unrolled loop that computes
	    // for (int j=0; j<N; j++)
	    //   op.t += op.partials[j] * x.getTangent<j>(i);
	    Sacado::mpl::for_each< mpl::range_c< int, 0, N > > f(op);
	  
	    this->fastAccessDx(op.i) = (-v * op.t) / xval2;
	  }
	}
	
      }

    }

    else {

      if (sz) {
	for (int i=0; i<sz; ++i)
	  this->fastAccessDx(i) /= xval;
      }

    }

  }

  if (update_val_)
    this->val() /= xval;

  return *this;
}