fastlwc.c

#include <fcntl.h>
#include <sys/mman.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>

#include <ctype.h>
#include <inttypes.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "simd.h"

#define BUFSIZE (sizeof(simd_vector) * 4096)

int main(int argc, char *argv[])
{
	if (argc < 2) {
		fputs("usage: fastlwc FILE\n", stderr);
		exit(EXIT_FAILURE);
	}

	simd_vector *buf = aligned_alloc(sizeof(simd_vector), BUFSIZE);
	if (!buf) {
		perror("fastlwc: alloc");
		exit(EXIT_FAILURE);
	}

	int fd = (strcmp(argv[1], "-") == 0 ? STDIN_FILENO : open(argv[1], O_RDONLY));
	if (fd < 0) {
		perror("fastlwc: open");
		exit(EXIT_FAILURE);
	}

	posix_fadvise(fd, 0, 0, POSIX_FADV_SEQUENTIAL);

	ssize_t len;
	size_t lcount = 0,
	       wcount = 0,
	       ccount = 0,
	       rem = 0;
	lcount_state lstate = LCOUNT_INITIAL;
	wcount_state wstate = WCOUNT_INITIAL;

	while ((len = read(fd, (char*)buf + rem, BUFSIZE - rem))) {
		if (len < 0) {
			perror("fastlwc: read");
			exit(EXIT_FAILURE);
		}

		rem += len;
		ccount += len;

		simd_vector *vp = buf;
		while (rem >= sizeof(simd_vector)) {
			lcount += count_lines(*vp, &lstate);
			wcount += count_words(*vp, &wstate);

			rem -= sizeof(simd_vector);
			vp++;
		}

		if (rem) // move rem leftover bytes to start of buf
			memmove(buf, vp, rem);
	}

	if (rem) {
		memset((char*)buf + rem, ' ', sizeof(simd_vector) - rem);
		simd_vector *vp = buf;
		lcount += count_lines(*vp, &lstate);
		wcount += count_words(*vp, &wstate);
	}

	lcount += count_lines_final(&lstate);
	wcount += count_words_final(&wstate);

	printf(" %7zu %7zu %7zu %s\n", lcount, wcount, ccount, argv[1]);

	close(fd);
	free(buf);
}