| 1 | // Function to read rawdata into Matlab
|
|---|
| 2 | //
|
|---|
| 3 | // Use 'mex matlabread.cc RawDataCTX.o' in Matlab to compile.
|
|---|
| 4 | // Use 'make' on console to generate RawDataCTX.o.
|
|---|
| 5 |
|
|---|
| 6 | #include <unistd.h>
|
|---|
| 7 |
|
|---|
| 8 | #include "/ihp/local/Matlab08/extern/include/mex.h"
|
|---|
| 9 | #include "../../drsdaq/RawDataCTX.h"
|
|---|
| 10 |
|
|---|
| 11 | // Print text date in temporary file, than clear file
|
|---|
| 12 | void PrintTempFile(FILE* File) {
|
|---|
| 13 | char Buffer[1000];
|
|---|
| 14 | rewind(File);
|
|---|
| 15 | while(fgets(Buffer, sizeof(Buffer), File) != NULL) printf("%s", Buffer);
|
|---|
| 16 | ftruncate(fileno(File),0);
|
|---|
| 17 | }
|
|---|
| 18 |
|
|---|
| 19 | // Interface function to Matlab
|
|---|
| 20 | void mexFunction (int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[]) {
|
|---|
| 21 |
|
|---|
| 22 | RawDataCTX* RD = new RawDataCTX(true); // Instantiate without console output
|
|---|
| 23 | FILE *Tmpfile = tmpfile(); // Temporary file for printing of headers
|
|---|
| 24 |
|
|---|
| 25 | // Check inputs and output arguments
|
|---|
| 26 | if (nrhs<1 || nrhs>2 || (nrhs==2 && !mxIsDouble(prhs[1]))) {
|
|---|
| 27 | mexErrMsgTxt("Parameters must be filename and (optional) event number.");
|
|---|
| 28 | }
|
|---|
| 29 | if (nlhs != 2) mexErrMsgTxt ("Need two return values.");
|
|---|
| 30 |
|
|---|
| 31 | switch (RD->OpenDataFile(mxArrayToString(prhs[0]), Tmpfile)) {
|
|---|
| 32 | case CTX_FOPEN: mexErrMsgTxt("Could not open file.");
|
|---|
| 33 | case CTX_RHEADER: mexErrMsgTxt("Could not read run header.");
|
|---|
| 34 | case CTX_BSTRUCT: mexErrMsgTxt("Could not read board structures.");
|
|---|
| 35 | }
|
|---|
| 36 | if (RD->RHeader->MagicNum == MAGICNUM_OPEN) {
|
|---|
| 37 | mexWarnMsgTxt("Magic number in run header indicates that the file has not "
|
|---|
| 38 | "been closed properly.");
|
|---|
| 39 | }
|
|---|
| 40 | if (RD->RHeader->MagicNum == MAGICNUM_ERROR) {
|
|---|
| 41 | mexWarnMsgTxt("Magic number in run header indicates that an error occurred "
|
|---|
| 42 | "while writing the file.");
|
|---|
| 43 | }
|
|---|
| 44 |
|
|---|
| 45 | // ...some abbrevation for convenience
|
|---|
| 46 | unsigned int Boards = RD->RHeader->NBoards;
|
|---|
| 47 | unsigned int Channels = RD->RHeader->NChannels;
|
|---|
| 48 | unsigned int Chips = RD->RHeader->NChips;
|
|---|
| 49 | unsigned int Samples = RD->RHeader->Samples;
|
|---|
| 50 |
|
|---|
| 51 | // Print header data
|
|---|
| 52 | if(RD->ReadEvent(mxIsDouble(prhs[1]) ? (int) mxGetScalar(prhs[1]):0, Tmpfile) != CTX_OK) {
|
|---|
| 53 | mexErrMsgTxt("Could not read event.");
|
|---|
| 54 | }
|
|---|
| 55 | else PrintTempFile(Tmpfile);
|
|---|
| 56 |
|
|---|
| 57 | // Allocate memory
|
|---|
| 58 | plhs[0] = mxCreateDoubleMatrix(Boards*Chips*Channels*Samples, 1, mxREAL);
|
|---|
| 59 | plhs[1] = mxCreateDoubleMatrix(Chips, Boards, mxREAL);
|
|---|
| 60 |
|
|---|
| 61 | // Fill trigger cell array
|
|---|
| 62 | for(unsigned int i=0; i<Boards*Chips; i++) {
|
|---|
| 63 | *(mxGetPr(plhs[1])+i) = *((int *) (RD->Data+i*sizeof(int)));
|
|---|
| 64 | }
|
|---|
| 65 |
|
|---|
| 66 | // Fill data array
|
|---|
| 67 | unsigned int Count = 0;
|
|---|
| 68 | for(unsigned int i=0; i<Boards; i++) {
|
|---|
| 69 | for(unsigned int j=0; j<Chips*Channels*Samples; j++) {
|
|---|
| 70 | *(mxGetPr(plhs[0])+Count) = (double) *((short *) (RD->Data+Boards*Chips*sizeof(int)) + Count++) * RD->BStruct[i].ScaleFactor;
|
|---|
| 71 | }
|
|---|
| 72 | }
|
|---|
| 73 |
|
|---|
| 74 | // Resize data array into boards, channels and samples
|
|---|
| 75 | mwSize Dimensions[] = {Samples, Chips*Channels, Boards};
|
|---|
| 76 | mxSetDimensions(plhs[0], Dimensions, sizeof(Dimensions)/sizeof(mwSize));
|
|---|
| 77 |
|
|---|
| 78 | // Clean up
|
|---|
| 79 | fclose(Tmpfile);
|
|---|
| 80 | RD->CloseDataFile();
|
|---|
| 81 | }
|
|---|