1 # This file is part of NIT ( http://www.nitlanguage.org ).
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
15 # Management and utilities on annotations
25 # Try to get its single annotation with a given name
26 # If there is no such an annotation, null is returned.
27 # If there is more than one annotation, a error message is printed and the first annotation is returned
28 fun get_single_annotation
(name
: String, modelbuilder
: ModelBuilder): nullable AAnnotation
30 var res
= get_annotations
(name
)
31 if res
.is_empty
then return null
32 if res
.length
> 1 then
33 modelbuilder
.error
(res
[1], "Error: multiple annotation `{name}`. A previous one is defined line {res[0].location.line_start}")
38 # Return all its annotations of a given name in the order of their declaration
39 # Retun an empty array if no such an annotation.
40 fun get_annotations
(name
: String): Array[AAnnotation]
42 var res
= new Array[AAnnotation]
43 var nas
= n_annotations
44 if nas
== null then return res
45 for na
in nas
.n_items
do
46 if na
.name
!= name
then continue
53 redef class AAnnotation
54 # The name of the annotation
57 return n_atid
.n_id
.text
60 # Get the single argument of `self` as a `String`.
61 # Raise error and return null on any inconsistency.
62 fun arg_as_string
(modelbuilder
: ModelBuilder): nullable String
65 if args
.length
== 1 then
66 var arg
= args
.first
.as_string
67 if arg
!= null then return arg
70 modelbuilder
.error
(self, "Annotation error: \"{name}\
" expects a single String as argument.")
74 # Get the single argument of `self` as an `Int`.
75 # Raise error and return null on any inconsistency.
76 fun arg_as_int
(modelbuilder
: ModelBuilder): nullable Int
79 if args
.length
== 1 then
80 var arg
= args
.first
.as_int
81 if arg
!= null then return arg
84 modelbuilder
.error
(self, "Annotation error: \"{name}\
" expects a single Int as argument.")
90 # Get `self` as a `String`.
91 # Return null if not a string.
92 fun as_string
: nullable String
94 if not self isa AExprAtArg then return null
96 if not nexpr
isa AStringFormExpr then return null
97 return nexpr
.value
.as(not null)
100 # Get `self` as an `Int`.
101 # Return null if not an integer.
102 fun as_int
: nullable Int
104 if not self isa AExprAtArg then return null
106 if not nexpr
isa AIntExpr then return null
107 return nexpr
.value
.as(not null)
110 # Get `self` as a single identifier.
111 # Return null if not a single identifier.
112 fun as_id
: nullable String
114 if not self isa AExprAtArg then return null
116 if not nexpr
isa ACallExpr then return null
117 if not nexpr
.n_expr
isa AImplicitSelfExpr then return null
118 if not nexpr
.n_args
.n_exprs
.is_empty
then return null
119 return nexpr
.n_id
.text