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

/*
 * File: mod.rs
 * Project: directivity
 * Created Date: 08/10/2023
 * Author: Shun Suzuki
 * -----
 * Last Modified: 21/11/2023
 * Modified By: Shun Suzuki (suzuki@hapis.k.u-tokyo.ac.jp)
 * -----
 * Copyright (c) 2023 Shun Suzuki. All rights reserved.
 *
 */

mod sphere;
mod t4010a1;

use crate::{
    defined::{float, PI},
    geometry::{Transducer, Vector3},
};

pub use sphere::Sphere;
pub use t4010a1::T4010A1;

/// Directivity
pub trait Directivity: Send + Sync {
    fn directivity(theta_deg: float) -> float;
    fn directivity_from_tr(tr: &Transducer, target: &Vector3) -> float {
        let dir = tr.z_direction();
        Self::directivity((dir.cross(target).norm()).atan2(dir.dot(target)) * 180. / PI)
    }
}

#[cfg(test)]
pub mod tests {
    use super::*;

    use crate::geometry::UnitQuaternion;

    pub struct TestDirectivity {}

    impl Directivity for TestDirectivity {
        fn directivity(t: float) -> float {
            t
        }
    }

    #[test]
    fn directivity_from_tr() {
        let tr =
            crate::geometry::Transducer::new(0, 0, Vector3::zeros(), UnitQuaternion::identity());

        assert_approx_eq::assert_approx_eq!(
            0.,
            TestDirectivity::directivity_from_tr(&tr, &tr.z_direction())
        );
        assert_approx_eq::assert_approx_eq!(
            90.,
            TestDirectivity::directivity_from_tr(&tr, &tr.x_direction())
        );
    }
}