prog8/examples/primes.p8

50 lines
1.2 KiB
Plaintext
Raw Normal View History

2019-01-27 00:02:45 +00:00
%import c64utils
%zeropage basicsafe
2019-01-27 00:02:45 +00:00
~ main {
ubyte[256] sieve
2019-02-25 00:37:05 +00:00
ubyte candidate_prime = 2 ; is increased in the loop
2019-01-27 00:02:45 +00:00
sub start() {
2019-02-25 00:37:05 +00:00
memset(sieve, 256, false) ; clear the sieve, to reset starting situation on subsequent runs
2019-01-27 00:02:45 +00:00
; calculate primes
c64scr.print("prime numbers up to 255:\n\n")
2019-02-25 00:37:05 +00:00
ubyte amount=0
2019-01-27 00:02:45 +00:00
while true {
ubyte prime = find_next_prime()
if prime==0
break
c64scr.print_ub(prime)
c64scr.print(", ")
2019-02-25 00:08:10 +00:00
amount++
2019-01-27 00:02:45 +00:00
}
c64.CHROUT('\n')
2019-02-25 00:37:05 +00:00
c64scr.print("number of primes (expected 54): ")
2019-02-25 00:08:10 +00:00
c64scr.print_ub(amount)
c64.CHROUT('\n')
2019-01-27 00:02:45 +00:00
}
sub find_next_prime() -> ubyte {
2019-02-25 00:37:05 +00:00
while sieve[candidate_prime] {
candidate_prime++
if candidate_prime==0
return 0 ; we wrapped; no more primes available in the sieve
}
; found next one, mark the multiples and return it.
sieve[candidate_prime] = true
uword multiple = candidate_prime
while multiple < len(sieve) {
sieve[lsb(multiple)] = true
multiple += candidate_prime
2019-01-27 00:02:45 +00:00
}
return candidate_prime
2019-01-27 00:02:45 +00:00
}
}