Skip to content

Commit b64a25e

Browse files
committed
Derive Float for newtypes
1 parent 9690611 commit b64a25e

File tree

2 files changed

+200
-2
lines changed

2 files changed

+200
-2
lines changed

src/lib.rs

+185-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
#![crate_type = "proc-macro"]
1212
#![doc(html_root_url = "https://docs.rs/num-derive/0.2")]
13-
#![recursion_limit="256"]
13+
#![recursion_limit="512"]
1414

1515
//! Procedural macros to derive numeric traits in Rust.
1616
//!
@@ -519,3 +519,187 @@ pub fn num(input: TokenStream) -> TokenStream {
519519
}
520520
}).into()
521521
}
522+
523+
#[proc_macro_derive(Float)]
524+
pub fn float(input: TokenStream) -> TokenStream {
525+
let ast: syn::DeriveInput = syn::parse(input).unwrap();
526+
let name = &ast.ident;
527+
let inner_ty = newtype_inner(&ast.data).expect(NEWTYPE_ONLY);
528+
dummy_const_trick("Float", &name, quote! {
529+
extern crate num_traits as _num_traits;
530+
impl _num_traits::Float for #name {
531+
fn nan() -> Self {
532+
#name(<#inner_ty as _num_traits::Float>::nan())
533+
}
534+
fn infinity() -> Self {
535+
#name(<#inner_ty as _num_traits::Float>::infinity())
536+
}
537+
fn neg_infinity() -> Self {
538+
#name(<#inner_ty as _num_traits::Float>::neg_infinity())
539+
}
540+
fn neg_zero() -> Self {
541+
#name(<#inner_ty as _num_traits::Float>::neg_zero())
542+
}
543+
fn min_value() -> Self {
544+
#name(<#inner_ty as _num_traits::Float>::min_value())
545+
}
546+
fn min_positive_value() -> Self {
547+
#name(<#inner_ty as _num_traits::Float>::min_positive_value())
548+
}
549+
fn max_value() -> Self {
550+
#name(<#inner_ty as _num_traits::Float>::max_value())
551+
}
552+
fn is_nan(self) -> bool {
553+
<#inner_ty as _num_traits::Float>::is_nan(self.0)
554+
}
555+
fn is_infinite(self) -> bool {
556+
<#inner_ty as _num_traits::Float>::is_infinite(self.0)
557+
}
558+
fn is_finite(self) -> bool {
559+
<#inner_ty as _num_traits::Float>::is_finite(self.0)
560+
}
561+
fn is_normal(self) -> bool {
562+
<#inner_ty as _num_traits::Float>::is_normal(self.0)
563+
}
564+
fn classify(self) -> ::std::num::FpCategory {
565+
<#inner_ty as _num_traits::Float>::classify(self.0)
566+
}
567+
fn floor(self) -> Self {
568+
#name(<#inner_ty as _num_traits::Float>::floor(self.0))
569+
}
570+
fn ceil(self) -> Self {
571+
#name(<#inner_ty as _num_traits::Float>::ceil(self.0))
572+
}
573+
fn round(self) -> Self {
574+
#name(<#inner_ty as _num_traits::Float>::round(self.0))
575+
}
576+
fn trunc(self) -> Self {
577+
#name(<#inner_ty as _num_traits::Float>::trunc(self.0))
578+
}
579+
fn fract(self) -> Self {
580+
#name(<#inner_ty as _num_traits::Float>::fract(self.0))
581+
}
582+
fn abs(self) -> Self {
583+
#name(<#inner_ty as _num_traits::Float>::abs(self.0))
584+
}
585+
fn signum(self) -> Self {
586+
#name(<#inner_ty as _num_traits::Float>::signum(self.0))
587+
}
588+
fn is_sign_positive(self) -> bool {
589+
<#inner_ty as _num_traits::Float>::is_sign_positive(self.0)
590+
}
591+
fn is_sign_negative(self) -> bool {
592+
<#inner_ty as _num_traits::Float>::is_sign_negative(self.0)
593+
}
594+
fn mul_add(self, a: Self, b: Self) -> Self {
595+
#name(<#inner_ty as _num_traits::Float>::mul_add(self.0, a.0, b.0))
596+
}
597+
fn recip(self) -> Self {
598+
#name(<#inner_ty as _num_traits::Float>::recip(self.0))
599+
}
600+
fn powi(self, n: i32) -> Self {
601+
#name(<#inner_ty as _num_traits::Float>::powi(self.0, n))
602+
}
603+
fn powf(self, n: Self) -> Self {
604+
#name(<#inner_ty as _num_traits::Float>::powf(self.0, n.0))
605+
}
606+
fn sqrt(self) -> Self {
607+
#name(<#inner_ty as _num_traits::Float>::sqrt(self.0))
608+
}
609+
fn exp(self) -> Self {
610+
#name(<#inner_ty as _num_traits::Float>::exp(self.0))
611+
}
612+
fn exp2(self) -> Self {
613+
#name(<#inner_ty as _num_traits::Float>::exp2(self.0))
614+
}
615+
fn ln(self) -> Self {
616+
#name(<#inner_ty as _num_traits::Float>::ln(self.0))
617+
}
618+
fn log(self, base: Self) -> Self {
619+
#name(<#inner_ty as _num_traits::Float>::log(self.0, base.0))
620+
}
621+
fn log2(self) -> Self {
622+
#name(<#inner_ty as _num_traits::Float>::log2(self.0))
623+
}
624+
fn log10(self) -> Self {
625+
#name(<#inner_ty as _num_traits::Float>::log10(self.0))
626+
}
627+
fn max(self, other: Self) -> Self {
628+
#name(<#inner_ty as _num_traits::Float>::max(self.0, other.0))
629+
}
630+
fn min(self, other: Self) -> Self {
631+
#name(<#inner_ty as _num_traits::Float>::min(self.0, other.0))
632+
}
633+
fn abs_sub(self, other: Self) -> Self {
634+
#name(<#inner_ty as _num_traits::Float>::abs_sub(self.0, other.0))
635+
}
636+
fn cbrt(self) -> Self {
637+
#name(<#inner_ty as _num_traits::Float>::cbrt(self.0))
638+
}
639+
fn hypot(self, other: Self) -> Self {
640+
#name(<#inner_ty as _num_traits::Float>::hypot(self.0, other.0))
641+
}
642+
fn sin(self) -> Self {
643+
#name(<#inner_ty as _num_traits::Float>::sin(self.0))
644+
}
645+
fn cos(self) -> Self {
646+
#name(<#inner_ty as _num_traits::Float>::cos(self.0))
647+
}
648+
fn tan(self) -> Self {
649+
#name(<#inner_ty as _num_traits::Float>::tan(self.0))
650+
}
651+
fn asin(self) -> Self {
652+
#name(<#inner_ty as _num_traits::Float>::asin(self.0))
653+
}
654+
fn acos(self) -> Self {
655+
#name(<#inner_ty as _num_traits::Float>::acos(self.0))
656+
}
657+
fn atan(self) -> Self {
658+
#name(<#inner_ty as _num_traits::Float>::atan(self.0))
659+
}
660+
fn atan2(self, other: Self) -> Self {
661+
#name(<#inner_ty as _num_traits::Float>::atan2(self.0, other.0))
662+
}
663+
fn sin_cos(self) -> (Self, Self) {
664+
let (x, y) = <#inner_ty as _num_traits::Float>::sin_cos(self.0);
665+
(#name(x), #name(y))
666+
}
667+
fn exp_m1(self) -> Self {
668+
#name(<#inner_ty as _num_traits::Float>::exp_m1(self.0))
669+
}
670+
fn ln_1p(self) -> Self {
671+
#name(<#inner_ty as _num_traits::Float>::ln_1p(self.0))
672+
}
673+
fn sinh(self) -> Self {
674+
#name(<#inner_ty as _num_traits::Float>::sinh(self.0))
675+
}
676+
fn cosh(self) -> Self {
677+
#name(<#inner_ty as _num_traits::Float>::cosh(self.0))
678+
}
679+
fn tanh(self) -> Self {
680+
#name(<#inner_ty as _num_traits::Float>::tanh(self.0))
681+
}
682+
fn asinh(self) -> Self {
683+
#name(<#inner_ty as _num_traits::Float>::asinh(self.0))
684+
}
685+
fn acosh(self) -> Self {
686+
#name(<#inner_ty as _num_traits::Float>::acosh(self.0))
687+
}
688+
fn atanh(self) -> Self {
689+
#name(<#inner_ty as _num_traits::Float>::atanh(self.0))
690+
}
691+
fn integer_decode(self) -> (u64, i16, i8) {
692+
<#inner_ty as _num_traits::Float>::integer_decode(self.0)
693+
}
694+
fn epsilon() -> Self {
695+
#name(<#inner_ty as _num_traits::Float>::epsilon())
696+
}
697+
fn to_degrees(self) -> Self {
698+
#name(<#inner_ty as _num_traits::Float>::to_degrees(self.0))
699+
}
700+
fn to_radians(self) -> Self {
701+
#name(<#inner_ty as _num_traits::Float>::to_radians(self.0))
702+
}
703+
}
704+
}).into()
705+
}

tests/newtype.rs

+15-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,8 @@ extern crate num as num_renamed;
22
#[macro_use]
33
extern crate num_derive;
44

5-
use num_renamed::{FromPrimitive, ToPrimitive, NumCast, One, Zero, Num};
5+
use num_renamed::{FromPrimitive, ToPrimitive, NumCast, One, Zero, Num, Float};
6+
use std::ops::Neg;
67

78
#[derive(
89
Debug,
@@ -17,9 +18,17 @@ use num_renamed::{FromPrimitive, ToPrimitive, NumCast, One, Zero, Num};
1718
One,
1819
Zero,
1920
Num,
21+
Float,
2022
)]
2123
struct MyFloat(f64);
2224

25+
impl Neg for MyFloat {
26+
type Output = MyFloat;
27+
fn neg(self) -> Self {
28+
MyFloat(self.0.neg())
29+
}
30+
}
31+
2332
#[test]
2433
fn test_from_primitive() {
2534
assert_eq!(MyFloat::from_u32(25), Some(MyFloat(25.0)));
@@ -58,3 +67,8 @@ fn test_one() {
5867
fn test_num() {
5968
assert_eq!(MyFloat::from_str_radix("25", 10).ok(), Some(MyFloat(25.0)));
6069
}
70+
71+
#[test]
72+
fn test_float() {
73+
assert_eq!(MyFloat(4.0).log(MyFloat(2.0)), MyFloat(2.0));
74+
}

0 commit comments

Comments
 (0)