mirror of
https://github.com/metabarcoding/obitools4.git
synced 2025-06-29 16:20:46 +00:00
Start of implementing binary file format
This commit is contained in:
@ -2,7 +2,6 @@ package obiseq
|
||||
|
||||
type SequencePredicate func(BioSequence) bool
|
||||
|
||||
|
||||
func (predicate1 SequencePredicate) And(predicate2 SequencePredicate) SequencePredicate {
|
||||
f := func(sequence BioSequence) bool {
|
||||
return predicate1(sequence) && predicate2(sequence)
|
||||
@ -74,4 +73,3 @@ func IsShorterOrEqualTo(length int) SequencePredicate {
|
||||
|
||||
return f
|
||||
}
|
||||
|
||||
|
105
pkg/obiseq/surge.go
Normal file
105
pkg/obiseq/surge.go
Normal file
@ -0,0 +1,105 @@
|
||||
package obiseq
|
||||
|
||||
import "github.com/renproject/surge"
|
||||
|
||||
func (sequence BioSequence) SizeHint() int {
|
||||
return surge.SizeHintString(sequence.sequence.id) +
|
||||
surge.SizeHintString(sequence.sequence.definition) +
|
||||
surge.SizeHintBytes(sequence.sequence.sequence) +
|
||||
surge.SizeHintBytes(sequence.sequence.qualities) +
|
||||
surge.SizeHintBytes(sequence.sequence.feature) +
|
||||
surge.SizeHint(sequence.sequence.annotations)
|
||||
}
|
||||
|
||||
func (sequence BioSequence) Marshal(buf []byte, rem int) ([]byte, int, error) {
|
||||
var err error
|
||||
if buf, rem, err = surge.MarshalString(sequence.sequence.id, buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
if buf, rem, err = surge.MarshalString(sequence.sequence.definition, buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
if buf, rem, err = surge.MarshalBytes(sequence.sequence.sequence, buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
if buf, rem, err = surge.MarshalBytes(sequence.sequence.qualities, buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
if buf, rem, err = surge.MarshalBytes(sequence.sequence.feature, buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
if buf, rem, err = surge.Marshal(sequence.sequence.annotations, buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
|
||||
return buf, rem, err
|
||||
}
|
||||
|
||||
// Unmarshal is the opposite of Marshal, and requires
|
||||
// a pointer receiver.
|
||||
func (sequence *BioSequence) Unmarshal(buf []byte, rem int) ([]byte, int, error) {
|
||||
var err error
|
||||
if buf, rem, err = surge.UnmarshalString(&(sequence.sequence.id), buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
if buf, rem, err = surge.UnmarshalString(&(sequence.sequence.definition), buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
if buf, rem, err = surge.UnmarshalBytes(&(sequence.sequence.sequence), buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
if buf, rem, err = surge.UnmarshalBytes(&(sequence.sequence.qualities), buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
if buf, rem, err = surge.UnmarshalBytes(&(sequence.sequence.feature), buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
if buf, rem, err = surge.Unmarshal(&(sequence.sequence.annotations), buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
return buf, rem, err
|
||||
}
|
||||
|
||||
func (sequences BioSequenceSlice) SizeHint() int {
|
||||
size := surge.SizeHintI64
|
||||
for _, s := range sequences {
|
||||
size += s.SizeHint()
|
||||
}
|
||||
|
||||
return size
|
||||
}
|
||||
|
||||
func (sequences BioSequenceSlice) Marshal(buf []byte, rem int) ([]byte, int, error) {
|
||||
var err error
|
||||
|
||||
if buf, rem, err = surge.MarshalI64(int64(len(sequences)), buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
|
||||
for _, s := range sequences {
|
||||
if buf, rem, err = s.Marshal(buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
}
|
||||
|
||||
return buf, rem, err
|
||||
}
|
||||
|
||||
func (sequences *BioSequenceSlice) Unmarshal(buf []byte, rem int) ([]byte, int, error) {
|
||||
var err error
|
||||
var length int64
|
||||
|
||||
if buf, rem, err = surge.UnmarshalI64(&length, buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
|
||||
*sequences = make(BioSequenceSlice, length)
|
||||
|
||||
for i := 0; i < int(length); i++ {
|
||||
if buf, rem, err = ((*sequences)[i]).Unmarshal(buf, rem); err != nil {
|
||||
return buf, rem, err
|
||||
}
|
||||
}
|
||||
|
||||
return buf, rem, err
|
||||
}
|
Reference in New Issue
Block a user