The Computer Language
23.03 Benchmarks Game

k-nucleotide Go #3 program

source code

/*
The Computer Language Benchmarks Game
https://salsa.debian.org/benchmarksgame-team/benchmarksgame/

Contributed by Olivier Mengué.
Based on work by Bert Gijsbers, Dirk Moerenhout, Jason Alan Palmer and the Go Authors.
*/

package main

import (
    "bufio"
    "bytes"
    "fmt"
    "os"
    "runtime"
    "sort"
)

// seqString is a sequence of nucleotides as a string: "ACGT..."
type seqString string

// seqChars is a sequence of nucleotides as chars: 'A', 'C', 'G', 'T'...
type seqChars []byte

// seqBits is a sequence of nucleotides as 2 low bits per byte: 0, 1, 3, 2...
type seqBits []byte

// toBits converts *in-place*
func (seq seqChars) toBits() seqBits {
    for i := 0; i < len(seq); i++ {
        // 'A' => 0, 'C' => 1, 'T' => 2, 'G' => 3
        seq[i] = seq[i] >> 1 & 3
    }
    return seqBits(seq)
}

func (seq seqString) seqBits() seqBits {
    return seqChars(seq).toBits()
}

// seq32 is a short (<= 16) sequence of nucleotides in a compact form
// length is not embedded
type seq32 uint32

// seq64 is a short (17..32) sequence of nucleotides in a compact form
// length is not embedded
type seq64 uint64

// seq32 converts a seqBits to a seq32
func (seq seqBits) seq32() seq32 {
    var num seq32
    for _, char := range seq {
        num = num<<2 | seq32(char)
    }
    return num
}

// seq64 converts a seqBits to a seq64
func (seq seqBits) seq64() seq64 {
    var num seq64
    for _, char := range seq {
        num = num<<2 | seq64(char)
    }
    return num
}

// seqString converts a seq32 to a human readable string
func (num seq32) seqString(length int) seqString {
    sequence := make(seqChars, length)
    for i := 0; i < length; i++ {
        sequence[length-i-1] = "ACTG"[num&3]
        num = num >> 2
    }
    return seqString(sequence)
}

type counter uint32

func (dna seqBits) count32(length int) map[seq32]*counter {
    counts := make(map[seq32]*counter)
    key := dna[0 : length-1].seq32()
    mask := seq32(1)<<uint(2*length) - 1
    for index := length - 1; index < len(dna); index++ {
        key = key<<2&mask | seq32(dna[index])
        pointer := counts[key]
        if pointer == nil {
            n := counter(1)
            counts[key] = &n
        } else {
            *pointer++
        }
    }
    return counts
}

func (dna seqBits) count64(length int) map[seq64]*counter {
    counts := make(map[seq64]*counter)
    key := dna[0 : length-1].seq64()
    mask := seq64(1)<<uint(2*length) - 1
    for index := length - 1; index < len(dna); index++ {
        key = key<<2&mask | seq64(dna[index])
        pointer := counts[key]
        if pointer == nil {
            n := counter(1)
            counts[key] = &n
        } else {
            *pointer++
        }
    }
    return counts
}

type job struct {
    run    func(dna seqBits)
    result chan string
}

func makeJob(j func(dna seqBits) string) job {
    r := make(chan string, 1)
    return job{
        run: func(dna seqBits) {
            r <- j(dna)
        },
        result: r,
    }
}

func frequencyReportJob(length int) job {
    return makeJob(func(dna seqBits) string {
        return frequencyReport(dna, length)
    })
}

func sequenceReportJob(sequence seqString) job {
    return makeJob(func(dna seqBits) string {
        return sequenceReport(dna, sequence)
    })
}

var jobs = [...]job{
    frequencyReportJob(1),
    frequencyReportJob(2),
    sequenceReportJob("GGT"),
    sequenceReportJob("GGTA"),
    sequenceReportJob("GGTATT"),
    sequenceReportJob("GGTATTTTAATT"),
    sequenceReportJob("GGTATTTTAATTTATAGT"),
}

func main() {
    dna := input()
    scheduleJobs(dna)
    for i := range jobs {
        fmt.Println(<-jobs[i].result)
    }
}

func scheduleJobs(dna seqBits) {
    command := make(chan int, len(jobs))
    for i := runtime.NumCPU(); i > 0; i-- {
        go worker(dna, command)
    }

    for i := range jobs {
        // longest job first, shortest job last
        command <- len(jobs) - 1 - i
    }
    close(command)
}

func worker(dna seqBits, command <-chan int) {
    for k := range command {
        jobs[k].run(dna)
    }
}

func input() (data seqBits) {
    return readSequence(">THREE").toBits()
}

func readSequence(prefix string) (data seqChars) {
    in, lineCount := findSequence(prefix)
    data = make(seqChars, 0, lineCount*61)
    for {
        line, err := in.ReadSlice('\n')
        if len(line) <= 1 || line[0] == '>' {
            break
        }

        last := len(line) - 1
        if line[last] == '\n' {
            line = line[0:last]
        }
        data = append(data, seqChars(line)...)

        if err != nil {
            break
        }
    }
    return
}

func findSequence(prefix string) (in *bufio.Reader, lineCount int) {
    pfx := []byte(prefix)
    in = bufio.NewReaderSize(os.Stdin, 1<<20)
    for {
        line, err := in.ReadSlice('\n')
        if err != nil {
            panic("read error")
        }
        lineCount++
        if line[0] == '>' && bytes.HasPrefix(line, pfx) {
            break
        }
    }
    return
}

type seqCount struct {
    seq   seqString
    count counter
}

type seqCounts []seqCount

func (ss seqCounts) Len() int { return len(ss) }

func (ss seqCounts) Swap(i, j int) { ss[i], ss[j] = ss[j], ss[i] }

// Less order descending by count then seq
func (ss seqCounts) Less(i, j int) bool {
    if ss[i].count == ss[j].count {
        return ss[i].seq > ss[j].seq
    }
    return ss[i].count > ss[j].count
}

func frequencyReport(dna seqBits, length int) string {
    counts := dna.count32(length)
    sortedSeqs := make(seqCounts, 0, len(counts))
    for num, pointer := range counts {
        sortedSeqs = append(
            sortedSeqs,
            seqCount{num.seqString(length), *pointer},
        )
    }
    sort.Sort(sortedSeqs)

    var buf bytes.Buffer
    buf.Grow((8 + length) * len(sortedSeqs))
    var scale float32 = 100.0 / float32(len(dna)-length+1)
    for _, sequence := range sortedSeqs {
        buf.WriteString(fmt.Sprintf(
            "%v %.3f\n", sequence.seq,
            float32(sequence.count)*scale),
        )
    }
    return buf.String()
}

func sequenceReport(dna seqBits, sequence seqString) string {
    var pointer *counter
    seq := sequence.seqBits()
    if len(sequence) <= 16 {
        counts := dna.count32(len(sequence))
        pointer = counts[seq.seq32()]
    } else {
        counts := dna.count64(len(sequence))
        pointer = counts[seq.seq64()]
    }
    var sequenceCount counter
    if pointer != nil {
        sequenceCount = *pointer
    }
    return fmt.Sprintf("%v\t%v", sequenceCount, sequence)
}
    

notes, command-line, and program output

NOTES:
64-bit Ubuntu quad core
go version go1.20 linux/amd64
GOAMD64=v2


Wed, 01 Feb 2023 21:27:58 GMT

MAKE:
/opt/src/go1.20/go/bin/go build -o knucleotide.go-3.go_run knucleotide.go-3.go

4.83s to complete and log all make actions

COMMAND LINE:
./knucleotide.go-3.go_run 0 < knucleotide-input25000000.txt

PROGRAM OUTPUT:
A 30.295
T 30.151
C 19.800
G 19.754

AA 9.177
TA 9.132
AT 9.131
TT 9.091
CA 6.002
AC 6.001
AG 5.987
GA 5.984
CT 5.971
TC 5.971
GT 5.957
TG 5.956
CC 3.917
GC 3.911
CG 3.909
GG 3.902

1471758	GGT
446535	GGTA
47336	GGTATT
893	GGTATTTTAATT
893	GGTATTTTAATTTATAGT