MatrixElement.h

Go to the documentation of this file.

/*
 * Copyright 2020 IFPEN-CEA
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * SPDX-License-Identifier: Apache-2.0
 */

 
#ifndef ALIEN_COMMON_UTILS_MATRIXELEMENT_H
#define ALIEN_COMMON_UTILS_MATRIXELEMENT_H
 
#include <alien/utils/Precomp.h>
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Alien
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

template <typename Builder>
class MatrixElementT
{
 public:
  MatrixElementT(
  const Arccore::Integer iIndex, const Arccore::Integer jIndex, Builder& parent)
  : m_iIndex(iIndex)
  , m_jIndex(jIndex)
  , m_parent(parent)
  {}

  Arccore::Real operator()() const { return m_parent.getData(m_iIndex, m_jIndex); }

  void operator+=(Real value)
  {
    m_parent.addData(m_iIndex, m_jIndex, value);
  }

  void operator-=(Real value)
  {
    m_parent.addData(m_iIndex, m_jIndex, -value);
  }

  void operator=(Real value)
  {
    m_parent.setData(m_iIndex, m_jIndex, value);
  }

  template <typename Builder2>
  bool operator=(const MatrixElementT<Builder2>& other)
  {
    bool test_pattern = (m_iIndex == other.m_iIndex) && (m_jIndex == other.m_jIndex);
    // TODO: Check values.
    return test_pattern;
  }
 
 private:
  const Arccore::Integer m_iIndex;
  const Arccore::Integer m_jIndex;
  Builder& m_parent;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Alien
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif /* ALIEN_COMMON_UTILS_MATRIXELEMENT_H */