com.facebook.presto.jdbc.internal.guava.graph.AbstractBaseGraph Maven / Gradle / Ivy
/*
* Copyright (C) 2017 The Guava Authors
*
* 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.
*/
package com.facebook.presto.jdbc.internal.guava.graph;
import static com.facebook.presto.jdbc.internal.guava.base.Preconditions.checkArgument;
import static com.facebook.presto.jdbc.internal.guava.base.Preconditions.checkNotNull;
import static com.facebook.presto.jdbc.internal.guava.base.Preconditions.checkState;
import static com.facebook.presto.jdbc.internal.guava.graph.GraphConstants.ENDPOINTS_MISMATCH;
import com.facebook.presto.jdbc.internal.guava.collect.ImmutableSet;
import com.facebook.presto.jdbc.internal.guava.collect.Iterators;
import com.facebook.presto.jdbc.internal.guava.collect.Sets;
import com.facebook.presto.jdbc.internal.guava.collect.UnmodifiableIterator;
import com.facebook.presto.jdbc.internal.guava.math.IntMath;
import com.facebook.presto.jdbc.internal.guava.primitives.Ints;
import java.util.AbstractSet;
import java.util.Set;
import com.facebook.presto.jdbc.internal.javax.annotation.CheckForNull;
/**
* This class provides a skeletal implementation of {@link BaseGraph}.
*
* The methods implemented in this class should not be overridden unless the subclass admits a
* more efficient implementation.
*
* @author James Sexton
* @param Node parameter type
*/
@ElementTypesAreNonnullByDefault
abstract class AbstractBaseGraph implements BaseGraph {
/**
* Returns the number of edges in this graph; used to calculate the size of {@link Graph#edges()}.
* This implementation requires O(|N|) time. Classes extending this one may manually keep track of
* the number of edges as the graph is updated, and override this method for better performance.
*/
protected long edgeCount() {
long degreeSum = 0L;
for (N node : nodes()) {
degreeSum += degree(node);
}
// According to the degree sum formula, this is equal to twice the number of edges.
checkState((degreeSum & 1) == 0);
return degreeSum >>> 1;
}
/**
* An implementation of {@link BaseGraph#edges()} defined in terms of {@link Graph#nodes()} and
* {@link #successors(Object)}.
*/
@Override
public Set> edges() {
return new AbstractSet>() {
@Override
public UnmodifiableIterator> iterator() {
return EndpointPairIterator.of(AbstractBaseGraph.this);
}
@Override
public int size() {
return Ints.saturatedCast(edgeCount());
}
@Override
public boolean remove(@CheckForNull Object o) {
throw new UnsupportedOperationException();
}
// Mostly safe: We check contains(u) before calling successors(u), so we perform unsafe
// operations only in weird cases like checking for an EndpointPair in a
// Graph.
@SuppressWarnings("unchecked")
@Override
public boolean contains(@CheckForNull Object obj) {
if (!(obj instanceof EndpointPair)) {
return false;
}
EndpointPair endpointPair = (EndpointPair) obj;
return isOrderingCompatible(endpointPair)
&& nodes().contains(endpointPair.nodeU())
&& successors((N) endpointPair.nodeU()).contains(endpointPair.nodeV());
}
};
}
@Override
public ElementOrder incidentEdgeOrder() {
return ElementOrder.unordered();
}
@Override
public Set> incidentEdges(N node) {
checkNotNull(node);
checkArgument(nodes().contains(node), "Node %s is not an element of this graph.", node);
return new IncidentEdgeSet(this, node) {
@Override
public UnmodifiableIterator> iterator() {
if (graph.isDirected()) {
return Iterators.unmodifiableIterator(
Iterators.concat(
Iterators.transform(
graph.predecessors(node).iterator(),
(N predecessor) -> EndpointPair.ordered(predecessor, node)),
Iterators.transform(
// filter out 'node' from successors (already covered by predecessors, above)
Sets.difference(graph.successors(node), ImmutableSet.of(node)).iterator(),
(N successor) -> EndpointPair.ordered(node, successor))));
} else {
return Iterators.unmodifiableIterator(
Iterators.transform(
graph.adjacentNodes(node).iterator(),
(N adjacentNode) -> EndpointPair.unordered(node, adjacentNode)));
}
}
};
}
@Override
public int degree(N node) {
if (isDirected()) {
return IntMath.saturatedAdd(predecessors(node).size(), successors(node).size());
} else {
Set neighbors = adjacentNodes(node);
int selfLoopCount = (allowsSelfLoops() && neighbors.contains(node)) ? 1 : 0;
return IntMath.saturatedAdd(neighbors.size(), selfLoopCount);
}
}
@Override
public int inDegree(N node) {
return isDirected() ? predecessors(node).size() : degree(node);
}
@Override
public int outDegree(N node) {
return isDirected() ? successors(node).size() : degree(node);
}
@Override
public boolean hasEdgeConnecting(N nodeU, N nodeV) {
checkNotNull(nodeU);
checkNotNull(nodeV);
return nodes().contains(nodeU) && successors(nodeU).contains(nodeV);
}
@Override
public boolean hasEdgeConnecting(EndpointPair endpoints) {
checkNotNull(endpoints);
if (!isOrderingCompatible(endpoints)) {
return false;
}
N nodeU = endpoints.nodeU();
N nodeV = endpoints.nodeV();
return nodes().contains(nodeU) && successors(nodeU).contains(nodeV);
}
/**
* Throws {@code IllegalArgumentException} if the ordering of {@code endpoints} is not compatible
* with the directionality of this graph.
*/
protected final void validateEndpoints(EndpointPair endpoints) {
checkNotNull(endpoints);
checkArgument(isOrderingCompatible(endpoints), ENDPOINTS_MISMATCH);
}
/**
* Returns {@code true} iff {@code endpoints}' ordering is compatible with the directionality of
* this graph.
*/
protected final boolean isOrderingCompatible(EndpointPair endpoints) {
return endpoints.isOrdered() == this.isDirected();
}
}