You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

121 lines
3.8 KiB
C

/*=========================================================================
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 "vtkGraphAlgorithm.h"
#include "vtkInfovisCoreModule.h" // For export macro
class vtkIntArray;
class VTKINFOVISCORE_EXPORT vtkKCoreDecomposition : public vtkGraphAlgorithm
{
public:
static vtkKCoreDecomposition* New();
vtkTypeMacro(vtkKCoreDecomposition, vtkGraphAlgorithm);
void PrintSelf(ostream& os, vtkIndent indent) 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() override;
int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
private:
char* OutputArrayName;
bool UseInDegreeNeighbors;
bool UseOutDegreeNeighbors;
bool CheckInputGraph;
// K-core partitioning implementation
void Cores(vtkGraph* g, vtkIntArray* KCoreNumbers);
vtkKCoreDecomposition(const vtkKCoreDecomposition&) = delete;
void operator=(const vtkKCoreDecomposition&) = delete;
};
#endif