split.c
上传用户:tsgydb
上传日期:2007-04-14
资源大小:10674k
文件大小:7k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. #include <stdio.h>
  2. #include <string.h>
  3. /*
  4.  - split - divide a string into fields, like awk split()
  5.  = int split(char *string, char *fields[], int nfields, char *sep);
  6.  */
  7. int /* number of fields, including overflow */
  8. split(string, fields, nfields, sep)
  9. char *string;
  10. char *fields[]; /* list is not NULL-terminated */
  11. int nfields; /* number of entries available in fields[] */
  12. char *sep; /* "" white, "c" single char, "ab" [ab]+ */
  13. {
  14. register char *p = string;
  15. register char c; /* latest character */
  16. register char sepc = sep[0];
  17. register char sepc2;
  18. register int fn;
  19. register char **fp = fields;
  20. register char *sepp;
  21. register int trimtrail;
  22. /* white space */
  23. if (sepc == '') {
  24. while ((c = *p++) == ' ' || c == 't')
  25. continue;
  26. p--;
  27. trimtrail = 1;
  28. sep = " t"; /* note, code below knows this is 2 long */
  29. sepc = ' ';
  30. } else
  31. trimtrail = 0;
  32. sepc2 = sep[1]; /* now we can safely pick this up */
  33. /* catch empties */
  34. if (*p == '')
  35. return(0);
  36. /* single separator */
  37. if (sepc2 == '') {
  38. fn = nfields;
  39. for (;;) {
  40. *fp++ = p;
  41. fn--;
  42. if (fn == 0)
  43. break;
  44. while ((c = *p++) != sepc)
  45. if (c == '')
  46. return(nfields - fn);
  47. *(p-1) = '';
  48. }
  49. /* we have overflowed the fields vector -- just count them */
  50. fn = nfields;
  51. for (;;) {
  52. while ((c = *p++) != sepc)
  53. if (c == '')
  54. return(fn);
  55. fn++;
  56. }
  57. /* not reached */
  58. }
  59. /* two separators */
  60. if (sep[2] == '') {
  61. fn = nfields;
  62. for (;;) {
  63. *fp++ = p;
  64. fn--;
  65. while ((c = *p++) != sepc && c != sepc2)
  66. if (c == '') {
  67. if (trimtrail && **(fp-1) == '')
  68. fn++;
  69. return(nfields - fn);
  70. }
  71. if (fn == 0)
  72. break;
  73. *(p-1) = '';
  74. while ((c = *p++) == sepc || c == sepc2)
  75. continue;
  76. p--;
  77. }
  78. /* we have overflowed the fields vector -- just count them */
  79. fn = nfields;
  80. while (c != '') {
  81. while ((c = *p++) == sepc || c == sepc2)
  82. continue;
  83. p--;
  84. fn++;
  85. while ((c = *p++) != '' && c != sepc && c != sepc2)
  86. continue;
  87. }
  88. /* might have to trim trailing white space */
  89. if (trimtrail) {
  90. p--;
  91. while ((c = *--p) == sepc || c == sepc2)
  92. continue;
  93. p++;
  94. if (*p != '') {
  95. if (fn == nfields+1)
  96. *p = '';
  97. fn--;
  98. }
  99. }
  100. return(fn);
  101. }
  102. /* n separators */
  103. fn = 0;
  104. for (;;) {
  105. if (fn < nfields)
  106. *fp++ = p;
  107. fn++;
  108. for (;;) {
  109. c = *p++;
  110. if (c == '')
  111. return(fn);
  112. sepp = sep;
  113. while ((sepc = *sepp++) != '' && sepc != c)
  114. continue;
  115. if (sepc != '') /* it was a separator */
  116. break;
  117. }
  118. if (fn < nfields)
  119. *(p-1) = '';
  120. for (;;) {
  121. c = *p++;
  122. sepp = sep;
  123. while ((sepc = *sepp++) != '' && sepc != c)
  124. continue;
  125. if (sepc == '') /* it wasn't a separator */
  126. break;
  127. }
  128. p--;
  129. }
  130. /* not reached */
  131. }
  132. #ifdef TEST_SPLIT
  133. /*
  134.  * test program
  135.  * pgm runs regression
  136.  * pgm sep splits stdin lines by sep
  137.  * pgm str sep splits str by sep
  138.  * pgm str sep n splits str by sep n times
  139.  */
  140. int
  141. main(argc, argv)
  142. int argc;
  143. char *argv[];
  144. {
  145. char buf[512];
  146. register int n;
  147. # define MNF 10
  148. char *fields[MNF];
  149. if (argc > 4)
  150. for (n = atoi(argv[3]); n > 0; n--) {
  151. (void) strcpy(buf, argv[1]);
  152. }
  153. else if (argc > 3)
  154. for (n = atoi(argv[3]); n > 0; n--) {
  155. (void) strcpy(buf, argv[1]);
  156. (void) split(buf, fields, MNF, argv[2]);
  157. }
  158. else if (argc > 2)
  159. dosplit(argv[1], argv[2]);
  160. else if (argc > 1)
  161. while (fgets(buf, sizeof(buf), stdin) != NULL) {
  162. buf[strlen(buf)-1] = ''; /* stomp newline */
  163. dosplit(buf, argv[1]);
  164. }
  165. else
  166. regress();
  167. exit(0);
  168. }
  169. dosplit(string, seps)
  170. char *string;
  171. char *seps;
  172. {
  173. # define NF 5
  174. char *fields[NF];
  175. register int nf;
  176. nf = split(string, fields, NF, seps);
  177. print(nf, NF, fields);
  178. }
  179. print(nf, nfp, fields)
  180. int nf;
  181. int nfp;
  182. char *fields[];
  183. {
  184. register int fn;
  185. register int bound;
  186. bound = (nf > nfp) ? nfp : nf;
  187. printf("%d:t", nf);
  188. for (fn = 0; fn < bound; fn++)
  189. printf(""%s"%s", fields[fn], (fn+1 < nf) ? ", " : "n");
  190. }
  191. #define RNF 5 /* some table entries know this */
  192. struct {
  193. char *str;
  194. char *seps;
  195. int nf;
  196. char *fi[RNF];
  197. } tests[] = {
  198. "", " ", 0, { "" },
  199. " ", " ", 2, { "", "" },
  200. "x", " ", 1, { "x" },
  201. "xy", " ", 1, { "xy" },
  202. "x y", " ", 2, { "x", "y" },
  203. "abc def  g ", " ", 5, { "abc", "def", "", "g", "" },
  204. "  a bcd", " ", 4, { "", "", "a", "bcd" },
  205. "a b c d e f", " ", 6, { "a", "b", "c", "d", "e f" },
  206. " a b c d ", " ", 6, { "", "a", "b", "c", "d " },
  207. "", " _", 0, { "" },
  208. " ", " _", 2, { "", "" },
  209. "x", " _", 1, { "x" },
  210. "x y", " _", 2, { "x", "y" },
  211. "ab _ cd", " _", 2, { "ab", "cd" },
  212. " a_b  c ", " _", 5, { "", "a", "b", "c", "" },
  213. "a b c_d e f", " _", 6, { "a", "b", "c", "d", "e f" },
  214. " a b c d ", " _", 6, { "", "a", "b", "c", "d " },
  215. "", " _~", 0, { "" },
  216. " ", " _~", 2, { "", "" },
  217. "x", " _~", 1, { "x" },
  218. "x y", " _~", 2, { "x", "y" },
  219. "ab _~ cd", " _~", 2, { "ab", "cd" },
  220. " a_b  c~", " _~", 5, { "", "a", "b", "c", "" },
  221. "a b_c d~e f", " _~", 6, { "a", "b", "c", "d", "e f" },
  222. "~a b c d ", " _~", 6, { "", "a", "b", "c", "d " },
  223. "", " _~-", 0, { "" },
  224. " ", " _~-", 2, { "", "" },
  225. "x", " _~-", 1, { "x" },
  226. "x y", " _~-", 2, { "x", "y" },
  227. "ab _~- cd", " _~-", 2, { "ab", "cd" },
  228. " a_b  c~", " _~-", 5, { "", "a", "b", "c", "" },
  229. "a b_c-d~e f", " _~-", 6, { "a", "b", "c", "d", "e f" },
  230. "~a-b c d ", " _~-", 6, { "", "a", "b", "c", "d " },
  231. "", "  ", 0, { "" },
  232. " ", "  ", 2, { "", "" },
  233. "x", "  ", 1, { "x" },
  234. "xy", "  ", 1, { "xy" },
  235. "x y", "  ", 2, { "x", "y" },
  236. "abc def  g ", "  ", 4, { "abc", "def", "g", "" },
  237. "  a bcd", "  ", 3, { "", "a", "bcd" },
  238. "a b c d e f", "  ", 6, { "a", "b", "c", "d", "e f" },
  239. " a b c d ", "  ", 6, { "", "a", "b", "c", "d " },
  240. "", "", 0, { "" },
  241. " ", "", 0, { "" },
  242. "x", "", 1, { "x" },
  243. "xy", "", 1, { "xy" },
  244. "x y", "", 2, { "x", "y" },
  245. "abc def  g ", "", 3, { "abc", "def", "g" },
  246. "t a bcd", "", 2, { "a", "bcd" },
  247. "  a tbt c ", "", 3, { "a", "b", "c" },
  248. "a b c d e ", "", 5, { "a", "b", "c", "d", "e" },
  249. "a btc d e f", "", 6, { "a", "b", "c", "d", "e f" },
  250. " a b c d e f ", "", 6, { "a", "b", "c", "d", "e f " },
  251. NULL, NULL, 0, { NULL },
  252. };
  253. regress()
  254. {
  255. char buf[512];
  256. register int n;
  257. char *fields[RNF+1];
  258. register int nf;
  259. register int i;
  260. register int printit;
  261. register char *f;
  262. for (n = 0; tests[n].str != NULL; n++) {
  263. (void) strcpy(buf, tests[n].str);
  264. fields[RNF] = NULL;
  265. nf = split(buf, fields, RNF, tests[n].seps);
  266. printit = 0;
  267. if (nf != tests[n].nf) {
  268. printf("split `%s' by `%s' gave %d fields, not %dn",
  269. tests[n].str, tests[n].seps, nf, tests[n].nf);
  270. printit = 1;
  271. } else if (fields[RNF] != NULL) {
  272. printf("split() went beyond array endn");
  273. printit = 1;
  274. } else {
  275. for (i = 0; i < nf && i < RNF; i++) {
  276. f = fields[i];
  277. if (f == NULL)
  278. f = "(NULL)";
  279. if (strcmp(f, tests[n].fi[i]) != 0) {
  280. printf("split `%s' by `%s', field %d is `%s', not `%s'n",
  281. tests[n].str, tests[n].seps,
  282. i, fields[i], tests[n].fi[i]);
  283. printit = 1;
  284. }
  285. }
  286. }
  287. if (printit)
  288. print(nf, RNF, fields);
  289. }
  290. }
  291. #endif