DistArray.h

Go to the documentation of this file.

//-----------------------------------------------------------------------bl-
//--------------------------------------------------------------------------
//
// QUESO - a library to support the Quantification of Uncertainty
// for Estimation, Simulation and Optimization
//
// Copyright (C) 2008-2017 The PECOS Development Team
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the Version 2.1 GNU Lesser General
// Public License as published by the Free Software Foundation.
//
// 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 Street, Fifth Floor,
// Boston, MA  02110-1301  USA
//
//-----------------------------------------------------------------------el-

#ifndef UQ_DIST_ARRAY_H
#define UQ_DIST_ARRAY_H

#include <queso/Defines.h>
#include <queso/Map.h>
#ifdef QUESO_HAS_TRILINOS
#include <EpetraExt_DistArray.h>
#endif
#include <ostream>

namespace QUESO {

template <typename T>
class DistArray
{
public:


  DistArray(const Map& inputMap,
                   const int         inputRowSize);

 ~DistArray();



        T&   operator    ()(int localElementId, int colId);

  const T&   operator    ()(int localElementId, int colId) const;

  int  GlobalLength() const;

  int  MyLength    () const;

  int  RowSize     () const;



  void print       (std::ostream& os) const;
  friend std::ostream & operator<<(std::ostream& os, const DistArray<T>& obj)
  {
    obj.print(os);

    return os;
  }

private:
  DistArray();

  DistArray(const DistArray<T>& src) : m_Map(src.m_Map) { }

  DistArray<T>& operator=(const DistArray<T>& rhs);

  Map                   m_Map;
#ifdef QUESO_HAS_TRILINOS
  EpetraExt::DistArray<T>*     m_epetraDistArray;
#else
  unsigned int                 m_rowSize;
  std::vector<std::vector<T> > m_elements;
#endif
};

}  // End namespace QUESO

#endif // UQ_DIST_ARRAY_H