mirror of
https://github.com/lh3/minimap2.git
synced 2026-09-24 18:08:12 +08:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
dd3d637c20 | ||
|
|
cdb7857841 | ||
|
|
3c0d05d272 | ||
|
|
47b646acbf | ||
|
|
a79cb3e991 | ||
|
|
367aed4271 | ||
|
|
081df6ac7d | ||
|
|
a3e7a575fb | ||
|
|
d90583b83c | ||
|
|
7fc03b0c32 | ||
|
|
20c104ce8d | ||
|
|
238b6bb3ea | ||
|
|
e026e18439 | ||
|
|
58c2251b18 | ||
|
|
03dc8d5d97 | ||
|
|
5cb61f8ee6 | ||
|
|
c18cd3ad2d | ||
|
|
c16a1742a3 | ||
|
|
4bd5a018c2 | ||
|
|
05974c80f1 | ||
|
|
7bc87b4175 | ||
|
|
6762368cf0 | ||
|
|
e60d78e0b1 | ||
|
|
c2aec88b84 | ||
|
|
97f67a2a0a | ||
|
|
189555503a | ||
|
|
69af86657e | ||
|
|
49c6d83a8e | ||
|
|
f64e426a5a | ||
|
|
2bb8cbbeef | ||
|
|
e80759c97a | ||
|
|
f4c844b143 | ||
|
|
be171aa2dc | ||
|
|
cdc730d573 | ||
|
|
6420acca6d | ||
|
|
371bc9513a | ||
|
|
e9a45a4e1c | ||
|
|
169216bfff | ||
|
|
6b391e3373 | ||
|
|
55e39c2d30 | ||
|
|
90b7b83ec7 | ||
|
|
d431dc0181 | ||
|
|
ccf1680aaf | ||
|
|
ea84fc0a53 | ||
|
|
19208fb06b | ||
|
|
e02bebd96d | ||
|
|
32ab6ce15b | ||
|
|
1739a260fb | ||
|
|
aaf3233818 | ||
|
|
8b05880f73 | ||
|
|
eba237f39d | ||
|
|
a8e1e3cbb8 | ||
|
|
597212b9f3 | ||
|
|
30abcf3cf9 | ||
|
|
48e230f40d | ||
|
|
f5e2176bc5 | ||
|
|
0b4be2996e | ||
|
|
feca68c71d | ||
|
|
6d9ce56721 | ||
|
|
f2f425890d | ||
|
|
58f4210dea | ||
|
|
a4782c7d7a | ||
|
|
2a7d071e8b | ||
|
|
9462da5159 |
@@ -0,0 +1,46 @@
|
|||||||
|
#### 1. Alignment different with option `-a` or `-c`?
|
||||||
|
|
||||||
|
Without `-a`, `-c` or `--cs`, minimap2 only finds *approximate* mapping
|
||||||
|
locations without detailed base alignment. In particular, the start and end
|
||||||
|
positions of the alignment are impricise. With one of those options, minimap2
|
||||||
|
will perform base alignment, which is generally more accurate but is much
|
||||||
|
slower.
|
||||||
|
|
||||||
|
#### 2. How to map Illumina short reads to noisy long reads?
|
||||||
|
|
||||||
|
No good solutions. The better approach is to assemble short reads into contigs
|
||||||
|
and then map noisy reads to contigs.
|
||||||
|
|
||||||
|
#### 3. The output SAM doesn't have a header.
|
||||||
|
|
||||||
|
By default, minimap2 indexes 4 billion reference bases (4Gb) in a batch and map
|
||||||
|
all reads against each reference batch. Given a reference longer than 4Gb,
|
||||||
|
minimap2 is unable to see all the sequences and thus can't produce a correct
|
||||||
|
SAM header. In this case, minimap2 doesn't output any SAM header. There are two
|
||||||
|
solutions to this issue. First, you may increase option `-I` to, for example,
|
||||||
|
`-I8g` to index more reference bases in a batch. This is preferred if your
|
||||||
|
machine has enough memory. Second, if your machines doesn't have enough memory
|
||||||
|
to hold the reference index, you can use the `--split-prefix` option in a
|
||||||
|
command line like:
|
||||||
|
```sh
|
||||||
|
minimap2 -ax map-ont --split-prefix=tmp ref.fa reads.fq
|
||||||
|
```
|
||||||
|
This second approach uses less memory, but it is slower and requires temporary
|
||||||
|
disk space.
|
||||||
|
|
||||||
|
#### 4. The output SAM is malformatted.
|
||||||
|
|
||||||
|
This typically happens when you use nohup to wrap a minimap2 command line.
|
||||||
|
Nohup is discouraged as it breaks piping. If you have to use nohup, please
|
||||||
|
specify an output file with option `-o`.
|
||||||
|
|
||||||
|
#### 5. How to output one alignment per read?
|
||||||
|
|
||||||
|
You can use `--secondary=no` to suppress secondary alignments (aka multiple
|
||||||
|
mappings), but you can't suppress supplementary alignment (aka split or
|
||||||
|
chimeric alignment) this way. You can use samtools to filter out these
|
||||||
|
alignments:
|
||||||
|
```sh
|
||||||
|
minimap2 -ax map-out ref.fa reads.fq | samtools view -F0x900
|
||||||
|
```
|
||||||
|
However, this is discouraged as supplementary alignment is informative.
|
||||||
@@ -2,13 +2,25 @@ CFLAGS= -g -Wall -O2 -Wc++-compat #-Wextra
|
|||||||
CPPFLAGS= -DHAVE_KALLOC
|
CPPFLAGS= -DHAVE_KALLOC
|
||||||
INCLUDES=
|
INCLUDES=
|
||||||
OBJS= kthread.o kalloc.o misc.o bseq.o sketch.o sdust.o options.o index.o chain.o align.o hit.o map.o format.o pe.o esterr.o splitidx.o ksw2_ll_sse.o
|
OBJS= kthread.o kalloc.o misc.o bseq.o sketch.o sdust.o options.o index.o chain.o align.o hit.o map.o format.o pe.o esterr.o splitidx.o ksw2_ll_sse.o
|
||||||
|
OBJS_SSE= ksw2_extz2_sse41.o ksw2_extd2_sse41.o ksw2_exts2_sse41.o ksw2_extz2_sse2.o ksw2_extd2_sse2.o ksw2_exts2_sse2.o
|
||||||
|
DISPATCH_FLAG=-msse4.1
|
||||||
PROG= minimap2
|
PROG= minimap2
|
||||||
PROG_EXTRA= sdust minimap2-lite
|
PROG_EXTRA= sdust minimap2-lite
|
||||||
LIBS= -lm -lz -lpthread
|
LIBS= -lm -lz -lpthread
|
||||||
|
|
||||||
ifeq ($(arm_neon),) # if arm_neon is not defined
|
ifeq ($(arm_neon),) # if arm_neon is not defined
|
||||||
ifeq ($(sse2only),) # if sse2only is not defined
|
ifeq ($(sse2only),) # if sse2only is not defined
|
||||||
OBJS+=ksw2_extz2_sse41.o ksw2_extd2_sse41.o ksw2_exts2_sse41.o ksw2_extz2_sse2.o ksw2_extd2_sse2.o ksw2_exts2_sse2.o ksw2_dispatch.o
|
ifeq ($(avx512),)
|
||||||
|
ifeq ($(avx2),)
|
||||||
|
OBJS+=$(OBJS_SSE) ksw2_dispatch.o
|
||||||
|
else
|
||||||
|
OBJS+=ksw2_extd2_avx2.o $(OBJS_SSE) ksw2_dispatch.o
|
||||||
|
DISPATCH_FLAG=-mavx2
|
||||||
|
endif
|
||||||
|
else
|
||||||
|
OBJS+=ksw2_extd2_avx512.o ksw2_extd2_avx2.o $(OBJS_SSE) ksw2_dispatch.o
|
||||||
|
DISPATCH_FLAG=-mavx512bw
|
||||||
|
endif
|
||||||
else # if sse2only is defined
|
else # if sse2only is defined
|
||||||
OBJS+=ksw2_extz2_sse.o ksw2_extd2_sse.o ksw2_exts2_sse.o
|
OBJS+=ksw2_extz2_sse.o ksw2_extd2_sse.o ksw2_exts2_sse.o
|
||||||
endif
|
endif
|
||||||
@@ -22,6 +34,16 @@ else #if aarch64 is defined
|
|||||||
endif
|
endif
|
||||||
endif
|
endif
|
||||||
|
|
||||||
|
ifneq ($(asan),)
|
||||||
|
CFLAGS+=-fsanitize=address
|
||||||
|
LIBS+=-fsanitize=address
|
||||||
|
endif
|
||||||
|
|
||||||
|
ifneq ($(tsan),)
|
||||||
|
CFLAGS+=-fsanitize=thread
|
||||||
|
LIBS+=-fsanitize=thread
|
||||||
|
endif
|
||||||
|
|
||||||
.PHONY:all extra clean depend
|
.PHONY:all extra clean depend
|
||||||
.SUFFIXES:.c .o
|
.SUFFIXES:.c .o
|
||||||
|
|
||||||
@@ -57,11 +79,17 @@ ksw2_extz2_sse41.o:ksw2_extz2_sse.c ksw2.h kalloc.h
|
|||||||
ksw2_extz2_sse2.o:ksw2_extz2_sse.c ksw2.h kalloc.h
|
ksw2_extz2_sse2.o:ksw2_extz2_sse.c ksw2.h kalloc.h
|
||||||
$(CC) -c $(CFLAGS) -msse2 -mno-sse4.1 $(CPPFLAGS) -DKSW_CPU_DISPATCH -DKSW_SSE2_ONLY $(INCLUDES) $< -o $@
|
$(CC) -c $(CFLAGS) -msse2 -mno-sse4.1 $(CPPFLAGS) -DKSW_CPU_DISPATCH -DKSW_SSE2_ONLY $(INCLUDES) $< -o $@
|
||||||
|
|
||||||
|
ksw2_extd2_avx2.o:ksw2_extd2_sse.c ksw2.h kalloc.h
|
||||||
|
$(CC) -c $(CFLAGS) -mavx2 $(CPPFLAGS) -DKSW_CPU_DISPATCH $(INCLUDES) $< -o $@
|
||||||
|
|
||||||
|
ksw2_extd2_avx512.o:ksw2_extd2_sse.c ksw2.h kalloc.h
|
||||||
|
$(CC) -c $(CFLAGS) -mavx512bw $(CPPFLAGS) -DKSW_CPU_DISPATCH $(INCLUDES) $< -o $@
|
||||||
|
|
||||||
ksw2_extd2_sse41.o:ksw2_extd2_sse.c ksw2.h kalloc.h
|
ksw2_extd2_sse41.o:ksw2_extd2_sse.c ksw2.h kalloc.h
|
||||||
$(CC) -c $(CFLAGS) -msse4.1 $(CPPFLAGS) -DKSW_CPU_DISPATCH $(INCLUDES) $< -o $@
|
$(CC) -c $(CFLAGS) -msse4.1 -mno-avx2 $(CPPFLAGS) -DKSW_CPU_DISPATCH $(INCLUDES) $< -o $@
|
||||||
|
|
||||||
ksw2_extd2_sse2.o:ksw2_extd2_sse.c ksw2.h kalloc.h
|
ksw2_extd2_sse2.o:ksw2_extd2_sse.c ksw2.h kalloc.h
|
||||||
$(CC) -c $(CFLAGS) -msse2 -mno-sse4.1 $(CPPFLAGS) -DKSW_CPU_DISPATCH -DKSW_SSE2_ONLY $(INCLUDES) $< -o $@
|
$(CC) -c $(CFLAGS) -msse2 -mno-sse4.1 -mno-avx2 $(CPPFLAGS) -DKSW_CPU_DISPATCH -DKSW_SSE2_ONLY $(INCLUDES) $< -o $@
|
||||||
|
|
||||||
ksw2_exts2_sse41.o:ksw2_exts2_sse.c ksw2.h kalloc.h
|
ksw2_exts2_sse41.o:ksw2_exts2_sse.c ksw2.h kalloc.h
|
||||||
$(CC) -c $(CFLAGS) -msse4.1 $(CPPFLAGS) -DKSW_CPU_DISPATCH $(INCLUDES) $< -o $@
|
$(CC) -c $(CFLAGS) -msse4.1 $(CPPFLAGS) -DKSW_CPU_DISPATCH $(INCLUDES) $< -o $@
|
||||||
@@ -70,7 +98,7 @@ ksw2_exts2_sse2.o:ksw2_exts2_sse.c ksw2.h kalloc.h
|
|||||||
$(CC) -c $(CFLAGS) -msse2 -mno-sse4.1 $(CPPFLAGS) -DKSW_CPU_DISPATCH -DKSW_SSE2_ONLY $(INCLUDES) $< -o $@
|
$(CC) -c $(CFLAGS) -msse2 -mno-sse4.1 $(CPPFLAGS) -DKSW_CPU_DISPATCH -DKSW_SSE2_ONLY $(INCLUDES) $< -o $@
|
||||||
|
|
||||||
ksw2_dispatch.o:ksw2_dispatch.c ksw2.h
|
ksw2_dispatch.o:ksw2_dispatch.c ksw2.h
|
||||||
$(CC) -c $(CFLAGS) -msse4.1 $(CPPFLAGS) -DKSW_CPU_DISPATCH $(INCLUDES) $< -o $@
|
$(CC) -c $(CFLAGS) $(DISPATCH_FLAG) $(CPPFLAGS) -DKSW_CPU_DISPATCH $(INCLUDES) $< -o $@
|
||||||
|
|
||||||
# NEON-specific targets on ARM
|
# NEON-specific targets on ARM
|
||||||
|
|
||||||
|
|||||||
@@ -1,3 +1,80 @@
|
|||||||
|
Release 2.17-r941 (4 May 2019)
|
||||||
|
------------------------------
|
||||||
|
|
||||||
|
Changes since the last release:
|
||||||
|
|
||||||
|
* Fixed flawed CIGARs like `5I6D7I` (#392).
|
||||||
|
|
||||||
|
* Bugfix: TLEN should be 0 when either end is unmapped (#373 and #365).
|
||||||
|
|
||||||
|
* Bugfix: mappy is unable to write index (#372).
|
||||||
|
|
||||||
|
* Added option `--junc-bed` to load known gene annotations in the BED12
|
||||||
|
format. Minimap2 prefers annotated junctions over novel junctions (#197 and
|
||||||
|
#348). GTF can be converted to BED12 with `paftools.js gff2bed`.
|
||||||
|
|
||||||
|
* Added option `--sam-hit-only` to suppress unmapped hits in SAM (#377).
|
||||||
|
|
||||||
|
* Added preset `splice:hq` for high-quality CCS or mRNA sequences. It applies
|
||||||
|
better scoring and improves the sensitivity to small exons. This preset may
|
||||||
|
introduce false small introns, but the overall accuracy should be higher.
|
||||||
|
|
||||||
|
This version produces nearly identical alignments to v2.16, except for CIGARs
|
||||||
|
affected by the bug mentioned above.
|
||||||
|
|
||||||
|
(2.17: 5 May 2019, r941)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
Release 2.16-r922 (28 February 2019)
|
||||||
|
------------------------------------
|
||||||
|
|
||||||
|
This release is 50% faster for mapping ultra-long nanopore reads at comparable
|
||||||
|
accuracy. For short-read mapping, long-read overlapping and ordinary long-read
|
||||||
|
mapping, the performance and accuracy remain similar. This speedup is achieved
|
||||||
|
with a new heuristic to limit the number of chaining iterations (#324). Users
|
||||||
|
can disable the heuristic by increasing a new option `--max-chain-iter` to a
|
||||||
|
huge number.
|
||||||
|
|
||||||
|
Other changes to minimap2:
|
||||||
|
|
||||||
|
* Implemented option `--paf-no-hit` to output unmapped query sequences in PAF.
|
||||||
|
The strand and reference name columns are both `*` at an unmapped line. The
|
||||||
|
hidden option is available in earlier minimap2 but had a different 2-column
|
||||||
|
output format instead of PAF.
|
||||||
|
|
||||||
|
* Fixed a bug that leads to wrongly calculated `de` tags when ambiguous bases
|
||||||
|
are involved (#309). This bug only affects v2.15.
|
||||||
|
|
||||||
|
* Fixed a bug when parsing command-line option `--splice` (#344). This bug was
|
||||||
|
introduced in v2.13.
|
||||||
|
|
||||||
|
* Fixed two division-by-zero cases (#326). They don't affect final alignments
|
||||||
|
because the results of the divisions are not used in both case.
|
||||||
|
|
||||||
|
* Added an option `-o` to output alignments to a specified file. It is still
|
||||||
|
recommended to use UNIX pipes for on-the-fly conversion or compression.
|
||||||
|
|
||||||
|
* Output a new `rl` tag to give the length of query regions harboring
|
||||||
|
repetitive seeds.
|
||||||
|
|
||||||
|
Changes to paftool.js:
|
||||||
|
|
||||||
|
* Added a new option to convert the MD tag to the long form of the cs tag.
|
||||||
|
|
||||||
|
Changes to mappy:
|
||||||
|
|
||||||
|
* Added the `mappy.Aligner.seq_names` method to return sequence names (#312).
|
||||||
|
|
||||||
|
For NA12878 ultra-long reads, this release changes the alignments of <0.1% of
|
||||||
|
reads in comparison to v2.15. All these reads have highly fragmented alignments
|
||||||
|
and are likely to be problematic anyway. For shorter or well aligned reads,
|
||||||
|
this release should produce mostly identical alignments to v2.15.
|
||||||
|
|
||||||
|
(2.16: 28 February 2019, r922)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
Release 2.15-r905 (10 January 2019)
|
Release 2.15-r905 (10 January 2019)
|
||||||
-----------------------------------
|
-----------------------------------
|
||||||
|
|
||||||
|
|||||||
@@ -9,8 +9,8 @@ cd minimap2 && make
|
|||||||
# long sequences against a reference genome
|
# long sequences against a reference genome
|
||||||
./minimap2 -a test/MT-human.fa test/MT-orang.fa > test.sam
|
./minimap2 -a test/MT-human.fa test/MT-orang.fa > test.sam
|
||||||
# create an index first and then map
|
# create an index first and then map
|
||||||
./minimap2 -d MT-human.mmi test/MT-human.fa
|
./minimap2 -x map-ont -d MT-human-ont.mmi test/MT-human.fa
|
||||||
./minimap2 -a MT-human.mmi test/MT-orang.fa > test.sam
|
./minimap2 -a MT-human-ont.mmi test/MT-orang.fa > test.sam
|
||||||
# use presets (no test data)
|
# use presets (no test data)
|
||||||
./minimap2 -ax map-pb ref.fa pacbio.fq.gz > aln.sam # PacBio genomic reads
|
./minimap2 -ax map-pb ref.fa pacbio.fq.gz > aln.sam # PacBio genomic reads
|
||||||
./minimap2 -ax map-ont ref.fa ont.fq.gz > aln.sam # Oxford Nanopore genomic reads
|
./minimap2 -ax map-ont ref.fa ont.fq.gz > aln.sam # Oxford Nanopore genomic reads
|
||||||
@@ -18,7 +18,7 @@ cd minimap2 && make
|
|||||||
./minimap2 -ax sr ref.fa read1.fa read2.fa > aln.sam # short genomic paired-end reads
|
./minimap2 -ax sr ref.fa read1.fa read2.fa > aln.sam # short genomic paired-end reads
|
||||||
./minimap2 -ax splice ref.fa rna-reads.fa > aln.sam # spliced long reads (strand unknown)
|
./minimap2 -ax splice ref.fa rna-reads.fa > aln.sam # spliced long reads (strand unknown)
|
||||||
./minimap2 -ax splice -uf -k14 ref.fa reads.fa > aln.sam # noisy Nanopore Direct RNA-seq
|
./minimap2 -ax splice -uf -k14 ref.fa reads.fa > aln.sam # noisy Nanopore Direct RNA-seq
|
||||||
./minimap2 -ax splice -uf -C5 ref.fa query.fa > aln.sam # Final PacBio Iso-seq or traditional cDNA
|
./minimap2 -ax splice:hq -uf ref.fa query.fa > aln.sam # Final PacBio Iso-seq or traditional cDNA
|
||||||
./minimap2 -cx asm5 asm1.fa asm2.fa > aln.paf # intra-species asm-to-asm alignment
|
./minimap2 -cx asm5 asm1.fa asm2.fa > aln.paf # intra-species asm-to-asm alignment
|
||||||
./minimap2 -x ava-pb reads.fa reads.fa > overlaps.paf # PacBio read overlap
|
./minimap2 -x ava-pb reads.fa reads.fa > overlaps.paf # PacBio read overlap
|
||||||
./minimap2 -x ava-ont reads.fa reads.fa > overlaps.paf # Nanopore read overlap
|
./minimap2 -x ava-ont reads.fa reads.fa > overlaps.paf # Nanopore read overlap
|
||||||
@@ -71,8 +71,8 @@ Detailed evaluations are available from the [minimap2 paper][doi] or the
|
|||||||
Minimap2 is optimized for x86-64 CPUs. You can acquire precompiled binaries from
|
Minimap2 is optimized for x86-64 CPUs. You can acquire precompiled binaries from
|
||||||
the [release page][release] with:
|
the [release page][release] with:
|
||||||
```sh
|
```sh
|
||||||
curl -L https://github.com/lh3/minimap2/releases/download/v2.15/minimap2-2.15_x64-linux.tar.bz2 | tar -jxvf -
|
curl -L https://github.com/lh3/minimap2/releases/download/v2.17/minimap2-2.17_x64-linux.tar.bz2 | tar -jxvf -
|
||||||
./minimap2-2.15_x64-linux/minimap2
|
./minimap2-2.17_x64-linux/minimap2
|
||||||
```
|
```
|
||||||
If you want to compile from the source, you need to have a C compiler, GNU make
|
If you want to compile from the source, you need to have a C compiler, GNU make
|
||||||
and zlib development files installed. Then type `make` in the source code
|
and zlib development files installed. Then type `make` in the source code
|
||||||
@@ -139,7 +139,7 @@ Nanopore reads.
|
|||||||
#### <a name="map-long-splice"></a>Map long mRNA/cDNA reads
|
#### <a name="map-long-splice"></a>Map long mRNA/cDNA reads
|
||||||
|
|
||||||
```sh
|
```sh
|
||||||
minimap2 -ax splice -uf -C5 ref.fa iso-seq.fq > aln.sam # PacBio Iso-seq/traditional cDNA
|
minimap2 -ax splice:hq -uf ref.fa iso-seq.fq > aln.sam # PacBio Iso-seq/traditional cDNA
|
||||||
minimap2 -ax splice ref.fa nanopore-cdna.fa > aln.sam # Nanopore 2D cDNA-seq
|
minimap2 -ax splice ref.fa nanopore-cdna.fa > aln.sam # Nanopore 2D cDNA-seq
|
||||||
minimap2 -ax splice -uf -k14 ref.fa direct-rna.fq > aln.sam # Nanopore Direct RNA-seq
|
minimap2 -ax splice -uf -k14 ref.fa direct-rna.fq > aln.sam # Nanopore Direct RNA-seq
|
||||||
minimap2 -ax splice --splice-flank=no SIRV.fa SIRV-seq.fa # mapping against SIRV control
|
minimap2 -ax splice --splice-flank=no SIRV.fa SIRV-seq.fa # mapping against SIRV control
|
||||||
@@ -315,16 +315,17 @@ highlighted in bold. The description may help to tune minimap2 parameters.
|
|||||||
### <a name="help"></a>Getting help
|
### <a name="help"></a>Getting help
|
||||||
|
|
||||||
Manpage [minimap2.1][manpage] provides detailed description of minimap2
|
Manpage [minimap2.1][manpage] provides detailed description of minimap2
|
||||||
command line options and optional tags. If you encounter bugs or have further
|
command line options and optional tags. The [FAQ](FAQ.md) page answers several
|
||||||
questions or requests, you can raise an issue at the [issue page][issue].
|
frequently asked questions. If you encounter bugs or have further questions or
|
||||||
There is not a specific mailing list for the time being.
|
requests, you can raise an issue at the [issue page][issue]. There is not a
|
||||||
|
specific mailing list for the time being.
|
||||||
|
|
||||||
### <a name="cite"></a>Citing minimap2
|
### <a name="cite"></a>Citing minimap2
|
||||||
|
|
||||||
If you use minimap2 in your work, please cite:
|
If you use minimap2 in your work, please cite:
|
||||||
|
|
||||||
> Li, H. (2018). Minimap2: pairwise alignment for nucleotide sequences.
|
> Li, H. (2018). Minimap2: pairwise alignment for nucleotide sequences.
|
||||||
> Bioinformatics. [doi:10.1093/bioinformatics/bty191][doi]
|
> *Bioinformatics*, **34**:3094-3100. [doi:10.1093/bioinformatics/bty191][doi]
|
||||||
|
|
||||||
## <a name="dguide"></a>Developers' Guide
|
## <a name="dguide"></a>Developers' Guide
|
||||||
|
|
||||||
|
|||||||
@@ -123,6 +123,25 @@ static void mm_fix_cigar(mm_reg1_t *r, const uint8_t *qseq, const uint8_t *tseq,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
assert(qoff == r->qe - r->qs && toff == r->re - r->rs);
|
assert(qoff == r->qe - r->qs && toff == r->re - r->rs);
|
||||||
|
for (k = 0; k < p->n_cigar - 2; ++k) { // fix CIGAR like 5I6D7I
|
||||||
|
if ((p->cigar[k]&0xf) > 0 && (p->cigar[k]&0xf) + (p->cigar[k+1]&0xf) == 3) {
|
||||||
|
uint32_t l, s[3] = {0,0,0};
|
||||||
|
for (l = k; l < p->n_cigar; ++l) { // count number of adjacent I and D
|
||||||
|
uint32_t op = p->cigar[l]&0xf;
|
||||||
|
if (op == 1 || op == 2 || p->cigar[l]>>4 == 0)
|
||||||
|
s[op] += p->cigar[l] >> 4;
|
||||||
|
else break;
|
||||||
|
}
|
||||||
|
if (s[1] > 0 && s[2] > 0 && l - k > 2) { // turn to a single I and a single D
|
||||||
|
p->cigar[k] = s[1]<<4|1;
|
||||||
|
p->cigar[k+1] = s[2]<<4|2;
|
||||||
|
for (k += 2; k < l; ++k)
|
||||||
|
p->cigar[k] &= 0xf;
|
||||||
|
to_shrink = 1;
|
||||||
|
}
|
||||||
|
k = l;
|
||||||
|
}
|
||||||
|
}
|
||||||
if (to_shrink) { // squeeze out zero-length operations
|
if (to_shrink) { // squeeze out zero-length operations
|
||||||
int32_t l = 0;
|
int32_t l = 0;
|
||||||
for (k = 0; k < p->n_cigar; ++k) // squeeze out zero-length operations
|
for (k = 0; k < p->n_cigar; ++k) // squeeze out zero-length operations
|
||||||
@@ -291,7 +310,7 @@ static void mm_append_cigar(mm_reg1_t *r, uint32_t n_cigar, uint32_t *cigar) //
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static void mm_align_pair(void *km, const mm_mapopt_t *opt, int qlen, const uint8_t *qseq, int tlen, const uint8_t *tseq, const int8_t *mat, int w, int end_bonus, int zdrop, int flag, ksw_extz_t *ez)
|
static void mm_align_pair(void *km, const mm_mapopt_t *opt, int qlen, const uint8_t *qseq, int tlen, const uint8_t *tseq, const uint8_t *junc, const int8_t *mat, int w, int end_bonus, int zdrop, int flag, ksw_extz_t *ez)
|
||||||
{
|
{
|
||||||
if (mm_dbg_flag & MM_DBG_PRINT_ALN_SEQ) {
|
if (mm_dbg_flag & MM_DBG_PRINT_ALN_SEQ) {
|
||||||
int i;
|
int i;
|
||||||
@@ -305,7 +324,7 @@ static void mm_align_pair(void *km, const mm_mapopt_t *opt, int qlen, const uint
|
|||||||
ksw_reset_extz(ez);
|
ksw_reset_extz(ez);
|
||||||
ez->zdropped = 1;
|
ez->zdropped = 1;
|
||||||
} else if (opt->flag & MM_F_SPLICE)
|
} else if (opt->flag & MM_F_SPLICE)
|
||||||
ksw_exts2_sse(km, qlen, qseq, tlen, tseq, 5, mat, opt->q, opt->e, opt->q2, opt->noncan, zdrop, flag, ez);
|
ksw_exts2_sse(km, qlen, qseq, tlen, tseq, 5, mat, opt->q, opt->e, opt->q2, opt->noncan, zdrop, opt->junc_bonus, flag, junc, ez);
|
||||||
else if (opt->q == opt->q2 && opt->e == opt->e2)
|
else if (opt->q == opt->q2 && opt->e == opt->e2)
|
||||||
ksw_extz2_sse(km, qlen, qseq, tlen, tseq, 5, mat, opt->q, opt->e, w, zdrop, end_bonus, flag, ez);
|
ksw_extz2_sse(km, qlen, qseq, tlen, tseq, 5, mat, opt->q, opt->e, w, zdrop, end_bonus, flag, ez);
|
||||||
else
|
else
|
||||||
@@ -547,7 +566,7 @@ static void mm_align1(void *km, const mm_mapopt_t *opt, const mm_idx_t *mi, int
|
|||||||
{
|
{
|
||||||
int is_sr = !!(opt->flag & MM_F_SR), is_splice = !!(opt->flag & MM_F_SPLICE);
|
int is_sr = !!(opt->flag & MM_F_SR), is_splice = !!(opt->flag & MM_F_SPLICE);
|
||||||
int32_t rid = a[r->as].x<<1>>33, rev = a[r->as].x>>63, as1, cnt1;
|
int32_t rid = a[r->as].x<<1>>33, rev = a[r->as].x>>63, as1, cnt1;
|
||||||
uint8_t *tseq, *qseq;
|
uint8_t *tseq, *qseq, *junc;
|
||||||
int32_t i, l, bw, dropped = 0, extra_flag = 0, rs0, re0, qs0, qe0;
|
int32_t i, l, bw, dropped = 0, extra_flag = 0, rs0, re0, qs0, qe0;
|
||||||
int32_t rs, re, qs, qe;
|
int32_t rs, re, qs, qe;
|
||||||
int32_t rs1, qs1, re1, qe1;
|
int32_t rs1, qs1, re1, qe1;
|
||||||
@@ -666,13 +685,16 @@ static void mm_align1(void *km, const mm_mapopt_t *opt, const mm_idx_t *mi, int
|
|||||||
|
|
||||||
assert(re0 > rs0);
|
assert(re0 > rs0);
|
||||||
tseq = (uint8_t*)kmalloc(km, re0 - rs0);
|
tseq = (uint8_t*)kmalloc(km, re0 - rs0);
|
||||||
|
junc = (uint8_t*)kmalloc(km, re0 - rs0);
|
||||||
|
|
||||||
if (qs > 0 && rs > 0) { // left extension; probably the condition can be changed to "qs > qs0 && rs > rs0"
|
if (qs > 0 && rs > 0) { // left extension; probably the condition can be changed to "qs > qs0 && rs > rs0"
|
||||||
qseq = &qseq0[rev][qs0];
|
qseq = &qseq0[rev][qs0];
|
||||||
mm_idx_getseq(mi, rid, rs0, rs, tseq);
|
mm_idx_getseq(mi, rid, rs0, rs, tseq);
|
||||||
|
mm_idx_bed_junc(mi, rid, rs0, rs, junc);
|
||||||
mm_seq_rev(qs - qs0, qseq);
|
mm_seq_rev(qs - qs0, qseq);
|
||||||
mm_seq_rev(rs - rs0, tseq);
|
mm_seq_rev(rs - rs0, tseq);
|
||||||
mm_align_pair(km, opt, qs - qs0, qseq, rs - rs0, tseq, mat, bw, opt->end_bonus, r->split_inv? opt->zdrop_inv : opt->zdrop, extra_flag|KSW_EZ_EXTZ_ONLY|KSW_EZ_RIGHT|KSW_EZ_REV_CIGAR, ez);
|
mm_seq_rev(rs - rs0, junc);
|
||||||
|
mm_align_pair(km, opt, qs - qs0, qseq, rs - rs0, tseq, junc, mat, bw, opt->end_bonus, r->split_inv? opt->zdrop_inv : opt->zdrop, extra_flag|KSW_EZ_EXTZ_ONLY|KSW_EZ_RIGHT|KSW_EZ_REV_CIGAR, ez);
|
||||||
if (ez->n_cigar > 0) {
|
if (ez->n_cigar > 0) {
|
||||||
mm_append_cigar(r, ez->n_cigar, ez->cigar);
|
mm_append_cigar(r, ez->n_cigar, ez->cigar);
|
||||||
r->p->dp_score += ez->max;
|
r->p->dp_score += ez->max;
|
||||||
@@ -698,6 +720,7 @@ static void mm_align1(void *km, const mm_mapopt_t *opt, const mm_idx_t *mi, int
|
|||||||
// perform alignment
|
// perform alignment
|
||||||
qseq = &qseq0[rev][qs];
|
qseq = &qseq0[rev][qs];
|
||||||
mm_idx_getseq(mi, rid, rs, re, tseq);
|
mm_idx_getseq(mi, rid, rs, re, tseq);
|
||||||
|
mm_idx_bed_junc(mi, rid, rs, re, junc);
|
||||||
if (is_sr) { // perform ungapped alignment
|
if (is_sr) { // perform ungapped alignment
|
||||||
assert(qe - qs == re - rs);
|
assert(qe - qs == re - rs);
|
||||||
ksw_reset_extz(ez);
|
ksw_reset_extz(ez);
|
||||||
@@ -707,11 +730,11 @@ static void mm_align1(void *km, const mm_mapopt_t *opt, const mm_idx_t *mi, int
|
|||||||
}
|
}
|
||||||
ez->cigar = ksw_push_cigar(km, &ez->n_cigar, &ez->m_cigar, ez->cigar, 0, qe - qs);
|
ez->cigar = ksw_push_cigar(km, &ez->n_cigar, &ez->m_cigar, ez->cigar, 0, qe - qs);
|
||||||
} else { // perform normal gapped alignment
|
} else { // perform normal gapped alignment
|
||||||
mm_align_pair(km, opt, qe - qs, qseq, re - rs, tseq, mat, bw1, -1, opt->zdrop, extra_flag|KSW_EZ_APPROX_MAX, ez); // first pass: with approximate Z-drop
|
mm_align_pair(km, opt, qe - qs, qseq, re - rs, tseq, junc, mat, bw1, -1, opt->zdrop, extra_flag|KSW_EZ_APPROX_MAX, ez); // first pass: with approximate Z-drop
|
||||||
}
|
}
|
||||||
// test Z-drop and inversion Z-drop
|
// test Z-drop and inversion Z-drop
|
||||||
if ((zdrop_code = mm_test_zdrop(km, opt, qseq, tseq, ez->n_cigar, ez->cigar, mat)) != 0)
|
if ((zdrop_code = mm_test_zdrop(km, opt, qseq, tseq, ez->n_cigar, ez->cigar, mat)) != 0)
|
||||||
mm_align_pair(km, opt, qe - qs, qseq, re - rs, tseq, mat, bw1, -1, zdrop_code == 2? opt->zdrop_inv : opt->zdrop, extra_flag, ez); // second pass: lift approximate
|
mm_align_pair(km, opt, qe - qs, qseq, re - rs, tseq, junc, mat, bw1, -1, zdrop_code == 2? opt->zdrop_inv : opt->zdrop, extra_flag, ez); // second pass: lift approximate
|
||||||
// update CIGAR
|
// update CIGAR
|
||||||
if (ez->n_cigar > 0)
|
if (ez->n_cigar > 0)
|
||||||
mm_append_cigar(r, ez->n_cigar, ez->cigar);
|
mm_append_cigar(r, ez->n_cigar, ez->cigar);
|
||||||
@@ -737,7 +760,8 @@ static void mm_align1(void *km, const mm_mapopt_t *opt, const mm_idx_t *mi, int
|
|||||||
if (!dropped && qe < qe0 && re < re0) { // right extension
|
if (!dropped && qe < qe0 && re < re0) { // right extension
|
||||||
qseq = &qseq0[rev][qe];
|
qseq = &qseq0[rev][qe];
|
||||||
mm_idx_getseq(mi, rid, re, re0, tseq);
|
mm_idx_getseq(mi, rid, re, re0, tseq);
|
||||||
mm_align_pair(km, opt, qe0 - qe, qseq, re0 - re, tseq, mat, bw, opt->end_bonus, opt->zdrop, extra_flag|KSW_EZ_EXTZ_ONLY, ez);
|
mm_idx_bed_junc(mi, rid, re, re0, junc);
|
||||||
|
mm_align_pair(km, opt, qe0 - qe, qseq, re0 - re, tseq, junc, mat, bw, opt->end_bonus, opt->zdrop, extra_flag|KSW_EZ_EXTZ_ONLY, ez);
|
||||||
if (ez->n_cigar > 0) {
|
if (ez->n_cigar > 0) {
|
||||||
mm_append_cigar(r, ez->n_cigar, ez->cigar);
|
mm_append_cigar(r, ez->n_cigar, ez->cigar);
|
||||||
r->p->dp_score += ez->max;
|
r->p->dp_score += ez->max;
|
||||||
@@ -760,6 +784,7 @@ static void mm_align1(void *km, const mm_mapopt_t *opt, const mm_idx_t *mi, int
|
|||||||
}
|
}
|
||||||
|
|
||||||
kfree(km, tseq);
|
kfree(km, tseq);
|
||||||
|
kfree(km, junc);
|
||||||
}
|
}
|
||||||
|
|
||||||
static int mm_align1_inv(void *km, const mm_mapopt_t *opt, const mm_idx_t *mi, int qlen, uint8_t *qseq0[2], const mm_reg1_t *r1, const mm_reg1_t *r2, mm_reg1_t *r_inv, ksw_extz_t *ez)
|
static int mm_align1_inv(void *km, const mm_mapopt_t *opt, const mm_idx_t *mi, int qlen, uint8_t *qseq0[2], const mm_reg1_t *r1, const mm_reg1_t *r2, mm_reg1_t *r_inv, ksw_extz_t *ez)
|
||||||
@@ -793,7 +818,7 @@ static int mm_align1_inv(void *km, const mm_mapopt_t *opt, const mm_idx_t *mi, i
|
|||||||
mm_seq_rev(tl, tseq);
|
mm_seq_rev(tl, tseq);
|
||||||
if (score < opt->min_dp_max) goto end_align1_inv;
|
if (score < opt->min_dp_max) goto end_align1_inv;
|
||||||
q_off = ql - (q_off + 1), t_off = tl - (t_off + 1);
|
q_off = ql - (q_off + 1), t_off = tl - (t_off + 1);
|
||||||
mm_align_pair(km, opt, ql - q_off, qseq + q_off, tl - t_off, tseq + t_off, mat, (int)(opt->bw * 1.5), -1, opt->zdrop, KSW_EZ_EXTZ_ONLY, ez);
|
mm_align_pair(km, opt, ql - q_off, qseq + q_off, tl - t_off, tseq + t_off, 0, mat, (int)(opt->bw * 1.5), -1, opt->zdrop, KSW_EZ_EXTZ_ONLY, ez);
|
||||||
if (ez->n_cigar == 0) goto end_align1_inv; // should never be here
|
if (ez->n_cigar == 0) goto end_align1_inv; // should never be here
|
||||||
mm_append_cigar(r_inv, ez->n_cigar, ez->cigar);
|
mm_append_cigar(r_inv, ez->n_cigar, ez->cigar);
|
||||||
r_inv->p->dp_score = ez->max;
|
r_inv->p->dp_score = ez->max;
|
||||||
|
|||||||
@@ -19,7 +19,7 @@ static inline int ilog2_32(uint32_t v)
|
|||||||
return (t = v>>8) ? 8 + LogTable256[t] : LogTable256[v];
|
return (t = v>>8) ? 8 + LogTable256[t] : LogTable256[v];
|
||||||
}
|
}
|
||||||
|
|
||||||
mm128_t *mm_chain_dp(int max_dist_x, int max_dist_y, int bw, int max_skip, int min_cnt, int min_sc, int is_cdna, int n_segs, int64_t n, mm128_t *a, int *n_u_, uint64_t **_u, void *km)
|
mm128_t *mm_chain_dp(int max_dist_x, int max_dist_y, int bw, int max_skip, int max_iter, int min_cnt, int min_sc, int is_cdna, int n_segs, int64_t n, mm128_t *a, int *n_u_, uint64_t **_u, void *km)
|
||||||
{ // TODO: make sure this works when n has more than 32 bits
|
{ // TODO: make sure this works when n has more than 32 bits
|
||||||
int32_t k, *f, *p, *t, *v, n_u, n_v;
|
int32_t k, *f, *p, *t, *v, n_u, n_v;
|
||||||
int64_t i, j, st = 0;
|
int64_t i, j, st = 0;
|
||||||
@@ -28,6 +28,10 @@ mm128_t *mm_chain_dp(int max_dist_x, int max_dist_y, int bw, int max_skip, int m
|
|||||||
mm128_t *b, *w;
|
mm128_t *b, *w;
|
||||||
|
|
||||||
if (_u) *_u = 0, *n_u_ = 0;
|
if (_u) *_u = 0, *n_u_ = 0;
|
||||||
|
if (n == 0 || a == 0) {
|
||||||
|
kfree(km, a);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
f = (int32_t*)kmalloc(km, n * 4);
|
f = (int32_t*)kmalloc(km, n * 4);
|
||||||
p = (int32_t*)kmalloc(km, n * 4);
|
p = (int32_t*)kmalloc(km, n * 4);
|
||||||
t = (int32_t*)kmalloc(km, n * 4);
|
t = (int32_t*)kmalloc(km, n * 4);
|
||||||
@@ -45,6 +49,7 @@ mm128_t *mm_chain_dp(int max_dist_x, int max_dist_y, int bw, int max_skip, int m
|
|||||||
int32_t max_f = q_span, n_skip = 0, min_d;
|
int32_t max_f = q_span, n_skip = 0, min_d;
|
||||||
int32_t sidi = (a[i].y & MM_SEED_SEG_MASK) >> MM_SEED_SEG_SHIFT;
|
int32_t sidi = (a[i].y & MM_SEED_SEG_MASK) >> MM_SEED_SEG_SHIFT;
|
||||||
while (st < i && ri > a[st].x + max_dist_x) ++st;
|
while (st < i && ri > a[st].x + max_dist_x) ++st;
|
||||||
|
if (i - st > max_iter) st = i - max_iter;
|
||||||
for (j = i - 1; j >= st; --j) {
|
for (j = i - 1; j >= st; --j) {
|
||||||
int64_t dr = ri - a[j].x;
|
int64_t dr = ri - a[j].x;
|
||||||
int32_t dq = qi - (int32_t)a[j].y, dd, sc, log_dd;
|
int32_t dq = qi - (int32_t)a[j].y, dd, sc, log_dd;
|
||||||
@@ -150,8 +155,8 @@ mm128_t *mm_chain_dp(int max_dist_x, int max_dist_y, int bw, int max_skip, int m
|
|||||||
memcpy(&a[k], &b[w[i].y>>32], n * sizeof(mm128_t));
|
memcpy(&a[k], &b[w[i].y>>32], n * sizeof(mm128_t));
|
||||||
k += n;
|
k += n;
|
||||||
}
|
}
|
||||||
memcpy(u, u2, n_u * 8);
|
if (n_u) memcpy(u, u2, n_u * 8);
|
||||||
memcpy(b, a, k * sizeof(mm128_t)); // write _a_ to _b_ and deallocate _a_ because _a_ is oversized, sometimes a lot
|
if (k) memcpy(b, a, k * sizeof(mm128_t)); // write _a_ to _b_ and deallocate _a_ because _a_ is oversized, sometimes a lot
|
||||||
kfree(km, a); kfree(km, w); kfree(km, u2);
|
kfree(km, a); kfree(km, w); kfree(km, u2);
|
||||||
return b;
|
return b;
|
||||||
}
|
}
|
||||||
|
|||||||
+2
-2
@@ -31,8 +31,8 @@ To acquire the data used in this cookbook and to install minimap2 and paftools,
|
|||||||
please follow the command lines below:
|
please follow the command lines below:
|
||||||
```sh
|
```sh
|
||||||
# install minimap2 executables
|
# install minimap2 executables
|
||||||
curl -L https://github.com/lh3/minimap2/releases/download/v2.15/minimap2-2.15_x64-linux.tar.bz2 | tar jxf -
|
curl -L https://github.com/lh3/minimap2/releases/download/v2.17/minimap2-2.17_x64-linux.tar.bz2 | tar jxf -
|
||||||
cp minimap2-2.15_x64-linux/{minimap2,k8,paftools.js} . # copy executables
|
cp minimap2-2.17_x64-linux/{minimap2,k8,paftools.js} . # copy executables
|
||||||
export PATH="$PATH:"`pwd` # put the current directory on PATH
|
export PATH="$PATH:"`pwd` # put the current directory on PATH
|
||||||
# download example datasets
|
# download example datasets
|
||||||
curl -L https://github.com/lh3/minimap2/releases/download/v2.10/cookbook-data.tgz | tar zxf -
|
curl -L https://github.com/lh3/minimap2/releases/download/v2.10/cookbook-data.tgz | tar zxf -
|
||||||
|
|||||||
@@ -35,6 +35,8 @@ int main(int argc, char *argv[])
|
|||||||
while ((mi = mm_idx_reader_read(r, n_threads)) != 0) { // traverse each part of the index
|
while ((mi = mm_idx_reader_read(r, n_threads)) != 0) { // traverse each part of the index
|
||||||
mm_mapopt_update(&mopt, mi); // this sets the maximum minimizer occurrence; TODO: set a better default in mm_mapopt_init()!
|
mm_mapopt_update(&mopt, mi); // this sets the maximum minimizer occurrence; TODO: set a better default in mm_mapopt_init()!
|
||||||
mm_tbuf_t *tbuf = mm_tbuf_init(); // thread buffer; for multi-threading, allocate one tbuf for each thread
|
mm_tbuf_t *tbuf = mm_tbuf_init(); // thread buffer; for multi-threading, allocate one tbuf for each thread
|
||||||
|
gzrewind(f);
|
||||||
|
kseq_rewind(ks);
|
||||||
while (kseq_read(ks) >= 0) { // each kseq_read() call reads one query sequence
|
while (kseq_read(ks) >= 0) { // each kseq_read() call reads one query sequence
|
||||||
mm_reg1_t *reg;
|
mm_reg1_t *reg;
|
||||||
int j, i, n_reg;
|
int j, i, n_reg;
|
||||||
|
|||||||
@@ -79,11 +79,11 @@ static char *mm_escape(char *s)
|
|||||||
return s;
|
return s;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void sam_write_rg_line(kstring_t *str, const char *s)
|
static int sam_write_rg_line(kstring_t *str, const char *s)
|
||||||
{
|
{
|
||||||
char *p, *q, *r, *rg_line = 0;
|
char *p, *q, *r, *rg_line = 0;
|
||||||
memset(mm_rg_id, 0, 256);
|
memset(mm_rg_id, 0, 256);
|
||||||
if (s == 0) return;
|
if (s == 0) return 0;
|
||||||
if (strstr(s, "@RG") != s) {
|
if (strstr(s, "@RG") != s) {
|
||||||
if (mm_verbose >= 1) fprintf(stderr, "[ERROR] the read group line is not started with @RG\n");
|
if (mm_verbose >= 1) fprintf(stderr, "[ERROR] the read group line is not started with @RG\n");
|
||||||
goto err_set_rg;
|
goto err_set_rg;
|
||||||
@@ -108,20 +108,23 @@ static void sam_write_rg_line(kstring_t *str, const char *s)
|
|||||||
for (q = p, r = mm_rg_id; *q && *q != '\t' && *q != '\n'; ++q)
|
for (q = p, r = mm_rg_id; *q && *q != '\t' && *q != '\n'; ++q)
|
||||||
*r++ = *q;
|
*r++ = *q;
|
||||||
mm_sprintf_lite(str, "%s\n", rg_line);
|
mm_sprintf_lite(str, "%s\n", rg_line);
|
||||||
|
return 0;
|
||||||
|
|
||||||
err_set_rg:
|
err_set_rg:
|
||||||
free(rg_line);
|
free(rg_line);
|
||||||
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
void mm_write_sam_hdr(const mm_idx_t *idx, const char *rg, const char *ver, int argc, char *argv[])
|
int mm_write_sam_hdr(const mm_idx_t *idx, const char *rg, const char *ver, int argc, char *argv[])
|
||||||
{
|
{
|
||||||
kstring_t str = {0,0,0};
|
kstring_t str = {0,0,0};
|
||||||
|
int ret = 0;
|
||||||
if (idx) {
|
if (idx) {
|
||||||
uint32_t i;
|
uint32_t i;
|
||||||
for (i = 0; i < idx->n_seq; ++i)
|
for (i = 0; i < idx->n_seq; ++i)
|
||||||
mm_sprintf_lite(&str, "@SQ\tSN:%s\tLN:%d\n", idx->seq[i].name, idx->seq[i].len);
|
mm_sprintf_lite(&str, "@SQ\tSN:%s\tLN:%d\n", idx->seq[i].name, idx->seq[i].len);
|
||||||
}
|
}
|
||||||
if (rg) sam_write_rg_line(&str, rg);
|
if (rg) ret = sam_write_rg_line(&str, rg);
|
||||||
mm_sprintf_lite(&str, "@PG\tID:minimap2\tPN:minimap2");
|
mm_sprintf_lite(&str, "@PG\tID:minimap2\tPN:minimap2");
|
||||||
if (ver) mm_sprintf_lite(&str, "\tVN:%s", ver);
|
if (ver) mm_sprintf_lite(&str, "\tVN:%s", ver);
|
||||||
if (argc > 1) {
|
if (argc > 1) {
|
||||||
@@ -132,6 +135,7 @@ void mm_write_sam_hdr(const mm_idx_t *idx, const char *rg, const char *ver, int
|
|||||||
}
|
}
|
||||||
mm_err_puts(str.s);
|
mm_err_puts(str.s);
|
||||||
free(str.s);
|
free(str.s);
|
||||||
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void write_cs_core(kstring_t *s, const uint8_t *tseq, const uint8_t *qseq, const mm_reg1_t *r, char *tmp, int no_iden, int write_tag)
|
static void write_cs_core(kstring_t *s, const uint8_t *tseq, const uint8_t *qseq, const mm_reg1_t *r, char *tmp, int no_iden, int write_tag)
|
||||||
@@ -270,7 +274,7 @@ double mm_event_identity(const mm_reg1_t *r)
|
|||||||
if (op == 1 || op == 2)
|
if (op == 1 || op == 2)
|
||||||
++n_gapo, n_gap += len;
|
++n_gapo, n_gap += len;
|
||||||
}
|
}
|
||||||
return (double)r->mlen / (r->blen - r->p->n_ambi - n_gap + n_gapo);
|
return (double)r->mlen / (r->blen - n_gap + n_gapo);
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline void write_tags(kstring_t *s, const mm_reg1_t *r)
|
static inline void write_tags(kstring_t *s, const mm_reg1_t *r)
|
||||||
@@ -301,11 +305,12 @@ static inline void write_tags(kstring_t *s, const mm_reg1_t *r)
|
|||||||
if (r->split) mm_sprintf_lite(s, "\tzd:i:%d", r->split);
|
if (r->split) mm_sprintf_lite(s, "\tzd:i:%d", r->split);
|
||||||
}
|
}
|
||||||
|
|
||||||
void mm_write_paf(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, const mm_reg1_t *r, void *km, int opt_flag)
|
void mm_write_paf3(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, const mm_reg1_t *r, void *km, int opt_flag, int rep_len)
|
||||||
{
|
{
|
||||||
s->l = 0;
|
s->l = 0;
|
||||||
if (r == 0) {
|
if (r == 0) {
|
||||||
mm_sprintf_lite(s, "%s\t%d", t->name, t->l_seq);
|
mm_sprintf_lite(s, "%s\t%d\t0\t0\t*\t*\t0\t0\t0\t0\t0\t0", t->name, t->l_seq);
|
||||||
|
if (rep_len >= 0) mm_sprintf_lite(s, "\trl:i:%d", rep_len);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
mm_sprintf_lite(s, "%s\t%d\t%d\t%d\t%c\t", t->name, t->l_seq, r->qs, r->qe, "+-"[r->rev]);
|
mm_sprintf_lite(s, "%s\t%d\t%d\t%d\t%c\t", t->name, t->l_seq, r->qs, r->qe, "+-"[r->rev]);
|
||||||
@@ -315,6 +320,7 @@ void mm_write_paf(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, const m
|
|||||||
mm_sprintf_lite(s, "\t%d\t%d", r->mlen, r->blen);
|
mm_sprintf_lite(s, "\t%d\t%d", r->mlen, r->blen);
|
||||||
mm_sprintf_lite(s, "\t%d", r->mapq);
|
mm_sprintf_lite(s, "\t%d", r->mapq);
|
||||||
write_tags(s, r);
|
write_tags(s, r);
|
||||||
|
if (rep_len >= 0) mm_sprintf_lite(s, "\trl:i:%d", rep_len);
|
||||||
if (r->p && (opt_flag & MM_F_OUT_CG)) {
|
if (r->p && (opt_flag & MM_F_OUT_CG)) {
|
||||||
uint32_t k;
|
uint32_t k;
|
||||||
mm_sprintf_lite(s, "\tcg:Z:");
|
mm_sprintf_lite(s, "\tcg:Z:");
|
||||||
@@ -327,6 +333,11 @@ void mm_write_paf(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, const m
|
|||||||
mm_sprintf_lite(s, "\t%s", t->comment);
|
mm_sprintf_lite(s, "\t%s", t->comment);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void mm_write_paf(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, const mm_reg1_t *r, void *km, int opt_flag)
|
||||||
|
{
|
||||||
|
mm_write_paf3(s, mi, t, r, km, opt_flag, -1);
|
||||||
|
}
|
||||||
|
|
||||||
static void sam_write_sq(kstring_t *s, char *seq, int l, int rev, int comp)
|
static void sam_write_sq(kstring_t *s, char *seq, int l, int rev, int comp)
|
||||||
{
|
{
|
||||||
extern unsigned char seq_comp_table[256];
|
extern unsigned char seq_comp_table[256];
|
||||||
@@ -368,6 +379,7 @@ static void write_sam_cigar(kstring_t *s, int sam_flag, int in_tag, int qlen, co
|
|||||||
if (clip_len[1]) mm_sprintf_lite(s, ",%u", clip_len[1]<<4|clip_char);
|
if (clip_len[1]) mm_sprintf_lite(s, ",%u", clip_len[1]<<4|clip_char);
|
||||||
} else {
|
} else {
|
||||||
int clip_char = (sam_flag&0x800) && !(opt_flag&MM_F_SOFTCLIP)? 'H' : 'S';
|
int clip_char = (sam_flag&0x800) && !(opt_flag&MM_F_SOFTCLIP)? 'H' : 'S';
|
||||||
|
assert(clip_len[0] < qlen && clip_len[1] < qlen);
|
||||||
if (clip_len[0]) mm_sprintf_lite(s, "%d%c", clip_len[0], clip_char);
|
if (clip_len[0]) mm_sprintf_lite(s, "%d%c", clip_len[0], clip_char);
|
||||||
for (k = 0; k < r->p->n_cigar; ++k)
|
for (k = 0; k < r->p->n_cigar; ++k)
|
||||||
mm_sprintf_lite(s, "%d%c", r->p->cigar[k]>>4, "MIDNSHP=XB"[r->p->cigar[k]&0xf]);
|
mm_sprintf_lite(s, "%d%c", r->p->cigar[k]>>4, "MIDNSHP=XB"[r->p->cigar[k]&0xf]);
|
||||||
@@ -376,7 +388,7 @@ static void write_sam_cigar(kstring_t *s, int sam_flag, int in_tag, int qlen, co
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void mm_write_sam2(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, int seg_idx, int reg_idx, int n_seg, const int *n_regss, const mm_reg1_t *const* regss, void *km, int opt_flag)
|
void mm_write_sam3(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, int seg_idx, int reg_idx, int n_seg, const int *n_regss, const mm_reg1_t *const* regss, void *km, int opt_flag, int rep_len)
|
||||||
{
|
{
|
||||||
const int max_bam_cigar_op = 65535;
|
const int max_bam_cigar_op = 65535;
|
||||||
int flag, n_regs = n_regss[seg_idx], cigar_in_tag = 0;
|
int flag, n_regs = n_regss[seg_idx], cigar_in_tag = 0;
|
||||||
@@ -452,17 +464,17 @@ void mm_write_sam2(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, int se
|
|||||||
int tlen = 0;
|
int tlen = 0;
|
||||||
if (this_rid >= 0 && r_next) {
|
if (this_rid >= 0 && r_next) {
|
||||||
if (this_rid == r_next->rid) {
|
if (this_rid == r_next->rid) {
|
||||||
int this_pos5 = r && r->rev? r->re - 1 : this_pos;
|
if (r) {
|
||||||
int next_pos5 = r_next->rev? r_next->re - 1 : r_next->rs;
|
int this_pos5 = r->rev? r->re - 1 : this_pos;
|
||||||
tlen = next_pos5 - this_pos5;
|
int next_pos5 = r_next->rev? r_next->re - 1 : r_next->rs;
|
||||||
|
tlen = next_pos5 - this_pos5;
|
||||||
|
}
|
||||||
mm_sprintf_lite(s, "\t=\t");
|
mm_sprintf_lite(s, "\t=\t");
|
||||||
} else mm_sprintf_lite(s, "\t%s\t", mi->seq[r_next->rid].name);
|
} else mm_sprintf_lite(s, "\t%s\t", mi->seq[r_next->rid].name);
|
||||||
mm_sprintf_lite(s, "%d\t", r_next->rs + 1);
|
mm_sprintf_lite(s, "%d\t", r_next->rs + 1);
|
||||||
} else if (r_next) { // && this_rid < 0
|
} else if (r_next) { // && this_rid < 0
|
||||||
mm_sprintf_lite(s, "\t%s\t%d\t", mi->seq[r_next->rid].name, r_next->rs + 1);
|
mm_sprintf_lite(s, "\t%s\t%d\t", mi->seq[r_next->rid].name, r_next->rs + 1);
|
||||||
} else if (this_rid >= 0) { // && r_next == NULL
|
} else if (this_rid >= 0) { // && r_next == NULL
|
||||||
int this_pos5 = this_rev? r->re - 1 : this_pos; // this_rev is only true when r != NULL
|
|
||||||
tlen = this_pos - this_pos5; // next_pos5 will be this_pos
|
|
||||||
mm_sprintf_lite(s, "\t=\t%d\t", this_pos + 1); // next segment will take r's coordinate
|
mm_sprintf_lite(s, "\t=\t%d\t", this_pos + 1); // next segment will take r's coordinate
|
||||||
} else mm_sprintf_lite(s, "\t*\t0\t"); // neither has coordinates
|
} else mm_sprintf_lite(s, "\t*\t0\t"); // neither has coordinates
|
||||||
if (tlen > 0) ++tlen;
|
if (tlen > 0) ++tlen;
|
||||||
@@ -527,6 +539,7 @@ void mm_write_sam2(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, int se
|
|||||||
if (cigar_in_tag)
|
if (cigar_in_tag)
|
||||||
write_sam_cigar(s, flag, 1, t->l_seq, r, opt_flag);
|
write_sam_cigar(s, flag, 1, t->l_seq, r, opt_flag);
|
||||||
}
|
}
|
||||||
|
if (rep_len >= 0) mm_sprintf_lite(s, "\trl:i:%d", rep_len);
|
||||||
|
|
||||||
if ((opt_flag & MM_F_COPY_COMMENT) && t->comment)
|
if ((opt_flag & MM_F_COPY_COMMENT) && t->comment)
|
||||||
mm_sprintf_lite(s, "\t%s", t->comment);
|
mm_sprintf_lite(s, "\t%s", t->comment);
|
||||||
@@ -534,6 +547,11 @@ void mm_write_sam2(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, int se
|
|||||||
s->s[s->l] = 0; // we always have room for an extra byte (see str_enlarge)
|
s->s[s->l] = 0; // we always have room for an extra byte (see str_enlarge)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void mm_write_sam2(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, int seg_idx, int reg_idx, int n_seg, const int *n_regss, const mm_reg1_t *const* regss, void *km, int opt_flag)
|
||||||
|
{
|
||||||
|
mm_write_sam3(s, mi, t, seg_idx, reg_idx, n_seg, n_regss, regss, km, opt_flag, -1);
|
||||||
|
}
|
||||||
|
|
||||||
void mm_write_sam(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, const mm_reg1_t *r, int n_regs, const mm_reg1_t *regs)
|
void mm_write_sam(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, const mm_reg1_t *r, int n_regs, const mm_reg1_t *regs)
|
||||||
{
|
{
|
||||||
int i;
|
int i;
|
||||||
|
|||||||
@@ -449,6 +449,7 @@ void mm_set_mapq(void *km, int n_regs, mm_reg1_t *regs, int min_chain_sc, int ma
|
|||||||
int64_t sum_sc = 0;
|
int64_t sum_sc = 0;
|
||||||
float uniq_ratio;
|
float uniq_ratio;
|
||||||
int i;
|
int i;
|
||||||
|
if (n_regs == 0) return;
|
||||||
for (i = 0; i < n_regs; ++i)
|
for (i = 0; i < n_regs; ++i)
|
||||||
if (regs[i].parent == regs[i].id)
|
if (regs[i].parent == regs[i].id)
|
||||||
sum_sc += regs[i].score;
|
sum_sc += regs[i].score;
|
||||||
|
|||||||
@@ -31,6 +31,16 @@ typedef struct mm_idx_bucket_s {
|
|||||||
void *h; // hash table indexing _p_ and minimizers appearing once
|
void *h; // hash table indexing _p_ and minimizers appearing once
|
||||||
} mm_idx_bucket_t;
|
} mm_idx_bucket_t;
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
int32_t st, en, max; // max is not used for now
|
||||||
|
int32_t score:30, strand:2;
|
||||||
|
} mm_idx_intv1_t;
|
||||||
|
|
||||||
|
typedef struct mm_idx_intv_s {
|
||||||
|
int32_t n, m;
|
||||||
|
mm_idx_intv1_t *a;
|
||||||
|
} mm_idx_intv_t;
|
||||||
|
|
||||||
mm_idx_t *mm_idx_init(int w, int k, int b, int flag)
|
mm_idx_t *mm_idx_init(int w, int k, int b, int flag)
|
||||||
{
|
{
|
||||||
mm_idx_t *mi;
|
mm_idx_t *mi;
|
||||||
@@ -55,6 +65,11 @@ void mm_idx_destroy(mm_idx_t *mi)
|
|||||||
kh_destroy(idx, (idxhash_t*)mi->B[i].h);
|
kh_destroy(idx, (idxhash_t*)mi->B[i].h);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if (mi->I) {
|
||||||
|
for (i = 0; i < mi->n_seq; ++i)
|
||||||
|
free(mi->I[i].a);
|
||||||
|
free(mi->I);
|
||||||
|
}
|
||||||
if (!mi->km) {
|
if (!mi->km) {
|
||||||
for (i = 0; i < mi->n_seq; ++i)
|
for (i = 0; i < mi->n_seq; ++i)
|
||||||
free(mi->seq[i].name);
|
free(mi->seq[i].name);
|
||||||
@@ -102,8 +117,8 @@ void mm_idx_stat(const mm_idx_t *mi)
|
|||||||
if (kh_key(h, k)&1) ++n1;
|
if (kh_key(h, k)&1) ++n1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
fprintf(stderr, "[M::%s::%.3f*%.2f] distinct minimizers: %d (%.2f%% are singletons); average occurrences: %.3lf; average spacing: %.3lf\n",
|
fprintf(stderr, "[M::%s::%.3f*%.2f] distinct minimizers: %d (%.2f%% are singletons); average occurrences: %.3lf; average spacing: %.3lf; total length: %ld\n",
|
||||||
__func__, realtime() - mm_realtime0, cputime() / (realtime() - mm_realtime0), n, 100.0*n1/n, (double)sum / n, (double)len / sum);
|
__func__, realtime() - mm_realtime0, cputime() / (realtime() - mm_realtime0), n, 100.0*n1/n, (double)sum / n, (double)len / sum, (long)len);
|
||||||
}
|
}
|
||||||
|
|
||||||
int mm_idx_index_name(mm_idx_t *mi)
|
int mm_idx_index_name(mm_idx_t *mi)
|
||||||
@@ -585,3 +600,127 @@ int mm_idx_reader_eof(const mm_idx_reader_t *r) // TODO: in extremely rare cases
|
|||||||
{
|
{
|
||||||
return r->is_idx? (feof(r->fp.idx) || ftell(r->fp.idx) == r->idx_size) : mm_bseq_eof(r->fp.seq);
|
return r->is_idx? (feof(r->fp.idx) || ftell(r->fp.idx) == r->idx_size) : mm_bseq_eof(r->fp.seq);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#include <ctype.h>
|
||||||
|
#include <zlib.h>
|
||||||
|
#include "ksort.h"
|
||||||
|
#include "kseq.h"
|
||||||
|
KSTREAM_DECLARE(gzFile, gzread)
|
||||||
|
|
||||||
|
#define sort_key_bed(a) ((a).st)
|
||||||
|
KRADIX_SORT_INIT(bed, mm_idx_intv1_t, sort_key_bed, 4)
|
||||||
|
|
||||||
|
mm_idx_intv_t *mm_idx_read_bed(const mm_idx_t *mi, const char *fn, int read_junc)
|
||||||
|
{
|
||||||
|
gzFile fp;
|
||||||
|
kstream_t *ks;
|
||||||
|
kstring_t str = {0,0,0};
|
||||||
|
mm_idx_intv_t *I;
|
||||||
|
|
||||||
|
fp = fn && strcmp(fn, "-")? gzopen(fn, "r") : gzdopen(fileno(stdin), "r");
|
||||||
|
if (fp == 0) return 0;
|
||||||
|
I = (mm_idx_intv_t*)calloc(mi->n_seq, sizeof(*I));
|
||||||
|
ks = ks_init(fp);
|
||||||
|
while (ks_getuntil(ks, KS_SEP_LINE, &str, 0) >= 0) {
|
||||||
|
mm_idx_intv_t *r;
|
||||||
|
mm_idx_intv1_t t = {-1,-1,-1,-1,0};
|
||||||
|
char *p, *q, *bl, *bs;
|
||||||
|
int32_t i, id = -1, n_blk = 0;
|
||||||
|
for (p = q = str.s, i = 0;; ++p) {
|
||||||
|
if (*p == 0 || isspace(*p)) {
|
||||||
|
int32_t c = *p;
|
||||||
|
*p = 0;
|
||||||
|
if (i == 0) { // chr
|
||||||
|
id = mm_idx_name2id(mi, q);
|
||||||
|
if (id < 0) break; // unknown name; TODO: throw a warning
|
||||||
|
} else if (i == 1) { // start
|
||||||
|
t.st = atol(q); // TODO: watch out integer overflow!
|
||||||
|
if (t.st < 0) break;
|
||||||
|
} else if (i == 2) { // end
|
||||||
|
t.en = atol(q);
|
||||||
|
if (t.en < 0) break;
|
||||||
|
} else if (i == 4) { // BED score
|
||||||
|
t.score = atol(q);
|
||||||
|
} else if (i == 5) { // strand
|
||||||
|
t.strand = *q == '+'? 1 : *q == '-'? -1 : 0;
|
||||||
|
} else if (i == 9) {
|
||||||
|
if (!isdigit(*q)) break;
|
||||||
|
n_blk = atol(q);
|
||||||
|
} else if (i == 10) {
|
||||||
|
bl = q;
|
||||||
|
} else if (i == 11) {
|
||||||
|
bs = q;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if (c == 0) break;
|
||||||
|
++i, q = p + 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (id < 0 || t.st < 0 || t.st >= t.en) continue;
|
||||||
|
r = &I[id];
|
||||||
|
if (i >= 11 && read_junc) { // BED12
|
||||||
|
int32_t st, sz, en;
|
||||||
|
st = strtol(bs, &bs, 10); ++bs;
|
||||||
|
sz = strtol(bl, &bl, 10); ++bl;
|
||||||
|
en = t.st + st + sz;
|
||||||
|
for (i = 1; i < n_blk; ++i) {
|
||||||
|
mm_idx_intv1_t s = t;
|
||||||
|
if (r->n == r->m) {
|
||||||
|
r->m = r->m? r->m + (r->m>>1) : 16;
|
||||||
|
r->a = (mm_idx_intv1_t*)realloc(r->a, sizeof(*r->a) * r->m);
|
||||||
|
}
|
||||||
|
st = strtol(bs, &bs, 10); ++bs;
|
||||||
|
sz = strtol(bl, &bl, 10); ++bl;
|
||||||
|
s.st = en, s.en = t.st + st;
|
||||||
|
en = t.st + st + sz;
|
||||||
|
if (s.en > s.st) r->a[r->n++] = s;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if (r->n == r->m) {
|
||||||
|
r->m = r->m? r->m + (r->m>>1) : 16;
|
||||||
|
r->a = (mm_idx_intv1_t*)realloc(r->a, sizeof(*r->a) * r->m);
|
||||||
|
}
|
||||||
|
r->a[r->n++] = t;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
free(str.s);
|
||||||
|
ks_destroy(ks);
|
||||||
|
gzclose(fp);
|
||||||
|
return I;
|
||||||
|
}
|
||||||
|
|
||||||
|
int mm_idx_bed_read(mm_idx_t *mi, const char *fn, int read_junc)
|
||||||
|
{
|
||||||
|
int32_t i;
|
||||||
|
if (mi->h == 0) mm_idx_index_name(mi);
|
||||||
|
mi->I = mm_idx_read_bed(mi, fn, read_junc);
|
||||||
|
if (mi->I == 0) return -1;
|
||||||
|
for (i = 0; i < mi->n_seq; ++i) // TODO: eliminate redundant intervals
|
||||||
|
radix_sort_bed(mi->I[i].a, mi->I[i].a + mi->I[i].n);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int mm_idx_bed_junc(const mm_idx_t *mi, int32_t ctg, int32_t st, int32_t en, uint8_t *s)
|
||||||
|
{
|
||||||
|
int32_t i, left, right;
|
||||||
|
mm_idx_intv_t *r;
|
||||||
|
memset(s, 0, en - st);
|
||||||
|
if (mi->I == 0 || ctg < 0 || ctg >= mi->n_seq) return -1;
|
||||||
|
r = &mi->I[ctg];
|
||||||
|
left = 0, right = r->n;
|
||||||
|
while (right > left) {
|
||||||
|
int32_t mid = left + ((right - left) >> 1);
|
||||||
|
if (r->a[mid].st >= st) right = mid;
|
||||||
|
else left = mid + 1;
|
||||||
|
}
|
||||||
|
for (i = left; i < r->n; ++i) {
|
||||||
|
if (st <= r->a[i].st && en >= r->a[i].en && r->a[i].strand != 0) {
|
||||||
|
if (r->a[i].strand > 0) {
|
||||||
|
s[r->a[i].st - st] |= 1, s[r->a[i].en - 1 - st] |= 2;
|
||||||
|
} else {
|
||||||
|
s[r->a[i].st - st] |= 8, s[r->a[i].en - 1 - st] |= 4;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return left;
|
||||||
|
}
|
||||||
|
|||||||
@@ -18,15 +18,14 @@
|
|||||||
* | | | |
|
* | | | |
|
||||||
* p=p->ptr->ptr->ptr->ptr p->ptr p->ptr->ptr p->ptr->ptr->ptr
|
* p=p->ptr->ptr->ptr->ptr p->ptr p->ptr->ptr p->ptr->ptr->ptr
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#define MIN_CORE_SIZE 0x80000
|
|
||||||
|
|
||||||
typedef struct header_t {
|
typedef struct header_t {
|
||||||
size_t size;
|
size_t size;
|
||||||
struct header_t *ptr;
|
struct header_t *ptr;
|
||||||
} header_t;
|
} header_t;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
|
void *par;
|
||||||
|
size_t min_core_size;
|
||||||
header_t base, *loop_head, *core_head; /* base is a zero-sized block always kept in the loop */
|
header_t base, *loop_head, *core_head; /* base is a zero-sized block always kept in the loop */
|
||||||
} kmem_t;
|
} kmem_t;
|
||||||
|
|
||||||
@@ -36,31 +35,39 @@ static void panic(const char *s)
|
|||||||
abort();
|
abort();
|
||||||
}
|
}
|
||||||
|
|
||||||
void *km_init(void)
|
void *km_init2(void *km_par, size_t min_core_size)
|
||||||
{
|
{
|
||||||
return calloc(1, sizeof(kmem_t));
|
kmem_t *km;
|
||||||
|
km = (kmem_t*)kcalloc(km_par, 1, sizeof(kmem_t));
|
||||||
|
km->par = km_par;
|
||||||
|
km->min_core_size = min_core_size > 0? min_core_size : 0x80000;
|
||||||
|
return (void*)km;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void *km_init(void) { return km_init2(0, 0); }
|
||||||
|
|
||||||
void km_destroy(void *_km)
|
void km_destroy(void *_km)
|
||||||
{
|
{
|
||||||
kmem_t *km = (kmem_t*)_km;
|
kmem_t *km = (kmem_t*)_km;
|
||||||
|
void *km_par;
|
||||||
header_t *p, *q;
|
header_t *p, *q;
|
||||||
if (km == NULL) return;
|
if (km == NULL) return;
|
||||||
|
km_par = km->par;
|
||||||
for (p = km->core_head; p != NULL;) {
|
for (p = km->core_head; p != NULL;) {
|
||||||
q = p->ptr;
|
q = p->ptr;
|
||||||
free(p);
|
kfree(km_par, p);
|
||||||
p = q;
|
p = q;
|
||||||
}
|
}
|
||||||
free(km);
|
kfree(km_par, km);
|
||||||
}
|
}
|
||||||
|
|
||||||
static header_t *morecore(kmem_t *km, size_t nu)
|
static header_t *morecore(kmem_t *km, size_t nu)
|
||||||
{
|
{
|
||||||
header_t *q;
|
header_t *q;
|
||||||
size_t bytes, *p;
|
size_t bytes, *p;
|
||||||
nu = (nu + 1 + (MIN_CORE_SIZE - 1)) / MIN_CORE_SIZE * MIN_CORE_SIZE; /* the first +1 for core header */
|
nu = (nu + 1 + (km->min_core_size - 1)) / km->min_core_size * km->min_core_size; /* the first +1 for core header */
|
||||||
bytes = nu * sizeof(header_t);
|
bytes = nu * sizeof(header_t);
|
||||||
q = (header_t*)malloc(bytes);
|
q = (header_t*)kmalloc(km->par, bytes);
|
||||||
if (!q) panic("[morecore] insufficient memory");
|
if (!q) panic("[morecore] insufficient memory");
|
||||||
q->ptr = km->core_head, q->size = nu, km->core_head = q;
|
q->ptr = km->core_head, q->size = nu, km->core_head = q;
|
||||||
p = (size_t*)(q + 1);
|
p = (size_t*)(q + 1);
|
||||||
@@ -125,7 +132,7 @@ void *kmalloc(void *_km, size_t n_bytes)
|
|||||||
|
|
||||||
if (n_bytes == 0) return 0;
|
if (n_bytes == 0) return 0;
|
||||||
if (km == NULL) return malloc(n_bytes);
|
if (km == NULL) return malloc(n_bytes);
|
||||||
n_units = (n_bytes + sizeof(size_t) + sizeof(header_t) - 1) / sizeof(header_t) + 1;
|
n_units = (n_bytes + sizeof(size_t) + sizeof(header_t) - 1) / sizeof(header_t); /* header+n_bytes requires at least this number of units */
|
||||||
|
|
||||||
if (!(q = km->loop_head)) /* the first time when kmalloc() is called, intialize it */
|
if (!(q = km->loop_head)) /* the first time when kmalloc() is called, intialize it */
|
||||||
q = km->loop_head = km->base.ptr = &km->base;
|
q = km->loop_head = km->base.ptr = &km->base;
|
||||||
@@ -160,18 +167,18 @@ void *kcalloc(void *_km, size_t count, size_t size)
|
|||||||
void *krealloc(void *_km, void *ap, size_t n_bytes) // TODO: this can be made more efficient in principle
|
void *krealloc(void *_km, void *ap, size_t n_bytes) // TODO: this can be made more efficient in principle
|
||||||
{
|
{
|
||||||
kmem_t *km = (kmem_t*)_km;
|
kmem_t *km = (kmem_t*)_km;
|
||||||
size_t n_units, *p, *q;
|
size_t cap, *p, *q;
|
||||||
|
|
||||||
if (n_bytes == 0) {
|
if (n_bytes == 0) {
|
||||||
kfree(km, ap); return 0;
|
kfree(km, ap); return 0;
|
||||||
}
|
}
|
||||||
if (km == NULL) return realloc(ap, n_bytes);
|
if (km == NULL) return realloc(ap, n_bytes);
|
||||||
if (ap == NULL) return kmalloc(km, n_bytes);
|
if (ap == NULL) return kmalloc(km, n_bytes);
|
||||||
n_units = (n_bytes + sizeof(size_t) + sizeof(header_t) - 1) / sizeof(header_t);
|
|
||||||
p = (size_t*)ap - 1;
|
p = (size_t*)ap - 1;
|
||||||
if (*p >= n_units) return ap; /* TODO: this prevents shrinking */
|
cap = (*p) * sizeof(header_t) - sizeof(size_t);
|
||||||
|
if (cap >= n_bytes) return ap; /* TODO: this prevents shrinking */
|
||||||
q = (size_t*)kmalloc(km, n_bytes);
|
q = (size_t*)kmalloc(km, n_bytes);
|
||||||
memcpy(q, ap, (*p - 1) * sizeof(header_t));
|
memcpy(q, ap, cap);
|
||||||
kfree(km, ap);
|
kfree(km, ap);
|
||||||
return q;
|
return q;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -17,6 +17,7 @@ void *kcalloc(void *km, size_t count, size_t size);
|
|||||||
void kfree(void *km, void *ptr);
|
void kfree(void *km, void *ptr);
|
||||||
|
|
||||||
void *km_init(void);
|
void *km_init(void);
|
||||||
|
void *km_init2(void *km_par, size_t min_core_size);
|
||||||
void km_destroy(void *km);
|
void km_destroy(void *km);
|
||||||
void km_stat(const void *_km, km_stat_t *s);
|
void km_stat(const void *_km, km_stat_t *s);
|
||||||
|
|
||||||
@@ -24,4 +25,13 @@ void km_stat(const void *_km, km_stat_t *s);
|
|||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#define KMALLOC(km, ptr, len) ((ptr) = (__typeof__(ptr))kmalloc((km), (len) * sizeof(*(ptr))))
|
||||||
|
#define KCALLOC(km, ptr, len) ((ptr) = (__typeof__(ptr))kcalloc((km), (len), sizeof(*(ptr))))
|
||||||
|
#define KREALLOC(km, ptr, len) ((ptr) = (__typeof__(ptr))krealloc((km), (ptr), (len) * sizeof(*(ptr))))
|
||||||
|
|
||||||
|
#define KEXPAND(km, a, m) do { \
|
||||||
|
(m) = (m) >= 4? (m) + ((m)>>1) : 16; \
|
||||||
|
KREALLOC((km), (a), (m)); \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@@ -73,13 +73,17 @@ static int ketopt(ketopt_t *s, int argc, char *argv[], int permute, const char *
|
|||||||
}
|
}
|
||||||
s->opt = 0, opt = '?', s->pos = -1;
|
s->opt = 0, opt = '?', s->pos = -1;
|
||||||
if (longopts) { /* parse long options */
|
if (longopts) { /* parse long options */
|
||||||
int k, n_matches = 0;
|
int k, n_exact = 0, n_partial = 0;
|
||||||
const ko_longopt_t *o = 0;
|
const ko_longopt_t *o = 0, *o_exact = 0, *o_partial = 0;
|
||||||
for (j = 2; argv[s->i][j] != '\0' && argv[s->i][j] != '='; ++j) {} /* find the end of the option name */
|
for (j = 2; argv[s->i][j] != '\0' && argv[s->i][j] != '='; ++j) {} /* find the end of the option name */
|
||||||
for (k = 0; longopts[k].name != 0; ++k)
|
for (k = 0; longopts[k].name != 0; ++k)
|
||||||
if (strncmp(&argv[s->i][2], longopts[k].name, j - 2) == 0)
|
if (strncmp(&argv[s->i][2], longopts[k].name, j - 2) == 0) {
|
||||||
++n_matches, o = &longopts[k];
|
if (longopts[k].name[j - 2] == 0) ++n_exact, o_exact = &longopts[k];
|
||||||
if (n_matches == 1) {
|
else ++n_partial, o_partial = &longopts[k];
|
||||||
|
}
|
||||||
|
if (n_exact > 1 || (n_exact == 0 && n_partial > 1)) return '?';
|
||||||
|
o = n_exact == 1? o_exact : n_partial == 1? o_partial : 0;
|
||||||
|
if (o) {
|
||||||
s->opt = opt = o->val, s->longidx = o - longopts;
|
s->opt = opt = o->val, s->longidx = o - longopts;
|
||||||
if (argv[s->i][j] == '=') s->arg = &argv[s->i][j + 1];
|
if (argv[s->i][j] == '=') s->arg = &argv[s->i][j + 1];
|
||||||
if (o->has_arg == 1 && argv[s->i][j] == '\0') {
|
if (o->has_arg == 1 && argv[s->i][j] == '\0') {
|
||||||
|
|||||||
@@ -37,6 +37,14 @@
|
|||||||
#define KS_SEP_LINE 2 // line separator: "\n" (Unix) or "\r\n" (Windows)
|
#define KS_SEP_LINE 2 // line separator: "\n" (Unix) or "\r\n" (Windows)
|
||||||
#define KS_SEP_MAX 2
|
#define KS_SEP_MAX 2
|
||||||
|
|
||||||
|
#ifndef klib_unused
|
||||||
|
#if (defined __clang__ && __clang_major__ >= 3) || (defined __GNUC__ && __GNUC__ >= 3)
|
||||||
|
#define klib_unused __attribute__ ((__unused__))
|
||||||
|
#else
|
||||||
|
#define klib_unused
|
||||||
|
#endif
|
||||||
|
#endif /* klib_unused */
|
||||||
|
|
||||||
#define __KS_TYPE(type_t) \
|
#define __KS_TYPE(type_t) \
|
||||||
typedef struct __kstream_t { \
|
typedef struct __kstream_t { \
|
||||||
int begin, end; \
|
int begin, end; \
|
||||||
@@ -64,7 +72,7 @@
|
|||||||
}
|
}
|
||||||
|
|
||||||
#define __KS_INLINED(__read) \
|
#define __KS_INLINED(__read) \
|
||||||
static inline int ks_getc(kstream_t *ks) \
|
static inline klib_unused int ks_getc(kstream_t *ks) \
|
||||||
{ \
|
{ \
|
||||||
if (ks->is_eof && ks->begin >= ks->end) return -1; \
|
if (ks->is_eof && ks->begin >= ks->end) return -1; \
|
||||||
if (ks->begin >= ks->end) { \
|
if (ks->begin >= ks->end) { \
|
||||||
|
|||||||
@@ -61,7 +61,7 @@ void ksw_extd2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
int8_t gapo, int8_t gape, int8_t gapo2, int8_t gape2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez);
|
int8_t gapo, int8_t gape, int8_t gapo2, int8_t gape2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez);
|
||||||
|
|
||||||
void ksw_exts2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
void ksw_exts2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
int8_t gapo, int8_t gape, int8_t gapo2, int8_t noncan, int zdrop, int flag, ksw_extz_t *ez);
|
int8_t gapo, int8_t gape, int8_t gapo2, int8_t noncan, int zdrop, int8_t junc_bonus, int flag, const uint8_t *junc, ksw_extz_t *ez);
|
||||||
|
|
||||||
void ksw_extf2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t mch, int8_t mis, int8_t e, int w, int xdrop, ksw_extz_t *ez);
|
void ksw_extf2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t mch, int8_t mis, int8_t e, int w, int xdrop, ksw_extz_t *ez);
|
||||||
|
|
||||||
|
|||||||
+30
-14
@@ -2,15 +2,16 @@
|
|||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include "ksw2.h"
|
#include "ksw2.h"
|
||||||
|
|
||||||
#define SIMD_SSE 0x1
|
#define SIMD_SSE 0x1
|
||||||
#define SIMD_SSE2 0x2
|
#define SIMD_SSE2 0x2
|
||||||
#define SIMD_SSE3 0x4
|
#define SIMD_SSE3 0x4
|
||||||
#define SIMD_SSSE3 0x8
|
#define SIMD_SSSE3 0x8
|
||||||
#define SIMD_SSE4_1 0x10
|
#define SIMD_SSE4_1 0x10
|
||||||
#define SIMD_SSE4_2 0x20
|
#define SIMD_SSE4_2 0x20
|
||||||
#define SIMD_AVX 0x40
|
#define SIMD_AVX 0x40
|
||||||
#define SIMD_AVX2 0x80
|
#define SIMD_AVX2 0x80
|
||||||
#define SIMD_AVX512F 0x100
|
#define SIMD_AVX512F 0x100
|
||||||
|
#define SIMD_AVX512BW 0x200
|
||||||
|
|
||||||
#ifndef _MSC_VER
|
#ifndef _MSC_VER
|
||||||
// adapted from https://github.com/01org/linux-sgx/blob/master/common/inc/internal/linux/cpuid_gnu.h
|
// adapted from https://github.com/01org/linux-sgx/blob/master/common/inc/internal/linux/cpuid_gnu.h
|
||||||
@@ -48,6 +49,7 @@ static int x86_simd(void)
|
|||||||
__cpuidex(cpuid, 7, 0);
|
__cpuidex(cpuid, 7, 0);
|
||||||
if (cpuid[1]>>5 &1) flag |= SIMD_AVX2;
|
if (cpuid[1]>>5 &1) flag |= SIMD_AVX2;
|
||||||
if (cpuid[1]>>16&1) flag |= SIMD_AVX512F;
|
if (cpuid[1]>>16&1) flag |= SIMD_AVX512F;
|
||||||
|
if (cpuid[1]>>30&1) flag |= SIMD_AVX512BW;
|
||||||
}
|
}
|
||||||
return flag;
|
return flag;
|
||||||
}
|
}
|
||||||
@@ -71,7 +73,21 @@ void ksw_extd2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez);
|
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez);
|
||||||
extern void ksw_extd2_sse41(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
extern void ksw_extd2_sse41(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez);
|
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez);
|
||||||
|
extern void ksw_extd2_avx2(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
|
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez);
|
||||||
|
extern void ksw_extd2_avx512(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
|
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez);
|
||||||
if (ksw_simd < 0) ksw_simd = x86_simd();
|
if (ksw_simd < 0) ksw_simd = x86_simd();
|
||||||
|
#if defined(__AVX512BW__)
|
||||||
|
if (ksw_simd & SIMD_AVX512BW)
|
||||||
|
ksw_extd2_avx512(km, qlen, query, tlen, target, m, mat, q, e, q2, e2, w, zdrop, end_bonus, flag, ez);
|
||||||
|
else
|
||||||
|
#endif
|
||||||
|
#if defined(__AVX2__)
|
||||||
|
if (ksw_simd & SIMD_AVX2)
|
||||||
|
ksw_extd2_avx2(km, qlen, query, tlen, target, m, mat, q, e, q2, e2, w, zdrop, end_bonus, flag, ez);
|
||||||
|
else
|
||||||
|
#endif
|
||||||
if (ksw_simd & SIMD_SSE4_1)
|
if (ksw_simd & SIMD_SSE4_1)
|
||||||
ksw_extd2_sse41(km, qlen, query, tlen, target, m, mat, q, e, q2, e2, w, zdrop, end_bonus, flag, ez);
|
ksw_extd2_sse41(km, qlen, query, tlen, target, m, mat, q, e, q2, e2, w, zdrop, end_bonus, flag, ez);
|
||||||
else if (ksw_simd & SIMD_SSE2)
|
else if (ksw_simd & SIMD_SSE2)
|
||||||
@@ -80,17 +96,17 @@ void ksw_extd2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
}
|
}
|
||||||
|
|
||||||
void ksw_exts2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
void ksw_exts2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
int8_t q, int8_t e, int8_t q2, int8_t noncan, int zdrop, int flag, ksw_extz_t *ez)
|
int8_t q, int8_t e, int8_t q2, int8_t noncan, int zdrop, int8_t junc_bonus, int flag, const uint8_t *junc, ksw_extz_t *ez)
|
||||||
{
|
{
|
||||||
extern void ksw_exts2_sse2(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
extern void ksw_exts2_sse2(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
int8_t q, int8_t e, int8_t q2, int8_t noncan, int zdrop, int flag, ksw_extz_t *ez);
|
int8_t q, int8_t e, int8_t q2, int8_t noncan, int zdrop, int8_t junc_bonus, int flag, const uint8_t *junc, ksw_extz_t *ez);
|
||||||
extern void ksw_exts2_sse41(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
extern void ksw_exts2_sse41(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
int8_t q, int8_t e, int8_t q2, int8_t noncan, int zdrop, int flag, ksw_extz_t *ez);
|
int8_t q, int8_t e, int8_t q2, int8_t noncan, int zdrop, int8_t junc_bonus, int flag, const uint8_t *junc, ksw_extz_t *ez);
|
||||||
if (ksw_simd < 0) ksw_simd = x86_simd();
|
if (ksw_simd < 0) ksw_simd = x86_simd();
|
||||||
if (ksw_simd & SIMD_SSE4_1)
|
if (ksw_simd & SIMD_SSE4_1)
|
||||||
ksw_exts2_sse41(km, qlen, query, tlen, target, m, mat, q, e, q2, noncan, zdrop, flag, ez);
|
ksw_exts2_sse41(km, qlen, query, tlen, target, m, mat, q, e, q2, noncan, zdrop, junc_bonus, flag, junc, ez);
|
||||||
else if (ksw_simd & SIMD_SSE2)
|
else if (ksw_simd & SIMD_SSE2)
|
||||||
ksw_exts2_sse2(km, qlen, query, tlen, target, m, mat, q, e, q2, noncan, zdrop, flag, ez);
|
ksw_exts2_sse2(km, qlen, query, tlen, target, m, mat, q, e, q2, noncan, zdrop, junc_bonus, flag, junc, ez);
|
||||||
else abort();
|
else abort();
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
+291
-158
@@ -4,7 +4,27 @@
|
|||||||
#include "ksw2.h"
|
#include "ksw2.h"
|
||||||
|
|
||||||
#ifdef __SSE2__
|
#ifdef __SSE2__
|
||||||
|
|
||||||
|
#if defined(__AVX512BW__)
|
||||||
|
#include <immintrin.h>
|
||||||
|
#define SIMD_INT __m512i
|
||||||
|
#define SIMD_SHIFT 6
|
||||||
|
#define simd_func(func) _mm512_##func
|
||||||
|
#define simd_funcw(func) _mm512_##func##_si512
|
||||||
|
|
||||||
|
#elif defined(__AVX2__)
|
||||||
|
#include <immintrin.h>
|
||||||
|
#define SIMD_INT __m256i
|
||||||
|
#define SIMD_SHIFT 5
|
||||||
|
#define simd_func(func) _mm256_##func
|
||||||
|
#define simd_funcw(func) _mm256_##func##_si256
|
||||||
|
|
||||||
|
#elif defined(__SSE2__)
|
||||||
#include <emmintrin.h>
|
#include <emmintrin.h>
|
||||||
|
#define SIMD_INT __m128i
|
||||||
|
#define SIMD_SHIFT 4
|
||||||
|
#define simd_func(func) _mm_##func
|
||||||
|
#define simd_funcw(func) _mm_##func##_si128
|
||||||
|
|
||||||
#ifdef KSW_SSE2_ONLY
|
#ifdef KSW_SSE2_ONLY
|
||||||
#undef __SSE4_1__
|
#undef __SSE4_1__
|
||||||
@@ -13,12 +33,39 @@
|
|||||||
#ifdef __SSE4_1__
|
#ifdef __SSE4_1__
|
||||||
#include <smmintrin.h>
|
#include <smmintrin.h>
|
||||||
#endif
|
#endif
|
||||||
|
#endif // defined(__SSE2__)
|
||||||
|
|
||||||
|
#define SIMD_WIDTH (1<<SIMD_SHIFT)
|
||||||
|
|
||||||
|
|
||||||
|
#if !defined(__AVX512BW__)
|
||||||
|
#if defined(__AVX2__)
|
||||||
|
static inline __m256i simd_slli_1(__m256i x)
|
||||||
|
{
|
||||||
|
return _mm256_insert_epi8(_mm256_slli_si256(x, 1), _mm256_extract_epi8(x, 15), 16);
|
||||||
|
}
|
||||||
|
static inline __m256i simd_srli_last(__m256i x)
|
||||||
|
{
|
||||||
|
return _mm256_insert_epi8(_mm256_setzero_si256(), _mm256_extract_epi8(x, 31), 0);
|
||||||
|
}
|
||||||
|
#elif defined(__SSE2__)
|
||||||
|
static inline __m128i simd_slli_1(__m128i x) { return _mm_slli_si128(x, 1); }
|
||||||
|
static inline __m128i simd_srli_last(__m128i x) { return _mm_srli_si128(x, 15); }
|
||||||
|
#endif
|
||||||
|
#endif // ~__AVX512BW__
|
||||||
|
|
||||||
|
|
||||||
#ifdef KSW_CPU_DISPATCH
|
#ifdef KSW_CPU_DISPATCH
|
||||||
#ifdef __SSE4_1__
|
#if defined(__AVX512BW__)
|
||||||
|
void ksw_extd2_avx512(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
|
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez)
|
||||||
|
#elif defined(__AVX2__)
|
||||||
|
void ksw_extd2_avx2(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
|
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez)
|
||||||
|
#elif defined(__SSE4_1__)
|
||||||
void ksw_extd2_sse41(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
void ksw_extd2_sse41(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez)
|
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez)
|
||||||
#else
|
#elif defined(__SSE2__)
|
||||||
void ksw_extd2_sse2(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
void ksw_extd2_sse2(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez)
|
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez)
|
||||||
#endif
|
#endif
|
||||||
@@ -27,64 +74,91 @@ void ksw_extd2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez)
|
int8_t q, int8_t e, int8_t q2, int8_t e2, int w, int zdrop, int end_bonus, int flag, ksw_extz_t *ez)
|
||||||
#endif // ~KSW_CPU_DISPATCH
|
#endif // ~KSW_CPU_DISPATCH
|
||||||
{
|
{
|
||||||
|
#if defined(__AVX512BW__)
|
||||||
#define __dp_code_block1 \
|
#define __dp_code_block1 \
|
||||||
z = _mm_load_si128(&s[t]); \
|
z = _mm512_load_si512(&s[t]); \
|
||||||
xt1 = _mm_load_si128(&x[t]); /* xt1 <- x[r-1][t..t+15] */ \
|
tmp = _mm512_loadu_si512((uint8_t*)&x[t] - 1); \
|
||||||
tmp = _mm_srli_si128(xt1, 15); /* tmp <- x[r-1][t+15] */ \
|
xt1 = _mm512_mask_blend_epi8(1, tmp, x1_); \
|
||||||
xt1 = _mm_or_si128(_mm_slli_si128(xt1, 1), x1_); /* xt1 <- x[r-1][t-1..t+14] */ \
|
x1_ = _mm512_maskz_set1_epi8(1, *((uint8_t*)&x[t] + 63)); \
|
||||||
|
tmp = _mm512_loadu_si512((uint8_t*)&v[t] - 1); \
|
||||||
|
vt1 = _mm512_mask_blend_epi8(1, tmp, v1_); \
|
||||||
|
v1_ = _mm512_maskz_set1_epi8(1, *((uint8_t*)&v[t] + 63)); \
|
||||||
|
a = _mm512_add_epi8(xt1, vt1); \
|
||||||
|
ut = _mm512_load_si512(&u[t]); \
|
||||||
|
b = _mm512_add_epi8(_mm512_load_si512(&y[t]), ut); \
|
||||||
|
tmp = _mm512_loadu_si512((uint8_t*)&x2[t] - 1); \
|
||||||
|
x2t1 = _mm512_mask_blend_epi8(1, tmp, x21_); \
|
||||||
|
x21_ = _mm512_maskz_set1_epi8(1, *((uint8_t*)&x2[t] + 63)); \
|
||||||
|
a2= _mm512_add_epi8(x2t1, vt1); \
|
||||||
|
b2= _mm512_add_epi8(_mm512_load_si512(&y2[t]), ut);
|
||||||
|
#else
|
||||||
|
#define __dp_code_block1 \
|
||||||
|
z = simd_funcw(load)(&s[t]); \
|
||||||
|
xt1 = simd_funcw(load)(&x[t]); /* xt1 <- x[r-1][t..t+15] */ \
|
||||||
|
tmp = simd_srli_last(xt1); /* tmp <- x[r-1][t+15] */ \
|
||||||
|
xt1 = simd_funcw(or)(simd_slli_1(xt1), x1_); /* xt1 <- x[r-1][t-1..t+14] */ \
|
||||||
x1_ = tmp; \
|
x1_ = tmp; \
|
||||||
vt1 = _mm_load_si128(&v[t]); /* vt1 <- v[r-1][t..t+15] */ \
|
vt1 = simd_funcw(load)(&v[t]); /* vt1 <- v[r-1][t..t+15] */ \
|
||||||
tmp = _mm_srli_si128(vt1, 15); /* tmp <- v[r-1][t+15] */ \
|
tmp = simd_srli_last(vt1); /* tmp <- v[r-1][t+15] */ \
|
||||||
vt1 = _mm_or_si128(_mm_slli_si128(vt1, 1), v1_); /* vt1 <- v[r-1][t-1..t+14] */ \
|
vt1 = simd_funcw(or)(simd_slli_1(vt1), v1_); /* vt1 <- v[r-1][t-1..t+14] */ \
|
||||||
v1_ = tmp; \
|
v1_ = tmp; \
|
||||||
a = _mm_add_epi8(xt1, vt1); /* a <- x[r-1][t-1..t+14] + v[r-1][t-1..t+14] */ \
|
a = simd_func(add_epi8)(xt1, vt1); /* a <- x[r-1][t-1..t+14] + v[r-1][t-1..t+14] */ \
|
||||||
ut = _mm_load_si128(&u[t]); /* ut <- u[t..t+15] */ \
|
ut = simd_funcw(load)(&u[t]); /* ut <- u[t..t+15] */ \
|
||||||
b = _mm_add_epi8(_mm_load_si128(&y[t]), ut); /* b <- y[r-1][t..t+15] + u[r-1][t..t+15] */ \
|
b = simd_func(add_epi8)(simd_funcw(load)(&y[t]), ut); /* b <- y[r-1][t..t+15] + u[r-1][t..t+15] */ \
|
||||||
x2t1= _mm_load_si128(&x2[t]); \
|
x2t1= simd_funcw(load)(&x2[t]); \
|
||||||
tmp = _mm_srli_si128(x2t1, 15); \
|
tmp = simd_srli_last(x2t1); \
|
||||||
x2t1= _mm_or_si128(_mm_slli_si128(x2t1, 1), x21_); \
|
x2t1= simd_funcw(or)(simd_slli_1(x2t1), x21_); \
|
||||||
x21_= tmp; \
|
x21_= tmp; \
|
||||||
a2= _mm_add_epi8(x2t1, vt1); \
|
a2= simd_func(add_epi8)(x2t1, vt1); \
|
||||||
b2= _mm_add_epi8(_mm_load_si128(&y2[t]), ut);
|
b2= simd_func(add_epi8)(simd_funcw(load)(&y2[t]), ut);
|
||||||
|
#endif // ~__AVX512BW__
|
||||||
|
|
||||||
#define __dp_code_block2 \
|
#define __dp_code_block2 \
|
||||||
_mm_store_si128(&u[t], _mm_sub_epi8(z, vt1)); /* u[r][t..t+15] <- z - v[r-1][t-1..t+14] */ \
|
simd_funcw(store)(&u[t], simd_func(sub_epi8)(z, vt1));/* u[r][t..t+15] <- z - v[r-1][t-1..t+14] */ \
|
||||||
_mm_store_si128(&v[t], _mm_sub_epi8(z, ut)); /* v[r][t..t+15] <- z - u[r-1][t..t+15] */ \
|
simd_funcw(store)(&v[t], simd_func(sub_epi8)(z, ut)); /* v[r][t..t+15] <- z - u[r-1][t..t+15] */ \
|
||||||
tmp = _mm_sub_epi8(z, q_); \
|
tmp = simd_func(sub_epi8)(z, q_); \
|
||||||
a = _mm_sub_epi8(a, tmp); \
|
a = simd_func(sub_epi8)(a, tmp); \
|
||||||
b = _mm_sub_epi8(b, tmp); \
|
b = simd_func(sub_epi8)(b, tmp); \
|
||||||
tmp = _mm_sub_epi8(z, q2_); \
|
tmp = simd_func(sub_epi8)(z, q2_); \
|
||||||
a2= _mm_sub_epi8(a2, tmp); \
|
a2= simd_func(sub_epi8)(a2, tmp); \
|
||||||
b2= _mm_sub_epi8(b2, tmp);
|
b2= simd_func(sub_epi8)(b2, tmp);
|
||||||
|
|
||||||
int r, t, qe = q + e, n_col_, *off = 0, *off_end = 0, tlen_, qlen_, last_st, last_en, wl, wr, max_sc, min_sc, long_thres, long_diff;
|
int r, t, qe = q + e, n_col_, *off = 0, *off_end = 0, tlen_, qlen_, last_st, last_en, wl, wr, max_sc, min_sc, long_thres, long_diff;
|
||||||
int with_cigar = !(flag&KSW_EZ_SCORE_ONLY), approx_max = !!(flag&KSW_EZ_APPROX_MAX);
|
int with_cigar = !(flag&KSW_EZ_SCORE_ONLY), approx_max = !!(flag&KSW_EZ_APPROX_MAX);
|
||||||
int32_t *H = 0, H0 = 0, last_H0_t = 0;
|
int32_t *H = 0, H0 = 0, last_H0_t = 0;
|
||||||
uint8_t *qr, *sf, *mem, *mem2 = 0;
|
uint8_t *qr, *sf, *mem, *mem2 = 0;
|
||||||
__m128i q_, q2_, qe_, qe2_, zero_, sc_mch_, sc_mis_, m1_, sc_N_;
|
SIMD_INT q_, q2_, qe_, qe2_, zero_, sc_mch_, sc_mis_, m1_, sc_N_, mask1_;
|
||||||
__m128i *u, *v, *x, *y, *x2, *y2, *s, *p = 0;
|
SIMD_INT *u, *v, *x, *y, *x2, *y2, *s, *p = 0;
|
||||||
|
|
||||||
ksw_reset_extz(ez);
|
ksw_reset_extz(ez);
|
||||||
if (m <= 1 || qlen <= 0 || tlen <= 0) return;
|
if (m <= 1 || qlen <= 0 || tlen <= 0) return;
|
||||||
|
|
||||||
if (q2 + e2 < q + e) t = q, q = q2, q2 = t, t = e, e = e2, e2 = t; // make sure q+e no larger than q2+e2
|
if (q2 + e2 < q + e) t = q, q = q2, q2 = t, t = e, e = e2, e2 = t; // make sure q+e no larger than q2+e2
|
||||||
|
|
||||||
zero_ = _mm_set1_epi8(0);
|
zero_ = simd_func(set1_epi8)(0);
|
||||||
q_ = _mm_set1_epi8(q);
|
q_ = simd_func(set1_epi8)(q);
|
||||||
q2_ = _mm_set1_epi8(q2);
|
q2_ = simd_func(set1_epi8)(q2);
|
||||||
qe_ = _mm_set1_epi8(q + e);
|
qe_ = simd_func(set1_epi8)(q + e);
|
||||||
qe2_ = _mm_set1_epi8(q2 + e2);
|
qe2_ = simd_func(set1_epi8)(q2 + e2);
|
||||||
sc_mch_ = _mm_set1_epi8(mat[0]);
|
sc_mch_ = simd_func(set1_epi8)(mat[0]);
|
||||||
sc_mis_ = _mm_set1_epi8(mat[1]);
|
sc_mis_ = simd_func(set1_epi8)(mat[1]);
|
||||||
sc_N_ = mat[m*m-1] == 0? _mm_set1_epi8(-e2) : _mm_set1_epi8(mat[m*m-1]);
|
sc_N_ = mat[m*m-1] == 0? simd_func(set1_epi8)(-e2) : simd_func(set1_epi8)(mat[m*m-1]);
|
||||||
m1_ = _mm_set1_epi8(m - 1); // wildcard
|
m1_ = simd_func(set1_epi8)(m - 1); // wildcard
|
||||||
|
|
||||||
|
#if defined(__AVX512BW__)
|
||||||
|
mask1_ = _mm512_maskz_set1_epi8(1, 0xff);
|
||||||
|
#elif defined(__AVX2__)
|
||||||
|
mask1_ = _mm256_setr_epi32(0xff, 0, 0, 0, 0, 0, 0, 0);
|
||||||
|
#elif defined(__SSE2__)
|
||||||
|
mask1_ = _mm_setr_epi32(0xff, 0, 0, 0);
|
||||||
|
#endif
|
||||||
|
|
||||||
if (w < 0) w = tlen > qlen? tlen : qlen;
|
if (w < 0) w = tlen > qlen? tlen : qlen;
|
||||||
wl = wr = w;
|
wl = wr = w;
|
||||||
tlen_ = (tlen + 15) / 16;
|
tlen_ = (tlen + SIMD_WIDTH - 1) / SIMD_WIDTH;
|
||||||
n_col_ = qlen < tlen? qlen : tlen;
|
n_col_ = qlen < tlen? qlen : tlen;
|
||||||
n_col_ = ((n_col_ < w + 1? n_col_ : w + 1) + 15) / 16 + 1;
|
n_col_ = ((n_col_ < w + 1? n_col_ : w + 1) + SIMD_WIDTH - 1) / SIMD_WIDTH + 1;
|
||||||
qlen_ = (qlen + 15) / 16;
|
qlen_ = (qlen + SIMD_WIDTH - 1) / SIMD_WIDTH;
|
||||||
for (t = 1, max_sc = mat[0], min_sc = mat[1]; t < m * m; ++t) {
|
for (t = 1, max_sc = mat[0], min_sc = mat[1]; t < m * m; ++t) {
|
||||||
max_sc = max_sc > mat[t]? max_sc : mat[t];
|
max_sc = max_sc > mat[t]? max_sc : mat[t];
|
||||||
min_sc = min_sc < mat[t]? min_sc : mat[t];
|
min_sc = min_sc < mat[t]? min_sc : mat[t];
|
||||||
@@ -96,23 +170,23 @@ void ksw_extd2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
++long_thres;
|
++long_thres;
|
||||||
long_diff = long_thres * (e - e2) - (q2 - q) - e2;
|
long_diff = long_thres * (e - e2) - (q2 - q) - e2;
|
||||||
|
|
||||||
mem = (uint8_t*)kcalloc(km, tlen_ * 8 + qlen_ + 1, 16);
|
mem = (uint8_t*)kcalloc(km, tlen_ * 8 + qlen_ + 1, SIMD_WIDTH);
|
||||||
u = (__m128i*)(((size_t)mem + 15) >> 4 << 4); // 16-byte aligned
|
u = (SIMD_INT*)(((size_t)mem + SIMD_WIDTH - 1) >> SIMD_SHIFT << SIMD_SHIFT); // 16-byte aligned
|
||||||
v = u + tlen_, x = v + tlen_, y = x + tlen_, x2 = y + tlen_, y2 = x2 + tlen_;
|
v = u + tlen_, x = v + tlen_, y = x + tlen_, x2 = y + tlen_, y2 = x2 + tlen_;
|
||||||
s = y2 + tlen_, sf = (uint8_t*)(s + tlen_), qr = sf + tlen_ * 16;
|
s = y2 + tlen_, sf = (uint8_t*)(s + tlen_), qr = sf + tlen_ * SIMD_WIDTH;
|
||||||
memset(u, -q - e, tlen_ * 16);
|
memset(u, -q - e, tlen_ * SIMD_WIDTH);
|
||||||
memset(v, -q - e, tlen_ * 16);
|
memset(v, -q - e, tlen_ * SIMD_WIDTH);
|
||||||
memset(x, -q - e, tlen_ * 16);
|
memset(x, -q - e, tlen_ * SIMD_WIDTH);
|
||||||
memset(y, -q - e, tlen_ * 16);
|
memset(y, -q - e, tlen_ * SIMD_WIDTH);
|
||||||
memset(x2, -q2 - e2, tlen_ * 16);
|
memset(x2, -q2 - e2, tlen_ * SIMD_WIDTH);
|
||||||
memset(y2, -q2 - e2, tlen_ * 16);
|
memset(y2, -q2 - e2, tlen_ * SIMD_WIDTH);
|
||||||
if (!approx_max) {
|
if (!approx_max) {
|
||||||
H = (int32_t*)kmalloc(km, tlen_ * 16 * 4);
|
H = (int32_t*)kmalloc(km, tlen_ * SIMD_WIDTH * 4);
|
||||||
for (t = 0; t < tlen_ * 16; ++t) H[t] = KSW_NEG_INF;
|
for (t = 0; t < tlen_ * SIMD_WIDTH; ++t) H[t] = KSW_NEG_INF;
|
||||||
}
|
}
|
||||||
if (with_cigar) {
|
if (with_cigar) {
|
||||||
mem2 = (uint8_t*)kmalloc(km, ((size_t)(qlen + tlen - 1) * n_col_ + 1) * 16);
|
mem2 = (uint8_t*)kmalloc(km, ((size_t)(qlen + tlen - 1) * n_col_ + 1) * SIMD_WIDTH);
|
||||||
p = (__m128i*)(((size_t)mem2 + 15) >> 4 << 4);
|
p = (SIMD_INT*)(((size_t)mem2 + SIMD_WIDTH - 1) >> SIMD_SHIFT << SIMD_SHIFT);
|
||||||
off = (int*)kmalloc(km, (qlen + tlen - 1) * sizeof(int) * 2);
|
off = (int*)kmalloc(km, (qlen + tlen - 1) * sizeof(int) * 2);
|
||||||
off_end = off + qlen + tlen - 1;
|
off_end = off + qlen + tlen - 1;
|
||||||
}
|
}
|
||||||
@@ -125,7 +199,7 @@ void ksw_extd2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
int8_t x1, x21, v1;
|
int8_t x1, x21, v1;
|
||||||
uint8_t *qrr = qr + (qlen - 1 - r);
|
uint8_t *qrr = qr + (qlen - 1 - r);
|
||||||
int8_t *u8 = (int8_t*)u, *v8 = (int8_t*)v, *x8 = (int8_t*)x, *x28 = (int8_t*)x2;
|
int8_t *u8 = (int8_t*)u, *v8 = (int8_t*)v, *x8 = (int8_t*)x, *x28 = (int8_t*)x2;
|
||||||
__m128i x1_, x21_, v1_;
|
SIMD_INT x1_, x21_, v1_;
|
||||||
// find the boundaries
|
// find the boundaries
|
||||||
if (st < r - qlen + 1) st = r - qlen + 1;
|
if (st < r - qlen + 1) st = r - qlen + 1;
|
||||||
if (en > r) en = r;
|
if (en > r) en = r;
|
||||||
@@ -136,7 +210,7 @@ void ksw_extd2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
st0 = st, en0 = en;
|
st0 = st, en0 = en;
|
||||||
st = st / 16 * 16, en = (en + 16) / 16 * 16 - 1;
|
st = st / SIMD_WIDTH * SIMD_WIDTH, en = (en + SIMD_WIDTH) / SIMD_WIDTH * SIMD_WIDTH - 1;
|
||||||
// set boundary conditions
|
// set boundary conditions
|
||||||
if (st > 0) {
|
if (st > 0) {
|
||||||
if (st - 1 >= last_st && st - 1 <= last_en) {
|
if (st - 1 >= last_st && st - 1 <= last_en) {
|
||||||
@@ -155,47 +229,53 @@ void ksw_extd2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
}
|
}
|
||||||
// loop fission: set scores first
|
// loop fission: set scores first
|
||||||
if (!(flag & KSW_EZ_GENERIC_SC)) {
|
if (!(flag & KSW_EZ_GENERIC_SC)) {
|
||||||
for (t = st0; t <= en0; t += 16) {
|
for (t = st0; t <= en0; t += SIMD_WIDTH) {
|
||||||
__m128i sq, st, tmp, mask;
|
SIMD_INT sq, st, tmp;
|
||||||
sq = _mm_loadu_si128((__m128i*)&sf[t]);
|
sq = simd_funcw(loadu)((SIMD_INT*)&sf[t]);
|
||||||
st = _mm_loadu_si128((__m128i*)&qrr[t]);
|
st = simd_funcw(loadu)((SIMD_INT*)&qrr[t]);
|
||||||
mask = _mm_or_si128(_mm_cmpeq_epi8(sq, m1_), _mm_cmpeq_epi8(st, m1_));
|
#if defined(__AVX512BW__)
|
||||||
tmp = _mm_cmpeq_epi8(sq, st);
|
__mmask64 mask = _mm512_cmpeq_epi8_mask(sq, m1_) | _mm512_cmpeq_epi8_mask(st, m1_);
|
||||||
#ifdef __SSE4_1__
|
tmp = _mm512_mask_blend_epi8(_mm512_cmpeq_epi8_mask(sq, st), sc_mis_, sc_mch_);
|
||||||
tmp = _mm_blendv_epi8(sc_mis_, sc_mch_, tmp);
|
tmp = _mm512_mask_blend_epi8(mask, tmp, sc_N_);
|
||||||
tmp = _mm_blendv_epi8(tmp, sc_N_, mask);
|
#elif defined(__SSE4_1__) || defined(__AVX2__)
|
||||||
#else
|
SIMD_INT mask = simd_funcw(or)(simd_func(cmpeq_epi8)(sq, m1_), simd_func(cmpeq_epi8)(st, m1_));
|
||||||
|
tmp = simd_func(cmpeq_epi8)(sq, st);
|
||||||
|
tmp = simd_func(blendv_epi8)(sc_mis_, sc_mch_, tmp);
|
||||||
|
tmp = simd_func(blendv_epi8)(tmp, sc_N_, mask);
|
||||||
|
#elif defined(__SSE2__) // emulate blendv
|
||||||
|
SIMD_INT mask = simd_funcw(or)(simd_func(cmpeq_epi8)(sq, m1_), simd_func(cmpeq_epi8)(st, m1_));
|
||||||
|
tmp = simd_func(cmpeq_epi8)(sq, st);
|
||||||
tmp = _mm_or_si128(_mm_andnot_si128(tmp, sc_mis_), _mm_and_si128(tmp, sc_mch_));
|
tmp = _mm_or_si128(_mm_andnot_si128(tmp, sc_mis_), _mm_and_si128(tmp, sc_mch_));
|
||||||
tmp = _mm_or_si128(_mm_andnot_si128(mask, tmp), _mm_and_si128(mask, sc_N_));
|
tmp = _mm_or_si128(_mm_andnot_si128(mask, tmp), _mm_and_si128(mask, sc_N_));
|
||||||
#endif
|
#endif
|
||||||
_mm_storeu_si128((__m128i*)((int8_t*)s + t), tmp);
|
simd_funcw(storeu)((SIMD_INT*)((int8_t*)s + t), tmp);
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
for (t = st0; t <= en0; ++t)
|
for (t = st0; t <= en0; ++t)
|
||||||
((uint8_t*)s)[t] = mat[sf[t] * m + qrr[t]];
|
((uint8_t*)s)[t] = mat[sf[t] * m + qrr[t]];
|
||||||
}
|
}
|
||||||
// core loop
|
// core loop
|
||||||
x1_ = _mm_cvtsi32_si128((uint8_t)x1);
|
x1_ = simd_funcw(and)(simd_func(set1_epi8)((uint8_t)x1), mask1_);
|
||||||
x21_ = _mm_cvtsi32_si128((uint8_t)x21);
|
x21_ = simd_funcw(and)(simd_func(set1_epi8)((uint8_t)x21), mask1_);
|
||||||
v1_ = _mm_cvtsi32_si128((uint8_t)v1);
|
v1_ = simd_funcw(and)(simd_func(set1_epi8)((uint8_t)v1), mask1_);
|
||||||
st_ = st / 16, en_ = en / 16;
|
st_ = st / SIMD_WIDTH, en_ = en / SIMD_WIDTH;
|
||||||
assert(en_ - st_ + 1 <= n_col_);
|
assert(en_ - st_ + 1 <= n_col_);
|
||||||
if (!with_cigar) { // score only
|
if (!with_cigar) { // score only
|
||||||
for (t = st_; t <= en_; ++t) {
|
for (t = st_; t <= en_; ++t) {
|
||||||
__m128i z, a, b, a2, b2, xt1, x2t1, vt1, ut, tmp;
|
SIMD_INT z, a, b, a2, b2, xt1, x2t1, vt1, ut, tmp;
|
||||||
__dp_code_block1;
|
__dp_code_block1;
|
||||||
#ifdef __SSE4_1__
|
#if defined(__SSE4_1__) || defined(__AVX2__) || defined(__AVX512BW__)
|
||||||
z = _mm_max_epi8(z, a);
|
z = simd_func(max_epi8)(z, a);
|
||||||
z = _mm_max_epi8(z, b);
|
z = simd_func(max_epi8)(z, b);
|
||||||
z = _mm_max_epi8(z, a2);
|
z = simd_func(max_epi8)(z, a2);
|
||||||
z = _mm_max_epi8(z, b2);
|
z = simd_func(max_epi8)(z, b2);
|
||||||
z = _mm_min_epi8(z, sc_mch_);
|
z = simd_func(min_epi8)(z, sc_mch_);
|
||||||
__dp_code_block2; // save u[] and v[]; update a, b, a2 and b2
|
__dp_code_block2; // save u[] and v[]; update a, b, a2 and b2
|
||||||
_mm_store_si128(&x[t], _mm_sub_epi8(_mm_max_epi8(a, zero_), qe_));
|
simd_funcw(store)(&x[t], simd_func(sub_epi8)(simd_func(max_epi8)(a, zero_), qe_));
|
||||||
_mm_store_si128(&y[t], _mm_sub_epi8(_mm_max_epi8(b, zero_), qe_));
|
simd_funcw(store)(&y[t], simd_func(sub_epi8)(simd_func(max_epi8)(b, zero_), qe_));
|
||||||
_mm_store_si128(&x2[t], _mm_sub_epi8(_mm_max_epi8(a2, zero_), qe2_));
|
simd_funcw(store)(&x2[t], simd_func(sub_epi8)(simd_func(max_epi8)(a2, zero_), qe2_));
|
||||||
_mm_store_si128(&y2[t], _mm_sub_epi8(_mm_max_epi8(b2, zero_), qe2_));
|
simd_funcw(store)(&y2[t], simd_func(sub_epi8)(simd_func(max_epi8)(b2, zero_), qe2_));
|
||||||
#else
|
#elif defined(__SSE2__)
|
||||||
tmp = _mm_cmpgt_epi8(a, z);
|
tmp = _mm_cmpgt_epi8(a, z);
|
||||||
z = _mm_or_si128(_mm_andnot_si128(tmp, z), _mm_and_si128(tmp, a));
|
z = _mm_or_si128(_mm_andnot_si128(tmp, z), _mm_and_si128(tmp, a));
|
||||||
tmp = _mm_cmpgt_epi8(b, z);
|
tmp = _mm_cmpgt_epi8(b, z);
|
||||||
@@ -218,22 +298,42 @@ void ksw_extd2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
} else if (!(flag&KSW_EZ_RIGHT)) { // gap left-alignment
|
} else if (!(flag&KSW_EZ_RIGHT)) { // gap left-alignment
|
||||||
__m128i *pr = p + (size_t)r * n_col_ - st_;
|
SIMD_INT *pr = p + (size_t)r * n_col_ - st_;
|
||||||
off[r] = st, off_end[r] = en;
|
off[r] = st, off_end[r] = en;
|
||||||
for (t = st_; t <= en_; ++t) {
|
for (t = st_; t <= en_; ++t) {
|
||||||
__m128i d, z, a, b, a2, b2, xt1, x2t1, vt1, ut, tmp;
|
SIMD_INT d, z, a, b, a2, b2, xt1, x2t1, vt1, ut, tmp;
|
||||||
__dp_code_block1;
|
__dp_code_block1;
|
||||||
#ifdef __SSE4_1__
|
#if defined(__AVX512BW__)
|
||||||
d = _mm_and_si128(_mm_cmpgt_epi8(a, z), _mm_set1_epi8(1)); // d = a > z? 1 : 0
|
d = _mm512_maskz_set1_epi8(_mm512_cmpgt_epi8_mask(a, z), 1);
|
||||||
z = _mm_max_epi8(z, a);
|
z = _mm512_max_epi8(z, a);
|
||||||
d = _mm_blendv_epi8(d, _mm_set1_epi8(2), _mm_cmpgt_epi8(b, z)); // d = b > z? 2 : d
|
d = _mm512_mask_blend_epi8(_mm512_cmpgt_epi8_mask(b, z), d, _mm512_set1_epi8(2));
|
||||||
z = _mm_max_epi8(z, b);
|
z = _mm512_max_epi8(z, b);
|
||||||
d = _mm_blendv_epi8(d, _mm_set1_epi8(3), _mm_cmpgt_epi8(a2, z)); // d = a2 > z? 3 : d
|
d = _mm512_mask_blend_epi8(_mm512_cmpgt_epi8_mask(a2, z), d, _mm512_set1_epi8(3));
|
||||||
z = _mm_max_epi8(z, a2);
|
z = _mm512_max_epi8(z, a2);
|
||||||
d = _mm_blendv_epi8(d, _mm_set1_epi8(4), _mm_cmpgt_epi8(b2, z)); // d = a2 > z? 3 : d
|
d = _mm512_mask_blend_epi8(_mm512_cmpgt_epi8_mask(b2, z), d, _mm512_set1_epi8(4));
|
||||||
z = _mm_max_epi8(z, b2);
|
z = _mm512_max_epi8(z, b2);
|
||||||
z = _mm_min_epi8(z, sc_mch_);
|
z = _mm512_min_epi8(z, sc_mch_);
|
||||||
#else // we need to emulate SSE4.1 intrinsics _mm_max_epi8() and _mm_blendv_epi8()
|
__dp_code_block2;
|
||||||
|
d = _mm512_or_si512(d, _mm512_maskz_set1_epi8(_mm512_cmpgt_epi8_mask(a, zero_), 0x08)); // d = a > 0? 1<<3 : 0
|
||||||
|
_mm512_store_si512(&x[t], _mm512_sub_epi8(_mm512_max_epi8(a, zero_), qe_));
|
||||||
|
d = _mm512_or_si512(d, _mm512_maskz_set1_epi8(_mm512_cmpgt_epi8_mask(b, zero_), 0x10)); // d = b > 0? 1<<4 : 0
|
||||||
|
_mm512_store_si512(&y[t], _mm512_sub_epi8(_mm512_max_epi8(b, zero_), qe_));
|
||||||
|
d = _mm512_or_si512(d, _mm512_maskz_set1_epi8(_mm512_cmpgt_epi8_mask(a2, zero_), 0x20)); // d = a2 > 0? 1<<5 : 0
|
||||||
|
_mm512_store_si512(&x2[t], _mm512_sub_epi8(_mm512_max_epi8(a2, zero_), qe2_));
|
||||||
|
d = _mm512_or_si512(d, _mm512_maskz_set1_epi8(_mm512_cmpgt_epi8_mask(b2, zero_), 0x40)); // d = b2 > 0? 1<<6 : 0
|
||||||
|
_mm512_store_si512(&y2[t], _mm512_sub_epi8(_mm512_max_epi8(b2, zero_), qe2_));
|
||||||
|
#else
|
||||||
|
#if defined(__SSE4_1__) || defined(__AVX2__)
|
||||||
|
d = simd_funcw(and)(simd_func(cmpgt_epi8)(a, z), simd_func(set1_epi8)(1)); // d = a > z? 1 : 0
|
||||||
|
z = simd_func(max_epi8)(z, a);
|
||||||
|
d = simd_func(blendv_epi8)(d, simd_func(set1_epi8)(2), simd_func(cmpgt_epi8)(b, z)); // d = b > z? 2 : d
|
||||||
|
z = simd_func(max_epi8)(z, b);
|
||||||
|
d = simd_func(blendv_epi8)(d, simd_func(set1_epi8)(3), simd_func(cmpgt_epi8)(a2, z)); // d = a2 > z? 3 : d
|
||||||
|
z = simd_func(max_epi8)(z, a2);
|
||||||
|
d = simd_func(blendv_epi8)(d, simd_func(set1_epi8)(4), simd_func(cmpgt_epi8)(b2, z)); // d = a2 > z? 3 : d
|
||||||
|
z = simd_func(max_epi8)(z, b2);
|
||||||
|
z = simd_func(min_epi8)(z, sc_mch_);
|
||||||
|
#elif defined(__SSE2__) // emulate SSE4.1 intrinsics _mm_max_epi8() and _mm_blendv_epi8()
|
||||||
tmp = _mm_cmpgt_epi8(a, z);
|
tmp = _mm_cmpgt_epi8(a, z);
|
||||||
d = _mm_and_si128(tmp, _mm_set1_epi8(1));
|
d = _mm_and_si128(tmp, _mm_set1_epi8(1));
|
||||||
z = _mm_or_si128(_mm_andnot_si128(tmp, z), _mm_and_si128(tmp, a));
|
z = _mm_or_si128(_mm_andnot_si128(tmp, z), _mm_and_si128(tmp, a));
|
||||||
@@ -248,39 +348,60 @@ void ksw_extd2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
z = _mm_or_si128(_mm_andnot_si128(tmp, z), _mm_and_si128(tmp, b2));
|
z = _mm_or_si128(_mm_andnot_si128(tmp, z), _mm_and_si128(tmp, b2));
|
||||||
tmp = _mm_cmplt_epi8(sc_mch_, z);
|
tmp = _mm_cmplt_epi8(sc_mch_, z);
|
||||||
z = _mm_or_si128(_mm_and_si128(tmp, sc_mch_), _mm_andnot_si128(tmp, z));
|
z = _mm_or_si128(_mm_and_si128(tmp, sc_mch_), _mm_andnot_si128(tmp, z));
|
||||||
#endif
|
#endif // ~__SSE2__
|
||||||
__dp_code_block2;
|
__dp_code_block2;
|
||||||
tmp = _mm_cmpgt_epi8(a, zero_);
|
tmp = simd_func(cmpgt_epi8)(a, zero_);
|
||||||
_mm_store_si128(&x[t], _mm_sub_epi8(_mm_and_si128(tmp, a), qe_));
|
simd_funcw(store)(&x[t], simd_func(sub_epi8)(simd_funcw(and)(tmp, a), qe_));
|
||||||
d = _mm_or_si128(d, _mm_and_si128(tmp, _mm_set1_epi8(0x08))); // d = a > 0? 1<<3 : 0
|
d = simd_funcw(or)(d, simd_funcw(and)(tmp, simd_func(set1_epi8)(0x08))); // d = a > 0? 1<<3 : 0
|
||||||
tmp = _mm_cmpgt_epi8(b, zero_);
|
tmp = simd_func(cmpgt_epi8)(b, zero_);
|
||||||
_mm_store_si128(&y[t], _mm_sub_epi8(_mm_and_si128(tmp, b), qe_));
|
simd_funcw(store)(&y[t], simd_func(sub_epi8)(simd_funcw(and)(tmp, b), qe_));
|
||||||
d = _mm_or_si128(d, _mm_and_si128(tmp, _mm_set1_epi8(0x10))); // d = b > 0? 1<<4 : 0
|
d = simd_funcw(or)(d, simd_funcw(and)(tmp, simd_func(set1_epi8)(0x10))); // d = b > 0? 1<<4 : 0
|
||||||
tmp = _mm_cmpgt_epi8(a2, zero_);
|
tmp = simd_func(cmpgt_epi8)(a2, zero_);
|
||||||
_mm_store_si128(&x2[t], _mm_sub_epi8(_mm_and_si128(tmp, a2), qe2_));
|
simd_funcw(store)(&x2[t], simd_func(sub_epi8)(simd_funcw(and)(tmp, a2), qe2_));
|
||||||
d = _mm_or_si128(d, _mm_and_si128(tmp, _mm_set1_epi8(0x20))); // d = a > 0? 1<<5 : 0
|
d = simd_funcw(or)(d, simd_funcw(and)(tmp, simd_func(set1_epi8)(0x20))); // d = a > 0? 1<<5 : 0
|
||||||
tmp = _mm_cmpgt_epi8(b2, zero_);
|
tmp = simd_func(cmpgt_epi8)(b2, zero_);
|
||||||
_mm_store_si128(&y2[t], _mm_sub_epi8(_mm_and_si128(tmp, b2), qe2_));
|
simd_funcw(store)(&y2[t], simd_func(sub_epi8)(simd_funcw(and)(tmp, b2), qe2_));
|
||||||
d = _mm_or_si128(d, _mm_and_si128(tmp, _mm_set1_epi8(0x40))); // d = b > 0? 1<<6 : 0
|
d = simd_funcw(or)(d, simd_funcw(and)(tmp, simd_func(set1_epi8)(0x40))); // d = b > 0? 1<<6 : 0
|
||||||
_mm_store_si128(&pr[t], d);
|
#endif // ~__AVX512BW__
|
||||||
|
simd_funcw(store)(&pr[t], d);
|
||||||
}
|
}
|
||||||
} else { // gap right-alignment
|
} else { // gap right-alignment
|
||||||
__m128i *pr = p + (size_t)r * n_col_ - st_;
|
SIMD_INT *pr = p + (size_t)r * n_col_ - st_;
|
||||||
off[r] = st, off_end[r] = en;
|
off[r] = st, off_end[r] = en;
|
||||||
for (t = st_; t <= en_; ++t) {
|
for (t = st_; t <= en_; ++t) {
|
||||||
__m128i d, z, a, b, a2, b2, xt1, x2t1, vt1, ut, tmp;
|
SIMD_INT d, z, a, b, a2, b2, xt1, x2t1, vt1, ut, tmp;
|
||||||
__dp_code_block1;
|
__dp_code_block1;
|
||||||
#ifdef __SSE4_1__
|
#if defined(__AVX512BW__)
|
||||||
d = _mm_andnot_si128(_mm_cmpgt_epi8(z, a), _mm_set1_epi8(1)); // d = z > a? 0 : 1
|
d = _mm512_maskz_set1_epi8(_mm512_cmpge_epi8_mask(a, z), 1);
|
||||||
z = _mm_max_epi8(z, a);
|
z = _mm512_max_epi8(z, a);
|
||||||
d = _mm_blendv_epi8(_mm_set1_epi8(2), d, _mm_cmpgt_epi8(z, b)); // d = z > b? d : 2
|
d = _mm512_mask_blend_epi8(_mm512_cmpge_epi8_mask(b, z), d, _mm512_set1_epi8(2));
|
||||||
z = _mm_max_epi8(z, b);
|
z = _mm512_max_epi8(z, b);
|
||||||
d = _mm_blendv_epi8(_mm_set1_epi8(3), d, _mm_cmpgt_epi8(z, a2)); // d = z > a2? d : 3
|
d = _mm512_mask_blend_epi8(_mm512_cmpge_epi8_mask(a2, z), d, _mm512_set1_epi8(3));
|
||||||
z = _mm_max_epi8(z, a2);
|
z = _mm512_max_epi8(z, a2);
|
||||||
d = _mm_blendv_epi8(_mm_set1_epi8(4), d, _mm_cmpgt_epi8(z, b2)); // d = z > b2? d : 4
|
d = _mm512_mask_blend_epi8(_mm512_cmpge_epi8_mask(b2, z), d, _mm512_set1_epi8(4));
|
||||||
z = _mm_max_epi8(z, b2);
|
z = _mm512_max_epi8(z, b2);
|
||||||
z = _mm_min_epi8(z, sc_mch_);
|
z = _mm512_min_epi8(z, sc_mch_);
|
||||||
#else // we need to emulate SSE4.1 intrinsics _mm_max_epi8() and _mm_blendv_epi8()
|
__dp_code_block2;
|
||||||
|
d = _mm512_or_si512(d, _mm512_maskz_set1_epi8(_mm512_cmpge_epi8_mask(a, zero_), 0x08)); // d = a >= 0? 1<<3 : 0
|
||||||
|
_mm512_store_si512(&x[t], _mm512_sub_epi8(_mm512_max_epi8(a, zero_), qe_));
|
||||||
|
d = _mm512_or_si512(d, _mm512_maskz_set1_epi8(_mm512_cmpge_epi8_mask(b, zero_), 0x10)); // d = b >= 0? 1<<4 : 0
|
||||||
|
_mm512_store_si512(&y[t], _mm512_sub_epi8(_mm512_max_epi8(b, zero_), qe_));
|
||||||
|
d = _mm512_or_si512(d, _mm512_maskz_set1_epi8(_mm512_cmpge_epi8_mask(a2, zero_), 0x20)); // d = a2 >= 0? 1<<5 : 0
|
||||||
|
_mm512_store_si512(&x2[t], _mm512_sub_epi8(_mm512_max_epi8(a2, zero_), qe2_));
|
||||||
|
d = _mm512_or_si512(d, _mm512_maskz_set1_epi8(_mm512_cmpge_epi8_mask(b2, zero_), 0x40)); // d = b2 >= 0? 1<<6 : 0
|
||||||
|
_mm512_store_si512(&y2[t], _mm512_sub_epi8(_mm512_max_epi8(b2, zero_), qe2_));
|
||||||
|
#else
|
||||||
|
#if defined(__SSE4_1__) || defined(__AVX2__)
|
||||||
|
d = simd_funcw(andnot)(simd_func(cmpgt_epi8)(z, a), simd_func(set1_epi8)(1)); // d = z > a? 0 : 1
|
||||||
|
z = simd_func(max_epi8)(z, a);
|
||||||
|
d = simd_func(blendv_epi8)(simd_func(set1_epi8)(2), d, simd_func(cmpgt_epi8)(z, b)); // d = z > b? d : 2
|
||||||
|
z = simd_func(max_epi8)(z, b);
|
||||||
|
d = simd_func(blendv_epi8)(simd_func(set1_epi8)(3), d, simd_func(cmpgt_epi8)(z, a2)); // d = z > a2? d : 3
|
||||||
|
z = simd_func(max_epi8)(z, a2);
|
||||||
|
d = simd_func(blendv_epi8)(simd_func(set1_epi8)(4), d, simd_func(cmpgt_epi8)(z, b2)); // d = z > b2? d : 4
|
||||||
|
z = simd_func(max_epi8)(z, b2);
|
||||||
|
z = simd_func(min_epi8)(z, sc_mch_);
|
||||||
|
#elif defined(__SSE2__)
|
||||||
tmp = _mm_cmpgt_epi8(z, a);
|
tmp = _mm_cmpgt_epi8(z, a);
|
||||||
d = _mm_andnot_si128(tmp, _mm_set1_epi8(1));
|
d = _mm_andnot_si128(tmp, _mm_set1_epi8(1));
|
||||||
z = _mm_or_si128(_mm_and_si128(tmp, z), _mm_andnot_si128(tmp, a));
|
z = _mm_or_si128(_mm_and_si128(tmp, z), _mm_andnot_si128(tmp, a));
|
||||||
@@ -295,52 +416,64 @@ void ksw_extd2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
z = _mm_or_si128(_mm_and_si128(tmp, z), _mm_andnot_si128(tmp, b2));
|
z = _mm_or_si128(_mm_and_si128(tmp, z), _mm_andnot_si128(tmp, b2));
|
||||||
tmp = _mm_cmplt_epi8(sc_mch_, z);
|
tmp = _mm_cmplt_epi8(sc_mch_, z);
|
||||||
z = _mm_or_si128(_mm_and_si128(tmp, sc_mch_), _mm_andnot_si128(tmp, z));
|
z = _mm_or_si128(_mm_and_si128(tmp, sc_mch_), _mm_andnot_si128(tmp, z));
|
||||||
#endif
|
#endif // ~__SSE2__
|
||||||
__dp_code_block2;
|
__dp_code_block2;
|
||||||
tmp = _mm_cmpgt_epi8(zero_, a);
|
tmp = simd_func(cmpgt_epi8)(zero_, a);
|
||||||
_mm_store_si128(&x[t], _mm_sub_epi8(_mm_andnot_si128(tmp, a), qe_));
|
simd_funcw(store)(&x[t], simd_func(sub_epi8)(simd_funcw(andnot)(tmp, a), qe_));
|
||||||
d = _mm_or_si128(d, _mm_andnot_si128(tmp, _mm_set1_epi8(0x08))); // d = a > 0? 1<<3 : 0
|
d = simd_funcw(or)(d, simd_funcw(andnot)(tmp, simd_func(set1_epi8)(0x08))); // d = a > 0? 1<<3 : 0
|
||||||
tmp = _mm_cmpgt_epi8(zero_, b);
|
tmp = simd_func(cmpgt_epi8)(zero_, b);
|
||||||
_mm_store_si128(&y[t], _mm_sub_epi8(_mm_andnot_si128(tmp, b), qe_));
|
simd_funcw(store)(&y[t], simd_func(sub_epi8)(simd_funcw(andnot)(tmp, b), qe_));
|
||||||
d = _mm_or_si128(d, _mm_andnot_si128(tmp, _mm_set1_epi8(0x10))); // d = b > 0? 1<<4 : 0
|
d = simd_funcw(or)(d, simd_funcw(andnot)(tmp, simd_func(set1_epi8)(0x10))); // d = b > 0? 1<<4 : 0
|
||||||
tmp = _mm_cmpgt_epi8(zero_, a2);
|
tmp = simd_func(cmpgt_epi8)(zero_, a2);
|
||||||
_mm_store_si128(&x2[t], _mm_sub_epi8(_mm_andnot_si128(tmp, a2), qe2_));
|
simd_funcw(store)(&x2[t], simd_func(sub_epi8)(simd_funcw(andnot)(tmp, a2), qe2_));
|
||||||
d = _mm_or_si128(d, _mm_andnot_si128(tmp, _mm_set1_epi8(0x20))); // d = a > 0? 1<<5 : 0
|
d = simd_funcw(or)(d, simd_funcw(andnot)(tmp, simd_func(set1_epi8)(0x20))); // d = a > 0? 1<<5 : 0
|
||||||
tmp = _mm_cmpgt_epi8(zero_, b2);
|
tmp = simd_func(cmpgt_epi8)(zero_, b2);
|
||||||
_mm_store_si128(&y2[t], _mm_sub_epi8(_mm_andnot_si128(tmp, b2), qe2_));
|
simd_funcw(store)(&y2[t], simd_func(sub_epi8)(simd_funcw(andnot)(tmp, b2), qe2_));
|
||||||
d = _mm_or_si128(d, _mm_andnot_si128(tmp, _mm_set1_epi8(0x40))); // d = b > 0? 1<<6 : 0
|
d = simd_funcw(or)(d, simd_funcw(andnot)(tmp, simd_func(set1_epi8)(0x40))); // d = b > 0? 1<<6 : 0
|
||||||
_mm_store_si128(&pr[t], d);
|
#endif // ~__AVX512BW__
|
||||||
|
simd_funcw(store)(&pr[t], d);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (!approx_max) { // find the exact max with a 32-bit score array
|
if (!approx_max) { // find the exact max with a 32-bit score array
|
||||||
int32_t max_H, max_t;
|
int32_t max_H, max_t;
|
||||||
// compute H[], max_H and max_t
|
// compute H[], max_H and max_t
|
||||||
if (r > 0) {
|
if (r > 0) {
|
||||||
int32_t HH[4], tt[4], en1 = st0 + (en0 - st0) / 4 * 4, i;
|
int32_t HH[SIMD_WIDTH/4], tt[SIMD_WIDTH/4], en1 = st0 + (en0 - st0) / (SIMD_WIDTH/4) * (SIMD_WIDTH/4), i;
|
||||||
__m128i max_H_, max_t_;
|
SIMD_INT max_H_, max_t_;
|
||||||
max_H = H[en0] = en0 > 0? H[en0-1] + u8[en0] : H[en0] + v8[en0]; // special casing the last element
|
max_H = H[en0] = en0 > 0? H[en0-1] + u8[en0] : H[en0] + v8[en0]; // special casing the last element
|
||||||
max_t = en0;
|
max_t = en0;
|
||||||
max_H_ = _mm_set1_epi32(max_H);
|
max_H_ = simd_func(set1_epi32)(max_H);
|
||||||
max_t_ = _mm_set1_epi32(max_t);
|
max_t_ = simd_func(set1_epi32)(max_t);
|
||||||
for (t = st0; t < en1; t += 4) { // this implements: H[t]+=v8[t]-qe; if(H[t]>max_H) max_H=H[t],max_t=t;
|
for (t = st0; t < en1; t += SIMD_WIDTH/4) { // this implements: H[t]+=v8[t]; if(H[t]>max_H) max_H=H[t],max_t=t;
|
||||||
__m128i H1, tmp, t_;
|
SIMD_INT H1, t_;
|
||||||
H1 = _mm_loadu_si128((__m128i*)&H[t]);
|
H1 = simd_funcw(loadu)((SIMD_INT*)&H[t]);
|
||||||
|
#if defined(__AVX512BW__)
|
||||||
|
t_ = _mm512_cvtepi8_epi32(_mm_loadu_si128((__m128i*)&v8[t]));
|
||||||
|
#elif defined(__AVX2__)
|
||||||
|
t_ = _mm256_setr_epi32(v8[t], v8[t+1], v8[t+2], v8[t+3], v8[t+4], v8[t+5], v8[t+6], v8[t+7]);
|
||||||
|
#elif defined(__SSE2__)
|
||||||
t_ = _mm_setr_epi32(v8[t], v8[t+1], v8[t+2], v8[t+3]);
|
t_ = _mm_setr_epi32(v8[t], v8[t+1], v8[t+2], v8[t+3]);
|
||||||
H1 = _mm_add_epi32(H1, t_);
|
#endif
|
||||||
_mm_storeu_si128((__m128i*)&H[t], H1);
|
H1 = simd_func(add_epi32)(H1, t_);
|
||||||
t_ = _mm_set1_epi32(t);
|
simd_funcw(storeu)((SIMD_INT*)&H[t], H1);
|
||||||
tmp = _mm_cmpgt_epi32(H1, max_H_);
|
t_ = simd_func(set1_epi32)(t);
|
||||||
#ifdef __SSE4_1__
|
#if defined(__AVX512BW__)
|
||||||
max_H_ = _mm_blendv_epi8(max_H_, H1, tmp);
|
__mmask64 tmp = _mm512_cmpgt_epi32_mask(H1, max_H_);
|
||||||
max_t_ = _mm_blendv_epi8(max_t_, t_, tmp);
|
max_H_ = _mm512_mask_blend_epi32(tmp, max_H_, H1);
|
||||||
#else
|
max_t_ = _mm512_mask_blend_epi32(tmp, max_t_, t_);
|
||||||
max_H_ = _mm_or_si128(_mm_and_si128(tmp, H1), _mm_andnot_si128(tmp, max_H_));
|
#elif defined(__SSE4_1__) || defined(__AVX2__)
|
||||||
max_t_ = _mm_or_si128(_mm_and_si128(tmp, t_), _mm_andnot_si128(tmp, max_t_));
|
SIMD_INT tmp = simd_func(cmpgt_epi32)(H1, max_H_);
|
||||||
|
max_H_ = simd_func(blendv_epi8)(max_H_, H1, tmp);
|
||||||
|
max_t_ = simd_func(blendv_epi8)(max_t_, t_, tmp);
|
||||||
|
#elif defined(__SSE2__)
|
||||||
|
SIMD_INT tmp = simd_func(cmpgt_epi32)(H1, max_H_);
|
||||||
|
max_H_ = simd_funcw(or)(simd_funcw(and)(tmp, H1), simd_funcw(andnot)(tmp, max_H_));
|
||||||
|
max_t_ = simd_funcw(or)(simd_funcw(and)(tmp, t_), simd_funcw(andnot)(tmp, max_t_));
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
_mm_storeu_si128((__m128i*)HH, max_H_);
|
simd_funcw(storeu)((SIMD_INT*)HH, max_H_);
|
||||||
_mm_storeu_si128((__m128i*)tt, max_t_);
|
simd_funcw(storeu)((SIMD_INT*)tt, max_t_);
|
||||||
for (i = 0; i < 4; ++i)
|
for (i = 0; i < SIMD_WIDTH/4; ++i)
|
||||||
if (max_H < HH[i]) max_H = HH[i], max_t = tt[i] + i;
|
if (max_H < HH[i]) max_H = HH[i], max_t = tt[i] + i;
|
||||||
for (; t < en0; ++t) { // for the rest of values that haven't been computed with SSE
|
for (; t < en0; ++t) { // for the rest of values that haven't been computed with SSE
|
||||||
H[t] += (int32_t)v8[t];
|
H[t] += (int32_t)v8[t];
|
||||||
@@ -381,12 +514,12 @@ void ksw_extd2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
if (with_cigar) { // backtrack
|
if (with_cigar) { // backtrack
|
||||||
int rev_cigar = !!(flag & KSW_EZ_REV_CIGAR);
|
int rev_cigar = !!(flag & KSW_EZ_REV_CIGAR);
|
||||||
if (!ez->zdropped && !(flag&KSW_EZ_EXTZ_ONLY)) {
|
if (!ez->zdropped && !(flag&KSW_EZ_EXTZ_ONLY)) {
|
||||||
ksw_backtrack(km, 1, rev_cigar, 0, (uint8_t*)p, off, off_end, n_col_*16, tlen-1, qlen-1, &ez->m_cigar, &ez->n_cigar, &ez->cigar);
|
ksw_backtrack(km, 1, rev_cigar, 0, (uint8_t*)p, off, off_end, n_col_*SIMD_WIDTH, tlen-1, qlen-1, &ez->m_cigar, &ez->n_cigar, &ez->cigar);
|
||||||
} else if (!ez->zdropped && (flag&KSW_EZ_EXTZ_ONLY) && ez->mqe + end_bonus > (int)ez->max) {
|
} else if (!ez->zdropped && (flag&KSW_EZ_EXTZ_ONLY) && ez->mqe + end_bonus > (int)ez->max) {
|
||||||
ez->reach_end = 1;
|
ez->reach_end = 1;
|
||||||
ksw_backtrack(km, 1, rev_cigar, 0, (uint8_t*)p, off, off_end, n_col_*16, ez->mqe_t, qlen-1, &ez->m_cigar, &ez->n_cigar, &ez->cigar);
|
ksw_backtrack(km, 1, rev_cigar, 0, (uint8_t*)p, off, off_end, n_col_*SIMD_WIDTH, ez->mqe_t, qlen-1, &ez->m_cigar, &ez->n_cigar, &ez->cigar);
|
||||||
} else if (ez->max_t >= 0 && ez->max_q >= 0) {
|
} else if (ez->max_t >= 0 && ez->max_q >= 0) {
|
||||||
ksw_backtrack(km, 1, rev_cigar, 0, (uint8_t*)p, off, off_end, n_col_*16, ez->max_t, ez->max_q, &ez->m_cigar, &ez->n_cigar, &ez->cigar);
|
ksw_backtrack(km, 1, rev_cigar, 0, (uint8_t*)p, off, off_end, n_col_*SIMD_WIDTH, ez->max_t, ez->max_q, &ez->m_cigar, &ez->n_cigar, &ez->cigar);
|
||||||
}
|
}
|
||||||
kfree(km, mem2); kfree(km, off);
|
kfree(km, mem2); kfree(km, off);
|
||||||
}
|
}
|
||||||
|
|||||||
+49
-16
@@ -17,14 +17,14 @@
|
|||||||
#ifdef KSW_CPU_DISPATCH
|
#ifdef KSW_CPU_DISPATCH
|
||||||
#ifdef __SSE4_1__
|
#ifdef __SSE4_1__
|
||||||
void ksw_exts2_sse41(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
void ksw_exts2_sse41(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
int8_t q, int8_t e, int8_t q2, int8_t noncan, int zdrop, int flag, ksw_extz_t *ez)
|
int8_t q, int8_t e, int8_t q2, int8_t noncan, int zdrop, int8_t junc_bonus, int flag, const uint8_t *junc, ksw_extz_t *ez)
|
||||||
#else
|
#else
|
||||||
void ksw_exts2_sse2(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
void ksw_exts2_sse2(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
int8_t q, int8_t e, int8_t q2, int8_t noncan, int zdrop, int flag, ksw_extz_t *ez)
|
int8_t q, int8_t e, int8_t q2, int8_t noncan, int zdrop, int8_t junc_bonus, int flag, const uint8_t *junc, ksw_extz_t *ez)
|
||||||
#endif
|
#endif
|
||||||
#else
|
#else
|
||||||
void ksw_exts2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
void ksw_exts2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uint8_t *target, int8_t m, const int8_t *mat,
|
||||||
int8_t q, int8_t e, int8_t q2, int8_t noncan, int zdrop, int flag, ksw_extz_t *ez)
|
int8_t q, int8_t e, int8_t q2, int8_t noncan, int zdrop, int8_t junc_bonus, int flag, const uint8_t *junc, ksw_extz_t *ez)
|
||||||
#endif // ~KSW_CPU_DISPATCH
|
#endif // ~KSW_CPU_DISPATCH
|
||||||
{
|
{
|
||||||
#define __dp_code_block1 \
|
#define __dp_code_block1 \
|
||||||
@@ -113,20 +113,53 @@ void ksw_exts2_sse(void *km, int qlen, const uint8_t *query, int tlen, const uin
|
|||||||
if (flag & (KSW_EZ_SPLICE_FOR|KSW_EZ_SPLICE_REV)) {
|
if (flag & (KSW_EZ_SPLICE_FOR|KSW_EZ_SPLICE_REV)) {
|
||||||
int semi_cost = flag&KSW_EZ_SPLICE_FLANK? -noncan/2 : 0; // GTr or yAG is worth 0.5 bit; see PMID:18688272
|
int semi_cost = flag&KSW_EZ_SPLICE_FLANK? -noncan/2 : 0; // GTr or yAG is worth 0.5 bit; see PMID:18688272
|
||||||
memset(donor, -noncan, tlen_ * 16);
|
memset(donor, -noncan, tlen_ * 16);
|
||||||
for (t = 0; t < tlen - 4; ++t) {
|
|
||||||
int can_type = 0; // type of canonical site: 0=none, 1=GT/AG only, 2=GTr/yAG
|
|
||||||
if ((flag & KSW_EZ_SPLICE_FOR) && target[t+1] == 2 && target[t+2] == 3) can_type = 1; // GTr...
|
|
||||||
if ((flag & KSW_EZ_SPLICE_REV) && target[t+1] == 1 && target[t+2] == 3) can_type = 1; // CTr...
|
|
||||||
if (can_type && (target[t+3] == 0 || target[t+3] == 2)) can_type = 2;
|
|
||||||
if (can_type) ((int8_t*)donor)[t] = can_type == 2? 0 : semi_cost;
|
|
||||||
}
|
|
||||||
memset(acceptor, -noncan, tlen_ * 16);
|
memset(acceptor, -noncan, tlen_ * 16);
|
||||||
for (t = 2; t < tlen; ++t) {
|
if (!(flag & KSW_EZ_REV_CIGAR)) {
|
||||||
int can_type = 0;
|
for (t = 0; t < tlen - 4; ++t) {
|
||||||
if ((flag & KSW_EZ_SPLICE_FOR) && target[t-1] == 0 && target[t] == 2) can_type = 1; // ...yAG
|
int can_type = 0; // type of canonical site: 0=none, 1=GT/AG only, 2=GTr/yAG
|
||||||
if ((flag & KSW_EZ_SPLICE_REV) && target[t-1] == 0 && target[t] == 1) can_type = 1; // ...yAC
|
if ((flag & KSW_EZ_SPLICE_FOR) && target[t+1] == 2 && target[t+2] == 3) can_type = 1; // GTr...
|
||||||
if (can_type && (target[t-2] == 1 || target[t-2] == 3)) can_type = 2;
|
if ((flag & KSW_EZ_SPLICE_REV) && target[t+1] == 1 && target[t+2] == 3) can_type = 1; // CTr...
|
||||||
if (can_type) ((int8_t*)acceptor)[t] = can_type == 2? 0 : semi_cost;
|
if (can_type && (target[t+3] == 0 || target[t+3] == 2)) can_type = 2;
|
||||||
|
if (can_type) ((int8_t*)donor)[t] = can_type == 2? 0 : semi_cost;
|
||||||
|
}
|
||||||
|
if (junc)
|
||||||
|
for (t = 0; t < tlen - 1; ++t)
|
||||||
|
if (((flag & KSW_EZ_SPLICE_FOR) && (junc[t+1]&1)) || ((flag & KSW_EZ_SPLICE_REV) && (junc[t+1]&8)))
|
||||||
|
((int8_t*)donor)[t] += junc_bonus;
|
||||||
|
for (t = 2; t < tlen; ++t) {
|
||||||
|
int can_type = 0;
|
||||||
|
if ((flag & KSW_EZ_SPLICE_FOR) && target[t-1] == 0 && target[t] == 2) can_type = 1; // ...yAG
|
||||||
|
if ((flag & KSW_EZ_SPLICE_REV) && target[t-1] == 0 && target[t] == 1) can_type = 1; // ...yAC
|
||||||
|
if (can_type && (target[t-2] == 1 || target[t-2] == 3)) can_type = 2;
|
||||||
|
if (can_type) ((int8_t*)acceptor)[t] = can_type == 2? 0 : semi_cost;
|
||||||
|
}
|
||||||
|
if (junc)
|
||||||
|
for (t = 0; t < tlen; ++t)
|
||||||
|
if (((flag & KSW_EZ_SPLICE_FOR) && (junc[t]&2)) || ((flag & KSW_EZ_SPLICE_REV) && (junc[t]&4)))
|
||||||
|
((int8_t*)acceptor)[t] += junc_bonus;
|
||||||
|
} else {
|
||||||
|
for (t = 0; t < tlen - 4; ++t) {
|
||||||
|
int can_type = 0; // type of canonical site: 0=none, 1=GT/AG only, 2=GTr/yAG
|
||||||
|
if ((flag & KSW_EZ_SPLICE_FOR) && target[t+1] == 2 && target[t+2] == 0) can_type = 1; // GAy...
|
||||||
|
if ((flag & KSW_EZ_SPLICE_REV) && target[t+1] == 1 && target[t+2] == 0) can_type = 1; // CAy...
|
||||||
|
if (can_type && (target[t+3] == 1 || target[t+3] == 3)) can_type = 2;
|
||||||
|
if (can_type) ((int8_t*)donor)[t] = can_type == 2? 0 : semi_cost;
|
||||||
|
}
|
||||||
|
if (junc)
|
||||||
|
for (t = 0; t < tlen - 1; ++t)
|
||||||
|
if (((flag & KSW_EZ_SPLICE_FOR) && (junc[t+1]&2)) || ((flag & KSW_EZ_SPLICE_REV) && (junc[t+1]&4)))
|
||||||
|
((int8_t*)donor)[t] += junc_bonus;
|
||||||
|
for (t = 2; t < tlen; ++t) {
|
||||||
|
int can_type = 0;
|
||||||
|
if ((flag & KSW_EZ_SPLICE_FOR) && target[t-1] == 3 && target[t] == 2) can_type = 1; // ...rTG
|
||||||
|
if ((flag & KSW_EZ_SPLICE_REV) && target[t-1] == 3 && target[t] == 1) can_type = 1; // ...rTC
|
||||||
|
if (can_type && (target[t-2] == 0 || target[t-2] == 2)) can_type = 2;
|
||||||
|
if (can_type) ((int8_t*)acceptor)[t] = can_type == 2? 0 : semi_cost;
|
||||||
|
}
|
||||||
|
if (junc)
|
||||||
|
for (t = 0; t < tlen; ++t)
|
||||||
|
if (((flag & KSW_EZ_SPLICE_FOR) && (junc[t]&1)) || ((flag & KSW_EZ_SPLICE_REV) && (junc[t]&8)))
|
||||||
|
((int8_t*)acceptor)[t] += junc_bonus;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,12 +1,13 @@
|
|||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
#include <errno.h>
|
||||||
#include "bseq.h"
|
#include "bseq.h"
|
||||||
#include "minimap.h"
|
#include "minimap.h"
|
||||||
#include "mmpriv.h"
|
#include "mmpriv.h"
|
||||||
#include "ketopt.h"
|
#include "ketopt.h"
|
||||||
|
|
||||||
#define MM_VERSION "2.15-r905"
|
#define MM_VERSION "2.17-r963-dirty"
|
||||||
|
|
||||||
#ifdef __linux__
|
#ifdef __linux__
|
||||||
#include <sys/resource.h>
|
#include <sys/resource.h>
|
||||||
@@ -62,6 +63,10 @@ static ko_longopt_t long_options[] = {
|
|||||||
{ "hard-mask-level",ko_no_argument, 336 },
|
{ "hard-mask-level",ko_no_argument, 336 },
|
||||||
{ "cap-sw-mem", ko_required_argument, 337 },
|
{ "cap-sw-mem", ko_required_argument, 337 },
|
||||||
{ "max-qlen", ko_required_argument, 338 },
|
{ "max-qlen", ko_required_argument, 338 },
|
||||||
|
{ "max-chain-iter", ko_required_argument, 339 },
|
||||||
|
{ "junc-bed", ko_required_argument, 340 },
|
||||||
|
{ "junc-bonus", ko_required_argument, 341 },
|
||||||
|
{ "sam-hit-only", ko_no_argument, 342 },
|
||||||
{ "help", ko_no_argument, 'h' },
|
{ "help", ko_no_argument, 'h' },
|
||||||
{ "max-intron-len", ko_required_argument, 'G' },
|
{ "max-intron-len", ko_required_argument, 'G' },
|
||||||
{ "version", ko_no_argument, 'V' },
|
{ "version", ko_no_argument, 'V' },
|
||||||
@@ -99,12 +104,12 @@ static inline void yes_or_no(mm_mapopt_t *opt, int flag, int long_idx, const cha
|
|||||||
|
|
||||||
int main(int argc, char *argv[])
|
int main(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
const char *opt_str = "2aSDw:k:K:t:r:f:Vv:g:G:I:d:XT:s:x:Hcp:M:n:z:A:B:O:E:m:N:Qu:R:hF:LC:yYP";
|
const char *opt_str = "2aSDw:k:K:t:r:f:Vv:g:G:I:d:XT:s:x:Hcp:M:n:z:A:B:O:E:m:N:Qu:R:hF:LC:yYPo:";
|
||||||
ketopt_t o = KETOPT_INIT;
|
ketopt_t o = KETOPT_INIT;
|
||||||
mm_mapopt_t opt;
|
mm_mapopt_t opt;
|
||||||
mm_idxopt_t ipt;
|
mm_idxopt_t ipt;
|
||||||
int i, c, n_threads = 3, n_parts, old_best_n = -1;
|
int i, c, n_threads = 3, n_parts, old_best_n = -1;
|
||||||
char *fnw = 0, *rg = 0, *s;
|
char *fnw = 0, *rg = 0, *junc_bed = 0, *s;
|
||||||
FILE *fp_help = stderr;
|
FILE *fp_help = stderr;
|
||||||
mm_idx_reader_t *idx_rdr;
|
mm_idx_reader_t *idx_rdr;
|
||||||
mm_idx_t *mi;
|
mm_idx_t *mi;
|
||||||
@@ -165,12 +170,21 @@ int main(int argc, char *argv[])
|
|||||||
else if (c == 'R') rg = o.arg;
|
else if (c == 'R') rg = o.arg;
|
||||||
else if (c == 'h') fp_help = stdout;
|
else if (c == 'h') fp_help = stdout;
|
||||||
else if (c == '2') opt.flag |= MM_F_2_IO_THREADS;
|
else if (c == '2') opt.flag |= MM_F_2_IO_THREADS;
|
||||||
|
else if (c == 'o') {
|
||||||
|
if (strcmp(o.arg, "-") != 0) {
|
||||||
|
if (freopen(o.arg, "wb", stdout) == NULL) {
|
||||||
|
fprintf(stderr, "[ERROR]\033[1;31m failed to write the output to file '%s'\033[0m: %s\n", o.arg, strerror(errno));
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
else if (c == 300) ipt.bucket_bits = atoi(o.arg); // --bucket-bits
|
else if (c == 300) ipt.bucket_bits = atoi(o.arg); // --bucket-bits
|
||||||
else if (c == 302) opt.seed = atoi(o.arg); // --seed
|
else if (c == 302) opt.seed = atoi(o.arg); // --seed
|
||||||
else if (c == 303) mm_dbg_flag |= MM_DBG_NO_KALLOC; // --no-kalloc
|
else if (c == 303) mm_dbg_flag |= MM_DBG_NO_KALLOC; // --no-kalloc
|
||||||
else if (c == 304) mm_dbg_flag |= MM_DBG_PRINT_QNAME; // --print-qname
|
else if (c == 304) mm_dbg_flag |= MM_DBG_PRINT_QNAME; // --print-qname
|
||||||
else if (c == 306) mm_dbg_flag |= MM_DBG_PRINT_QNAME | MM_DBG_PRINT_SEED, n_threads = 1; // --print-seed
|
else if (c == 306) mm_dbg_flag |= MM_DBG_PRINT_QNAME | MM_DBG_PRINT_SEED, n_threads = 1; // --print-seed
|
||||||
else if (c == 307) opt.max_chain_skip = atoi(o.arg); // --max-chain-skip
|
else if (c == 307) opt.max_chain_skip = atoi(o.arg); // --max-chain-skip
|
||||||
|
else if (c == 339) opt.max_chain_iter = atoi(o.arg); // --max-chain-iter
|
||||||
else if (c == 308) opt.min_ksw_len = atoi(o.arg); // --min-dp-len
|
else if (c == 308) opt.min_ksw_len = atoi(o.arg); // --min-dp-len
|
||||||
else if (c == 309) mm_dbg_flag |= MM_DBG_PRINT_QNAME | MM_DBG_PRINT_ALN_SEQ, n_threads = 1; // --print-aln-seq
|
else if (c == 309) mm_dbg_flag |= MM_DBG_PRINT_QNAME | MM_DBG_PRINT_ALN_SEQ, n_threads = 1; // --print-aln-seq
|
||||||
else if (c == 310) opt.flag |= MM_F_SPLICE; // --splice
|
else if (c == 310) opt.flag |= MM_F_SPLICE; // --splice
|
||||||
@@ -194,6 +208,9 @@ int main(int argc, char *argv[])
|
|||||||
else if (c == 336) opt.flag |= MM_F_HARD_MLEVEL; // --hard-mask-level
|
else if (c == 336) opt.flag |= MM_F_HARD_MLEVEL; // --hard-mask-level
|
||||||
else if (c == 337) opt.max_sw_mat = mm_parse_num(o.arg); // --cap-sw-mat
|
else if (c == 337) opt.max_sw_mat = mm_parse_num(o.arg); // --cap-sw-mat
|
||||||
else if (c == 338) opt.max_qlen = mm_parse_num(o.arg); // --max-qlen
|
else if (c == 338) opt.max_qlen = mm_parse_num(o.arg); // --max-qlen
|
||||||
|
else if (c == 340) junc_bed = o.arg; // --junc-bed
|
||||||
|
else if (c == 341) opt.junc_bonus = atoi(o.arg); // --junc-bonus
|
||||||
|
else if (c == 342) opt.flag |= MM_F_SAM_HIT_ONLY; // --sam-hit-only
|
||||||
else if (c == 314) { // --frag
|
else if (c == 314) { // --frag
|
||||||
yes_or_no(&opt, MM_F_FRAG_MODE, o.longidx, o.arg, 1);
|
yes_or_no(&opt, MM_F_FRAG_MODE, o.longidx, o.arg, 1);
|
||||||
} else if (c == 315) { // --secondary
|
} else if (c == 315) { // --secondary
|
||||||
@@ -268,7 +285,7 @@ int main(int argc, char *argv[])
|
|||||||
fprintf(fp_help, " Indexing:\n");
|
fprintf(fp_help, " Indexing:\n");
|
||||||
fprintf(fp_help, " -H use homopolymer-compressed k-mer (preferrable for PacBio)\n");
|
fprintf(fp_help, " -H use homopolymer-compressed k-mer (preferrable for PacBio)\n");
|
||||||
fprintf(fp_help, " -k INT k-mer size (no larger than 28) [%d]\n", ipt.k);
|
fprintf(fp_help, " -k INT k-mer size (no larger than 28) [%d]\n", ipt.k);
|
||||||
fprintf(fp_help, " -w INT minizer window size [%d]\n", ipt.w);
|
fprintf(fp_help, " -w INT minimizer window size [%d]\n", ipt.w);
|
||||||
fprintf(fp_help, " -I NUM split index for every ~NUM input bases [4G]\n");
|
fprintf(fp_help, " -I NUM split index for every ~NUM input bases [4G]\n");
|
||||||
fprintf(fp_help, " -d FILE dump index to FILE []\n");
|
fprintf(fp_help, " -d FILE dump index to FILE []\n");
|
||||||
fprintf(fp_help, " Mapping:\n");
|
fprintf(fp_help, " Mapping:\n");
|
||||||
@@ -293,7 +310,7 @@ int main(int argc, char *argv[])
|
|||||||
fprintf(fp_help, " -u CHAR how to find GT-AG. f:transcript strand, b:both strands, n:don't match GT-AG [n]\n");
|
fprintf(fp_help, " -u CHAR how to find GT-AG. f:transcript strand, b:both strands, n:don't match GT-AG [n]\n");
|
||||||
fprintf(fp_help, " Input/Output:\n");
|
fprintf(fp_help, " Input/Output:\n");
|
||||||
fprintf(fp_help, " -a output in the SAM format (PAF by default)\n");
|
fprintf(fp_help, " -a output in the SAM format (PAF by default)\n");
|
||||||
fprintf(fp_help, " -Q don't output base quality in SAM\n");
|
fprintf(fp_help, " -o FILE output alignments to FILE [stdout]\n");
|
||||||
fprintf(fp_help, " -L write CIGAR with >65535 ops at the CG tag\n");
|
fprintf(fp_help, " -L write CIGAR with >65535 ops at the CG tag\n");
|
||||||
fprintf(fp_help, " -R STR SAM read group line in a format like '@RG\\tID:foo\\tSM:bar' []\n");
|
fprintf(fp_help, " -R STR SAM read group line in a format like '@RG\\tID:foo\\tSM:bar' []\n");
|
||||||
fprintf(fp_help, " -c output CIGAR in PAF\n");
|
fprintf(fp_help, " -c output CIGAR in PAF\n");
|
||||||
@@ -307,11 +324,11 @@ int main(int argc, char *argv[])
|
|||||||
fprintf(fp_help, " --version show version number\n");
|
fprintf(fp_help, " --version show version number\n");
|
||||||
fprintf(fp_help, " Preset:\n");
|
fprintf(fp_help, " Preset:\n");
|
||||||
fprintf(fp_help, " -x STR preset (always applied before other options; see minimap2.1 for details) []\n");
|
fprintf(fp_help, " -x STR preset (always applied before other options; see minimap2.1 for details) []\n");
|
||||||
fprintf(fp_help, " - map-pb/map-ont: PacBio/Nanopore vs reference mapping\n");
|
fprintf(fp_help, " - map-pb/map-ont - PacBio/Nanopore vs reference mapping\n");
|
||||||
fprintf(fp_help, " - ava-pb/ava-ont: PacBio/Nanopore read overlap\n");
|
fprintf(fp_help, " - ava-pb/ava-ont - PacBio/Nanopore read overlap\n");
|
||||||
fprintf(fp_help, " - asm5/asm10/asm20: asm-to-ref mapping, for ~0.1/1/5%% sequence divergence\n");
|
fprintf(fp_help, " - asm5/asm10/asm20 - asm-to-ref mapping, for ~0.1/1/5%% sequence divergence\n");
|
||||||
fprintf(fp_help, " - splice: long-read spliced alignment\n");
|
fprintf(fp_help, " - splice/splice:hq - long-read/Pacbio-CCS spliced alignment\n");
|
||||||
fprintf(fp_help, " - sr: genomic short-read mapping\n");
|
fprintf(fp_help, " - sr - genomic short-read mapping\n");
|
||||||
fprintf(fp_help, "\nSee `man ./minimap2.1' for detailed description of these and other advanced command-line options.\n");
|
fprintf(fp_help, "\nSee `man ./minimap2.1' for detailed description of these and other advanced command-line options.\n");
|
||||||
return fp_help == stdout? 0 : 1;
|
return fp_help == stdout? 0 : 1;
|
||||||
}
|
}
|
||||||
@@ -322,7 +339,7 @@ int main(int argc, char *argv[])
|
|||||||
}
|
}
|
||||||
idx_rdr = mm_idx_reader_open(argv[o.ind], &ipt, fnw);
|
idx_rdr = mm_idx_reader_open(argv[o.ind], &ipt, fnw);
|
||||||
if (idx_rdr == 0) {
|
if (idx_rdr == 0) {
|
||||||
fprintf(stderr, "[ERROR] failed to open file '%s'\n", argv[o.ind]);
|
fprintf(stderr, "[ERROR] failed to open file '%s': %s\n", argv[o.ind], strerror(errno));
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
if (!idx_rdr->is_idx && fnw == 0 && argc - o.ind < 2) {
|
if (!idx_rdr->is_idx && fnw == 0 && argc - o.ind < 2) {
|
||||||
@@ -340,19 +357,26 @@ int main(int argc, char *argv[])
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
if ((opt.flag & MM_F_OUT_SAM) && idx_rdr->n_parts == 1) {
|
if ((opt.flag & MM_F_OUT_SAM) && idx_rdr->n_parts == 1) {
|
||||||
|
int ret;
|
||||||
if (mm_idx_reader_eof(idx_rdr)) {
|
if (mm_idx_reader_eof(idx_rdr)) {
|
||||||
mm_write_sam_hdr(mi, rg, MM_VERSION, argc, argv);
|
ret = mm_write_sam_hdr(mi, rg, MM_VERSION, argc, argv);
|
||||||
} else {
|
} else {
|
||||||
mm_write_sam_hdr(0, rg, MM_VERSION, argc, argv);
|
ret = mm_write_sam_hdr(0, rg, MM_VERSION, argc, argv);
|
||||||
if (opt.split_prefix == 0 && mm_verbose >= 2)
|
if (opt.split_prefix == 0 && mm_verbose >= 2)
|
||||||
fprintf(stderr, "[WARNING]\033[1;31m For a multi-part index, no @SQ lines will be outputted. Please use --split-prefix.\033[0m\n");
|
fprintf(stderr, "[WARNING]\033[1;31m For a multi-part index, no @SQ lines will be outputted. Please use --split-prefix.\033[0m\n");
|
||||||
}
|
}
|
||||||
|
if (ret != 0) {
|
||||||
|
mm_idx_destroy(mi);
|
||||||
|
mm_idx_reader_close(idx_rdr);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
if (mm_verbose >= 3)
|
if (mm_verbose >= 3)
|
||||||
fprintf(stderr, "[M::%s::%.3f*%.2f] loaded/built the index for %d target sequence(s)\n",
|
fprintf(stderr, "[M::%s::%.3f*%.2f] loaded/built the index for %d target sequence(s)\n",
|
||||||
__func__, realtime() - mm_realtime0, cputime() / (realtime() - mm_realtime0), mi->n_seq);
|
__func__, realtime() - mm_realtime0, cputime() / (realtime() - mm_realtime0), mi->n_seq);
|
||||||
if (argc != o.ind + 1) mm_mapopt_update(&opt, mi);
|
if (argc != o.ind + 1) mm_mapopt_update(&opt, mi);
|
||||||
if (mm_verbose >= 3) mm_idx_stat(mi);
|
if (mm_verbose >= 3) mm_idx_stat(mi);
|
||||||
|
if (junc_bed) mm_idx_bed_read(mi, junc_bed, 1);
|
||||||
if (!(opt.flag & MM_F_FRAG_MODE)) {
|
if (!(opt.flag & MM_F_FRAG_MODE)) {
|
||||||
for (i = o.ind + 1; i < argc; ++i)
|
for (i = o.ind + 1; i < argc; ++i)
|
||||||
mm_map_file(mi, argv[i], &opt, n_threads);
|
mm_map_file(mi, argv[i], &opt, n_threads);
|
||||||
@@ -368,7 +392,7 @@ int main(int argc, char *argv[])
|
|||||||
mm_split_merge(argc - (o.ind + 1), (const char**)&argv[o.ind + 1], &opt, n_parts);
|
mm_split_merge(argc - (o.ind + 1), (const char**)&argv[o.ind + 1], &opt, n_parts);
|
||||||
|
|
||||||
if (fflush(stdout) == EOF) {
|
if (fflush(stdout) == EOF) {
|
||||||
fprintf(stderr, "[ERROR] failed to write the results\n");
|
perror("[ERROR] failed to write the results");
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
#include <assert.h>
|
#include <assert.h>
|
||||||
|
#include <errno.h>
|
||||||
#include "kthread.h"
|
#include "kthread.h"
|
||||||
#include "kvec.h"
|
#include "kvec.h"
|
||||||
#include "kalloc.h"
|
#include "kalloc.h"
|
||||||
@@ -312,7 +313,7 @@ void mm_map_frag(const mm_idx_t *mi, int n_segs, const int *qlens, const char **
|
|||||||
if (max_chain_gap_ref < opt->max_gap) max_chain_gap_ref = opt->max_gap;
|
if (max_chain_gap_ref < opt->max_gap) max_chain_gap_ref = opt->max_gap;
|
||||||
} else max_chain_gap_ref = opt->max_gap;
|
} else max_chain_gap_ref = opt->max_gap;
|
||||||
|
|
||||||
a = mm_chain_dp(max_chain_gap_ref, max_chain_gap_qry, opt->bw, opt->max_chain_skip, opt->min_cnt, opt->min_chain_score, is_splice, n_segs, n_a, a, &n_regs0, &u, b->km);
|
a = mm_chain_dp(max_chain_gap_ref, max_chain_gap_qry, opt->bw, opt->max_chain_skip, opt->max_chain_iter, opt->min_cnt, opt->min_chain_score, is_splice, n_segs, n_a, a, &n_regs0, &u, b->km);
|
||||||
|
|
||||||
if (opt->max_occ > opt->mid_occ && rep_len > 0) {
|
if (opt->max_occ > opt->mid_occ && rep_len > 0) {
|
||||||
int rechain = 0;
|
int rechain = 0;
|
||||||
@@ -334,7 +335,7 @@ void mm_map_frag(const mm_idx_t *mi, int n_segs, const int *qlens, const char **
|
|||||||
kfree(b->km, mini_pos);
|
kfree(b->km, mini_pos);
|
||||||
if (opt->flag & MM_F_HEAP_SORT) a = collect_seed_hits_heap(b->km, opt, opt->max_occ, mi, qname, &mv, qlen_sum, &n_a, &rep_len, &n_mini_pos, &mini_pos);
|
if (opt->flag & MM_F_HEAP_SORT) a = collect_seed_hits_heap(b->km, opt, opt->max_occ, mi, qname, &mv, qlen_sum, &n_a, &rep_len, &n_mini_pos, &mini_pos);
|
||||||
else a = collect_seed_hits(b->km, opt, opt->max_occ, mi, qname, &mv, qlen_sum, &n_a, &rep_len, &n_mini_pos, &mini_pos);
|
else a = collect_seed_hits(b->km, opt, opt->max_occ, mi, qname, &mv, qlen_sum, &n_a, &rep_len, &n_mini_pos, &mini_pos);
|
||||||
a = mm_chain_dp(max_chain_gap_ref, max_chain_gap_qry, opt->bw, opt->max_chain_skip, opt->min_cnt, opt->min_chain_score, is_splice, n_segs, n_a, a, &n_regs0, &u, b->km);
|
a = mm_chain_dp(max_chain_gap_ref, max_chain_gap_qry, opt->bw, opt->max_chain_skip, opt->max_chain_iter, opt->min_cnt, opt->min_chain_score, is_splice, n_segs, n_a, a, &n_regs0, &u, b->km);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
b->frag_gap = max_chain_gap_ref;
|
b->frag_gap = max_chain_gap_ref;
|
||||||
@@ -584,16 +585,16 @@ static void *worker_pipeline(void *shared, int step, void *in)
|
|||||||
if ((p->opt->flag & MM_F_NO_PRINT_2ND) && r->id != r->parent)
|
if ((p->opt->flag & MM_F_NO_PRINT_2ND) && r->id != r->parent)
|
||||||
continue;
|
continue;
|
||||||
if (p->opt->flag & MM_F_OUT_SAM)
|
if (p->opt->flag & MM_F_OUT_SAM)
|
||||||
mm_write_sam2(&p->str, mi, t, i - seg_st, j, s->n_seg[k], &s->n_reg[seg_st], (const mm_reg1_t*const*)&s->reg[seg_st], km, p->opt->flag);
|
mm_write_sam3(&p->str, mi, t, i - seg_st, j, s->n_seg[k], &s->n_reg[seg_st], (const mm_reg1_t*const*)&s->reg[seg_st], km, p->opt->flag, s->rep_len[i]);
|
||||||
else
|
else
|
||||||
mm_write_paf(&p->str, mi, t, r, km, p->opt->flag);
|
mm_write_paf3(&p->str, mi, t, r, km, p->opt->flag, s->rep_len[i]);
|
||||||
mm_err_puts(p->str.s);
|
mm_err_puts(p->str.s);
|
||||||
}
|
}
|
||||||
} else if (p->opt->flag & (MM_F_OUT_SAM|MM_F_PAF_NO_HIT)) { // output an empty hit, if requested
|
} else if ((p->opt->flag & MM_F_PAF_NO_HIT) || ((p->opt->flag & MM_F_OUT_SAM) && !(p->opt->flag & MM_F_SAM_HIT_ONLY))) { // output an empty hit, if requested
|
||||||
if (p->opt->flag & MM_F_OUT_SAM)
|
if (p->opt->flag & MM_F_OUT_SAM)
|
||||||
mm_write_sam2(&p->str, mi, t, i - seg_st, -1, s->n_seg[k], &s->n_reg[seg_st], (const mm_reg1_t*const*)&s->reg[seg_st], km, p->opt->flag);
|
mm_write_sam3(&p->str, mi, t, i - seg_st, -1, s->n_seg[k], &s->n_reg[seg_st], (const mm_reg1_t*const*)&s->reg[seg_st], km, p->opt->flag, s->rep_len[i]);
|
||||||
else
|
else
|
||||||
mm_write_paf(&p->str, mi, t, 0, 0, p->opt->flag);
|
mm_write_paf3(&p->str, mi, t, 0, 0, p->opt->flag, s->rep_len[i]);
|
||||||
mm_err_puts(p->str.s);
|
mm_err_puts(p->str.s);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -622,7 +623,7 @@ static mm_bseq_file_t **open_bseqs(int n, const char **fn)
|
|||||||
for (i = 0; i < n; ++i) {
|
for (i = 0; i < n; ++i) {
|
||||||
if ((fp[i] = mm_bseq_open(fn[i])) == 0) {
|
if ((fp[i] = mm_bseq_open(fn[i])) == 0) {
|
||||||
if (mm_verbose >= 1)
|
if (mm_verbose >= 1)
|
||||||
fprintf(stderr, "ERROR: failed to open file '%s'\n", fn[i]);
|
fprintf(stderr, "ERROR: failed to open file '%s': %s\n", fn[i], strerror(errno));
|
||||||
for (j = 0; j < i; ++j)
|
for (j = 0; j < i; ++j)
|
||||||
mm_bseq_close(fp[j]);
|
mm_bseq_close(fp[j]);
|
||||||
free(fp);
|
free(fp);
|
||||||
|
|||||||
@@ -35,6 +35,7 @@
|
|||||||
#define MM_F_PAF_NO_HIT 0x8000000 // output unmapped reads to PAF
|
#define MM_F_PAF_NO_HIT 0x8000000 // output unmapped reads to PAF
|
||||||
#define MM_F_NO_END_FLT 0x10000000
|
#define MM_F_NO_END_FLT 0x10000000
|
||||||
#define MM_F_HARD_MLEVEL 0x20000000
|
#define MM_F_HARD_MLEVEL 0x20000000
|
||||||
|
#define MM_F_SAM_HIT_ONLY 0x40000000
|
||||||
|
|
||||||
#define MM_I_HPC 0x1
|
#define MM_I_HPC 0x1
|
||||||
#define MM_I_NO_SEQ 0x2
|
#define MM_I_NO_SEQ 0x2
|
||||||
@@ -66,6 +67,7 @@ typedef struct {
|
|||||||
mm_idx_seq_t *seq; // sequence name, length and offset
|
mm_idx_seq_t *seq; // sequence name, length and offset
|
||||||
uint32_t *S; // 4-bit packed sequence
|
uint32_t *S; // 4-bit packed sequence
|
||||||
struct mm_idx_bucket_s *B; // index (hidden)
|
struct mm_idx_bucket_s *B; // index (hidden)
|
||||||
|
struct mm_idx_intv_s *I; // intervals (hidden)
|
||||||
void *km, *h;
|
void *km, *h;
|
||||||
} mm_idx_t;
|
} mm_idx_t;
|
||||||
|
|
||||||
@@ -103,16 +105,16 @@ typedef struct {
|
|||||||
} mm_idxopt_t;
|
} mm_idxopt_t;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
|
int64_t flag; // see MM_F_* macros
|
||||||
int seed;
|
int seed;
|
||||||
int sdust_thres; // score threshold for SDUST; 0 to disable
|
int sdust_thres; // score threshold for SDUST; 0 to disable
|
||||||
int flag; // see MM_F_* macros
|
|
||||||
|
|
||||||
int max_qlen; // max query length
|
int max_qlen; // max query length
|
||||||
|
|
||||||
int bw; // bandwidth
|
int bw; // bandwidth
|
||||||
int max_gap, max_gap_ref; // break a chain if there are no minimizers in a max_gap window
|
int max_gap, max_gap_ref; // break a chain if there are no minimizers in a max_gap window
|
||||||
int max_frag_len;
|
int max_frag_len;
|
||||||
int max_chain_skip;
|
int max_chain_skip, max_chain_iter;
|
||||||
int min_cnt; // min number of minimizers on each chain
|
int min_cnt; // min number of minimizers on each chain
|
||||||
int min_chain_score; // min chaining score
|
int min_chain_score; // min chaining score
|
||||||
|
|
||||||
@@ -127,6 +129,7 @@ typedef struct {
|
|||||||
int a, b, q, e, q2, e2; // matching score, mismatch, gap-open and gap-ext penalties
|
int a, b, q, e, q2, e2; // matching score, mismatch, gap-open and gap-ext penalties
|
||||||
int sc_ambi; // score when one or both bases are "N"
|
int sc_ambi; // score when one or both bases are "N"
|
||||||
int noncan; // cost of non-canonical splicing sites
|
int noncan; // cost of non-canonical splicing sites
|
||||||
|
int junc_bonus;
|
||||||
int zdrop, zdrop_inv; // break alignment if alignment score drops too fast along the diagonal
|
int zdrop, zdrop_inv; // break alignment if alignment score drops too fast along the diagonal
|
||||||
int end_bonus;
|
int end_bonus;
|
||||||
int min_dp_max; // drop an alignment if the score of the max scoring segment is below this threshold
|
int min_dp_max; // drop an alignment if the score of the max scoring segment is below this threshold
|
||||||
@@ -365,6 +368,9 @@ int mm_idx_index_name(mm_idx_t *mi);
|
|||||||
int mm_idx_name2id(const mm_idx_t *mi, const char *name);
|
int mm_idx_name2id(const mm_idx_t *mi, const char *name);
|
||||||
int mm_idx_getseq(const mm_idx_t *mi, uint32_t rid, uint32_t st, uint32_t en, uint8_t *seq);
|
int mm_idx_getseq(const mm_idx_t *mi, uint32_t rid, uint32_t st, uint32_t en, uint8_t *seq);
|
||||||
|
|
||||||
|
int mm_idx_bed_read(mm_idx_t *mi, const char *fn, int read_junc);
|
||||||
|
int mm_idx_bed_junc(const mm_idx_t *mi, int32_t ctg, int32_t st, int32_t en, uint8_t *s);
|
||||||
|
|
||||||
// deprecated APIs for backward compatibility
|
// deprecated APIs for backward compatibility
|
||||||
void mm_mapopt_init(mm_mapopt_t *opt);
|
void mm_mapopt_init(mm_mapopt_t *opt);
|
||||||
mm_idx_t *mm_idx_build(const char *fn, int w, int k, int flag, int n_threads);
|
mm_idx_t *mm_idx_build(const char *fn, int w, int k, int flag, int n_threads);
|
||||||
|
|||||||
+42
-7
@@ -1,4 +1,4 @@
|
|||||||
.TH minimap2 1 "10 January 2019" "minimap2-2.15 (r905)" "Bioinformatics tools"
|
.TH minimap2 1 "4 May 2019" "minimap2-2.17 (r941)" "Bioinformatics tools"
|
||||||
.SH NAME
|
.SH NAME
|
||||||
.PP
|
.PP
|
||||||
minimap2 - mapping and alignment between collections of DNA sequences
|
minimap2 - mapping and alignment between collections of DNA sequences
|
||||||
@@ -232,13 +232,19 @@ Honor option
|
|||||||
and disable a heurstic to save unmapped subsequences.
|
and disable a heurstic to save unmapped subsequences.
|
||||||
.TP
|
.TP
|
||||||
.BI --max-chain-skip \ INT
|
.BI --max-chain-skip \ INT
|
||||||
A heuristics that stops chaining early [50]. Minimap2 uses dynamic programming
|
A heuristics that stops chaining early [25]. Minimap2 uses dynamic programming
|
||||||
for chaining. The time complexity is quadratic in the number of seeds. This
|
for chaining. The time complexity is quadratic in the number of seeds. This
|
||||||
option makes minimap2 exits the inner loop if it repeatedly sees seeds already
|
option makes minimap2 exits the inner loop if it repeatedly sees seeds already
|
||||||
on chains. Set
|
on chains. Set
|
||||||
.I INT
|
.I INT
|
||||||
to a large number to switch off this heurstics.
|
to a large number to switch off this heurstics.
|
||||||
.TP
|
.TP
|
||||||
|
.BI --max-chain-iter \ INT
|
||||||
|
Check up to
|
||||||
|
.I INT
|
||||||
|
partial chains during chaining [5000]. This is a heuristic to avoid quadratic
|
||||||
|
time complexity in the worst case.
|
||||||
|
.TP
|
||||||
.B --no-long-join
|
.B --no-long-join
|
||||||
Disable the long gap patching heuristic. When this option is applied, the
|
Disable the long gap patching heuristic. When this option is applied, the
|
||||||
maximum alignment gap is mostly controlled by
|
maximum alignment gap is mostly controlled by
|
||||||
@@ -358,6 +364,17 @@ on SIRV data, please add
|
|||||||
.B --splice-flank=no
|
.B --splice-flank=no
|
||||||
to the command line.
|
to the command line.
|
||||||
.TP
|
.TP
|
||||||
|
.BR --junc-bed \ FILE
|
||||||
|
Gene annotations in the BED12 format (aka 12-column BED), or intron positions
|
||||||
|
in 5-column BED. With this option, minimap2 prefers splicing in annotations.
|
||||||
|
BED12 file can be converted from GTF/GFF3 with `paftools.js gff2bed anno.gtf'
|
||||||
|
[].
|
||||||
|
.TP
|
||||||
|
.BR --junc-bonus \ INT
|
||||||
|
Score bonus for a splice donor or acceptor found in annotation (effective with
|
||||||
|
.BR --junc-bed )
|
||||||
|
[0].
|
||||||
|
.TP
|
||||||
.BI --end-seed-pen \ INT
|
.BI --end-seed-pen \ INT
|
||||||
Drop a terminal anchor if
|
Drop a terminal anchor if
|
||||||
.IR s <log( g )+ INT ,
|
.IR s <log( g )+ INT ,
|
||||||
@@ -384,6 +401,11 @@ Set 0 to disable [0].
|
|||||||
Generate CIGAR and output alignments in the SAM format. Minimap2 outputs in PAF
|
Generate CIGAR and output alignments in the SAM format. Minimap2 outputs in PAF
|
||||||
by default.
|
by default.
|
||||||
.TP
|
.TP
|
||||||
|
.BI -o \ FILE
|
||||||
|
Output alignments to
|
||||||
|
.I FILE
|
||||||
|
[stdout].
|
||||||
|
.TP
|
||||||
.B -Q
|
.B -Q
|
||||||
Ignore base quality in the input file.
|
Ignore base quality in the input file.
|
||||||
.TP
|
.TP
|
||||||
@@ -463,7 +485,12 @@ Filter out query sequences longer than
|
|||||||
.IR NUM .
|
.IR NUM .
|
||||||
.TP
|
.TP
|
||||||
.B --paf-no-hit
|
.B --paf-no-hit
|
||||||
In PAF, output query name and length for an unmapped sequence.
|
In PAF, output unmapped queries; the strand and the reference name fields are
|
||||||
|
set to `*'. Warning: some paftools.js commands may not work with such output
|
||||||
|
for the moment.
|
||||||
|
.TP
|
||||||
|
.B --sam-hit-only
|
||||||
|
In SAM, don't output unmapped reads.
|
||||||
.TP
|
.TP
|
||||||
.B --version
|
.B --version
|
||||||
Print version number to stdout
|
Print version number to stdout
|
||||||
@@ -494,7 +521,7 @@ is determined by the sequencing error mode.
|
|||||||
.B asm5
|
.B asm5
|
||||||
Long assembly to reference mapping
|
Long assembly to reference mapping
|
||||||
.RB ( -k19
|
.RB ( -k19
|
||||||
.B -w19 -A1 -B19 -O39,81 -E3,1 -s200 -z200
|
.B -w19 -A1 -B19 -O39,81 -E3,1 -s200 -z200 -N50
|
||||||
.BR --min-occ-floor=100 ).
|
.BR --min-occ-floor=100 ).
|
||||||
Typically, the alignment will not extend to regions with 5% or higher sequence
|
Typically, the alignment will not extend to regions with 5% or higher sequence
|
||||||
divergence. Only use this preset if the average divergence is far below 5%.
|
divergence. Only use this preset if the average divergence is far below 5%.
|
||||||
@@ -502,14 +529,14 @@ divergence. Only use this preset if the average divergence is far below 5%.
|
|||||||
.B asm10
|
.B asm10
|
||||||
Long assembly to reference mapping
|
Long assembly to reference mapping
|
||||||
.RB ( -k19
|
.RB ( -k19
|
||||||
.B -w19 -A1 -B9 -O16,41 -E2,1 -s200 -z200
|
.B -w19 -A1 -B9 -O16,41 -E2,1 -s200 -z200 -N50
|
||||||
.BR --min-occ-floor=100 ).
|
.BR --min-occ-floor=100 ).
|
||||||
Up to 10% sequence divergence.
|
Up to 10% sequence divergence.
|
||||||
.TP
|
.TP
|
||||||
.B asm20
|
.B asm20
|
||||||
Long assembly to reference mapping
|
Long assembly to reference mapping
|
||||||
.RB ( -k19
|
.RB ( -k19
|
||||||
.B -w10 -A1 -B4 -O6,26 -E2,1 -s200 -z200
|
.B -w10 -A1 -B4 -O6,26 -E2,1 -s200 -z200 -N50
|
||||||
.BR --min-occ-floor=100 ).
|
.BR --min-occ-floor=100 ).
|
||||||
Up to 20% sequence divergence.
|
Up to 20% sequence divergence.
|
||||||
.TP
|
.TP
|
||||||
@@ -531,7 +558,7 @@ is that this preset is not using HPC minimizers.
|
|||||||
.B splice
|
.B splice
|
||||||
Long-read spliced alignment
|
Long-read spliced alignment
|
||||||
.RB ( -k15
|
.RB ( -k15
|
||||||
.B -w5 --splice -g2000 -G200k -A1 -B2 -O2,32 -E1,0 -C9 -z200 -ub
|
.B -w5 --splice -g2000 -G200k -A1 -B2 -O2,32 -E1,0 -C9 -z200 -ub --junc-bonus=9
|
||||||
.BR --splice-flank=yes ).
|
.BR --splice-flank=yes ).
|
||||||
In the splice mode, 1) long deletions are taken as introns and represented as
|
In the splice mode, 1) long deletions are taken as introns and represented as
|
||||||
the
|
the
|
||||||
@@ -541,6 +568,12 @@ costs are different during chaining; 4) the computation of the
|
|||||||
.RB ` ms '
|
.RB ` ms '
|
||||||
tag ignores introns to demote hits to pseudogenes.
|
tag ignores introns to demote hits to pseudogenes.
|
||||||
.TP
|
.TP
|
||||||
|
.B splice:hq
|
||||||
|
Long-read splice alignment for PacBio CCS reads
|
||||||
|
.RB ( -xsplice
|
||||||
|
.B -C5 -O6,24
|
||||||
|
.BR -B4 ).
|
||||||
|
.TP
|
||||||
.B sr
|
.B sr
|
||||||
Short single-end reads without splicing
|
Short single-end reads without splicing
|
||||||
.RB ( -k21
|
.RB ( -k21
|
||||||
@@ -605,6 +638,7 @@ s2 i Chaining score of the best secondary chain
|
|||||||
NM i Total number of mismatches and gaps in the alignment
|
NM i Total number of mismatches and gaps in the alignment
|
||||||
MD Z To generate the ref sequence in the alignment
|
MD Z To generate the ref sequence in the alignment
|
||||||
AS i DP alignment score
|
AS i DP alignment score
|
||||||
|
SA Z List of other supplementary alignments
|
||||||
ms i DP score of the max scoring segment in the alignment
|
ms i DP score of the max scoring segment in the alignment
|
||||||
nn i Number of ambiguous bases in the alignment
|
nn i Number of ambiguous bases in the alignment
|
||||||
ts A Transcript strand (splice mode only)
|
ts A Transcript strand (splice mode only)
|
||||||
@@ -612,6 +646,7 @@ cg Z CIGAR string (only in PAF)
|
|||||||
cs Z Difference string
|
cs Z Difference string
|
||||||
dv f Approximate per-base sequence divergence
|
dv f Approximate per-base sequence divergence
|
||||||
de f Gap-compressed per-base sequence divergence
|
de f Gap-compressed per-base sequence divergence
|
||||||
|
rl i Length of query regions harboring repetitive seeds
|
||||||
.TE
|
.TE
|
||||||
|
|
||||||
.PP
|
.PP
|
||||||
|
|||||||
@@ -125,7 +125,7 @@ void mm_err_puts(const char *str)
|
|||||||
int ret;
|
int ret;
|
||||||
ret = puts(str);
|
ret = puts(str);
|
||||||
if (ret == EOF) {
|
if (ret == EOF) {
|
||||||
fprintf(stderr, "[ERROR] failed to write the results\n");
|
perror("[ERROR] failed to write the results");
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -135,7 +135,7 @@ void mm_err_fwrite(const void *p, size_t size, size_t nitems, FILE *fp)
|
|||||||
int ret;
|
int ret;
|
||||||
ret = fwrite(p, size, nitems, fp);
|
ret = fwrite(p, size, nitems, fp);
|
||||||
if (ret == EOF) {
|
if (ret == EOF) {
|
||||||
fprintf(stderr, "[ERROR] failed to write data\n");
|
perror("[ERROR] failed to write data");
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -145,7 +145,7 @@ void mm_err_fread(void *p, size_t size, size_t nitems, FILE *fp)
|
|||||||
int ret;
|
int ret;
|
||||||
ret = fread(p, size, nitems, fp);
|
ret = fread(p, size, nitems, fp);
|
||||||
if (ret == EOF) {
|
if (ret == EOF) {
|
||||||
fprintf(stderr, "[ERROR] failed to read data\n");
|
perror("[ERROR] failed to read data");
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+54
-31
@@ -1,6 +1,6 @@
|
|||||||
#!/usr/bin/env k8
|
#!/usr/bin/env k8
|
||||||
|
|
||||||
var paftools_version = '2.15-r905';
|
var paftools_version = '2.17-r949-dirty';
|
||||||
|
|
||||||
/*****************************
|
/*****************************
|
||||||
***** Library functions *****
|
***** Library functions *****
|
||||||
@@ -433,6 +433,7 @@ function paf_call(args)
|
|||||||
while (file.readline(buf) >= 0) {
|
while (file.readline(buf) >= 0) {
|
||||||
var line = buf.toString();
|
var line = buf.toString();
|
||||||
var m, t = line.split("\t", 12);
|
var m, t = line.split("\t", 12);
|
||||||
|
if (t.length < 12 || t[5] == '*') continue; // unmapped
|
||||||
for (var i = 6; i <= 11; ++i)
|
for (var i = 6; i <= 11; ++i)
|
||||||
t[i] = parseInt(t[i]);
|
t[i] = parseInt(t[i]);
|
||||||
if (t[10] < min_cov_len || t[11] < min_mapq) continue;
|
if (t[10] < min_cov_len || t[11] < min_mapq) continue;
|
||||||
@@ -680,13 +681,12 @@ function paf_asmstat(args)
|
|||||||
var t = line.split("\t");
|
var t = line.split("\t");
|
||||||
t[1] = parseInt(t[1]);
|
t[1] = parseInt(t[1]);
|
||||||
if (t[1] < min_query_len) continue;
|
if (t[1] < min_query_len) continue;
|
||||||
if (t.length >= 2) {
|
if (t.length < 2) continue;
|
||||||
query[t[0]] = t[1];
|
query[t[0]] = t[1];
|
||||||
if (qinfo[t[0]] == null) qinfo[t[0]] = {};
|
if (qinfo[t[0]] == null) qinfo[t[0]] = {};
|
||||||
qinfo[t[0]].len = t[1];
|
qinfo[t[0]].len = t[1];
|
||||||
qinfo[t[0]].bp = [];
|
qinfo[t[0]].bp = [];
|
||||||
}
|
if (t.length < 9 || t[5] == "*") continue;
|
||||||
if (t.length < 9) continue;
|
|
||||||
if (!/\ttp:A:[PI]/.test(line)) continue;
|
if (!/\ttp:A:[PI]/.test(line)) continue;
|
||||||
if ((m = /\tcg:Z:(\S+)/.exec(line)) == null) continue;
|
if ((m = /\tcg:Z:(\S+)/.exec(line)) == null) continue;
|
||||||
var cigar = m[1];
|
var cigar = m[1];
|
||||||
@@ -967,7 +967,9 @@ function paf_stat(args)
|
|||||||
var t = line.split("\t", 12);
|
var t = line.split("\t", 12);
|
||||||
var m, rs, cigar = null, is_pri = false, is_sam = false, is_rev = false, tname = null;
|
var m, rs, cigar = null, is_pri = false, is_sam = false, is_rev = false, tname = null;
|
||||||
var atlen = null, aqlen, qs, qe, mapq, ori_qlen;
|
var atlen = null, aqlen, qs, qe, mapq, ori_qlen;
|
||||||
if (t[4] == '+' || t[4] == '-') { // PAF
|
if (t.length < 2) continue;
|
||||||
|
if (t[4] == '+' || t[4] == '-' || t[4] == '*') { // PAF
|
||||||
|
if (t[4] == '*') continue; // unmapped
|
||||||
if (!/\ts2:i:\d+/.test(line)) {
|
if (!/\ts2:i:\d+/.test(line)) {
|
||||||
++n_2nd;
|
++n_2nd;
|
||||||
continue;
|
continue;
|
||||||
@@ -1467,15 +1469,21 @@ function paf_view(args)
|
|||||||
|
|
||||||
function paf_gff2bed(args)
|
function paf_gff2bed(args)
|
||||||
{
|
{
|
||||||
var c, fn_ucsc_fai = null, is_short = false, keep_gff = false;
|
var c, fn_ucsc_fai = null, is_short = false, keep_gff = false, print_junc = false;
|
||||||
while ((c = getopt(args, "u:sg")) != null) {
|
while ((c = getopt(args, "u:sgj")) != null) {
|
||||||
if (c == 'u') fn_ucsc_fai = getopt.arg;
|
if (c == 'u') fn_ucsc_fai = getopt.arg;
|
||||||
else if (c == 's') is_short = true;
|
else if (c == 's') is_short = true;
|
||||||
else if (c == 'g') keep_gff = true;
|
else if (c == 'g') keep_gff = true;
|
||||||
|
else if (c == 'j') print_junc = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (getopt.ind == args.length) {
|
if (getopt.ind == args.length) {
|
||||||
print("Usage: paftools.js gff2bed [-g] [-u ucsc-genome.fa.fai] <in.gff>");
|
print("Usage: paftools.js gff2bed [options] <in.gff>");
|
||||||
|
print("Options:");
|
||||||
|
print(" -j Output junction BED");
|
||||||
|
print(" -s Print names in the short form");
|
||||||
|
print(" -u FILE hg38.fa.fai for chr name conversion");
|
||||||
|
print(" -g Output GFF (used with -u)");
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1501,17 +1509,23 @@ function paf_gff2bed(args)
|
|||||||
|
|
||||||
var colors = {
|
var colors = {
|
||||||
'protein_coding':'0,128,255',
|
'protein_coding':'0,128,255',
|
||||||
|
'mRNA':'0,128,255',
|
||||||
'lincRNA':'0,192,0',
|
'lincRNA':'0,192,0',
|
||||||
'snRNA':'0,192,0',
|
'snRNA':'0,192,0',
|
||||||
'miRNA':'0,192,0',
|
'miRNA':'0,192,0',
|
||||||
'misc_RNA':'0,192,0'
|
'misc_RNA':'0,192,0'
|
||||||
};
|
};
|
||||||
|
|
||||||
function print_bed12(exons, cds_st, cds_en, is_short)
|
function print_bed12(exons, cds_st, cds_en, is_short, print_junc)
|
||||||
{
|
{
|
||||||
if (exons.length == 0) return;
|
if (exons.length == 0) return;
|
||||||
var name = is_short? exons[0][7] + "|" + exons[0][5] : exons[0].slice(4, 7).join("|");
|
var name = is_short? exons[0][7] + "|" + exons[0][5] : exons[0].slice(4, 7).join("|");
|
||||||
var a = exons.sort(function(a,b) {return a[1]-b[1]});
|
var a = exons.sort(function(a,b) {return a[1]-b[1]});
|
||||||
|
if (print_junc) {
|
||||||
|
for (var i = 1; i < a.length; ++i)
|
||||||
|
print(a[i][0], a[i-1][2], a[i][1], name, 1000, a[i][3]);
|
||||||
|
return;
|
||||||
|
}
|
||||||
var sizes = [], starts = [], st, en;
|
var sizes = [], starts = [], st, en;
|
||||||
st = a[0][1];
|
st = a[0][1];
|
||||||
en = a[a.length - 1][2];
|
en = a[a.length - 1][2];
|
||||||
@@ -1528,8 +1542,8 @@ function paf_gff2bed(args)
|
|||||||
print(a[0][0], st, en, name, 1000, a[0][3], cds_st, cds_en, color, a.length, sizes.join(",") + ",", starts.join(",") + ",");
|
print(a[0][0], st, en, name, 1000, a[0][3], cds_st, cds_en, color, a.length, sizes.join(",") + ",", starts.join(",") + ",");
|
||||||
}
|
}
|
||||||
|
|
||||||
var re_gtf = /(transcript_id|transcript_type|transcript_biotype|gene_name|transcript_name) "([^"]+)";/g;
|
var re_gtf = /\b(transcript_id|transcript_type|transcript_biotype|gene_name|gene_id|gbkey|transcript_name) "([^"]+)";/g;
|
||||||
var re_gff3 = /(transcript_id|transcript_type|transcript_biotype|gene_name|transcript_name)=([^;]+)/g;
|
var re_gff3 = /\b(transcript_id|transcript_type|transcript_biotype|gene_name|gene_id|gbkey|transcript_name)=([^;]+)/g;
|
||||||
var buf = new Bytes();
|
var buf = new Bytes();
|
||||||
var file = args[getopt.ind] == '-'? new File() : new File(args[getopt.ind]);
|
var file = args[getopt.ind] == '-'? new File() : new File(args[getopt.ind]);
|
||||||
|
|
||||||
@@ -1546,25 +1560,25 @@ function paf_gff2bed(args)
|
|||||||
if (t[2] != "CDS" && t[2] != "exon") continue;
|
if (t[2] != "CDS" && t[2] != "exon") continue;
|
||||||
t[3] = parseInt(t[3]) - 1;
|
t[3] = parseInt(t[3]) - 1;
|
||||||
t[4] = parseInt(t[4]);
|
t[4] = parseInt(t[4]);
|
||||||
var id = null, type = "", gname = "N/A", biotype = "", m, tname = "N/A";
|
var id = null, type = "", name = "N/A", biotype = "", m, tname = "N/A";
|
||||||
while ((m = re_gtf.exec(t[8])) != null) {
|
while ((m = re_gtf.exec(t[8])) != null) {
|
||||||
if (m[1] == "transcript_id") id = m[2];
|
if (m[1] == "transcript_id") id = m[2];
|
||||||
else if (m[1] == "transcript_type") type = m[2];
|
else if (m[1] == "transcript_type") type = m[2];
|
||||||
else if (m[1] == "transcript_biotype") biotype = m[2];
|
else if (m[1] == "transcript_biotype" || m[1] == "gbkey") biotype = m[2];
|
||||||
else if (m[1] == "gene_name") name = m[2];
|
else if (m[1] == "gene_name" || m[1] == "gene_id") name = m[2];
|
||||||
else if (m[1] == "transcript_name") tname = m[2];
|
else if (m[1] == "transcript_name") tname = m[2];
|
||||||
}
|
}
|
||||||
while ((m = re_gff3.exec(t[8])) != null) {
|
while ((m = re_gff3.exec(t[8])) != null) {
|
||||||
if (m[1] == "transcript_id") id = m[2];
|
if (m[1] == "transcript_id") id = m[2];
|
||||||
else if (m[1] == "transcript_type") type = m[2];
|
else if (m[1] == "transcript_type") type = m[2];
|
||||||
else if (m[1] == "transcript_biotype") biotype = m[2];
|
else if (m[1] == "transcript_biotype" || m[1] == "gbkey") biotype = m[2];
|
||||||
else if (m[1] == "gene_name") name = m[2];
|
else if (m[1] == "gene_name" || m[1] == "gene_id") name = m[2];
|
||||||
else if (m[1] == "transcript_name") tname = m[2];
|
else if (m[1] == "transcript_name") tname = m[2];
|
||||||
}
|
}
|
||||||
if (type == "" && biotype != "") type = biotype;
|
if (type == "" && biotype != "") type = biotype;
|
||||||
if (id == null) throw Error("No transcript_id");
|
if (id == null) throw Error("No transcript_id");
|
||||||
if (id != last_id) {
|
if (id != last_id) {
|
||||||
print_bed12(exons, cds_st, cds_en, is_short);
|
print_bed12(exons, cds_st, cds_en, is_short, print_junc);
|
||||||
exons = [], cds_st = 1<<30, cds_en = 0;
|
exons = [], cds_st = 1<<30, cds_en = 0;
|
||||||
last_id = id;
|
last_id = id;
|
||||||
}
|
}
|
||||||
@@ -1582,7 +1596,7 @@ function paf_gff2bed(args)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (last_id != null)
|
if (last_id != null)
|
||||||
print_bed12(exons, cds_st, cds_en, is_short);
|
print_bed12(exons, cds_st, cds_en, is_short, print_junc);
|
||||||
|
|
||||||
file.close();
|
file.close();
|
||||||
buf.destroy();
|
buf.destroy();
|
||||||
@@ -1590,11 +1604,16 @@ function paf_gff2bed(args)
|
|||||||
|
|
||||||
function paf_sam2paf(args)
|
function paf_sam2paf(args)
|
||||||
{
|
{
|
||||||
var c, pri_only = false, use_eq = false;
|
var c, pri_only = false, long_cs = false;
|
||||||
while ((c = getopt(args, "p")) != null)
|
while ((c = getopt(args, "pL")) != null) {
|
||||||
if (c == 'p') pri_only = true;
|
if (c == 'p') pri_only = true;
|
||||||
|
else if (c == 'L') long_cs = true;
|
||||||
|
}
|
||||||
if (args.length == getopt.ind) {
|
if (args.length == getopt.ind) {
|
||||||
print("Usage: paftools.js sam2paf [-p] <in.sam>");
|
print("Usage: paftools.js sam2paf [options] <in.sam>");
|
||||||
|
print("Options:");
|
||||||
|
print(" -p convert primary or supplementary alignments only");
|
||||||
|
print(" -L output the cs tag in the long form");
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1623,13 +1642,14 @@ function paf_sam2paf(args)
|
|||||||
var tlen = ctg_len[t[2]];
|
var tlen = ctg_len[t[2]];
|
||||||
if (tlen == null) throw Error("at line " + lineno + ": can't find the length of contig " + t[2]);
|
if (tlen == null) throw Error("at line " + lineno + ": can't find the length of contig " + t[2]);
|
||||||
// find tags
|
// find tags
|
||||||
var nn = 0, NM = null, MD = null, md_list = [];
|
var nn = 0, NM = null, MD = null, cs_str = null, md_list = [];
|
||||||
while ((m = re_tag.exec(line)) != null) {
|
while ((m = re_tag.exec(line)) != null) {
|
||||||
if (m[1] == "NM:i") NM = parseInt(m[2]);
|
if (m[1] == "NM:i") NM = parseInt(m[2]);
|
||||||
else if (m[1] == "nn:i") nn = parseInt(m[2]);
|
else if (m[1] == "nn:i") nn = parseInt(m[2]);
|
||||||
else if (m[1] == "MD:Z") MD = m[2];
|
else if (m[1] == "MD:Z") MD = m[2];
|
||||||
|
else if (m[1] == "cs:Z") cs_str = m[2];
|
||||||
}
|
}
|
||||||
if (t[9] == '*') MD = null;
|
if (t[9] == '*') MD = cs_str = null;
|
||||||
// infer various lengths from CIGAR
|
// infer various lengths from CIGAR
|
||||||
var clip = [0, 0], soft_clip = 0, I = [0, 0], D = [0, 0], M = 0, N = 0, mm = 0, have_M = false, have_ext = false, cigar = [];
|
var clip = [0, 0], soft_clip = 0, I = [0, 0], D = [0, 0], M = 0, N = 0, mm = 0, have_M = false, have_ext = false, cigar = [];
|
||||||
while ((m = re.exec(t[5])) != null) {
|
while ((m = re.exec(t[5])) != null) {
|
||||||
@@ -1665,8 +1685,8 @@ function paf_sam2paf(args)
|
|||||||
}
|
}
|
||||||
// parse MD
|
// parse MD
|
||||||
var cs = [];
|
var cs = [];
|
||||||
if (MD != null) {
|
if (MD != null && cs_str == null && t[9] != "*") {
|
||||||
var k = 0, cx = 0, cy = 0, mx = 0, my = 0;
|
var k = 0, cx = 0, cy = 0, mx = 0, my = 0; // cx: cigar ref position; cy: cigar query; mx: MD ref; my: MD query
|
||||||
while ((m = re_MD.exec(MD)) != null) {
|
while ((m = re_MD.exec(MD)) != null) {
|
||||||
if (m[2] != null) { // deletion from the reference
|
if (m[2] != null) { // deletion from the reference
|
||||||
var len = m[2].length - 1;
|
var len = m[2].length - 1;
|
||||||
@@ -1680,13 +1700,15 @@ function paf_sam2paf(args)
|
|||||||
if (my + ml < cy + cl) {
|
if (my + ml < cy + cl) {
|
||||||
if (ml > 0) {
|
if (ml > 0) {
|
||||||
if (m[3] != null) cs.push('*', m[3], t[9][my]);
|
if (m[3] != null) cs.push('*', m[3], t[9][my]);
|
||||||
|
else if (long_cs) cs.push('=', t[9].substr(my, ml));
|
||||||
else cs.push(':', ml);
|
else cs.push(':', ml);
|
||||||
}
|
}
|
||||||
mx += ml, my += ml, ml = 0;
|
mx += ml, my += ml, ml = 0;
|
||||||
break;
|
break;
|
||||||
} else {
|
} else {
|
||||||
var dl = cy + cl - my;
|
var dl = cy + cl - my;
|
||||||
cs.push(':', dl);
|
if (long_cs) cs.push('=', t[9].substr(my, dl));
|
||||||
|
else cs.push(':', dl);
|
||||||
cx += cl, cy += cl, ++k;
|
cx += cl, cy += cl, ++k;
|
||||||
mx += dl, my += dl, ml -= dl;
|
mx += dl, my += dl, ml -= dl;
|
||||||
}
|
}
|
||||||
@@ -1731,7 +1753,8 @@ function paf_sam2paf(args)
|
|||||||
var tags = ["tp:A:" + type];
|
var tags = ["tp:A:" + type];
|
||||||
if (NM != null) tags.push("mm:i:"+mm);
|
if (NM != null) tags.push("mm:i:"+mm);
|
||||||
tags.push("gn:i:"+(I[1]+D[1]), "go:i:"+(I[0]+D[0]), "cg:Z:" + t[5].replace(/\d+[SH]/g, ''));
|
tags.push("gn:i:"+(I[1]+D[1]), "go:i:"+(I[0]+D[0]), "cg:Z:" + t[5].replace(/\d+[SH]/g, ''));
|
||||||
if (cs.length > 0) tags.push("cs:Z:" + cs.join(""));
|
if (cs_str != null) tags.push("cs:Z:" + cs_str);
|
||||||
|
else if (cs.length > 0) tags.push("cs:Z:" + cs.join(""));
|
||||||
// print out
|
// print out
|
||||||
var a = [qname, qlen, qs, qe, flag&16? '-' : '+', t[2], tlen, ts, te, mlen, blen, t[4]];
|
var a = [qname, qlen, qs, qe, flag&16? '-' : '+', t[2], tlen, ts, te, mlen, blen, t[4]];
|
||||||
print(a.join("\t"), tags.join("\t"));
|
print(a.join("\t"), tags.join("\t"));
|
||||||
|
|||||||
@@ -59,15 +59,17 @@ uint32_t ks_ksmall_uint32_t(size_t n, uint32_t arr[], size_t kk);
|
|||||||
|
|
||||||
void mm_sketch(void *km, const char *str, int len, int w, int k, uint32_t rid, int is_hpc, mm128_v *p);
|
void mm_sketch(void *km, const char *str, int len, int w, int k, uint32_t rid, int is_hpc, mm128_v *p);
|
||||||
|
|
||||||
void mm_write_sam_hdr(const mm_idx_t *mi, const char *rg, const char *ver, int argc, char *argv[]);
|
int mm_write_sam_hdr(const mm_idx_t *mi, const char *rg, const char *ver, int argc, char *argv[]);
|
||||||
void mm_write_paf(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, const mm_reg1_t *r, void *km, int opt_flag);
|
void mm_write_paf(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, const mm_reg1_t *r, void *km, int opt_flag);
|
||||||
|
void mm_write_paf3(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, const mm_reg1_t *r, void *km, int opt_flag, int rep_len);
|
||||||
void mm_write_sam(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, const mm_reg1_t *r, int n_regs, const mm_reg1_t *regs);
|
void mm_write_sam(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, const mm_reg1_t *r, int n_regs, const mm_reg1_t *regs);
|
||||||
void mm_write_sam2(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, int seg_idx, int reg_idx, int n_seg, const int *n_regs, const mm_reg1_t *const* regs, void *km, int opt_flag);
|
void mm_write_sam2(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, int seg_idx, int reg_idx, int n_seg, const int *n_regs, const mm_reg1_t *const* regs, void *km, int opt_flag);
|
||||||
|
void mm_write_sam3(kstring_t *s, const mm_idx_t *mi, const mm_bseq1_t *t, int seg_idx, int reg_idx, int n_seg, const int *n_regss, const mm_reg1_t *const* regss, void *km, int opt_flag, int rep_len);
|
||||||
|
|
||||||
void mm_idxopt_init(mm_idxopt_t *opt);
|
void mm_idxopt_init(mm_idxopt_t *opt);
|
||||||
const uint64_t *mm_idx_get(const mm_idx_t *mi, uint64_t minier, int *n);
|
const uint64_t *mm_idx_get(const mm_idx_t *mi, uint64_t minier, int *n);
|
||||||
int32_t mm_idx_cal_max_occ(const mm_idx_t *mi, float f);
|
int32_t mm_idx_cal_max_occ(const mm_idx_t *mi, float f);
|
||||||
mm128_t *mm_chain_dp(int max_dist_x, int max_dist_y, int bw, int max_skip, int min_cnt, int min_sc, int is_cdna, int n_segs, int64_t n, mm128_t *a, int *n_u_, uint64_t **_u, void *km);
|
mm128_t *mm_chain_dp(int max_dist_x, int max_dist_y, int bw, int max_skip, int max_iter, int min_cnt, int min_sc, int is_cdna, int n_segs, int64_t n, mm128_t *a, int *n_u_, uint64_t **_u, void *km);
|
||||||
mm_reg1_t *mm_align_skeleton(void *km, const mm_mapopt_t *opt, const mm_idx_t *mi, int qlen, const char *qstr, int *n_regs_, mm_reg1_t *regs, mm128_t *a);
|
mm_reg1_t *mm_align_skeleton(void *km, const mm_mapopt_t *opt, const mm_idx_t *mi, int qlen, const char *qstr, int *n_regs_, mm_reg1_t *regs, mm128_t *a);
|
||||||
|
|
||||||
mm_reg1_t *mm_gen_regs(void *km, uint32_t hash, int qlen, int n_u, uint64_t *u, mm128_t *a);
|
mm_reg1_t *mm_gen_regs(void *km, uint32_t hash, int qlen, int n_u, uint64_t *u, mm128_t *a);
|
||||||
|
|||||||
@@ -23,6 +23,7 @@ void mm_mapopt_init(mm_mapopt_t *opt)
|
|||||||
opt->max_gap = 5000;
|
opt->max_gap = 5000;
|
||||||
opt->max_gap_ref = -1;
|
opt->max_gap_ref = -1;
|
||||||
opt->max_chain_skip = 25;
|
opt->max_chain_skip = 25;
|
||||||
|
opt->max_chain_iter = 5000;
|
||||||
|
|
||||||
opt->mask_level = 0.5f;
|
opt->mask_level = 0.5f;
|
||||||
opt->pri_ratio = 0.8f;
|
opt->pri_ratio = 0.8f;
|
||||||
@@ -119,13 +120,16 @@ int mm_set_opt(const char *preset, mm_idxopt_t *io, mm_mapopt_t *mo)
|
|||||||
mo->mid_occ = 1000;
|
mo->mid_occ = 1000;
|
||||||
mo->max_occ = 5000;
|
mo->max_occ = 5000;
|
||||||
mo->mini_batch_size = 50000000;
|
mo->mini_batch_size = 50000000;
|
||||||
} else if (strcmp(preset, "splice") == 0 || strcmp(preset, "cdna") == 0) {
|
} else if (strncmp(preset, "splice", 6) == 0 || strcmp(preset, "cdna") == 0) {
|
||||||
io->flag = 0, io->k = 15, io->w = 5;
|
io->flag = 0, io->k = 15, io->w = 5;
|
||||||
mo->flag |= MM_F_SPLICE | MM_F_SPLICE_FOR | MM_F_SPLICE_REV | MM_F_SPLICE_FLANK;
|
mo->flag |= MM_F_SPLICE | MM_F_SPLICE_FOR | MM_F_SPLICE_REV | MM_F_SPLICE_FLANK;
|
||||||
mo->max_gap = 2000, mo->max_gap_ref = mo->bw = 200000;
|
mo->max_gap = 2000, mo->max_gap_ref = mo->bw = 200000;
|
||||||
mo->a = 1, mo->b = 2, mo->q = 2, mo->e = 1, mo->q2 = 32, mo->e2 = 0;
|
mo->a = 1, mo->b = 2, mo->q = 2, mo->e = 1, mo->q2 = 32, mo->e2 = 0;
|
||||||
mo->noncan = 9;
|
mo->noncan = 9;
|
||||||
|
mo->junc_bonus = 9;
|
||||||
mo->zdrop = 200, mo->zdrop_inv = 100; // because mo->a is halved
|
mo->zdrop = 200, mo->zdrop_inv = 100; // because mo->a is halved
|
||||||
|
if (strcmp(preset, "splice:hq") == 0)
|
||||||
|
mo->junc_bonus = 5, mo->b = 4, mo->q = 6, mo->q2 = 24;
|
||||||
} else return -1;
|
} else return -1;
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -114,6 +114,12 @@ This method retrieves a (sub)sequence from the index and returns it as a Python
|
|||||||
string. :code:`None` is returned if :code:`name` is not present in the index or
|
string. :code:`None` is returned if :code:`name` is not present in the index or
|
||||||
the start/end coordinates are invalid.
|
the start/end coordinates are invalid.
|
||||||
|
|
||||||
|
.. code:: python
|
||||||
|
|
||||||
|
mappy.Aligner.seq_names
|
||||||
|
|
||||||
|
This property gives the array of sequence names in the index.
|
||||||
|
|
||||||
Class mappy.Alignment
|
Class mappy.Alignment
|
||||||
~~~~~~~~~~~~~~~~~~~~~
|
~~~~~~~~~~~~~~~~~~~~~
|
||||||
|
|
||||||
|
|||||||
+5
-3
@@ -10,13 +10,14 @@ cdef extern from "minimap.h":
|
|||||||
uint64_t batch_size
|
uint64_t batch_size
|
||||||
|
|
||||||
ctypedef struct mm_mapopt_t:
|
ctypedef struct mm_mapopt_t:
|
||||||
|
int64_t flag
|
||||||
int seed
|
int seed
|
||||||
int sdust_thres
|
int sdust_thres
|
||||||
int flag
|
int max_qlen
|
||||||
int bw
|
int bw
|
||||||
int max_gap, max_gap_ref
|
int max_gap, max_gap_ref
|
||||||
int max_frag_len
|
int max_frag_len
|
||||||
int max_chain_skip
|
int max_chain_skip, max_chain_iter
|
||||||
int min_cnt
|
int min_cnt
|
||||||
int min_chain_score
|
int min_chain_score
|
||||||
float mask_level
|
float mask_level
|
||||||
@@ -24,10 +25,11 @@ cdef extern from "minimap.h":
|
|||||||
int best_n
|
int best_n
|
||||||
int max_join_long, max_join_short
|
int max_join_long, max_join_short
|
||||||
int min_join_flank_sc
|
int min_join_flank_sc
|
||||||
float min_join_flank_ratio;
|
float min_join_flank_ratio
|
||||||
int a, b, q, e, q2, e2
|
int a, b, q, e, q2, e2
|
||||||
int sc_ambi
|
int sc_ambi
|
||||||
int noncan
|
int noncan
|
||||||
|
int junc_bonus
|
||||||
int zdrop, zdrop_inv
|
int zdrop, zdrop_inv
|
||||||
int end_bonus
|
int end_bonus
|
||||||
int min_dp_max
|
int min_dp_max
|
||||||
|
|||||||
+42
-20
@@ -3,7 +3,7 @@ from libc.stdlib cimport free
|
|||||||
cimport cmappy
|
cimport cmappy
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
__version__ = '2.15'
|
__version__ = '2.17'
|
||||||
|
|
||||||
cmappy.mm_reset_timer()
|
cmappy.mm_reset_timer()
|
||||||
|
|
||||||
@@ -113,6 +113,7 @@ cdef class Aligner:
|
|||||||
cdef cmappy.mm_mapopt_t map_opt
|
cdef cmappy.mm_mapopt_t map_opt
|
||||||
|
|
||||||
def __cinit__(self, fn_idx_in=None, preset=None, k=None, w=None, min_cnt=None, min_chain_score=None, min_dp_score=None, bw=None, best_n=None, n_threads=3, fn_idx_out=None, max_frag_len=None, extra_flags=None, seq=None, scoring=None):
|
def __cinit__(self, fn_idx_in=None, preset=None, k=None, w=None, min_cnt=None, min_chain_score=None, min_dp_score=None, bw=None, best_n=None, n_threads=3, fn_idx_out=None, max_frag_len=None, extra_flags=None, seq=None, scoring=None):
|
||||||
|
self._idx = NULL
|
||||||
cmappy.mm_set_opt(NULL, &self.idx_opt, &self.map_opt) # set the default options
|
cmappy.mm_set_opt(NULL, &self.idx_opt, &self.map_opt) # set the default options
|
||||||
if preset is not None:
|
if preset is not None:
|
||||||
cmappy.mm_set_opt(str.encode(preset), &self.idx_opt, &self.map_opt) # apply preset
|
cmappy.mm_set_opt(str.encode(preset), &self.idx_opt, &self.map_opt) # apply preset
|
||||||
@@ -142,7 +143,7 @@ cdef class Aligner:
|
|||||||
if fn_idx_out is None:
|
if fn_idx_out is None:
|
||||||
r = cmappy.mm_idx_reader_open(str.encode(fn_idx_in), &self.idx_opt, NULL)
|
r = cmappy.mm_idx_reader_open(str.encode(fn_idx_in), &self.idx_opt, NULL)
|
||||||
else:
|
else:
|
||||||
r = cmappy.mm_idx_reader_open(str.encode(fn_idx_in), &self.idx_opt, fn_idx_out)
|
r = cmappy.mm_idx_reader_open(str.encode(fn_idx_in), &self.idx_opt, str.encode(fn_idx_out))
|
||||||
if r is not NULL:
|
if r is not NULL:
|
||||||
self._idx = cmappy.mm_idx_reader_read(r, n_threads) # NB: ONLY read the first part
|
self._idx = cmappy.mm_idx_reader_read(r, n_threads) # NB: ONLY read the first part
|
||||||
cmappy.mm_idx_reader_close(r)
|
cmappy.mm_idx_reader_close(r)
|
||||||
@@ -170,6 +171,7 @@ cdef class Aligner:
|
|||||||
cdef void *km
|
cdef void *km
|
||||||
cdef cmappy.mm_mapopt_t map_opt
|
cdef cmappy.mm_mapopt_t map_opt
|
||||||
|
|
||||||
|
if self._idx == NULL: return
|
||||||
map_opt = self.map_opt
|
map_opt = self.map_opt
|
||||||
if max_frag_len is not None: map_opt.max_frag_len = max_frag_len
|
if max_frag_len is not None: map_opt.max_frag_len = max_frag_len
|
||||||
if extra_flags is not None: map_opt.flag |= extra_flags
|
if extra_flags is not None: map_opt.flag |= extra_flags
|
||||||
@@ -186,27 +188,36 @@ cdef class Aligner:
|
|||||||
_seq2 = seq2 if isinstance(seq2, bytes) else seq2.encode()
|
_seq2 = seq2 if isinstance(seq2, bytes) else seq2.encode()
|
||||||
regs = cmappy.mm_map_aux(self._idx, _seq, _seq2, &n_regs, b._b, &map_opt)
|
regs = cmappy.mm_map_aux(self._idx, _seq, _seq2, &n_regs, b._b, &map_opt)
|
||||||
|
|
||||||
for i in range(n_regs):
|
try:
|
||||||
cmappy.mm_reg2hitpy(self._idx, ®s[i], &h)
|
i = 0
|
||||||
cigar, _cs, _MD = [], '', ''
|
while i < n_regs:
|
||||||
for k in range(h.n_cigar32): # convert the 32-bit CIGAR encoding to Python array
|
cmappy.mm_reg2hitpy(self._idx, ®s[i], &h)
|
||||||
c = h.cigar32[k]
|
cigar, _cs, _MD = [], '', ''
|
||||||
cigar.append([c>>4, c&0xf])
|
for k in range(h.n_cigar32): # convert the 32-bit CIGAR encoding to Python array
|
||||||
if cs or MD: # generate the cs and/or the MD tag, if requested
|
c = h.cigar32[k]
|
||||||
if cs:
|
cigar.append([c>>4, c&0xf])
|
||||||
l_cs_str = cmappy.mm_gen_cs(km, &cs_str, &m_cs_str, self._idx, ®s[i], _seq, 1)
|
if cs or MD: # generate the cs and/or the MD tag, if requested
|
||||||
_cs = cs_str[:l_cs_str] if isinstance(cs_str, str) else cs_str[:l_cs_str].decode()
|
if cs:
|
||||||
if MD:
|
l_cs_str = cmappy.mm_gen_cs(km, &cs_str, &m_cs_str, self._idx, ®s[i], _seq, 1)
|
||||||
l_cs_str = cmappy.mm_gen_MD(km, &cs_str, &m_cs_str, self._idx, ®s[i], _seq)
|
_cs = cs_str[:l_cs_str] if isinstance(cs_str, str) else cs_str[:l_cs_str].decode()
|
||||||
_MD = cs_str[:l_cs_str] if isinstance(cs_str, str) else cs_str[:l_cs_str].decode()
|
if MD:
|
||||||
yield Alignment(h.ctg, h.ctg_len, h.ctg_start, h.ctg_end, h.strand, h.qry_start, h.qry_end, h.mapq, cigar, h.is_primary, h.mlen, h.blen, h.NM, h.trans_strand, h.seg_id, _cs, _MD)
|
l_cs_str = cmappy.mm_gen_MD(km, &cs_str, &m_cs_str, self._idx, ®s[i], _seq)
|
||||||
cmappy.mm_free_reg1(®s[i])
|
_MD = cs_str[:l_cs_str] if isinstance(cs_str, str) else cs_str[:l_cs_str].decode()
|
||||||
free(regs)
|
yield Alignment(h.ctg, h.ctg_len, h.ctg_start, h.ctg_end, h.strand, h.qry_start, h.qry_end, h.mapq, cigar, h.is_primary, h.mlen, h.blen, h.NM, h.trans_strand, h.seg_id, _cs, _MD)
|
||||||
free(cs_str)
|
cmappy.mm_free_reg1(®s[i])
|
||||||
|
i += 1
|
||||||
|
finally:
|
||||||
|
while i < n_regs:
|
||||||
|
cmappy.mm_free_reg1(®s[i])
|
||||||
|
i += 1
|
||||||
|
free(regs)
|
||||||
|
free(cs_str)
|
||||||
|
|
||||||
def seq(self, str name, int start=0, int end=0x7fffffff):
|
def seq(self, str name, int start=0, int end=0x7fffffff):
|
||||||
cdef int l
|
cdef int l
|
||||||
cdef char *s = cmappy.mappy_fetch_seq(self._idx, name.encode(), start, end, &l)
|
cdef char *s
|
||||||
|
if self._idx == NULL: return
|
||||||
|
s = cmappy.mappy_fetch_seq(self._idx, name.encode(), start, end, &l)
|
||||||
if l == 0: return None
|
if l == 0: return None
|
||||||
r = s[:l] if isinstance(s, str) else s[:l].decode()
|
r = s[:l] if isinstance(s, str) else s[:l].decode()
|
||||||
free(s)
|
free(s)
|
||||||
@@ -221,6 +232,17 @@ cdef class Aligner:
|
|||||||
@property
|
@property
|
||||||
def n_seq(self): return self._idx.n_seq
|
def n_seq(self): return self._idx.n_seq
|
||||||
|
|
||||||
|
@property
|
||||||
|
def seq_names(self):
|
||||||
|
cdef char *p
|
||||||
|
if self._idx == NULL: return
|
||||||
|
sn = []
|
||||||
|
for i in range(self._idx.n_seq):
|
||||||
|
p = self._idx.seq[i].name
|
||||||
|
s = p if isinstance(p, str) else p.decode()
|
||||||
|
sn.append(s)
|
||||||
|
return sn
|
||||||
|
|
||||||
def fastx_read(fn, read_comment=False):
|
def fastx_read(fn, read_comment=False):
|
||||||
cdef cmappy.kseq_t *ks
|
cdef cmappy.kseq_t *ks
|
||||||
ks = cmappy.mm_fastx_open(str.encode(fn))
|
ks = cmappy.mm_fastx_open(str.encode(fn))
|
||||||
|
|||||||
@@ -33,7 +33,7 @@ def readme():
|
|||||||
|
|
||||||
setup(
|
setup(
|
||||||
name = 'mappy',
|
name = 'mappy',
|
||||||
version = '2.15',
|
version = '2.17',
|
||||||
url = 'https://github.com/lh3/minimap2',
|
url = 'https://github.com/lh3/minimap2',
|
||||||
description = 'Minimap2 python binding',
|
description = 'Minimap2 python binding',
|
||||||
long_description = readme(),
|
long_description = readme(),
|
||||||
|
|||||||
+11
-7
@@ -2,6 +2,7 @@
|
|||||||
#include <assert.h>
|
#include <assert.h>
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
|
#include <errno.h>
|
||||||
#include "mmpriv.h"
|
#include "mmpriv.h"
|
||||||
|
|
||||||
FILE *mm_split_init(const char *prefix, const mm_idx_t *mi)
|
FILE *mm_split_init(const char *prefix, const mm_idx_t *mi)
|
||||||
@@ -11,14 +12,17 @@ FILE *mm_split_init(const char *prefix, const mm_idx_t *mi)
|
|||||||
uint32_t i, k = mi->k;
|
uint32_t i, k = mi->k;
|
||||||
fn = (char*)calloc(strlen(prefix) + 10, 1);
|
fn = (char*)calloc(strlen(prefix) + 10, 1);
|
||||||
sprintf(fn, "%s.%.4d.tmp", prefix, mi->index);
|
sprintf(fn, "%s.%.4d.tmp", prefix, mi->index);
|
||||||
fp = fopen(fn, "wb");
|
if ((fp = fopen(fn, "wb")) == NULL) {
|
||||||
assert(fp);
|
if (mm_verbose >= 1)
|
||||||
|
fprintf(stderr, "[ERROR]\033[1;31m failed to write to temporary file '%s'\033[0m: %s\n", fn, strerror(errno));
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
mm_err_fwrite(&k, 4, 1, fp);
|
mm_err_fwrite(&k, 4, 1, fp);
|
||||||
mm_err_fwrite(&mi->n_seq, 4, 1, fp);
|
mm_err_fwrite(&mi->n_seq, 4, 1, fp);
|
||||||
for (i = 0; i < mi->n_seq; ++i) {
|
for (i = 0; i < mi->n_seq; ++i) {
|
||||||
uint8_t l;
|
uint32_t l;
|
||||||
l = strlen(mi->seq[i].name);
|
l = strlen(mi->seq[i].name);
|
||||||
mm_err_fwrite(&l, 1, 1, fp);
|
mm_err_fwrite(&l, 1, 4, fp);
|
||||||
mm_err_fwrite(mi->seq[i].name, 1, l, fp);
|
mm_err_fwrite(mi->seq[i].name, 1, l, fp);
|
||||||
mm_err_fwrite(&mi->seq[i].len, 4, 1, fp);
|
mm_err_fwrite(&mi->seq[i].len, 4, 1, fp);
|
||||||
}
|
}
|
||||||
@@ -38,7 +42,7 @@ mm_idx_t *mm_split_merge_prep(const char *prefix, int n_splits, FILE **fp, uint3
|
|||||||
sprintf(fn, "%s.%.4d.tmp", prefix, i);
|
sprintf(fn, "%s.%.4d.tmp", prefix, i);
|
||||||
if ((fp[i] = fopen(fn, "rb")) == 0) {
|
if ((fp[i] = fopen(fn, "rb")) == 0) {
|
||||||
if (mm_verbose >= 1)
|
if (mm_verbose >= 1)
|
||||||
fprintf(stderr, "ERROR: failed to open temporary file '%s'\n", fn);
|
fprintf(stderr, "ERROR: failed to open temporary file '%s': %s\n", fn, strerror(errno));
|
||||||
for (j = 0; j < i; ++j)
|
for (j = 0; j < i; ++j)
|
||||||
fclose(fp[j]);
|
fclose(fp[j]);
|
||||||
free(fn);
|
free(fn);
|
||||||
@@ -57,8 +61,8 @@ mm_idx_t *mm_split_merge_prep(const char *prefix, int n_splits, FILE **fp, uint3
|
|||||||
for (i = j = 0; i < n_splits; ++i) {
|
for (i = j = 0; i < n_splits; ++i) {
|
||||||
uint32_t k;
|
uint32_t k;
|
||||||
for (k = 0; k < n_seq_part[i]; ++k, ++j) {
|
for (k = 0; k < n_seq_part[i]; ++k, ++j) {
|
||||||
uint8_t l;
|
uint32_t l;
|
||||||
mm_err_fread(&l, 1, 1, fp[i]);
|
mm_err_fread(&l, 1, 4, fp[i]);
|
||||||
mi->seq[j].name = (char*)calloc(l + 1, 1);
|
mi->seq[j].name = (char*)calloc(l + 1, 1);
|
||||||
mm_err_fread(mi->seq[j].name, 1, l, fp[i]);
|
mm_err_fread(mi->seq[j].name, 1, l, fp[i]);
|
||||||
mm_err_fread(&mi->seq[j].len, 4, 1, fp[i]);
|
mm_err_fread(&mi->seq[j].len, 4, 1, fp[i]);
|
||||||
|
|||||||
Reference in New Issue
Block a user