Add filter options to export cmd.

[lightning.git] / export.go

diff --git a/export.go b/export.go
index 6bc1007b235cadffdcdda86ef7d5d0c65bc90468..9546926c015fe98d02357b17fb4f27e54e2a1951 100644 (file)
--- a/export.go
+++ b/export.go
@@ -60,6 +60,7 @@ type exporter struct {
        outputPerChrom bool
        compress       bool
        maxTileSize    int
+       filter         filter
 }
 
 func (cmd *exporter) RunCommand(prog string, args []string, stdin io.Reader, stdout, stderr io.Writer) int {
@@ -85,6 +86,7 @@ func (cmd *exporter) RunCommand(prog string, args []string, stdin io.Reader, std
        flags.BoolVar(&cmd.compress, "z", false, "write gzip-compressed output files")
        labelsFilename := flags.String("output-labels", "", "also output genome labels csv `file`")
        flags.IntVar(&cmd.maxTileSize, "max-tile-size", 50000, "don't try to make annotations for tiles bigger than given `size`")
+       cmd.filter.Flags(flags)
        err = flags.Parse(args)
        if err == flag.ErrHelp {
                err = nil
@@ -151,6 +153,7 @@ func (cmd *exporter) RunCommand(prog string, args []string, stdin io.Reader, std
                        "-output-dir", "/mnt/output",
                        "-z=" + fmt.Sprintf("%v", cmd.compress),
                }
+               runner.Args = append(runner.Args, cmd.filter.Args()...)
                var output string
                output, err = runner.Run()
                if err != nil {
@@ -182,6 +185,9 @@ func (cmd *exporter) RunCommand(prog string, args []string, stdin io.Reader, std
                return 1
        }
 
+       log.Infof("filtering: %+v", cmd.filter)
+       cmd.filter.Apply(tilelib)
+
        names := cgnames(tilelib)
        for _, name := range names {
                cgs = append(cgs, CompactGenome{Name: name, Variants: tilelib.compactGenomes[name]})