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