1ast_enum! {
2 #[cfg_attr(docsrs, doc(cfg(any(feature = "full", feature = "derive"))))]
4 #[non_exhaustive]
5 pub enum BinOp {
6 Add(Token![+]),
8 Sub(Token![-]),
10 Mul(Token![*]),
12 Div(Token![/]),
14 Rem(Token![%]),
16 And(Token![&&]),
18 Or(Token![||]),
20 BitXor(Token![^]),
22 BitAnd(Token![&]),
24 BitOr(Token![|]),
26 Shl(Token![<<]),
28 Shr(Token![>>]),
30 Eq(Token![==]),
32 Lt(Token![<]),
34 Le(Token![<=]),
36 Ne(Token![!=]),
38 Ge(Token![>=]),
40 Gt(Token![>]),
42 AddAssign(Token![+=]),
44 SubAssign(Token![-=]),
46 MulAssign(Token![*=]),
48 DivAssign(Token![/=]),
50 RemAssign(Token![%=]),
52 BitXorAssign(Token![^=]),
54 BitAndAssign(Token![&=]),
56 BitOrAssign(Token![|=]),
58 ShlAssign(Token![<<=]),
60 ShrAssign(Token![>>=]),
62 }
63}
64
65#[doc = r" A unary operator: `*`, `!`, `-`."]
#[non_exhaustive]
pub enum UnOp {
#[doc = r" The `*` operator for dereferencing"]
Deref(crate::token::Star),
#[doc = r" The `!` operator for logical inversion"]
Not(crate::token::Not),
#[doc = r" The `-` operator for negation"]
Neg(crate::token::Minus),
}ast_enum! {
66 #[cfg_attr(docsrs, doc(cfg(any(feature = "full", feature = "derive"))))]
68 #[non_exhaustive]
69 pub enum UnOp {
70 Deref(Token![*]),
72 Not(Token![!]),
74 Neg(Token![-]),
76 }
77}
78
79#[cfg(feature = "parsing")]
80pub(crate) mod parsing {
81 use crate::error::Result;
82 use crate::op::{BinOp, UnOp};
83 use crate::parse::{Parse, ParseStream};
84
85 #[cfg_attr(docsrs, doc(cfg(feature = "parsing")))]
86 impl Parse for BinOp {
87 fn parse(input: ParseStream) -> Result<Self> {
88 if input.peek(crate::token::PlusEqToken![+=]) {
89 input.parse().map(BinOp::AddAssign)
90 } else if input.peek(crate::token::MinusEqToken![-=]) {
91 input.parse().map(BinOp::SubAssign)
92 } else if input.peek(crate::token::StarEqToken![*=]) {
93 input.parse().map(BinOp::MulAssign)
94 } else if input.peek(crate::token::SlashEqToken![/=]) {
95 input.parse().map(BinOp::DivAssign)
96 } else if input.peek(crate::token::PercentEqToken![%=]) {
97 input.parse().map(BinOp::RemAssign)
98 } else if input.peek(crate::token::CaretEqToken![^=]) {
99 input.parse().map(BinOp::BitXorAssign)
100 } else if input.peek(crate::token::AndEqToken![&=]) {
101 input.parse().map(BinOp::BitAndAssign)
102 } else if input.peek(crate::token::OrEqToken![|=]) {
103 input.parse().map(BinOp::BitOrAssign)
104 } else if input.peek(crate::token::ShlEqToken![<<=]) {
105 input.parse().map(BinOp::ShlAssign)
106 } else if input.peek(crate::token::ShrEqToken![>>=]) {
107 input.parse().map(BinOp::ShrAssign)
108 } else if input.peek(crate::token::AndAndToken![&&]) {
109 input.parse().map(BinOp::And)
110 } else if input.peek(crate::token::OrOrToken![||]) {
111 input.parse().map(BinOp::Or)
112 } else if input.peek(crate::token::ShlToken![<<]) {
113 input.parse().map(BinOp::Shl)
114 } else if input.peek(crate::token::ShrToken![>>]) {
115 input.parse().map(BinOp::Shr)
116 } else if input.peek(crate::token::EqEqToken![==]) {
117 input.parse().map(BinOp::Eq)
118 } else if input.peek(crate::token::LeToken![<=]) {
119 input.parse().map(BinOp::Le)
120 } else if input.peek(crate::token::NeToken![!=]) {
121 input.parse().map(BinOp::Ne)
122 } else if input.peek(crate::token::GeToken![>=]) {
123 input.parse().map(BinOp::Ge)
124 } else if input.peek(crate::token::PlusToken![+]) {
125 input.parse().map(BinOp::Add)
126 } else if input.peek(crate::token::MinusToken![-]) {
127 input.parse().map(BinOp::Sub)
128 } else if input.peek(crate::token::StarToken![*]) {
129 input.parse().map(BinOp::Mul)
130 } else if input.peek(crate::token::SlashToken![/]) {
131 input.parse().map(BinOp::Div)
132 } else if input.peek(crate::token::PercentToken![%]) {
133 input.parse().map(BinOp::Rem)
134 } else if input.peek(crate::token::CaretToken![^]) {
135 input.parse().map(BinOp::BitXor)
136 } else if input.peek(crate::token::AndToken![&]) {
137 input.parse().map(BinOp::BitAnd)
138 } else if input.peek(crate::token::OrToken![|]) {
139 input.parse().map(BinOp::BitOr)
140 } else if input.peek(crate::token::LtToken![<]) {
141 input.parse().map(BinOp::Lt)
142 } else if input.peek(crate::token::GtToken![>]) {
143 input.parse().map(BinOp::Gt)
144 } else {
145 Err(input.error("expected binary operator"))
146 }
147 }
148 }
149
150 #[cfg_attr(docsrs, doc(cfg(feature = "parsing")))]
151 impl Parse for UnOp {
152 fn parse(input: ParseStream) -> Result<Self> {
153 let lookahead = input.lookahead1();
154 if lookahead.peek(crate::token::StarToken![*]) {
155 input.parse().map(UnOp::Deref)
156 } else if lookahead.peek(crate::token::NotToken![!]) {
157 input.parse().map(UnOp::Not)
158 } else if lookahead.peek(crate::token::MinusToken![-]) {
159 input.parse().map(UnOp::Neg)
160 } else {
161 Err(lookahead.error())
162 }
163 }
164 }
165}
166
167#[cfg(feature = "printing")]
168mod printing {
169 use crate::op::{BinOp, UnOp};
170 use proc_macro2::TokenStream;
171 use quote::ToTokens;
172
173 #[cfg_attr(docsrs, doc(cfg(feature = "printing")))]
174 impl ToTokens for BinOp {
175 fn to_tokens(&self, tokens: &mut TokenStream) {
176 match self {
177 BinOp::Add(t) => t.to_tokens(tokens),
178 BinOp::Sub(t) => t.to_tokens(tokens),
179 BinOp::Mul(t) => t.to_tokens(tokens),
180 BinOp::Div(t) => t.to_tokens(tokens),
181 BinOp::Rem(t) => t.to_tokens(tokens),
182 BinOp::And(t) => t.to_tokens(tokens),
183 BinOp::Or(t) => t.to_tokens(tokens),
184 BinOp::BitXor(t) => t.to_tokens(tokens),
185 BinOp::BitAnd(t) => t.to_tokens(tokens),
186 BinOp::BitOr(t) => t.to_tokens(tokens),
187 BinOp::Shl(t) => t.to_tokens(tokens),
188 BinOp::Shr(t) => t.to_tokens(tokens),
189 BinOp::Eq(t) => t.to_tokens(tokens),
190 BinOp::Lt(t) => t.to_tokens(tokens),
191 BinOp::Le(t) => t.to_tokens(tokens),
192 BinOp::Ne(t) => t.to_tokens(tokens),
193 BinOp::Ge(t) => t.to_tokens(tokens),
194 BinOp::Gt(t) => t.to_tokens(tokens),
195 BinOp::AddAssign(t) => t.to_tokens(tokens),
196 BinOp::SubAssign(t) => t.to_tokens(tokens),
197 BinOp::MulAssign(t) => t.to_tokens(tokens),
198 BinOp::DivAssign(t) => t.to_tokens(tokens),
199 BinOp::RemAssign(t) => t.to_tokens(tokens),
200 BinOp::BitXorAssign(t) => t.to_tokens(tokens),
201 BinOp::BitAndAssign(t) => t.to_tokens(tokens),
202 BinOp::BitOrAssign(t) => t.to_tokens(tokens),
203 BinOp::ShlAssign(t) => t.to_tokens(tokens),
204 BinOp::ShrAssign(t) => t.to_tokens(tokens),
205 }
206 }
207 }
208
209 #[cfg_attr(docsrs, doc(cfg(feature = "printing")))]
210 impl ToTokens for UnOp {
211 fn to_tokens(&self, tokens: &mut TokenStream) {
212 match self {
213 UnOp::Deref(t) => t.to_tokens(tokens),
214 UnOp::Not(t) => t.to_tokens(tokens),
215 UnOp::Neg(t) => t.to_tokens(tokens),
216 }
217 }
218 }
219}