Use tilelib to load for export and pca.

[lightning.git] / tilelib.go

diff --git a/tilelib.go b/tilelib.go
index 5c0ebdfd4688b4cbc5bfb3ad546a7d2dfab8b3c8..31fbb314a5eabf79187c90cdf800c28c07a41d11 100644 (file)
--- a/tilelib.go
+++ b/tilelib.go
@@ -53,6 +53,7 @@ type tileLibrary struct {
        taglib         *tagLibrary
        variant        [][][blake2b.Size256]byte
        refseqs        map[string]map[string][]tileLibRef
+       compactGenomes map[string][]tileVariantID
        // count [][]int
        // seq map[[blake2b.Size]byte][]byte
        variants int
@@ -151,6 +152,11 @@ func (tilelib *tileLibrary) loadCompactGenomes(cgs []CompactGenome, variantmap m
                                        return
                                }
                        }
+                       if tilelib.compactGenomes != nil {
+                               tilelib.mtx.Lock()
+                               defer tilelib.mtx.Unlock()
+                               tilelib.compactGenomes[cg.Name] = cg.Variants
+                       }
                }()
        }
        wg.Wait()