indexamajig: Wall clock profiling

1 files changed, 199 insertions, 0 deletions

diff --git a/src/time-accounts.c b/src/time-accounts.c
new file mode 100644
index 00000000..2ff22d6c
--- /dev/null
+++ b/src/time-accounts.c
@@ -0,0 +1,199 @@
+/*
+ * time-accounts.c
+ *
+ * Simple profiling according to wall clock time
+ *
+ * Copyright © 2016 Deutsches Elektronen-Synchrotron DESY,
+ *                  a research centre of the Helmholtz Association.
+ *
+ * Authors:
+ *  2016 Thomas White <taw@physics.org>
+ *
+ * This file is part of CrystFEL.
+ *
+ * CrystFEL is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * CrystFEL is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with CrystFEL.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <time.h>
+#include <assert.h>
+
+#include "time-accounts.h"
+
+#define MAX_ACCOUNTS 256
+
+struct _timeaccounts
+{
+	enum timeaccount accs[MAX_ACCOUNTS];
+	time_t sec[MAX_ACCOUNTS];
+	long nsec[MAX_ACCOUNTS];
+	int n_accs;
+	enum timeaccount cur_acc;
+	time_t cur_sec;
+	long cur_nsec;
+};
+
+
+TimeAccounts *time_accounts_init()
+{
+	TimeAccounts *accs = malloc(sizeof(struct _timeaccounts));
+	if ( accs == NULL ) return NULL;
+
+	accs->n_accs = 0;
+	accs->cur_acc = TACC_NOTHING;
+
+	return accs;
+}
+
+
+void time_accounts_free(TimeAccounts *accs)
+{
+	free(accs);
+}
+
+
+static int find_account(TimeAccounts *accs, enum timeaccount acc)
+{
+	int i;
+	static int warned_toomany = 0;
+
+	for ( i=0; i<accs->n_accs; i++ ) {
+		if ( accs->accs[i] == acc ) return i;
+	}
+
+	if ( i == MAX_ACCOUNTS ) {
+		if ( !warned_toomany ) printf("Too many time accounts used!\n");
+		warned_toomany = 1;
+		return MAX_ACCOUNTS;
+	}
+
+	/* This is the first time the account is used */
+	accs->accs[i] = acc;
+	accs->sec[i] = 0;
+	accs->nsec[i] = 0;
+	accs->n_accs++;
+	return i;
+}
+
+
+#ifdef HAVE_CLOCK_GETTIME
+
+void time_accounts_set(TimeAccounts *accs, enum timeaccount new_acc)
+{
+	struct timespec tp;
+
+	if ( accs == NULL ) return;
+
+	clock_gettime(CLOCK_MONOTONIC_RAW, &tp);
+
+	/* Record time used on the previous account */
+	if ( accs->cur_acc != TACC_NOTHING ) {
+		int i = find_account(accs, accs->cur_acc);
+		if ( i == MAX_ACCOUNTS ) {
+			printf("Too many time accounts!\n");
+		} else {
+
+			time_t sec = tp.tv_sec - accs->cur_sec;
+			long nsec = tp.tv_nsec - accs->cur_nsec;
+
+			if ( nsec < 0 ) {
+				sec -= 1;
+				nsec += 1000000000;
+			}
+			accs->sec[i] += sec;
+			accs->nsec[i] += nsec;
+
+			while ( accs->nsec[i] > 1000000000 ) {
+				accs->sec[i] += 1;
+				accs->nsec[i] -= 1000000000;
+			}
+
+		}
+	}
+
+	accs->cur_acc = new_acc;
+	accs->cur_sec = tp.tv_sec;
+	accs->cur_nsec = tp.tv_nsec;
+}
+
+#else
+
+void time_accounts_set(TimeAccounts *accs, enum timeaccount new_acc)
+{
+	if ( accs == NULL ) return;
+
+	/* Record time used on the previous account */
+	if ( accs->cur_acc != TACC_NOTHING ) {
+		int i = find_account(accs, accs->cur_acc);
+		if ( i == MAX_ACCOUNTS ) {
+			printf("Too many time accounts!\n");
+		} else {
+			/* Do nothing because we have no timer */
+		}
+	}
+
+	accs->cur_acc = new_acc;
+	accs->cur_sec = 0;
+	accs->cur_nsec = 0;
+}
+
+#endif
+
+static const char *taccname(enum timeaccount acc)
+{
+	switch ( acc ) {
+		case TACC_NOTHING : return "Nothing";
+		case TACC_SELECT : return "select()";
+		case TACC_STREAMREAD : return "Stream read";
+		case TACC_SIGNALS : return "Checking signals";
+		case TACC_QUEUETOPUP : return "Topping up queue";
+		case TACC_STATUS : return "Printing status";
+		case TACC_ENDCHECK : return "Checking end";
+		case TACC_WAKEUP : return "Waking up workers";
+		case TACC_WAITPID : return "Waiting on workers";
+		case TACC_HDF5OPEN : return "Opening HDF5 file";
+		case TACC_HDF5READ : return "Reading HDF5 file";
+		case TACC_FILTER : return "Image filters";
+		case TACC_RESRANGE : return "Resolution range";
+		case TACC_PEAKSEARCH : return "Peak search";
+		case TACC_INDEXING : return "Indexing";
+		case TACC_PREDPARAMS : return "Prediction parameters";
+		case TACC_INTEGRATION : return "Integration";
+		case TACC_TOTALS : return "Crystal totals";
+		case TACC_WRITESTREAM : return "Writing stream";
+		case TACC_CLEANUP : return "Image cleanup";
+		case TACC_EVENTWAIT : return "Waiting for event";
+		case TACC_FINALCLEANUP : return "Final cleanup";
+		default : return "Unknown";
+	}
+}
+
+
+void time_accounts_print(TimeAccounts *accs)
+{
+	int i;
+	printf("Wall clock time budget:\n");
+	printf("-----------------------\n");
+	for ( i=0; i<accs->n_accs; i++ ) {
+		printf("%25s: %10lli sec %10li nsec\n", taccname(accs->accs[i]),
+		       (long long)accs->sec[i], accs->nsec[i]);
+	}
+}