From c1e144cc1e693f529160a56c59a16f76a4566b03 Mon Sep 17 00:00:00 2001
From: Guillaume Dalle <22795598+gdalle@users.noreply.github.com>
Date: Thu, 8 Aug 2024 10:12:50 +0200
Subject: [PATCH] Remove preparation to speed up ADTypes interface

---
 docs/src/api.md  |  8 --------
 src/interface.jl | 27 +++++++++------------------
 2 files changed, 9 insertions(+), 26 deletions(-)

diff --git a/docs/src/api.md b/docs/src/api.md
index bc40ce5a..26bde61c 100644
--- a/docs/src/api.md
+++ b/docs/src/api.md
@@ -30,14 +30,6 @@ column_groups
 row_groups
 ```
 
-### ADTypes interface
-
-```@docs
-column_coloring
-row_coloring
-symmetric_coloring
-```
-
 ## Public, not exported
 
 ### Decompression
diff --git a/src/interface.jl b/src/interface.jl
index 765be34e..2e294561 100644
--- a/src/interface.jl
+++ b/src/interface.jl
@@ -179,29 +179,20 @@ end
 
 ## ADTypes interface
 
-"""
-    ADTypes.column_coloring(S::AbstractMatrix, algo::GreedyColoringAlgorithm)
-
-Call [`column_coloring_detailed`](@ref) and return only the vector of column colors.
-"""
 function ADTypes.column_coloring(S::AbstractMatrix, algo::GreedyColoringAlgorithm)
-    return column_colors(column_coloring_detailed(S, algo))
+    bg = bipartite_graph(S)
+    color = partial_distance2_coloring(bg, Val(2), algo.order)
+    return color
 end
 
-"""
-    ADTypes.row_coloring(S::AbstractMatrix, algo::GreedyColoringAlgorithm)
-
-Call [`row_coloring_detailed`](@ref) and return only the vector of column colors.
-"""
 function ADTypes.row_coloring(S::AbstractMatrix, algo::GreedyColoringAlgorithm)
-    return row_colors(row_coloring_detailed(S, algo))
+    bg = bipartite_graph(S)
+    color = partial_distance2_coloring(bg, Val(1), algo.order)
+    return color
 end
 
-"""
-    ADTypes.symmetric_coloring(S::AbstractMatrix, algo::GreedyColoringAlgorithm)
-
-Call [`symmetric_coloring_detailed`](@ref) and return only the vector of column colors.
-"""
 function ADTypes.symmetric_coloring(S::AbstractMatrix, algo::GreedyColoringAlgorithm)
-    return column_colors(symmetric_coloring_detailed(S, algo))
+    ag = adjacency_graph(S)
+    color, star_set = star_coloring(ag, algo.order)
+    return color
 end