1
我正在使用Ruby C API。我需要从一个函数内创建一个结构,但我想我正在做一些分配错误。这是我的代码C - 使用结构指针作为参数在函数中分配结构
#include <stdio.h>
#include <ruby.h>
#include <ruby/thread.h>
typedef struct {
double *matrix;
int nrows;
int ncols;
}Matrix;
void createMatrix(VALUE matrix, Matrix *mat) {
printf("In\n");
mat->nrows = RARRAY_LEN(matrix);
VALUE firstElement = rb_ary_entry(matrix, 0);
mat->ncols = RARRAY_LEN(firstElement);
printf("Matrix shape: (%d,%d)\n", mat->nrows, mat->ncols);
int i,j;
double *tempMat = (double *)malloc(mat->nrows * mat->ncols * sizeof(double));
printf("Allocated\n");
VALUE row;
for (i=0; i<mat->nrows; i++)
{
row = rb_ary_entry(matrix, i);
for (j=0; j<mat->ncols; j++)
{
tempMat[i * mat->ncols + j] = NUM2DBL(rb_ary_entry(row, j));
// printf("Matrix A Element(%d,%d)=%f\n", i, j, matA[i * colsA + j]);
}
}
mat->matrix = tempMat;
for (i=0; i<mat->nrows; i++)
{
for (j=0; j<mat->ncols; j++)
{
printf("Matrix temp Element(%d,%d)=%f\n", i, j, mat->matrix[i * mat->ncols + j]);
}
}
printf("Assigned\n");
return;
}
VALUE matmat_mul(VALUE self, VALUE matrixA, VALUE matrixB)
{
int i,j;
Matrix *matA;
createMatrix(matrixA, matA);
Matrix *matB;
createMatrix(matrixB, matB);
return Qnil;
}
void Init_la_ruby_ext()
{
VALUE rg = rb_define_module("RG");
VALUE linalg = rb_define_module_under(rg, "LinearAlgebra");
VALUE operation = rb_define_class_under(linalg, "Operation", rb_cObject);
rb_define_method(operation, "matmat_mul", matmat_mul, 2);
}
的extconf.rb
文件
require 'mkmf'
extension_name = 'la_ruby_ext'
create_makefile(extension_name)
,你可以运行
require './la_ruby_ext'
rows = 3
cols = 3
mat = Array.new(rows){Array.new(cols)}
mat[0] = [0.0, 1.0, 2.0]
mat[1] = [3.0, 4.0, 5.0]
mat[2] = [6.0, 7.0, 8.0]
operation = RG::LinearAlgebra::Operation.new
matC = operation.matmat_mul(mat, mat.transpose)
测试你可以从运行测试的代码给段错误的看第二次我呼叫功能createMatrix
。这是我的理解:
- 我首先创建一个指向
struct Matrix
- 我那么这个指针作为参数传递给
createMatrix
- 我分配一个指针到一个二维数组
- 创建二维数组
- 我将指针指定给矩阵 - > nmatrix
任何想法?这是正确的方法吗?