18387296101

Committed 09 Oct 2025 07:45PM UTC coverage: 88.772% (+0.08%) from 88.694%

Build # 18387296101

Build Type

push

github

Committed by

rjfarmer

Commit Message

More typing

Coverage Stats

18 of 22 new or added lines in 4 files covered. (81.82%)

1012 of 1140 relevant lines covered (88.77%)

0.89 hits per line

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

86.81

/gfModParser/modules/expressions.py

# SPDX-License-Identifier: GPL-2.0+

import numpy as np
from typing import List, Tuple

from .. import utils

from . import procedures


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,
            "UNKNOWN": ExpUnknown,
        }

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

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

    @property
    def typespec(self):
        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 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 ExpOp(ExpGeneric):

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

    @property
    def unary_args(self):
        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):
        if self._type == "CHARACTER":
            return int(self._args[3])

    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):
        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:
        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 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):
        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) -> int:
        return self._typespec[6]

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

rjfarmer / gfModParser / 18387296101

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