1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
|
//mexData.C
// can efficiently extract data from a file which contains ASCII data and text.
// written by Mike.Zhang, 5-5/28/2002, Copyright reserved.
// run under MatLab enviroment and compiled by MatLab embeded Lcc C compiler.
// Copyright (C) 2002-2022 ABINIT group (Mike Zhang)
// This file is distributed under the terms of the
// GNU General Public License, see ~abinit/COPYING
// or http://www.gnu.org/copyleft/gpl.txt .
// For the initials of contributors, see ~abinit/doc/developers/contributors.txt .
#include "mex.h"
#include <stdio.h>
#include <string.h>
mxArray *data;
double *Data;
int dataDims;
int M, N, L; // Support max dimensions is up to 3
char strSign[128] = "Eigenvalues ( eV ) for nkpt=",
strSign2[128] = "",
strInnerInterval[128] = "coord)",
strOuterInterval[128] = "";
int fstrstr(FILE *fp, const char* Str, const int ScanLen)
{
int ScanChar;
long curFP, bytes=0;
if (ScanLen == 0)
return 1; // null string always return TRUE.
curFP = ftell(fp);
do {
ScanChar = fgetc(fp);
if ((char)ScanChar == Str[0])
{
int i;
for (i=1; i<ScanLen; i++)
if ((ScanChar = fgetc(fp)) != Str[i])
break;
if (i == ScanLen) // Success!
return 1;
else
fseek(fp, -i, SEEK_CUR);
}
} while (ScanChar != EOF);
// Search again from the beginning of the file
fseek(fp, 0, SEEK_SET);
do {
ScanChar = fgetc(fp);
if ((char)ScanChar == Str[0])
{
int i;
for (i=1; i<ScanLen; i++)
if ((ScanChar = fgetc(fp)) != Str[i])
break;
if (i == ScanLen) // Success!
return 1;
else
fseek(fp, -i, SEEK_CUR);
}
} while (++bytes < curFP);
// If search failed, the fp will return old position.
return 0;
}
int readData(char *filename)
{
FILE *fp;
int row, col, z;
int ScanLength = strlen(strSign);
int ScanLength2 = strlen(strSign2);
int InnerIntervalLength = strlen(strInnerInterval);
int OuterIntervalLength = strlen(strOuterInterval);
if (!(fp = fopen(filename, "rb")))
return -1;
// Now scan the buffer string:
fseek(fp, 0L, SEEK_SET);
if (!fstrstr(fp, strSign, ScanLength))
return -3;
if (!fstrstr(fp, strSign2, ScanLength2))
return -7;
for (z=0; z<L; z++)
{
if (!fstrstr(fp, strOuterInterval, OuterIntervalLength))
return -9;
for (row=0; row<M; row++)
{
if (!fstrstr(fp, strInnerInterval, InnerIntervalLength)) // Bypass "kpt#...coord)" parts
return -5;
for (col=0; col<N; col++)
fscanf(fp, "%lf", &Data[row+M*col+z*M*N]);
}
}
fclose(fp);
return 1;
}
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{
char name[128];
int dims[3]={1, 1, 1}, i, status;
double *reDIMS;
if (nrhs <= 1)
{
mexPrintf("\nData Extractor\n"
"Usage: mexData([M N L], ReadFileNameString, SignatureString, InnerIntervalString, OuterIntervalString, SubSignatureString);\n"
" mexData extracts data from ASCII code file which contains useful data as well as text infomation.\n\n"
" As for two-dimension data, mexData searches Signature_String which indicates the useful data will appear,\n"
" then searches the Inner_Interval_String to locate the accurate position of the useful data.\n"
" After the N columns data are read, mexData again searches Interval_String to prepare reading another row data.\n"
" When all M-row by N-column data are extracted, mexData returns a double float M-by-N matrix and finishes its work.\n\n"
" As for three-dimension M-by-N-by-L data, Outer_Interval_String is used to differentiate every L dimension.\n"
" mexData returns M-by-N-by-L matrix.\n\n"
" Modified at 7/30/2002, add OuterIntervalString, SubSignatureString.\n"
" Modified at 8/11/2002, correct L dimension disposition(use Data[row+M*col+z*M*N] instead of Data[L*row+M*L*col+z]).\n"
" Used by your own warranty, Mike.Zhang(mz24cn@hotmail.com), June 2002.\n\n");
return;
}
reDIMS = mxGetPr(prhs[0]);
dataDims = mxGetN(prhs[0]);
for (i=0; i<dataDims; i++)
dims[i] = (int)reDIMS[i];
M = dims[0];
N = dims[1];
L = dims[2];
data = mxCreateNumericArray(dataDims, dims, mxDOUBLE_CLASS, mxREAL);
Data = mxGetPr(data);
if (nrhs >= 2)
{
mxGetString(prhs[1], name, 127);
if (nrhs >= 3)
mxGetString(prhs[2], strSign, 127);
if (nrhs >= 4)
mxGetString(prhs[3], strInnerInterval, 127);
if (nrhs >= 5)
mxGetString(prhs[4], strOuterInterval, 127);
if (nrhs >= 6)
mxGetString(prhs[5], strSign2, 127);
}
mexPrintf("Data Extractor\n"
"Read file : '%s'\n"
"Signature string : '%s'\n"
"InnerInterval string : '%s'\n"
"OuterInterval string : '%s'\n"
"2nd Signature string : '%s'\n"
"Output array dimensions: %d x %d x %d\n",
name, strSign, strInnerInterval, strOuterInterval, strSign2, M, N, L);
status = readData(name);
if (status == 1)
mexPrintf("Data extracted success.\n");
else if (status == -1)
mexPrintf("Error: Can't open file '%s'.\n", name);
else if (status == -3)
mexPrintf("Error: Can't find string '%s'.\n", strSign);
else if (status == -5)
mexPrintf("Error: Can't find string '%s'.\n", strInnerInterval);
else if (status == -7)
mexPrintf("Error: Can't find string '%s'.\n", strSign2);
else if (status == -9)
mexPrintf("Error: Can't find string '%s'.\n", strOuterInterval);
plhs[0] = data;
}
|
