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
任何想法?這是正確的方法嗎?