/*
  This file is part of p4est.
  p4est is a C library to manage a collection (a forest) of multiple
  connected adaptive quadtrees or octrees in parallel.

  Copyright (C) 2010 The University of Texas System
  Additional copyright (C) 2011 individual authors
  Written by Carsten Burstedde, Lucas C. Wilcox, and Tobin Isaac

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

  p4est 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 General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with p4est; if not, write to the Free Software Foundation, Inc.,
  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/

/********************************************************************
 *                          IMPORTANT NOTE                          *
 *                                                                  *
 * These interfaces are intended for those who like finer control.  *
 * The API offers extended versions of some basic p4est functions.  *
 * The API may change without notice.                               *
 ********************************************************************/

/** \file p8est_extended.h
 *
 * Interface routines with extended capabilities.
 *
 * \ingroup p8est
 */

#ifndef P8EST_EXTENDED_H
#define P8EST_EXTENDED_H

#include <sc_uint128.h>
#include <p8est_mesh.h>
#include <p8est_iterate.h>
#include <p8est_lnodes.h>
#include <p8est_io.h>

SC_EXTERN_C_BEGIN;

/** A datatype to handle the linear id in 3D.
 * We use the implementation of unsigned 128 bit
 * integer in libsc, i.e., a struct with the
 * members high_bits and low_bits (both uint64_t).
 */
typedef sc_uint128_t p8est_lid_t;

/* Data pertaining to selecting, inspecting, and profiling algorithms.
 * A pointer to this structure is hooked into the p8est main structure.
 *
 * TODO: Describe the purpose of various switches, counters, and timings.
 *
 * The balance_ranges and balance_notify* times are collected
 * whenever an inspect structure is present in p8est.
 */
struct p8est_inspect
{
  /** Use sc_ranges to determine the asymmetric communication pattern.
   * If \a use_balance_ranges is false (the default), sc_notify is used. */
  int                 use_balance_ranges;
  /** If true, call both sc_ranges and sc_notify and verify consistency.
   * Which is actually used is still determined by \a use_balance_ranges. */
  int                 use_balance_ranges_notify;
  /** Verify sc_ranges and/or sc_notify as applicable. */
  int                 use_balance_verify;
  /** If positive and smaller than p8est_num ranges, overrides it */
  int                 balance_max_ranges;
  size_t              balance_A_count_in;
  size_t              balance_A_count_out;
  size_t              balance_comm_sent;
  size_t              balance_comm_nzpeers;
  size_t              balance_B_count_in;
  size_t              balance_B_count_out;
  size_t              balance_zero_sends[2], balance_zero_receives[2];
  double              balance_A;
  double              balance_comm;
  double              balance_B;
  double              balance_ranges;   /**< time spent in sc_ranges */
  double              balance_notify;   /**< time spent in sc_notify */
  /** time spent in sc_notify_allgather */
  double              balance_notify_allgather;
  int                 use_B;
};

/** Callback function prototype to replace one set of quadrants with another.
 *
 * This is used by extended routines when the quadrants of an existing, valid
 * p8est are changed.  The callback allows the user to make changes to newly
 * initialized quadrants before the quadrants that they replace are destroyed.
 *
 * \param [in] p8est        A valid forest for context.
 * \param [in] which_tree   Tree index of the invocation.
 * \param [in] num_outgoing The number of outgoing quadrants.
 * \param [in] outgoing     The outgoing quadrants: after the callback, the
 *                          user_data, if \a p8est->data_size is nonzero,
 *                          will be destroyed.
 * \param [in] num_incoming The number of incoming quadrants.
 * \param [in,out] incoming The incoming quadrants: prior to the callback,
 *                          the user_data, if \a p8est->data_size is nonzero,
 *                          is allocated, and the p8est_init_t callback,
 *                          if it has been provided, will be called.
 *
 * If the mesh is being refined, num_outgoing will be 1 and num_incoming will
 * be 8, and vice versa if the mesh is being coarsened.
 */
typedef void        (*p8est_replace_t) (p8est_t * p8est,
                                        p4est_topidx_t which_tree,
                                        int num_outgoing,
                                        p8est_quadrant_t * outgoing[],
                                        int num_incoming,
                                        p8est_quadrant_t * incoming[]);

/** Compare the p8est_lid_t \a a and the p8est_lid_t \a b.
 * \param [in]  a A pointer to a p8est_lid_t.
 * \param [in]  b A pointer to a p8est_lid_t.
 * \return        Returns -1 if a < b,
 *                         1 if a > b and
 *                         0 if a == b.
 */
int                 p8est_lid_compare (const p8est_lid_t * a,
                                       const p8est_lid_t * b);

/** Checks if the p8est_lid_t \a a and the p8est_lid_t \a b are equal.
 * \param [in]  a A pointer to a p8est_lid_t.
 * \param [in]  b A pointer to a p8est_lid_t.
 * \return        Returns a true value if \a a and \a b are equal,
 *                false otherwise
 */
int                 p8est_lid_is_equal (const p8est_lid_t * a,
                                        const p8est_lid_t * b);

/** Initializes a linear index to a given value.
 * \param [in,out] input    A pointer to the p8est_lid_t that will be
 *                          initialized.
 * \param [in]     high     The given high bits to initialize \a a.
 * \param [in]     low      The given low bits to initialize \a a.
 */
void                p8est_lid_init (p8est_lid_t * input, uint64_t high,
                                    uint64_t low);

/** Initializes a linear index to zero.
 * \param [out] input     A pointer to a p4est_lid_t that will be initialized.
 */
void                p8est_lid_set_zero (p8est_lid_t * input);

/** Initializes a linear index to one.
 * \param [out] input     A pointer to a p4est_lid_t that will be initialized.
 */
void                p8est_lid_set_one (p8est_lid_t * input);

/** Initializes a linear index to an unsigned 64 bit integer.
 * \param [out] input     A pointer to a p4est_lid_t that will be initialized.
 */
void                p8est_lid_set_uint64 (p8est_lid_t * input, uint64_t u);

/** Returns the bit_number-th bit of \a input.
 * This function checks a bit of an existing, initialized value.
 * \param [in]     input      A pointer to a p8est_lid_t.
 * \param[in]      bit_number The bit (counted from the right hand side)
 *                            that is checked by logical and.
 *                            Require 0 <= \a bit_number < 128.
 * \return                    True if bit is set, false if not.
 */
int                 p8est_lid_chk_bit (const p8est_lid_t * input,
                                       int bit_number);

/** Sets the exponent-th bit of \a input to one.
 * This function modifies an existing, initialized value.
 * \param [in,out] input      A pointer to a p8est_lid_t.
 * \param[in]      bit_number The bit (counted from the right hand side)
 *                            that is set to one by logical or.
 *                            Require 0 <= \a bit_number < 128.
 */
void                p8est_lid_set_bit (p8est_lid_t * input, int bit_number);

/** Copies an initialized p8est_lid_t to a p8est_lid_t.
 * \param [in]     input    A pointer to the sc_uint128 that is copied.
 * \param [in,out] output   A pointer to a p8est_lid_t.
 *                          The high and low bits of \a output will
 *                          be set to the high and low bits of
 *                          \a input, respectively.
 */
void                p8est_lid_copy (const p8est_lid_t * input,
                                    p8est_lid_t * output);

/** Adds the uint128_t \a b to the uint128_t \a a.
 * \a result == \a a or \a result == \a b is not allowed.
 * \a a == \a b is allowed.
 * \param [in]  a       A pointer to a p8est_lid_t.
 * \param [in]  b       A pointer to a p8est_lid_t.
 * \param[out]  result  A pointer to a p8est_lid_t.
 *                      The sum \a a + \a b will be saved in \a result.
 */
void                p8est_lid_add (const p8est_lid_t * a,
                                   const p8est_lid_t * b,
                                   p8est_lid_t * result);

/** Subtracts the p8est_lid_t \a b from the p8est_lid_t \a a.
 * This function assumes that the result is >= 0.
 * \a result == \a a or \a result == \a b is not allowed.
 * \a a == \a b is allowed.
 * \param [in]  a       A pointer to a p8est_lid_t.
 * \param [in]  b       A pointer to a p8est_lid_t.
 * \param[out]  result  A pointer to a p8est_lid_t.
 *                      The difference \a a - \a b will be saved in \a result.
 */
void                p8est_lid_sub (const p8est_lid_t * a,
                                   const p8est_lid_t * b,
                                   p8est_lid_t * result);

/** Calculates the bitwise negation of the uint128_t \a a.
 * \a a == \a result is allowed.
 * \param[in]  a        A pointer to a p8est_lid_t.
 * \param[out] result   A pointer to a p8est_lid_t.
 *                      The bitwise negation of \a a will be saved in
 *                      \a result.
 */
void                p8est_lid_bitwise_neg (const p8est_lid_t * a,
                                           p8est_lid_t * result);

/** Calculates the bitwise or of the uint128_t \a a and \a b.
 * \a a == \a result is allowed. Furthermore, \a a == \a result
 * and/or \a b == \a result is allowed.
 * \param[in]  a        A pointer to a p8est_lid_t.
 * \param[in]  b        A pointer to a p8est_lid_t.
 * \param[out] result   A pointer to a p8est_lid_t.
 *                      The bitwise or of \a a and \a b will be
 *                      saved in \a result.
 */
void                p8est_lid_bitwise_or (const p8est_lid_t * a,
                                          const p8est_lid_t * b,
                                          p8est_lid_t * result);

/** Calculates the bitwise and of the uint128_t \a a and the uint128_t \a b.
 * \a a == \a result is allowed. Furthermore, \a a == \a result
 * and/or \a b == \a result is allowed.
 * \param [in]  a       A pointer to a p8est_lid_t.
 * \param [in]  b       A pointer to a p8est_lid_t.
 * \param[out]  result  A pointer to a p8est_lid_t.
 *                      The bitwise and of \a a and \a b will be saved.
 *                      in \a result.
 */
void                p8est_lid_bitwise_and (const p8est_lid_t * a,
                                           const p8est_lid_t * b,
                                           p8est_lid_t * result);

/** Calculates the bit right shift of uint128_t \a input by shift_count bits.
 * We shift in zeros from the left. If \a shift_count >= 128, \a result is 0.
 * All bits right from the zeroth bit (counted from the right hand side)
 * drop out. \a input == \a result is allowed.
 * \param [in]      input       A pointer to a p8est_lid_t.
 * \param [in]      shift_count Bits to shift. \a shift_count >= 0.
 * \param [in,out]  result      A pointer to a p8est_lid_t.
 *                              The right shifted number will be saved
 *                              in \a result.
 */
void                p8est_lid_shift_right (const p8est_lid_t * input,
                                           unsigned shift_count,
                                           p8est_lid_t * result);

/** Calculates the bit left shift of uint128_t \a input by shift_count bits.
 * We shift in zeros from the right. If \a shift_count >= 128, \a result is 0.
 * All bits left from the 127th bit (counted zero based from the right
 * hand side) drop out. \a input == \a result is allowed.
 * \param [in]      input       A pointer to a p8est_lid_t.
 * \param [in]      shift_count Bits to shift. \a shift_count >= 0.
 * \param [in,out]  result      A pointer to a p8est_lid_t.
 *                              The left shifted number will be saved
 *                              in \a result.
 */
void                p8est_lid_shift_left (const p8est_lid_t * input,
                                          unsigned shift_count,
                                          p8est_lid_t * result);

/** Adds the p8est_lid_t \a b to the p8est_lid_t \a a.
 * The result is saved in \a a. \a a == \a b is allowed.
 * \param [in, out] a   A pointer to a p8est_lid_t. \a a
 *                      will be overwritten by \a a + \a b.
 *	\param [in] b       A pointer to a p8est_lid_t.
 */
void                p8est_lid_add_inplace (p8est_lid_t * a,
                                           const p8est_lid_t * b);

/** Subtracts the uint128_t \a b from the uint128_t \a a.
 * The result is saved in \a a. \a a == \a b is allowed.
 * This function assumes that the result is >= 0.
 * \param [in,out]  a   A pointer to a p8est_lid_t.
 *                      \a a will be overwritten by \a a - \a b.
 * \param [in]      b   A pointer to a p8est_lid_t.
 */
void                p8est_lid_sub_inplace (p8est_lid_t * a,
                                           const p8est_lid_t * b);

/** Calculates the bitwise or of the uint128_t \a a and the uint128_t \a b.
 * \a a == \a b is allowed.
 * \param [in,out]  a   A pointer to a p8est_lid_t.
 *                      The bitwise or will be saved in \a a.
 * \param [in]      b   A pointer to a p8est_lid_t.
 */
void                p8est_lid_bitwise_or_inplace (p8est_lid_t * a,
                                                  const p8est_lid_t * b);

/** Calculates the bitwise and of the uint128_t \a a and the uint128_t \a b.
 * \a a == \a b is allowed.
 * \param [in,out]  a   A pointer to a p8est_lid_t.
 *                      The bitwise and will be saved in \a a.
 * \param [in]      b   A pointer to a p8est_lid_t.
 */
void                p8est_lid_bitwise_and_inplace (p8est_lid_t * a,
                                                   const p8est_lid_t * b);
/** Computes the linear position as p8est_lid_t of a quadrant in a uniform grid.
 * The grid and quadrant levels need not coincide.
 * If they do, this is the inverse of \ref p4est_quadrant_set_morton.
 * \param [in] quadrant  Quadrant whose linear index will be computed.
 *                       If the quadrant is smaller than the grid (has a higher
 *                       quadrant->level), the result is computed from its
 *                       ancestor at the grid's level.
 *                       If the quadrant has a smaller level than the grid (it
 *                       is bigger than a grid cell), the grid cell sharing its
 *                       lower left corner is used as reference.
 * \param [in] level     The level of the regular grid compared to which the
 *                       linear position is to be computed.
 * \param[in,out] id     A pointer to an allocated or static p8est_lid_t.
 *                       id will be the linear position of this quadrant on a
 *                       uniform grid.
 * \note The user_data of \a quadrant is never modified.
 */
void                p8est_quadrant_linear_id_ext128 (const p8est_quadrant_t *
                                                     quadrant, int level,
                                                     p8est_lid_t * id);

/** Set quadrant Morton indices based on linear position given as p8est_lid_t in uniform grid.
 * This is the inverse operation of \ref p4est_quadrant_linear_id.
 * \param [in,out] quadrant  Quadrant whose Morton indices will be set.
 * \param [in]     level     Level of the grid and of the resulting quadrant.
 * \param [in]     id        Linear index of the quadrant on a uniform grid.
 * \note The user_data of \a quadrant is never modified.
 */
void                p8est_quadrant_set_morton_ext128 (p8est_quadrant_t *
                                                      quadrant, int level,
                                                      const p8est_lid_t * id);

/** Create a new forest.
 * This is a more general form of \ref p8est_new.
 * The forest created is either uniformly refined at a given level
 * or created with the coarsest possible refinement that fits the
 * exact partition that would have been created in the uniform mode.
 * The latter, coarse refinement depends on the number of MPI processes!
 * The initial level is currently limited to \ref P8EST_OLD_QMAXLEVEL.
 * Regardless, \ref p8est_refine can go as deep as \ref P8EST_QMAXLEVEL.
 *
 * \param [in] mpicomm          A valid MPI communicator.
 * \param [in] connectivity     This is the connectivity information that
 *                              the forest is built with.  Note the forest
 *                              does not take ownership of the memory.
 * \param [in] min_quadrants    Minimum initial quadrants per processor.
 *                              Makes the refinement pattern mpisize-specific.
 *                              For maximum reproducibility, set this to 0.
 * \param [in] min_level        The forest is refined at most to this level.
 *                              Later coarsening and refinement is unaffected.
 *                              May be negative or 0, then it has no effect.
 * \param [in] fill_uniform     If true, fill the forest with a uniform mesh
 *                              instead of the coarsest possible one.
 *                              The latter is partition-specific, which
 *                              is not a good idea wrt. reproducibility.
 * \param [in] data_size        The size of data for each quadrant.
 * \param [in] init_fn          Callback function to initialize the user_data
 *                              which is internally allocated using data_size.
 * \param [in] user_pointer     Assigned to the user_pointer member of the
 *                              forest before init_fn is called the first time.
 * \return                      Valid p8est object.
 */
p8est_t            *p8est_new_ext (sc_MPI_Comm mpicomm,
                                   p8est_connectivity_t * connectivity,
                                   p4est_locidx_t min_quadrants,
                                   int min_level, int fill_uniform,
                                   size_t data_size, p8est_init_t init_fn,
                                   void *user_pointer);

/** Create a new mesh.
 * This function sets a subset of the mesh creation parameters. For full control
 * use \ref p8est_mesh_new_params.
 * \param [in] p8est                A forest that is fully 2:1 balanced.
 * \param [in] ghost                The ghost layer created from the
 *                                  provided p4est.
 * \param [in] compute_tree_index   Boolean to decide whether to allocate and
 *                                  compute the quad_to_tree list.
 * \param [in] compute_level_lists  Boolean to decide whether to compute the
 *                                  level lists in quad_level.
 * \param [in] btype                Flag indicating the connection types (face,
                                    edge, corner) stored in the mesh.
 * \return                          A fully allocated mesh structure.
 */
p8est_mesh_t       *p8est_mesh_new_ext (p8est_t * p8est,
                                        p8est_ghost_t * ghost,
                                        int compute_tree_index,
                                        int compute_level_lists,
                                        p8est_connect_type_t btype);

/** Make a deep copy of a p8est.
 * The connectivity is not duplicated.
 * Copying of quadrant user data is optional.
 * If old and new data sizes are 0, the user_data field is copied regardless.
 * The inspect member of the copy is set to NULL.
 * The revision counter of the copy is set to zero.
 *
 * \param [in]  input      Valid forest to return a copy of.
 * \param [in]  copy_data  If true, data are copied.
 *                         If false, data_size is set to 0.
 * \param [in]  duplicate_mpicomm  If true, MPI communicator is copied.
 * \return  Returns a valid p8est that does not depend on the input,
 *                         except for borrowing the same connectivity.
 *                         Its revision counter is 0.
 */
p8est_t            *p8est_copy_ext (p8est_t * input, int copy_data,
                                    int duplicate_mpicomm);

/** Refine a forest with a bounded refinement level and a replace option.
 * \param [in,out] p8est The forest is changed in place.
 * \param [in] refine_recursive Boolean to decide on recursive refinement.
 * \param [in] maxlevel   Maximum allowed refinement level (inclusive).
 *                        If this is negative the level is restricted only
 *                        by the compile-time constant QMAXLEVEL in p8est.h.
 * \param [in] refine_fn  Callback function that must return true if a quadrant
 *                        shall be refined.  If refine_recursive is true,
 *                        refine_fn is called for every existing and newly
 *                        created quadrant.  Otherwise, it is called for every
 *                        existing quadrant.  It is possible that a refinement
 *                        request made by the callback is ignored.  To catch
 *                        this case, you can examine whether init_fn or
 *                        replace_fn gets called.
 * \param [in] init_fn    Callback function to initialize the user_data for
 *                        newly created quadrants, which is guaranteed to be
 *                        allocated.  This function pointer may be NULL.
 * \param [in] replace_fn Callback function that allows the user to change
 *                        incoming quadrants based on the quadrants they
 *                        replace; may be NULL.
 */
void                p8est_refine_ext (p8est_t * p8est,
                                      int refine_recursive, int maxlevel,
                                      p8est_refine_t refine_fn,
                                      p8est_init_t init_fn,
                                      p8est_replace_t replace_fn);

/** Coarsen a forest.
 * \param [in,out] p8est The forest is changed in place.
 * \param [in] coarsen_recursive Boolean to decide on recursive coarsening.
 * \param [in] callback_orphans Boolean to enable calling coarsen_fn even on
 *                        non-families.  In this case, the second quadrant
 *                        pointer in the argument list of the callback is NULL,
 *                        subsequent pointers are undefined, and the return
 *                        value is ignored.  If coarsen_recursive is true, it
 *                        is possible that a quadrant is called once or more as
 *                        an orphan and eventually becomes part of a family.
 *                        With coarsen_recursive false and callback_orphans true,
 *                        it is guaranteed that every quadrant is passed exactly
 *                        once into the coarsen_fn callback.
 * \param [in] coarsen_fn Callback function that returns true if a
 *                        family of quadrants shall be coarsened.
 * \param [in] init_fn    Callback function to initialize the user_data
 *                        which is already allocated automatically.
 * \param [in] replace_fn Callback function that allows the user to change
 *                        incoming quadrants based on the quadrants they
 *                        replace.
 */
void                p8est_coarsen_ext (p8est_t * p8est, int coarsen_recursive,
                                       int callback_orphans,
                                       p8est_coarsen_t coarsen_fn,
                                       p8est_init_t init_fn,
                                       p8est_replace_t replace_fn);

/** 2:1 balance the size differences of neighboring elements in a forest.
 * \param [in,out] p8est  The p8est to be worked on.
 * \param [in] btype      Balance type (face, edge, or corner/full).
 *                        Corner balance is almost never required when
 *                        discretizing a PDE; just causes smoother mesh grading.
 * \param [in] init_fn    Callback function to initialize the user_data
 *                        which is already allocated automatically.
 * \param [in] replace_fn Callback function that allows the user to change
 *                        incoming quadrants based on the quadrants they
 *                        replace.
 */
void                p8est_balance_ext (p8est_t * p8est,
                                       p8est_connect_type_t btype,
                                       p8est_init_t init_fn,
                                       p8est_replace_t replace_fn);

void                p8est_balance_subtree_ext (p8est_t * p8est,
                                               p8est_connect_type_t btype,
                                               p4est_topidx_t which_tree,
                                               p8est_init_t init_fn,
                                               p8est_replace_t replace_fn);

/** Repartition the forest.
 *
 * The forest is partitioned between processors such that each processor
 * has an approximately equal number of quadrants (or weight).
 *
 * The user data of a \ref p8est_quadrant is transferred along within this
 * function.
 *
 * The only extension of this function over \ref p8est_partition is the
 * return value.
 *
 * \param [in,out] p8est      The forest that will be partitioned.
 * \param [in]     partition_for_coarsening     If true, the partition
 *                            is modified to allow one level of coarsening.
 * \param [in]     weight_fn  A weighting function or NULL
 *                            for uniform partitioning. A weighting function
 *                            with constant weight 1 on each quadrant is
 *                            equivalent to weight_fn == NULL but other constant
 *                            weightings may result in different uniform
 *                            partitionings.
 * \return         The global number of shipped quadrants
 */
p4est_gloidx_t      p8est_partition_ext (p8est_t * p8est,
                                         int partition_for_coarsening,
                                         p8est_weight_t weight_fn);

/** Correct partition to allow one level of coarsening.
 *
 * \param [in] p8est                     forest whose partition is corrected
 * \param [in,out] num_quadrants_in_proc partition that will be corrected
 * \return                               absolute number of moved quadrants
 */
p4est_gloidx_t      p8est_partition_for_coarsening (p8est_t * p8est,
                                                    p4est_locidx_t *
                                                    num_quadrants_in_proc);

/** p8est_iterate_ext adds the option \a remote: if this is false, then it is
 * the same as p8est_iterate; if this is true, then corner/edge callbacks are
 * also called on corners/edges for hanging faces/edges touched by local
 * quadrants.
 */
void                p8est_iterate_ext (p8est_t * p8est,
                                       p8est_ghost_t * ghost_layer,
                                       void *user_data,
                                       p8est_iter_volume_t iter_volume,
                                       p8est_iter_face_t iter_face,
                                       p8est_iter_edge_t iter_edge,
                                       p8est_iter_corner_t iter_corner,
                                       int remote);

/** Save the complete connectivity/p8est data to disk.  This is a collective
 * operation that all MPI processes need to call.  All processes write
 * into the same file, so the filename given needs to be identical over
 * all parallel invocations.
 * See p8est_load_ext for information on the autopartition parameter.
 * \param [in] filename    Name of the file to write.
 * \param [in] p8est       Valid forest structure.
 * \param [in] save_data   If true, the element data is saved.
 *                         Otherwise, a data size of 0 is saved.
 * \param [in] save_partition   If false, save file as if 1 core was used.
 *                              If true, save core count and partition.
 *                         Advantage: Partition can be recovered on loading
 *                              with same mpisize and autopartition false.
 *                         Disadvantage: Makes the file depend on mpisize.
 *                  Either way the file can be loaded with autopartition true.
 * \note            Aborts on file errors.
 */
void                p8est_save_ext (const char *filename, p8est_t * p8est,
                                    int save_data, int save_partition);

/** Load the complete connectivity/p4est structure from disk.
 * It is possible to load the file with a different number of processors
 * than has been used to write it.  The partition will then be uniform.
 * \param [in] filename         Name of the file to read.
 * \param [in] mpicomm          A valid MPI communicator.
 * \param [in] data_size        Size of data for each quadrant which can be
 *                              zero.  Then user_data_pool is set to NULL.
 *                              If data_size is zero, load_data is ignored.
 * \param [in] load_data        If true, the element data is loaded.  This is
 *                              only permitted if the saved data size matches.
 *                              If false, the stored data size is ignored.
 * \param [in] autopartition    Ignore saved partition and make it uniform.
 * \param [in] broadcasthead    Have only rank 0 read headers and bcast them.
 * \param [in] user_pointer     Assign to the user_pointer member of the p4est
 *                              before init_fn is called the first time.
 * \param [out] connectivity    Connectivity must be destroyed separately.
 * \return          Returns a valid forest structure. A pointer to a valid
 *                  connectivity structure is returned through the last
 *                  argument.
 * \note            Aborts on file errors or invalid file contents.
 */
p8est_t            *p8est_load_ext (const char *filename, sc_MPI_Comm mpicomm,
                                    size_t data_size, int load_data,
                                    int autopartition, int broadcasthead,
                                    void *user_pointer,
                                    p8est_connectivity_t ** connectivity);

/** The same as p8est_load_ext, but reading the connectivity/p8est from an
 * open sc_io_source_t stream.
 */
p8est_t            *p8est_source_ext (sc_io_source_t * src,
                                      sc_MPI_Comm mpicomm, size_t data_size,
                                      int load_data, int autopartition,
                                      int broadcasthead, void *user_pointer,
                                      p8est_connectivity_t ** connectivity);

#ifdef P4EST_ENABLE_FILE_DEPRECATED

/** Open a file for reading without knowing the p4est that is associated
 * with the mesh-related data in the file (cf. \ref p8est_file_open_read).
 * For more general comments on open_read see the documentation of
 * \ref p8est_file_open_read.
 * The parameters that are not documented are the same as in \ref
 * p8est_file_open_read.
 *
 * \param [in]  mpicomm   The MPI communicator that is used to read the file.
 */
p8est_file_context_t *p8est_file_open_read_ext (sc_MPI_Comm mpicomm,
                                                const char *filename,
                                                char *user_string,
                                                p4est_gloidx_t *
                                                global_num_quadrants,
                                                int *errcode);

/** Read a data field and specify the partition for reading in parallel.
 * See also the documentation of \ref p8est_file_read_field.
 *
 * \param [in]  gfq   An array of the size mpisize + 1 that contains the global
 *                    first quadrants per rank and
 *                    gfq[mpisize] == global_num_quadrants. This defines
 *                    partition that is used to read the data field in parallel.
 */
p8est_file_context_t *p8est_file_read_field_ext (p8est_file_context_t * fc,
                                                 p4est_gloidx_t * gfq,
                                                 size_t quadrant_size,
                                                 sc_array_t * quadrant_data,
                                                 char *user_string,
                                                 int *errcode);

#endif /* P4EST_ENABLE_FILE_DEPRECATED */

/** Create the data necessary to create a PETsc DMPLEX representation of a
 * forest, as well as the accompanying lnodes and ghost layer.  The forest
 * must be at least face balanced (see p4est_balance()).  See
 * test/test_plex2.c for example usage.
 *
 * All arrays should be initialized to hold sizeof (p4est_locidx_t), except
 * for \a out_remotes, which should be initialized to hold
 * (2 * sizeof (p4est_locidx_t)).
 *
 * \param[in]     p8est                 the forest
 * \param[out]    ghost                 the ghost layer
 * \param[out]    lnodes                the lnodes
 * \param[in]     ctype                 the type of adjacency for the overlap
 * \param[in]     overlap               the number of layers of overlap (zero
 *                                      is acceptable)
 * \param[out]    first_local_quad      the local quadrants are assigned
 *                                      contiguous plex indices, starting with
 *                                      this index
 * \param[in,out] out_points_per_dim    filled with argument for
 *                                      DMPlexCreateFromDAG()
 * \param[in,out] out_cone_sizes        filled with argument for
 *                                      DMPlexCreateFromDAG()
 * \param[in,out] out_cones             filled with argument for
 *                                      DMPlexCreateFromDAG()
 * \param[in,out] out_cone_orientations filled with argument for
 *                                      DMPlexCreateFromDAG()
 * \param[in,out] out_vertex_coords     filled with argument for
 *                                      DMPlexCreateFromDAG()
 * \param[in,out] out_children          filled with argument for
 *                                      DMPlexSetTree()
 * \param[in,out] out_parents           filled with argument for
 *                                      DMPlexSetTree()
 * \param[in,out] out_childids          filled with argument for
 *                                      DMPlexSetTree()
 * \param[in,out] out_leaves            filled with argument for
 *                                      PetscSFSetGraph()
 * \param[in,out] out_remotes           filled with argument for
 *                                      PetscSFSetGraph()
 * \param[in]     custom_numbering      Whether or use the default numbering
 *                                      (0) of DMPlex child ids or the custom
 *                                      (1).
 */
void                p8est_get_plex_data_ext (p8est_t * p8est,
                                             p8est_ghost_t ** ghost,
                                             p8est_lnodes_t ** lnodes,
                                             p8est_connect_type_t ctype,
                                             int overlap,
                                             p4est_locidx_t *
                                             first_local_quad,
                                             sc_array_t * out_points_per_dim,
                                             sc_array_t * out_cone_sizes,
                                             sc_array_t * out_cones,
                                             sc_array_t *
                                             out_cone_orientations,
                                             sc_array_t * out_vertex_coords,
                                             sc_array_t * out_children,
                                             sc_array_t * out_parents,
                                             sc_array_t * out_childids,
                                             sc_array_t * out_leaves,
                                             sc_array_t * out_remotes,
                                             int custom_numbering);

SC_EXTERN_C_END;

#endif /* !P8EST_EXTENDED_H */
