TryFrom< B512 > for b256, u256

sway-lib-std/src/lib.sw

@@ -1,5 +1,6 @@
 library;
 
+pub mod constants;
 pub mod error_signals;
 pub mod logging;
 pub mod revert;
@@ -16,17 +17,16 @@ pub mod bytes;
 pub mod math;
 pub mod flags;
 pub mod u128;
+pub mod b512;
 pub mod primitive_conversions;
 pub mod alias;
 pub mod hash;
 pub mod asset_id;
 pub mod contract_id;
 pub mod execution;
-pub mod constants;
 pub mod call_frames;
 pub mod context;
 pub mod external;
-pub mod b512;
 pub mod tx;
 pub mod outputs;
 pub mod address;

sway-lib-std/src/primitive_conversions/b256.sw

@@ -1,9 +1,11 @@
 library;
 
 use ::bytes::Bytes;
+use ::constants::ZERO_B256;
 use ::convert::{From, TryFrom};
 use ::option::Option::{self, *};
 use ::u128::U128;
+use ::b512::B512;
 
 impl TryFrom<Bytes> for b256 {
     fn try_from(b: Bytes) -> Option<Self> {
@@ -18,6 +20,41 @@ impl TryFrom<Bytes> for b256 {
     }
 }
 
+impl TryFrom<B512> for b256 {
+    /// Attempts conversion from a `B512` to a `b256`.
+    ///
+    /// # Additional Information
+    ///
+    /// If the high bits of the `B512` are not zero, the conversion will fail.
+    ///
+    /// # Arguments
+    ///
+    /// * `val`: [B512] - The `B512` to be converted.
+    ///
+    /// # Returns
+    ///
+    /// * [Option<b256>] - The `b256` representation of the `B512` value.
+    ///
+    /// # Examples
+    ///
+    /// ```sway
+    /// use std::b512::B512;
+    ///
+    /// fn foo() {
+    ///     let b512_value = B512::new();
+    ///     let b256_value = b256::try_from(b512_value).unwrap();
+    /// }
+    /// ```
+    fn try_from(val: B512) -> Option<Self> {
+        let bits = val.bits();
+        if bits[0] == ZERO_B256 {
+            Some(bits[1])
+        } else {
+            None
+        }
+    }
+}
+
 impl From<u256> for b256 {
     /// Casts a `u256` to raw `b256` data.
     ///
@@ -152,3 +189,16 @@ fn test_b256_from_tuple() {
         b256_value == 0x0000000000000001000000000000000200000000000000030000000000000004,
     );
 }
+
+#[test]
+fn test_b256_try_from_b512() {
+    use ::assert::assert;
+
+    let b512_value = B512::new();
+    let b256_value = b256::try_from(b512_value);
+    assert(b256_value.is_some());
+
+    let b512_value = B512::from((0x0000000000000000000000000000000000000000000000000000000000000001, ZERO_B256));
+    let b256_value = b256::try_from(b512_value);
+    assert(b256_value.is_none());
+}

sway-lib-std/src/primitive_conversions/u256.sw

@@ -1,7 +1,45 @@
 library;
 
-use ::convert::From;
+use ::constants::ZERO_B256;
+use ::convert::{From, TryFrom};
+use ::option::Option::{self, *};
 use ::u128::U128;
+use ::b512::B512;
+
+impl TryFrom<B512> for u256 {
+    /// Attempts conversion from a `B512` to a `u256`.
+    ///
+    /// # Additional Information
+    ///
+    /// If the high bits of the `B512` are not zero, the conversion will fail.
+    ///
+    /// # Arguments
+    ///
+    /// * `val`: [B512] - The `B512` to be converted.
+    ///
+    /// # Returns
+    ///
+    /// * [Option<u256>] - The `u256` representation of the `B512` value.
+    ///
+    /// # Examples
+    ///
+    /// ```sway
+    /// use std::b512::B512;
+    ///
+    /// fn foo() {
+    ///     let b512_value = B512::new();
+    ///     let u256_value = u256::try_from(b512_value).unwrap();
+    /// }
+    /// ```
+    fn try_from(val: B512) -> Option<Self> {
+        let bits = val.bits();
+        if bits[0] == ZERO_B256 {
+            Some(bits[1].as_u256())
+        } else {
+            None
+        }
+    }
+}
 
 /// Functions for casting between `u256` and other types.
 impl From<u8> for u256 {
@@ -255,3 +293,16 @@ fn test_u256_from_tuple() {
         u256_value == 0x0000000000000001000000000000000200000000000000030000000000000004_u256,
     );
 }
+
+#[test]
+fn test_u256_try_from_b512() {
+    use ::assert::assert;
+
+    let b512_value = B512::new();
+    let u256_value = u256::try_from(b512_value);
+    assert(u256_value.is_some());
+
+    let b512_value = B512::from((0x0000000000000000000000000000000000000000000000000000000000000001, ZERO_B256));
+    let u256_value = u256::try_from(b512_value);
+    assert(u256_value.is_none());
+}

test/src/e2e_vm_tests/reduced_std_libs/sway-lib-std-conversions/reduced_lib.config

@@ -16,6 +16,7 @@ registers.sw
 flags.sw
 math.sw
 u128.sw
+b512.sw
 bytes_conversions.sw
 bytes_conversions/b256.sw
 bytes_conversions/u16.sw

test/src/e2e_vm_tests/reduced_std_libs/sway-lib-std-conversions/src/lib.sw

@@ -1,5 +1,6 @@
 library;
 
+pub mod constants;
 pub mod error_signals;
 pub mod logging;
 pub mod revert;
@@ -16,6 +17,7 @@ pub mod registers;
 pub mod flags;
 pub mod math;
 pub mod u128;
+pub mod b512;
 pub mod primitive_conversions;
 pub mod array_conversions;
 pub mod bytes_conversions;