mirror of
https://github.com/nushell/nushell.git
synced 2025-04-11 14:58:21 +02:00
MathCeil, MathFloor and MathMode
This commit is contained in:
parent
756773a6ed
commit
5ae823612f
@ -62,6 +62,7 @@ pub fn create_default_context() -> EngineState {
|
|||||||
Math,
|
Math,
|
||||||
MathAbs,
|
MathAbs,
|
||||||
MathAvg,
|
MathAvg,
|
||||||
|
MathCeil,
|
||||||
MathFloor,
|
MathFloor,
|
||||||
MathMax,
|
MathMax,
|
||||||
MathMin,
|
MathMin,
|
||||||
|
73
crates/nu-command/src/math/ceil.rs
Normal file
73
crates/nu-command/src/math/ceil.rs
Normal file
@ -0,0 +1,73 @@
|
|||||||
|
use nu_protocol::ast::Call;
|
||||||
|
use nu_protocol::engine::{Command, EngineState, Stack};
|
||||||
|
use nu_protocol::{Example, PipelineData, ShellError, Signature, Span, Value};
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct SubCommand;
|
||||||
|
|
||||||
|
impl Command for SubCommand {
|
||||||
|
fn name(&self) -> &str {
|
||||||
|
"math ceil"
|
||||||
|
}
|
||||||
|
|
||||||
|
fn signature(&self) -> Signature {
|
||||||
|
Signature::build("math ceil")
|
||||||
|
}
|
||||||
|
|
||||||
|
fn usage(&self) -> &str {
|
||||||
|
"Applies the ceil function to a list of numbers"
|
||||||
|
}
|
||||||
|
|
||||||
|
fn run(
|
||||||
|
&self,
|
||||||
|
engine_state: &EngineState,
|
||||||
|
_stack: &mut Stack,
|
||||||
|
call: &Call,
|
||||||
|
input: PipelineData,
|
||||||
|
) -> Result<nu_protocol::PipelineData, nu_protocol::ShellError> {
|
||||||
|
let head = call.head;
|
||||||
|
input.map(
|
||||||
|
move |value| operate(value, head),
|
||||||
|
engine_state.ctrlc.clone(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn examples(&self) -> Vec<Example> {
|
||||||
|
vec![Example {
|
||||||
|
description: "Apply the ceil function to a list of numbers",
|
||||||
|
example: "[1.5 2.3 -3.1] | math ceil",
|
||||||
|
result: Some(Value::List {
|
||||||
|
vals: vec![Value::test_int(2), Value::test_int(3), Value::test_int(-3)],
|
||||||
|
span: Span::unknown(),
|
||||||
|
}),
|
||||||
|
}]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn operate(value: Value, head: Span) -> Value {
|
||||||
|
match value {
|
||||||
|
Value::Int { .. } => value,
|
||||||
|
Value::Float { val, span } => Value::Float {
|
||||||
|
val: val.ceil(),
|
||||||
|
span,
|
||||||
|
},
|
||||||
|
other => Value::Error {
|
||||||
|
error: ShellError::UnsupportedInput(
|
||||||
|
String::from("Only numerical values are supported"),
|
||||||
|
other.span().unwrap_or(head),
|
||||||
|
),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod test {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_examples() {
|
||||||
|
use crate::test_examples;
|
||||||
|
|
||||||
|
test_examples(SubCommand {})
|
||||||
|
}
|
||||||
|
}
|
@ -1,5 +1,6 @@
|
|||||||
mod abs;
|
mod abs;
|
||||||
mod avg;
|
mod avg;
|
||||||
|
mod ceil;
|
||||||
pub mod command;
|
pub mod command;
|
||||||
mod floor;
|
mod floor;
|
||||||
mod max;
|
mod max;
|
||||||
@ -14,6 +15,7 @@ mod utils;
|
|||||||
|
|
||||||
pub use abs::SubCommand as MathAbs;
|
pub use abs::SubCommand as MathAbs;
|
||||||
pub use avg::SubCommand as MathAvg;
|
pub use avg::SubCommand as MathAvg;
|
||||||
|
pub use ceil::SubCommand as MathCeil;
|
||||||
pub use command::MathCommand as Math;
|
pub use command::MathCommand as Math;
|
||||||
pub use floor::SubCommand as MathFloor;
|
pub use floor::SubCommand as MathFloor;
|
||||||
pub use max::SubCommand as MathMax;
|
pub use max::SubCommand as MathMax;
|
||||||
|
@ -88,7 +88,7 @@ pub fn mode(values: &[Value], head: &Span) -> Result<Value, ShellError> {
|
|||||||
// But f64 doesn't implement Hash, so we get the binary representation to use as
|
// But f64 doesn't implement Hash, so we get the binary representation to use as
|
||||||
// key in the HashMap
|
// key in the HashMap
|
||||||
let hashable_values: Result<Vec<HashableType>, ShellError> = values
|
let hashable_values: Result<Vec<HashableType>, ShellError> = values
|
||||||
.into_iter()
|
.iter()
|
||||||
.map(|val| match val {
|
.map(|val| match val {
|
||||||
Value::Int { val, .. } => Ok(HashableType::new(val.to_be_bytes(), NumberTypes::Int)),
|
Value::Int { val, .. } => Ok(HashableType::new(val.to_be_bytes(), NumberTypes::Int)),
|
||||||
Value::Duration { val, .. } => {
|
Value::Duration { val, .. } => {
|
||||||
|
Loading…
Reference in New Issue
Block a user