p4est.h

Go to the documentation of this file.

/*
  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.
*/
 
#ifndef P4EST_H
#define P4EST_H
 
#ifdef P4EST_TO_P8EST_H
#error "The include files p4est.h and p4est_to_p8est.h cannot be combined"
#endif
 
/* p4est_connectivity.h includes p4est_base.h sc_containers.h */
#include <p4est_connectivity.h>
 
SC_EXTERN_C_BEGIN;
 
#define P4EST_OLD_MAXLEVEL 30   /* in 2D, the maxlevel has always been 30 */
#define P4EST_MAXLEVEL 30
 
#define P4EST_OLD_QMAXLEVEL 29  /* in 2D, the qmaxlevel has always been 29 */
#define P4EST_QMAXLEVEL 29
 
#define P4EST_ROOT_LEN ((p4est_qcoord_t) 1 << P4EST_MAXLEVEL)
 
#define P4EST_QUADRANT_LEN(l) ((p4est_qcoord_t) 1 << (P4EST_MAXLEVEL - (l)))
 
#define P4EST_QUADRANT_MASK(l) (~(P4EST_QUADRANT_LEN (l) - 1))
 
#define P4EST_LAST_OFFSET(l) (P4EST_ROOT_LEN - P4EST_QUADRANT_LEN (l))
 
typedef struct p4est_quadrant
{
  p4est_qcoord_t      x, y;  
  int8_t              level,    
                      pad8;     
  int16_t             pad16;    
  union p4est_quadrant_data
  {
    void               *user_data;      
    long                user_long;      
    int                 user_int;       
    p4est_topidx_t      which_tree;     
    struct
    {
      p4est_topidx_t      which_tree;
      int                 owner_rank;
    }
    piggy1; 
    struct
    {
      p4est_topidx_t      which_tree;
      p4est_topidx_t      from_tree;
    }
    piggy2; 
    struct
    {
      p4est_topidx_t      which_tree;
      p4est_locidx_t      local_num;
    }
    piggy3; 
  }
  p; 
}
p4est_quadrant_t;
 
typedef struct p4est_tree
{
  sc_array_t          quadrants;             
  p4est_quadrant_t    first_desc,            
                      last_desc;             
  p4est_locidx_t      quadrants_offset;      
  p4est_locidx_t      quadrants_per_level[P4EST_MAXLEVEL + 1];
  int8_t              maxlevel;              
}
p4est_tree_t;
 
typedef struct p4est_inspect p4est_inspect_t;
 
typedef struct p4est
{
  sc_MPI_Comm         mpicomm;          
  int                 mpisize,          
                      mpirank;          
  int                 mpicomm_owned;    
  size_t              data_size;        
  void               *user_pointer;     
  long                revision;         
  p4est_topidx_t      first_local_tree; 
  p4est_topidx_t      last_local_tree;  
  p4est_locidx_t      local_num_quadrants;   
  p4est_gloidx_t      global_num_quadrants;  
  p4est_gloidx_t     *global_first_quadrant; 
  p4est_quadrant_t   *global_first_position; 
  p4est_connectivity_t *connectivity; 
  sc_array_t         *trees;          
  sc_mempool_t       *user_data_pool; 
  /*   WARNING: This is NULL if data size
     equals zero. */
  sc_mempool_t       *quadrant_pool;  
  p4est_inspect_t    *inspect;        
}
p4est_t;
 
size_t              p4est_memory_used (p4est_t * p4est);
 
long                p4est_revision (p4est_t * p4est);
 
typedef void        (*p4est_init_t) (p4est_t * p4est,
                                     p4est_topidx_t which_tree,
                                     p4est_quadrant_t * quadrant);
 
typedef int         (*p4est_refine_t) (p4est_t * p4est,
                                       p4est_topidx_t which_tree,
                                       p4est_quadrant_t * quadrant);
 
typedef int         (*p4est_coarsen_t) (p4est_t * p4est,
                                        p4est_topidx_t which_tree,
                                        p4est_quadrant_t * quadrants[]);
 
typedef int         (*p4est_weight_t) (p4est_t * p4est,
                                       p4est_topidx_t which_tree,
                                       p4est_quadrant_t * quadrant);
 
extern void        *P4EST_DATA_UNINITIALIZED;
 
#define P4EST_QUADRANT_INIT(q) \
  ((void) memset ((q), -1, sizeof (p4est_quadrant_t)))
 
void                p4est_qcoord_to_vertex (p4est_connectivity_t *
                                            connectivity,
                                            p4est_topidx_t treeid,
                                            p4est_qcoord_t x,
                                            p4est_qcoord_t y, double vxyz[3]);
 
p4est_t            *p4est_new (sc_MPI_Comm mpicomm,
                               p4est_connectivity_t * connectivity,
                               size_t data_size,
                               p4est_init_t init_fn, void *user_pointer);
 
void                p4est_destroy (p4est_t * p4est);
 
p4est_t            *p4est_copy (p4est_t * input, int copy_data);
 
void                p4est_reset_data (p4est_t * p4est, size_t data_size,
                                      p4est_init_t init_fn,
                                      void *user_pointer);
 
void                p4est_refine (p4est_t * p4est,
                                  int refine_recursive,
                                  p4est_refine_t refine_fn,
                                  p4est_init_t init_fn);
 
void                p4est_coarsen (p4est_t * p4est,
                                   int coarsen_recursive,
                                   p4est_coarsen_t coarsen_fn,
                                   p4est_init_t init_fn);
 
void                p4est_balance (p4est_t * p4est,
                                   p4est_connect_type_t btype,
                                   p4est_init_t init_fn);
 
void                p4est_partition (p4est_t * p4est,
                                     int allow_for_coarsening,
                                     p4est_weight_t weight_fn);
 
unsigned            p4est_checksum (p4est_t * p4est);
 
unsigned            p4est_checksum_partition (p4est_t * p4est);
 
void                p4est_save (const char *filename, p4est_t * p4est,
                                int save_data);
 
p4est_t            *p4est_load (const char *filename, sc_MPI_Comm mpicomm,
                                size_t data_size, int load_data,
                                void *user_pointer,
                                p4est_connectivity_t ** connectivity);
 
/*@unused@*/
static inline p4est_tree_t *
p4est_tree_array_index (sc_array_t * array, p4est_topidx_t it)
{
  P4EST_ASSERT (array->elem_size == sizeof (p4est_tree_t));
  P4EST_ASSERT (it >= 0 && (size_t) it < array->elem_count);
 
  return (p4est_tree_t *) (array->array +
                           sizeof (p4est_tree_t) * (size_t) it);
}
 
/*@unused@*/
static inline p4est_quadrant_t *
p4est_quadrant_array_index (sc_array_t * array, size_t it)
{
  P4EST_ASSERT (array->elem_size == sizeof (p4est_quadrant_t));
  P4EST_ASSERT (it < array->elem_count);
 
  return (p4est_quadrant_t *) (array->array + sizeof (p4est_quadrant_t) * it);
}
 
static inline p4est_quadrant_t *
p4est_quadrant_array_push_copy (sc_array_t * array,
                                const p4est_quadrant_t *qsrc)
{
  p4est_quadrant_t *q;
 
  P4EST_ASSERT (array->elem_size == sizeof (p4est_quadrant_t));
 
  q = (p4est_quadrant_t *) sc_array_push (array);
  *q = *qsrc;
  return q;
}
 
/*@unused@*/
static inline p4est_quadrant_t *
p4est_quadrant_array_push (sc_array_t * array)
{
  p4est_quadrant_t *q;
 
  P4EST_ASSERT (array->elem_size == sizeof (p4est_quadrant_t));
 
  q = (p4est_quadrant_t *) sc_array_push (array);
  P4EST_QUADRANT_INIT(q);
  return q;
}
 
/*@unused@*/
static inline p4est_quadrant_t *
p4est_quadrant_mempool_alloc (sc_mempool_t * mempool)
{
  P4EST_ASSERT (mempool->elem_size == sizeof (p4est_quadrant_t));
 
  return (p4est_quadrant_t *) sc_mempool_alloc (mempool);
}
 
/*@unused@*/
static inline p4est_quadrant_t *
p4est_quadrant_list_pop (sc_list_t * list)
{
  return (p4est_quadrant_t *) sc_list_pop (list);
}
 
SC_EXTERN_C_END;
 
#endif /* !P4EST_H */