Function signature_size 

pub fn signature_size(dim: usize, max_depth: usize) -> usize

Computes the number of signature components for given dim and depth.

Total terms = Σ_{k=1}^{depth} dim^k = dim * (dim^depth - 1) / (dim - 1) For dim=1, it’s just depth.