RTOpPack_TOpEleWiseDivide.hpp

// @HEADER
// *****************************************************************************
// RTOp: Interfaces and Support Software for Vector Reduction Transformation
//       Operations
//
// Copyright 2006 NTESS and the RTOp contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef RTOPPACK_TOP_ELE_WISE_DIVIDE_HPP
#define RTOPPACK_TOP_ELE_WISE_DIVIDE_HPP

#include "RTOpPack_RTOpTHelpers.hpp"


namespace RTOpPack {


template<class Scalar>
class TOpEleWiseDivideEleWiseTransformation
{
public:
  TOpEleWiseDivideEleWiseTransformation( const Scalar &alpha )
    : alpha_(alpha)
    {}
  void operator()( const Scalar &v0, const Scalar &v1, Scalar &z0 ) const
    {
      z0 += alpha_ * v0 / v1;
    }
private:
  Scalar alpha_;
};


template<class Scalar>
class TOpEleWiseDivide
  : public TOp_2_1_Base<Scalar, TOpEleWiseDivideEleWiseTransformation<Scalar> >
{
public:
  typedef TOp_2_1_Base<Scalar, TOpEleWiseDivideEleWiseTransformation<Scalar> > base_t;
  TOpEleWiseDivide( const Scalar &alpha )
    : base_t(TOpEleWiseDivideEleWiseTransformation<Scalar>(alpha))
    {
      this->setOpNameBase("TOpEleWiseDivide");
    }
};


} // namespace RTOpPack


#endif // RTOPPACK_TOP_ELE_WISE_DIVIDE_HPP