Add support for tables
This commit is contained in:
parent
10aa558152
commit
1018655f8c
@ -326,6 +326,24 @@ namespace MatFileHandler.Tests
|
||||
Assert.That(obj[2]["x"].ConvertToDoubleArray(), Is.EqualTo(new[] { -2.0 }));
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void TestTable()
|
||||
{
|
||||
var matFile = GetTests("good")["table"];
|
||||
var obj = matFile["table_"].Value as IMatObject;
|
||||
var table = new TableAdapter(obj);
|
||||
Assert.That(table.NumberOfRows, Is.EqualTo(3));
|
||||
Assert.That(table.NumberOfVariables, Is.EqualTo(2));
|
||||
Assert.That(table.Description, Is.EqualTo("Some table"));
|
||||
Assert.That(table.VariableNames, Is.EqualTo(new[] { "variable1", "variable2" }));
|
||||
var variable1 = table["variable1"] as ICellArray;
|
||||
Assert.That((variable1[0] as ICharArray).String, Is.EqualTo("First row"));
|
||||
Assert.That((variable1[1] as ICharArray).String, Is.EqualTo("Second row"));
|
||||
Assert.That((variable1[2] as ICharArray).String, Is.EqualTo("Third row"));
|
||||
var variable2 = table["variable2"];
|
||||
Assert.That(variable2.ConvertToDoubleArray(), Is.EqualTo(new[] { 1.0, 3.0, 5.0, 2.0, 4.0, 6.0 }));
|
||||
}
|
||||
|
||||
private static AbstractTestDataFactory<IMatFile> GetTests(string factoryName) =>
|
||||
new MatTestDataFactory(Path.Combine(TestDirectory, factoryName));
|
||||
|
||||
|
BIN
MatFileHandler.Tests/test-data/good/table.mat
Normal file
BIN
MatFileHandler.Tests/test-data/good/table.mat
Normal file
Binary file not shown.
60
MatFileHandler/TableAdapter.cs
Normal file
60
MatFileHandler/TableAdapter.cs
Normal file
@ -0,0 +1,60 @@
|
||||
using System;
|
||||
using System.Linq;
|
||||
|
||||
namespace MatFileHandler
|
||||
{
|
||||
public class TableAdapter
|
||||
{
|
||||
private readonly IMatObject matObject;
|
||||
|
||||
public TableAdapter(IArray array)
|
||||
{
|
||||
matObject = array as IMatObject;
|
||||
if (matObject?.ClassName != "table")
|
||||
{
|
||||
throw new ArgumentException("The object provided is not a table.");
|
||||
}
|
||||
var cellArray = matObject["varnames"] as ICellArray;
|
||||
VariableNames = Enumerable
|
||||
.Range(0, cellArray.Count)
|
||||
.Select(i => (cellArray[i] as ICharArray).String)
|
||||
.ToArray();
|
||||
NumberOfVariables = VariableNames.Length;
|
||||
var props = matObject["props"] as IStructureArray;
|
||||
Description = (props["Description"] as ICharArray).String;
|
||||
NumberOfRows = (int) matObject["nrows"].ConvertToDoubleArray()[0];
|
||||
var rowNamesArrays = matObject["rownames"] as ICellArray;
|
||||
RowNames = Enumerable
|
||||
.Range(0, rowNamesArrays.Count)
|
||||
.Select(i => (cellArray[i] as ICharArray).String)
|
||||
.ToArray();
|
||||
}
|
||||
|
||||
public int NumberOfRows { get; }
|
||||
|
||||
public string[] RowNames { get; }
|
||||
|
||||
public int NumberOfVariables { get; }
|
||||
|
||||
public string[] VariableNames { get; }
|
||||
|
||||
public IArray this[string variableName]
|
||||
{
|
||||
get
|
||||
{
|
||||
var maybeIndex = Enumerable.Range(0, VariableNames.Length)
|
||||
.Where(i => VariableNames[i] == variableName)
|
||||
.Select(i => (int?)i)
|
||||
.FirstOrDefault();
|
||||
if (!(maybeIndex is int index))
|
||||
{
|
||||
throw new IndexOutOfRangeException($"Variable '{variableName}' not found.");
|
||||
}
|
||||
var data = matObject["data"] as ICellArray;
|
||||
return data[index];
|
||||
}
|
||||
}
|
||||
|
||||
public string Description { get; }
|
||||
}
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user