aboutsummaryrefslogtreecommitdiff
path: root/toys/other/factor.c
blob: e3992a8bfb43dfc9ecd700a253dac24192c94392 (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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
/* factor.c - Factor integers
 *
 * Copyright 2014 Rob Landley <rob@landley.net>
 *
 * No standard, but it's in coreutils

USE_FACTOR(NEWTOY(factor, 0, TOYFLAG_USR|TOYFLAG_BIN))

config FACTOR
  bool "factor"
  default y
  help
    usage: factor NUMBER...

    Factor integers.
*/

#include "toys.h"

static void factor(char *s)
{
  long l, ll;

  l = strtol(s, &s, 0);
  if (*s) {
    error_msg("%s: not integer");
    return;
  }

  printf("%ld:", l);

  // Negative numbers have -1 as a factor
  if (l < 0) {
    printf(" -1");
    l *= -1;
  }

  // Deal with 0 and 1 (and 2 since we're here)
  if (l < 3) {
    printf(" %ld\n", l);
    return;
  }

  // Special case factors of 2
  while (l && !(l&1)) {
    printf(" 2");
    l >>= 1;
  }

  // test odd numbers.
  for (ll=3; ;ll += 2) {
    if (ll*ll>l) {
      if (l>1) printf(" %ld", l);
      break;
    }
    while (!(l%ll)) {
      printf(" %ld", ll);
      l /= ll;
    }
  }
  xputc('\n');
}

void factor_main(void)
{
  if (toys.optc) {
    char **ss;

    for (ss = toys.optargs; *ss; ss++) factor(*ss);
  } else for (;;) {
    char *s = 0;
    size_t len = 0;

    if (-1 == getline(&s, &len, stdin)) break;
    factor(s);
  }
}