FlattenTransposed()

Nixill · Dec 12, 2024 · 0c9ae8f · 0c9ae8f
1 parent b2ba15b
commit 0c9ae8f
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/CSharp.Nixill/src/Collections/Grid/Grid.cs b/CSharp.Nixill/src/Collections/Grid/Grid.cs
@@ -226,6 +226,9 @@ public bool Contains(T? item)
     public IEnumerable<(T? Item, GridReference Reference)> Flatten()
       => this.SelectMany((r, y) => r.Select((i, x) => (i, GridReference.XY(x, y))));
 
+    public IEnumerable<(T? Item, GridReference Reference)> FlattenTransposed()
+      => this.Columns.SelectMany((c, x) => c.Select((i, y) => (i, GridReference.XY(x, y))));
+
     public IEnumerator<IEnumerable<T?>> GetColumnEnumerator() => Columns.GetEnumerator();
     public IEnumerator<IEnumerable<T?>> GetEnumerator() => Rows.GetEnumerator();
 

diff --git a/CSharp.Nixill/src/Collections/Grid/IGrid.cs b/CSharp.Nixill/src/Collections/Grid/IGrid.cs
@@ -147,10 +147,16 @@ public interface IGrid<T> : IEnumerable<IEnumerable<T?>>
 
     /// <summary>
     /// Returns the items and references within a grid as a
-    /// one-dimensional enumerable.
+    /// one-dimensional enumerable, rows-first.
     /// </summary>
     IEnumerable<(T? Item, GridReference Reference)> Flatten();
 
+    /// <summary>
+    /// Returns the items and references within a grid as a
+    /// one-dimensional enumerable, columns-first.
+    /// </summary>
+    IEnumerable<(T? Item, GridReference Reference)> FlattenTransposed();
+
     /// <summary>
     /// Returns a single column as a subclass of IList.
     /// </summary>

diff --git a/CSharp.Nixill/src/Collections/Grid/OffsetGrid.cs b/CSharp.Nixill/src/Collections/Grid/OffsetGrid.cs
@@ -160,7 +160,10 @@ public void Clear()
     public bool Contains(T? item) => BackingGrid.Contains(item);
 
     public IEnumerable<(T? Item, GridReference Reference)> Flatten()
-      => this.SelectMany((l, r) => l.Select((i, c) => (i, GridReference.XY(c - ColumnOffset, r - RowOffset))));
+      => this.SelectMany((r, y) => r.Select((i, x) => (i, GridReference.XY(x - ColumnOffset, y - RowOffset))));
+
+    public IEnumerable<(T? Item, GridReference Reference)> FlattenTransposed()
+      => this.Columns.SelectMany((c, x) => c.Select((i, y) => (i, GridReference.XY(x - ColumnOffset, y - RowOffset))));
 
     public IList<T?> GetColumn(int index) => BackingGrid.GetColumn(index + ColumnOffset);
     public IEnumerator<IEnumerable<T?>> GetColumnEnumerator() => BackingGrid.GetColumnEnumerator();