lib/graphcut.h

   1 /*
   2     graphcut- a graphcut implementation based on the Boykov Kolmogorov algorithm
   3
   4     Part of the swftools package.
   5
   6     Copyright (c) 2007,2008,2009 Matthias Kramm <kramm@quiss.org>
   7
   8     This program is free software: you can redistribute it and/or modify
   9     it under the terms of the GNU General Public License as published by
  10     the Free Software Foundation, either version 3 of the License, or
  11     (at your option) any later version.
  12
  13     This program is distributed in the hope that it will be useful,
  14     but WITHOUT ANY WARRANTY; without even the implied warranty of
  15     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16     GNU General Public License for more details.
  17
  18     You should have received a copy of the GNU General Public License
  19     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21 #ifndef __graphcut_h__
  22 #define __graphcut_h__
  23
  24 typedef signed int weight_t;
  25
  26 typedef struct _halfedge halfedge_t;
  27 typedef struct _node node_t;
  28 typedef struct _graph graph_t;
  29
  30 struct _halfedge {
  31     node_t*node;
  32     struct _halfedge*fwd;
  33     weight_t weight;
  34     weight_t init_weight;
  35     char used;
  36     halfedge_t*next;
  37 };
  38
  39 struct _node {
  40     halfedge_t*edges;
  41     union {
  42         int tmp;
  43         int component;
  44         int color;
  45     };
  46     int nr;
  47 };
  48
  49 struct _graph {
  50     node_t* nodes;
  51     int num_nodes;
  52 };
  53
  54 graph_t* graph_new(int num_nodes);
  55 halfedge_t*graph_add_edge(node_t*from, node_t*to, weight_t forward_weight, weight_t backward_weight);
  56 weight_t graph_maxflow(graph_t*graph, node_t*pos1, node_t*pos2);
  57 int graph_find_components(graph_t*g);
  58 void graph_delete(graph_t*);
  59
  60 #endif