aboutsummaryrefslogtreecommitdiff
path: root/src/refinetest3d2.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/refinetest3d2.c')
-rw-r--r--src/refinetest3d2.c191
1 files changed, 191 insertions, 0 deletions
diff --git a/src/refinetest3d2.c b/src/refinetest3d2.c
new file mode 100644
index 0000000..38ea72e
--- /dev/null
+++ b/src/refinetest3d2.c
@@ -0,0 +1,191 @@
+/*
+ * refinetest3d2.c
+ *
+ * Unit test for refinement procedure in 3D
+ *
+ * (c) 2007-2008 Thomas White <taw27@cam.ac.uk>
+ *
+ * dtr - Diffraction Tomography Reconstruction
+ *
+ */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <stdlib.h>
+#include <math.h>
+
+#include "basis.h"
+#include "reflections.h"
+#include "image.h"
+#include "reproject.h"
+#include "refine.h"
+#include "utils.h"
+#include "mapping.h"
+#include "control.h"
+#include "displaywindow.h"
+#include "mrc.h"
+
+static int check_cell(Basis *cell, Basis *cell_real) {
+
+ int fail;
+
+ printf(" Refinement Actual\n");
+ printf("---------------------------\n");
+ printf("ax %+8f %+8f\n", cell->a.x/1e9, cell_real->a.x/1e9);
+ printf("ay %+8f %+8f\n", cell->a.y/1e9, cell_real->a.y/1e9);
+ printf("az %+8f %+8f\n", cell->a.z/1e9, cell_real->a.z/1e9);
+ printf("bx %+8f %+8f\n", cell->b.x/1e9, cell_real->b.x/1e9);
+ printf("by %+8f %+8f\n", cell->b.y/1e9, cell_real->b.y/1e9);
+ printf("bz %+8f %+8f\n", cell->b.z/1e9, cell_real->b.z/1e9);
+ printf("cx %+8f %+8f\n", cell->c.x/1e9, cell_real->c.x/1e9);
+ printf("cy %+8f %+8f\n", cell->c.y/1e9, cell_real->c.y/1e9);
+ printf("cz %+8f %+8f\n", cell->c.z/1e9, cell_real->c.z/1e9);
+
+ fail = 0;
+ if ( fabs(cell->a.x - cell_real->a.x) > 0.01e9 ) {
+ fprintf(stderr, "refinetest3d: ax not determined correctly (got %8f, should be %8f)\n",
+ cell->a.x/1e9, cell_real->a.x/1e9);
+ fail = 1;
+ } else {
+ printf("ax is OK.\n");
+ }
+ if ( fabs(cell->a.y - cell_real->a.y) > 0.01e9 ) {
+ fprintf(stderr, "refinetest3d: ay not determined correctly (got %8f, should be %8f)\n",
+ cell->a.y/1e9, cell_real->a.y/1e9);
+ fail = 1;
+ } else {
+ printf("ay is OK.\n");
+ }
+ if ( fabs(cell->a.z - cell_real->a.z) > 0.01e9 ) {
+ fprintf(stderr, "refinetest3d: az not determined correctly (got %8f, should be %8f)\n",
+ cell->a.z/1e9, cell_real->a.z/1e9);
+ fail = 1;
+ } else {
+ printf("az is OK.\n");
+ }
+ if ( fabs(cell->b.x - cell_real->b.x) > 0.01e9 ) {
+ fprintf(stderr, "refinetest3d: bx not determined correctly (got %8f, should be %8f)\n",
+ cell->b.x/1e9, cell_real->b.x/1e9);
+ fail = 1;
+ } else {
+ printf("bx is OK.\n");
+ }
+ if ( fabs(cell->b.y - cell_real->b.y) > 0.01e9 ) {
+ fprintf(stderr, "refinetest3d: by not determined correctly (got %8f, should be %8f)\n",
+ cell->b.y/1e9, cell_real->b.y/1e9);
+ fail = 1;
+ } else {
+ printf("by is OK.\n");
+ }
+ if ( fabs(cell->b.z - cell_real->b.z) > 0.01e9 ) {
+ fprintf(stderr, "refinetest3d: bz not determined correctly (got %8f, should be %8f)\n",
+ cell->b.z/1e9, cell_real->b.z/1e9);
+ fail = 1;
+ } else {
+ printf("bz is OK.\n");
+ }
+ if ( fabs(cell->c.x - cell_real->c.x) > 0.01e9 ) {
+ fprintf(stderr, "refinetest3d: cx not determined correctly (got %8f, should be %8f)\n",
+ cell->c.x/1e9, cell_real->c.x/1e9);
+ fail = 1;
+ } else {
+ printf("cx is OK.\n");
+ }
+ if ( fabs(cell->c.y - cell_real->c.y) > 0.01e9 ) {
+ fprintf(stderr, "refinetest3d: cy not determined correctly (got %8f, should be %8f)\n",
+ cell->c.y/1e9, cell_real->c.y/1e9);
+ fail = 1;
+ } else {
+ printf("cx is OK.\n");
+ }
+ if ( fabs(cell->c.z - cell_real->c.z) > 0.01e9 ) {
+ fprintf(stderr, "refinetest3d: cz not determined correctly (got %8f, should be %8f)\n",
+ cell->c.z/1e9, cell_real->c.z/1e9);
+ fail = 1;
+ } else {
+ printf("cz is OK.\n");
+ }
+
+ return fail;
+
+}
+
+int main(int argc, char *argv[]) {
+
+ ControlContext *ctx;
+ ReflectionList *reflections_real;
+ Basis *cell_real;
+ int fail;
+
+ ctx = control_ctx_new();
+ ctx->omega = deg2rad(0.0);
+ ctx->lambda = lambda(300.0e3); /* 300 keV */
+ ctx->fmode = FORMULATION_PIXELSIZE;
+ ctx->x_centre = 256;
+ ctx->y_centre = 256;
+ ctx->pixel_size = 5e7;
+ image_add(ctx->images, NULL, 512, 512, deg2rad(00.0), ctx);
+ ctx->images->images[0].features = image_feature_list_new();
+ image_add(ctx->images, NULL, 512, 512, deg2rad(90.0), ctx);
+ ctx->images->images[1].features = image_feature_list_new();
+ ctx->omega = deg2rad(90.0);
+ image_add(ctx->images, NULL, 512, 512, deg2rad(90.0), ctx);
+ ctx->images->images[2].features = image_feature_list_new();
+
+ /* Fudge to avoid horrifying pointer-related death */
+ ctx->dw = malloc(sizeof(DisplayWindow));
+ ctx->dw->cur_image = 0;
+
+ /* The "true" cell */
+ cell_real = malloc(sizeof(Basis));
+ cell_real->a.x = 5.0e9; cell_real->a.y = 0.0e9; cell_real->a.z = 0.0e9;
+ cell_real->b.x = 0.0e9; cell_real->b.y = 5.0e9; cell_real->b.z = 0.0e9;
+ cell_real->c.x = 0.0e9; cell_real->c.y = 0.0e9; cell_real->c.z = 5.0e9;
+ /* The "real" reflections */
+ reflections_real = reflection_list_from_cell(cell_real);
+ ctx->images->images[0].features = reproject_get_reflections(&ctx->images->images[0], reflections_real);
+ ctx->images->images[1].features = reproject_get_reflections(&ctx->images->images[1], reflections_real);
+ ctx->images->images[2].features = reproject_get_reflections(&ctx->images->images[2], reflections_real);
+
+ /* The "model" cell to be refined */
+ ctx->cell = malloc(sizeof(Basis));
+ ctx->cell->a.x = 5.0e9; ctx->cell->a.y = 0.1e9; ctx->cell->a.z = 0.1e9;
+ ctx->cell->b.x = 0.1e9; ctx->cell->b.y = 5.0e9; ctx->cell->b.z = 0.1e9;
+ ctx->cell->c.x = 0.1e9; ctx->cell->c.y = 0.1e9; ctx->cell->c.z = 5.0e9;
+ ctx->cell_lattice = reflection_list_from_cell(ctx->cell);
+ ctx->images->images[0].rflist = reproject_get_reflections(&ctx->images->images[0], ctx->cell_lattice);
+ reproject_partner_features(ctx->images->images[0].rflist, &ctx->images->images[0]);
+ ctx->images->images[1].rflist = reproject_get_reflections(&ctx->images->images[1], ctx->cell_lattice);
+ reproject_partner_features(ctx->images->images[1].rflist, &ctx->images->images[1]);
+ ctx->images->images[2].rflist = reproject_get_reflections(&ctx->images->images[2], ctx->cell_lattice);
+ reproject_partner_features(ctx->images->images[2].rflist, &ctx->images->images[2]);
+
+ refine_do_cell(ctx);
+ image_feature_list_free(ctx->images->images[0].rflist);
+ image_feature_list_free(ctx->images->images[1].rflist);
+ image_feature_list_free(ctx->images->images[2].rflist);
+
+ fail = check_cell(ctx->cell, cell_real);
+
+ free(ctx);
+
+ if ( fail ) return 1;
+
+ printf("\n3D refinement test OK.\n");
+
+ return 0;
+
+}
+
+/* Dummy function stubs */
+#include "gtk-valuegraph.h"
+void displaywindow_update_imagestack(DisplayWindow *dw) { };
+void displaywindow_enable_cell_functions(DisplayWindow *dw, gboolean g) { };
+void displaywindow_update(DisplayWindow *dw) { };
+void displaywindow_error(const char *msg, DisplayWindow *dw) { };
+guint gtk_value_graph_get_type() { return 0; };
+GtkWidget *gtk_value_graph_new() { return NULL; };
+void gtk_value_graph_set_data(GtkValueGraph *vg, double *data, unsigned int n) { };
+