array als rückgabe einer funktion (mex)

Diese Seite verwendet Cookies. Durch die Nutzung unserer Seite erklären Sie sich damit einverstanden, dass wir Cookies setzen. Weitere Informationen

  • array als rückgabe einer funktion (mex)

    hallo erstmal :)

    das problem gabs zwar schonmal (easy-coding.de/array-und-funktionen-t1877.html), aber leider versteh ich die antwort nicht so recht :roll: da ich im bezug auf arrays absolut gar keine ahnung habe :) im speziellen, wie diese in c++ initiiert und definiert werden...

    daher hier mein spezifisches problem:
    ich muss eine matrix mit einem vektor multiplizieren. das ganze soll so ablaufen, dass die hauptprozedur eine funktion aufruft, der die matrix und der vektor übergeben werden. die funktion berechnet dann das matrix-vektor-produkt.

    hier mal der code:

    Quellcode

    1. #include "mex.h"
    3. double mult(int nor, int noc, double matA[][], double vecx[])
    4. {
    5. int i = 0;
    6. int j = 0;
    7. double res[nor];
    8. for (i = 0; i < nor; i++)
    9. {
    10. res[i] = 0.0;
    11. }
    12. for (i = 0; i < nor; i++)
    13. {
    14. for (j = 0; j < noc; j++)
    15. {
    16. res[i] = res[i] + (matA[i][j] * vecx[j]);
    17. }
    18. }
    19. return res;
    20. }
    22. void mexFunction(int nlhs, mxArray *plhs[ ],int nrhs, const mxArray *prhs[ ])
    23. {
    24. /* check number of input arguments */
    25. if (nrhs != 2)
    26. {
    27. mexErrMsgTxt("Must have two input arguments.");
    28. }
    29. /* check number of output arguments */
    30. if (nlhs != 1)
    31. {
    32. mexErrMsgTxt("Must have one output argument.");
    33. }
    36. /* check if second input argument is a vector. If yes check if it is a column vector, else GGF. PROBLEM! */
    37. int vecCheckN = mxGetN(prhs[1]); /* number of columns of the input vector */
    38. int vecCheckM = mxGetM(prhs[1]); /* number of rows of the input vector */
    39. if (vecCheckN != 1)
    40. {
    41. if (vecCheckM != 1)
    42. {
    43. mexErrMsgTxt("Second input argument must be a vector.");
    44. }
    45. mexErrMsgTxt("Second input argument must be a column vector.");
    46. // ZEILENVEKTOR EINGEGEBEN! GGF. PROBLEME BEACHTEN!
    47. // vecCheckM = vecCheckN;
    48. }
    51. /* check if matrix and vector dimensions agree */
    52. int n = mxGetN(prhs[0]); /* number of columns of the input matrix */
    53. if (n != vecCheckN)
    54. {
    55. mexErrMsgTxt("Matrix and vector dimensions disagree.");
    56. }
    59. /* declare variables */
    60. int checkSparse = 0;
    61. int m = mxGetM(prhs[0]); /* number of rows of the input matrix */
    62. double matrixA[m][n];
    63. double vectorx[n];
    64. double result[m]
    66. // SCHLEIFE ZUM BEFÜLLEN VON matrixA UND vectorx EINFÜGEN
    68. /* check if matrix is a sparse matrix */
    69. if ( mxIsClass(prhs[0],"sparse"))
    70. {
    71. checkSparse = 1;
    72. }
    74. /* choose way to compute result and compute*/
    75. switch (checkSparse)
    76. {
    77. case 0:
    78. result = mult(m,n,matrixA,vectorx);
    79. break;
    81. case 1: // IS SPARSE: BERECHNUNG PER mexCallMATLAB
    82. break;
    84. default:
    85. break;
    86. }
    87. }
    88. }
    Alles anzeigen


    ok, wie ihr seht, geht es um mex-programmierung :)
    mein problem ist die definition der funktion mult, sowie ihrer übergabewerte, usw. das programm müsste möglichst "schnell" sein, sprich es geht nicht um schöne formulierung, sondern um möglichst gute performance.

    kurze zusammenfassung vom code:
    in zeile 78 wird die funktion mult aufgerufen, die ein produkt von einem 2-d array und einem 1-d array berechnet (was wiederrum ein 1-d array ist)
    und dieser ergebnissarray soll dann halt in result gespeichert werden.

    hoffe ihr könnt mir helfen :) bin noch c/c++ neuling und fang grad erst an :) also bitte nich haun ;)