C - Выделить структуру внутри функции с помощью указателя struct в качестве аргумента

Question

Я работаю с API Ruby 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. Это мое понимание:

1. Я сначала создать указатель на struct Matrix
2. Я затем передать этот указатель в качестве аргумента createMatrix
3. выделяю указатель на 2D массива
4. Создание 2D массива
5. Я назначаю указатель на матрицу-> nmatrix

Любая идея? Правильно ли это?

Answer 1

Учитывая определение void createMatrix(VALUE xxx, Matrix *mat) и использование mat->nrows = … и т.д. внутри функции, вызовы к createMatrix() в matmul_mul() неверны.

У вас есть:

VALUE matmat_mul(VALUE self, VALUE matrixA, VALUE matrixB) 
{ 
    int i,j; 

    Matrix *matA; 
    createMatrix(matrixA, matA); 

    Matrix *matB; 
    createMatrix(matrixB, matB); 

Понадобится:

VALUE matmat_mul(VALUE self, VALUE matrixA, VALUE matrixB) 
{ 
    int i,j; 

    Matrix matA; 
    createMatrix(matrixA, &matA); 

    Matrix matB; 
    createMatrix(matrixB, &matB); 

В текущем коде, вы передаете неинициализированные указатели на createMatrix(), а затем назначить «случайного» памяти, указатели указывают на. Это редко приводит к счастью. В обновленном коде у вас есть пара значений Matrix, и вы передаете указатели на них функции createMatrix(), которая затем заполняет детали для вас.