main.c 2.67 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
/*
 * 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>

23
24
25
26
27
28
#include "cell.h"
#include "image.h"
#include "utils.h"
#include "hdf5-file.h"
#include "templates.h"

29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

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;
44
45
46
	UnitCell *cell;
	TemplateList *templates;
	struct image template_parameters;
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

	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;
	}

72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
	/* 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);
90

Thomas White's avatar
Thomas White committed
91
92
93
	printf("%i files to index:\n", nin);
	printf("     #:                        Omega     Tilt\n");
	printf("--------------------------------------------------\n");
94
	for ( i=0; i<nin; i++ ) {
95
96
97

		struct image image;

Thomas White's avatar
Thomas White committed
98
		printf("%6i: %20s ", i+1, in_files[i]);
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116

		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);

Thomas White's avatar
Thomas White committed
117
118
		printf("%+8.2f %+8.2f deg\n", rad2deg(image.omega),
		                              rad2deg(image.tilt));
Thomas White's avatar
Thomas White committed
119

120
121
122
123
	}

	return 0;
}