Stokhos_Sparse3Tensor.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//                           Stokhos Package
//
// Copyright 2009 NTESS and the Stokhos contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef STOKHOS_SPARSE3TENSOR_HPP
#define STOKHOS_SPARSE3TENSOR_HPP

#include <ostream>
#include <map>
#include "Stokhos_SparseArray.hpp"

namespace Stokhos {

  template <typename ordinal_type, typename value_type>
  class Sparse3Tensor {

  private:

    typedef std::map<const ordinal_type, value_type> i_map;
    typedef std::map<const ordinal_type, i_map> ji_map;
    typedef std::map<const ordinal_type, ji_map> kji_map;

    typedef std::map<const ordinal_type, value_type> j_map;
    typedef std::map<const ordinal_type, j_map> kj_map;
    typedef std::map<const ordinal_type, kj_map> ikj_map;

    typedef SparseArray<ordinal_type, value_type> i_sparse_array;
    typedef SparseArray<ordinal_type, i_sparse_array> ji_sparse_array;
    typedef SparseArray<ordinal_type, ji_sparse_array> kji_sparse_array;

    typedef SparseArray<ordinal_type, value_type> j_sparse_array;
    typedef SparseArray<ordinal_type, j_sparse_array> kj_sparse_array;
    typedef SparseArray<ordinal_type, kj_sparse_array> ikj_sparse_array;

  public:

    typedef typename kji_sparse_array::const_iterator k_iterator;

    typedef typename kji_sparse_array::const_reverse_iterator k_reverse_iterator;

    typedef typename ji_sparse_array::const_iterator kj_iterator;

    typedef typename j_sparse_array::const_iterator kji_iterator;

    typedef typename ikj_sparse_array::const_iterator i_iterator;

    typedef typename ikj_sparse_array::const_reverse_iterator i_reverse_iterator;

    typedef typename kj_sparse_array::const_iterator ik_iterator;

    typedef typename j_sparse_array::const_iterator ikj_iterator;
    
    Sparse3Tensor();
    
    ~Sparse3Tensor() {}

    void add_term(ordinal_type i, ordinal_type j, ordinal_type k,
      const value_type& c);

    void sum_term(ordinal_type i, ordinal_type j, ordinal_type k,
      const value_type& c);


    void fillComplete();

    bool fillCompleted() const;

    void print(std::ostream& os) const;

    value_type getValue(ordinal_type i, ordinal_type j, ordinal_type k) const;

    ordinal_type num_entries() const;


    ordinal_type num_k() const;

    ordinal_type num_j(const k_iterator& k) const;

    ordinal_type num_i(const kj_iterator& j) const;

    k_iterator find_k(ordinal_type k) const;

    kj_iterator find_j(const k_iterator& k, ordinal_type j) const;

    kji_iterator find_i(const kj_iterator& j, ordinal_type i) const;

    k_iterator k_begin() const;

    k_iterator k_end() const;

    k_reverse_iterator k_rbegin() const;

    k_reverse_iterator k_rend() const;

    kj_iterator j_begin(const k_iterator& k) const;

    kj_iterator j_end(const k_iterator& k) const;

    kj_iterator j_begin(const k_reverse_iterator& k) const;

    kj_iterator j_end(const k_reverse_iterator& k) const;

    kji_iterator i_begin(const kj_iterator& j) const;

    kji_iterator i_end(const kj_iterator& j) const;



    ordinal_type num_i() const;

    ordinal_type num_k(const i_iterator& i) const;

    ordinal_type num_j(const ik_iterator& k) const;

    i_iterator find_i(ordinal_type i) const;

    ik_iterator find_k(const i_iterator& i, ordinal_type k) const;

    ikj_iterator find_j(const ik_iterator& k, ordinal_type j) const;

    i_iterator i_begin() const;

    i_iterator i_end() const;

    i_reverse_iterator i_rbegin() const;

    i_reverse_iterator i_rend() const;

    ik_iterator k_begin(const i_iterator& i) const;

    ik_iterator k_end(const i_iterator& i) const;

    ik_iterator k_begin(const i_reverse_iterator& i) const;

    ik_iterator k_end(const i_reverse_iterator& i) const;

    ikj_iterator j_begin(const ik_iterator& k) const;

    ikj_iterator j_end(const ik_iterator& k) const;


  private:

    // Prohibit copying
    Sparse3Tensor(const Sparse3Tensor&);

    // Prohibit Assignment
    Sparse3Tensor& operator=(const Sparse3Tensor& b);

  protected:

    bool fill_completed;


    kji_map kji_data;

    kji_sparse_array kji_array;



    ikj_map ikj_data;

    ikj_sparse_array ikj_array;


  }; // class Sparse3Tensor

  template <typename ordinal_type, typename value_type>
  std::ostream& 
  operator << (std::ostream& os, 
         const Sparse3Tensor<ordinal_type, value_type>& Cijk) {
    Cijk.print(os);
    return os;
  }

  template <typename index_iterator, typename value_iterator>
  typename SparseArrayIterator<index_iterator, value_iterator>::value_type
  index(const SparseArrayIterator<index_iterator, value_iterator>& it) {
    return *it;
  }

  template <typename index_iterator, typename value_iterator>
  typename SparseArrayReverseIterator<index_iterator, value_iterator>::value_type
  index(const SparseArrayReverseIterator<index_iterator, value_iterator>& it)
  {
    return *it;
  }

  template <typename index_iterator, typename value_iterator>
  typename SparseArrayIterator<index_iterator, value_iterator>::value_reference
  value(const SparseArrayIterator<index_iterator, value_iterator>& it) {
    return it.value();
  }

} // namespace Stokhos

// Include template definitions
#include "Stokhos_Sparse3TensorImp.hpp"

#endif // STOKHOS_SPARSE3TENSOR_HPP