1
0
Fork 0
arangodb/arangod/Pregel/Algos/SCC.h

69 lines
2.5 KiB
C++

////////////////////////////////////////////////////////////////////////////////
/// DISCLAIMER
///
/// Copyright 2016 ArangoDB GmbH, Cologne, Germany
///
/// 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
///
/// http://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.
///
/// Copyright holder is ArangoDB GmbH, Cologne, Germany
///
/// @author Simon Grätzer
////////////////////////////////////////////////////////////////////////////////
#ifndef ARANGODB_PREGEL_ALGOS_SCC_H
#define ARANGODB_PREGEL_ALGOS_SCC_H 1
#include "Pregel/Algorithm.h"
#include "Pregel/CommonFormats.h"
namespace arangodb {
namespace pregel {
namespace algos {
/// Finds strongly connected components of the graph.
///
/// 1. Each vertex starts with its vertex id as its "color".
/// 2. Remove vertices which cannot be in a SCC (no incoming or no outgoing
/// edges)
/// 3. Propagate the color forward from each vertex, accept a neighbor's color
/// if it's smaller than yours.
/// At convergence, vertices with the same color represents all nodes that
/// are visitable from the root of that color.
/// 4. Reverse the graph.
/// 5. Start at all roots, walk the graph. Visit a neighbor if it has the same
/// color as you.
/// All nodes visited belongs to the SCC identified by the root color.
struct SCC : public SimpleAlgorithm<SCCValue, int8_t, SenderMessage<uint64_t>> {
public:
explicit SCC(application_features::ApplicationServer& server, VPackSlice userParams)
: SimpleAlgorithm<SCCValue, int8_t, SenderMessage<uint64_t>>(server, "SCC", userParams) {
}
GraphFormat<SCCValue, int8_t>* inputFormat() const override;
MessageFormat<SenderMessage<uint64_t>>* messageFormat() const override {
return new SenderMessageFormat<uint64_t>();
}
VertexComputation<SCCValue, int8_t, SenderMessage<uint64_t>>* createComputation(
WorkerConfig const*) const override;
MasterContext* masterContext(VPackSlice userParams) const override;
IAggregator* aggregator(std::string const& name) const override;
};
} // namespace algos
} // namespace pregel
} // namespace arangodb
#endif