doc: book sections on metadata

Author: molpopgen

book/src/SUMMARY.md

@@ -15,6 +15,12 @@
   - [Working with trees](./tree_sequence_tree.md)
   - [Miscellaneous operations](./tree_sequence_miscellaneous.md)
 
+* [Metadata](./metadata.md)
+  - [Defining metadata types in rust](./metadata_derive.md)
+  - [Metadata and tables](./metadata_tables.md)
+  - [Metadata schema](./metadata_schema.md)
+  
+
 [Crate prelude](./prelude.md)
 [Changelog](./changelog.md)
 [Migration Guide](./migration_guide.md)

book/src/metadata.md

@@ -0,0 +1 @@
+# Metadata <img align="right" width="73" height="45" src="https://raw.githubusercontent.com/tskit-dev/administrative/main/logos/svg/tskit-rust/Tskit_rust_logo.eps.svg">

book/src/metadata_derive.md

@@ -0,0 +1,12 @@
+# Defining metadata types in rust
+
+Using the `tskit` cargo feature `derive`, we can use procedural macros to define metadata types.
+Here, we define a metadata type for a mutation table:
+
+```rust, noplayground, ignore
+{{#include ../../tests/book_metadata.rs:metadata_derive}}
+```
+
+We require that you also manually specify the `serde` derive macros because the metadata API
+itself does not depend on `serde`.
+Rather, it expects raw bytes and `serde` happens to be a good way to get them from your data types.

book/src/metadata_schema.md

@@ -0,0 +1,17 @@
+# Metadata schema
+
+For useful data interchange with `tskit-python`, we need to define [metadata schema](https://tskit.dev/tskit/docs/stable/metadata.html).
+
+There are currently several points slowing down a rust API for schema:
+
+* It is not clear which `serde` formats are compatible with metadata on the Python side.
+* Experiments have shown that `serde_json` works with `tskit-python`.
+  * Ideally, we would also like a binary format compatible with the Python `struct`
+    module.
+* However, we have not found a solution eliminating the need to manually write the
+  schema as a string and add it to the tables.
+  Various crates to generate JSON schema from rust structs return schema that are over-specified
+  and fail to validate in `tskit-python`.
+* We also have the problem that we will need to add some Python to our CI to prove to ourselves
+  that some reasonable tests can pass.
+

book/src/metadata_tables.md

@@ -0,0 +1,36 @@
+# Metadata and tables
+
+Let us create a table and add a row with our mutation metadata:
+
+```rust, noplayground, ignore
+{{#include ../../tests/book_metadata.rs:add_mutation_table_row_with_metadata}}
+```
+
+Meta data is optional on a per-row basis:
+
+```rust, noplayground, ignore
+{{#include ../../tests/book_metadata.rs:add_mutation_table_row_without_metadata}}
+```
+
+We can confirm that we have one row with, and one without, metadata:
+
+```rust, noplayground, ignore
+{{#include ../../tests/book_metadata.rs:validate_metadata_row_contents}}
+```
+
+Fetching our metadata from the table requires specifying the metadata type.
+The result of a metadata retrieval is `Option<Result, TskitError>`.
+The `None` variant occurs if a row does not have metadata or if a row id does not exist.
+The error state occurs if decoding raw bytes into the metadata type fails.
+The details of the error variant are [here](https://docs.rs/tskit/latest/tskit/error/enum.TskitError.html#variant.MetadataError).
+The reason why the error type holds `Box<dyn Error>` is that the API is very general.
+We assume nothing about the API used to encode/decode metadata.
+Therefore, the error could be anything.
+
+```rust, noplayground, ignore
+{{#include ../../tests/book_metadata.rs:metadata_retrieval}}
+```
+
+```rust, noplayground, ignore
+{{#include ../../tests/book_metadata.rs:metadata_retrieval_none}}
+```

tests/book_metadata.rs

@@ -0,0 +1,89 @@
+#[cfg(feature = "derive")]
+#[test]
+fn book_mutation_metadata() {
+    // ANCHOR: metadata_derive
+    #[derive(serde::Serialize, serde::Deserialize, tskit::metadata::MutationMetadata)]
+    #[serializer("serde_json")]
+    struct MutationMetadata {
+        effect_size: f64,
+        dominance: f64,
+    }
+    // ANCHOR_END: metadata_derive
+
+    // ANCHOR: add_mutation_table_row_with_metadata
+    let mut tables = tskit::TableCollection::new(50.0).unwrap();
+
+    let md = MutationMetadata {
+        effect_size: 1e-3,
+        dominance: 1.0,
+    };
+
+    tables
+        .add_mutation_with_metadata(
+            0,    // site id
+            0,    // node id
+            -1,   // mutation parent id
+            0.0,  // time
+            None, // derived state is Option<&[u8]>
+            &md,  // metadata for this row
+        )
+        .unwrap();
+    // ANCHOR_END: add_mutation_table_row_with_metadata
+
+    // ANCHOR: add_mutation_table_row_without_metadata
+    tables
+        .add_mutation(
+            0,    // site id
+            0,    // node id
+            -1,   // mutation parent id
+            0.0,  // time
+            None, // derived state is Option<&[u8]>
+        )
+        .unwrap();
+    // ANCHOR_END: add_mutation_table_row_without_metadata
+
+    // ANCHOR: validate_metadata_row_contents
+    assert_eq!(
+        tables
+            .mutations_iter()
+            .filter(|m| m.metadata.is_some())
+            .count(),
+        1
+    );
+    assert_eq!(
+        tables
+            .mutations_iter()
+            .filter(|m| m.metadata.is_none())
+            .count(),
+        1
+    );
+    // ANCHOR_END: validate_metadata_row_contents
+
+    // ANCHOR: metadata_retrieval
+    let fetched_md = match tables.mutations().metadata::<MutationMetadata>(0.into()) {
+        Some(Ok(m)) => m,
+        Some(Err(e)) => panic!("metadata decoding failed: {:?}", e),
+        None => panic!("hmmm...row 0 should have been a valid row with metadata..."),
+    };
+
+    assert_eq!(md.effect_size, fetched_md.effect_size);
+    assert_eq!(md.dominance, fetched_md.dominance);
+    // ANCHOR_END: metadata_retrieval
+
+    // ANCHOR: metadata_retrieval_none
+    // There is no metadata at row 1, so
+    // you get None back
+    assert!(tables
+        .mutations()
+        .metadata::<MutationMetadata>(1.into())
+        .is_none());
+
+    // There is also no metadata at row 2,
+    // because that row does not exist, so
+    // you get None back
+    assert!(tables
+        .mutations()
+        .metadata::<MutationMetadata>(2.into())
+        .is_none());
+    // ANCHOR_END: metadata_retrieval_none
+}