That-One-Nerd/Nerd_STF
1
0
Fork 0
Code Issues 63 Pull Requests Actions Packages Projects Releases 19 Wiki Activity

Added a new ND array. Maybe a list will come soon.

Browse Source
This commit is contained in:
That_One_Nerd 2023-06-07 12:53:24 -04:00
parent f6ff0aac6c
commit 51ad1974d5
1 changed files with 115 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

115
Nerd_STF/NDArray.cs Normal file
View File

@ -0,0 +1,115 @@
namespace Nerd_STF;
public class NDArray<T> : IEnumerable<T>, IEquatable<NDArray<T>>
{
public int Dimensions => dimensions;
public int[] Lengths => sizes;
public long FullLength
{
get
{
long prod = 1;
foreach (int size in sizes) prod *= size;
return prod;
}
}
protected readonly T[] arr;
protected readonly int dimensions;
protected readonly int[] sizes;
public NDArray()
{
arr = Array.Empty<T>();
dimensions = 0;
sizes = Array.Empty<int>();
}
public NDArray(int dimensions)
{
arr = Array.Empty<T>();
this.dimensions = dimensions;
sizes = new int[dimensions];
Array.Fill(sizes, 0);
}
public NDArray(int dimensions, int allLengths)
{
this.dimensions = dimensions;
sizes = new int[dimensions];
Array.Fill(sizes, allLengths);
arr = new T[Mathf.Product(sizes)];
Console.WriteLine("Gay ass balls");
}
public NDArray(int[] lengths)
{
arr = new T[Mathf.Product(lengths)];
dimensions = lengths.Length;
sizes = lengths;
}
public NDArray(int dimensions, int[] lengths)
{
if (dimensions != lengths.Length) throw new InvalidSizeException("Dimension count doesn't match length count.");
arr = new T[Mathf.Product(lengths)];
this.dimensions = lengths.Length;
sizes = lengths;
}
public NDArray(T[] items, int[] lengths)
{
arr = items;
dimensions = lengths.Length;
sizes = lengths;
if (arr.Length != Mathf.Product(lengths)) throw new InvalidSizeException("Too many or too few items were provided.");
}
public NDArray(T[] items, int dimensions, int[] lengths)
{
if (dimensions != lengths.Length) throw new InvalidSizeException("Dimension count doesn't match length count.");
arr = items;
this.dimensions = lengths.Length;
sizes = lengths;
if (arr.Length != Mathf.Product(lengths)) throw new InvalidSizeException("Too many or too few items were provided.");
}
public T this[params int[] indexes]
{
get => arr[FlattenIndex(indexes)];
set => arr[FlattenIndex(indexes)] = value;
}
private int FlattenIndex(params int[] indexes)
{
if (indexes.Length != sizes.Length) throw new InvalidSizeException("Too many or too few indexes were provided.");
int ind = indexes[^1];
Console.WriteLine($"Start at {ind}");
for (int i = indexes.Length - 2; i >= 0; i--)
{
Console.WriteLine($"[{ind}] * {sizes[i]} + {indexes[i]}");
ind = ind * sizes[i] + indexes[i];
}
return ind;
}
IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
public IEnumerator<T> GetEnumerator()
{
foreach (T item in arr) yield return item;
}
public override bool Equals(object? obj)
{
if (obj is null) return false;
else if (obj is NDArray<T> arr) return Equals(arr);
return false;
}
public bool Equals(NDArray<T>? obj) => obj is not null && arr.Equals(obj.arr);
public override int GetHashCode() => base.GetHashCode();
public static bool operator ==(NDArray<T> a, NDArray<T> b) => a.Equals(b);
public static bool operator !=(NDArray<T> a, NDArray<T> b) => !a.Equals(b);
}