下面是基于 C 语言写的一段 vector 与 xarray 互相转换的代码,能够兼容一维、二维、三维情况:
``` #include <stdio.h> #include <stdlib.h>
#define MAX_DIMS 3
typedef struct { void *data; int dims[MAX_DIMS]; int ndims; } xarray;
typedef struct { void *data; int size; } vector;
/* 将一维数组转换为 vector 结构 */ vector array_to_vector(void *data, int size) { vector v; v.data = malloc(size * sizeof(char)); memcpy(v.data, data, size * sizeof(char)); v.size = size; return v; }
/* 将 vector 结构转换为一维数组 */ void *vector_to_array(vector v) { void *data = malloc(v.size * sizeof(char)); memcpy(data, v.data, v.size * sizeof(char)); return data; }
/* 将 xarray 结构转换为 vector 结构 */ vector xarray_to_vector(xarray a) { vector v; int size = 1; for (int i = 0; i < a.ndims; i++) { size *= a.dims[i]; } v = array_to_vector(a.data, size); return v; }
/* 将 vector 结构转换为 xarray 结构 */ xarray vector_to_xarray(vector v, int *dims, int ndims) { xarray a; a.data = vector_to_array(v); memcpy(a.dims, dims, ndims * sizeof(int)); a.ndims = ndims; return a; }
/* 将一维数组转换为 xarray 结构 */ xarray array_to_xarray(void *data, int size) { int dims[1] = { size }; return vector_to_xarray(array_to_vector(data, size), dims, 1); }
/* 将二维数组转换为 xarray 结构 */ xarray array2d_to_xarray(void *data, int nrows, int ncols) { int dims[2] = { nrows, ncols }; return vector_to_xarray(array_to_vector(data, nrows * ncols), dims, 2); }
/* 将三维数组转换为 xarray 结构 */ xarray array3d_to_xarray(void *data, int dim1, int dim2, int dim3) { int dims[3] = { dim1, dim2, dim3 }; return vector_to_xarray(array_to_vector(data, dim1 * dim2 * dim3), dims, 3); }
int main() { /* 示例代码 */ int a[3] = { 1, 2, 3 }; xarray arr1 = array_to_xarray(a, 3); vector vec1 = xarray_to_vector(arr1); int *b = vector_to_array(vec1); for (int i = 0; i < vec1.size / sizeof(int); i++) { printf("%d ", b[i]); } printf("\n");
int c[2][3] = { { 1, 2, 3 }, { 4, 5, 6 } };
xarray arr2 = array2d_to_xarray(c, 2, 3);
vector vec2 = xarray_to_vector(arr2);
int (*d)[3] = vector_to_array(vec2);
for (int i = 0