mirror of
https://github.com/metabarcoding/obitools4.git
synced 2025-06-29 16:20:46 +00:00
Correct for a strange bug...
This commit is contained in:
@ -7,7 +7,6 @@ import (
|
|||||||
|
|
||||||
"git.metabarcoding.org/lecasofts/go/obitools/pkg/obioptions"
|
"git.metabarcoding.org/lecasofts/go/obitools/pkg/obioptions"
|
||||||
"git.metabarcoding.org/lecasofts/go/obitools/pkg/obitools/obiconvert"
|
"git.metabarcoding.org/lecasofts/go/obitools/pkg/obitools/obiconvert"
|
||||||
"git.metabarcoding.org/lecasofts/go/obitools/pkg/obitools/obimultiplex"
|
|
||||||
"git.metabarcoding.org/lecasofts/go/obitools/pkg/obitools/obipairing"
|
"git.metabarcoding.org/lecasofts/go/obitools/pkg/obitools/obipairing"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -29,7 +28,7 @@ func main() {
|
|||||||
trace.Start(ftrace)
|
trace.Start(ftrace)
|
||||||
defer trace.Stop()
|
defer trace.Stop()
|
||||||
|
|
||||||
optionParser := obioptions.GenerateOptionParser(obimultiplex.OptionSet)
|
optionParser := obioptions.GenerateOptionParser(obipairing.OptionSet)
|
||||||
|
|
||||||
optionParser(os.Args)
|
optionParser(os.Args)
|
||||||
pairs, _ := obipairing.IBatchPairedSequence()
|
pairs, _ := obipairing.IBatchPairedSequence()
|
||||||
|
153
pkg/obiformats/binary_write.go
Normal file
153
pkg/obiformats/binary_write.go
Normal file
@ -0,0 +1,153 @@
|
|||||||
|
package obiformats
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"git.metabarcoding.org/lecasofts/go/obitools/pkg/obiseq"
|
||||||
|
"github.com/renproject/surge"
|
||||||
|
)
|
||||||
|
|
||||||
|
func WriteBinary(iterator obiseq.IBioSequence, file io.Writer) error {
|
||||||
|
singleseq := make(obiseq.BioSequenceSlice, 1)
|
||||||
|
blob := make([]byte, 0, 1024)
|
||||||
|
for iterator.Next() {
|
||||||
|
singleseq[0] = iterator.Get()
|
||||||
|
blobsize := singleseq.SizeHint()
|
||||||
|
if blobsize > cap(blob) {
|
||||||
|
blob = make([]byte, 0, blobsize*2+8)
|
||||||
|
}
|
||||||
|
_, _, err := surge.MarshalI64(int64(blobsize), blob, 8)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
data := blob[8 : 8+blobsize]
|
||||||
|
_, _, err = singleseq.Marshal(data, blobsize)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
file.Write(blob[0 : 8+blobsize])
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func WriteBinaryToFile(iterator obiseq.IBioSequence,
|
||||||
|
filename string) error {
|
||||||
|
|
||||||
|
file, err := os.Create(filename)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("open file error: %v", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return WriteBinary(iterator, file)
|
||||||
|
}
|
||||||
|
|
||||||
|
func WriteBinaryToStdout(iterator obiseq.IBioSequence) error {
|
||||||
|
return WriteBinary(iterator, os.Stdout)
|
||||||
|
}
|
||||||
|
|
||||||
|
func WriteBinaryBatch(iterator obiseq.IBioSequenceBatch, file io.Writer, options ...WithOption) (obiseq.IBioSequenceBatch, error) {
|
||||||
|
opt := MakeOptions(options)
|
||||||
|
|
||||||
|
buffsize := iterator.BufferSize()
|
||||||
|
newIter := obiseq.MakeIBioSequenceBatch(buffsize)
|
||||||
|
|
||||||
|
nwriters := opt.ParallelWorkers()
|
||||||
|
|
||||||
|
chunkchan := make(chan FileChunck)
|
||||||
|
|
||||||
|
newIter.Add(nwriters)
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
newIter.Wait()
|
||||||
|
for len(chunkchan) > 0 {
|
||||||
|
time.Sleep(time.Millisecond)
|
||||||
|
}
|
||||||
|
close(chunkchan)
|
||||||
|
for len(newIter.Channel()) > 0 {
|
||||||
|
time.Sleep(time.Millisecond)
|
||||||
|
}
|
||||||
|
close(newIter.Channel())
|
||||||
|
}()
|
||||||
|
|
||||||
|
ff := func(iterator obiseq.IBioSequenceBatch) {
|
||||||
|
blob := make([]byte, 0, 10240)
|
||||||
|
for iterator.Next() {
|
||||||
|
batch := iterator.Get()
|
||||||
|
blobsize := batch.Slice().SizeHint()
|
||||||
|
if blobsize > cap(blob) {
|
||||||
|
blob = make([]byte, 0, blobsize*2+8)
|
||||||
|
}
|
||||||
|
_, _, err := surge.MarshalI64(int64(blobsize), blob, 8)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("error in reading binary file %v\n", err)
|
||||||
|
}
|
||||||
|
data := blob[8 : 8+blobsize]
|
||||||
|
_, _, err = batch.Slice().Marshal(data, blobsize)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("error in reading binary file %v\n", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
chunkchan <- FileChunck{
|
||||||
|
data,
|
||||||
|
batch.Order(),
|
||||||
|
}
|
||||||
|
newIter.Channel() <- batch
|
||||||
|
}
|
||||||
|
newIter.Done()
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Println("Start of the binary file writing")
|
||||||
|
go ff(iterator)
|
||||||
|
for i := 0; i < nwriters-1; i++ {
|
||||||
|
go ff(iterator.Split())
|
||||||
|
}
|
||||||
|
|
||||||
|
next_to_send := 0
|
||||||
|
received := make(map[int]FileChunck, 100)
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
for chunk := range chunkchan {
|
||||||
|
if chunk.order == next_to_send {
|
||||||
|
file.Write(chunk.text)
|
||||||
|
next_to_send++
|
||||||
|
chunk, ok := received[next_to_send]
|
||||||
|
for ok {
|
||||||
|
file.Write(chunk.text)
|
||||||
|
delete(received, next_to_send)
|
||||||
|
next_to_send++
|
||||||
|
chunk, ok = received[next_to_send]
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
received[chunk.order] = chunk
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
return newIter, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func WriteBinaryBatchToStdout(iterator obiseq.IBioSequenceBatch, options ...WithOption) (obiseq.IBioSequenceBatch, error) {
|
||||||
|
return WriteBinaryBatch(iterator, os.Stdout, options...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func WriteBinaryBatchToFile(iterator obiseq.IBioSequenceBatch,
|
||||||
|
filename string,
|
||||||
|
options ...WithOption) (obiseq.IBioSequenceBatch, error) {
|
||||||
|
|
||||||
|
file, err := os.Create(filename)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("open file error: %v", err)
|
||||||
|
return obiseq.NilIBioSequenceBatch, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return WriteBinaryBatch(iterator, file, options...)
|
||||||
|
}
|
@ -143,11 +143,11 @@ func MakeOptions(setters []WithOption) Options {
|
|||||||
func _ExtractBarcodeSlice(ngslibrary NGSLibrary,
|
func _ExtractBarcodeSlice(ngslibrary NGSLibrary,
|
||||||
sequences obiseq.BioSequenceSlice,
|
sequences obiseq.BioSequenceSlice,
|
||||||
options Options) obiseq.BioSequenceSlice {
|
options Options) obiseq.BioSequenceSlice {
|
||||||
newSlice := make(obiseq.BioSequenceSlice,0,len(sequences))
|
newSlice := make(obiseq.BioSequenceSlice, 0, len(sequences))
|
||||||
|
|
||||||
for _, seq := range sequences {
|
for _, seq := range sequences {
|
||||||
s, err := ngslibrary.ExtractBarcode(seq,true)
|
s, err := ngslibrary.ExtractBarcode(seq, true)
|
||||||
if err==nil || ! options.pointer.discardErrors {
|
if err == nil || !options.pointer.discardErrors {
|
||||||
newSlice = append(newSlice, s)
|
newSlice = append(newSlice, s)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -179,4 +179,3 @@ func ExtractBarcodeSliceWorker(ngslibrary NGSLibrary,
|
|||||||
|
|
||||||
return worker
|
return worker
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -27,7 +27,6 @@ func GenerateOptionParser(optionset ...func(*getoptions.GetOpt)) ArgumentParser
|
|||||||
options.Description("Number of parallele threads computing the result"))
|
options.Description("Number of parallele threads computing the result"))
|
||||||
|
|
||||||
options.IntVar(&_MaxAllowedCPU, "max-cpu", _MaxAllowedCPU,
|
options.IntVar(&_MaxAllowedCPU, "max-cpu", _MaxAllowedCPU,
|
||||||
options.Alias("w"),
|
|
||||||
options.Description("Number of parallele threads computing the result"))
|
options.Description("Number of parallele threads computing the result"))
|
||||||
|
|
||||||
for _, o := range optionset {
|
for _, o := range optionset {
|
||||||
|
@ -30,7 +30,7 @@ func PairingOptionSet(options *getoptions.GetOpt) {
|
|||||||
options.Alias("O"),
|
options.Alias("O"),
|
||||||
options.Description("Minimum ovelap between both the reads to consider the aligment"))
|
options.Description("Minimum ovelap between both the reads to consider the aligment"))
|
||||||
options.Float64Var(&_MinIdentity, "min-identity", _MinIdentity,
|
options.Float64Var(&_MinIdentity, "min-identity", _MinIdentity,
|
||||||
options.Alias("O"),
|
options.Alias("X"),
|
||||||
options.Description("Minimum identity between ovelaped regions of the reads to consider the aligment"))
|
options.Description("Minimum identity between ovelaped regions of the reads to consider the aligment"))
|
||||||
options.Float64Var(&_GapPenality, "gap-penality", _GapPenality,
|
options.Float64Var(&_GapPenality, "gap-penality", _GapPenality,
|
||||||
options.Alias("G"),
|
options.Alias("G"),
|
||||||
|
@ -105,7 +105,7 @@ func JoinPairedSequence(seqA, seqB obiseq.BioSequence, inplace bool) obiseq.BioS
|
|||||||
// input sequence.
|
// input sequence.
|
||||||
//
|
//
|
||||||
func AssemblePESequences(seqA, seqB obiseq.BioSequence,
|
func AssemblePESequences(seqA, seqB obiseq.BioSequence,
|
||||||
gap float64, delta, minOverlap int, minIdentity float64,withStats bool,
|
gap float64, delta, minOverlap int, minIdentity float64, withStats bool,
|
||||||
inplace bool,
|
inplace bool,
|
||||||
arenaAlign obialign.PEAlignArena) obiseq.BioSequence {
|
arenaAlign obialign.PEAlignArena) obiseq.BioSequence {
|
||||||
|
|
||||||
@ -119,7 +119,7 @@ func AssemblePESequences(seqA, seqB obiseq.BioSequence,
|
|||||||
}
|
}
|
||||||
lcons := cons.Length()
|
lcons := cons.Length()
|
||||||
aliLength := lcons - _Abs(left) - _Abs(right)
|
aliLength := lcons - _Abs(left) - _Abs(right)
|
||||||
identity := float64(match)/float64(aliLength)
|
identity := float64(match) / float64(aliLength)
|
||||||
|
|
||||||
if aliLength >= minOverlap && identity >= minIdentity {
|
if aliLength >= minOverlap && identity >= minIdentity {
|
||||||
if withStats {
|
if withStats {
|
||||||
@ -268,7 +268,6 @@ func IAssemblePESequencesBatch(iterator obiseq.IPairedBioSequenceBatch,
|
|||||||
go f(iterator.Split(), i)
|
go f(iterator.Split(), i)
|
||||||
}
|
}
|
||||||
go f(iterator, nworkers-1)
|
go f(iterator, nworkers-1)
|
||||||
|
|
||||||
return newIter
|
return newIter
|
||||||
|
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user