# distutils: language=c++

from cython.operator import dereference, preincrement

from libcpp cimport bool as bool_t
from libcpp.vector cimport vector
from libcpp.utility cimport pair
from libcpp.string cimport string
from libcpp.unordered_set cimport unordered_set

from .base cimport _Algorithm, Algorithm
from .structures cimport edgeid, index, count, node, edgeweight

cdef extern from "<algorithm>" namespace "std":
	void swap[T](T &a,  T &b)
	_Graph move( _Graph t ) nogil
	vector[double] move(vector[double])

cdef extern from "cython_helper.h":
	void throw_runtime_error(string message)

cdef extern from "<networkit/Globals.hpp>" namespace "NetworKit":

	index _none "NetworKit::none"

cdef extern from "<networkit/graph/Graph.hpp>":

	cdef struct Edge "NetworKit::Edge":
		node u
		node v

	cdef struct WeightedEdge "NetworKit::WeightedEdge":
		node u
		node v
		edgeweight weight

	cdef cppclass _Graph "NetworKit::Graph":
		_Graph() except +
		_Graph(count, bool_t, bool_t, bool_t) except +
		_Graph(const _Graph& other) except +
		_Graph(const _Graph& other, bool_t weighted, bool_t directed, bool_t edgesIndexed) except +
		void indexEdges(bool_t) except +
		bool_t hasEdgeIds() except +
		edgeid edgeId(node, node) except +
		count numberOfNodes() except +
		count numberOfEdges() except +
		index upperNodeIdBound() except +
		index upperEdgeIdBound() except +
		count degree(node u) except +
		count degreeIn(node u) except +
		count degreeOut(node u) except +
		double weightedDegree(node u, bool_t) except +
		double weightedDegreeIn(node u, bool_t) except +
		bool_t isIsolated(node u) except +
		node addNode() except +
		node addNodes(node) except +
		void removeNode(node u) except +
		bool_t hasNode(node u) except +
		void restoreNode(node u) except +
		bool_t addEdge(node u, node v, edgeweight w, bool_t checkMultiEdge) except +
		void setWeight(node u, node v, edgeweight w) except +
		void increaseWeight(node u, node v, edgeweight w) except +
		void removeEdge(node u, node v) except +
		void removeAllEdges() except +
		void removeSelfLoops() except +
		void removeMultiEdges() except +
		void swapEdge(node s1, node t1, node s2, node t2) except +
		void compactEdges() except +
		void sortEdges() except +
		bool_t hasEdge(node u, node v) except +
		edgeweight weight(node u, node v) except +
		void forEdges[Callback](Callback c) except +
		void forNodes[Callback](Callback c) except +
		void forNodePairs[Callback](Callback c) except +
		void forNodesInRandomOrder[Callback](Callback c) except +
		void forEdgesOf[Callback](node u, Callback c) except +
		void forInEdgesOf[Callback](node u, Callback c) except +
		bool_t isWeighted() except +
		bool_t isDirected() except +
		edgeweight totalEdgeWeight() except +
		count numberOfSelfLoops() except +
		bool_t checkConsistency() except +
		_NodeRange nodeRange() except +
		_EdgeRange edgeRange() except +
		_EdgeWeightRange edgeWeightRange() except +
		_OutNeighborRange neighborRange(node u) except +
		_InNeighborRange inNeighborRange(node u) except +
		_OutNeighborWeightRange weightNeighborRange(node u) except +
		_InNeighborWeightRange weightInNeighborRange(node u) except +
		_NodeIntAttribute attachNodeIntAttribute(string) except +
		_NodeDoubleAttribute attachNodeDoubleAttribute(string) except +
		_NodeStringAttribute attachNodeStringAttribute(string) except +
		_NodeIntAttribute getNodeIntAttribute(string) except +
		_NodeDoubleAttribute getNodeDoubleAttribute(string) except +
		_NodeStringAttribute getNodeStringAttribute(string) except +
		void detachNodeAttribute(string) except +
		_EdgeIntAttribute attachEdgeIntAttribute(string) except +
		_EdgeDoubleAttribute attachEdgeDoubleAttribute(string) except +
		_EdgeStringAttribute attachEdgeStringAttribute(string) except +
		_EdgeIntAttribute getEdgeIntAttribute(string) except +
		_EdgeDoubleAttribute getEdgeDoubleAttribute(string) except +
		_EdgeStringAttribute getEdgeStringAttribute(string) except +
		void detachEdgeAttribute(string) except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _NodeIterator "NetworKit::Graph::NodeIterator":
		_NodeIterator& operator++() except +
		_NodeIterator operator++(int) except +
		bool_t operator!=(const _NodeIterator) except +
		node operator*() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _NodeRange "NetworKit::Graph::NodeRange":
		_NodeIterator begin() except +
		_NodeIterator end() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _EdgeWeightIterator "NetworKit::Graph::EdgeWeightIterator":
		_EdgeWeightIterator& operator++() except +
		_EdgeWeightIterator operator++(int) except +
		bool_t operator!=(const _EdgeWeightIterator) except +
		WeightedEdge operator*() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _EdgeWeightRange "NetworKit::Graph::EdgeWeightRange":
		_EdgeWeightIterator begin() except +
		_EdgeWeightIterator end() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _EdgeIterator "NetworKit::Graph::EdgeIterator":
		_EdgeIterator& operator++() except +
		_EdgeIterator operator++(int) except +
		bool_t operator!=(const _EdgeIterator) except +
		Edge operator*() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _EdgeRange "NetworKit::Graph::EdgeRange":
		_EdgeIterator begin() except +
		_EdgeIterator end() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _NeighborIterator "NetworKit::Graph::NeighborIterator":
		_NeighborIterator& operator++() except +
		_NeighborIterator operator++(int) except +
		bool_t operator!=(const _NeighborIterator) except +
		node operator*() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _OutNeighborRange "NetworKit::Graph::OutNeighborRange":
		_NeighborIterator begin() except +
		_NeighborIterator end() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _InNeighborRange "NetworKit::Graph::InNeighborRange":
		_NeighborIterator begin() except +
		_NeighborIterator end() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _NeighborWeightIterator "NetworKit::Graph::NeighborWeightIterator":
		_NeighborWeightIterator& operator++() except +
		_NeighborWeightIterator operator++(int) except +
		bool_t operator!=(const _NeighborWeightIterator) except +
		pair[node, edgeweight] operator*() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _OutNeighborWeightRange "NetworKit::Graph::OutNeighborWeightRange":
		_NeighborWeightIterator begin() except +
		_NeighborWeightIterator end() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _InNeighborWeightRange "NetworKit::Graph::InNeighborWeightRange":
		_NeighborWeightIterator begin() except +
		_NeighborWeightIterator end() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _NodeIntAttribute "NetworKit::Graph::NodeIntAttribute":
		cppclass _AttributeIterator "Iterator":
			_AttributeIterator operator++()
			pair[node, int] operator*()
			bool_t operator==(const _AttributeIterator)
			bool_t operator!=(const _AttributeIterator)
		_AttributeIterator begin() except +
		_AttributeIterator end()
		index size()
		void set(node, int) except +
		int get(node) except +
		int get(node, int) except +
		void write(string) except +
		void read(string) except +
		void swap(_NodeIntAttribute& other)
		string getName() except +	

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _NodeDoubleAttribute "NetworKit::Graph::NodeDoubleAttribute":
		cppclass _AttributeIterator "Iterator":
			_AttributeIterator operator++()
			pair[node, double] operator*()
			bool_t operator==(const _AttributeIterator)
			bool_t operator!=(const _AttributeIterator)
		_AttributeIterator begin() except +
		_AttributeIterator end()
		index size()
		void set(node, double) except +
		double get(node) except +
		double get(node, double) except +
		void write(string) except +
		void read(string) except +
		void swap(_NodeDoubleAttribute& other)
		string getName() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _NodeStringAttribute "NetworKit::Graph::NodeStringAttribute":
		cppclass _AttributeIterator "Iterator":
			_AttributeIterator operator++()
			pair[node, string] operator*()
			bool_t operator==(const _AttributeIterator)
			bool_t operator!=(const _AttributeIterator)
		_AttributeIterator begin() except +
		_AttributeIterator end()
		index size()
		void set(node, string) except +
		string get(node) except +
		string get(node, string) except +
		void write(string) except +
		void read(string) except +
		void swap(_NodeStringAttribute& other)
		string getName() except +

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _EdgeIntAttribute "NetworKit::Graph::EdgeIntAttribute":
		cppclass _AttributeIterator "Iterator":
			_AttributeIterator operator++()
			pair[edgeid, int] operator*()
			bool_t operator==(const _AttributeIterator)
			bool_t operator!=(const _AttributeIterator)
		_AttributeIterator begin() except +
		_AttributeIterator end()
		index size()
		void set(edgeid, int) except +
		void set2(node, node, int) except +
		int get(edgeid) except +
		int get2(node, node) except +
		int get(edgeid, int) except +
		int get2(node, node, int) except +
		void write(string) except +
		void read(string) except +
		void swap(_EdgeIntAttribute& other)

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _EdgeDoubleAttribute "NetworKit::Graph::EdgeDoubleAttribute":
		cppclass _AttributeIterator "Iterator":
			_AttributeIterator operator++()
			pair[edgeid, double] operator*()
			bool_t operator==(const _AttributeIterator)
			bool_t operator!=(const _AttributeIterator)
		_AttributeIterator begin() except +
		_AttributeIterator end()
		index size()
		void set(edgeid, double) except +
		void set2(node, node, double) except +
		double get(edgeid) except +
		double get2(node, node) except +
		double get(edgeid, double) except +
		double get2(node, node, double) except +
		void write(string) except +
		void read(string) except +
		void swap(_EdgeDoubleAttribute& other)

cdef extern from "<networkit/graph/Graph.hpp>":
	cdef cppclass _EdgeStringAttribute "NetworKit::Graph::EdgeStringAttribute":
		cppclass _AttributeIterator "Iterator":
			_AttributeIterator operator++()
			pair[edgeid, string] operator*()
			bool_t operator==(const _AttributeIterator)
			bool_t operator!=(const _AttributeIterator)
		_AttributeIterator begin() except +
		_AttributeIterator end()
		index size()
		void set(edgeid, string) except +
		void set2(node, node, string) except +
		string get(edgeid) except +
		string get2(node, node) except +
		string get(edgeid, string) except +
		string get2(node, node, string) except +
		void write(string) except +
		void read(string) except +
		void swap(_EdgeStringAttribute& other)

cdef class Graph:
	cdef _Graph _this
	cdef setThis(self, _Graph& other)

cdef class NodeIntAttribute:
	cdef _NodeIntAttribute _this
	cdef _NodeIntAttribute._AttributeIterator _iter
	cdef _NodeIntAttribute._AttributeIterator _stopiter
	cdef _Graph* _G
	cdef setThis(self, _NodeIntAttribute& other, _Graph* graph)

cdef class NodeDoubleAttribute:
	cdef _NodeDoubleAttribute _this
	cdef _NodeDoubleAttribute._AttributeIterator _iter
	cdef _NodeDoubleAttribute._AttributeIterator _stopiter
	cdef _Graph* _G
	cdef setThis(self, _NodeDoubleAttribute& other, _Graph* graph)

cdef class NodeStringAttribute:
	cdef _NodeStringAttribute _this
	cdef _NodeStringAttribute._AttributeIterator _iter
	cdef _NodeStringAttribute._AttributeIterator _stopiter
	cdef _Graph* _G
	cdef setThis(self, _NodeStringAttribute& other, _Graph* graph)

cdef class EdgeIntAttribute:
	cdef _EdgeIntAttribute _this
	cdef _EdgeIntAttribute._AttributeIterator _iter
	cdef _EdgeIntAttribute._AttributeIterator _stopiter
	cdef _Graph* _G
	cdef setThis(self, _EdgeIntAttribute& other, _Graph* graph)

cdef class EdgeDoubleAttribute:
	cdef _EdgeDoubleAttribute _this
	cdef _EdgeDoubleAttribute._AttributeIterator _iter
	cdef _EdgeDoubleAttribute._AttributeIterator _stopiter
	cdef _Graph* _G
	cdef setThis(self, _EdgeDoubleAttribute& other, _Graph* graph)

cdef class EdgeStringAttribute:
	cdef _EdgeStringAttribute _this
	cdef _EdgeStringAttribute._AttributeIterator _iter
	cdef _EdgeStringAttribute._AttributeIterator _stopiter
	cdef _Graph* _G
	cdef setThis(self, _EdgeStringAttribute& other, _Graph* graph)

cdef extern from "<networkit/graph/SpanningForest.hpp>":

	cdef cppclass _SpanningForest "NetworKit::SpanningForest":
		_SpanningForest(_Graph) except +
		void run() except + nogil
		_Graph getForest() except +

cdef extern from "<networkit/graph/KruskalMSF.hpp>":
	
	cdef cppclass _KruskalMSF "NetworKit::KruskalMSF"(_SpanningForest):
		_KruskalMSF(_Graph) except +
		edgeweight getTotalWeight() except +

cdef extern from "<networkit/graph/PrimMSF.hpp>":

	cdef cppclass _PrimMSF "NetworKit::PrimMSF"(_SpanningForest):
		_PrimMSF(_Graph) except +
		edgeweight getTotalWeight() except +

cdef extern from "<networkit/graph/RandomMaximumSpanningForest.hpp>":

	cdef cppclass _RandomMaximumSpanningForest "NetworKit::RandomMaximumSpanningForest"(_Algorithm):
		_RandomMaximumSpanningForest(_Graph) except +
		_RandomMaximumSpanningForest(_Graph, vector[double]) except +
		_Graph getMSF(bool_t move) except +
		vector[bool_t] getAttribute(bool_t move) except +
		bool_t inMSF(edgeid eid) except +
		bool_t inMSF(node u, node v) except +

cdef class RandomMaximumSpanningForest(Algorithm):
	cdef vector[double] _attribute
	cdef Graph _G


cdef extern from "<networkit/graph/UnionMaximumSpanningForest.hpp>":

	cdef cppclass _UnionMaximumSpanningForest "NetworKit::UnionMaximumSpanningForest"(_Algorithm):
		_UnionMaximumSpanningForest(_Graph) except +
		_UnionMaximumSpanningForest(_Graph, vector[double]) except +
		_Graph getUMSF(bool_t move) except +
		vector[bool_t] getAttribute(bool_t move) except +
		bool_t inUMSF(edgeid eid) except +
		bool_t inUMSF(node u, node v) except +

cdef class UnionMaximumSpanningForest(Algorithm):
	cdef Graph _G
