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
78
79
|
/***************************************************************************
* PHAST: PHylogenetic Analysis with Space/Time models
* Copyright (c) 2002-2005 University of California, 2006-2010 Cornell
* University. All rights reserved.
*
* This source code is distributed under a BSD-style license. See the
* file LICENSE.txt for details.
***************************************************************************/
#include <stdlib.h>
#include <stdio.h>
#include <getopt.h>
#include <misc.h>
#include <stringsplus.h>
#include <sys/types.h>
#include <unistd.h>
void usage(char *prog) {
printf("\n\
PROGRAM: %s\n\
DESCRIPTION: Randomly choose k lines from a file of n lines, for 0 < k < n.\n\
USAGE: %s [OPTIONS] <infile>\n\
OPTIONS:\n\
-k <k> Number of lines to choose (default is all lines).\n\
-h Print this help message.\n\n", prog, prog);
exit(0);
}
int main(int argc, char *argv[]) {
FILE *INF, *F;
int i, n, k = -1;
String *line = str_new(STR_MED_LEN);
int *chosen;
signed char c;
while ((c = getopt(argc, argv, "k:rh")) != -1) {
switch (c) {
case 'k':
k = atoi(optarg);
if (k <= 0) die("ERROR: k must be greater than 0.\n");
break;
case 'h':
usage(argv[0]);
case '?':
die("Bad argument. Try '%s -h'.\n", argv[0]);
}
}
if (optind != argc - 1)
die("Input filename required. Try '%s -h'.\n", argv[0]);
set_seed(-1);
INF = phast_fopen(argv[optind], "r");
/* scan for number of lines */
if (!strcmp(argv[optind], "-")) { /* if stdin, need temp file */
char tmpstr[20];
sprintf(tmpstr, "choose_lines.%d", getpid());
F = phast_fopen(tmpstr, "w+");
for (n = 0; str_readline(line, INF) != EOF; n++)
fprintf(F, "%s", line->chars);
phast_fclose(F);
INF = phast_fopen(tmpstr, "r");
}
else
for (n = 0; str_readline(line, INF) != EOF; n++);
if (k == -1) k = n;
chosen = smalloc(n * sizeof(int));
choose(chosen, n, k);
rewind(INF);
for (i = 0; str_readline(line, INF) != EOF; i++)
if (chosen[i]) printf("%s", line->chars);
return 0;
}
|
