【学Rust写CAD】15 定点数实现(fixed.rs)

源代码

fixed.rs文件实现了一个定点数(Fixed Point)类型 Fixed,用于在整数运算中模拟小数运算。代码如下:

rust 复制代码
//小数位数
const FIXED_FRACTION_BITS: u32 = 16;
//用于 双线性插值(Bilinear Interpolation) 的计算,它决定了插值权重(weight)的精度位数。一般为4或8
const BILINEAR_INTERPOLATION_BITS: u32 = 4;

#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
pub struct Fixed(i32);

impl Fixed {
    pub const ONE: Fixed = Fixed(1 << FIXED_FRACTION_BITS);
    pub const HALF: Fixed = Fixed(1 << (FIXED_FRACTION_BITS-1));//FIXED_ONE
    
    /// 创建新的Fixed值
    pub fn new(value: i32) -> Self {
        Fixed(value)
    }
    
    /// 将浮点数转换为定点数
    pub fn from_float(x: f32) -> Self {
        Fixed(((x * (1 << FIXED_FRACTION_BITS) as f32) + 0.5) as i32)
    }
    
    /// 将定点数转换为整数(截断小数部分)
    pub fn to_int(self) -> i32 {
        self.0 >> FIXED_FRACTION_BITS
    }
    
    /// 获取双线性插值权重
    pub fn bilinear_weight(self) -> u32 {
        // 丢弃不需要的精度位
        let reduced = self.0 >> (FIXED_FRACTION_BITS - BILINEAR_INTERPOLATION_BITS);
        // 提取剩余的小数部分
        let fraction = reduced & ((1 << BILINEAR_INTERPOLATION_BITS) - 1);
        fraction as u32
    }
    
    /// 获取内部值
    pub fn raw_value(self) -> i32 {
        self.0
    }
}

// 实现一些运算符重载以便更方便地使用
impl std::ops::Add for Fixed {
    type Output = Self;
    fn add(self, rhs: Self) -> Self {
        Fixed(self.0 + rhs.0)
    }
}

impl std::ops::Sub for Fixed {
    type Output = Self;
    fn sub(self, rhs: Self) -> Self {
        Fixed(self.0 - rhs.0)
    }
}

impl std::ops::Mul for Fixed {
    type Output = Self;
    fn mul(self, rhs: Self) -> Self {
        // 定点数乘法需要调整小数位
        Fixed((self.0 as i64 * rhs.0 as i64 >> FIXED_FRACTION_BITS) as i32)
    }
}

impl std::ops::Div for Fixed {
    type Output = Self;
    fn div(self, rhs: Self) -> Self {
        // 定点数除法需要调整小数位
        Fixed(((self.0 as i64 << FIXED_FRACTION_BITS) / rhs.0 as i64) as i32)
    }
}

impl std::ops::Shr<u32> for Fixed {
    type Output = Self;
    fn shr(self, rhs: u32) -> Self {
        Fixed(self.0 >> rhs)
    }
}

impl std::ops::Shl<u32> for Fixed {
    type Output = Self;
    fn shl(self, rhs: u32) -> Self {
        Fixed(self.0 << rhs)
    }
}

代码解析:

常量定义
  1. FIXED_FRACTION_BITS: u32 = 16:
  • 表示小数部分占用的位数

  • 使用16位表示小数部分,意味着有16位小数和16位整数(在i32中)

  1. BILINEAR_INTERPOLATION_BITS: u32 = 4:
  • 用于双线性插值计算的精度位数

  • 通常设置为4或8位,决定了插值权重的精度

Fixed 结构体

Fixed 是一个包装了 i32 的新类型,用于表示定点数。

重要常量
  1. ONE: 表示定点数1.0,值为 1 << 16 (65536)

  2. HALF: 表示定点数0.5,值为 1 << 15 (32768)

主要方法
  1. new(value: i32): 直接从一个i32值创建Fixed数

  2. from_float(x: f32): 将浮点数转换为定点数

  • 公式:x * (1 << 16) + 0.5 (0.5用于四舍五入)
  1. to_int(): 将定点数转换为整数(截断小数部分)
  • 右移16位丢弃小数部分
  1. bilinear_weight(): 获取双线性插值权重
  • 先丢弃不需要的精度位(保留BILINEAR_INTERPOLATION_BITS位)

  • 然后提取剩余的小数部分作为权重

  1. raw_value(): 获取内部存储的原始i32值
运算符重载

实现了基本的算术运算,注意乘法和除法的特殊处理:

  1. 乘法:
  • 需要先将操作数扩展为i64防止溢出

  • 结果右移16位调整小数位

  1. 除法:
  • 被除数左移16位扩展

  • 然后进行除法运算

定点数表示原理

这个实现使用Q16.16格式的定点数:

  • 32位整数(i32)中,高16位表示整数部分,低16位表示小数部分

  • 例如:0x00010000 表示1.0 (1 << 16)

  • 例如:0x00008000 表示0.5 (1 << 15)

应用场景

这种定点数实现常用于:

  1. 需要高性能小数运算但不想用浮点数的场合

  2. 图形处理中的坐标计算

  3. 嵌入式系统等不支持浮点运算硬件的环境

  4. 双线性插值等需要精确控制精度的算法

双线性插值中使用时,bilinear_weight()方法提供了权重计算,通过控制BILINEAR_INTERPOLATION_BITS可以调整插值精度。

相关推荐
第一程序员12 小时前
Rust trait 入门:把 AI 客户端抽象成可替换接口
python·rust·github
分布式存储与RustFS12 小时前
RustFS Beta.10 性能解读:PUT 全面反超 MinIO,Rust 重写对象存储成了?
开发语言·后端·rust
l1t15 小时前
测试用rust重写的postgresql: pgrust
开发语言·postgresql·rust
@atweiwei16 小时前
Langchainrust:中LLM-as-a-Judge,用 Rust 实现一套 LLM 评估系统
开发语言·后端·ai·rust·llm·rag
程序员爱钓鱼19 小时前
Rust match 模式匹配详解:比 if 更强大的条件分支
后端·rust
爱吃牛肉的大老虎1 天前
rust基础之环境搭建
java·开发语言·rust
openKylin1 天前
与全球技术演进同频,openKylin 3.0从C迈向Rust
c语言·开发语言·rust·开源·开放原子·openkylin
Source.Liu2 天前
【A11】Duration —— 精确时长结构体实现
rust·time
右耳朵猫AI_2 天前
用 Rust 重写 Bun
rust·bun
独孤留白2 天前
Rust 可变性完整指南 —— 从默认不可变到多线程安全修改
rust