py/objstr: Add a helper to set mp_obj_str_t data.

jimmo · dpgeorge · commit c44b3927b8f4 · 2022-10-11T17:50:19.000+11:00
Signed-off-by: Jim Mussared &lt;jim.mussared@gmail.com&gt;
diff --git a/py/objstr.c b/py/objstr.c
@@ -2045,6 +2045,12 @@ mp_int_t mp_obj_str_get_buffer(mp_obj_t self_in, mp_buffer_info_t *bufinfo, mp_u
     }
 }
 
+void mp_obj_str_set_data(mp_obj_str_t *str, const byte *data, size_t len) {
+    str->data = data;
+    str->len = len;
+    str->hash = qstr_compute_hash(data, len);
+}
+
 // This locals table is used for the following types: str, bytes, bytearray, array.array.
 // Each type takes a different section (start to end offset) of this table.
 STATIC const mp_rom_map_elem_t array_bytearray_str_bytes_locals_table[] = {
diff --git a/py/objstr.h b/py/objstr.h
@@ -94,6 +94,8 @@ mp_obj_t mp_obj_new_str_of_type(const mp_obj_type_t *type, const byte *data, siz
 mp_obj_t mp_obj_str_binary_op(mp_binary_op_t op, mp_obj_t lhs_in, mp_obj_t rhs_in);
 mp_int_t mp_obj_str_get_buffer(mp_obj_t self_in, mp_buffer_info_t *bufinfo, mp_uint_t flags);
 
+void mp_obj_str_set_data(mp_obj_str_t *str, const byte *data, size_t len);
+
 const byte *str_index_to_ptr(const mp_obj_type_t *type, const byte *self_data, size_t self_len,
     mp_obj_t index, bool is_slice);
 const byte *find_subbytes(const byte *haystack, size_t hlen, const byte *needle, size_t nlen, int direction);

Original file line number	Diff line number	Diff line change
`@@ -2045,6 +2045,12 @@ mp_int_t mp_obj_str_get_buffer(mp_obj_t self_in, mp_buffer_info_t *bufinfo, mp_u`
`2045`	`2045`	`}`
`2046`	`2046`	`}`
`2047`	`2047`
	`2048`	`+void mp_obj_str_set_data(mp_obj_str_t str, const byte data, size_t len) {`
	`2049`	`+ str->data = data;`
	`2050`	`+ str->len = len;`
	`2051`	`+ str->hash = qstr_compute_hash(data, len);`
	`2052`	`+}`
	`2053`	`+`
`2048`	`2054`	`// This locals table is used for the following types: str, bytes, bytearray, array.array.`
`2049`	`2055`	`// Each type takes a different section (start to end offset) of this table.`
`2050`	`2056`	`STATIC const mp_rom_map_elem_t array_bytearray_str_bytes_locals_table[] = {`