Skip to main content

extendr_api/
lib.rs

1//! An ergonomic, opinionated, safe and user-friendly wrapper to the R-API
2//!
3//! This library aims to provide an interface that will be familiar to
4//! first-time users of Rust or indeed any compiled language.
5//!
6//! See [`Robj`] for much of the content of this crate.
7//! [`Robj`] provides a safe wrapper for the R object type.
8//!
9//! ## Examples
10//!
11//! Use attributes and macros to export to R.
12//!
13//! ```rust,ignore
14//! use extendr_api::prelude::*;
15//! // Export a function or impl to R.
16//! #[extendr]
17//! fn fred(a: i32) -> i32 {
18//!     a + 1
19//! }
20//!
21//! // define exports using extendr_module
22//! extendr_module! {
23//!    mod mymodule;
24//!    fn fred;
25//! }
26//! ```
27//!
28//! In R:
29//!
30//! ```r
31//! result <- fred(1)
32//! ```
33//!
34//! [`Robj`] is a wrapper for R objects.
35//! The [`r!`] and `R!` macros let you build R objects
36//! using Rust and R syntax respectively.
37//! ```
38//! use extendr_api::prelude::*;
39//! test! {
40//!     // An R object with a single string "hello"
41//!     let character = r!("hello");
42//!     let character = r!(["hello", "goodbye"]);
43//!
44//!     // An R integer object with a single number 1L.
45//!     // Note that in Rust, 1 is an integer and 1.0 is a real.
46//!     let integer = r!(1);
47//!
48//!     // An R real object with a single number 1.
49//!     // Note that in R, 1 is a real and 1L is an integer.
50//!     let real = r!(1.0);
51//!
52//!     // An R real vector.
53//!     let real_vector = r!([1.0, 2.0]);
54//!     let real_vector = &[1.0, 2.0].iter().collect_robj();
55//!     let real_vector = r!(vec![1.0, 2.0]);
56//!
57//!     // An R function object.
58//!     let function = R!("function(x, y) { x + y }")?;
59//!
60//!     // A named list using the list! macro.
61//!     let list = list!(a = 1, b = 2);
62//!
63//!     // An unnamed list (of R objects) using the List wrapper.
64//!     let list = r!(List::from_values(vec![1, 2, 3]));
65//!     let list = r!(List::from_values(vec!["a", "b", "c"]));
66//!     let list = r!(List::from_values(&[r!("a"), r!(1), r!(2.0)]));
67//!
68//!     // A symbol
69//!     let sym = sym!(wombat);
70//!
71//!     // A R vector using collect_robj()
72//!     let vector = (0..3).map(|x| x * 10).collect_robj();
73//! }
74//! ```
75//!
76//! In Rust, we prefer to use iterators rather than loops.
77//!
78//! ```
79//! use extendr_api::prelude::*;
80//! test! {
81//!     // 1 ..= 100 is the same as 1:100
82//!     let res = r!(1 ..= 100);
83//!     assert_eq!(res, R!("1:100")?);
84//!
85//!     // Rust arrays are zero-indexed so it is more common to use 0 .. 100.
86//!     let res = r!(0 .. 100);
87//!     assert_eq!(res.len(), 100);
88//!
89//!     // Using map is a super fast way to generate vectors.
90//!     let iter = (0..3).map(|i| format!("fred{}", i));
91//!     let character = iter.collect_robj();
92//!     assert_eq!(character, r!(["fred0", "fred1", "fred2"]));
93//! }
94//! ```
95//!
96//! To index a vector, first convert it to a slice and then
97//! remember to use 0-based indexing. In Rust, going out of bounds
98//! will cause and error (a panic) unlike C++ which may crash.
99//! ```
100//! use extendr_api::prelude::*;
101//! test! {
102//!     let vals = r!([1.0, 2.0]);
103//!     let slice = vals.as_real_slice().ok_or("expected slice")?;
104//!     let one = slice[0];
105//!     let two = slice[1];
106//!     // let error = slice[2];
107//!     assert_eq!(one, 1.0);
108//!     assert_eq!(two, 2.0);
109//! }
110//! ```
111//!
112//! Much slower, but more general are these methods:
113//! ```
114//! use extendr_api::prelude::*;
115//! test! {
116//!     let vals = r!([1.0, 2.0, 3.0]);
117//!
118//!     // one-based indexing [[i]], returns an object.
119//!     assert_eq!(vals.index(1)?, r!(1.0));
120//!
121//!     // one-based slicing [x], returns an object.
122//!     assert_eq!(vals.slice(1..=2)?, r!([1.0, 2.0]));
123//!
124//!     // $ operator, returns an object
125//!     let list = list!(a = 1.0, b = "xyz");
126//!     assert_eq!(list.dollar("a")?, r!(1.0));
127//! }
128//! ```
129//!
130//! The [`R!`] macro lets you embed R code in Rust
131//! and takes Rust expressions in `{{ }}` pairs.
132//!
133//! The [`Rraw!`] macro will not expand the `{{ }}` pairs.
134//! ```
135//! use extendr_api::prelude::*;
136//! test! {
137//!     // The text "1 + 1" is parsed as R source code.
138//!     // The result is 1.0 + 1.0 in Rust.
139//!     assert_eq!(R!("1 + 1")?, r!(2.0));
140//!
141//!     let a = 1.0;
142//!     assert_eq!(R!("1 + {{a}}")?, r!(2.0));
143//!
144//!     assert_eq!(R!(r"
145//!         x <- {{ a }}
146//!         x + 1
147//!     ")?, r!(2.0));
148//!
149//!     assert_eq!(R!(r#"
150//!         x <- "hello"
151//!         x
152//!     "#)?, r!("hello"));
153//!
154//!     // Use the R meaning of {{ }} and do not expand.
155//!     assert_eq!(Rraw!(r"
156//!         x <- {{ 1 }}
157//!         x + 1
158//!     ")?, r!(2.0));
159//! }
160//! ```
161//!
162//! The [`r!`] macro converts a rust object to an R object
163//! and takes parameters.
164//! ```
165//! use extendr_api::prelude::*;
166//! test! {
167//!     // The text "1.0+1.0" is parsed as Rust source code.
168//!     let one = 1.0;
169//!     assert_eq!(r!(one+1.0), r!(2.0));
170//! }
171//! ```
172//!
173//! Rust has a concept of "Owned" and "Borrowed" objects.
174//!
175//! Owned objects, such as [`Vec`] and [`String`] allocate memory
176//! which is released when the object lifetime ends.
177//!
178//! Borrowed objects such as `&[i32]` and `&str` are just pointers
179//! to annother object's memory and can't live longer than the
180//! object they reference.
181//!
182//! Borrowed objects are much faster than owned objects and use less
183//! memory but are used only for temporary access.
184//!
185//! When we take a slice of an R vector, for example, we need the
186//! original R object to be alive or the data will be corrupted.
187//!
188//! ```
189//! use extendr_api::prelude::*;
190//! test! {
191//!     // robj is an "Owned" object that controls the memory allocated.
192//!     let robj = r!([1, 2, 3]);
193//!
194//!     // Here slice is a "borrowed" reference to the bytes in robj.
195//!     // and cannot live longer than robj.
196//!     let slice = robj.as_integer_slice().ok_or("expected slice")?;
197//!     assert_eq!(slice.len(), 3);
198//! }
199//! ```
200//!
201//! ## Writing tests
202//!
203//! To test the functions exposed to R, wrap your code in the [`test!`] macro.
204//! This macro starts up the necessary R machinery for tests to work.
205//!
206//! ```rust,no_run
207//! use extendr_api::prelude::*;
208//!
209//! #[extendr]
210//! fn things() ->  Strings {
211//!     Strings::from_values(vec!["Test", "this"])
212//! }
213//!
214//! // define exports using extendr_module
215//! extendr_module! {
216//!    mod mymodule;
217//!    fn things;
218//! }
219//!
220//!
221//! #[cfg(test)]
222//! mod test {
223//!     use super::*;
224//!     use extendr_api::prelude::*;
225//!
226//!     #[test]
227//!     fn test_simple_function() {
228//!         assert_eq!(things().elt(0), "Test")
229//!     }
230//! }
231//! ```
232//!
233//! ## Returning `Result<T, E>` to R
234//!
235//! Two experimental features for returning error-aware R `list`s, `result_list` and `result_condition`,
236//! can be toggled to avoid panics on `Err`. Instead, an `Err` `x` is returned as either
237//!  - list: `list(ok=NULL, err=x)` when `result_list` is enabled,
238//!  - error condition: `<error: extendr_error>`, with `x` placed in `condition$value`, when `resultd_condition` is enabled.
239//!
240//! It is currently solely up to the user to handle any result on R side.
241//!
242//! There is an added overhead of wrapping Rust results in an R `list` object.
243//!
244//! ```rust,ignore
245//! use extendr_api::prelude::*;
246//! // simple function always returning an Err string
247//! #[extendr]
248//! fn oups(a: i32) -> std::result::Result<i32, String> {
249//!     Err("I did it again".to_string())
250//! }
251//!
252//! // define exports using extendr_module
253//! extendr_module! {
254//!    mod mymodule;
255//!    fn oups;
256//! }
257//!
258//! ```
259//!
260//! In R:
261//!
262//! ```rust,ignore
263//! # default result_panic feature
264//! oups(1)
265//! > ... long panic traceback from rust printed to stderr
266//!
267//! # result_list feature
268//! lst <- oups(1)
269//! print(lst)
270//! > list(ok = NULL, err = "I did it again")
271//!
272//! # result_condition feature
273//! cnd <- oups(1)
274//! print(cnd)
275//! > <error: extendr_error>
276//! print(cnd$value)
277//! > "I did it again"
278//!
279//! # handling example for result_condition
280//! oups_handled <- function(a) {
281//!   val_or_err <- oups(1)
282//!   if (inherits(val_or_err, "extendr_error")) stop(val_or_err)
283//!   val_or_err
284//! }
285//! ```
286//!
287//! ## Feature gates
288//!
289//! extendr-api has some optional features behind these feature gates:
290//!
291//! - `ndarray`: provides the conversion between R's matrices and [`ndarray`](https://docs.rs/ndarray/latest/ndarray/).
292//! - `num-complex`: provides the conversion between R's complex numbers and [`num-complex`](https://docs.rs/num-complex/latest/num_complex/).
293//! - `serde`: provides the [`serde`](https://serde.rs/) support.
294//! - `graphics`: provides the functionality to control or implement graphics devices.
295//! - `either`: provides implementation of type conversion traits for `Either<L, R>` from [`either`](https://docs.rs/either/latest/either/) if `L` and `R` both implement those traits.
296//! - `faer`: provides conversion between R's matrices and [`faer`](https://docs.rs/faer/latest/faer/).
297//!
298//! extendr-api supports three ways of returning a Result<T,E> to R.
299//! Only one behavior feature can be enabled at a time.
300//! - `result_panic`: Default behavior, return `Ok` as is, panic! on any `Err`
301//!
302//! Default behavior can be overridden by specifying `extend_api` features, i.e. `extendr-api = {..., default-features = false, features= ["result_condition"]}`
303//! These features are experimental and are subject to change.
304//! - `result_list`: return `Ok` as `list(ok=?, err=NULL)` or `Err` `list(ok=NULL, err=?)`
305//! - `result_condition`: return `Ok` as is or `Err` as $value in an R error condition.
306#![doc(
307    html_logo_url = "https://raw.githubusercontent.com/extendr/extendr/main/extendr-logo-256.png"
308)]
309
310pub mod conditions;
311pub mod error;
312pub mod functions;
313pub mod io;
314pub mod iter;
315pub mod lang_macros;
316pub mod metadata;
317pub mod na;
318pub mod optional;
319pub mod ownership;
320pub mod prelude;
321pub mod rmacros;
322pub mod robj;
323pub mod scalar;
324pub mod thread_safety;
325pub mod wrapper;
326
327pub use robj::Robj;
328pub use std::convert::{TryFrom, TryInto};
329pub use std::ops::Deref;
330pub use std::ops::DerefMut;
331pub use std::str::FromStr;
332
333#[cfg(feature = "serde")]
334pub mod serializer;
335
336#[cfg(feature = "serde")]
337pub mod deserializer;
338
339#[cfg(feature = "graphics")]
340pub mod graphics;
341
342pub(crate) mod conversions;
343
344//////////////////////////////////////////////////
345// Note these pub use statements are deprecated
346//
347// `use extendr_api::prelude::*;`
348//
349// instead.
350
351pub use error::*;
352pub use functions::*;
353pub use lang_macros::*;
354pub use na::*;
355pub use robj::*;
356pub use thread_safety::{catch_r_error, single_threaded, throw_r_error};
357pub use wrapper::*;
358
359pub use extendr_macros::*;
360
361use extendr_ffi::SEXPTYPE;
362use scalar::Rbool;
363
364//////////////////////////////////////////////////
365
366/// TRUE value eg. `r!(TRUE)`
367pub const TRUE: Rbool = Rbool::true_value();
368
369/// FALSE value eg. `r!(FALSE)`
370pub const FALSE: Rbool = Rbool::false_value();
371
372/// NULL value eg. `Robj::null()`
373pub const NULL: () = ();
374
375/// NA value for integers eg. `r!(NA_INTEGER)`
376pub const NA_INTEGER: Option<i32> = None;
377
378/// NA value for real values eg. `r!(NA_REAL)`
379pub const NA_REAL: Option<f64> = None;
380
381/// NA value for strings. `r!(NA_STRING)`
382pub const NA_STRING: Option<&str> = None;
383
384/// NA value for logical. `r!(NA_LOGICAL)`
385pub const NA_LOGICAL: Rbool = Rbool::na_value();
386
387/// This is needed for the generation of wrappers.
388#[doc(hidden)]
389pub use extendr_ffi::DllInfo;
390
391/// This is necessary for `#[extendr]`-impl
392#[doc(hidden)]
393pub use extendr_ffi::R_ExternalPtrAddr;
394
395/// This is used in `#[extendr(use_rng = true)]` on `fn`-items.
396#[doc(hidden)]
397pub use extendr_ffi::GetRNGstate;
398
399/// This is used in `#[extendr(use_rng = true)]` on `fn`-items.
400#[doc(hidden)]
401pub use extendr_ffi::PutRNGstate;
402
403#[doc(hidden)]
404pub use extendr_ffi::SEXP;
405
406#[doc(hidden)]
407pub use extendr_ffi::{R_NilValue, Rf_errorcall, Rf_warningcall};
408
409pub use metadata::Metadata;
410
411#[doc(hidden)]
412pub struct CallMethod {
413    pub call_symbol: std::ffi::CString,
414    pub func_ptr: *const u8,
415    pub num_args: i32,
416}
417
418unsafe fn make_method_def(
419    cstrings: &mut Vec<std::ffi::CString>,
420    rmethods: &mut Vec<extendr_ffi::R_CallMethodDef>,
421    func: &metadata::Func,
422    wrapped_name: &str,
423) {
424    cstrings.push(std::ffi::CString::new(wrapped_name).unwrap());
425    rmethods.push(extendr_ffi::R_CallMethodDef {
426        name: cstrings.last().unwrap().as_ptr(),
427        fun: Some(std::mem::transmute::<
428            *const u8,
429            unsafe extern "C" fn() -> *mut std::ffi::c_void,
430        >(func.func_ptr)),
431        numArgs: func.args.len() as i32,
432    });
433}
434
435// Internal function used to implement the .Call interface.
436// This is called from the code generated by the #[extendr] attribute.
437#[doc(hidden)]
438pub unsafe fn register_call_methods(info: *mut extendr_ffi::DllInfo, metadata: Metadata) {
439    let mut rmethods = Vec::new();
440    let mut cstrings = Vec::new();
441    for func in metadata.functions {
442        make_method_def(&mut cstrings, &mut rmethods, &func, func.c_name);
443    }
444
445    for imp in metadata.impls {
446        for func in imp.methods {
447            make_method_def(&mut cstrings, &mut rmethods, &func, func.c_name);
448        }
449    }
450
451    rmethods.push(extendr_ffi::R_CallMethodDef {
452        name: std::ptr::null(),
453        fun: None,
454        numArgs: 0,
455    });
456
457    extendr_ffi::R_registerRoutines(
458        info,
459        std::ptr::null(),
460        rmethods.as_ptr(),
461        std::ptr::null(),
462        std::ptr::null(),
463    );
464
465    // This seems to allow both symbols and strings,
466    extendr_ffi::R_useDynamicSymbols(info, extendr_ffi::Rboolean::FALSE);
467    extendr_ffi::R_forceSymbols(info, extendr_ffi::Rboolean::FALSE);
468}
469
470/// Type of R objects used by [Robj::rtype].
471#[derive(Debug, PartialEq)]
472pub enum Rtype {
473    Null,        // NILSXP
474    Symbol,      // SYMSXP
475    Pairlist,    // LISTSXP
476    Function,    // CLOSXP
477    Environment, // ENVSXP
478    Promise,     // PROMSXP
479    Language,    // LANGSXP
480    Special,     // SPECIALSXP
481    Builtin,     // BUILTINSXP
482    Rstr,        // CHARSXP
483    Logicals,    // LGLSXP
484    Integers,    // INTSXP
485    Doubles,     // REALSXP
486    Complexes,   // CPLXSXP
487    Strings,     // STRSXP
488    Dot,         // DOTSXP
489    Any,         // ANYSXP
490    List,        // VECSXP
491    Expressions, // EXPRSXP
492    Bytecode,    // BCODESXP
493    ExternalPtr, // EXTPTRSXP
494    WeakRef,     // WEAKREFSXP
495    Raw,         // RAWSXP
496    S4,          // S4SXP
497    Unknown,
498}
499
500/// Enum use to unpack R objects into their specialist wrappers.
501// Todo: convert all Robj types to wrappers.
502// Note: this only works if the wrappers are all just SEXPs.
503#[derive(Debug, PartialEq)]
504pub enum Rany<'a> {
505    Null(&'a Robj),               // NILSXP
506    Symbol(&'a Symbol),           // SYMSXP
507    Pairlist(&'a Pairlist),       // LISTSXP
508    Function(&'a Function),       // CLOSXP
509    Environment(&'a Environment), // ENVSXP
510    Promise(&'a Promise),         // PROMSXP
511    Language(&'a Language),       // LANGSXP
512    Special(&'a Primitive),       // SPECIALSXP
513    Builtin(&'a Primitive),       // BUILTINSXP
514    Rstr(&'a Rstr),               // CHARSXP
515    Logicals(&'a Logicals),       // LGLSXP
516    Integers(&'a Integers),       // INTSXP
517    Doubles(&'a Doubles),         // REALSXP
518    Complexes(&'a Complexes),     // CPLXSXP
519    Strings(&'a Strings),         // STRSXP
520    Dot(&'a Robj),                // DOTSXP
521    Any(&'a Robj),                // ANYSXP
522    List(&'a List),               // VECSXP
523    Expressions(&'a Expressions), // EXPRSXP
524    Bytecode(&'a Robj),           // BCODESXP
525    ExternalPtr(&'a Robj),        // EXTPTRSXP
526    WeakRef(&'a Robj),            // WEAKREFSXP
527    Raw(&'a Raw),                 // RAWSXP
528    S4(&'a S4),                   // S4SXP
529    Unknown(&'a Robj),
530}
531
532/// Convert extendr's Rtype to R's SEXPTYPE.
533/// Panics if the type is Unknown.
534pub fn rtype_to_sxp(rtype: Rtype) -> SEXPTYPE {
535    use extendr_ffi::SEXPTYPE;
536    match rtype {
537        Rtype::Null => SEXPTYPE::NILSXP,
538        Rtype::Symbol => SEXPTYPE::SYMSXP,
539        Rtype::Pairlist => SEXPTYPE::LISTSXP,
540        Rtype::Function => SEXPTYPE::CLOSXP,
541        Rtype::Environment => SEXPTYPE::ENVSXP,
542        Rtype::Promise => SEXPTYPE::PROMSXP,
543        Rtype::Language => SEXPTYPE::LANGSXP,
544        Rtype::Special => SEXPTYPE::SPECIALSXP,
545        Rtype::Builtin => SEXPTYPE::BUILTINSXP,
546        Rtype::Rstr => SEXPTYPE::CHARSXP,
547        Rtype::Logicals => SEXPTYPE::LGLSXP,
548        Rtype::Integers => SEXPTYPE::INTSXP,
549        Rtype::Doubles => SEXPTYPE::REALSXP,
550        Rtype::Complexes => SEXPTYPE::CPLXSXP,
551        Rtype::Strings => SEXPTYPE::STRSXP,
552        Rtype::Dot => SEXPTYPE::DOTSXP,
553        Rtype::Any => SEXPTYPE::ANYSXP,
554        Rtype::List => SEXPTYPE::VECSXP,
555        Rtype::Expressions => SEXPTYPE::EXPRSXP,
556        Rtype::Bytecode => SEXPTYPE::BCODESXP,
557        Rtype::ExternalPtr => SEXPTYPE::EXTPTRSXP,
558        Rtype::WeakRef => SEXPTYPE::WEAKREFSXP,
559        Rtype::Raw => SEXPTYPE::RAWSXP,
560        #[cfg(not(use_objsxp))]
561        Rtype::S4 => SEXPTYPE::S4SXP,
562        #[cfg(use_objsxp)]
563        Rtype::S4 => SEXPTYPE::OBJSXP,
564        Rtype::Unknown => panic!("attempt to use Unknown Rtype"),
565    }
566}
567
568/// Convert R's SEXPTYPE to extendr's Rtype.
569pub fn sxp_to_rtype(sxptype: SEXPTYPE) -> Rtype {
570    match sxptype {
571        SEXPTYPE::NILSXP => Rtype::Null,
572        SEXPTYPE::SYMSXP => Rtype::Symbol,
573        SEXPTYPE::LISTSXP => Rtype::Pairlist,
574        SEXPTYPE::CLOSXP => Rtype::Function,
575        SEXPTYPE::ENVSXP => Rtype::Environment,
576        SEXPTYPE::PROMSXP => Rtype::Promise,
577        SEXPTYPE::LANGSXP => Rtype::Language,
578        SEXPTYPE::SPECIALSXP => Rtype::Special,
579        SEXPTYPE::BUILTINSXP => Rtype::Builtin,
580        SEXPTYPE::CHARSXP => Rtype::Rstr,
581        SEXPTYPE::LGLSXP => Rtype::Logicals,
582        SEXPTYPE::INTSXP => Rtype::Integers,
583        SEXPTYPE::REALSXP => Rtype::Doubles,
584        SEXPTYPE::CPLXSXP => Rtype::Complexes,
585        SEXPTYPE::STRSXP => Rtype::Strings,
586        SEXPTYPE::DOTSXP => Rtype::Dot,
587        SEXPTYPE::ANYSXP => Rtype::Any,
588        SEXPTYPE::VECSXP => Rtype::List,
589        SEXPTYPE::EXPRSXP => Rtype::Expressions,
590        SEXPTYPE::BCODESXP => Rtype::Bytecode,
591        SEXPTYPE::EXTPTRSXP => Rtype::ExternalPtr,
592        SEXPTYPE::WEAKREFSXP => Rtype::WeakRef,
593        SEXPTYPE::RAWSXP => Rtype::Raw,
594        #[cfg(not(use_objsxp))]
595        SEXPTYPE::S4SXP => Rtype::S4,
596        #[cfg(use_objsxp)]
597        SEXPTYPE::OBJSXP => Rtype::S4,
598        _ => Rtype::Unknown,
599    }
600}
601
602const PRINTF_NO_FMT_CSTRING: &[std::os::raw::c_char] = &[37, 115, 0]; // same as "%s\0"
603#[doc(hidden)]
604pub fn print_r_output<T: Into<Vec<u8>>>(s: T) {
605    let cs = std::ffi::CString::new(s).expect("NulError");
606    unsafe {
607        extendr_ffi::Rprintf(PRINTF_NO_FMT_CSTRING.as_ptr(), cs.as_ptr());
608    }
609}
610
611#[doc(hidden)]
612pub fn print_r_error<T: Into<Vec<u8>>>(s: T) {
613    let cs = std::ffi::CString::new(s).expect("NulError");
614    unsafe {
615        extendr_ffi::REprintf(PRINTF_NO_FMT_CSTRING.as_ptr(), cs.as_ptr());
616    }
617}
618
619#[cfg(test)]
620mod tests {
621    use super::prelude::*;
622    use crate as extendr_api;
623
624    use extendr_macros::extendr;
625    use extendr_macros::extendr_module;
626    use extendr_macros::pairlist;
627
628    #[allow(clippy::too_many_arguments)]
629    #[extendr]
630    pub fn inttypes(a: i8, b: u8, c: i16, d: u16, e: i32, f: u32, g: i64, h: u64) {
631        assert_eq!(a, 1);
632        assert_eq!(b, 2);
633        assert_eq!(c, 3);
634        assert_eq!(d, 4);
635        assert_eq!(e, 5);
636        assert_eq!(f, 6);
637        assert_eq!(g, 7);
638        assert_eq!(h, 8);
639    }
640
641    #[extendr]
642    pub fn floattypes(a: f32, b: f64) {
643        assert_eq!(a, 1.);
644        assert_eq!(b, 2.);
645    }
646
647    #[extendr]
648    pub fn strtypes(a: &str, b: String) {
649        assert_eq!(a, "abc");
650        assert_eq!(b, "def");
651    }
652
653    #[extendr]
654    pub fn vectortypes(a: Vec<i32>, b: Vec<f64>) {
655        assert_eq!(a, [1, 2, 3]);
656        assert_eq!(b, [4., 5., 6.]);
657    }
658
659    #[extendr]
660    pub fn robjtype(a: Robj) {
661        assert_eq!(a, Robj::from(1))
662    }
663
664    #[extendr]
665    pub fn return_u8() -> u8 {
666        123
667    }
668
669    #[extendr]
670    pub fn return_u16() -> u16 {
671        123
672    }
673
674    #[extendr]
675    pub fn return_u32() -> u32 {
676        123
677    }
678
679    #[extendr]
680    pub fn return_u64() -> u64 {
681        123
682    }
683
684    #[extendr]
685    pub fn return_i8() -> i8 {
686        123
687    }
688
689    #[extendr]
690    pub fn return_i16() -> i16 {
691        123
692    }
693
694    #[extendr]
695    pub fn return_i32() -> i32 {
696        123
697    }
698
699    #[extendr]
700    pub fn return_i64() -> i64 {
701        123
702    }
703
704    #[extendr]
705    pub fn return_f32() -> f32 {
706        123.
707    }
708
709    #[extendr]
710    pub fn return_f64() -> f64 {
711        123.
712    }
713
714    #[extendr]
715    pub fn f64_slice(x: &[f64]) -> &[f64] {
716        x
717    }
718
719    #[extendr]
720    pub fn i32_slice(x: &[i32]) -> &[i32] {
721        x
722    }
723
724    #[extendr]
725    pub fn bool_slice(x: &[Rbool]) -> &[Rbool] {
726        x
727    }
728
729    #[extendr]
730    pub fn f64_iter(x: Doubles) -> Doubles {
731        x
732    }
733
734    #[extendr]
735    pub fn i32_iter(x: Integers) -> Integers {
736        x
737    }
738
739    // #[extendr]
740    // pub fn bool_iter(x: Logicals) -> Logicals {
741    //     x
742    // }
743
744    #[extendr]
745    pub fn symbol(x: Symbol) -> Symbol {
746        x
747    }
748
749    #[extendr]
750    pub fn matrix(x: RMatrix<f64>) -> RMatrix<f64> {
751        x
752    }
753
754    #[extendr]
755    struct Person {
756        pub name: String,
757    }
758
759    #[extendr]
760    /// impl comment.
761    impl Person {
762        fn new() -> Self {
763            Self {
764                name: "".to_string(),
765            }
766        }
767
768        fn set_name(&mut self, name: &str) {
769            self.name = name.to_string();
770        }
771
772        fn name(&self) -> &str {
773            self.name.as_str()
774        }
775    }
776
777    // see metadata_test for the following comments.
778
779    /// comment #1
780    /// comment #2
781    /**
782        comment #3
783        comment #4
784    **/
785    #[extendr]
786    /// aux_func doc comment.
787    fn aux_func(_person: &Person) {}
788
789    // Macro to generate exports
790    extendr_module! {
791        mod my_module;
792        fn aux_func;
793        impl Person;
794    }
795
796    #[test]
797    fn export_test() {
798        test! {
799            use super::*;
800            // Call the exported functions through their generated C wrappers.
801            unsafe {
802                wrap__inttypes(
803                    Robj::from(1).get(),
804                    Robj::from(2).get(),
805                    Robj::from(3).get(),
806                    Robj::from(4).get(),
807                    Robj::from(5).get(),
808                    Robj::from(6).get(),
809                    Robj::from(7).get(),
810                    Robj::from(8).get(),
811                );
812                wrap__inttypes(
813                    Robj::from(1.).get(),
814                    Robj::from(2.).get(),
815                    Robj::from(3.).get(),
816                    Robj::from(4.).get(),
817                    Robj::from(5.).get(),
818                    Robj::from(6.).get(),
819                    Robj::from(7.).get(),
820                    Robj::from(8.).get(),
821                );
822                wrap__floattypes(Robj::from(1.).get(), Robj::from(2.).get());
823                wrap__floattypes(Robj::from(1).get(), Robj::from(2).get());
824                wrap__strtypes(Robj::from("abc").get(), Robj::from("def").get());
825                wrap__vectortypes(
826                    Robj::from(&[1, 2, 3] as &[i32]).get(),
827                    Robj::from(&[4., 5., 6.] as &[f64]).get(),
828                );
829                wrap__robjtype(Robj::from(1).get());
830
831                // General integer types.
832                assert_eq!(Robj::from_sexp(wrap__return_u8()), Robj::from(123_u8));
833                assert_eq!(Robj::from_sexp(wrap__return_u16()), Robj::from(123));
834                assert_eq!(Robj::from_sexp(wrap__return_u32()), Robj::from(123.));
835                assert_eq!(Robj::from_sexp(wrap__return_u64()), Robj::from(123.));
836                assert_eq!(Robj::from_sexp(wrap__return_i8()), Robj::from(123));
837                assert_eq!(Robj::from_sexp(wrap__return_i16()), Robj::from(123));
838                assert_eq!(Robj::from_sexp(wrap__return_i32()), Robj::from(123));
839                assert_eq!(Robj::from_sexp(wrap__return_i64()), Robj::from(123.));
840
841                // Floating point types.
842                assert_eq!(Robj::from_sexp(wrap__return_f32()), Robj::from(123.));
843                assert_eq!(Robj::from_sexp(wrap__return_f64()), Robj::from(123.));
844            }
845        }
846    }
847
848    #[test]
849    fn class_wrapper_test() {
850        test! {
851            let mut person = Person::new();
852            person.set_name("fred");
853            let robj = r!(person);
854            assert_eq!(robj.check_external_ptr_type::<Person>(), true);
855            let person2 = <&Person>::try_from(&robj).unwrap();
856            assert_eq!(person2.name(), "fred");
857        }
858    }
859
860    #[test]
861    fn slice_test() {
862        test! {
863            unsafe {
864                // #[extendr]
865                // pub fn f64_slice(x: &[f64]) -> &[f64] { x }
866
867                let robj = r!([1., 2., 3.]);
868                assert_eq!(Robj::from_sexp(wrap__f64_slice(robj.get())), robj);
869
870                // #[extendr]
871                // pub fn i32_slice(x: &[i32]) -> &[i32] { x }
872
873                let robj = r!([1, 2, 3]);
874                assert_eq!(Robj::from_sexp(wrap__i32_slice(robj.get())), robj);
875
876                // #[extendr]
877                // pub fn bool_slice(x: &[Rbool]) -> &[Rbool] { x }
878
879                let robj = r!([TRUE, FALSE, TRUE]);
880                assert_eq!(Robj::from_sexp(wrap__bool_slice(robj.get())), robj);
881
882                // #[extendr]
883                // pub fn f64_iter(x: Doubles) -> Doubles { x }
884
885                let robj = r!([1., 2., 3.]);
886                assert_eq!(Robj::from_sexp(wrap__f64_iter(robj.get())), robj);
887
888                // #[extendr]
889                // pub fn i32_iter(x: Integers) -> Integers { x }
890
891                let robj = r!([1, 2, 3]);
892                assert_eq!(Robj::from_sexp(wrap__i32_iter(robj.get())), robj);
893
894                // #[extendr]
895                // pub fn bool_iter(x: Logicals) -> Logicals { x }
896
897                // TODO: reinstate this test.
898                // let robj = r!([TRUE, FALSE, TRUE]);
899                // assert_eq!(Robj::from_sexp(wrap__bool_iter(robj.get())), robj);
900
901                // #[extendr]
902                // pub fn symbol(x: Symbol) -> Symbol { x }
903
904                let robj = sym!(fred);
905                assert_eq!(Robj::from_sexp(wrap__symbol(robj.get())), robj);
906
907                // #[extendr]
908                // pub fn matrix(x: Matrix<&[f64]>) -> Matrix<&[f64]> { x }
909
910                let m = RMatrix::new_matrix(1, 2, |r, c| if r == c {1.0} else {0.});
911                let robj = r!(m);
912                assert_eq!(Robj::from_sexp(wrap__matrix(robj.get())), robj);
913            }
914        }
915    }
916
917    #[test]
918    fn r_output_test() {
919        // R equivalent
920        // > txt_con <- textConnection("test_con", open = "w")
921        // > sink(txt_con)
922        // > cat("Hello world")
923        // > sink()
924        // > close(txt_con)
925        // > expect_equal(test_con, "Hello world")
926        //
927
928        test! {
929            let txt_con = R!(r#"textConnection("test_con", open = "w")"#).unwrap();
930            call!("sink", &txt_con).unwrap();
931            rprintln!("Hello world %%!"); //%% checks printf formatting is off, yields one % if on
932            call!("sink").unwrap();
933            call!("close", &txt_con).unwrap();
934            let result = R!("test_con").unwrap();
935            assert_eq!(result, r!("Hello world %%!"));
936        }
937    }
938
939    #[test]
940    fn test_na_str() {
941        assert_ne!(<&str>::na().as_ptr(), "NA".as_ptr());
942        assert_eq!(<&str>::na(), "NA");
943        assert!(!"NA".is_na());
944        assert!(<&str>::na().is_na());
945    }
946
947    #[test]
948    fn metadata_test() {
949        test! {
950            // Rust interface.
951            let metadata = get_my_module_metadata();
952            assert_eq!(metadata.functions[0].doc, " comment #1\n comment #2\n\n        comment #3\n        comment #4\n    *\n aux_func doc comment.");
953            assert_eq!(metadata.functions[0].rust_name, "aux_func");
954            assert_eq!(metadata.functions[0].mod_name, "aux_func");
955            assert_eq!(metadata.functions[0].r_name, "aux_func");
956            assert_eq!(metadata.functions[0].args[0].name, "_person");
957            assert_eq!(metadata.functions[1].rust_name, "get_my_module_metadata");
958            assert_eq!(metadata.impls[0].name, "Person");
959            assert_eq!(metadata.impls[0].methods.len(), 3);
960
961            // R interface
962            let robj = unsafe { Robj::from_sexp(wrap__get_my_module_metadata()) };
963            let functions = robj.dollar("functions").unwrap();
964            let impls = robj.dollar("impls").unwrap();
965            assert_eq!(functions.len(), 3);
966            assert_eq!(impls.len(), 1);
967        }
968    }
969
970    #[test]
971    fn pairlist_macro_works() {
972        test! {
973            assert_eq!(pairlist!(1, 2, 3), Pairlist::from_pairs(&[("", 1), ("", 2), ("", 3)]));
974            assert_eq!(pairlist!(a=1, 2, 3), Pairlist::from_pairs(&[("a", 1), ("", 2), ("", 3)]));
975            assert_eq!(pairlist!(1, b=2, 3), Pairlist::from_pairs(&[("", 1), ("b", 2), ("", 3)]));
976            assert_eq!(pairlist!(a=1, b=2, c=3), Pairlist::from_pairs(&[("a", 1), ("b", 2), ("c", 3)]));
977            assert_eq!(pairlist!(a=NULL), Pairlist::from_pairs(&[("a", ())]));
978            assert_eq!(pairlist!(), Pairlist::from(()));
979        }
980    }
981
982    #[test]
983    fn big_r_macro_works() {
984        test! {
985            assert_eq!(R!("1")?, r!(1.0));
986            assert_eq!(R!(r"1")?, r!(1.0));
987            assert_eq!(R!(r"
988                x <- 1
989                x
990            ")?, r!(1.0));
991            assert_eq!(R!(r"
992                x <- {{ 1.0 }}
993                x
994            ")?, r!(1.0));
995            assert_eq!(R!(r"
996                x <- {{ (0..4).collect_robj() }}
997                x
998            ")?, r!([0, 1, 2, 3]));
999            assert_eq!(R!(r#"
1000                x <- "hello"
1001                x
1002            "#)?, r!("hello"));
1003            assert_eq!(Rraw!(r"
1004                x <- {{ 1 }}
1005                x
1006            ")?, r!(1.0));
1007        }
1008    }
1009}