RTOp Package Browser (Single Doxygen Collection) Version of the Day
RTOpPack_TOpSetElement.hpp
Go to the documentation of this file.
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 // Redistribution and use in source and binary forms, with or without
00012 // modification, are permitted provided that the following conditions are
00013 // met:
00014 //
00015 // 1. Redistributions of source code must retain the above copyright
00016 // notice, this list of conditions and the following disclaimer.
00017 //
00018 // 2. Redistributions in binary form must reproduce the above copyright
00019 // notice, this list of conditions and the following disclaimer in the
00020 // documentation and/or other materials provided with the distribution.
00021 //
00022 // 3. Neither the name of the Corporation nor the names of the
00023 // contributors may be used to endorse or promote products derived from
00024 // this software without specific prior written permission.
00025 //
00026 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
00027 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
00028 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
00029 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
00030 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
00031 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
00032 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
00033 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
00034 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
00035 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
00036 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
00037 //
00038 // Questions? Contact Roscoe A. Bartlett (rabartl@sandia.gov) 
00039 // 
00040 // ***********************************************************************
00041 // @HEADER
00042 
00043 #ifndef RTOPPACK_TOP_SET_ELEMENT_HPP
00044 #define RTOPPACK_TOP_SET_ELEMENT_HPP
00045 
00046 #include "RTOpPack_RTOpTHelpers.hpp"
00047 #include "Teuchos_as.hpp"
00048 
00049 
00050 namespace RTOpPack {
00051 
00052 
00054 template<class Scalar>
00055 class TOpSetElementEleWiseTransformation
00056 {
00057 public:
00059   TOpSetElementEleWiseTransformation( const Ordinal &global_i_in = -1,
00060     const Scalar &val_i_in = static_cast<Scalar>(0.0) )
00061     :global_i_(global_i_in), val_i_(val_i_in)
00062     {}
00064   Ordinal global_i() const
00065     {
00066       return global_i_;
00067     }
00069   void operator()( const Ordinal global_i_in, Scalar &z0 ) const
00070     {
00071       if (global_i_in == global_i_) {
00072         z0 = val_i_;
00073       }
00074     }
00075 private:
00076   Ordinal global_i_;
00077   Scalar val_i_;
00078 };
00079 
00080 
00083 template<class Scalar>
00084 class TOpSetElement
00085   : public TOp_0_1_CoordVariantBase<Scalar, TOpSetElementEleWiseTransformation<Scalar> >
00086 {
00087 public:
00089   TOpSetElement(const Ordinal &global_i_in = -1,
00090     const Scalar &val_i_in = static_cast<Scalar>(0.0))
00091     {
00092       this->setOpNameBase("TOpSetElement");
00093       this->setEleWiseTransformation(
00094         TOpSetElementEleWiseTransformation<Scalar>(global_i_in, val_i_in));
00095     }
00097   void initialize(const Ordinal &global_i_in, const Scalar &val_i_in)
00098     { 
00099       this->setEleWiseTransformation(
00100         TOpSetElementEleWiseTransformation<Scalar>(global_i_in, val_i_in));
00101     }
00102 protected:
00104   virtual Range1D range_impl() const
00105     {
00106       const Ordinal i = this->getEleWiseTransformation().global_i();
00107       return Range1D(i, i);
00108     }
00109 };
00110 
00111 
00112 } // namespace RTOpPack
00113 
00114 
00115 #endif // RTOPPACK_TOP_SET_ELEMENT_HPP
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines