18542693371

Committed 15 Oct 2025 09:07PM UTC coverage: 88.199% (-0.6%) from 88.811%

Build # 18542693371

Build Type

push

github

Committed by

rjfarmer

Commit Message

Fix string test

Run Details

1009 of 1144 relevant lines covered (88.2%)

0.88 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

87.1

/gfModParser/modules/expressions.py

# SPDX-License-Identifier: GPL-2.0+

import numpy as np

from .. import utils

from . import procedures


class ExpGeneric:
    def __init__(self, type, kind, args, *, version):
        self._args = args
        self.version = version
        self._type = type
        self.kind = kind

    def __str__(self):
        return self._type

    def __repr__(self):
        return self._type

    @property
    def value(self):
        return None

    def __eq__(self, key):
        return self._type == key


class Expression:
    def __init__(self, expression, *, version):
        self._expression = expression
        self.version = version
        t = self._expression[0]

        map = {
            "OP": ExpOp,
            "FUNCTION": ExpFunction,
            "CONSTANT": ExpConstant,
            "VARIABLE": ExpVariable,
            "SUBSTRING": ExpSubString,
            "STRUCTURE": ExpStructure,
            "ARRAY": ExpArray,
            "NULL": ExpNull,
            "COMPCALL": ExpCompCall,
            "PPC": ExpPPC,
            "CONDITIONAL": ExpConditional,
            "UNKNOWN": ExpUnknown,
        }

        self._exp = map[t](
            self.typespec.type,
            self.typespec.kind,
            self._expression,
            version=self.version,
        )

    @property
    def type(self) -> ExpGeneric:
        return self._exp

    @property
    def typespec(self) -> "typespec":
        return typespec(self._expression[1], version=self.version)

    @property
    def rank(self) -> int:
        return int(self._expression[2])

    @property
    def arglist(self) -> list:
        raise NotImplementedError
        # if len(self._exp._args) == 7:
        #     return procedures.actual_arglist(self._exp._args[6])
        # return []

    @property
    def value(self):
        return self._exp.value

    def __str__(self):
        return self._exp.__str__()

    def __repr__(self):
        return self._exp.__repr__()

    @property
    def len(self) -> int:
        return self._exp.len

    @property
    def kind(self) -> int:
        return self.typespec.kind


class ExpOp(ExpGeneric):

    @property
    def unary_op(self):
        return self._args[3]

    @property
    def unary_args(self) -> tuple[Expression, Expression]:
        return Expression(self._args[4], version=self.version), Expression(
            self._args[5], version=self.version
        )


class ExpNotImplemented(ExpGeneric):
    @property
    def value(self):
        raise NotImplementedError


class ExpFunction(ExpGeneric):
    @property
    def value(self):
        return self._args[3]

    @property
    def args(self):
        return Expression(self._args[4], version=self.version)


class ExpConstant(ExpGeneric):
    @property
    def value(self):
        if self._type == "REAL":
            return utils.hextofloat(utils.string_clean(self._args[3]), self.kind)
        elif self._type == "INTEGER" or self._type == "UNSIGNED":
            return int(utils.string_clean(self._args[3]))
        elif self._type == "CHARACTER":
            return utils.string_clean(self._args[4])
        elif self._type == "COMPLEX":
            return complex(
                utils.hextofloat(utils.string_clean(self._args[3]), self.kind),
                utils.hextofloat(utils.string_clean(self._args[4]), self.kind),
            )
        elif self._type == "LOGICAL":
            return int(self._args[3]) == 1
        else:
            raise NotImplementedError(f"Type={self._type} args3={self._args[3]}")

    @property
    def len(self) -> int:
        if self._type == "CHARACTER":
            return int(self._args[3])
        return -1

    def __str__(self):
        if self._type == "CHARACTER":
            return f"CHARACTER(kind={self.kind},len={self.len})"
        else:
            return f"{self.type}(kind={self.kind})"

    @property
    def type(self) -> str:
        return self._type

    @property
    def raw(self) -> str:
        return utils.string_clean(self._args[3])


class ExpVariable(ExpGeneric):
    @property
    def value(self):
        return self._args[3]

    def __str__(self):
        if self._type == "CHARACTER":
            return f"CHARACTER(kind={self.kind},len={self.len})"
        else:
            return f"{self._type}"


class ExpArray(ExpGeneric):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self._value = None

    @property
    def value(self) -> np.ndarray:
        if self._value is None:
            self._value = []
            for i in self._args[3]:
                self._value.append(Expression(i[0], version=self.version))

        value = []

        for v in self._value:
            value.append(v.value)

        return np.array(value, dtype=self.dtype).reshape(self.shape)

    @property
    def shape(self) -> tuple[int, ...]:
        return tuple([int(utils.string_clean(i)) for i in self._args[4]])

    @property
    def dtype(self) -> np.dtype:
        v = Expression(self._args[3][0][0], version=self.version)
        return utils.dtype(v.type, self.kind, len=v.len)

    def __str__(self):
        v = Expression(self._args[3][0][0], version=self.version)
        if v.type == "CHARACTER":
            return f"CHARACTER(kind={v.kind},len={v.len}),dimension{self.shape}"
        else:
            return f"{v.type},dimension{self.shape}"


class ExpSubString(ExpNotImplemented):
    pass


class ExpStructure(ExpNotImplemented):
    pass


class ExpNull(ExpNotImplemented):
    pass


class ExpCompCall(ExpNotImplemented):
    pass


class ExpPPC(ExpNotImplemented):
    pass


class ExpConditional(ExpNotImplemented):
    pass


class ExpUnknown(ExpNotImplemented):
    pass


# Need to store this here as we get a cyclic dependency
# between expressions and typespec
class typespec:
    def __init__(self, typespec, *, version):
        self._typespec = typespec
        self.version = version

    @property
    def type(self) -> str:
        return self._typespec[0]

    def _isclass(self) -> bool:
        return self.type == "CLASS" or self.type == "DERIVED"

    @property
    def kind(self) -> int:
        if not self._isclass():
            return int(self._typespec[1])
        return -1

    @property
    def class_ref(self) -> int:
        if self._isclass():
            return int(self._typespec[1])
        return -1

    @property
    def interface(self):
        return self._typespec[2]

    @property
    def is_c_interop(self) -> bool:
        return int(self._typespec[3]) == 1

    @property
    def is_iso_c(self) -> bool:
        return int(self._typespec[4]) == 1

    @property
    def type2(self):
        # Whats this?
        return self._typespec[5]

    @property
    def charlen(self) -> Expression:
        if len(self._typespec[6]):
            return Expression(self._typespec[6][0], version=self.version)
        raise AttributeError("Object does not have a character length")

    #     try:
    #         if not args[6][0]:
    #             self.charlen = -1
    #         else:
    #             self.charlen = Expression(
    #                 *args[6][0]
    #             )  # TODO: might this need to be iterated for mulit-d strings?
    #     except IndexError:
    #         self.charlen = -1

    @property
    def deferred_cl(self) -> bool:
        if len(self._typespec) == 8:
            return self._typespec[7] == "DEFERRED_CL"

        return False

1	# SPDX-License-Identifier: GPL-2.0+
2
3	import numpy as np	1✔
4
5	from .. import utils	1✔
6
7	from . import procedures	1✔
8
9
10	class ExpGeneric:	1✔
11	def __init__(self, type, kind, args, *, version):	1✔
12	self._args = args	1✔
13	self.version = version	1✔
14	self._type = type	1✔
15	self.kind = kind	1✔
16
17	def __str__(self):	1✔
18	return self._type	×
19
20	def __repr__(self):	1✔
21	return self._type	1✔
22
23	@property	1✔
24	def value(self):	1✔
25	return None	×
26
27	def __eq__(self, key):	1✔
28	return self._type == key	1✔
29
30
31	class Expression:	1✔
32	def __init__(self, expression, *, version):	1✔
33	self._expression = expression	1✔
34	self.version = version	1✔
35	t = self._expression[0]	1✔
36
37	map = {	1✔
38	"OP": ExpOp,
39	"FUNCTION": ExpFunction,
40	"CONSTANT": ExpConstant,
41	"VARIABLE": ExpVariable,
42	"SUBSTRING": ExpSubString,
43	"STRUCTURE": ExpStructure,
44	"ARRAY": ExpArray,
45	"NULL": ExpNull,
46	"COMPCALL": ExpCompCall,
47	"PPC": ExpPPC,
48	"CONDITIONAL": ExpConditional,
49	"UNKNOWN": ExpUnknown,
50	}
51
52	self._exp = map[t](	1✔
53	self.typespec.type,
54	self.typespec.kind,
55	self._expression,
56	version=self.version,
57	)
58
59	@property	1✔
60	def type(self) -> ExpGeneric:	1✔
61	return self._exp	1✔
62
63	@property	1✔
64	def typespec(self) -> "typespec":	1✔
65	return typespec(self._expression[1], version=self.version)	1✔
66
67	@property	1✔
68	def rank(self) -> int:	1✔
69	return int(self._expression[2])	1✔
70
71	@property	1✔
72	def arglist(self) -> list:	1✔
73	raise NotImplementedError	×
74	# if len(self._exp._args) == 7:
75	# return procedures.actual_arglist(self._exp._args[6])
76	# return []
77
78	@property	1✔
79	def value(self):	1✔
80	return self._exp.value	1✔
81
82	def __str__(self):	1✔
83	return self._exp.__str__()	1✔
84
85	def __repr__(self):	1✔
86	return self._exp.__repr__()	1✔
87
88	@property	1✔
89	def len(self) -> int:	1✔
90	return self._exp.len	1✔
91
92	@property	1✔
93	def kind(self) -> int:	1✔
94	return self.typespec.kind	1✔
95
96
97	class ExpOp(ExpGeneric):	1✔
98
99	@property	1✔
100	def unary_op(self):	1✔
101	return self._args[3]	×
102
103	@property	1✔
104	def unary_args(self) -> tuple[Expression, Expression]:	1✔
105	return Expression(self._args[4], version=self.version), Expression(	×
106	self._args[5], version=self.version
107	)
108
109
110	class ExpNotImplemented(ExpGeneric):	1✔
111	@property	1✔
112	def value(self):	1✔
113	raise NotImplementedError	×
114
115
116	class ExpFunction(ExpGeneric):	1✔
117	@property	1✔
118	def value(self):	1✔
119	return self._args[3]	×
120
121	@property	1✔
122	def args(self):	1✔
123	return Expression(self._args[4], version=self.version)	×
124
125
126	class ExpConstant(ExpGeneric):	1✔
127	@property	1✔
128	def value(self):	1✔
129	if self._type == "REAL":	1✔
130	return utils.hextofloat(utils.string_clean(self._args[3]), self.kind)	1✔
131	elif self._type == "INTEGER" or self._type == "UNSIGNED":	1✔
132	return int(utils.string_clean(self._args[3]))	1✔
133	elif self._type == "CHARACTER":	1✔
134	return utils.string_clean(self._args[4])	1✔
135	elif self._type == "COMPLEX":	1✔
136	return complex(	1✔
137	utils.hextofloat(utils.string_clean(self._args[3]), self.kind),
138	utils.hextofloat(utils.string_clean(self._args[4]), self.kind),
139	)
140	elif self._type == "LOGICAL":	1✔
141	return int(self._args[3]) == 1	1✔
142	else:
143	raise NotImplementedError(f"Type={self._type} args3={self._args[3]}")	×
144
145	@property	1✔
146	def len(self) -> int:	1✔
147	if self._type == "CHARACTER":	1✔
148	return int(self._args[3])	1✔
149	return -1	1✔
150
151	def __str__(self):	1✔
152	if self._type == "CHARACTER":	1✔
153	return f"CHARACTER(kind={self.kind},len={self.len})"	1✔
154	else:
155	return f"{self.type}(kind={self.kind})"	1✔
156
157	@property	1✔
158	def type(self) -> str:	1✔
159	return self._type	1✔
160
161	@property	1✔
162	def raw(self) -> str:	1✔
163	return utils.string_clean(self._args[3])	×
164
165
166	class ExpVariable(ExpGeneric):	1✔
167	@property	1✔
168	def value(self):	1✔
169	return self._args[3]	×
170
171	def __str__(self):	1✔
172	if self._type == "CHARACTER":	×
173	return f"CHARACTER(kind={self.kind},len={self.len})"	×
174	else:
175	return f"{self._type}"	×
176
177
178	class ExpArray(ExpGeneric):	1✔
179	def __init__(self, args, *kwargs):	1✔
180	super().__init__(args, *kwargs)	1✔
181	self._value = None	1✔
182
183	@property	1✔
184	def value(self) -> np.ndarray:	1✔
185	if self._value is None:	1✔
186	self._value = []	1✔
187	for i in self._args[3]:	1✔
188	self._value.append(Expression(i[0], version=self.version))	1✔
189
190	value = []	1✔
191
192	for v in self._value:	1✔
193	value.append(v.value)	1✔
194
195	return np.array(value, dtype=self.dtype).reshape(self.shape)	1✔
196
197	@property	1✔
198	def shape(self) -> tuple[int, ...]:	1✔
199	return tuple([int(utils.string_clean(i)) for i in self._args[4]])	1✔
200
201	@property	1✔
202	def dtype(self) -> np.dtype:	1✔
203	v = Expression(self._args[3][0][0], version=self.version)	1✔
204	return utils.dtype(v.type, self.kind, len=v.len)	1✔
205
206	def __str__(self):	1✔
207	v = Expression(self._args[3][0][0], version=self.version)	1✔
208	if v.type == "CHARACTER":	1✔
209	return f"CHARACTER(kind={v.kind},len={v.len}),dimension{self.shape}"	1✔
210	else:
211	return f"{v.type},dimension{self.shape}"	1✔
212
213
214	class ExpSubString(ExpNotImplemented):	1✔
215	pass	1✔
216
217
218	class ExpStructure(ExpNotImplemented):	1✔
219	pass	1✔
220
221
222	class ExpNull(ExpNotImplemented):	1✔
223	pass	1✔
224
225
226	class ExpCompCall(ExpNotImplemented):	1✔
227	pass	1✔
228
229
230	class ExpPPC(ExpNotImplemented):	1✔
231	pass	1✔
232
233
234	class ExpConditional(ExpNotImplemented):	1✔
235	pass	1✔
236
237
238	class ExpUnknown(ExpNotImplemented):	1✔
239	pass	1✔
240
241
242	# Need to store this here as we get a cyclic dependency
243	# between expressions and typespec
244	class typespec:	1✔
245	def __init__(self, typespec, *, version):	1✔
246	self._typespec = typespec	1✔
247	self.version = version	1✔
248
249	@property	1✔
250	def type(self) -> str:	1✔
251	return self._typespec[0]	1✔
252
253	def _isclass(self) -> bool:	1✔
254	return self.type == "CLASS" or self.type == "DERIVED"	1✔
255
256	@property	1✔
257	def kind(self) -> int:	1✔
258	if not self._isclass():	1✔
259	return int(self._typespec[1])	1✔
260	return -1	×
261
262	@property	1✔
263	def class_ref(self) -> int:	1✔
264	if self._isclass():	×
265	return int(self._typespec[1])	×
266	return -1	×
267
268	@property	1✔
269	def interface(self):	1✔
270	return self._typespec[2]	×
271
272	@property	1✔
273	def is_c_interop(self) -> bool:	1✔
274	return int(self._typespec[3]) == 1	1✔
275
276	@property	1✔
277	def is_iso_c(self) -> bool:	1✔
278	return int(self._typespec[4]) == 1	1✔
279
280	@property	1✔
281	def type2(self):	1✔
282	# Whats this?
283	return self._typespec[5]	×
284
285	@property	1✔
286	def charlen(self) -> Expression:	1✔
287	if len(self._typespec[6]):	1✔
288	return Expression(self._typespec[6][0], version=self.version)	1✔
289	raise AttributeError("Object does not have a character length")	×
290
291	# try:
292	# if not args[6][0]:
293	# self.charlen = -1
294	# else:
295	# self.charlen = Expression(
296	# *args[6][0]
297	# ) # TODO: might this need to be iterated for mulit-d strings?
298	# except IndexError:
299	# self.charlen = -1
300
301	@property	1✔
302	def deferred_cl(self) -> bool:	1✔
303	if len(self._typespec) == 8:	×
304	return self._typespec[7] == "DEFERRED_CL"	×
305
306	return False	×

rjfarmer / gfModParser / 18542693371

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous