|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
using System.IO;
using System.Text;
using Microsoft.ML.Data;
using Microsoft.ML.Runtime;
namespace Microsoft.ML.TestFramework
{
/// <summary>
/// On demand open up new streams over a source of bytes.
/// </summary>
public sealed class BytesStreamSource : IMultiStreamSource
{
private readonly byte[] _data;
public int Count => 1;
public BytesStreamSource(byte[] data)
{
Contracts.AssertValue(data);
_data = data;
}
public BytesStreamSource(string data)
: this(Encoding.UTF8.GetBytes(data))
{
}
public string GetPathOrNull(int index)
{
Contracts.Check(index == 0);
return null;
}
public Stream Open(int index)
{
Contracts.Check(index == 0);
return new MemoryStream(_data, writable: false);
}
public TextReader OpenTextReader(int index)
{
return new StreamReader(Open(index));
}
}
}
|