aboutsummaryrefslogtreecommitdiff
path: root/crates/hir_ty/src/op.rs
blob: 8533e1ed87e88c33039cd35ffdfaeb74fa8de16c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
//! Helper functions for binary operator type inference.
use chalk_ir::TyVariableKind;
use hir_def::expr::{ArithOp, BinaryOp, CmpOp};

use crate::{Interner, Scalar, Ty, TyKind};

pub(super) fn binary_op_return_ty(op: BinaryOp, lhs_ty: Ty, rhs_ty: Ty) -> Ty {
    match op {
        BinaryOp::LogicOp(_) | BinaryOp::CmpOp(_) => TyKind::Scalar(Scalar::Bool).intern(&Interner),
        BinaryOp::Assignment { .. } => Ty::unit(),
        BinaryOp::ArithOp(ArithOp::Shl) | BinaryOp::ArithOp(ArithOp::Shr) => {
            match lhs_ty.kind(&Interner) {
                TyKind::Scalar(Scalar::Int(_))
                | TyKind::Scalar(Scalar::Uint(_))
                | TyKind::Scalar(Scalar::Float(_)) => lhs_ty,
                TyKind::InferenceVar(_, TyVariableKind::Integer)
                | TyKind::InferenceVar(_, TyVariableKind::Float) => lhs_ty,
                _ => TyKind::Unknown.intern(&Interner),
            }
        }
        BinaryOp::ArithOp(_) => match rhs_ty.kind(&Interner) {
            TyKind::Scalar(Scalar::Int(_))
            | TyKind::Scalar(Scalar::Uint(_))
            | TyKind::Scalar(Scalar::Float(_)) => rhs_ty,
            TyKind::InferenceVar(_, TyVariableKind::Integer)
            | TyKind::InferenceVar(_, TyVariableKind::Float) => rhs_ty,
            _ => TyKind::Unknown.intern(&Interner),
        },
    }
}

pub(super) fn binary_op_rhs_expectation(op: BinaryOp, lhs_ty: Ty) -> Ty {
    match op {
        BinaryOp::LogicOp(..) => TyKind::Scalar(Scalar::Bool).intern(&Interner),
        BinaryOp::Assignment { op: None } => lhs_ty,
        BinaryOp::CmpOp(CmpOp::Eq { .. }) => match lhs_ty.kind(&Interner) {
            TyKind::Scalar(_) | TyKind::Str => lhs_ty,
            TyKind::InferenceVar(_, TyVariableKind::Integer)
            | TyKind::InferenceVar(_, TyVariableKind::Float) => lhs_ty,
            _ => TyKind::Unknown.intern(&Interner),
        },
        BinaryOp::ArithOp(ArithOp::Shl) | BinaryOp::ArithOp(ArithOp::Shr) => {
            TyKind::Unknown.intern(&Interner)
        }
        BinaryOp::CmpOp(CmpOp::Ord { .. })
        | BinaryOp::Assignment { op: Some(_) }
        | BinaryOp::ArithOp(_) => match lhs_ty.kind(&Interner) {
            TyKind::Scalar(Scalar::Int(_))
            | TyKind::Scalar(Scalar::Uint(_))
            | TyKind::Scalar(Scalar::Float(_)) => lhs_ty,
            TyKind::InferenceVar(_, TyVariableKind::Integer)
            | TyKind::InferenceVar(_, TyVariableKind::Float) => lhs_ty,
            _ => TyKind::Unknown.intern(&Interner),
        },
    }
}