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

/// The documentation is taken from original [C++ library by Joshua Tippetts](http://accidentalnoise.sourceforge.net/docs.html).

use super::implicit_base::ImplicitModuleBase;
use super::ImplicitModule;
use super::cellular_gen::*;

use std::rc::Rc;
use std::cell::RefCell;

/// The Cellular function provides an interface to a [`CellularGenerator`](struct.CellularGenerator.html) to generate cellular noise based on the function originally written by Steve Worley. The function calculates F1,F2,F3 and F4 as distances to the first, second, third and fourth nearest seed points, applies the specified coefficients to the distances and sums them together, returning the result.
pub struct ImplicitCellular {
    base: ImplicitModuleBase,
    generator: Option<Rc<RefCell<CellularGenerator>>>,
    coefficients: [f64; 4],
}

impl ImplicitCellular {
    pub fn new() -> ImplicitCellular {
        ImplicitCellular {
            base: Default::default(),
            generator: None,
            coefficients: [1.0, 0.0, 0.0, 0.0],
        }
    }

    pub fn with_coefficients(a: f64, b: f64, c: f64, d: f64) -> ImplicitCellular {
        ImplicitCellular {
            base: Default::default(),
            generator: None,
            coefficients: [a, b, c, d],
        }
    }

    pub fn set_coefficients(&mut self, a: f64, b: f64, c: f64, d: f64) {
        self.coefficients[0] = a;
        self.coefficients[1] = b;
        self.coefficients[2] = c;
        self.coefficients[3] = d;
    }

    pub fn set_cellular_source(&mut self, source: Option<Rc<RefCell<CellularGenerator>>>) {
        self.generator = source;
    }
}

impl ImplicitModule for ImplicitCellular {
    fn set_seed(&mut self, seed: u32) {
        if let Some(ref mut g) = self.generator {
            g.borrow_mut().set_seed(seed);
        }
    }

    fn get_2d(&mut self, x: f64, y: f64) -> f64 {
        match self.generator {
            None => 0.0,
            Some(ref mut g) => {
                let mut b = g.borrow_mut();
                let c = b.get_2d(x, y);
                c.f[0] * self.coefficients[0] + c.f[1] * self.coefficients[1] + c.f[2] * self.coefficients[2] + c.f[3] * self.coefficients[3]
            }
        }
    }
    fn get_3d(&mut self, x: f64, y: f64, z: f64) -> f64 {
        match self.generator {
            None => 0.0,
            Some(ref mut g) => {
                let mut b = g.borrow_mut();
                let c = b.get_3d(x, y, z);
                c.f[0] * self.coefficients[0] + c.f[1] * self.coefficients[1] + c.f[2] * self.coefficients[2] + c.f[3] * self.coefficients[3]
            }
        }
    }
    fn get_4d(&mut self, x: f64, y: f64, z: f64, w: f64) -> f64 {
        match self.generator {
            None => 0.0,
            Some(ref mut g) => {
                let mut b = g.borrow_mut();
                let c = b.get_4d(x, y, z, w);
                c.f[0] * self.coefficients[0] + c.f[1] * self.coefficients[1] + c.f[2] * self.coefficients[2] + c.f[3] * self.coefficients[3]
            }
        }
    }
    fn get_6d(&mut self, x: f64, y: f64, z: f64, w: f64, u: f64, v: f64) -> f64 {
        match self.generator {
            None => 0.0,
            Some(ref mut g) => {
                let mut b = g.borrow_mut();
                let c = b.get_6d(x, y, z, w, u, v);
                c.f[0] * self.coefficients[0] + c.f[1] * self.coefficients[1] + c.f[2] * self.coefficients[2] + c.f[3] * self.coefficients[3]
            }
        }
    }


    fn set_deriv_spacing(&mut self, s: f64) {
        self.base.spacing = s;
    }
    fn spacing(&self) -> f64 {
        self.base.spacing
    }
}