Sacado_Fad_Vector.hpp

Go to the documentation of this file.

// $Id$ 
// $Source$ 
// @HEADER
// ***********************************************************************
// 
//                           Sacado Package
//                 Copyright (2006) Sandia Corporation
// 
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
// the U.S. Government retains certain rights in this software.
// 
// This library is free software; you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation; either version 2.1 of the
// License, or (at your option) any later version.
//  
// This library is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//  
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301
// USA
// Questions? Contact David M. Gay (dmgay@sandia.gov) or Eric T. Phipps
// (etphipp@sandia.gov).
// 
// ***********************************************************************
// @HEADER

#ifndef SACADO_FAD_VECTOR_HPP
#define SACADO_FAD_VECTOR_HPP

#include <vector>
#include "Sacado_Fad_DVFad.hpp"

namespace Sacado {

  namespace Fad {

    enum VectorDerivOrientation {
      Row,      
      Column    
    };

    template <typename OrdinalType, typename FadType >
    class Vector {
    public:

      typedef typename Sacado::ValueType<FadType>::type ValueType;

      Vector(OrdinalType vec_size, OrdinalType deriv_sz,
       VectorDerivOrientation orient = Row) :
  deriv_size_(deriv_sz), vec_(vec_size) {
  for (OrdinalType i=0; i<vec_size; i++)
    vec_[i] = FadType(deriv_size_, ValueType(0.0));
      }

      Vector(const Vector& fv) : deriv_size_(fv.deriv_size_), vec_(fv.vec_) {}

      ~Vector() {}

      Vector& operator=(const Vector& fv) {
  deriv_size_ = fv.deriv_size_;
  vec_ = fv.vec_;
  return *this;
      }

      OrdinalType size() const { return vec_.size(); }

      OrdinalType deriv_size() const { return deriv_size_; }

      OrdinalType deriv_stride() const { return 1; }

      VectorDerivOrientation deriv_orientation() const { return Column; }

      FadType& operator[] (OrdinalType i) { return vec_[i]; }

      const FadType& operator[](OrdinalType i) const { return vec_[i]; }

    protected:

      OrdinalType deriv_size_;

      std::vector<FadType> vec_;

    }; // class Vector

    template <typename OrdinalType, typename ValueType>
    class Vector< OrdinalType, Sacado::Fad::DVFad<ValueType> > {
    public:

      typedef Sacado::Fad::DVFad<ValueType> FadType;

      Vector(OrdinalType vec_size, OrdinalType deriv_size, 
       VectorDerivOrientation orient = Row);

      Vector(const Vector& fv);

      ~Vector();

      Vector& operator=(const Vector& fv);

      OrdinalType size() const { return vec_.size(); }

      OrdinalType deriv_size() const { return deriv_size_; }

      OrdinalType deriv_stride() const { return stride_; }

      VectorDerivOrientation deriv_orientation() const { return orient_; }

      FadType& operator[] (OrdinalType i) { return vec_[i]; }

      const FadType& operator[](OrdinalType i) const { return vec_[i]; }

      ValueType* vals();

      const ValueType* vals() const;

      ValueType* dx();

      const ValueType* dx() const;

    protected:

      OrdinalType deriv_size_;

      VectorDerivOrientation orient_;

      OrdinalType stride_;

      std::vector<FadType> vec_;

    }; // class Vector

  } // namespace Fad

} // namespace Sacado

#include "Sacado_Fad_VectorImp.hpp"

#endif // SACADO_FAD_VECTOR_HPP