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
|
/*
* sim-main.c
*
* Simulate test data
*
* (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 "image.h"
#include "relrod.h"
#include "cell.h"
#include "utils.h"
#include "hdf5-file.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, i;
ImageList *list;
UnitCell *cell;
struct image image;
int nrefl;
while ((c = getopt(argc, argv, "h")) != -1) {
switch ( c ) {
case 'h' : {
main_show_help(argv[0]);
return 0;
}
}
}
/* 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));
printf("Generating test data...\n");
/* Define image parameters */
list = image_list_new();
image.width = 512;
image.height = 512;
image.tilt = 0.0;
image.omega = 0.0;
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 */
image.data = malloc(512*512*2);
image_add(list, &image);
/* Calculate reflections */
get_reflections(&image, cell);
/* Construct the image */
nrefl = image_feature_count(image.rflist);
for ( i=0; i<nrefl; i++ ) {
struct imagefeature *f;
int x, y;
f = image_get_feature(image.rflist, i);
x = f->x;
y = f->y; /* Discards digits after the decimal point */
image.data[y*image.width+x] = 1;
}
/* Write the output file */
hdf5_write("simulated.h5", image.data, image.width, image.height);
return 0;
}
|