sway_core/language/ty/declaration/
trait_fn.rs

1use std::{
2    fmt,
3    hash::{Hash, Hasher},
4};
5
6use monomorphization::MonomorphizeHelper;
7use sway_types::{Ident, Named, Span, Spanned};
8
9use crate::{
10    engine_threading::*,
11    has_changes,
12    language::{parsed::TraitFn, ty::*, Purity},
13    transform,
14    type_system::*,
15};
16
17#[derive(Clone, Debug)]
18pub struct TyTraitFn {
19    pub name: Ident,
20    pub(crate) span: Span,
21    pub(crate) purity: Purity,
22    pub parameters: Vec<TyFunctionParameter>,
23    pub return_type: GenericArgument,
24    pub attributes: transform::Attributes,
25}
26
27impl TyDeclParsedType for TyTraitFn {
28    type ParsedType = TraitFn;
29}
30
31impl DebugWithEngines for TyTraitFn {
32    fn fmt(&self, f: &mut fmt::Formatter<'_>, engines: &Engines) -> fmt::Result {
33        write!(
34            f,
35            "{:?}({}):{}",
36            self.name,
37            self.parameters
38                .iter()
39                .map(|p| format!(
40                    "{}:{}",
41                    p.name.as_str(),
42                    engines.help_out(p.type_argument.initial_type_id())
43                ))
44                .collect::<Vec<_>>()
45                .join(", "),
46            engines.help_out(self.return_type.initial_type_id()),
47        )
48    }
49}
50
51impl Named for TyTraitFn {
52    fn name(&self) -> &Ident {
53        &self.name
54    }
55}
56
57impl Spanned for TyTraitFn {
58    fn span(&self) -> Span {
59        self.span.clone()
60    }
61}
62
63impl IsConcrete for TyTraitFn {
64    fn is_concrete(&self, engines: &Engines) -> bool {
65        self.type_parameters()
66            .iter()
67            .all(|tp| tp.is_concrete(engines))
68            && self
69                .return_type
70                .type_id()
71                .is_concrete(engines, TreatNumericAs::Concrete)
72            && self.parameters().iter().all(|t| {
73                t.type_argument
74                    .type_id()
75                    .is_concrete(engines, TreatNumericAs::Concrete)
76            })
77    }
78}
79
80impl declaration::FunctionSignature for TyTraitFn {
81    fn parameters(&self) -> &Vec<TyFunctionParameter> {
82        &self.parameters
83    }
84
85    fn return_type(&self) -> &GenericArgument {
86        &self.return_type
87    }
88}
89
90impl EqWithEngines for TyTraitFn {}
91impl PartialEqWithEngines for TyTraitFn {
92    fn eq(&self, other: &Self, ctx: &PartialEqWithEnginesContext) -> bool {
93        let type_engine = ctx.engines().te();
94        self.name == other.name
95            && self.purity == other.purity
96            && self.parameters.eq(&other.parameters, ctx)
97            && type_engine
98                .get(self.return_type.type_id())
99                .eq(&type_engine.get(other.return_type.type_id()), ctx)
100            && self.attributes == other.attributes
101    }
102}
103
104impl HashWithEngines for TyTraitFn {
105    fn hash<H: Hasher>(&self, state: &mut H, engines: &Engines) {
106        let TyTraitFn {
107            name,
108            purity,
109            parameters,
110            return_type,
111            // these fields are not hashed because they aren't relevant/a
112            // reliable source of obj v. obj distinction
113            span: _,
114            attributes: _,
115        } = self;
116        let type_engine = engines.te();
117        name.hash(state);
118        parameters.hash(state, engines);
119        type_engine.get(return_type.type_id()).hash(state, engines);
120        purity.hash(state);
121    }
122}
123
124impl SubstTypes for TyTraitFn {
125    fn subst_inner(&mut self, ctx: &SubstTypesContext) -> HasChanges {
126        has_changes! {
127            self.parameters.subst(ctx);
128            self.return_type.subst(ctx);
129        }
130    }
131}
132
133impl MonomorphizeHelper for TyTraitFn {
134    fn name(&self) -> &Ident {
135        &self.name
136    }
137
138    fn type_parameters(&self) -> &[TypeParameter] {
139        &[]
140    }
141
142    fn has_self_type_param(&self) -> bool {
143        false
144    }
145}
sway_core/language/ty/declaration/trait_fn.rs

sway_core/language/ty/declaration/
trait_fn.rs
