obitools/obitools4
2
0
Fork 0
mirror of https://github.com/metabarcoding/obitools4.git synced 2025-06-29 16:20:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make the --no-progressbar option effective on graph building progress too.

Browse Source 
Former-commit-id: 3fabecfc9de69ad406d6840c8f821c16eebaef46
This commit is contained in:
Eric Coissac
2024-04-22 12:01:43 +02:00
parent df74b2b2bd
commit 8f08959ebb
1 changed files with 36 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
pkg/obitools/obiclean/obiclean.go
View File

@ -7,6 +7,7 @@ import (
"git.metabarcoding.org/obitools/obitools4/obitools4/pkg/obiiter"
"git.metabarcoding.org/obitools/obitools4/obitools4/pkg/obioptions"
"git.metabarcoding.org/obitools/obitools4/obitools4/pkg/obiseq"
"git.metabarcoding.org/obitools/obitools4/obitools4/pkg/obitools/obiconvert"
"git.metabarcoding.org/obitools/obitools4/obitools4/pkg/obiutils"
"github.com/schollz/progressbar/v3"
log "github.com/sirupsen/logrus"
@ -300,6 +301,10 @@ func CLIOBIClean(itertator obiiter.IBioSequence) obiiter.IBioSequence {
obioptions.CLIParallelWorkers())
if RatioMax() < 1.0 {
bar := (*progressbar.ProgressBar)(nil)
if obiconvert.CLIProgressBar() {
pbopt := make([]progressbar.Option, 0, 5)
pbopt = append(pbopt,
progressbar.OptionSetWriter(os.Stderr),
@ -309,16 +314,22 @@ func CLIOBIClean(itertator obiiter.IBioSequence) obiiter.IBioSequence {
progressbar.OptionSetDescription("[Filter graph on abundance ratio]"),
)
bar := progressbar.NewOptions(len(samples), pbopt...)
bar = progressbar.NewOptions(len(samples), pbopt...)
}
for _, seqs := range samples {
FilterGraphOnRatio(seqs, RatioMax())
if bar != nil {
bar.Add(1)
}
}
}
Mutation(samples)
bar := (*progressbar.ProgressBar)(nil)
if obiconvert.CLIProgressBar() {
pbopt := make([]progressbar.Option, 0, 5)
pbopt = append(pbopt,
progressbar.OptionSetWriter(os.Stderr),
@ -328,7 +339,8 @@ func CLIOBIClean(itertator obiiter.IBioSequence) obiiter.IBioSequence {
progressbar.OptionSetDescription("[Annotate sequence status]"),
)
bar := progressbar.NewOptions(len(samples), pbopt...)
bar = progressbar.NewOptions(len(samples), pbopt...)
}
for name, seqs := range samples {
for _, pcr := range *seqs {
@ -338,8 +350,11 @@ func CLIOBIClean(itertator obiiter.IBioSequence) obiiter.IBioSequence {
obiweight := Weight(pcr.Sequence)
obiweight[name] = pcr.Weight
}
if bar != nil {
bar.Add(1)
}
}
if SaveGraphToFiles() {
SaveGMLGraphs(GraphFilesDirectory(), samples, MinCountToEvalMutationRate())