用可视化案例讲Rust编程4. 用泛型和特性实现自适配shapefile的读取

news2024/11/16 4:34:30

本节已经涉及Rust学习曲线上的一个大坑：泛型和特性了，属于语言的深水区，如果初学者，建议看一眼知道有这个功能即可。

如果我们立足于功能实现，那么做到像上一节那样就可以了，从原理上来说，每个函数满足唯一的功能，是一种好的设计，软件工程里面“高内聚低耦合”是有利于系统的独立性的。

但是在调用的时候，就需要有一点的心智了，例如我们要读取点线面三种不同的shapefile，就要记住三种不同的函数名和三种不同的绘图要素构建程序，那也太麻烦了，有没有一种方法，让我们调用一种方法，根据不同的参数类型自动适配不同的处理逻辑呢？

如果你是学Python/Java/JavaScript这样的高级语言的同学，你肯定已经有了自己的实现方法，例如Python/Javascript作为动态类型的语言，可以只写一个入口函数，然后根据函数的输入参数自动去选择不同的分支来处理。

而Java也有同名方法覆盖，利用不同的函数签名（参数类型）来解决用同名函数实现不同的逻辑流程。

而C++则是可以用泛型和函数重载来实现这个功能。

那么在Rust里面如何解决呢？

首先，Rust不支持函数签名的类型推断。所有函数签名的类型必须写清楚，这就杜绝了我们用动态类型的语言那样根据输入的参数来匹配。
第二，Rust也不（直接）支持函数重载，为什么是不直接支持呢，因为可以通过泛型和特性来实现函数重载，也就是我们今天要做的内容。

老规矩，先看结果：

用同名的一个方法，实现对于三种不同的shapefile读取，并且获取三种不同的返回类型值，那么这是这么实现的呢？

先看代码：

架构定义

首先定义了一个结构体，这个空结构体的作用类似于Java里面的class。
然后定义了一个特性（trait），特性这个东西，在Rust中，类似于Java里面的接口，或者C++里面的虚函数，但是与接口和虚函数不同的时候，特性可以直接在里面写实现，也可以留空，如果你是学Java的，你把下面的代码理解为定义了一个接口，并且定义了一个实现接口的工厂方法即可。
在这里的特性中，我们定义一个泛型T，然后定义了一个方法，就叫做read_shp，输入产生就是一个shapefile的路径，返回值是一个T ,这个T就是Rust中的泛型，即代表一切可能的类型。
- 如果是没有接触过泛型的同学，可能会问，他既然代表一切类型，那么你不是说了Rust是类型严格的语言么？那系统编译器怎么知道，这个T到底是哪种类型呢？继续往下看

pub struct shp;
pub trait ReadShapfile<T> {
    fn read_shp(shp_path:&str) -> T;
}

然后我们开始写针对不同类型的shapefile的实现，实现如下：

//实现1：读取polygon类型的shapefile，通过返回值来匹配。
impl ReadShapfile<Vec<Polygon>> for shp{
    fn read_shp(shp_path:&str) -> Vec<Polygon>{
        let shp_read = shapefile::read_as::<_,
        shapefile::Polygon, shapefile::dbase::Record>(shp_path)
        .expect(&format!("Could not open polygon-shapefile, error: {}", shp_path));

    let mut polygons:Vec<Polygon> = Vec::new();
    for (polygon, polygon_record) in shp_read {
        let geo_mpolygon: geo_types::MultiPolygon<f64> = polygon.into();
        for poly in geo_mpolygon.iter(){
            polygons.push(poly.to_owned());
        }
    }
    polygons
    }
}

//实现2：读取Polyline类型的shapefile，通过返回值来匹配。
impl ReadShapfile<Vec<LineString>> for shp{
    fn read_shp(shp_path:&str) -> Vec<LineString>{
        let shp_read = shapefile::read_as::<_,
            shapefile::Polyline, shapefile::dbase::Record>(shp_path)
        .expect(&format!("Could not open polyline-shapefile, error: {}", shp_path));

        let mut linestrings:Vec<LineString> = Vec::new();
        for (pline, pline_record) in shp_read {
            let geo_mline: geo_types::MultiLineString<f64> = pline.into();
            for line in geo_mline.iter(){
                linestrings.push(line.to_owned());
            }
        }
        linestrings
    }
}
//实现2：读取Point类型的shapefile，通过返回值来匹配。
impl ReadShapfile<Vec<Point>> for shp{
    fn read_shp(shp_path:&str) -> Vec<Point>{
        let shp_read = shapefile::read_as::<_,
            shapefile::Point, shapefile::dbase::Record>(shp_path)
        .expect(&format!("Could not open polyline-shapefile, error: {}", shp_path));

        let mut pnts:Vec<Point> = Vec::new();
        for (pnt, pnt_record) in shp_read {
            let geo_pnt: geo_types::Point<f64> = pnt.into();
            pnts.push(geo_pnt.to_owned());
        }
        pnts
    }
}

简要的解释一下：

在Rust的语法中，impl 特性名称<返回值> for 结构名 这个语法，代表了具体的实现，而里面那个 <返回值> 部分，就是我们定义的泛型T，注意，T只是一个代称，习惯性用大写字母，你用ABCDEFG，或者abcdefg都是可以的。
然后在这个实现里面，去写具体方法的read_shp的实现，写完之后，就可以通过在调用的时候，再指定返回类型，来适配具体调用哪个方法了。

看到这里，可能有同学又有问题了：

上一节，直接写多个方法，多实用啊，读polygon就是polygon方法，读point就是point方法，干净整洁又高效，干嘛搞这么麻烦？