Skip to content
Snippets Groups Projects
Commit d80d9dd7 authored by Udo Eisenbarth's avatar Udo Eisenbarth :speech_balloon:
Browse files

Add OpticPort

parent a4b7ea9b
No related branches found
No related tags found
No related merge requests found
......@@ -4,6 +4,7 @@
pub mod optic_scenery;
/// The basic structure representing an optical element
pub mod optic_node;
pub mod optic_port;
pub mod nodes;
......
#[derive(Debug, PartialEq, Clone)]
pub enum OpticPortDirection {
Incoming,
Outgoing,
}
impl OpticPortDirection {
fn invert(self) -> OpticPortDirection {
if self == OpticPortDirection::Incoming {
OpticPortDirection::Outgoing
} else {
OpticPortDirection::Incoming
}
}
}
#[derive(Debug)]
pub struct OpticPort {
name: String,
direction: OpticPortDirection,
}
impl OpticPort {
pub fn new(name: &str, direction: OpticPortDirection) -> Self {
Self{ name: name.into(), direction: direction }
}
pub fn set_name(&mut self, name: &str) {
self.name = name.into();
}
pub fn name(&self) -> &str {
self.name.as_ref()
}
pub fn set_direction(&mut self, direction: OpticPortDirection) {
self.direction = direction;
}
pub fn direction(&self) -> &OpticPortDirection {
&self.direction
}
pub fn invert(&mut self) {
self.direction = self.direction.clone().invert();
}
}
#[cfg(test)]
mod test {
use crate::optic_port::{OpticPort,OpticPortDirection};
#[test]
fn new() {
let port = OpticPort::new("Test", OpticPortDirection::Incoming);
assert_eq!(port.name, "Test");
assert_eq!(port.direction, OpticPortDirection::Incoming);
}
#[test]
fn set_name() {
let mut port = OpticPort::new("Test", OpticPortDirection::Incoming);
port.set_name("Test2");
assert_eq!(port.name, "Test2");
assert_eq!(port.direction, OpticPortDirection::Incoming);
}
#[test]
fn name() {
let port = OpticPort::new("Test", OpticPortDirection::Incoming);
assert_eq!(port.name(), "Test");
}
#[test]
fn set_direction() {
let mut port = OpticPort::new("Test", OpticPortDirection::Incoming);
port.set_direction(OpticPortDirection::Outgoing);
assert_eq!(port.name, "Test");
assert_eq!(port.direction, OpticPortDirection::Outgoing);
}
#[test]
fn direction() {
let port = OpticPort::new("Test", OpticPortDirection::Incoming);
assert_eq!(port.direction(), &OpticPortDirection::Incoming);
}
#[test]
fn invert() {
let mut port = OpticPort::new("Test", OpticPortDirection::Incoming);
port.invert();
assert_eq!(port.direction, OpticPortDirection::Outgoing);
assert_eq!(port.name, "Test");
}
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment