summaryrefslogtreecommitdiffstats
path: root/src/internal/bytealg/count_arm.s
blob: f704ea0c69e102e8ad8fdab662adca0795f84e12 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
// Copyright 2019 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

#include "go_asm.h"
#include "textflag.h"

TEXT ·Count(SB),NOSPLIT,$0-20
	MOVW	b_base+0(FP), R0
	MOVW	b_len+4(FP), R1
	MOVBU	c+12(FP), R2
	MOVW	$ret+16(FP), R7
	B	countbytebody<>(SB)

TEXT ·CountString(SB),NOSPLIT,$0-16
	MOVW	s_base+0(FP), R0
	MOVW	s_len+4(FP), R1
	MOVBU	c+8(FP), R2
	MOVW	$ret+12(FP), R7
	B	countbytebody<>(SB)

// Input:
// R0: data
// R1: data length
// R2: byte to find
// R7: address to put result
//
// On exit:
// R4 and R8 are clobbered
TEXT countbytebody<>(SB),NOSPLIT,$0
	MOVW	$0, R8	// R8 = count of byte to search
	CMP	$0, R1
	B.EQ	done	// short path to handle 0-byte case
	ADD	R0, R1	// R1 is the end of the range
byte_loop:
	MOVBU.P	1(R0), R4
	CMP	R4, R2
	ADD.EQ	$1, R8
	CMP	R0, R1
	B.NE	byte_loop
done:
	MOVW	R8, (R7)
	RET