BLI_astar.h

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2014 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#pragma once

 
#include "DNA_listBase.h"
 
#include "BLI_bitmap.h"
 
/* -------------------------------------------------------------------- */
 
struct BLI_AStarGNLink {
  int nodes[2];
  float cost;
 
  void *custom_data;
};
 
struct BLI_AStarGNode {
  struct ListBase neighbor_links;
 
  void *custom_data;
};
 
struct BLI_AStarSolution {
  /* Final 'most useful' data. */
  int steps;
  int *prev_nodes;
  BLI_AStarGNLink **prev_links;
 
  void *custom_data;
 
  /* Mostly runtime data. */
  BLI_bitmap *done_nodes;
  float *g_costs;
  int *g_steps;
 
  struct MemArena *mem; /* Memory arena. */
};
 
struct BLI_AStarGraph {
  int node_num;
  BLI_AStarGNode *nodes;
 
  void *custom_data;
 
  struct MemArena *mem; /* Memory arena. */
};

void BLI_astar_node_init(BLI_AStarGraph *as_graph, int node_index, void *custom_data);
void BLI_astar_node_link_add(
    BLI_AStarGraph *as_graph, int node1_index, int node2_index, float cost, void *custom_data);
int BLI_astar_node_link_other_node(BLI_AStarGNLink *lnk, int idx);

void BLI_astar_solution_init(BLI_AStarGraph *as_graph,
                             BLI_AStarSolution *as_solution,
                             void *custom_data);
void BLI_astar_solution_clear(BLI_AStarSolution *as_solution);
void BLI_astar_solution_free(BLI_AStarSolution *as_solution);

using astar_f_cost = float (*)(BLI_AStarGraph *as_graph,
                               BLI_AStarSolution *as_solution,
                               BLI_AStarGNLink *link,
                               int node_idx_curr,
                               int node_idx_next,
                               int node_idx_dst);

void BLI_astar_graph_init(BLI_AStarGraph *as_graph, int node_num, void *custom_data);
void BLI_astar_graph_free(BLI_AStarGraph *as_graph);
bool BLI_astar_graph_solve(BLI_AStarGraph *as_graph,
                           int node_index_src,
                           int node_index_dst,
                           astar_f_cost f_cost_cb,
                           BLI_AStarSolution *r_solution,
                           int max_steps);