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 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023
//! Convert data from Rust to Julia and back. Call Julia functions. use crate::array::{ArrayData, ArrayDataMut, Dimensions}; use crate::error::{JlrsError, JlrsResult}; use crate::frame::Output; use crate::global::Global; use crate::module::Module; use crate::symbol::Symbol; use crate::traits::{private::Internal, ArrayDatatype, Frame, IntoJulia, JuliaType, TryUnbox, TemporarySymbol}; use jl_sys::{ jl_alloc_array_1d, jl_alloc_array_2d, jl_alloc_array_3d, jl_apply_array_type, jl_array_data, jl_array_dim, jl_array_dims, jl_array_eltype, jl_array_ndims, jl_array_nrows, jl_call, jl_call0, jl_call1, jl_call2, jl_call3, jl_exception_occurred, jl_field_index, jl_field_names, jl_fieldref, jl_fieldref_noalloc, jl_get_nth_field, jl_get_nth_field_noalloc, jl_is_array, jl_is_tuple, jl_new_array, jl_nfields, jl_ptr_to_array, jl_ptr_to_array_1d, jl_svec_data, jl_svec_len, jl_typeis, jl_typeof, jl_typeof_str, jl_value_t, }; use std::ffi::CStr; use std::fmt::{Debug, Formatter, Result as FmtResult}; use std::marker::PhantomData; use std::slice; /// This type alias is used to encode the result of a function call: `Ok` indicates the call was /// successful and contains the function's result, while `Err` indicates an exception was thrown /// and contains said exception. pub type CallResult<'frame, 'data> = Result<Value<'frame, 'data>, Value<'frame, 'data>>; /// Several values that are allocated consecutively. This can be used in combination with /// [`Value::call_values`] and [`Value::call_values_output`]. /// /// [`Value::call_values`]: struct.Value.html#method.call_values /// [`Value::call_values_output`]: struct.Value.html#method.call_values_output #[derive(Copy, Clone)] pub struct Values<'frame>(*mut *mut jl_value_t, usize, PhantomData<&'frame ()>); impl<'frame> Values<'frame> { pub(crate) fn wrap(ptr: *mut *mut jl_value_t, n: usize) -> Self { Values(ptr, n, PhantomData) } pub(crate) unsafe fn ptr(self) -> *mut *mut jl_value_t { self.0 } /// Returns the number of `Value`s in this group. pub fn len(&self) -> usize { self.1 } /// Get a specific `Value` in this group. Returns an error if the index is out of bounds. pub fn value(&self, index: usize) -> JlrsResult<Value<'frame, 'static>> { if index >= self.len() { return Err(JlrsError::OutOfBounds(index, self.len()).into()); } unsafe { Ok(Value(*(self.ptr().add(index)), PhantomData, PhantomData)) } } /// Allocate several values of the same type, this type must implement [`IntoJulia`]. The /// values will be protected from garbage collection inside the frame used to create them. /// This takes as many slots on the GC stack as values that are allocated. /// /// Returns an error if there is not enough space on the stack. /// /// [`IntoJulia`]: ../traits/trait.IntoJulia.html pub fn new<T, V, F>(frame: &mut F, data: V) -> JlrsResult<Self> where T: IntoJulia, V: AsRef<[T]>, F: Frame<'frame>, { frame .create_many(data.as_ref(), Internal) .map_err(Into::into) } /// Allocate several values of possibly different types, these types must implement /// [`IntoJulia`]. The values will be protected from garbage collection inside the frame used /// to create them. This takes as many slots on the GC stack as values that are allocated. /// /// Returns an error if there is not enough space on the stack. /// /// [`IntoJulia`]: ../traits/trait.IntoJulia.html pub fn new_dyn<'v, V, F>(frame: &mut F, data: V) -> JlrsResult<Self> where V: AsRef<[&'v dyn IntoJulia]>, F: Frame<'frame>, { frame .create_many_dyn(data.as_ref(), Internal) .map_err(Into::into) } } /// Except modules and symbols, all Julia data is represented as a `Value` in `jlrs`. /// /// A `Value` wraps around the pointer-value from the Julia C API and applies some restrictions /// through lifetimes to ensure it can only be used while it's protected from garbage collection /// and its contents are valid. /// /// The methods that create a new `Value` come in two varieties: `method` and `method_output`. The /// first will use a slot in the current frame to protect the value from garbage collection, while /// the latter uses a slot in an earlier frame. Other features offered by `Value` include /// accessing the fields of these values and (im)mutably borrowing their underlying array data. /// /// [`Value::new`]: struct.Value.html#method.new /// [`Module`]: ../module/struct.Module.html #[repr(transparent)] #[derive(Copy, Clone)] pub struct Value<'frame, 'data>( pub(crate) *mut jl_value_t, PhantomData<&'frame ()>, PhantomData<&'data ()>, ); impl<'frame, 'data> Value<'frame, 'data> { pub(crate) unsafe fn wrap(ptr: *mut jl_value_t) -> Value<'frame, 'static> { Value(ptr, PhantomData, PhantomData) } pub(crate) unsafe fn ptr(self) -> *mut jl_value_t { self.0 } /// Create a new Julia value, any type that implements [`IntoJulia`] can be converted using /// this function. The value will be protected from garbage collection inside the frame used /// to create it. One free slot on the garbage collection stack is required for this function /// to succeed, returns an error if no slot is available. /// /// [`IntoJulia`]: ../traits/trait.IntoJulia.html pub fn new<V, F>(frame: &mut F, value: V) -> JlrsResult<Value<'frame, 'static>> where V: IntoJulia, F: Frame<'frame>, { unsafe { frame .protect(value.into_julia(Internal), Internal) .map_err(Into::into) } } /// Create a new Julia value, any type that implements [`IntoJulia`] can be converted using /// this function. The value will be protected from garbage collection inside the frame used /// to create it. One free slot on the garbage collection stack is required for this function /// to succeed, returns an error if no slot is available. /// /// [`IntoJulia`]: ../traits/trait.IntoJulia.html pub fn new_output<'output, V, F>( frame: &mut F, output: Output<'output>, value: V, ) -> Value<'output, 'static> where V: IntoJulia, F: Frame<'frame>, { unsafe { frame.assign_output(output, value.into_julia(Internal), Internal) } } /// Returns true if the value is of type `T`. pub fn is<T: JuliaType>(&self) -> bool { unsafe { jl_typeis(self.ptr(), T::julia_type(Internal).cast()) } } /// Returns the type name of this value. pub fn type_name(&self) -> &str { unsafe { let type_name = jl_typeof_str(self.ptr()); let type_name_ref = CStr::from_ptr(type_name); type_name_ref.to_str().unwrap() } } /// Returns true if the value is an array. pub fn is_array(&self) -> bool { unsafe { jl_is_array(self.ptr()) } } /// Returns true if the value is a tuple. pub fn is_tuple(&self) -> bool { unsafe { jl_is_tuple(self.ptr()) } } /// Returns true if the value is an array with elements of type `T`. pub fn is_array_of<T: JuliaType>(&self) -> bool { unsafe { self.is_array() && jl_array_eltype(self.ptr()) as *mut jl_value_t == T::julia_type(Internal) } } /// Returns the field names of this value as a slice of `Symbol`s. These symbols can be used /// to access their fields with [`Value::get_field`]. /// /// [`Value::get_field`]: struct.Value.html#method.get_field pub fn field_names<'base>(&self, _: Global<'base>) -> &[Symbol<'base>] { unsafe { let tp = jl_typeof(self.ptr()); let field_names = jl_field_names(tp.cast()); let len = jl_svec_len(field_names); let items: *mut Symbol = jl_svec_data(field_names).cast(); slice::from_raw_parts(items.cast(), len) } } /// Returns the number of fields the underlying Julia value has. These fields can be accessed /// with [`Value::get_field_n`]. /// /// [`Value::get_field_n`]: struct.Value.html#method.get_field_n pub fn n_fields(&self) -> usize { unsafe { jl_nfields(self.ptr()) as _ } } /// Returns the field at index `idx` if it exists. If it does not exist /// `JlrsError::OutOfBounds` is returned. This function assumes the field must be protected /// from garbage collection, so calling this function will take a single slot on the GC stack. /// If there is no slot available `JlrsError::AllocError` is returned. pub fn get_nth_field<'fr, F>(&self, frame: &mut F, idx: usize) -> JlrsResult<Value<'fr, 'data>> where F: Frame<'fr>, { unsafe { if idx >= self.n_fields() { return Err(JlrsError::OutOfBounds(idx, self.n_fields()).into()); } frame .protect(jl_fieldref(self.ptr(), idx), Internal) .map_err(Into::into) } } /// Returns the field at index `idx` if it exists. If it does not exist /// `JlrsError::OutOfBounds` is returned. This function assumes the field must be protected /// from garbage collection and uses the provided output to do so. pub fn get_nth_field_output<'output, 'fr, F>( &self, frame: &mut F, output: Output<'output>, idx: usize, ) -> JlrsResult<Value<'output, 'data>> where F: Frame<'fr>, { unsafe { if idx >= self.n_fields() { return Err(JlrsError::OutOfBounds(idx, self.n_fields()).into()); } Ok(frame.assign_output(output, jl_fieldref(self.ptr(), idx), Internal)) } } /// Returns the field at index `idx` if it exists and no allocation is required to return it. /// If it does not exist `JlrsError::NoSuchField` is returned. If allocating is required to /// return the field, an `assert` will fail and the program will abort. pub fn get_nth_field_noalloc(&self, idx: usize) -> JlrsResult<Value<'frame, 'data>> { unsafe { if idx >= self.n_fields() { return Err(JlrsError::OutOfBounds(idx, self.n_fields()).into()); } Ok(Value::wrap(jl_fieldref_noalloc(self.ptr(), idx))) } } /// Returns the field with the name `field_name` if it exists. If it does not exist /// `JlrsError::NoSuchField` is returned. This function assumes the field must be protected /// from garbage collection, so calling this function will take a single slot on the GC stack. /// If there is no slot available `JlrsError::AllocError` is returned. pub fn get_field<'fr, N, F>(self, frame: &mut F, field_name: N) -> JlrsResult<Value<'fr, 'data>> where N: TemporarySymbol, F: Frame<'fr>, { unsafe { let symbol = field_name.temporary_symbol(Internal); let jl_type = jl_typeof(self.ptr()).cast(); let idx = jl_field_index(jl_type, symbol.ptr(), 0); if idx < 0 { return Err(JlrsError::NoSuchField(symbol.into()).into()); } frame .protect(jl_get_nth_field(self.ptr(), idx as _), Internal) .map_err(Into::into) } } /// Returns the field with the name `field_name` if it exists. If it does not exist /// `JlrsError::NoSuchField` is returned. This function assumes the field must be protected /// from garbage collection and uses the provided output to do so. pub fn get_field_output<'output, 'fr, N, F>( self, frame: &mut F, output: Output<'output>, field_name: N, ) -> JlrsResult<Value<'output, 'data>> where N: TemporarySymbol, F: Frame<'fr>, { unsafe { let symbol = field_name.temporary_symbol(Internal); let jl_type = jl_typeof(self.ptr()).cast(); let idx = jl_field_index(jl_type, symbol.ptr(), 0); if idx < 0 { return Err(JlrsError::NoSuchField(symbol.into()).into()); } Ok(frame.assign_output(output, jl_get_nth_field(self.ptr(), idx as _), Internal)) } } /// Returns the field with the name `field_name` if it exists and no allocation is required /// to return it. If it does not exist `JlrsError::NoSuchField` is returned. If allocating is /// required to return the field, an `assert` will fail and the program will abort. pub fn get_field_noalloc<N>(self, field_name: N) -> JlrsResult<Value<'frame, 'data>> where N: TemporarySymbol, { unsafe { let symbol = field_name.temporary_symbol(Internal); let jl_type = jl_typeof(self.ptr()).cast(); let idx = jl_field_index(jl_type, symbol.ptr(), 0); if idx < 0 { return Err(JlrsError::NoSuchField(symbol.into()).into()); } Ok(Value::wrap(jl_get_nth_field_noalloc(self.ptr(), idx as _))) } } /// If you call a function with one or more borrowed arrays as arguments, its result can only /// be used when all the borrows are active. If this result doesn't reference any borrowed /// data this function can be used to relax its second lifetime to `'static`. /// /// Safety: The value must not contain a reference any borrowed data. pub unsafe fn assume_owned(self) -> Value<'frame, 'static> { Value::wrap(self.ptr()) } /// Extend the `Value`'s lifetime to the `Output's lifetime. The original value will still be /// valid after calling this method, the data will be protected from garbage collection until /// the `Output`'s frame goes out of scope. pub fn extend<'output, F>(self, frame: &mut F, output: Output<'output>) -> Value<'output, 'data> where F: Frame<'frame>, { unsafe { frame.assign_output(output, self.ptr().cast(), Internal) } } /// Allocates a new n-dimensional array in Julia. /// /// Allocating an array with one, two, or three dimensions requires one slot on the GC stack. /// If you allocate an array with more dimensions, an extra frame is created with `n + 1` /// slots, temporarily taking `n + 3` additional slots. This latter case requires that /// `jlrs.jl` has been included. /// /// This function returns an error if there are not enough slots available, or if `jlrs.jl` /// has not been included when allocating arrays with four or more dimensions. pub fn array<T, D, F>(frame: &mut F, dimensions: D) -> JlrsResult<Value<'frame, 'static>> where T: JuliaType, D: Into<Dimensions>, F: Frame<'frame>, { unsafe { let array = array::<T, _, _>(frame, dimensions)?; frame.protect(array, Internal).map_err(Into::into) } } /// Allocates a new n-dimensional array in Julia using an `Output`. /// /// Because an `Output` is used, no additional slot in the current frame is used. If you /// allocate an array with more dimensions, an extra frame is created with `n + 1` slots, /// temporarily taking `n + 3` additional slots. This latter case requires that `jlrs.jl` has /// been included. /// /// This function returns an error if there are not enough slots available, or if `jlrs.jl` /// has not been included when allocating arrays with four or more dimensions. pub fn array_output<'output, T, D, F>( frame: &mut F, output: Output<'output>, dimensions: D, ) -> JlrsResult<Value<'output, 'static>> where T: JuliaType, D: Into<Dimensions>, F: Frame<'frame>, { unsafe { let array = array::<T, _, _>(frame, dimensions)?; Ok(frame.assign_output(output, array, Internal)) } } /// Borrows an n-dimensional array from Rust for use in Julia. /// /// Borrowing an array with one dimension requires one slot on the GC stack. If you borrow an /// array with more dimensions, an extra frame is created with `n + 1` slots, temporarily /// taking `n + 3` additional slots. This latter case requires that `jlrs.jl` has been /// included. /// /// This function returns an error if there are not enough slots available, or if `jlrs.jl` /// has not been included when borrowing arrays with two or more dimensions. /// /// This function is unsafe to call because you must ensure that the lifetime of this value is /// never extended through an `Output` by returning it from a Julia function, is never /// assigned to a global in Julia, and is never referenced from a value with a longer lifetime /// in Julia. pub fn borrow_array<T, D, V, F>( frame: &mut F, data: &'data mut V, dimensions: D, ) -> JlrsResult<Value<'frame, 'data>> where T: JuliaType, D: Into<Dimensions>, V: AsMut<[T]>, F: Frame<'frame>, { unsafe { let array = borrow_array(frame, data, dimensions)?; frame.protect(array, Internal).map_err(Into::into) } } /// Borrows an n-dimensional array from Rust for use in Julia using an `Output`. /// /// Because an `Output` is used, no additional slot in the current frame is used. If you /// borrow an array with more dimensions, an extra frame is created with `n + 1` slots, /// temporarily taking `n + 3` additional slots. This latter case requires that `jlrs.jl` has /// been included. /// /// This function returns an error if there are not enough slots available, or if `jlrs.jl` /// has not been included when borrowing arrays with two or more dimensions. /// /// This function is unsafe to call because you must ensure that the lifetime of this value is /// never extended through an `Output` by returning it from a Julia function, is never /// assigned to a global in Julia, and is never referenced from a value with a longer lifetime /// in Julia. pub fn borrow_array_output<'output, 'borrow, T, D, V, F>( frame: &mut F, output: Output<'output>, data: &'borrow mut V, dimensions: D, ) -> JlrsResult<Value<'output, 'borrow>> where 'borrow: 'output, T: JuliaType, D: Into<Dimensions>, V: AsMut<[T]>, F: Frame<'frame>, { unsafe { let array = borrow_array(frame, data, dimensions)?; Ok(frame.assign_output(output, array, Internal)) } } /// Moves an n-dimensional array from Rust to Julia using an `Output`. /// /// Because an `Output` is used, no additional slot in the current frame is used. If you move /// an array with more dimensions, an extra frame is created with `n + 1` slots, temporarily /// taking `n + 3` additional slots. This latter case requires that `jlrs.jl` has been /// included. /// /// This function returns an error if there are not enough slots available, or if `jlrs.jl` /// has not been included when moving arrays with two or more dimensions. pub fn move_array<T, D, F>( frame: &mut F, data: Vec<T>, dimensions: D, ) -> JlrsResult<Value<'frame, 'static>> where T: JuliaType, D: Into<Dimensions>, F: Frame<'frame>, { unsafe { let array = move_array(frame, data, dimensions)?; frame.protect(array, Internal).map_err(Into::into) } } /// Moves an n-dimensional array from Rust to Julia. /// /// Moving an array with one dimension requires one slot on the GC stack. If you move an /// array with more dimensions, an extra frame is created with `n + 1` slots, temporarily /// taking `n + 3` additional slots. This latter case requires that `jlrs.jl` has been /// included. /// /// This function returns an error if there are not enough slots available, or if `jlrs.jl` /// has not been included when moving arrays with two or more dimensions. pub fn move_array_output<'output, T, D, F>( frame: &mut F, output: Output<'output>, data: Vec<T>, dimensions: D, ) -> JlrsResult<Value<'output, 'static>> where T: JuliaType, D: Into<Dimensions>, F: Frame<'frame>, { unsafe { let array = move_array(frame, data, dimensions)?; Ok(frame.assign_output(output, array, Internal)) } } /// Immutably borrow array data, you can borrow data from multiple arrays at the same time. /// This data can only be borrowed if it contains floating point numbers or (unsigned) /// integers. Returns `JlrsError::NotAnArray` if this value is not an array or /// `JlrsError::WrongType` if the type of the elements is incorrect. pub fn array_data<'borrow, T: ArrayDatatype, F: Frame<'frame>>( &'borrow self, frame: &'borrow F, ) -> JlrsResult<ArrayData<'borrow, 'frame, T, F>> { if !self.is_array() { Err(JlrsError::NotAnArray)?; } if !self.is_array_of::<T>() { Err(JlrsError::WrongType)?; } unsafe { let ptr = self.ptr(); let jl_data = jl_array_data(ptr).cast(); let ptr = ptr.cast(); let n_dims = jl_array_ndims(ptr); let dimensions: Dimensions = match n_dims { 0 => return Err(JlrsError::ZeroDimension.into()), 1 => Into::into(jl_array_nrows(ptr) as usize), 2 => Into::into((jl_array_dim(ptr, 0), jl_array_dim(ptr, 1))), 3 => Into::into(( jl_array_dim(ptr, 0), jl_array_dim(ptr, 1), jl_array_dim(ptr, 2), )), ndims => Into::into(jl_array_dims(ptr, ndims as _)), }; // the lifetime is constrained to the lifetime of the borrow let data = slice::from_raw_parts(jl_data, dimensions.size()); Ok(ArrayData::new(data, dimensions, frame)) } } /// Mutably borrow array data, you can borrow data from a single array at the same time. /// This data can only be borrowed if it contains floating point numbers or (unsigned) /// integers. Returns `JlrsError::NotAnArray` if this value is not an array or /// `JlrsError::WrongType` if the type of the elements is incorrect. pub fn array_data_mut<'borrow, T: ArrayDatatype, F: Frame<'frame>>( &'borrow mut self, frame: &'borrow mut F, ) -> JlrsResult<ArrayDataMut<'borrow, 'frame, T, F>> { if !self.is_array_of::<T>() { Err(JlrsError::NotAnArray)?; } unsafe { let jl_data = jl_array_data(self.ptr()).cast(); let ptr = self.ptr().cast(); let n_dims = jl_array_ndims(ptr); let dimensions: Dimensions = match n_dims { 0 => return Err(JlrsError::ZeroDimension.into()), 1 => (jl_array_nrows(ptr) as usize).into(), 2 => (jl_array_dim(ptr, 0), jl_array_dim(ptr, 1)).into(), 3 => ( jl_array_dim(ptr, 0), jl_array_dim(ptr, 1), jl_array_dim(ptr, 2), ) .into(), ndims => jl_array_dims(ptr, ndims as _).into(), }; // the lifetime is constrained to the lifetime of the borrow let data = slice::from_raw_parts_mut(jl_data, dimensions.size()); Ok(ArrayDataMut::new(data, dimensions, frame)) } } /// Try to copy data from Julia to Rust. You can only copy data if the output type implements /// [`TryUnbox`]; this trait is implemented by all types that implement [`IntoJulia`] and /// arrays whose contents implement [`ArrayData`] through [`Array`]. Returns an error if the /// requested type does not match the actual type of the data. /// /// [`TryUnbox`]: ../traits/trait.TryUnbox.html /// [`IntoJulia`]: ../traits/trait.IntoJulia.html /// [`ArrayData`]: ../traits/trait.ArrayData.html /// [`Array`]: ../array/struct.Array.html pub fn try_unbox<T>(self) -> JlrsResult<T> where T: TryUnbox, { unsafe { T::try_unbox(self.ptr(), Internal) } } /// Call this value as a function that takes zero arguments, this takes one slot on the GC /// stack. Returns the result of this function call if no exception is thrown, the exception /// if one is, or an error if no space is left on the stack. pub fn call0<F>(self, frame: &mut F) -> JlrsResult<CallResult<'frame, 'static>> where F: Frame<'frame>, { unsafe { let res = jl_call0(self.ptr()); try_protect(frame, res) } } /// Call this value as a function that takes zero arguments and use the `Output` to extend the /// result's lifetime. This takes no space on the GC stack. Returns the result of this /// function call if no exception is thrown or the exception if one is. pub fn call0_output<'output, F>( self, frame: &mut F, output: Output<'output>, ) -> JlrsResult<CallResult<'output, 'static>> where F: Frame<'frame>, { unsafe { let res = jl_call0(self.ptr()); assign(frame, output, res) } } /// Call this value as a function that takes one argument, this takes one slot on the GC /// stack. Returns the result of this function call if no exception is thrown, the exception /// if one is, or an error if no space is left on the stack. pub fn call1<'borrow, F>( self, frame: &mut F, arg: Value<'_, 'borrow>, ) -> JlrsResult<CallResult<'frame, 'borrow>> where F: Frame<'frame>, { unsafe { let res = jl_call1(self.ptr().cast(), arg.ptr()); try_protect(frame, res) } } /// Call this value as a function that takes one argument and use the `Output` to extend the /// result's lifetime. This takes no space on the GC stack. Returns the result of this /// function call if no exception is thrown or the exception if one is. pub fn call1_output<'output, 'borrow, F>( self, frame: &mut F, output: Output<'output>, arg: Value<'_, 'borrow>, ) -> JlrsResult<CallResult<'output, 'borrow>> where 'borrow: 'output, F: Frame<'frame>, { unsafe { let res = jl_call1(self.ptr().cast(), arg.ptr()); assign(frame, output, res) } } /// Call this value as a function that takes two arguments, this takes one slot on the GC /// stack. Returns the result of this function call if no exception is thrown, the exception /// if one is, or an error if no space is left on the stack. pub fn call2<'borrow, F>( self, frame: &mut F, arg0: Value<'_, 'borrow>, arg1: Value<'_, 'borrow>, ) -> JlrsResult<CallResult<'frame, 'borrow>> where F: Frame<'frame>, { unsafe { let res = jl_call2(self.ptr().cast(), arg0.ptr(), arg1.ptr()); try_protect(frame, res) } } /// Call this value as a function that takes two arguments and use the `Output` to extend the /// result's lifetime. This takes no space on the GC stack. Returns the result of this /// function call if no exception is thrown or the exception if one is. pub fn call2_output<'output, 'borrow, F>( self, frame: &mut F, output: Output<'output>, arg0: Value<'_, 'borrow>, arg1: Value<'_, 'borrow>, ) -> JlrsResult<CallResult<'output, 'borrow>> where 'borrow: 'output, F: Frame<'frame>, { unsafe { let res = jl_call2(self.ptr().cast(), arg0.ptr(), arg1.ptr()); assign(frame, output, res) } } /// Call this value as a function that takes three arguments, this takes one slot on the GC /// stack. Returns the result of this function call if no exception is thrown, the exception /// if one is, or an error if no space is left on the stack. pub fn call3<'borrow, F>( self, frame: &mut F, arg0: Value<'_, 'borrow>, arg1: Value<'_, 'borrow>, arg2: Value<'_, 'borrow>, ) -> JlrsResult<CallResult<'frame, 'borrow>> where F: Frame<'frame>, { unsafe { let res = jl_call3(self.ptr().cast(), arg0.ptr(), arg1.ptr(), arg2.ptr()); try_protect(frame, res) } } /// Call this value as a function that takes three arguments and use the `Output` to extend /// the result's lifetime. This takes no space on the GC stack. Returns the result of this /// function call if no exception is thrown or the exception if one is. pub fn call3_output<'output, 'borrow, F>( self, frame: &mut F, output: Output<'output>, arg0: Value<'_, 'borrow>, arg1: Value<'_, 'borrow>, arg2: Value<'_, 'borrow>, ) -> JlrsResult<CallResult<'output, 'borrow>> where 'borrow: 'output, F: Frame<'frame>, { unsafe { let res = jl_call3(self.ptr().cast(), arg0.ptr(), arg1.ptr(), arg2.ptr()); assign(frame, output, res) } } /// Call this value as a function that takes several arguments, this takes one slot on the GC /// stack. Returns the result of this function call if no exception is thrown, the exception /// if one is, or an error if no space is left on the stack. pub fn call<'value, 'borrow, V, F>( self, frame: &mut F, mut args: V, ) -> JlrsResult<CallResult<'frame, 'borrow>> where V: AsMut<[Value<'value, 'borrow>]>, F: Frame<'frame>, { unsafe { let args = args.as_mut(); let n = args.len(); let res = jl_call(self.ptr().cast(), args.as_mut_ptr().cast(), n as _); try_protect(frame, res) } } /// Call this value as a function that takes several arguments and use the `Output` to extend /// the result's lifetime. This takes no space on the GC stack. Returns the result of this /// function call if no exception is thrown or the exception if one is. pub fn call_output<'output, 'value, 'borrow, V, F>( self, frame: &mut F, output: Output<'output>, mut args: V, ) -> JlrsResult<CallResult<'output, 'borrow>> where 'borrow: 'output, V: AsMut<[Value<'value, 'borrow>]>, F: Frame<'frame>, { unsafe { let args = args.as_mut(); let n = args.len(); let res = jl_call(self.ptr().cast(), args.as_mut_ptr().cast(), n as _); assign(frame, output, res) } } /// Call this value as a function that takes several arguments in a single `Values`, this /// takes one slot on the GC stack. Returns the result of this function call if no exception /// is thrown, the exception if one is, or an error if no space is left on the stack. pub fn call_values<F>( self, frame: &mut F, args: Values, ) -> JlrsResult<CallResult<'frame, 'static>> where F: Frame<'frame>, { unsafe { let res = jl_call(self.ptr().cast(), args.ptr(), args.len() as _); try_protect(frame, res) } } /// Call this value as a function that takes several arguments in a single `Values` and use /// the `Output` to extend the result's lifetime. This takes no space on the GC stack. Returns /// the result of this function call if no exception is thrown or the exception if one is. pub fn call_values_output<'output, F>( self, frame: &mut F, output: Output<'output>, args: Values, ) -> JlrsResult<CallResult<'output, 'static>> where F: Frame<'frame>, { unsafe { let res = jl_call(self.ptr().cast(), args.ptr(), args.len() as _); assign(frame, output, res) } } /// Returns an anonymous function that wraps this value in a try-catch block. Calling this /// anonymous function with some arguments will call the value as a function with those /// arguments and return its result, or catch the exception, print the stackstrace, and /// rethrow that exception. This takes one slot on the GC stack. You must include `jlrs.jl` to /// use this function. pub fn tracing_call<F>(self, frame: &mut F) -> JlrsResult<CallResult<'frame, 'data>> where F: Frame<'frame>, { unsafe { let global = Global::new(); let func = Module::main(global) .submodule("Jlrs")? .function("tracingcall")?; let res = jl_call1(func.ptr(), self.ptr()); try_protect(frame, res) } } /// Returns an anonymous function that wraps this value in a try-catch block. Calling this /// anonymous function with some arguments will call the value as a function with those /// arguments and return its result, or catch the exception and throw a new one with two /// fields, `exc` and `stacktrace`, containing the original exception and the stacktrace /// respectively. This takes one slot on the GC stack. You must include `jlrs.jl` to use this /// function. pub fn attach_stacktrace<F>(self, frame: &mut F) -> JlrsResult<CallResult<'frame, 'data>> where F: Frame<'frame>, { unsafe { let global = Global::new(); let func = Module::main(global) .submodule("Jlrs")? .function("attachstacktrace")?; let res = jl_call1(func.ptr(), self.ptr()); try_protect(frame, res) } } } impl<'frame, 'data> Debug for Value<'frame, 'data> { fn fmt(&self, f: &mut Formatter<'_>) -> FmtResult { f.debug_tuple("Value").field(&self.type_name()).finish() } } unsafe fn array<'frame, T, D, F>(frame: &mut F, dimensions: D) -> JlrsResult<*mut jl_value_t> where T: JuliaType, D: Into<Dimensions>, F: Frame<'frame>, { let dims = dimensions.into(); let array_type = jl_apply_array_type(T::julia_type(Internal), dims.n_dimensions()); match dims.n_dimensions() { 1 => Ok(jl_alloc_array_1d(array_type, dims.n_elements(0)).cast()), 2 => Ok(jl_alloc_array_2d(array_type, dims.n_elements(0), dims.n_elements(1)).cast()), 3 => Ok(jl_alloc_array_3d( array_type, dims.n_elements(0), dims.n_elements(1), dims.n_elements(2), ) .cast()), n => frame.frame(n as usize + 1, |frame| { let global = Global::new(); let jlrs = Symbol::new(global, "Jlrs"); let arraydims = Symbol::new(global, "arraydims"); let func = Module::main(global).submodule(jlrs)?.function(arraydims)?; let v = Values::new(frame, dims.as_slice())?; let dims = func.call_values(frame, v)?.unwrap(); Ok(jl_new_array(array_type, dims.ptr()).cast()) }), } } unsafe fn borrow_array<'data, 'frame, T, D, V, F>( frame: &mut F, data: &'data mut V, dimensions: D, ) -> JlrsResult<*mut jl_value_t> where T: JuliaType, D: Into<Dimensions>, V: AsMut<[T]>, F: Frame<'frame>, { let dims = dimensions.into(); let array_type = jl_apply_array_type(T::julia_type(Internal), dims.n_dimensions()); match dims.n_dimensions() { 1 => Ok(jl_ptr_to_array_1d( array_type, data.as_mut().as_mut_ptr().cast(), dims.n_elements(0), 0, ) .cast()), n => frame.frame(n as usize + 1, |frame| { let global = Global::new(); let jlrs = Symbol::new(global, "Jlrs"); let arraydims = Symbol::new(global, "arraydims"); let func = Module::main(global).submodule(jlrs)?.function(arraydims)?; let v = Values::new(frame, dims.as_slice())?; let dims = func.call_values(frame, v)?.unwrap(); Ok( jl_ptr_to_array(array_type, data.as_mut().as_mut_ptr().cast(), dims.ptr(), 0) .cast(), ) }), } } unsafe fn move_array<'frame, T, D, F>( frame: &mut F, data: Vec<T>, dimensions: D, ) -> JlrsResult<*mut jl_value_t> where T: JuliaType, D: Into<Dimensions>, F: Frame<'frame>, { let dims = dimensions.into(); let array_type = jl_apply_array_type(T::julia_type(Internal), dims.n_dimensions()); match dims.n_dimensions() { 1 => Ok(jl_ptr_to_array_1d( array_type, Box::into_raw(data.into_boxed_slice()).cast(), dims.n_elements(0), 1, ) .cast()), n => frame.frame(n as usize + 1, |frame| { let global = Global::new(); let jlrs = Symbol::new(global, "Jlrs"); let arraydims = Symbol::new(global, "arraydims"); let func = Module::main(global).submodule(jlrs)?.function(arraydims)?; let v = Values::new(frame, dims.as_slice())?; let dims = func.call_values(frame, v)?.unwrap(); Ok(jl_ptr_to_array( array_type, Box::into_raw(data.into_boxed_slice()).cast(), dims.ptr(), 1, ) .cast()) }), } } unsafe fn try_protect<'frame, F>( frame: &mut F, res: *mut jl_value_t, ) -> JlrsResult<CallResult<'frame, 'static>> where F: Frame<'frame>, { let exc = jl_sys::jl_exception_occurred(); if !exc.is_null() { match frame.protect(exc, Internal) { Ok(exc) => Ok(Err(exc)), Err(a) => Err(a.into()), } } else { match frame.protect(res, Internal) { Ok(v) => Ok(Ok(v)), Err(a) => Err(a.into()), } } } unsafe fn assign<'output, 'frame, F>( frame: &mut F, output: Output<'output>, res: *mut jl_value_t, ) -> JlrsResult<CallResult<'output, 'static>> where F: Frame<'frame>, { let exc = jl_exception_occurred(); if !exc.is_null() { Ok(Err(frame.assign_output(output, exc, Internal))) } else { Ok(Ok(frame.assign_output(output, res, Internal))) } }