parallel_sort.h

Go to the documentation of this file.

// The libMesh Finite Element Library.
// Copyright (C) 2002-2024 Benjamin S. Kirk, John W. Peterson, Roy H. Stogner

// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.

// 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA


#ifndef LIBMESH_PARALLEL_SORT_H
#define LIBMESH_PARALLEL_SORT_H

// Local Includes
#include "libmesh/libmesh_common.h"
#include "libmesh/parallel_object.h"

// C++ Includes
#include <vector>

namespace libMesh
{

namespace Parallel
{

// Forward declarations
class Communicator;

template <typename KeyType, typename IdxType=unsigned int>
class Sort : public ParallelObject
{
public:
  Sort (const Parallel::Communicator & comm,
        std::vector<KeyType> & d);


  void sort();

  const std::vector<KeyType> & bin();

private:

  const processor_id_type _n_procs;

  const processor_id_type _proc_id;

  bool _bin_is_sorted;

  std::vector<KeyType> & _data;

  std::vector<IdxType> _local_bin_sizes;

  std::vector<KeyType> _my_bin;

  void binsort ();

  void communicate_bins();

  void sort_local_bin();

};
}

} // namespace libMesh

#endif // LIBMESH_PARALLEL_SORT_H