get_hkl.c 4.2 KB
Newer Older
1
2
3
4
5
/*
 * get_hkl.c
 *
 * Small program to write out a list of h,k,l,I values given a structure
 *
Thomas White's avatar
Thomas White committed
6
 * (c) 2006-2010 Thomas White <taw@physics.org>
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
 *
 * Part of CrystFEL - crystallography with a FEL
 *
 */


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

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

#include "utils.h"
#include "sfac.h"
#include "reflections.h"


static void show_help(const char *s)
{
	printf("Syntax: %s [options]\n\n", s);
	printf(
"Write idealised intensity lists.\n"
"\n"
Thomas White's avatar
Thomas White committed
35
36
"  -h, --help                 Display this help message.\n"
"\n"
Thomas White's avatar
Thomas White committed
37
38
"  -t, --template=<filename>  Only include reflections mentioned in file.\n"
"      --poisson              Simulate Poisson samples.\n"
Thomas White's avatar
Thomas White committed
39
"      --twin                 Generate twinned data.\n"
Thomas White's avatar
Thomas White committed
40
41
42
43
"  -o  --output=<filename>    Output filename (default: stdout).\n"
"      --zone-axis            Generate hk0 intensities only (and add\n"
"                              Synth2D-style header.\n"
);
Thomas White's avatar
Thomas White committed
44
45
46
}


47
static int template_reflections(const char *filename, unsigned int *counts)
Thomas White's avatar
Thomas White committed
48
49
50
51
52
53
{
	char *rval;
	FILE *fh;

	fh = fopen(filename, "r");
	if ( fh == NULL ) {
54
		return 1;
Thomas White's avatar
Thomas White committed
55
56
57
58
59
60
61
62
63
64
65
66
67
	}

	do {

		char line[1024];
		int r;
		signed int h, k, l;

		rval = fgets(line, 1023, fh);

		r = sscanf(line, "%i %i %i", &h, &k, &l);
		if ( r != 3 ) continue;

68
		set_count(counts, h, k, l, 1);
Thomas White's avatar
Thomas White committed
69
70
71
72
73

	} while ( rval != NULL );

	fclose(fh);

74
	return 0;
Thomas White's avatar
Thomas White committed
75
76
77
78
79
80
81
82
83
84
85
86
87
}


/* Apply Poisson noise to all reflections */
static void noisify_reflections(double *ref)
{
	signed int h, k, l;

	for ( h=-INDMAX; h<INDMAX; h++ ) {
	for ( k=-INDMAX; k<INDMAX; k++ ) {
	for ( l=-INDMAX; l<INDMAX; l++ ) {

		double val;
Thomas White's avatar
Thomas White committed
88
		int c;
Thomas White's avatar
Thomas White committed
89
90

		val = lookup_intensity(ref, h, k, l);
Thomas White's avatar
Thomas White committed
91
92
		c = poisson_noise(val);
		set_intensity(ref, h, k, l, c);
Thomas White's avatar
Thomas White committed
93
94
95

	}
	}
Thomas White's avatar
Thomas White committed
96
	progress_bar(h+INDMAX, 2*INDMAX, "Simulating noise");
Thomas White's avatar
Thomas White committed
97
	}
98
99
100
101
102
103
}


int main(int argc, char *argv[])
{
	int c;
Thomas White's avatar
Thomas White committed
104
	double *ideal_ref;
105
	struct molecule *mol;
Thomas White's avatar
Thomas White committed
106
107
	char *template = NULL;
	int config_noisify = 0;
Thomas White's avatar
Thomas White committed
108
	int config_twin = 0;
109
	int config_za = 0;
Thomas White's avatar
Thomas White committed
110
	char *output = NULL;
Thomas White's avatar
Thomas White committed
111
	unsigned int *counts;
112
	unsigned int *cts;
Thomas White's avatar
Thomas White committed
113
	signed int h, k, l;
114
115
116
117

	/* Long options */
	const struct option longopts[] = {
		{"help",               0, NULL,               'h'},
Thomas White's avatar
Thomas White committed
118
119
		{"template",           1, NULL,               't'},
		{"poisson",            0, &config_noisify,     1},
Thomas White's avatar
Thomas White committed
120
		{"output",             1, NULL,               'o'},
Thomas White's avatar
Thomas White committed
121
		{"twin",               0, &config_twin,        1},
122
		{"zone-axis",          0, &config_za,          1},
123
124
125
126
		{0, 0, NULL, 0}
	};

	/* Short options */
Thomas White's avatar
Thomas White committed
127
	while ((c = getopt_long(argc, argv, "ht:o:", longopts, NULL)) != -1) {
128
129
130
131
132
133
134

		switch (c) {
		case 'h' : {
			show_help(argv[0]);
			return 0;
		}

Thomas White's avatar
Thomas White committed
135
136
137
138
139
		case 't' : {
			template = strdup(optarg);
			break;
		}

Thomas White's avatar
Thomas White committed
140
141
142
143
144
		case 'o' : {
			output = strdup(optarg);
			break;
		}

145
146
147
148
149
150
151
152
153
154
155
156
		case 0 : {
			break;
		}

		default : {
			return 1;
		}
		}

	}

	mol = load_molecule();
157
158
	cts = new_list_count();
	ideal_ref = get_reflections(mol, eV_to_J(1790.0), 1/(0.6e-9), cts);
159

Thomas White's avatar
Thomas White committed
160
	counts = new_list_count();
Thomas White's avatar
Thomas White committed
161

Thomas White's avatar
Thomas White committed
162
	if ( template != NULL ) {
Thomas White's avatar
Thomas White committed
163

164
165
		if ( template_reflections(template, counts) != 0 ) {
			ERROR("Failed to template reflections.\n");
Thomas White's avatar
Thomas White committed
166
167
			return 1;
		}
Thomas White's avatar
Thomas White committed
168
169
170

	} else {

171
172
		/* No template? Then only mark reflections which were
		 * calculated. */
Thomas White's avatar
Thomas White committed
173
174
175
		for ( h=-INDMAX; h<=INDMAX; h++ ) {
		for ( k=-INDMAX; k<=INDMAX; k++ ) {
		for ( l=-INDMAX; l<=INDMAX; l++ ) {
176
177
178
			unsigned int c;
			c = lookup_count(cts, h, k, l);
			set_count(counts, h, k, l, c);
Thomas White's avatar
Thomas White committed
179
180
181
		}
		}
		}
Thomas White's avatar
Thomas White committed
182

Thomas White's avatar
Thomas White committed
183
184
	}

185
	if ( config_noisify ) noisify_reflections(ideal_ref);
Thomas White's avatar
Thomas White committed
186

Thomas White's avatar
Thomas White committed
187
188
189
190
191
192
	if ( config_twin ) {

		for ( h=-INDMAX; h<=INDMAX; h++ ) {
		for ( k=-INDMAX; k<=INDMAX; k++ ) {
		for ( l=-INDMAX; l<=INDMAX; l++ ) {

193
194
195
196
197
198
199
			double a, b, c, d;
			double t;

			if ( abs(h+k) > INDMAX ) {
				set_intensity(ideal_ref, h, k, l, 0.0);
				continue;
			}
Thomas White's avatar
Thomas White committed
200

201
202
			a = lookup_intensity(ideal_ref, h, k, l);
			b = lookup_intensity(ideal_ref, k, h, -l);
203
204
205
206
			c = lookup_intensity(ideal_ref, -(h+k), k, -l);
			d = lookup_intensity(ideal_ref, -(h+k), h, l);

			t = (a+b+c+d)/4.0;
Thomas White's avatar
Thomas White committed
207

208
209
210
211
			set_intensity(ideal_ref, h, k, l, t);
			set_intensity(ideal_ref, k, h, -l, t);
			set_intensity(ideal_ref, -(h+k), h, l, t);
			set_intensity(ideal_ref, -(h+k), k, -l, t);
Thomas White's avatar
Thomas White committed
212
213
214
215
216
217
218

		}
		}
		}

	}

219
	write_reflections(output, counts, ideal_ref, config_za, mol->cell);
220
221
222

	return 0;
}