aboutsummaryrefslogtreecommitdiff
path: root/src/main.c
blob: 9c678667173cf5c9de3a41140882fe4c5a81b3de (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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/*
 * main.c
 *
 * "Main"
 *
 * (c) 2006-2009 Thomas White <thomas.white@desy.de>
 *
 * template_index - Indexing diffraction patterns by template matching
 *
 */


#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <stdarg.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>

#include "cell.h"
#include "image.h"
#include "utils.h"
#include "hdf5-file.h"
#include "templates.h"


static void main_show_help(const char *s)
{
	printf("Syntax: %s <file1.h5> <file2.h5> [...]\n\n", s);
	printf("Index diffraction patterns\n\n");
	printf("  -h              Display this help message\n");
}


int main(int argc, char *argv[])
{
	int c;
	char **in_files;
	size_t nin;
	size_t i;
	UnitCell *cell;
	TemplateList *templates;
	struct image template_parameters;

	while ((c = getopt(argc, argv, "h")) != -1) {

		switch ( c ) {

			case 'h' : {
				main_show_help(argv[0]);
				return 0;
			}

		}

	}

	if ( optind < argc ) {
		nin = argc-optind;
		in_files = malloc(nin*sizeof(char *));
		for ( i=0; i<nin; i++ ) {
			in_files[i] = strdup(argv[optind+i]);
		}
	} else {
		fprintf(stderr, "No input files!\n");
		return 1;
	}

	/* Define unit cell */
	cell = cell_new_from_parameters(28.10e-9,
	                                28.10e-9,
	                                16.52e-9,
	                                deg2rad(90.0),
	                                deg2rad(90.0),
	                                deg2rad(120.0));

	/* Generate templates */
	template_parameters.width = 512;
	template_parameters.height = 512;
	template_parameters.fmode = FORMULATION_CLEN;
	template_parameters.x_centre = 255.5;
	template_parameters.y_centre = 255.5;
	template_parameters.camera_len = 0.2;  /* 20 cm */
	template_parameters.resolution = 5120; /* 512 pixels in 10 cm */
	template_parameters.lambda = 0.2e-9;   /* LCLS wavelength */
	templates = generate_templates(cell, template_parameters);

	printf("Input files (%i):\n", nin);
	for ( i=0; i<nin; i++ ) {

		struct image image;

		printf("%6i: %s  ", i+1, in_files[i]);

		image.width = 512;
		image.height = 512;
		image.fmode = FORMULATION_CLEN;
		image.x_centre = 255.5;
		image.y_centre = 255.5;
		image.camera_len = 0.2;  /* 20 cm */
		image.resolution = 5120; /* 512 pixels in 10 cm */
		image.lambda = 0.2e-9;   /* LCLS wavelength */

		if ( hdf5_read(&image, in_files[i]) ) {
			fprintf(stderr, "Couldn't read file '%s'\n",
			        in_files[i]);
			continue;
		}

		try_templates(&image, templates);

		printf("%6.2f %6.2f\n", rad2deg(image.omega),
		                        rad2deg(image.tilt));

	}

	return 0;
}