ref_oct_test.c

/* Copyright 2006, 2014, 2021 United States Government as represented
 * by the Administrator of the National Aeronautics and Space
 * Administration. No copyright is claimed in the United States under
 * Title 17, U.S. Code.  All Other Rights Reserved.
 *
 * The refine version 3 unstructured grid adaptation platform is
 * 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
 * https://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.
 */

#include "ref_oct.h"

#include <math.h>
#include <stdio.h>
#include <stdlib.h>

#include "ref_args.h"
#include "ref_cell.h"
#include "ref_grid.h"
#include "ref_import.h"
#include "ref_mpi.h"
#include "ref_node.h"

int main(int argc, char *argv[]) {
  REF_INT pos;
  REF_MPI ref_mpi;
  RSS(ref_mpi_start(argc, argv), "start");
  RSS(ref_mpi_create(&ref_mpi), "make mpi");

  RXS(ref_args_find(argc, argv, "--tec", &pos), REF_NOT_FOUND, "arg search");
  if (pos != REF_EMPTY && pos + 1 < argc) {
    REF_OCT ref_oct;
    RSS(ref_oct_create(&ref_oct), "make oct");
    RSS(ref_oct_split(ref_oct, 0), "split root");
    RSS(ref_oct_split(ref_oct, 1), "split first child");
    RSS(ref_oct_split(ref_oct, 9), "split second gen");
    RSS(ref_oct_tec(ref_oct, argv[pos + 1]), "tec");
    RSS(ref_oct_free(ref_oct), "free oct");
    RSS(ref_mpi_free(ref_mpi), "mpi free");
    RSS(ref_mpi_stop(), "stop");
    return 0;
  }

  RXS(ref_args_find(argc, argv, "--point", &pos), REF_NOT_FOUND, "arg search");
  if (pos != REF_EMPTY && pos + 1 < argc) {
    REF_OCT ref_oct;
    REF_DBL xyz[3], h;
    RSS(ref_oct_create(&ref_oct), "make oct");
    xyz[0] = 0.3;
    xyz[1] = 0.4;
    xyz[2] = 0.49;
    h = 0.01;
    RSS(ref_oct_split_at(ref_oct, xyz, h), "split xyz h");
    RSS(ref_oct_tec(ref_oct, argv[pos + 1]), "tec");
    RSS(ref_oct_free(ref_oct), "search oct");
    RSS(ref_mpi_free(ref_mpi), "mpi free");
    RSS(ref_mpi_stop(), "stop");
    return 0;
  }

  RXS(ref_args_find(argc, argv, "--surf", &pos), REF_NOT_FOUND, "arg search");
  if (pos != REF_EMPTY && pos + 2 < argc) {
    REF_OCT ref_oct;

    RSS(ref_oct_create(&ref_oct), "make oct");
    {
      REF_GRID ref_grid;
      REF_CELL ref_cell;
      REF_NODE ref_node;
      REF_INT cell, nodes[REF_CELL_MAX_SIZE_PER];

      RSS(ref_import_by_extension(&ref_grid, ref_mpi, argv[pos + 1]), "import");
      ref_node = ref_grid_node(ref_grid);
      ref_cell = ref_grid_tri(ref_grid);
      each_ref_cell_valid_cell_with_nodes(ref_cell, cell, nodes) {
        REF_DBL xyz[3], h, area;
        REF_INT i, j;
        RSS(ref_node_tri_area(ref_node, nodes, &area), "tri area");
        h = sqrt(area);
        for (j = 0; j < 3; j++) {
          xyz[j] = 0;
          for (i = 0; i < 3; i++) {
            xyz[j] += ref_node_xyz(ref_node, j, nodes[i]) / 3.0;
          }
        }
        RSS(ref_oct_split_at(ref_oct, xyz, h), "split xyz h");
      }
      RSS(ref_grid_free(ref_grid), "free grid");
    }
    printf("writing %d vox to %s from %s\n", ref_oct->n, argv[pos + 2],
           argv[pos + 1]);
    RSS(ref_oct_tec(ref_oct, argv[pos + 2]), "tec");
    RSS(ref_oct_free(ref_oct), "search oct");
    RSS(ref_mpi_free(ref_mpi), "mpi free");
    RSS(ref_mpi_stop(), "stop");
    return 0;
  }

  { /* create */
    REF_OCT ref_oct;
    RSS(ref_oct_create(&ref_oct), "make oct");
    RSS(ref_oct_free(ref_oct), "free oct");
  }

  { /* child's bbox */
    REF_DBL parent_bbox[6], child_bbox[6];
    REF_INT child_index = 0;
    REF_DBL tol = -1.0;
    parent_bbox[0] = 0.0;
    parent_bbox[1] = 1.0;
    parent_bbox[2] = 0.0;
    parent_bbox[3] = 1.0;
    parent_bbox[4] = 0.0;
    parent_bbox[5] = 1.0;
    RSS(ref_oct_child_bbox(parent_bbox, child_index, child_bbox), "bbox");
    RWDS(0.0, child_bbox[0], tol, "not zero");
  }

  { /* bbox diag */
    REF_DBL bbox[6];
    REF_DBL diag, tol = -1.0;
    bbox[0] = 0.0;
    bbox[1] = 0.5;
    bbox[2] = 0.5;
    bbox[3] = 1.0;
    bbox[4] = 0.2;
    bbox[5] = 0.7;
    RSS(ref_oct_bbox_diag(bbox, &diag), "bbox diag");
    RWDS(sqrt(0.75), diag, tol, "wrong size");
  }

  { /* split root */
    REF_OCT ref_oct;
    RSS(ref_oct_create(&ref_oct), "make oct");
    RSS(ref_oct_split(ref_oct, 0), "split oct");
    RSS(ref_oct_free(ref_oct), "free oct");
  }

  { /* one refinment, matching bbox */
    REF_OCT ref_oct;
    REF_DBL bbox[6], h;
    bbox[0] = 0.0;
    bbox[1] = 1.0;
    bbox[2] = 0.0;
    bbox[3] = 1.0;
    bbox[4] = 0.0;
    bbox[5] = 1.0;
    h = 0.9 * sqrt(3.0);
    RSS(ref_oct_create(&ref_oct), "make oct");
    RSS(ref_oct_split_touching(ref_oct, bbox, h), "split main");
    REIS(9, ref_oct_n(ref_oct), "expect one refinement, match")
    RSS(ref_oct_free(ref_oct), "free oct");
  }

  { /* contains root */
    REF_OCT ref_oct;
    REF_DBL xyz[] = {0.1, 0.1, 0.1};
    REF_DBL bbox[6];
    REF_INT node;
    RSS(ref_oct_create(&ref_oct), "make oct");
    RSS(ref_oct_contains(ref_oct, xyz, &node, bbox), "contains oct");
    REIS(0, node, "expects root");
    xyz[0] = 100.0;
    RSS(ref_oct_contains(ref_oct, xyz, &node, bbox), "contains oct");
    REIS(REF_EMPTY, node, "expects empty outside root");
    RSS(ref_oct_free(ref_oct), "free oct");
  }

  { /* contains child */
    REF_OCT ref_oct;
    REF_DBL xyz[] = {0.1, 0.1, 0.1};
    REF_DBL bbox[6];
    REF_INT node;
    RSS(ref_oct_create(&ref_oct), "make oct");
    RSS(ref_oct_split(ref_oct, 0), "split oct");
    RSS(ref_oct_contains(ref_oct, xyz, &node, bbox), "contains oct");
    REIS(1, node, "expects first child");
    xyz[0] = 0.9;
    xyz[1] = 0.1;
    xyz[2] = 0.1;
    RSS(ref_oct_contains(ref_oct, xyz, &node, bbox), "contains oct");
    REIS(2, node, "expects last child");
    xyz[0] = 0.1;
    xyz[1] = 0.9;
    xyz[2] = 0.1;
    RSS(ref_oct_contains(ref_oct, xyz, &node, bbox), "contains oct");
    REIS(4, node, "expects last child");
    xyz[0] = 0.9;
    xyz[1] = 0.9;
    xyz[2] = 0.9;
    RSS(ref_oct_contains(ref_oct, xyz, &node, bbox), "contains oct");
    REIS(7, node, "expects last child");
    RSS(ref_oct_free(ref_oct), "free oct");
  }

  {
    REF_OCT ref_oct;
    REF_DBL xyz[] = {0.01, 0.01, 0.01};
    REF_DBL bbox[6];
    REF_INT node;

    RSS(ref_oct_create(&ref_oct), "make oct");
    RSS(ref_oct_split(ref_oct, 0), "split root");
    RSS(ref_oct_split(ref_oct, 1), "split first child");
    RSS(ref_oct_split(ref_oct, 9), "split second gen");

    RSS(ref_oct_contains(ref_oct, xyz, &node, bbox), "contains oct");
    REIS(17, node, "expects third level");
    RSS(ref_oct_free(ref_oct), "free oct");
  }

  {
    REF_DBL bbox0[6], bbox1[6];
    REF_BOOL overlap;
    bbox0[0] = 0.0;
    bbox0[1] = 1.0;
    bbox0[2] = 0.0;
    bbox0[3] = 1.0;
    bbox0[4] = 0.0;
    bbox0[5] = 1.0;
    bbox1[0] = 0.1;
    bbox1[1] = 0.2;
    bbox1[2] = 0.1;
    bbox1[3] = 0.2;
    bbox1[4] = 0.1;
    bbox1[5] = 0.2;
    RSS(ref_oct_bbox_overlap(bbox0, bbox1, &overlap), "overlap");
    REIS(REF_TRUE, overlap, "expect overlap");
    bbox1[0] = 1.1;
    bbox1[1] = 1.2;
    bbox1[2] = 0.1;
    bbox1[3] = 0.2;
    bbox1[4] = 0.1;
    bbox1[5] = 0.2;
    RSS(ref_oct_bbox_overlap(bbox0, bbox1, &overlap), "overlap");
    REIS(REF_FALSE, overlap, "did expect overlap");
  }

  RSS(ref_mpi_free(ref_mpi), "mpi free");
  RSS(ref_mpi_stop(), "stop");
  return 0;
}