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
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
|
from libc.stdio cimport *
from libc.string cimport memcpy, strcmp, strstr, strcpy
IF UNAME_SYSNAME == 'Windows':
cdef extern from 'io.h':
int dup(int fd)
ELSE:
cdef extern from 'unistd.h':
int dup(int fd)
IF UNAME_SYSNAME == 'Linux':
cdef extern from 'locale.h':
ctypedef struct locale_t:
pass
locale_t uselocale(locale_t __dataset)
locale_t newlocale(int __category_mask, const char *__locale,
locale_t __base)
void freelocale(locale_t __dataset)
enum: LC_NUMERIC_MASK
import numpy as np
cimport numpy as np
np.import_array()
cdef packed struct Facet:
np.float32_t n[3]
np.float32_t v[3][3]
np.uint16_t attr
dtype = np.dtype([
('normals', np.float32, 3),
('vectors', np.float32, (3, 3)),
('attr', np.uint16, 1),
])
DEF ALLOC_SIZE = 200000
DEF BUF_SIZE = 8192
DEF LINE_SIZE = 8192
cdef struct s_State:
FILE* fp
char buf[BUF_SIZE]
char line[LINE_SIZE]
size_t pos
size_t size
size_t line_num
int recoverable
ctypedef s_State State
cdef char* readline(State* state) except NULL:
cdef size_t line_pos = 0
cdef char current;
while True:
if state.pos == state.size:
if feof(state.fp):
if line_pos != 0:
state.line[line_pos] = '\0'
return state.line
raise RuntimeError(state.recoverable, 'Unexpected EOF')
state.size = fread(state.buf, 1, BUF_SIZE, state.fp)
state.pos = 0
state.recoverable = 0
if line_pos == LINE_SIZE:
raise RuntimeError(
state.recoverable, 'Line longer than %d, probably non-ascii' %
LINE_SIZE)
current = state.buf[state.pos]
state.pos += 1
if line_pos != 0 or (current != ' ' \
and current != '\t' \
and current != '\r'):
if current == '\n':
state.line_num += 1
if line_pos != 0:
state.line[line_pos] = '\0'
return state.line
elif 0x40 < current < 0x5b:
# Change all ascii characters to lower case
state.line[line_pos] = current | 0x60
line_pos += 1
else:
state.line[line_pos] = current
line_pos += 1
def ascii_read(fh, buf):
cdef char* line
cdef char name[LINE_SIZE]
cdef np.ndarray[Facet, cast=True] arr = np.zeros(ALLOC_SIZE, dtype = dtype)
cdef size_t offset;
cdef Facet* facet = <Facet*>arr.data
cdef size_t pos = 0
cdef State state
IF UNAME_SYSNAME == 'Linux':
cdef locale_t new_locale = newlocale(LC_NUMERIC_MASK, 'C',
<locale_t>NULL)
cdef locale_t old_locale = uselocale(new_locale)
freelocale(new_locale)
try:
state.size = len(buf)
memcpy(state.buf, <char*> buf, state.size)
state.pos = 0
state.line_num = 0
state.recoverable = 1
state.fp = fdopen(dup(fh.fileno()), 'rb')
fseek(state.fp, fh.tell(), SEEK_SET)
line = readline(&state)
if strstr(line, 'solid') != line:
raise RuntimeError(state.recoverable,
'Solid name not found (%i:%s)' % (state.line_num, line))
strcpy(name, line+5)
while True:
line = readline(&state)
line = state.line
if strstr(line, 'endsolid') != NULL:
arr.resize(facet - <Facet*>arr.data, refcheck=False)
return (<object>name).strip(), arr
if strcmp(line, 'color') == 0:
readline(&state)
continue
elif sscanf(line, '%*s %*s %e %e %e',
facet.n, facet.n+1, facet.n+2) != 3:
raise RuntimeError(state.recoverable,
'Cannot read normals (%i:%s)' % (state.line_num, line))
readline(&state) # outer loop
for i in range(3):
line = readline(&state)
if sscanf(line, '%*s %e %e %e',
facet.v[i], facet.v[i]+1, facet.v[i]+2) != 3:
raise RuntimeError(state.recoverable,
'Cannot read vertex (%i:%s)' % (state.line_num, line))
readline(&state) # endloop
readline(&state) # endfacet
facet += 1
offset = facet - <Facet*>arr.data
if arr.shape[0] == offset:
arr.resize(arr.shape[0] + ALLOC_SIZE, refcheck=False)
facet = <Facet*>arr.data + offset
finally:
if state.recoverable == 0:
pos = ftell(state.fp) - state.size + state.pos
fclose(state.fp)
fh.seek(pos, SEEK_SET)
IF UNAME_SYSNAME == 'Linux':
uselocale(old_locale)
freelocale(old_locale)
def ascii_write(fh, name, np.ndarray[Facet, mode = 'c', cast=True] arr):
cdef FILE* fp
cdef Facet* facet = <Facet*>arr.data
cdef Facet* end = <Facet*>arr.data + arr.shape[0]
cdef size_t pos = 0
try:
fp = fdopen(dup(fh.fileno()), 'wb')
fseek(fp, fh.tell(), SEEK_SET)
fprintf(fp, 'solid %s\n', <char*>name)
while facet != end:
fprintf(fp,
'facet normal %f %f %f\n'
' outer loop\n'
' vertex %f %f %f\n'
' vertex %f %f %f\n'
' vertex %f %f %f\n'
' endloop\n'
'endfacet\n',
facet.n[0], facet.n[1], facet.n[2],
facet.v[0][0], facet.v[0][1], facet.v[0][2],
facet.v[1][0], facet.v[1][1], facet.v[1][2],
facet.v[2][0], facet.v[2][1], facet.v[2][2])
facet += 1
fprintf(fp, 'endsolid %s\n', <char*>name)
finally:
pos = ftell(fp)
fclose(fp)
fh.seek(pos, SEEK_SET)
|
