MatFileHandler/MatFileHandler/TableAdapter.cs

101 lines
3.8 KiB
C#

// Copyright 2017-2018 Alexander Luzgarev
using System;
using System.Linq;
namespace MatFileHandler
{
/// <summary>
/// A better interface for using table objects.
/// </summary>
public class TableAdapter
{
private readonly IMatObject matObject;
/// <summary>
/// Initializes a new instance of the <see cref="TableAdapter"/> class.
/// </summary>
/// <param name="array">Source table object.</param>
public TableAdapter(IArray array)
{
matObject = array as IMatObject
?? throw new HandlerException("Table adapter must be initialized with a MATLAB object.");
if (matObject.ClassName != "table")
{
throw new ArgumentException("The object provided is not a table.");
}
var cellArray = matObject["varnames"] as ICellArray
?? throw new HandlerException("Table variable names must be in a cell array.");
VariableNames = Enumerable
.Range(0, cellArray.Count)
.Select(i =>
(cellArray[i] as ICharArray ??
throw new HandlerException("Variable name must be a char array.")).String)
.ToArray();
NumberOfVariables = VariableNames.Length;
var props = matObject["props"] as IStructureArray
?? throw new HandlerException("Table properties must be a structure array.");
Description = (props["Description"] as ICharArray
?? throw new HandlerException("Table description must be a char array.")).String;
NumberOfRows = (int)(matObject["nrows"].ConvertToDoubleArray()
?? throw new HandlerException("Cannot find number of rows in a table."))[0];
var rowNamesArrays = matObject["rownames"] as ICellArray
?? throw new HandlerException("Table row names must be a cell array.");
RowNames = Enumerable
.Range(0, rowNamesArrays.Count)
.Select(i => (cellArray[i] as ICharArray
?? throw new HandlerException("Each table row name must be a char array.")).String)
.ToArray();
}
/// <summary>
/// Gets table description.
/// </summary>
public string Description { get; }
/// <summary>
/// Gets the number of rows in the table.
/// </summary>
public int NumberOfRows { get; }
/// <summary>
/// Gets the number of variables in the table.
/// </summary>
public int NumberOfVariables { get; }
/// <summary>
/// Gets row names.
/// </summary>
public string[] RowNames { get; }
/// <summary>
/// Gets variable names.
/// </summary>
public string[] VariableNames { get; }
/// <summary>
/// Gets all the data for a given variable.
/// </summary>
/// <param name="variableName">Variable name.</param>
/// <returns>All data associated with the variable.</returns>
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
?? throw new HandlerException("Table data must be stored in a cell array.");
return data[index];
}
}
}
}