C#:将通用指针转换为数组

时间:2009-06-12 08:44:34

标签: c# arrays unsafe

我想将byte*转换为byte[],但我还希望有一个可重复使用的功能来执行此操作:

public unsafe static T[] Create<T>(T* ptr, int length)
{
    T[] array = new T[length];

    for (int i = 0; i < length; i++)
        array[i] = ptr[i];

    return array;
}

不幸的是我收到编译器错误,因为T可能是“.NET托管类型”,我们无法指向那些。更令人沮丧的是,没有泛型类型约束可以将T限制为“非托管类型”。是否有内置的.NET函数来执行此操作?有什么想法吗?

5 个答案:

答案 0 :(得分:5)

可以匹配您尝试执行的操作的方法是Marshal.Copy,但它没有采用适当的参数来制作通用方法。

虽然不可能用通用约束来编写可以描述可能性的泛型方法,但并不是每种类型都可以使用“不安全”方式进行复制。有一些例外;课程就是其中之一。

以下是示例代码:

    public unsafe static T[] Create<T>(void* source, int length)
    {
        var type = typeof(T);
        var sizeInBytes =  Marshal.SizeOf(typeof(T));

        T[] output = new T[length];

        if (type.IsPrimitive)
        {
            // Make sure the array won't be moved around by the GC 
            var handle = GCHandle.Alloc(output, GCHandleType.Pinned);

            var destination = (byte*)handle.AddrOfPinnedObject().ToPointer();
            var byteLength = length * sizeInBytes;

            // There are faster ways to do this, particularly by using wider types or by 
            // handling special lengths.
            for (int i = 0; i < byteLength; i++)
                destination[i] = ((byte*)source)[i];

            handle.Free();
        }
        else if (type.IsValueType)
        {
            if (!type.IsLayoutSequential && !type.IsExplicitLayout)
            {
                throw new InvalidOperationException(string.Format("{0} does not define a StructLayout attribute", type));
            }

            IntPtr sourcePtr = new IntPtr(source);

            for (int i = 0; i < length; i++)
            {
                IntPtr p = new IntPtr((byte*)source + i * sizeInBytes);

                output[i] = (T)System.Runtime.InteropServices.Marshal.PtrToStructure(p, typeof(T));
            }
        }
        else 
        {
            throw new InvalidOperationException(string.Format("{0} is not supported", type));
        }

        return output;
    }

    unsafe static void Main(string[] args)
    {
        var arrayDouble = Enumerable.Range(1, 1024)
                                    .Select(i => (double)i)
                                    .ToArray();

        fixed (double* p = arrayDouble)
        {
            var array2 = Create<double>(p, arrayDouble.Length);

            Assert.AreEqual(arrayDouble, array2);
        }

        var arrayPoint = Enumerable.Range(1, 1024)
                                   .Select(i => new Point(i, i * 2 + 1))
                                   .ToArray();

        fixed (Point* p = arrayPoint)
        {
            var array2 = Create<Point>(p, arrayPoint.Length);

            Assert.AreEqual(arrayPoint, array2);
        }
    }

该方法可以是通用的,但它不能采用泛型类型的指针。这不是一个问题,因为指针协方差有帮助,但这会产生阻止泛型参数类型的隐式解析的不幸影响。然后,您必须明确指定MakeArray。

我为结构添加了一个特例,最好有类型指定struct layout。在您的情况下,这可能不是问题,但如果指针数据来自本机C或C ++代码,则指定布局类型很重要(CLR可能会选择重新排序字段以获得更好的内存对齐)。

但是如果指针完全来自托管代码生成的数据,那么您可以删除检查。

此外,如果性能存在问题,则复制数据的算法比逐字节更好。 (请参阅memcpy的无数实现以供参考)

答案 1 :(得分:1)

似乎问题变成:如何将泛型类型指定为简单类型。

unsafe void Foo<T>() : where T : struct
{
   T* p;
}

给出错误:
不能取地址,获取大小,或声明指向托管类型的指针('T')

答案 2 :(得分:1)

这个怎么样?

static unsafe T[] MakeArray<T>(void* t, int length, int tSizeInBytes) where T:struct
{
    T[] result = new T[length];
    for (int i = 0; i < length; i++)
    {
        IntPtr p = new IntPtr((byte*)t + (i * tSizeInBytes));
        result[i] = (T)System.Runtime.InteropServices.Marshal.PtrToStructure(p, typeof(T));
    }

    return result;
}

我们不能在这里使用sizeof(T),但调用者可以执行类似

的操作
byte[] b = MakeArray<byte>(pBytes, lenBytes, sizeof(byte));

答案 3 :(得分:0)

我不知道如果以下内容可行,但它可能(至少它编译:):

public unsafe static T[] Create<T>(void* ptr, int length) where T : struct
{
    T[] array = new T[length];

    for (int i = 0; i < length; i++)
    {
        array[i] = (T)Marshal.PtrToStructure(new IntPtr(ptr), typeof(T));
    }

    return array;
}

关键是使用Marshal.PtrToStructure转换为正确的类型。

答案 4 :(得分:0)

从C#7.3开始,这可能如下:

public unsafe static T[] Create<T>(T* ptr, int length) where T: unmanaged
{
    T[] array = new T[length];
    for (int i = 0; i < length; i++)
        array[i] = ptr[i];
    return array;
}