RTOpPack_ROpDotProd.hpp

00001 // @HEADER
00002 // ***********************************************************************
00003 // 
00004 // RTOp: Interfaces and Support Software for Vector Reduction Transformation
00005 //       Operations
00006 //                Copyright (2006) Sandia Corporation
00007 // 
00008 // Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
00009 // license for use of this work by or on behalf of the U.S. Government.
00010 // 
00011 // This library is free software; you can redistribute it and/or modify
00012 // it under the terms of the GNU Lesser General Public License as
00013 // published by the Free Software Foundation; either version 2.1 of the
00014 // License, or (at your option) any later version.
00015 //  
00016 // This library is distributed in the hope that it will be useful, but
00017 // WITHOUT ANY WARRANTY; without even the implied warranty of
00018 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
00019 // Lesser General Public License for more details.
00020 //  
00021 // You should have received a copy of the GNU Lesser General Public
00022 // License along with this library; if not, write to the Free Software
00023 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
00024 // USA
00025 // Questions? Contact Roscoe A. Bartlett (rabartl@sandia.gov) 
00026 // 
00027 // ***********************************************************************
00028 // @HEADER
00029 
00030 #ifndef RTOPPACK_ROP_DOT_PROD_HPP
00031 #define RTOPPACK_ROP_DOT_PROD_HPP
00032 
00033 #include "RTOpPack_RTOpTHelpers.hpp"
00034 
00035 namespace RTOpPack {
00036 
00039 template<class Scalar>
00040 class ROpDotProd : public ROpScalarReductionBase<Scalar> {
00041 public:
00043   ROpDotProd() : RTOpT<Scalar>("ROpDotProd") {}
00045   Scalar operator()(const ReductTarget& reduct_obj ) const { return this->getRawVal(reduct_obj); }
00049   void apply_op(
00050     const int   num_vecs,       const ConstSubVectorView<Scalar>    sub_vecs[]
00051     ,const int  num_targ_vecs,  const SubVectorView<Scalar>         targ_sub_vecs[]
00052     ,ReductTarget *_reduct_obj
00053     ) const
00054     {
00055       using Teuchos::dyn_cast;
00056       typedef Teuchos::ScalarTraits<Scalar> ST;
00057       ReductTargetScalar<Scalar> &reduct_obj = dyn_cast<ReductTargetScalar<Scalar> >(*_reduct_obj); 
00058       RTOP_APPLY_OP_2_0(num_vecs,sub_vecs,num_targ_vecs,targ_sub_vecs);
00059       Scalar dot_prod = reduct_obj.get();
00060       if( v0_s == 1 && v1_s == 1 ) {
00061         for( Teuchos_Index i = 0; i < subDim; ++i )
00062           dot_prod += ST::conjugate(*v0_val++)*(*v1_val++);
00063       }
00064       else {
00065         for( Teuchos_Index i = 0; i < subDim; ++i, v0_val += v0_s, v1_val += v1_s )
00066           dot_prod += ST::conjugate(*v0_val)*(*v1_val);
00067       }
00068       reduct_obj.set(dot_prod);
00069     }
00071 }; // class ROpDotProd
00072 
00073 } // namespace RTOpPack
00074 
00075 #endif // RTOPPACK_ROP_DOT_PROD_HPP

Generated on Thu Sep 18 12:30:40 2008 for Collection of Concrete Vector Reduction/Transformation Operator Implementations by doxygen 1.3.9.1