forked from Kitware/VTK
-
Notifications
You must be signed in to change notification settings - Fork 0
/
vtkKCoreDecomposition.h
122 lines (101 loc) · 3.84 KB
/
vtkKCoreDecomposition.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
/*=========================================================================
Program: Visualization Toolkit
Module: vtkKCoreDecomposition.h
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================*/
/*-------------------------------------------------------------------------
Copyright 2008 Sandia Corporation.
Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
the U.S. Government retains certain rights in this software.
-------------------------------------------------------------------------*/
/**
* @class vtkKCoreDecomposition
* @brief Compute the k-core decomposition of the input graph.
*
*
* The k-core decomposition is a graph partitioning strategy that is useful for
* analyzing the structure of large networks. A k-core of a graph G is a maximal
* connected subgraph of G in which all vertices have degree at least k. The k-core
* membership for each vertex of the input graph is found on the vertex data of the
* output graph as an array named 'KCoreDecompositionNumbers' by default. The algorithm
* used to find the k-cores has O(number of graph edges) running time, and is described
* in the following reference paper.
*
* An O(m) Algorithm for Cores Decomposition of Networks
* V. Batagelj, M. Zaversnik, 2001
*
* @par Thanks:
* Thanks to Thomas Otahal from Sandia National Laboratories for providing this
* implementation.
*/
#ifndef vtkKCoreDecomposition_h
#define vtkKCoreDecomposition_h
#include "vtkInfovisCoreModule.h" // For export macro
#include "vtkGraphAlgorithm.h"
class vtkIntArray;
class VTKINFOVISCORE_EXPORT vtkKCoreDecomposition : public vtkGraphAlgorithm
{
public:
static vtkKCoreDecomposition *New();
vtkTypeMacro(vtkKCoreDecomposition, vtkGraphAlgorithm);
void PrintSelf(ostream& os, vtkIndent indent) VTK_OVERRIDE;
//@{
/**
* Set the output array name. If no output array name is
* set then the name 'KCoreDecompositionNumbers' is used.
*/
vtkSetStringMacro(OutputArrayName);
//@}
//@{
/**
* Directed graphs only. Use only the in edges to
* compute the vertex degree of a vertex. The default
* is to use both in and out edges to compute vertex
* degree.
*/
vtkSetMacro(UseInDegreeNeighbors, bool);
vtkGetMacro(UseInDegreeNeighbors, bool);
vtkBooleanMacro(UseInDegreeNeighbors, bool);
//@}
//@{
/**
* Directed graphs only. Use only the out edges to
* compute the vertex degree of a vertex. The default
* is to use both in and out edges to compute vertex
* degree.
*/
vtkSetMacro(UseOutDegreeNeighbors, bool);
vtkGetMacro(UseOutDegreeNeighbors, bool);
vtkBooleanMacro(UseOutDegreeNeighbors, bool);
//@}
//@{
/**
* Check the input graph for self loops and parallel
* edges. The k-core is not defined for graphs that
* contain either of these. Default is on.
*/
vtkSetMacro(CheckInputGraph, bool);
vtkGetMacro(CheckInputGraph, bool);
vtkBooleanMacro(CheckInputGraph, bool);
//@}
protected:
vtkKCoreDecomposition();
~vtkKCoreDecomposition() VTK_OVERRIDE;
int RequestData(vtkInformation *, vtkInformationVector **, vtkInformationVector *) VTK_OVERRIDE;
private:
char* OutputArrayName;
bool UseInDegreeNeighbors;
bool UseOutDegreeNeighbors;
bool CheckInputGraph;
// K-core partitioning implementation
void Cores(vtkGraph* g,
vtkIntArray* KCoreNumbers);
vtkKCoreDecomposition(const vtkKCoreDecomposition&) VTK_DELETE_FUNCTION;
void operator=(const vtkKCoreDecomposition&) VTK_DELETE_FUNCTION;
};
#endif