fuzzy-logic_rs 0.5.0

An implementation of fuzzy logic in rust!
Documentation 
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

use crate::membership_functions::{
    linear_membership, GetDegree, MembershipFunction, TSKMembershipFunction,
};
use crate::membership_ranges::MembershipRange;

#[derive(Debug, Clone)]
pub struct InputVariable {
    name: String,
    range: (f64, f64),
    mfs: Vec<MembershipFunction>,
}

impl InputVariable {
    pub fn new(name: String, range: (f64, f64)) -> Self {
        Self {
            name,
            range,
            mfs: Vec::new(),
        }
    }
    pub fn add_membership(&mut self, mf: MembershipFunction) {
        self.mfs.push(mf);
    }

    pub fn fuzzify(&self, idx: usize, x: f64) -> f64 {
        self.mfs[idx].get_degree(x)
    }
    pub fn membership_function_name(&self, idx: i32) -> String {
        self.mfs[idx as usize].get_name().clone()
    }
    pub fn get_name(&self) -> &String {
        &self.name
    }

    pub fn get_range(&self) -> &(f64, f64) {
        &self.range
    }
}

#[derive(Debug)]
pub struct OutputVariable {
    name: String,
    mrs: Vec<MembershipRange>,
    universe: Vec<f64>,
}

impl OutputVariable {
    pub fn new(name: String, range: (f64, f64), n: i32) -> Self {
        let mut universe = Vec::new();
        let (start, stop) = range;
        let delta = (stop - start) / (n as f64);
        for i in 0..n {
            universe.push(start + delta * (i as f64))
        }
        Self {
            name,
            mrs: Vec::new(),
            universe,
        }
    }
    pub fn add_membership(&mut self, membership_rang: MembershipRange) {
        self.mrs.push(membership_rang)
    }

    pub fn get_mu(&self, idx: usize) -> &Vec<f64> {
        self.mrs[idx].get_mu()
    }

    pub fn get_universe(&self) -> &Vec<f64> {
        &self.universe
    }

    pub fn get_name(&self) -> &String {
        &self.name
    }

    pub fn get_universe_by_idx(&self, idx: usize) -> f64 {
        self.universe[idx].clone()
    }
}

#[derive(Debug)]
pub struct TSKOutputVariable {
    name: String,
    mfs: Vec<TSKMembershipFunction>,
}

impl TSKOutputVariable {
    pub fn new(name: String) -> Self {
        Self {
            name,
            mfs: Vec::new(),
        }
    }

    pub fn add_membership(&mut self, membership: TSKMembershipFunction) {
        self.mfs.push(membership);
    }

    pub fn add_constant_membership(&mut self, value: f64) {
        self.mfs.push(TSKMembershipFunction::Constant(value));
    }

    pub fn add_linear_membership(&mut self, coefficients: Vec<f64>) {
        self.mfs.push(TSKMembershipFunction::Linear(coefficients));
    }

    pub fn get_mu(&self, idx: usize, input_vec: &Vec<f64>) -> f64 {
        match &self.mfs[idx] {
            TSKMembershipFunction::Constant(c) => *c,
            TSKMembershipFunction::Linear(coeff) => linear_membership(&coeff, input_vec),
            TSKMembershipFunction::Custom(fun) => fun(input_vec),
        }
    }

    pub fn get_name(&self) -> &String {
        &self.name
    }
}