Partition.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
 */

 
#pragma once
 
#include <arccore/base/String.h>
 
#include <alien/data/Space.h>
#include <alien/distribution/MatrixDistribution.h>
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Alien
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

class ALIEN_EXPORT Partition
{
 public:
  Partition(const ISpace& space, const MatrixDistribution& distribution);

  ~Partition() = default;

  void create(const Arccore::UniqueArray<Arccore::String>& tags);

  Arccore::Integer nbTaggedParts() const { return m_tagged_parts.size(); }

  Arccore::String tag(Arccore::Integer i) const { return m_tags[i]; }

  const Arccore::UniqueArray<Arccore::Integer>& taggedPart(Arccore::Integer i) const
  {
    return m_tagged_parts[i];
  }

  bool hasUntaggedPart() const { return !m_untagged_part.empty(); }

  const Arccore::UniqueArray<Arccore::Integer>& untaggedPart() const
  {
    return m_untagged_part;
  }
 
 private:
  const ISpace& m_space;
  const MatrixDistribution& m_distribution;
  Arccore::UniqueArray<Arccore::String> m_tags;
  Arccore::UniqueArray<Arccore::UniqueArray<Arccore::Integer>> m_tagged_parts;
  // The array of local ids for untagged entries
  Arccore::UniqueArray<Arccore::Integer> m_untagged_part;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Alien
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/