From 83338a2f753ebbb7ba7b154cc60b2f427b6121b0 Mon Sep 17 00:00:00 2001
From: Sebastiano Tronto <sebastiano@tronto.net>
Date: Wed, 26 Mar 2025 19:22:50 +0100
Subject: Added inverse coordinate for CO

this is necessary for DR coordinate
---
 src/arch/avx2.h | 49 ++++++++++++++++++++++++++++++++++---------------
 1 file changed, 34 insertions(+), 15 deletions(-)

(limited to 'src/arch/avx2.h')

diff --git a/src/arch/avx2.h b/src/arch/avx2.h
index ebcf84f..4682e33 100644
--- a/src/arch/avx2.h
+++ b/src/arch/avx2.h
@@ -188,6 +188,25 @@ coord_co(cube_t c)
 	return ret;
 }
 
+STATIC_INLINE cube_t
+invcoord_co(int64_t coord)
+{
+	int64_t i, c, p, co, mem[4] = {0};
+	cube_t cube, cc;
+
+	for (i = 0, p = 0, c = coord; i < 8; i++, c /= 3) {
+		co = i == 7 ? ((3 - (p % 3)) % 3) : (c % 3);
+		p += co;
+		mem[0] |= (int64_t)(i + (co << COSHIFT)) << (int64_t)(8 * i);
+	}
+
+	cc = _mm256_loadu_si256((const __m256i *)mem);
+	cube = SOLVED_CUBE;
+	copy_corners(&cube, cc);
+
+	return cube;
+}
+
 STATIC_INLINE int64_t
 coord_csep(cube_t c)
 {
@@ -251,6 +270,21 @@ coord_esep(cube_t c)
 	return ret1 * 70 + ret2;
 }
 
+STATIC_INLINE cube_t
+invcoord_esep(int64_t esep)
+{
+	cube_t eee, ret;
+	uint8_t mem[32] = {0};
+
+	invcoord_esep_array(esep % 70, esep / 70, mem+16);
+
+	ret = SOLVED_CUBE;
+	eee = _mm256_loadu_si256((__m256i_u *)&mem);
+	copy_edges(&ret, eee);
+
+	return ret;
+}
+
 STATIC_INLINE void
 copy_corners(cube_t dest[static 1], cube_t src)
 {
@@ -287,18 +321,3 @@ set_eo(cube_t cube[static 1], int64_t eo)
 	*cube = _mm256_andnot_si256(EO_AVX2, *cube);
 	*cube = _mm256_or_si256(*cube, veo);
 }
-
-STATIC_INLINE cube_t
-invcoord_esep(int64_t esep)
-{
-	cube_t eee, ret;
-	uint8_t mem[32] = {0};
-
-	invcoord_esep_array(esep % 70, esep / 70, mem+16);
-
-	ret = SOLVED_CUBE;
-	eee = _mm256_loadu_si256((__m256i_u *)&mem);
-	copy_edges(&ret, eee);
-
-	return ret;
-}
-- 
cgit v1.3