solution_indexer.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: © 2025 PRISMS Center at the University of Michigan
// SPDX-License-Identifier: GNU Lesser General Public Version 2.1
 
#pragma once
 
#include <prismspf/core/group_solution_handler.h>
 
#include <prismspf/config.h>
 
#include <vector>
 
PRISMS_PF_BEGIN_NAMESPACE

template <unsigned int dim, typename number>
class SolutionIndexer
{
public:
  SolutionIndexer() = default;

  void
  init(unsigned int                                            num_fields,
       const std::vector<GroupSolutionHandler<dim, number> *> &solution_handlers);

  [[nodiscard]] const SolutionVector<number> &
  get_solution_vector(unsigned int global_index) const;
  [[nodiscard]] SolutionVector<number> &
  get_solution_vector(unsigned int global_index);

  [[nodiscard]] const SolutionVector<number> &
  get_old_solution_vector(unsigned int age, unsigned int global_index) const;
  [[nodiscard]] SolutionVector<number> &
  get_old_solution_vector(unsigned int age, unsigned int global_index);

  [[nodiscard]] const SolveBlock &
  get_solve_block(unsigned int index) const;

  [[nodiscard]] std::pair<const SolutionLevel<dim, number> *, unsigned int>
  get_solution_level_and_block_index(unsigned int index,
                                     unsigned int relative_level = -1) const;

  [[nodiscard]] unsigned int
  get_block_index(unsigned int global_index) const;

  [[nodiscard]] std::vector<unsigned int>
  get_block_indices() const;
 
private:
  std::vector<GroupSolutionHandler<dim, number> *> solutions;
  std::vector<unsigned int>                        block_indices;
};
 
PRISMS_PF_END_NAMESPACE