AllocValue for more int types

Reviewed By: bobyangyf

Differential Revision: D38712858

fbshipit-source-id: 72e823078530ebcdb9cfc97157ebab311c23bff9

Author: stepancheg

starlark/src/values/types/bigint/convert.rs

@@ -24,6 +24,30 @@ use crate::values::FrozenValue;
 use crate::values::Heap;
 use crate::values::Value;
 
+impl StarlarkTypeRepr for u32 {
+    fn starlark_type_repr() -> String {
+        i32::starlark_type_repr()
+    }
+}
+
+impl<'v> AllocValue<'v> for u32 {
+    fn alloc_value(self, heap: &'v Heap) -> Value<'v> {
+        match i32::try_from(self) {
+            Ok(x) => Value::new_int(x),
+            Err(_) => StarlarkBigInt::alloc_bigint(self.into(), heap),
+        }
+    }
+}
+
+impl AllocFrozenValue for u32 {
+    fn alloc_frozen_value(self, heap: &FrozenHeap) -> FrozenValue {
+        match i32::try_from(self) {
+            Ok(x) => FrozenValue::new_int(x),
+            Err(_) => StarlarkBigInt::alloc_bigint_frozen(self.into(), heap),
+        }
+    }
+}
+
 impl StarlarkTypeRepr for u64 {
     fn starlark_type_repr() -> String {
         i32::starlark_type_repr()
@@ -71,3 +95,51 @@ impl AllocFrozenValue for i64 {
         }
     }
 }
+
+impl StarlarkTypeRepr for usize {
+    fn starlark_type_repr() -> String {
+        i32::starlark_type_repr()
+    }
+}
+
+impl<'v> AllocValue<'v> for usize {
+    fn alloc_value(self, heap: &'v Heap) -> Value<'v> {
+        match i32::try_from(self) {
+            Ok(x) => Value::new_int(x),
+            Err(_) => StarlarkBigInt::alloc_bigint(self.into(), heap),
+        }
+    }
+}
+
+impl AllocFrozenValue for usize {
+    fn alloc_frozen_value(self, heap: &FrozenHeap) -> FrozenValue {
+        match i32::try_from(self) {
+            Ok(x) => FrozenValue::new_int(x),
+            Err(_) => StarlarkBigInt::alloc_bigint_frozen(self.into(), heap),
+        }
+    }
+}
+
+impl StarlarkTypeRepr for isize {
+    fn starlark_type_repr() -> String {
+        i32::starlark_type_repr()
+    }
+}
+
+impl<'v> AllocValue<'v> for isize {
+    fn alloc_value(self, heap: &'v Heap) -> Value<'v> {
+        match i32::try_from(self) {
+            Ok(x) => Value::new_int(x),
+            Err(_) => StarlarkBigInt::alloc_bigint(self.into(), heap),
+        }
+    }
+}
+
+impl AllocFrozenValue for isize {
+    fn alloc_frozen_value(self, heap: &FrozenHeap) -> FrozenValue {
+        match i32::try_from(self) {
+            Ok(x) => FrozenValue::new_int(x),
+            Err(_) => StarlarkBigInt::alloc_bigint_frozen(self.into(), heap),
+        }
+    }
+}