RTOpPack_TOpLinearCombination_decl.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_LINEAR_COMBINATION_DECL_HPP
#define RTOPPACK_TOP_LINEAR_COMBINATION_DECL_HPP

#include "RTOpPack_RTOpTHelpers.hpp"
#include "Teuchos_Workspace.hpp"


namespace RTOpPack {


template<class Scalar>
class TOpLinearCombination : public RTOpT<Scalar> {
public:

  TOpLinearCombination(
    const ArrayView<const Scalar> &alpha_in = Teuchos::null,
    const Scalar &beta = Teuchos::ScalarTraits<Scalar>::zero()
    );

  void alpha( const ArrayView<const Scalar> &alpha_in );

  const ArrayView<const Scalar> alpha() const;

  void beta( const Scalar& beta_in );

  Scalar beta() const;

  int num_vecs() const;


  void apply_op_impl(
    const ArrayView<const ConstSubVectorView<Scalar> > &sub_vecs,
    const ArrayView<const SubVectorView<Scalar> > &targ_sub_vecs,
    const Ptr<ReductTarget> &reduct_obj_inout
    ) const;


private:

  Scalar beta_;
  Array<Scalar> alpha_;

};


} // namespace RTOpPack


#endif // RTOPPACK_TOP_LINEAR_COMBINATION_DECL_HPP