From e4ca31f5b0f447695949012c316f35b8e1a75404 Mon Sep 17 00:00:00 2001 From: Gregory Tucker Date: Fri, 17 Nov 2023 15:17:15 +0100 Subject: [PATCH] [Remove] errant test C file --- test/splitRunTest.c | 11848 ------------------------------------------ 1 file changed, 11848 deletions(-) delete mode 100644 test/splitRunTest.c diff --git a/test/splitRunTest.c b/test/splitRunTest.c deleted file mode 100644 index 4ca8295..0000000 --- a/test/splitRunTest.c +++ /dev/null @@ -1,11848 +0,0 @@ -/* Automatically generated file. Do not edit. - * Format: ANSI C source code - * Creator: McStas - * Instrument: None (splitRunTest) - * Date: 2023-10-16 12:14:23.566672 - * File: None - * CFLAGS= - */ - -#define MCCODE_STRING "McStas" -#define FLAVOR "mcstas" -#define FLAVOR_UPPER "MCSTAS" -#define MC_USE_DEFAULT_MAIN - - - -#include - -typedef double MCNUM; -typedef struct {MCNUM x, y, z;} Coords; -typedef MCNUM Rotation[3][3]; -#define MCCODE_BASE_TYPES - -#ifndef MC_NUSERVAR -#define MC_NUSERVAR 10 -#endif - -/* Particle JUMP control logic */ -struct particle_logic_struct { - int dummy; -}; -struct _struct_particle { - double x,y,z; /* position [m] */ - double vx,vy,vz; /* velocity [m/s] */ - double sx,sy,sz; /* spin [0-1] */ - int mcgravitation; /* gravity-state */ - void *mcMagnet; /* precession-state */ - int allow_backprop; /* allow backprop */ - unsigned long randstate[7]; - double t, p; /* time, event weight */ - long long _uid; /* event ID */ - long _index; /* component index where to send this event */ - long _absorbed; /* flag set to TRUE when this event is to be removed/ignored */ - long _scattered; /* flag set to TRUE when this event has interacted with the last component instance */ - long _restore; /* set to true if neutron event must be restored */ - long flag_nocoordschange; /* set to true if particle is jumping */ - struct particle_logic_struct _logic; - -}; -typedef struct _struct_particle _class_particle; - -_class_particle _particle_global_randnbuse_var; -_class_particle* _particle = &_particle_global_randnbuse_var; - -// Below lines relating to mcgenstate / setstate are in principle McStas - centric, we ought to generate -//this function based on "project" -#pragma acc routine -_class_particle mcgenstate(void); -#pragma acc routine -_class_particle mcsetstate(double x, double y, double z, double vx, double vy, double vz, - double t, double sx, double sy, double sz, double p, int mcgravitation, void *mcMagnet, int mcallowbackprop); - -extern int mcgravitation; /* flag to enable gravitation */ -#pragma acc declare create ( mcgravitation ) -int mcallowbackprop; -#pragma acc declare create ( mcallowbackprop ) - -_class_particle mcgenstate(void) { - _class_particle particle = mcsetstate(0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, mcgravitation, NULL, mcallowbackprop); - return(particle); -} -/*Generated user variable handlers:*/ - -#pragma acc routine -double particle_getvar(_class_particle *p, char *name, int *suc); - -#ifdef OPENACC -#pragma acc routine -int str_comp(char *str1, char *str2); -#endif - -double particle_getvar(_class_particle *p, char *name, int *suc){ -#ifndef OPENACC -#define str_comp strcmp -#endif - int s=1; - double rval=0; - if(!str_comp("x",name)){rval=p->x;s=0;} - if(!str_comp("y",name)){rval=p->y;s=0;} - if(!str_comp("z",name)){rval=p->z;s=0;} - if(!str_comp("vx",name)){rval=p->vx;s=0;} - if(!str_comp("vy",name)){rval=p->vy;s=0;} - if(!str_comp("vz",name)){rval=p->vz;s=0;} - if(!str_comp("sx",name)){rval=p->sx;s=0;} - if(!str_comp("sy",name)){rval=p->sy;s=0;} - if(!str_comp("sz",name)){rval=p->sz;s=0;} - if(!str_comp("t",name)){rval=p->t;s=0;} - if(!str_comp("p",name)){rval=p->p;s=0;} - - if (suc!=0x0) {*suc=s;} - return rval; -} - -#pragma acc routine -void* particle_getvar_void(_class_particle *p, char *name, int *suc); - -#ifdef OPENACC -#pragma acc routine -int str_comp(char *str1, char *str2); -#endif - -void* particle_getvar_void(_class_particle *p, char *name, int *suc){ -#ifndef OPENACC -#define str_comp strcmp -#endif - int s=1; - void* rval=0; - if(!str_comp("x",name)) {rval=(void*)&(p->x); s=0;} - if(!str_comp("y",name)) {rval=(void*)&(p->y); s=0;} - if(!str_comp("z",name)) {rval=(void*)&(p->z); s=0;} - if(!str_comp("vx",name)){rval=(void*)&(p->vx);s=0;} - if(!str_comp("vy",name)){rval=(void*)&(p->vy);s=0;} - if(!str_comp("vz",name)){rval=(void*)&(p->vz);s=0;} - if(!str_comp("sx",name)){rval=(void*)&(p->sx);s=0;} - if(!str_comp("sy",name)){rval=(void*)&(p->sy);s=0;} - if(!str_comp("sz",name)){rval=(void*)&(p->sz);s=0;} - if(!str_comp("t",name)) {rval=(void*)&(p->t); s=0;} - if(!str_comp("p",name)) {rval=(void*)&(p->p); s=0;} - - if (suc!=0x0) {*suc=s;} - return rval; -} - -#pragma acc routine -int particle_setvar_void(_class_particle *, char *, void*); - -int particle_setvar_void(_class_particle *p, char *name, void* value){ -#ifndef OPENACC -#define str_comp strcmp -#endif - int rval=1; - if(!str_comp("x",name)) {memcpy(&(p->x), value, sizeof(double)); rval=0;} - if(!str_comp("y",name)) {memcpy(&(p->y), value, sizeof(double)); rval=0;} - if(!str_comp("z",name)) {memcpy(&(p->z), value, sizeof(double)); rval=0;} - if(!str_comp("vx",name)){memcpy(&(p->vx), value, sizeof(double)); rval=0;} - if(!str_comp("vy",name)){memcpy(&(p->vy), value, sizeof(double)); rval=0;} - if(!str_comp("vz",name)){memcpy(&(p->vz), value, sizeof(double)); rval=0;} - if(!str_comp("sx",name)){memcpy(&(p->sx), value, sizeof(double)); rval=0;} - if(!str_comp("sy",name)){memcpy(&(p->sy), value, sizeof(double)); rval=0;} - if(!str_comp("sz",name)){memcpy(&(p->sz), value, sizeof(double)); rval=0;} - if(!str_comp("p",name)) {memcpy(&(p->p), value, sizeof(double)); rval=0;} - if(!str_comp("t",name)) {memcpy(&(p->t), value, sizeof(double)); rval=0;} - - return rval; -} - -#pragma acc routine -int particle_setvar_void_array(_class_particle *, char *, void*, int); - -int particle_setvar_void_array(_class_particle *p, char *name, void* value, int elements){ -#ifndef OPENACC -#define str_comp strcmp -#endif - int rval=1; - - return rval; -} - -// Function to handle a particle restore of physical particle params -#pragma acc routine -void particle_restore(_class_particle *p, _class_particle *p0); -void particle_restore(_class_particle *p, _class_particle *p0) { - p->x = p0->x; p->y = p0->y; p->z = p0->z; - p->vx = p0->vx; p->vy = p0->vy; p->vz = p0->vz; - p->sx = p0->sx; p->sy = p0->sy; p->sz = p0->sz; - p->t = p0->t; p->p = p0->p; - p->_absorbed=0; p->_restore=0; -} - -#pragma acc routine -double particle_getuservar_byid(_class_particle *p, int id, int *suc){ - int s=1; - double rval=0; - switch(id){ - - } - if (suc!=0x0) {*suc=s;} - return rval; -} - -#pragma acc routine -void particle_uservar_init(_class_particle *p){ - -} - -#define MC_EMBEDDED_RUNTIME -/* embedding file "/home/g/.cache/mccode-libc/main/mcstas-d.h" */ -#ifndef MCSTAS_D_H -#define MCSTAS_D_H - -#ifndef MCCODE_STRING -# define MCCODE_STRING "McStas" -#endif - -#ifndef MCCODE_DATE -# define MCCODE_DATE "2023-07-22" -#endif - -#ifndef MCCODE_VERSION -# define MCCODE_VERSION "4" -#endif - -#ifndef MCCODE_NAME -# define MCCODE_NAME "MCSTAS" -#endif - -#ifndef MCCODE_PARTICLE -# define MCCODE_PARTICLE "neutron" -#endif - -#ifndef MCCODE_PARTICLE_CODE -# define MCCODE_PARTICLE_CODE 2112 -#endif - -#ifndef MCCODE_LIBENV -# define MCCODE_LIBENV "MCSTAS" -#endif - -#ifndef FLAVOR_UPPER -# define FLAVOR_UPPER MCCODE_NAME -#endif - -#ifndef MCCODE_PROJECT -# define MCCODE_PROJECT 1 -#endif - -#ifndef FLAVOR_LIB -# define FLAVOR_LIB "nlib" -#endif - -#endif /* MCSTAS_D_H */ -/* end of file "/home/g/.cache/mccode-libc/main/mcstas-d.h" */ -/* embedding file "/home/g/.cache/mccode-libc/main/mccode-r.h" */ -/******************************************************************************* -* -* McCode, neutron/xray ray-tracing package -* Copyright (C) 1997-2009, All rights reserved -* Risoe National Laboratory, Roskilde, Denmark -* Institut Laue Langevin, Grenoble, France -* -* Runtime: share/mccode-r.h -* -* %Identification -* Written by: KN -* Date: Aug 29, 1997 -* Release: @MCCODE_NAME@ @MCCODE_VERSION@ -* Version: $Revision$ -* -* Runtime system header for McStas/McXtrace. -* -* In order to use this library as an external library, the following variables -* and macros must be declared (see details in the code) -* -* struct mcinputtable_struct mcinputtable[]; -* int numipar; -* metadata_table_t metadata_table[]; -* int num_metadata; -* char instrument_name[], instrument_source[]; -* int traceenabled, defaultmain; -* extern MCNUM mccomp_storein[]; -* extern MCNUM mcAbsorbProp[]; -* extern MCNUM mcScattered; -* #define MCCODE_STRING "the McStas/McXtrace version" -* -* Usage: Automatically embbeded in the c code. -* -* $Id$ -* -*******************************************************************************/ - -#ifndef MCCODE_R_H -#define MCCODE_R_H "$Revision$" - -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#ifdef OPENACC -#include -#ifndef GCCOFFLOAD -#include -#else -#include -#endif -#pragma acc routine -int noprintf(); -#pragma acc routine -size_t str_len(const char *s); -#else -#include -#endif - -/* If the runtime is embedded in the simulation program, some definitions can - be made static. */ - -#ifdef MC_EMBEDDED_RUNTIME -# define mcstatic -#else -# define mcstatic -#endif - -#ifdef __dest_os -# if (__dest_os == __mac_os) -# define MAC -# endif -#endif - -#ifdef __FreeBSD__ -# define NEED_STAT_H -#endif - -#if defined(__APPLE__) && defined(__GNUC__) -# define NEED_STAT_H -#endif - -#ifdef WIN32 -# define NEED_STAT_H -# define NEED_TYPES_H -#endif - -#ifdef NEED_STAT_H -# include -#endif - -#ifdef NEED_TYPES_H -# include -#endif - -#ifndef MC_PATHSEP_C -# ifdef WIN32 -# define MC_PATHSEP_C '\\' -# define MC_PATHSEP_S "\\" -# else /* !WIN32 */ -# define MC_PATHSEP_C '/' -# define MC_PATHSEP_S "/" -# endif /* !WIN32 */ -#endif /* MC_PATHSEP_C */ - -#ifdef WIN32 -# define mkdir(a,b) mkdir(a) -# define getpid() NULL -#endif - -#ifdef MC_PORTABLE -# ifndef NOSIGNALS -# define NOSIGNALS 1 -# endif -#endif - -#ifdef MAC -# ifndef NOSIGNALS -# define NOSIGNALS 1 -# endif -#endif - -#if (USE_MPI == 0) -# undef USE_MPI -#endif - -#ifdef USE_MPI /* default is to disable signals with MPI, as MPICH uses them to communicate */ -# ifndef NOSIGNALS -# define NOSIGNALS 1 -# endif -#endif - -#ifdef OPENACC /* default is to disable signals with PGI/OpenACC */ -# ifndef NOSIGNALS -# define NOSIGNALS 1 -# endif -#endif - -#ifndef OPENACC -# ifndef USE_OFF /* default is to enable OFF when not using PGI/OpenACC */ -# define USE_OFF -# endif -# ifndef CPUFUNNEL /* allow to enable FUNNEL-mode on CPU */ -# ifdef FUNNEL /* by default disable FUNNEL-mode when not using PGI/OpenACC */ -# undef FUNNEL -# endif -# endif -#endif - -#if (NOSIGNALS == 0) -# undef NOSIGNALS -#endif - -/** Header information for metadata-r.c ----------------------------------------------------------------------------- */ -struct metadata_table_struct { /* stores metadata strings from components */ - char * source; // component name which provided the metadata - char * name; // the name of the metadata - char * type; // the MIME type of the metadata (free form, valid identifier) - char * value; // the metadata string contents -}; -typedef struct metadata_table_struct metadata_table_t; -char * metadata_table_key_component(char* key); -char * metadata_table_key_literal(char * key); -int metadata_table_defined(int, metadata_table_t *, char *); -char * metadata_table_type(int, metadata_table_t *, char *); -char * metadata_table_literal(int, metadata_table_t *, char *); -void metadata_table_print_all_keys(int no, metadata_table_t * tab); -int metadata_table_print_all_components(int no, metadata_table_t * tab); -int metadata_table_print_component_keys(int no, metadata_table_t * tab, char * key); -/* -------------------------------------------------------------------------- Header information for metadata-r.c --- */ - -/* Note: the enum instr_formal_types definition MUST be kept - synchronized with the one in mccode.h and with the - instr_formal_type_names array in cogen.c. */ -enum instr_formal_types - { - instr_type_int, - instr_type_string, instr_type_char, - instr_type_vector, instr_type_double - }; -struct mcinputtable_struct { /* defines instrument parameters */ - char *name; /* name of parameter */ - void *par; /* pointer to instrument parameter (variable) */ - enum instr_formal_types type; - char *val; /* default value */ - char *unit; /* expected unit for parameter; informational only */ -}; - - -#ifndef MCCODE_BASE_TYPES -typedef double MCNUM; -typedef struct {MCNUM x, y, z;} Coords; -typedef MCNUM Rotation[3][3]; -#endif - -/* the following variables are defined in the McStas generated C code - but should be defined externally in case of independent library usage */ -#ifndef DANSE -extern struct mcinputtable_struct mcinputtable[]; /* list of instrument parameters */ -extern int numipar; /* number of instrument parameters */ -extern metadata_table_t metadata_table[]; /* list of component-defined string metadata */ -extern int num_metadata; /* number of component-defined string metadata */ -extern char instrument_name[], instrument_source[]; /* instrument name and filename */ -extern char *instrument_exe; /* executable path = argv[0] or NULL */ -extern char instrument_code[]; /* contains the initial 'instr' file */ - -#ifndef MC_ANCIENT_COMPATIBILITY -extern int traceenabled, defaultmain; -#endif -#endif - - -/* Useful macros ============================================================ */ - - -/* SECTION: Dynamic Arrays */ -typedef int* IArray1d; -IArray1d create_iarr1d(int n); -void destroy_iarr1d(IArray1d a); - -typedef int** IArray2d; -IArray2d create_iarr2d(int nx, int ny); -void destroy_iarr2d(IArray2d a); - -typedef int*** IArray3d; -IArray3d create_iarr3d(int nx, int ny, int nz); -void destroy_iarr3d(IArray3d a); - -typedef double* DArray1d; -DArray1d create_darr1d(int n); -void destroy_darr1d(DArray1d a); - -typedef double** DArray2d; -DArray2d create_darr2d(int nx, int ny); -void destroy_darr2d(DArray2d a); - -typedef double*** DArray3d; -DArray3d create_darr3d(int nx, int ny, int nz); -void destroy_darr3d(DArray3d a); - - -/* MPI stuff */ -#ifdef USE_MPI -#include "mpi.h" - -#ifdef OMPI_MPI_H /* openmpi does not use signals: we may install our sighandler */ -#ifndef OPENACC /* ... but only if we are not also running on GPU */ -#undef NOSIGNALS -#endif -#endif - -/* - * MPI_MASTER(i): - * execution of i only on master node - */ -#define MPI_MASTER(statement) { \ - if(mpi_node_rank == mpi_node_root)\ - { statement; } \ -} - -#ifndef MPI_REDUCE_BLOCKSIZE -#define MPI_REDUCE_BLOCKSIZE 1000 -#endif - -int mc_MPI_Sum(double* buf, long count); -int mc_MPI_Send(void *sbuf, long count, MPI_Datatype dtype, int dest); -int mc_MPI_Recv(void *rbuf, long count, MPI_Datatype dtype, int source); - -/* MPI_Finalize exits gracefully and should be preferred to MPI_Abort */ -#define exit(code) do { \ - MPI_Finalize(); \ - exit(code); \ - } while(0) - -#else /* !USE_MPI */ -#define MPI_MASTER(instr) instr -#endif /* USE_MPI */ - - -#ifdef USE_MPI -static int mpi_node_count; -#endif - -#ifdef USE_THREADS /* user want threads */ -#error Threading (USE_THREADS) support has been removed for very poor efficiency. Use MPI/SSH grid instead. -#endif - - -void mcset_ncount(unsigned long long count); /* wrapper to get mcncount */ -#pragma acc routine -unsigned long long int mcget_ncount(void); /* wrapper to set mcncount */ -unsigned long long mcget_run_num(void); /* wrapper to get mcrun_num=0:mcncount-1 */ - -/* Following part is only embedded when not redundant with mccode.h ========= */ - -#ifndef MCCODE_H - -#ifndef NOSIGNALS -#include -char *mcsig_message; -#define SIG_MESSAGE(msg) mcsig_message=(char *)(msg); -#else -#define SIG_MESSAGE(...) -#endif /* !NOSIGNALS */ - - -/* Useful macros and constants ============================================== */ - - -#ifndef FLT_MAX -#define FLT_MAX 3.40282347E+38F /* max decimal value of a "float" */ -#endif - -#ifndef MIN -#define MIN(a, b) (((a) < (b)) ? (a) : (b)) -#endif -#ifndef MAX -#define MAX(a, b) (((a) > (b)) ? (a) : (b)) -#endif -#ifndef SQR -#define SQR(x) ( (x) * (x) ) -#endif -#ifndef SIGN -#define SIGN(x) (((x)>0.0)?(1):(-1)) -#endif - - -# ifndef M_E -# define M_E 2.71828182845904523536 // e -# endif -# ifndef M_LOG2E -# define M_LOG2E 1.44269504088896340736 // log2(e) -# endif -# ifndef M_LOG10E -# define M_LOG10E 0.434294481903251827651 // log10(e) -# endif -# ifndef M_LN2 -# define M_LN2 0.693147180559945309417 // ln(2) -# endif -# ifndef M_LN10 -# define M_LN10 2.30258509299404568402 // ln(10) -# endif -# ifndef M_PI -# define M_PI 3.14159265358979323846 // pi -# endif -# ifndef PI -# define PI M_PI // pi - also used in some places -# endif -# ifndef M_PI_2 -# define M_PI_2 1.57079632679489661923 // pi/2 -# endif -# ifndef M_PI_4 -# define M_PI_4 0.785398163397448309616 // pi/4 -# endif -# ifndef M_1_PI -# define M_1_PI 0.318309886183790671538 // 1/pi -# endif -# ifndef M_2_PI -# define M_2_PI 0.636619772367581343076 // 2/pi -# endif -# ifndef M_2_SQRTPI -# define M_2_SQRTPI 1.12837916709551257390 // 2/sqrt(pi) -# endif -# ifndef M_SQRT2 -# define M_SQRT2 1.41421356237309504880 // sqrt(2) -# endif -# ifndef M_SQRT1_2 -# define M_SQRT1_2 0.707106781186547524401 // 1/sqrt(2) -# endif - -#define RAD2MIN ((180*60)/PI) -#define MIN2RAD (PI/(180*60)) -#define DEG2RAD (PI/180) -#define RAD2DEG (180/PI) -#define FWHM2RMS 0.424660900144 /* Convert between full-width-half-max and */ -#define RMS2FWHM 2.35482004503 /* root-mean-square (standard deviation) */ -#define HBAR 1.05457168e-34 /* [Js] h bar Planck constant CODATA 2002 */ -#define MNEUTRON 1.67492728e-27 /* [kg] mass of neutron CODATA 2002 */ -#define GRAVITY 9.81 /* [m/s^2] gravitational acceleration */ -#define NA 6.02214179e23 /* [#atoms/g .mole] Avogadro's number*/ - - -#define UNSET nan("0x6E6F74736574") -int nans_match(double, double); -int is_unset(double); -int is_valid(double); -int is_set(double); -int all_unset(int n, ...); -int all_set(int n, ...); -int any_unset(int n, ...); -int any_set(int n, ...); - - -/* wrapper to get absolute and relative position of comp */ -/* mccomp_posa and mccomp_posr are defined in McStas generated C code */ -#define POS_A_COMP_INDEX(index) (instrument->_position_absolute[index]) -#define POS_R_COMP_INDEX(index) (instrument->_position_relative[index]) - -/* setting parameters based COMP_GETPAR (returned as pointer) */ -/* compname must be given as a string, type and par are symbols. */ -#define COMP_GETPAR3(type, compname, par) \ - &( ((_class_ ## type ##_parameters *) _getvar_parameters(compname))->par ) -/* the body of this function depends on component instances, and is cogen'd */ -void* _getvar_parameters(char* compname); - -int _getcomp_index(char* compname); - -/* Note: The two-stage approach to COMP_GETPAR is NOT redundant; without it, -* after #define C sample, COMP_GETPAR(C,x) would refer to component C, not to -* component sample. Such are the joys of ANSI C. - -* Anyway the usage of COMP_GETPAR requires that we use sometimes bare names... -* NOTE: This can ONLY be used in instrument descriptions, not components. -*/ -#define COMP_GETPAR2(comp, par) (_ ## comp ## _var._parameters.par) -#define COMP_GETPAR(comp, par) COMP_GETPAR2(comp,par) - -#define INSTRUMENT_GETPAR(par) (_instrument_var._parameters.par) - -/* Current component name, index, position and orientation */ -/* These macros work because, using class-based functions, "comp" is usually -* the local variable of the active/current component. */ -#define INDEX_CURRENT_COMP (_comp->_index) -#define NAME_CURRENT_COMP (_comp->_name) -#define TYPE_CURRENT_COMP (_comp->_type) -#define POS_A_CURRENT_COMP (_comp->_position_absolute) -#define POS_R_CURRENT_COMP (_comp->_position_relative) -#define ROT_A_CURRENT_COMP (_comp->_rotation_absolute) -#define ROT_R_CURRENT_COMP (_comp->_rotation_relative) - -#define NAME_INSTRUMENT (instrument->_name) - - -/* MCDISPLAY/trace and debugging message sent to stdout */ -#ifdef MC_TRACE_ENABLED -#define DEBUG -#endif - -#ifdef DEBUG -#define DEBUG_INSTR() if(!mcdotrace); else { printf("INSTRUMENT:\n"); printf("Instrument '%s' (%s)\n", instrument_name, instrument_source); } -#define DEBUG_COMPONENT(name,c,t) if(!mcdotrace); else {\ - printf("COMPONENT: \"%s\"\n" \ - "POS: %g, %g, %g, %g, %g, %g, %g, %g, %g, %g, %g, %g\n", \ - name, c.x, c.y, c.z, t[0][0], t[0][1], t[0][2], \ - t[1][0], t[1][1], t[1][2], t[2][0], t[2][1], t[2][2]); \ - printf("Component %30s AT (%g,%g,%g)\n", name, c.x, c.y, c.z); \ - } -#define DEBUG_INSTR_END() if(!mcdotrace); else printf("INSTRUMENT END:\n"); -#define DEBUG_ENTER() if(!mcdotrace); else printf("ENTER:\n"); -#define DEBUG_COMP(c) if(!mcdotrace); else printf("COMP: \"%s\"\n", c); -#define DEBUG_LEAVE() if(!mcdotrace); else printf("LEAVE:\n"); -#define DEBUG_ABSORB() if(!mcdotrace); else printf("ABSORB:\n"); -#else -#define DEBUG_INSTR() -#define DEBUG_COMPONENT(name,c,t) -#define DEBUG_INSTR_END() -#define DEBUG_ENTER() -#define DEBUG_COMP(c) -#define DEBUG_LEAVE() -#define DEBUG_ABSORB() -#endif - -// mcDEBUG_STATE and mcDEBUG_SCATTER are defined by mcstas-r.h and mcxtrace-r.h - - - -#ifdef TEST -#define test_printf printf -#else -#define test_printf while(0) printf -#endif - -/* send MCDISPLAY message to stdout to show gemoetry */ -void mcdis_magnify(char *what); -void mcdis_line(double x1, double y1, double z1, - double x2, double y2, double z2); -void mcdis_dashed_line(double x1, double y1, double z1, - double x2, double y2, double z2, int n); -void mcdis_multiline(int count, ...); -void mcdis_rectangle(char* plane, double x, double y, double z, - double width, double height); -void mcdis_box(double x, double y, double z, - double width, double height, double length); -void mcdis_circle(char *plane, double x, double y, double z, double r); -void mcdis_Circle(double x, double y, double z, double r, double nx, double ny, double nz); -void mcdis_cylinder( double x, double y, double z, - double r, double height, int N, double nx, double ny, double nz); -void mcdis_sphere(double x, double y, double z, double r, int N); - - -/* random number generation. ================================================ */ - -/* available random number generators */ -#define _RNG_ALG_MT 1 -#define _RNG_ALG_KISS 2 - -/* selection of random number generator */ -#ifndef RNG_ALG -# define RNG_ALG _RNG_ALG_KISS -#endif - - -#if RNG_ALG == _RNG_ALG_MT // MT (currently not functional for gpu) -# define MC_RAND_MAX ((unsigned long)0xffffffff) -# define randstate_t unsigned long // this could be anything -# define RANDSTATE_LEN 1 -# define srandom(seed) mt_srandom_empty() -# define random() mt_random() -# define _random() mt_random() -#elif RNG_ALG == _RNG_ALG_KISS // KISS -# ifndef ULONG_MAX -# define ULONG_MAX ((unsigned long)0xffffffffffffffffUL) -# endif -# define MC_RAND_MAX ULONG_MAX -# define randstate_t unsigned long -# define RANDSTATE_LEN 7 -# define srandom(seed) kiss_srandom(_particle->randstate, seed) -# define random() kiss_random(_particle->randstate) -# define _random() kiss_random(state) -#endif - -#pragma acc routine -double _randnorm2(randstate_t* state); - - -// component writers interface -#define randnorm() _randnorm2(_particle->randstate) // NOTE: can not use _randnorm on gpu -#define rand01() _rand01(_particle->randstate) -#define randpm1() _randpm1(_particle->randstate) -#define rand0max(p1) _rand0max(p1, _particle->randstate) -#define randminmax(p1, p2) _randminmax(p1, p2, _particle->randstate) -#define randtriangle() _randtriangle(_particle->randstate) - -// Mersenne Twister rng -unsigned long mt_random(void); -void mt_srandom (unsigned long x); -void mt_srandom_empty(); - -// KISS rng -#pragma acc routine -unsigned long *kiss_srandom(unsigned long state[7], unsigned long seed); -#pragma acc routine -unsigned long kiss_random(unsigned long state[7]); - -// Scrambler / hash function -#pragma acc routine seq -randstate_t _hash(randstate_t x); - -// internal RNG (transforms) interface -#pragma acc routine -double _rand01(randstate_t* state); -#pragma acc routine -double _randpm1(randstate_t* state); -#pragma acc routine -double _rand0max(double max, randstate_t* state); -#pragma acc routine -double _randminmax(double min, double max, randstate_t* state); -#pragma acc routine -double _randtriangle(randstate_t* state); - - -#ifdef USE_OPENCL -#include "opencl-lib.h" -#include "opencl-lib.c" -#endif - -#ifndef DANSE -int init(void); -int raytrace(_class_particle*); -int save(FILE *); -int finally(void); -int display(void); -#endif - - -/* GPU related algorithms =================================================== */ - -/* -* Divide-and-conquer strategy for parallel sort absorbed last. -*/ -#ifdef FUNNEL -long sort_absorb_last(_class_particle* particles, _class_particle* pbuffer, long len, long buffer_len, long flag_split, long* multiplier); -#endif -long sort_absorb_last_serial(_class_particle* particles, long len); - - -/* simple vector algebra ==================================================== */ - - -#define vec_prod(x, y, z, x1, y1, z1, x2, y2, z2) \ - vec_prod_func(&x, &y, &z, x1, y1, z1, x2, y2, z2) -#pragma acc routine seq -mcstatic void vec_prod_func(double *x, double *y, double *z, - double x1, double y1, double z1, double x2, double y2, double z2); - -#pragma acc routine seq -mcstatic double scalar_prod( - double x1, double y1, double z1, double x2, double y2, double z2); - -#pragma acc routine seq -mcstatic void norm_func(double *x, double *y, double *z); -#define NORM(x,y,z) norm_func(&x, &y, &z) - -#pragma acc routine seq -void normal_vec(double *nx, double *ny, double *nz, - double x, double y, double z); - -/** - * Rotate the vector vx,vy,vz psi radians around the vector ax,ay,az - * and put the result in x,y,z. - */ -#define rotate(x, y, z, vx, vy, vz, phi, ax, ay, az) \ - do { \ - double mcrt_tmpx = (ax), mcrt_tmpy = (ay), mcrt_tmpz = (az); \ - double mcrt_vp, mcrt_vpx, mcrt_vpy, mcrt_vpz; \ - double mcrt_vnx, mcrt_vny, mcrt_vnz, mcrt_vn1x, mcrt_vn1y, mcrt_vn1z; \ - double mcrt_bx, mcrt_by, mcrt_bz; \ - double mcrt_cos, mcrt_sin; \ - NORM(mcrt_tmpx, mcrt_tmpy, mcrt_tmpz); \ - mcrt_vp = scalar_prod((vx), (vy), (vz), mcrt_tmpx, mcrt_tmpy, mcrt_tmpz); \ - mcrt_vpx = mcrt_vp*mcrt_tmpx; \ - mcrt_vpy = mcrt_vp*mcrt_tmpy; \ - mcrt_vpz = mcrt_vp*mcrt_tmpz; \ - mcrt_vnx = (vx) - mcrt_vpx; \ - mcrt_vny = (vy) - mcrt_vpy; \ - mcrt_vnz = (vz) - mcrt_vpz; \ - vec_prod(mcrt_bx, mcrt_by, mcrt_bz, \ - mcrt_tmpx, mcrt_tmpy, mcrt_tmpz, mcrt_vnx, mcrt_vny, mcrt_vnz); \ - mcrt_cos = cos((phi)); mcrt_sin = sin((phi)); \ - mcrt_vn1x = mcrt_vnx*mcrt_cos + mcrt_bx*mcrt_sin; \ - mcrt_vn1y = mcrt_vny*mcrt_cos + mcrt_by*mcrt_sin; \ - mcrt_vn1z = mcrt_vnz*mcrt_cos + mcrt_bz*mcrt_sin; \ - (x) = mcrt_vpx + mcrt_vn1x; \ - (y) = mcrt_vpy + mcrt_vn1y; \ - (z) = mcrt_vpz + mcrt_vn1z; \ - } while(0) - -/** - * Mirror (xyz) in the plane given by the point (rx,ry,rz) and normal (nx,ny,nz) - * - * TODO: This define is seemingly never used... - */ -#define mirror(x,y,z,rx,ry,rz,nx,ny,nz) \ - do { \ - double mcrt_tmpx= (nx), mcrt_tmpy = (ny), mcrt_tmpz = (nz); \ - double mcrt_tmpt; \ - NORM(mcrt_tmpx, mcrt_tmpy, mcrt_tmpz); \ - mcrt_tmpt=scalar_prod((rx),(ry),(rz),mcrt_tmpx,mcrt_tmpy,mcrt_tmpz); \ - (x) = rx -2 * mcrt_tmpt*mcrt_rmpx; \ - (y) = ry -2 * mcrt_tmpt*mcrt_rmpy; \ - (z) = rz -2 * mcrt_tmpt*mcrt_rmpz; \ - } while (0) - -#pragma acc routine -Coords coords_set(MCNUM x, MCNUM y, MCNUM z); -#pragma acc routine -Coords coords_get(Coords a, MCNUM *x, MCNUM *y, MCNUM *z); -#pragma acc routine -Coords coords_add(Coords a, Coords b); -#pragma acc routine -Coords coords_sub(Coords a, Coords b); -#pragma acc routine -Coords coords_neg(Coords a); -#pragma acc routine -Coords coords_scale(Coords b, double scale); -#pragma acc routine -double coords_sp(Coords a, Coords b); -#pragma acc routine -Coords coords_xp(Coords b, Coords c); -#pragma acc routine -double coords_len(Coords a); -#pragma acc routine seq -void coords_print(Coords a); -#pragma acc routine seq -mcstatic void coords_norm(Coords* c); - -#pragma acc routine seq -void rot_set_rotation(Rotation t, double phx, double phy, double phz); -#pragma acc routine seq -int rot_test_identity(Rotation t); -#pragma acc routine seq -void rot_mul(Rotation t1, Rotation t2, Rotation t3); -#pragma acc routine seq -void rot_copy(Rotation dest, Rotation src); -#pragma acc routine seq -void rot_transpose(Rotation src, Rotation dst); -#pragma acc routine seq -Coords rot_apply(Rotation t, Coords a); - -#pragma acc routine seq -void mccoordschange(Coords a, Rotation t, _class_particle *particle); -#pragma acc routine seq -void mccoordschange_polarisation(Rotation t, double *sx, double *sy, double *sz); - -double mcestimate_error(double N, double p1, double p2); -void mcreadparams(void); - -/* this is now in mcstas-r.h and mcxtrace-r.h as the number of state parameters -is no longer equal */ - -_class_particle mcgenstate(void); - -// trajectory/shape intersection routines -#pragma acc routine seq -int inside_rectangle(double, double, double, double); -#pragma acc routine seq -int box_intersect(double *dt_in, double *dt_out, double x, double y, double z, - double vx, double vy, double vz, double dx, double dy, double dz); -#pragma acc routine seq -int cylinder_intersect(double *t0, double *t1, double x, double y, double z, - double vx, double vy, double vz, double r, double h); -#pragma acc routine seq -int sphere_intersect(double *t0, double *t1, double x, double y, double z, - double vx, double vy, double vz, double r); -// second order equation roots -#pragma acc routine seq -int solve_2nd_order(double *t1, double *t2, - double A, double B, double C); - -// random vector generation to shape -// defines silently introducing _particle as the last argument -#define randvec_target_circle(xo, yo, zo, solid_angle, xi, yi, zi, radius) \ - _randvec_target_circle(xo, yo, zo, solid_angle, xi, yi, zi, radius, _particle) -#define randvec_target_rect_angular(xo, yo, zo, solid_angle, xi, yi, zi, height, width, A) \ - _randvec_target_rect_angular(xo, yo, zo, solid_angle, xi, yi, zi, height, width, A, _particle) -#define randvec_target_rect_real(xo, yo, zo, solid_angle, xi, yi, zi, height, width, A, lx, ly, lz, order) \ - _randvec_target_rect_real(xo, yo, zo, solid_angle, xi, yi, zi, height, width, A, lx, ly, lz, order, _particle) -// defines forwarding to "inner" functions -#define randvec_target_sphere randvec_target_circle -#define randvec_target_rect(p0,p1,p2,p3,p4,p5,p6,p7,p8,p9) \ - randvec_target_rect_real(p0,p1,p2,p3,p4,p5,p6,p7,p8,p9,0,0,0,1) -// headers for randvec -#pragma acc routine seq -void _randvec_target_circle(double *xo, double *yo, double *zo, - double *solid_angle, double xi, double yi, double zi, double radius, - _class_particle* _particle); -#pragma acc routine seq -void _randvec_target_rect_angular(double *xo, double *yo, double *zo, - double *solid_angle, double xi, double yi, double zi, double height, - double width, Rotation A, - _class_particle* _particle); -#pragma acc routine seq -void _randvec_target_rect_real(double *xo, double *yo, double *zo, double *solid_angle, - double xi, double yi, double zi, double height, double width, Rotation A, - double lx, double ly, double lz, int order, - _class_particle* _particle); - - -// this is the main() -int mccode_main(int argc, char *argv[]); - - -#endif /* !MCCODE_H */ - -#ifndef MCCODE_R_IO_H -#define MCCODE_R_IO_H "$Revision$" - -#if (USE_NEXUS == 0) -#undef USE_NEXUS -#endif - -#ifndef CHAR_BUF_LENGTH -#define CHAR_BUF_LENGTH 1024 -#endif - - -/* I/O section part ========================================================= */ - -/* ========================================================================== */ - -/* MCCODE_R_IO_C */ - -/* ========================================================================== */ - - -/* main DETECTOR structure which stores most information to write to data files */ -struct mcdetector_struct { - char filename[CHAR_BUF_LENGTH]; /* file name of monitor */ - char position[CHAR_BUF_LENGTH]; /* position of detector component */ - char component[CHAR_BUF_LENGTH]; /* component instance name */ - char instrument[CHAR_BUF_LENGTH]; /* instrument name */ - char type[CHAR_BUF_LENGTH]; /* data type, e.g. 0d, 1d, 2d, 3d */ - char user[CHAR_BUF_LENGTH]; /* user name, e.g. HOME */ - char date[CHAR_BUF_LENGTH]; /* date of simulation end/write time */ - char title[CHAR_BUF_LENGTH]; /* title of detector */ - char xlabel[CHAR_BUF_LENGTH]; /* X axis label */ - char ylabel[CHAR_BUF_LENGTH]; /* Y axis label */ - char zlabel[CHAR_BUF_LENGTH]; /* Z axis label */ - char xvar[CHAR_BUF_LENGTH]; /* X variable name */ - char yvar[CHAR_BUF_LENGTH]; /* Y variable name */ - char zvar[CHAR_BUF_LENGTH]; /* Z variable name */ - char ncount[CHAR_BUF_LENGTH]; /* number of events initially generated */ - char limits[CHAR_BUF_LENGTH]; /* X Y Z limits, e.g. [xmin xmax ymin ymax zmin zmax] */ - char variables[CHAR_BUF_LENGTH]; /* variables written into data block */ - char statistics[CHAR_BUF_LENGTH]; /* center, mean and half width along axis */ - char signal[CHAR_BUF_LENGTH]; /* min max and mean of signal (data block) */ - char values[CHAR_BUF_LENGTH]; /* integrated values e.g. [I I_err N] */ - double xmin,xmax; /* min max of axes */ - double ymin,ymax; - double zmin,zmax; - double intensity; /* integrated values for data block */ - double error; - double events; - double min; /* statistics for data block */ - double max; - double mean; - double centerX; /* statistics for axes */ - double halfwidthX; - double centerY; - double halfwidthY; - int rank; /* dimensionaly of monitor, e.g. 0 1 2 3 */ - char istransposed; /* flag to transpose matrix for some formats */ - - long m,n,p; /* dimensions of data block and along axes */ - long date_l; /* same as date, but in sec since 1970 */ - - double *p0, *p1, *p2; /* pointers to saved data, NULL when freed */ - char format[CHAR_BUF_LENGTH]; /* format for file generation */ -}; - -typedef struct mcdetector_struct MCDETECTOR; - -static char *dirname = NULL; /* name of output directory */ -static char *siminfo_name = "mccode"; /* default output sim file name */ -char *mcformat = NULL; /* NULL (default) or a specific format */ - -/* file I/O definitions and function prototypes */ - -#ifndef MC_EMBEDDED_RUNTIME /* the mcstatic variables (from mccode-r.c) */ -extern FILE * siminfo_file; /* handle to the output siminfo file */ -extern int mcgravitation; /* flag to enable gravitation */ -extern int mcdotrace; /* flag to print MCDISPLAY messages */ -#else -mcstatic FILE *siminfo_file = NULL; -#endif - -/* I/O function prototypes ================================================== */ - -// from msysgit: https://code.google.com/p/msysgit/source/browse/compat/strcasestr.c -char *strcasestr(const char *haystack, const char *needle); - -/* output functions */ -MCDETECTOR mcdetector_out_0D(char *t, double p0, double p1, double p2, char *c, Coords pos); -MCDETECTOR mcdetector_out_1D(char *t, char *xl, char *yl, - char *xvar, double x1, double x2, long n, - double *p0, double *p1, double *p2, char *f, char *c, Coords pos); -MCDETECTOR mcdetector_out_2D(char *t, char *xl, char *yl, - double x1, double x2, double y1, double y2, long m, - long n, double *p0, double *p1, double *p2, char *f, - char *c, Coords pos); -MCDETECTOR mcdetector_out_list(char *t, char *xl, char *yl, - long m, long n, - double *p1, char *f, - char *c, Coords posa); - -/* wrappers to output functions, that automatically set NAME and POSITION */ -#define DETECTOR_OUT(p0,p1,p2) mcdetector_out_0D(NAME_CURRENT_COMP,p0,p1,p2,NAME_CURRENT_COMP,POS_A_CURRENT_COMP) -#define DETECTOR_OUT_0D(t,p0,p1,p2) mcdetector_out_0D(t,p0,p1,p2,NAME_CURRENT_COMP,POS_A_CURRENT_COMP) -#define DETECTOR_OUT_1D(t,xl,yl,xvar,x1,x2,n,p0,p1,p2,f) \ - mcdetector_out_1D(t,xl,yl,xvar,x1,x2,n,p0,p1,p2,f,NAME_CURRENT_COMP,POS_A_CURRENT_COMP) -#define DETECTOR_OUT_2D(t,xl,yl,x1,x2,y1,y2,m,n,p0,p1,p2,f) \ - mcdetector_out_2D(t,xl,yl,x1,x2,y1,y2,m,n,p0,p1,p2,f,NAME_CURRENT_COMP,POS_A_CURRENT_COMP) - -#ifdef USE_NEXUS -#include "napi.h" -NXhandle nxhandle; -#endif - -#endif /* ndef MCCODE_R_IO_H */ - -#endif /* MCCODE_R_H */ -/* end of file "/home/g/.cache/mccode-libc/main/mccode-r.h" */ -/* embedding file "/home/g/.cache/mccode-libc/main/nlib/mcstas-r.h" */ -/******************************************************************************* -* -* McStas, neutron ray-tracing package -* Copyright (C) 1997-2009, All rights reserved -* Risoe National Laboratory, Roskilde, Denmark -* Institut Laue Langevin, Grenoble, France -* -* Runtime: share/mcstas-r.h -* -* %Identification -* Written by: KN -* Date: Aug 29, 1997 -* Release: McStas X.Y -* Version: $Revision$ -* -* Runtime system header for McStas. -* -* In order to use this library as an external library, the following variables -* and macros must be declared (see details in the code) -* -* struct mcinputtable_struct mcinputtable[]; -* int mcnumipar; -* char instrument_name[], instrument_source[]; -* int traceenabled, defaultmain; -* extern MCNUM mccomp_storein[]; -* extern MCNUM instrument.counter_AbsorbProp[]; -* extern MCNUM mcScattered; -* #define MCCODE_STRING "the McStas version" -* -* Usage: Automatically embbeded in the c code. -* -* $Id$ -* -*******************************************************************************/ - -#ifndef MCSTAS_R_H -#define MCSTAS_R_H "$Revision$" - -/* Following part is only embedded when not redundent with mcstas.h */ - -#ifndef MCCODE_H - -#define AA2MS 629.622368 /* Convert k[1/AA] to v[m/s] */ -#define MS2AA 1.58825361e-3 /* Convert v[m/s] to k[1/AA] */ -#define K2V AA2MS -#define V2K MS2AA -#define Q2V AA2MS -#define V2Q MS2AA -#define SE2V 437.393377 /* Convert sqrt(E)[meV] to v[m/s] */ -#define VS2E 5.22703725e-6 /* Convert (v[m/s])**2 to E[meV] */ - -#define SCATTER0 do {DEBUG_SCATTER(); SCATTERED++;} while(0) -#define SCATTER SCATTER0 - -#define JUMPTOCOMP(comp) mcneutron->_index = INDEX_COMP(comp); - -#define MAGNET_ON \ - do { \ - mcMagnet = 1; \ - } while(0) - -#define MAGNET_OFF \ - do { \ - mcMagnet = 0; \ - } while(0) - -#define ALLOW_BACKPROP \ - do { \ - mcallowbackprop = 1; \ - } while(0) - -#define DISALLOW_BACKPROP \ - do { \ - mcallowbackprop = 0; \ - } while(0) - -#define PROP_MAGNET(dt) \ - do { \ - } while (0) - /* change coordinates from local system to magnet system */ -/* Rotation rotLM, rotTemp; \ - Coords posLM = coords_sub(POS_A_CURRENT_COMP, mcMagnetPos); \ - rot_transpose(ROT_A_CURRENT_COMP, rotTemp); \ - rot_mul(rotTemp, mcMagnetRot, rotLM); \ - mcMagnetPrecession(x, y, z, t, vx, vy, vz, \ - &sx, &sy, &sz, dt, posLM, rotLM); \ - } while(0) -*/ - -#define mcPROP_DT(dt) \ - do { \ - if (mcMagnet && dt > 0) PROP_MAGNET(dt);\ - x += vx*(dt); \ - y += vy*(dt); \ - z += vz*(dt); \ - t += (dt); \ - if (isnan(p) || isinf(p)) { ABSORB; }\ - } while(0) - -/* ADD: E. Farhi, Aug 6th, 2001 PROP_GRAV_DT propagation with acceleration */ -#define PROP_GRAV_DT(dt, Ax, Ay, Az) \ - do { \ - if(dt < 0 && mcallowbackprop == 0) { ABSORB; }\ - if (mcMagnet) /*printf("Spin precession gravity\n")*/; \ - x += vx*(dt) + (Ax)*(dt)*(dt)/2; \ - y += vy*(dt) + (Ay)*(dt)*(dt)/2; \ - z += vz*(dt) + (Az)*(dt)*(dt)/2; \ - vx += (Ax)*(dt); \ - vy += (Ay)*(dt); \ - vz += (Az)*(dt); \ - t += (dt); \ - DISALLOW_BACKPROP;\ - } while(0) - - -#define PROP_DT(dt) \ - do { \ - if(dt < 0) { RESTORE=1; ABSORB; }; \ - if (mcgravitation) { Coords mcLocG; double mc_gx, mc_gy, mc_gz; \ - mcLocG = rot_apply(ROT_A_CURRENT_COMP, coords_set(0,-GRAVITY,0)); \ - coords_get(mcLocG, &mc_gx, &mc_gy, &mc_gz); \ - PROP_GRAV_DT(dt, mc_gx, mc_gy, mc_gz); } \ - else mcPROP_DT(dt); \ - DISALLOW_BACKPROP;\ - } while(0) - - -#define PROP_Z0 \ - do { \ - if (mcgravitation) { Coords mcLocG; int mc_ret; \ - double mc_dt, mc_gx, mc_gy, mc_gz; \ - mcLocG = rot_apply(ROT_A_CURRENT_COMP, coords_set(0,-GRAVITY,0)); \ - coords_get(mcLocG, &mc_gx, &mc_gy, &mc_gz); \ - mc_ret = solve_2nd_order(&mc_dt, NULL, -mc_gz/2, -vz, -z); \ - if (mc_ret) {PROP_GRAV_DT(mc_dt, mc_gx, mc_gy, mc_gz); z=0;}\ - else if (mcallowbackprop == 0 && mc_dt < 0) { ABSORB; }; } \ - else mcPROP_Z0; \ - DISALLOW_BACKPROP;\ - } while(0) - -#define mcPROP_Z0 \ - do { \ - double mc_dt; \ - if(vz == 0) { ABSORB; }; \ - mc_dt = -z/vz; \ - if(mc_dt < 0 && mcallowbackprop == 0) { ABSORB; }; \ - mcPROP_DT(mc_dt); \ - z = 0; \ - DISALLOW_BACKPROP;\ - } while(0) - -#define PROP_X0 \ - do { \ - if (mcgravitation) { Coords mcLocG; int mc_ret; \ - double mc_dt, mc_gx, mc_gy, mc_gz; \ - mcLocG = rot_apply(ROT_A_CURRENT_COMP, coords_set(0,-GRAVITY,0)); \ - coords_get(mcLocG, &mc_gx, &mc_gy, &mc_gz); \ - mc_ret = solve_2nd_order(&mc_dt, NULL, -mc_gx/2, -vx, -x); \ - if (mc_ret) {PROP_GRAV_DT(mc_dt, mc_gx, mc_gy, mc_gz); x=0;}\ - else if (mcallowbackprop == 0 && mc_dt < 0) { ABSORB; }; } \ - else mcPROP_X0; \ - DISALLOW_BACKPROP;\ - } while(0) - -#define mcPROP_X0 \ - do { \ - double mc_dt; \ - if(vx == 0) { ABSORB; }; \ - mc_dt = -x/vx; \ - if(mc_dt < 0 && mcallowbackprop == 0) { ABSORB; }; \ - mcPROP_DT(mc_dt); \ - x = 0; \ - DISALLOW_BACKPROP;\ - } while(0) - -#define PROP_Y0 \ - do { \ - if (mcgravitation) { Coords mcLocG; int mc_ret; \ - double mc_dt, mc_gx, mc_gy, mc_gz; \ - mcLocG = rot_apply(ROT_A_CURRENT_COMP, coords_set(0,-GRAVITY,0)); \ - coords_get(mcLocG, &mc_gx, &mc_gy, &mc_gz); \ - mc_ret = solve_2nd_order(&mc_dt, NULL, -mc_gy/2, -vy, -y); \ - if (mc_ret) {PROP_GRAV_DT(mc_dt, mc_gx, mc_gy, mc_gz); y=0;}\ - else if (mcallowbackprop == 0 && mc_dt < 0) { ABSORB; }; } \ - else mcPROP_Y0; \ - DISALLOW_BACKPROP;\ - } while(0) - - -#define mcPROP_Y0 \ - do { \ - double mc_dt; \ - if(vy == 0) { ABSORB; }; \ - mc_dt = -y/vy; \ - if(mc_dt < 0 && mcallowbackprop == 0) { ABSORB; }; \ - mcPROP_DT(mc_dt); \ - y = 0; \ - DISALLOW_BACKPROP; \ - } while(0) - - -#ifdef DEBUG - -#define DEBUG_STATE() if(!mcdotrace); else \ - printf("STATE: %g, %g, %g, %g, %g, %g, %g, %g, %g, %g, %g\n", \ - x,y,z,vx,vy,vz,t,sx,sy,sz,p); -#define DEBUG_SCATTER() if(!mcdotrace); else \ - printf("SCATTER: %g, %g, %g, %g, %g, %g, %g, %g, %g, %g, %g\n", \ - x,y,z,vx,vy,vz,t,sx,sy,sz,p); - -#else - -#define DEBUG_STATE() -#define DEBUG_SCATTER() - -#endif - -#endif /* !MCCODE_H */ - -#endif /* MCSTAS_R_H */ -/* end of file "/home/g/.cache/mccode-libc/main/nlib/mcstas-r.h" */ -/* embedding file "/home/g/.cache/mccode-libc/main/mccode-r.c" */ -/******************************************************************************* -* -* McCode, neutron/xray ray-tracing package -* Copyright (C) 1997-2009, All rights reserved -* Risoe National Laboratory, Roskilde, Denmark -* Institut Laue Langevin, Grenoble, France -* -* Runtime: share/mccode-r.c -* -* %Identification -* Written by: KN -* Date: Aug 29, 1997 -* Release: McStas X.Y/McXtrace X.Y -* Version: $Revision$ -* -* Runtime system for McStas and McXtrace. -* Embedded within instrument in runtime mode. -* Contains SECTIONS: -* MPI handling (sum, send, recv) -* format definitions -* I/O -* mcdisplay support -* random numbers -* coordinates handling -* vectors math (solve 2nd order, normals, randvec...) -* parameter handling -* signal and main handlers -* -* Usage: Automatically embbeded in the c code whenever required. -* -* $Id$ -* -*******************************************************************************/ - -/******************************************************************************* -* The I/O format definitions and functions -*******************************************************************************/ - - -/** Include header files to avoid implicit declarations (not allowed on LLVM) */ -#include -#include - -// UNIX specific headers (non-Windows) -#if defined(__unix__) || defined(__APPLE__) -#include -#include -#endif - - -#ifndef DANSE -#ifdef MC_ANCIENT_COMPATIBILITY -int traceenabled = 0; -int defaultmain = 0; -#endif -/* else defined directly in the McCode generated C code */ - -static long mcseed = 0; /* seed for random generator */ -#pragma acc declare create ( mcseed ) -static long mcstartdate = 0; /* start simulation time */ -static int mcdisable_output_files = 0; /* --no-output-files */ -mcstatic int mcgravitation = 0; /* use gravitation flag, for PROP macros */ -mcstatic int mcdotrace = 0; /* flag for --trace and messages for DISPLAY */ -#pragma acc declare create ( mcdotrace ) -int mcallowbackprop = 0; /* flag to enable negative/backprop */ - -/* OpenACC-related segmentation parameters: */ -int vecsize = 128; -int numgangs = 7813; -long gpu_innerloop = 2147483647; - -/* Monitor_nD list/buffer-size default */ -long MONND_BUFSIZ = 1000000; - -/* Number of particle histories to simulate. */ -#ifdef NEUTRONICS -mcstatic unsigned long long int mcncount = 1; -mcstatic unsigned long long int mcrun_num = 0; -#else -#ifdef MCDEFAULT_NCOUNT -mcstatic unsigned long long int mcncount = MCDEFAULT_NCOUNT; -#else -mcstatic unsigned long long int mcncount = 1000000; -#endif -#pragma acc declare create ( mcncount ) -mcstatic unsigned long long int mcrun_num = 0; -#pragma acc declare create ( mcrun_num ) -#endif /* NEUTRONICS */ - -#else -#include "mcstas-globals.h" -#endif /* !DANSE */ - -#ifndef NX_COMPRESSION -#define NX_COMPRESSION NX_COMP_NONE -#endif - -/* String nullification on GPU and other replacements */ -#ifdef OPENACC -int noprintf() { - return 0; -} - -int str_comp(char *str1, char *str2) { - while (*str1 && *str1 == *str2) { - str1++; - str2++; - } - return (*str1 - *str2); -} - -size_t str_len(const char *s) -{ - size_t len = 0; - if(s != NULL) - { - while(*s != '\0') - { - ++len; - ++s; - } - } - return len; -} - -#endif - -/* SECTION: Predefine (component) parameters ================================= */ - -int nans_match(double a, double b){ - return (*(uint64_t*)&a == *(uint64_t*)&b); -} -int is_unset(double x){ - return nans_match(x, UNSET); -} -int is_set(double x){ - return !nans_match(x, UNSET); -} -int is_valid(double x){ - return !isnan(x)||is_unset(x); -} -int all_unset(int n, ...){ - va_list ptr; - va_start(ptr, n); - int ret=1; - for (int i=0; i count-1) length=count-offset; - else length=MPI_REDUCE_BLOCKSIZE; - if (MPI_Allreduce((double*)(sbuf+offset), (double*)(rbuf+offset), - length, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD) != MPI_SUCCESS) - return MPI_ERR_COUNT; - offset += length; - } - - for (i=0; i count-1) length=count-offset; - else length=MPI_REDUCE_BLOCKSIZE; - if (MPI_Send((void*)(sbuf+offset*dsize), length, dtype, dest, tag++, MPI_COMM_WORLD) != MPI_SUCCESS) - return MPI_ERR_COUNT; - offset += length; - } - - return MPI_SUCCESS; -} /* mc_MPI_Send */ - -/******************************************************************************* -* mc_MPI_Recv: Receives arrays from MPI nodes by blocks to avoid buffer limit -* the buffer must have been allocated previously. -*******************************************************************************/ -int mc_MPI_Recv(void *sbuf, - long count, MPI_Datatype dtype, - int source) -{ - int dsize; - long offset=0; - int tag=1; - int length=MPI_REDUCE_BLOCKSIZE; /* defined in mccode-r.h */ - - if (!sbuf || count <= 0) return(MPI_SUCCESS); /* nothing to recv */ - MPI_Type_size(dtype, &dsize); - - while (offset < count) { - if (offset+length > count-1) length=count-offset; - else length=MPI_REDUCE_BLOCKSIZE; - if (MPI_Recv((void*)(sbuf+offset*dsize), length, dtype, source, tag++, - MPI_COMM_WORLD, MPI_STATUS_IGNORE) != MPI_SUCCESS) - return MPI_ERR_COUNT; - offset += length; - } - - return MPI_SUCCESS; -} /* mc_MPI_Recv */ - -#endif /* USE_MPI */ - -/* SECTION: parameters handling ============================================= */ - -/* Instrument input parameter type handling. */ -/******************************************************************************* -* mcparm_double: extract double value from 's' into 'vptr' -*******************************************************************************/ -static int -mcparm_double(char *s, void *vptr) -{ - char *p; - double *v = (double *)vptr; - - if (!s) { *v = 0; return(1); } - *v = strtod(s, &p); - if(*s == '\0' || (p != NULL && *p != '\0') || errno == ERANGE) - return 0; /* Failed */ - else - return 1; /* Success */ -} - -/******************************************************************************* -* mcparminfo_double: display parameter type double -*******************************************************************************/ -static char * -mcparminfo_double(char *parmname) -{ - return "double"; -} - -/******************************************************************************* -* mcparmerror_double: display error message when failed extract double -*******************************************************************************/ -static void -mcparmerror_double(char *parm, char *val) -{ - fprintf(stderr, "Error: Invalid value '%s' for floating point parameter %s (mcparmerror_double)\n", - val, parm); -} - -/******************************************************************************* -* mcparmprinter_double: convert double to string -*******************************************************************************/ -static void -mcparmprinter_double(char *f, void *vptr) -{ - double *v = (double *)vptr; - sprintf(f, "%g", *v); -} - -/******************************************************************************* -* mcparm_int: extract int value from 's' into 'vptr' -*******************************************************************************/ -static int -mcparm_int(char *s, void *vptr) -{ - char *p; - int *v = (int *)vptr; - long x; - - if (!s) { *v = 0; return(1); } - *v = 0; - x = strtol(s, &p, 10); - if(x < INT_MIN || x > INT_MAX) - return 0; /* Under/overflow */ - *v = x; - if(*s == '\0' || (p != NULL && *p != '\0') || errno == ERANGE) - return 0; /* Failed */ - else - return 1; /* Success */ -} - -/******************************************************************************* -* mcparminfo_int: display parameter type int -*******************************************************************************/ -static char * -mcparminfo_int(char *parmname) -{ - return "int"; -} - -/******************************************************************************* -* mcparmerror_int: display error message when failed extract int -*******************************************************************************/ -static void -mcparmerror_int(char *parm, char *val) -{ - fprintf(stderr, "Error: Invalid value '%s' for integer parameter %s (mcparmerror_int)\n", - val, parm); -} - -/******************************************************************************* -* mcparmprinter_int: convert int to string -*******************************************************************************/ -static void -mcparmprinter_int(char *f, void *vptr) -{ - int *v = (int *)vptr; - sprintf(f, "%d", *v); -} - -/******************************************************************************* -* mcparm_string: extract char* value from 's' into 'vptr' (copy) -*******************************************************************************/ -static int -mcparm_string(char *s, void *vptr) -{ - char **v = (char **)vptr; - if (!s) { *v = NULL; return(1); } - *v = (char *)malloc(strlen(s) + 1); - if(*v == NULL) - { - exit(-fprintf(stderr, "Error: Out of memory %li (mcparm_string).\n", (long)strlen(s) + 1)); - } - strcpy(*v, s); - return 1; /* Success */ -} - -/******************************************************************************* -* mcparminfo_string: display parameter type string -*******************************************************************************/ -static char * -mcparminfo_string(char *parmname) -{ - return "string"; -} - -/******************************************************************************* -* mcparmerror_string: display error message when failed extract string -*******************************************************************************/ -static void -mcparmerror_string(char *parm, char *val) -{ - fprintf(stderr, "Error: Invalid value '%s' for string parameter %s (mcparmerror_string)\n", - val, parm); -} - -/******************************************************************************* -* mcparmprinter_string: convert string to string (including esc chars) -*******************************************************************************/ -static void -mcparmprinter_string(char *f, void *vptr) -{ - char **v = (char **)vptr; - char *p; - - if (!*v) { *f='\0'; return; } - strcpy(f, ""); - for(p = *v; *p != '\0'; p++) - { - switch(*p) - { - case '\n': - strcat(f, "\\n"); - break; - case '\r': - strcat(f, "\\r"); - break; - case '"': - strcat(f, "\\\""); - break; - case '\\': - strcat(f, "\\\\"); - break; - default: - strncat(f, p, 1); - } - } - /* strcat(f, "\""); */ -} /* mcparmprinter_string */ - -/* now we may define the parameter structure, using previous functions */ -static struct - { - int (*getparm)(char *, void *); - char * (*parminfo)(char *); - void (*error)(char *, char *); - void (*printer)(char *, void *); -} mcinputtypes[] = { - { - mcparm_int, mcparminfo_int, mcparmerror_int, - mcparmprinter_int - }, { - mcparm_string, mcparminfo_string, mcparmerror_string, - mcparmprinter_string - }, { - mcparm_string, mcparminfo_string, mcparmerror_string, - mcparmprinter_string - }, { - mcparm_double, mcparminfo_double, mcparmerror_double, - mcparmprinter_double - }, { - mcparm_double, mcparminfo_double, mcparmerror_double, - mcparmprinter_double - } -}; - -/******************************************************************************* -* mcestimate_error: compute sigma from N,p,p2 in Gaussian large numbers approx -*******************************************************************************/ -double mcestimate_error(double N, double p1, double p2) -{ - double pmean, n1; - if(N <= 1) - return p1; - pmean = p1 / N; - n1 = N - 1; - /* Note: underflow may cause p2 to become zero; the fabs() below guards - against this. */ - return sqrt((N/n1)*fabs(p2 - pmean*pmean)); -} - -double (*mcestimate_error_p) - (double V2, double psum, double p2sum)=mcestimate_error; - -/* ========================================================================== */ - -/* MCCODE_R_IO_C */ - -/* ========================================================================== */ - -#ifndef MCCODE_R_IO_C -#define MCCODE_R_IO_C "$Revision$" - -/* SECTION: file i/o handling ================================================ */ - -#ifndef HAVE_STRCASESTR -// from msysgit: https://code.google.com/p/msysgit/source/browse/compat/strcasestr.c -char *strcasestr(const char *haystack, const char *needle) -{ - int nlen = strlen(needle); - int hlen = strlen(haystack) - nlen + 1; - int i; - - for (i = 0; i < hlen; i++) { - int j; - for (j = 0; j < nlen; j++) { - unsigned char c1 = haystack[i+j]; - unsigned char c2 = needle[j]; - if (toupper(c1) != toupper(c2)) - goto next; - } - return (char *) haystack + i; - next: - ; - } - return NULL; -} - - -#endif -#ifndef HAVE_STRCASECMP -int strcasecmp( const char *s1, const char *s2 ) -{ - int c1, c2; - do { - c1 = tolower( (unsigned char) *s1++ ); - c2 = tolower( (unsigned char) *s2++ ); - } while (c1 == c2 && c1 != 0); - return c2 > c1 ? -1 : c1 > c2; -} -#endif - -#ifndef STRACPY -/* this is a replacement to strncpy, but ensures that the copy ends with NULL */ -/* http://stracpy.blogspot.fr/2011/04/stracpy-strncpy-replacement.html */ -#define STRACPY -char *stracpy(char *destination, const char *source, size_t amount) -{ - if (!destination || !source || !amount) return(NULL); - while(amount--) - if((*destination++ = *source++) == '\0') break; - *destination = '\0'; - return destination; -} -#endif - -/******************************************************************************* -* mcfull_file: allocates a full file name=dirname+file. Catenate extension if missing. -*******************************************************************************/ -char *mcfull_file(char *name, char *ext) -{ - int dirlen=0; - char *mem =NULL; - - dirlen = dirname ? strlen(dirname) : 0; - mem = (char*)malloc(dirlen + strlen(name) + CHAR_BUF_LENGTH); - if(!mem) { - exit(-fprintf(stderr, "Error: Out of memory %li (mcfull_file)\n", (long)(dirlen + strlen(name) + 256))); - } - strcpy(mem, ""); - - /* prepend directory name to path if name does not contain a path */ - if (dirlen > 0 && !strchr(name, MC_PATHSEP_C)) { - strcat(mem, dirname); - strcat(mem, MC_PATHSEP_S); - } /* dirlen */ - - strcat(mem, name); - if (!strchr(name, '.') && ext && strlen(ext)) - { /* add extension if not in file name already */ - strcat(mem, "."); - strcat(mem, ext); - } - return(mem); -} /* mcfull_file */ - -/******************************************************************************* -* mcnew_file: opens a new file within dirname if non NULL -* the file is opened in "a" (append, create if does not exist) -* the extension 'ext' is added if the file name does not include one. -* the last argument is set to 0 if file did not exist, else to 1. -*******************************************************************************/ -FILE *mcnew_file(char *name, char *ext, int *exists) -{ - char *mem; - FILE *file=NULL; - - if (!name || strlen(name) == 0 || mcdisable_output_files) return(NULL); - - mem = mcfull_file(name, ext); /* create dirname/name.ext */ - - /* check for existence */ - file = fopen(mem, "r"); /* for reading -> fails if does not exist */ - if (file) { - fclose(file); - *exists=1; - } else - *exists=0; - - /* open the file for writing/appending */ -#ifdef USE_NEXUS - if (mcformat && strcasestr(mcformat, "NeXus")) { - /* NXhandle nxhandle is defined in the .h with USE_NEXUS */ - NXaccess mode = (*exists ? NXACC_CREATE5 | NXACC_RDWR : NXACC_CREATE5); - - if (NXopen(mem, mode, &nxhandle) != NX_OK) - file = NULL; - else - file = (FILE*)&nxhandle; /* to make it non NULL */ - } else -#endif - file = fopen(mem, "a+"); - - if(!file) - fprintf(stderr, "Warning: could not open output file '%s' for %s (mcnew_file)\n", - mem, *exists ? "append" : "create"); - free(mem); - - return file; -} /* mcnew_file */ - -/******************************************************************************* -* mcdetector_statistics: compute detector statistics, error bars, [x I I_err N] 1D -* RETURN: updated detector structure -* Used by: detector_import -*******************************************************************************/ -MCDETECTOR mcdetector_statistics( - MCDETECTOR detector) -{ - - if (!detector.p1 || !detector.m) - return(detector); - - /* compute statistics and update MCDETECTOR structure ===================== */ - double sum_z = 0, min_z = 0, max_z = 0; - double fmon_x =0, smon_x = 0, fmon_y =0, smon_y=0, mean_z=0; - double Nsum=0, P2sum=0; - - double sum_xz = 0, sum_yz = 0, sum_x = 0, sum_y = 0, sum_x2z = 0, sum_y2z = 0; - int i,j; - char hasnan=0, hasinf=0; - char israw = ((char*)strcasestr(detector.format,"raw") != NULL); - double *this_p1=NULL; /* new 1D McCode array [x I E N]. Freed after writing data */ - - /* if McCode/PGPLOT and rank==1 we create a new m*4 data block=[x I E N] */ - if (detector.rank == 1 && strcasestr(detector.format,"McCode")) { - this_p1 = (double *)calloc(detector.m*detector.n*detector.p*4, sizeof(double)); - if (!this_p1) - exit(-fprintf(stderr, "Error: Out of memory creating %li 1D " MCCODE_STRING " data set for file '%s' (detector_import)\n", - detector.m*detector.n*detector.p*4*sizeof(double*), detector.filename)); - } - - max_z = min_z = detector.p1[0]; - - /* compute sum and moments (not for lists) */ - if (!strcasestr(detector.format,"list") && detector.m) - for(j = 0; j < detector.n*detector.p; j++) - { - for(i = 0; i < detector.m; i++) - { - double x,y,z; - double N, E; - long index= !detector.istransposed ? i*detector.n*detector.p + j : i+j*detector.m; - char hasnaninf=0; - - if (detector.m) - x = detector.xmin + (i + 0.5)/detector.m*(detector.xmax - detector.xmin); - else x = 0; - if (detector.n && detector.p) - y = detector.ymin + (j + 0.5)/detector.n/detector.p*(detector.ymax - detector.ymin); - else y = 0; - z = detector.p1[index]; - N = detector.p0 ? detector.p0[index] : 1; - E = detector.p2 ? detector.p2[index] : 0; - if (detector.p2 && !israw) - detector.p2[index] = (*mcestimate_error_p)(detector.p0[index],detector.p1[index],detector.p2[index]); /* set sigma */ - - if (detector.rank == 1 && this_p1 && strcasestr(detector.format,"McCode")) { - /* fill-in 1D McCode array [x I E N] */ - this_p1[index*4] = x; - this_p1[index*4+1] = z; - this_p1[index*4+2] = detector.p2 ? detector.p2[index] : 0; - this_p1[index*4+3] = N; - } - - if (isnan(z) || isnan(E) || isnan(N)) hasnaninf=hasnan=1; - if (isinf(z) || isinf(E) || isinf(N)) hasnaninf=hasinf=1; - - /* compute stats integrals */ - if (!hasnaninf) { - sum_xz += x*z; - sum_yz += y*z; - sum_x += x; - sum_y += y; - sum_z += z; - sum_x2z += x*x*z; - sum_y2z += y*y*z; - if (z > max_z) max_z = z; - if (z < min_z) min_z = z; - - Nsum += N; - P2sum += E; - } - - } - } /* for j */ - - /* compute 1st and 2nd moments. For lists, sum_z=0 so this is skipped. */ - if (sum_z && detector.n*detector.m*detector.p) - { - fmon_x = sum_xz/sum_z; - fmon_y = sum_yz/sum_z; - smon_x = sum_x2z/sum_z-fmon_x*fmon_x; smon_x = smon_x > 0 ? sqrt(smon_x) : 0; - smon_y = sum_y2z/sum_z-fmon_y*fmon_y; smon_y = smon_y > 0 ? sqrt(smon_y) : 0; - mean_z = sum_z/detector.n/detector.m/detector.p; - } - /* store statistics into detector */ - detector.intensity = sum_z; - detector.error = Nsum ? (*mcestimate_error_p)(Nsum, sum_z, P2sum) : 0; - detector.events = Nsum; - detector.min = min_z; - detector.max = max_z; - detector.mean = mean_z; - detector.centerX = fmon_x; - detector.halfwidthX= smon_x; - detector.centerY = fmon_y; - detector.halfwidthY= smon_y; - - /* if McCode/PGPLOT and rank==1 replace p1 with new m*4 1D McCode and clear others */ - if (detector.rank == 1 && this_p1 && strcasestr(detector.format,"McCode")) { - - detector.p1 = this_p1; - detector.n = detector.m; detector.m = 4; - detector.p0 = detector.p2 = NULL; - detector.istransposed = 1; - } - - if (detector.n*detector.m*detector.p > 1) - snprintf(detector.signal, CHAR_BUF_LENGTH, - "Min=%g; Max=%g; Mean=%g;", detector.min, detector.max, detector.mean); - else - strcpy(detector.signal, "None"); - snprintf(detector.values, CHAR_BUF_LENGTH, - "%g %g %g", detector.intensity, detector.error, detector.events); - - switch (detector.rank) { - case 1: snprintf(detector.statistics, CHAR_BUF_LENGTH, "X0=%g; dX=%g;", - detector.centerX, detector.halfwidthX); break; - case 2: - case 3: snprintf(detector.statistics, CHAR_BUF_LENGTH, "X0=%g; dX=%g; Y0=%g; dY=%g;", - detector.centerX, detector.halfwidthX, detector.centerY, detector.halfwidthY); - break; - default: strcpy(detector.statistics, "None"); - } - - if (hasnan) - printf("WARNING: Nan detected in component/file %s %s\n", - detector.component, strlen(detector.filename) ? detector.filename : ""); - if (hasinf) - printf("WARNING: Inf detected in component/file %s %s\n", - detector.component, strlen(detector.filename) ? detector.filename : ""); - - return(detector); - -} /* mcdetector_statistics */ - -/******************************************************************************* -* detector_import: build detector structure, merge non-lists from MPI -* compute basic stat, write "Detector:" line -* RETURN: detector structure. Invalid data if detector.p1 == NULL -* Invalid detector sets m=0 and filename="" -* Simulation data sets m=0 and filename=siminfo_name -* This function is equivalent to the old 'mcdetector_out', returning a structure -*******************************************************************************/ -MCDETECTOR detector_import( - char *format, - char *component, char *title, - long m, long n, long p, - char *xlabel, char *ylabel, char *zlabel, - char *xvar, char *yvar, char *zvar, - double x1, double x2, double y1, double y2, double z1, double z2, - char *filename, - double *p0, double *p1, double *p2, - Coords position) -{ - time_t t; /* for detector.date */ - long date_l; /* date as a long number */ - char istransposed=0; - char c[CHAR_BUF_LENGTH]; /* temp var for signal label */ - - MCDETECTOR detector; - - /* build MCDETECTOR structure ============================================= */ - /* make sure we do not have NULL for char fields */ - - /* these also apply to simfile */ - strncpy (detector.filename, filename ? filename : "", CHAR_BUF_LENGTH); - strncpy (detector.format, format ? format : "McCode" , CHAR_BUF_LENGTH); - /* add extension if missing */ - if (strlen(detector.filename) && !strchr(detector.filename, '.')) - { /* add extension if not in file name already */ - strcat(detector.filename, ".dat"); - } - strncpy (detector.component, component ? component : MCCODE_STRING " component", CHAR_BUF_LENGTH); - - snprintf(detector.instrument, CHAR_BUF_LENGTH, "%s (%s)", instrument_name, instrument_source); - snprintf(detector.user, CHAR_BUF_LENGTH, "%s on %s", - getenv("USER") ? getenv("USER") : MCCODE_NAME, - getenv("HOST") ? getenv("HOST") : "localhost"); - time(&t); /* get current write time */ - date_l = (long)t; /* same but as a long */ - snprintf(detector.date, CHAR_BUF_LENGTH, "%s", ctime(&t)); - if (strlen(detector.date)) detector.date[strlen(detector.date)-1] = '\0'; /* remove last \n in date */ - detector.date_l = date_l; - - if (!mcget_run_num() || mcget_run_num() >= mcget_ncount()) - snprintf(detector.ncount, CHAR_BUF_LENGTH, "%llu", mcget_ncount() -#ifdef USE_MPI -*mpi_node_count -#endif - ); - else - snprintf(detector.ncount, CHAR_BUF_LENGTH, "%g/%g", (double)mcget_run_num(), (double)mcget_ncount()); - - detector.p0 = p0; - detector.p1 = p1; - detector.p2 = p2; - - /* handle transposition (not for NeXus) */ - if (!strcasestr(detector.format, "NeXus")) { - if (m<0 || n<0 || p<0) istransposed = !istransposed; - if (strcasestr(detector.format, "transpose")) istransposed = !istransposed; - if (istransposed) { /* do the swap once for all */ - long i=m; m=n; n=i; - } - } - - m=labs(m); n=labs(n); p=labs(p); /* make sure dimensions are positive */ - detector.istransposed = istransposed; - - /* determine detector rank (dimensionality) */ - if (!m || !n || !p || !p1) detector.rank = 4; /* invalid: exit with m=0 filename="" */ - else if (m*n*p == 1) detector.rank = 0; /* 0D */ - else if (n == 1 || m == 1) detector.rank = 1; /* 1D */ - else if (p == 1) detector.rank = 2; /* 2D */ - else detector.rank = 3; /* 3D */ - - /* from rank, set type */ - switch (detector.rank) { - case 0: strcpy(detector.type, "array_0d"); m=n=p=1; break; - case 1: snprintf(detector.type, CHAR_BUF_LENGTH, "array_1d(%ld)", m*n*p); m *= n*p; n=p=1; break; - case 2: snprintf(detector.type, CHAR_BUF_LENGTH, "array_2d(%ld, %ld)", m, n*p); n *= p; p=1; break; - case 3: snprintf(detector.type, CHAR_BUF_LENGTH, "array_3d(%ld, %ld, %ld)", m, n, p); break; - default: m=0; strcpy(detector.type, ""); strcpy(detector.filename, "");/* invalid */ - } - - detector.m = m; - detector.n = n; - detector.p = p; - - /* these only apply to detector files ===================================== */ - - snprintf(detector.position, CHAR_BUF_LENGTH, "%g %g %g", position.x, position.y, position.z); - /* may also store actual detector orientation in the future */ - - strncpy(detector.title, title && strlen(title) ? title : component, CHAR_BUF_LENGTH); - strncpy(detector.xlabel, xlabel && strlen(xlabel) ? xlabel : "X", CHAR_BUF_LENGTH); /* axis labels */ - strncpy(detector.ylabel, ylabel && strlen(ylabel) ? ylabel : "Y", CHAR_BUF_LENGTH); - strncpy(detector.zlabel, zlabel && strlen(zlabel) ? zlabel : "Z", CHAR_BUF_LENGTH); - strncpy(detector.xvar, xvar && strlen(xvar) ? xvar : "x", CHAR_BUF_LENGTH); /* axis variables */ - strncpy(detector.yvar, yvar && strlen(yvar) ? yvar : detector.xvar, CHAR_BUF_LENGTH); - strncpy(detector.zvar, zvar && strlen(zvar) ? zvar : detector.yvar, CHAR_BUF_LENGTH); - - /* set "variables" as e.g. "I I_err N" */ - strcpy(c, "I "); - if (strlen(detector.zvar)) strncpy(c, detector.zvar,32); - else if (strlen(detector.yvar)) strncpy(c, detector.yvar,32); - else if (strlen(detector.xvar)) strncpy(c, detector.xvar,32); - - if (detector.rank == 1) - snprintf(detector.variables, CHAR_BUF_LENGTH, "%s %s %s_err N", detector.xvar, c, c); - else - snprintf(detector.variables, CHAR_BUF_LENGTH, "%s %s_err N", c, c); - - /* limits */ - detector.xmin = x1; - detector.xmax = x2; - detector.ymin = y1; - detector.ymax = y2; - detector.zmin = z1; - detector.zmax = z2; - if (abs(detector.rank) == 1) - snprintf(detector.limits, CHAR_BUF_LENGTH, "%g %g", x1, x2); - else if (detector.rank == 2) - snprintf(detector.limits, CHAR_BUF_LENGTH, "%g %g %g %g", x1, x2, y1, y2); - else - snprintf(detector.limits, CHAR_BUF_LENGTH, "%g %g %g %g %g %g", x1, x2, y1, y2, z1, z2); - - /* if MPI and nodes_nb > 1: reduce data sets when using MPI =============== */ -#ifdef USE_MPI - if (!strcasestr(detector.format,"list") && mpi_node_count > 1 && m) { - /* we save additive data: reduce everything into mpi_node_root */ - if (p0) mc_MPI_Sum(p0, m*n*p); - if (p1) mc_MPI_Sum(p1, m*n*p); - if (p2) mc_MPI_Sum(p2, m*n*p); - if (!p0) { /* additive signal must be then divided by the number of nodes */ - int i; - for (i=0; i CHAR_BUF_LENGTH) break; - snprintf(ThisParam, CHAR_BUF_LENGTH, " %s(%s)", mcinputtable[i].name, - (*mcinputtypes[mcinputtable[i].type].parminfo) - (mcinputtable[i].name)); - if (strlen(Parameters) + strlen(ThisParam) + 1 >= CHAR_BUF_LENGTH) break; - strcat(Parameters, ThisParam); - } - - /* output data ============================================================ */ - if (f != stdout) - fprintf(f, "%sFile: %s%c%s\n", pre, dirname, MC_PATHSEP_C, siminfo_name); - else - fprintf(f, "%sCreator: %s\n", pre, MCCODE_STRING); - - fprintf(f, "%sSource: %s\n", pre, instrument_source); - fprintf(f, "%sParameters: %s\n", pre, Parameters); - - fprintf(f, "%sTrace_enabled: %s\n", pre, traceenabled ? "yes" : "no"); - fprintf(f, "%sDefault_main: %s\n", pre, defaultmain ? "yes" : "no"); - fprintf(f, "%sEmbedded_runtime: %s\n", pre, -#ifdef MC_EMBEDDED_RUNTIME - "yes" -#else - "no" -#endif - ); - - fflush(f); -} /* mcinfo_out */ - -/******************************************************************************* -* mcruninfo_out: output simulation tags/info (both in SIM and data files) -* Used in: siminfo_init (ascii case), mcdetector_out_xD_ascii -*******************************************************************************/ -static void mcruninfo_out(char *pre, FILE *f) -{ - int i; - char Parameters[CHAR_BUF_LENGTH]; - - if (!f || mcdisable_output_files) return; - - fprintf(f, "%sFormat: %s%s\n", pre, - mcformat && strlen(mcformat) ? mcformat : MCCODE_NAME, - mcformat && strcasestr(mcformat,"McCode") ? " with text headers" : ""); - fprintf(f, "%sURL: %s\n", pre, "http://www.mccode.org"); - fprintf(f, "%sCreator: %s\n", pre, MCCODE_STRING); - fprintf(f, "%sInstrument: %s\n", pre, instrument_source); - fprintf(f, "%sNcount: %llu\n", pre, mcget_ncount()); - fprintf(f, "%sTrace: %s\n", pre, mcdotrace ? "yes" : "no"); - fprintf(f, "%sGravitation: %s\n", pre, mcgravitation ? "yes" : "no"); - snprintf(Parameters, CHAR_BUF_LENGTH, "%ld", mcseed); - fprintf(f, "%sSeed: %s\n", pre, Parameters); - fprintf(f, "%sDirectory: %s\n", pre, dirname ? dirname : "."); -#ifdef USE_MPI - if (mpi_node_count > 1) - fprintf(f, "%sNodes: %i\n", pre, mpi_node_count); -#endif - - // TODO Consider replacing this by a a call to `mcparameterinfo_out(pre+"Param: ", f)` - /* output parameter string ================================================ */ - for(i = 0; i < numipar; i++) { - if (mcinputtable[i].par){ - /* Parameters with a default value */ - if(mcinputtable[i].val && strlen(mcinputtable[i].val)){ - (*mcinputtypes[mcinputtable[i].type].printer)(Parameters, mcinputtable[i].par); - fprintf(f, "%sParam: %s=%s\n", pre, mcinputtable[i].name, Parameters); - /* ... and those without */ - }else{ - fprintf(f, "%sParam: %s=NULL\n", pre, mcinputtable[i].name); - } - } - } - fflush(f); -} /* mcruninfo_out */ - -/******************************************************************************* - * @brief Print parameter information to the specified file - * @param pre any beginning-of-line padding - * @param f the output file - */ -static void mcparameterinfo_out(char * pre, FILE *f){ - if (!f || mcdisable_output_files) return; - - unsigned int nchar = 4; - for (int i=0; i < numipar; ++i){ - if (mcinputtable[i].par && mcinputtable[i].val && strlen(mcinputtable[i].val) > nchar) - nchar = strlen(mcinputtable[i].val); - } - char * buffer = calloc(nchar+1, sizeof(char)); - - if (!buffer) { - exit(1); - } - - for (int i=0; i < numipar; ++i) { - if (mcinputtable[i].par) { - char * name = mcinputtable[i].name; - if (mcinputtable[i].val && strlen(mcinputtable[i].val)) { - mcinputtypes[mcinputtable[i].type].printer(buffer, mcinputtable[i].par); - } else { - strcpy(buffer, "NULL"); - } - if (strlen(mcinputtable[i].unit)){ - //fprintf(f, "%s%s %s (\"%s\") = %s\n", pre, mcinputtypes[mcinputtable[i].type].parminfo(name), name, mcinputtable[i].unit, buffer); - fprintf(f, "%s%s %s/\"%s\" = %s\n", pre, mcinputtypes[mcinputtable[i].type].parminfo(name), name, mcinputtable[i].unit, buffer); - } else { - fprintf(f, "%s%s %s = %s\n", pre, mcinputtypes[mcinputtable[i].type].parminfo(name), name, buffer); - } - } - } - - free(buffer); -} - -/******************************************************************************* -* siminfo_out: wrapper to fprintf(siminfo_file) -*******************************************************************************/ -void siminfo_out(char *format, ...) -{ - va_list ap; - - if(siminfo_file && !mcdisable_output_files) - { - va_start(ap, format); - vfprintf(siminfo_file, format, ap); - va_end(ap); - } -} /* siminfo_out */ - - -/******************************************************************************* -* mcdatainfo_out: output detector header -* mcdatainfo_out(prefix, file_handle, detector) writes info to data file -*******************************************************************************/ -static void -mcdatainfo_out(char *pre, FILE *f, MCDETECTOR detector) -{ - if (!f || !detector.m || mcdisable_output_files) return; - - /* output data ============================================================ */ - fprintf(f, "%sDate: %s (%li)\n", pre, detector.date, detector.date_l); - fprintf(f, "%stype: %s\n", pre, detector.type); - fprintf(f, "%sSource: %s\n", pre, detector.instrument); - fprintf(f, "%scomponent: %s\n", pre, detector.component); - fprintf(f, "%sposition: %s\n", pre, detector.position); - - fprintf(f, "%stitle: %s\n", pre, detector.title); - fprintf(f, !mcget_run_num() || mcget_run_num() >= mcget_ncount() ? - "%sNcount: %s\n" : - "%sratio: %s\n", pre, detector.ncount); - - if (strlen(detector.filename)) { - fprintf(f, "%sfilename: %s\n", pre, detector.filename); - } - - fprintf(f, "%sstatistics: %s\n", pre, detector.statistics); - fprintf(f, "%ssignal: %s\n", pre, detector.signal); - fprintf(f, "%svalues: %s\n", pre, detector.values); - - if (detector.rank >= 1) - { - fprintf(f, "%sxvar: %s\n", pre, detector.xvar); - fprintf(f, "%syvar: %s\n", pre, detector.yvar); - fprintf(f, "%sxlabel: %s\n", pre, detector.xlabel); - fprintf(f, "%sylabel: %s\n", pre, detector.ylabel); - if (detector.rank > 1) { - fprintf(f, "%szvar: %s\n", pre, detector.zvar); - fprintf(f, "%szlabel: %s\n", pre, detector.zlabel); - } - } - - fprintf(f, - abs(detector.rank)==1 ? - "%sxlimits: %s\n" : - "%sxylimits: %s\n", pre, detector.limits); - fprintf(f, "%svariables: %s\n", pre, - strcasestr(detector.format, "list") ? detector.ylabel : detector.variables); - - fflush(f); - -} /* mcdatainfo_out */ - -/* mcdetector_out_array_ascii: output a single array to a file - * m: columns - * n: rows - * p: array - * f: file handle (already opened) - */ -static void mcdetector_out_array_ascii(long m, long n, double *p, FILE *f, char istransposed) -{ - if(f) - { - int i,j; - for(j = 0; j < n; j++) - { - for(i = 0; i < m; i++) - { - fprintf(f, "%.10g ", p[!istransposed ? i*n + j : j*m+i]); - } - fprintf(f,"\n"); - } - } -} /* mcdetector_out_array_ascii */ - -/******************************************************************************* -* mcdetector_out_0D_ascii: called by mcdetector_out_0D for ascii output -*******************************************************************************/ -MCDETECTOR mcdetector_out_0D_ascii(MCDETECTOR detector) -{ - int exists=0; - FILE *outfile = NULL; - - /* Write data set information to simulation description file. */ - MPI_MASTER( - siminfo_out("\nbegin data\n"); // detector.component - mcdatainfo_out(" ", siminfo_file, detector); - siminfo_out("end data\n"); - /* Don't write if filename is NULL: mcnew_file handles this (return NULL) */ - outfile = mcnew_file(detector.component, "dat", &exists); - if(outfile) - { - /* write data file header and entry in simulation description file */ - mcruninfo_out( "# ", outfile); - mcdatainfo_out("# ", outfile, detector); - /* write I I_err N */ - fprintf(outfile, "%g %g %g\n", - detector.intensity, detector.error, detector.events); - fclose(outfile); - } - ); /* MPI_MASTER */ - return(detector); -} /* mcdetector_out_0D_ascii */ - -/******************************************************************************* -* mcdetector_out_1D_ascii: called by mcdetector_out_1D for ascii output -*******************************************************************************/ -MCDETECTOR mcdetector_out_1D_ascii(MCDETECTOR detector) -{ - int exists=0; - FILE *outfile = NULL; - - MPI_MASTER( - /* Write data set information to simulation description file. */ - siminfo_out("\nbegin data\n"); // detector.filename - mcdatainfo_out(" ", siminfo_file, detector); - siminfo_out("end data\n"); - /* Loop over array elements, writing to file. */ - /* Don't write if filename is NULL: mcnew_file handles this (return NULL) */ - outfile = mcnew_file(detector.filename, "dat", &exists); - if(outfile) - { - /* write data file header and entry in simulation description file */ - mcruninfo_out( "# ", outfile); - mcdatainfo_out("# ", outfile, detector); - /* output the 1D array columns */ - mcdetector_out_array_ascii(detector.m, detector.n, detector.p1, outfile, detector.istransposed); - - fclose(outfile); - } - ); /* MPI_MASTER */ - return(detector); - -} /* mcdetector_out_1D_ascii */ - -/******************************************************************************* -* mcdetector_out_2D_ascii: called by mcdetector_out_2D for ascii output -*******************************************************************************/ -MCDETECTOR mcdetector_out_2D_ascii(MCDETECTOR detector) -{ - int exists=0; - FILE *outfile = NULL; - - MPI_MASTER( - /* Loop over array elements, writing to file. */ - /* Don't write if filename is NULL: mcnew_file handles this (return NULL) */ - outfile = mcnew_file(detector.filename, "dat", &exists); - if(outfile) - { - /* write header only if file has just been created (not appending) */ - if (!exists) { - /* Write data set information to simulation description file. */ - siminfo_out("\nbegin data\n"); // detector.filename - mcdatainfo_out(" ", siminfo_file, detector); - siminfo_out("end data\n"); - - mcruninfo_out( "# ", outfile); - mcdatainfo_out("# ", outfile, detector); - fprintf(outfile, "# Data [%s/%s] %s:\n", detector.component, detector.filename, detector.zvar); - } - mcdetector_out_array_ascii(detector.m, detector.n*detector.p, detector.p1, - outfile, detector.istransposed); - if (detector.p2) { - fprintf(outfile, "# Errors [%s/%s] %s_err:\n", detector.component, detector.filename, detector.zvar); - mcdetector_out_array_ascii(detector.m, detector.n*detector.p, detector.p2, - outfile, detector.istransposed); - } - if (detector.p0) { - fprintf(outfile, "# Events [%s/%s] N:\n", detector.component, detector.filename); - mcdetector_out_array_ascii(detector.m, detector.n*detector.p, detector.p0, - outfile, detector.istransposed); - } - fclose(outfile); - - if (!exists) { - if (strcasestr(detector.format, "list")) - printf("Events: \"%s\"\n", - strlen(detector.filename) ? detector.filename : detector.component); - } - } /* if outfile */ - ); /* MPI_MASTER */ -#ifdef USE_MPI - if (strcasestr(detector.format, "list") && mpi_node_count > 1) { - int node_i=0; - /* loop along MPI nodes to write sequentially */ - for(node_i=0; node_i strlen(original)) n = strlen(original); - else original += strlen(original)-n; - strncpy(valid, original, n); - - for (i=0; i < n; i++) - { - if ( (valid[i] > 122) - || (valid[i] < 32) - || (strchr("!\"#$%&'()*+,-.:;<=>?@[\\]^`/ \n\r\t", valid[i]) != NULL) ) - { - if (i) valid[i] = '_'; else valid[i] = 'm'; - } - } - valid[i] = '\0'; - - return(valid); -} /* strcpy_valid */ - -/* end ascii output section ================================================= */ - - - - - - - -#ifdef USE_NEXUS - -/* ========================================================================== */ - -/* NeXus output */ - -/* ========================================================================== */ - -#define nxprintf(...) nxstr('d', __VA_ARGS__) -#define nxprintattr(...) nxstr('a', __VA_ARGS__) - -/******************************************************************************* -* nxstr: output a tag=value data set (char) in NeXus/current group -* when 'format' is larger that 1024 chars it is used as value for the 'tag' -* else the value is assembled with format and following arguments. -* type='d' -> data set -* 'a' -> attribute for current data set -*******************************************************************************/ -static int nxstr(char type, NXhandle *f, char *tag, char *format, ...) -{ - va_list ap; - char value[CHAR_BUF_LENGTH]; - int i; - int ret=NX_OK; - - if (!tag || !format || !strlen(tag) || !strlen(format)) return(NX_OK); - - /* assemble the value string */ - if (strlen(format) < CHAR_BUF_LENGTH) { - va_start(ap, format); - ret = vsnprintf(value, CHAR_BUF_LENGTH, format, ap); - va_end(ap); - - i = strlen(value); - } else { - i = strlen(format); - } - - if (type == 'd') { - /* open/put/close data set */ - if (NXmakedata (f, tag, NX_CHAR, 1, &i) != NX_OK) return(NX_ERROR); - NXopendata (f, tag); - if (strlen(format) < CHAR_BUF_LENGTH) - ret = NXputdata (f, value); - else - ret = NXputdata (f, format); - NXclosedata(f); - } else { - if (strlen(format) < CHAR_BUF_LENGTH) - ret = NXputattr (f, tag, value, strlen(value), NX_CHAR); - else - ret = NXputattr (f, tag, format, strlen(format), NX_CHAR); - } - - return(ret); - -} /* nxstr */ - -/******************************************************************************* -* mcinfo_readfile: read a full file into a string buffer which is allocated -* Think to free the buffer after use. -* Used in: mcinfo_out_nexus (nexus) -*******************************************************************************/ -char *mcinfo_readfile(char *filename) -{ - FILE *f = fopen(filename, "rb"); - if (!f) return(NULL); - fseek(f, 0, SEEK_END); - long fsize = ftell(f); - rewind(f); - char *string = malloc(fsize + 1); - if (string) { - int n = fread(string, fsize, 1, f); - fclose(f); - - string[fsize] = 0; - } - return(string); -} - -/******************************************************************************* -* mcinfo_out: output instrument/simulation groups in NeXus file -* Used in: siminfo_init (nexus) -*******************************************************************************/ -static void mcinfo_out_nexus(NXhandle f) -{ - FILE *fid; /* for intrument source code/C/IDF */ - char *buffer=NULL; - time_t t =time(NULL); /* for date */ - char entry0[CHAR_BUF_LENGTH]; - int count=0; - char name[CHAR_BUF_LENGTH]; - char class[CHAR_BUF_LENGTH]; - - if (!f || mcdisable_output_files) return; - - /* write NeXus NXroot attributes */ - /* automatically added: file_name, HDF5_Version, file_time, NeXus_version */ - nxprintattr(f, "creator", "%s generated with " MCCODE_STRING, instrument_name); - - /* count the number of existing NXentry and create the next one */ - NXgetgroupinfo(f, &count, name, class); - sprintf(entry0, "entry%i", count+1); - - /* create the main NXentry (mandatory in NeXus) */ - if (NXmakegroup(f, entry0, "NXentry") == NX_OK) - if (NXopengroup(f, entry0, "NXentry") == NX_OK) { - - nxprintf(nxhandle, "program_name", MCCODE_STRING); - nxprintf(f, "start_time", ctime(&t)); - nxprintf(f, "title", "%s%s%s simulation generated by instrument %s", - dirname && strlen(dirname) ? dirname : ".", MC_PATHSEP_S, siminfo_name, - instrument_name); - nxprintattr(f, "program_name", MCCODE_STRING); - nxprintattr(f, "instrument", instrument_name); - nxprintattr(f, "simulation", "%s%s%s", - dirname && strlen(dirname) ? dirname : ".", MC_PATHSEP_S, siminfo_name); - - /* write NeXus instrument group */ - if (NXmakegroup(f, "instrument", "NXinstrument") == NX_OK) - if (NXopengroup(f, "instrument", "NXinstrument") == NX_OK) { - int i; - char *string=NULL; - - /* write NeXus parameters(types) data =================================== */ - string = (char*)malloc(CHAR_BUF_LENGTH); - if (string) { - strcpy(string, ""); - for(i = 0; i < numipar; i++) - { - char ThisParam[CHAR_BUF_LENGTH]; - snprintf(ThisParam, CHAR_BUF_LENGTH, " %s(%s)", mcinputtable[i].name, - (*mcinputtypes[mcinputtable[i].type].parminfo) - (mcinputtable[i].name)); - if (strlen(string) + strlen(ThisParam) < CHAR_BUF_LENGTH) - strcat(string, ThisParam); - } - nxprintattr(f, "Parameters", string); - free(string); - } - - nxprintattr(f, "name", instrument_name); - nxprintf (f, "name", instrument_name); - nxprintattr(f, "Source", instrument_source); - - nxprintattr(f, "Trace_enabled", traceenabled ? "yes" : "no"); - nxprintattr(f, "Default_main", defaultmain ? "yes" : "no"); - nxprintattr(f, "Embedded_runtime", - #ifdef MC_EMBEDDED_RUNTIME - "yes" - #else - "no" - #endif - ); - - /* add instrument source code when available */ - buffer = mcinfo_readfile(instrument_source); - if (buffer && strlen(buffer)) { - long length=strlen(buffer); - nxprintf (f, "description", buffer); - NXopendata(f,"description"); - nxprintattr(f, "file_name", instrument_source); - nxprintattr(f, "file_size", "%li", length); - nxprintattr(f, "MCCODE_STRING", MCCODE_STRING); - NXclosedata(f); - nxprintf (f,"instrument_source", "%s " MCCODE_NAME " " MCCODE_PARTICLE " Monte Carlo simulation", instrument_name); - free(buffer); - } else - nxprintf (f, "description", "File %s not found (instrument description %s is missing)", - instrument_source, instrument_name); - - /* add Mantid/IDF.xml when available */ - char *IDFfile=NULL; - IDFfile = (char*)malloc(CHAR_BUF_LENGTH); - sprintf(IDFfile,"%s%s",instrument_source,".xml"); - buffer = mcinfo_readfile(IDFfile); - if (buffer && strlen(buffer)) { - NXmakegroup (nxhandle, "instrument_xml", "NXnote"); - NXopengroup (nxhandle, "instrument_xml", "NXnote"); - nxprintf(f, "data", buffer); - nxprintf(f, "description", "IDF.xml file found with instrument %s", instrument_source); - nxprintf(f, "type", "text/xml"); - NXclosegroup(f); /* instrument_xml */ - free(buffer); - } - free(IDFfile); - NXclosegroup(f); /* instrument */ - } /* NXinstrument */ - - /* write NeXus simulation group */ - if (NXmakegroup(f, "simulation", "NXnote") == NX_OK) - if (NXopengroup(f, "simulation", "NXnote") == NX_OK) { - - nxprintattr(f, "name", "%s%s%s", - dirname && strlen(dirname) ? dirname : ".", MC_PATHSEP_S, siminfo_name); - - nxprintf (f, "name", "%s", siminfo_name); - nxprintattr(f, "Format", mcformat && strlen(mcformat) ? mcformat : MCCODE_NAME); - nxprintattr(f, "URL", "http://www.mccode.org"); - nxprintattr(f, "program", MCCODE_STRING); - nxprintattr(f, "Instrument",instrument_source); - nxprintattr(f, "Trace", mcdotrace ? "yes" : "no"); - nxprintattr(f, "Gravitation",mcgravitation ? "yes" : "no"); - nxprintattr(f, "Seed", "%li", mcseed); - nxprintattr(f, "Directory", dirname); - #ifdef USE_MPI - if (mpi_node_count > 1) - nxprintf(f, "Nodes", "%i", mpi_node_count); - #endif - - /* output parameter string ================================================ */ - if (NXmakegroup(f, "Param", "NXparameters") == NX_OK) - if (NXopengroup(f, "Param", "NXparameters") == NX_OK) { - int i; - char string[CHAR_BUF_LENGTH]; - for(i = 0; i < numipar; i++) { - if (mcget_run_num() || (mcinputtable[i].val && strlen(mcinputtable[i].val))) { - if (mcinputtable[i].par == NULL) - strncpy(string, (mcinputtable[i].val ? mcinputtable[i].val : ""), CHAR_BUF_LENGTH); - else - (*mcinputtypes[mcinputtable[i].type].printer)(string, mcinputtable[i].par); - - nxprintf(f, mcinputtable[i].name, "%s", string); - nxprintattr(f, mcinputtable[i].name, string); - } - } - NXclosegroup(f); /* Param */ - } /* NXparameters */ - - NXclosegroup(f); /* simulation */ - } /* NXsimulation */ - - /* create a group to hold all monitors */ - NXmakegroup(f, "data", "NXdetector"); - - /* leave the NXentry opened (closed at exit) */ - } /* NXentry */ -} /* mcinfo_out_nexus */ - -/******************************************************************************* -* mcdatainfo_out_nexus: output detector header -* mcdatainfo_out_nexus(detector) create group and write info to NeXus data file -* open data:NXdetector then filename:NXdata and write headers/attributes -* requires: NXentry to be opened -*******************************************************************************/ -static void -mcdatainfo_out_nexus(NXhandle f, MCDETECTOR detector) -{ - char data_name[CHAR_BUF_LENGTH]; - if (!f || !detector.m || mcdisable_output_files) return; - - strcpy_valid(data_name, - detector.filename && strlen(detector.filename) ? - detector.filename : detector.component); - - /* the NXdetector group has been created in mcinfo_out_nexus (siminfo_init) */ - if (NXopengroup(f, "data", "NXdetector") == NX_OK) { - - /* create and open the data group */ - /* this may fail when appending to list -> ignore/skip */ - NXMDisableErrorReporting(); /* unactivate NeXus error messages, as creation may fail */ - - if (NXmakegroup(f, data_name, "NXdata") == NX_OK) - if (NXopengroup(f, data_name, "NXdata") == NX_OK) { - - /* output metadata (as attributes) ======================================== */ - nxprintattr(f, "Date", detector.date); - nxprintattr(f, "type", detector.type); - nxprintattr(f, "Source", detector.instrument); - nxprintattr(f, "component", detector.component); - nxprintattr(f, "position", detector.position); - - nxprintattr(f, "title", detector.title); - nxprintattr(f, !mcget_run_num() || mcget_run_num() >= mcget_ncount() ? - "Ncount" : - "ratio", detector.ncount); - - if (strlen(detector.filename)) { - nxprintattr(f, "filename", detector.filename); - } - - nxprintattr(f, "statistics", detector.statistics); - nxprintattr(f, "signal", detector.signal); - nxprintattr(f, "values", detector.values); - - if (detector.rank >= 1) - { - nxprintattr(f, "xvar", detector.xvar); - nxprintattr(f, "yvar", detector.yvar); - nxprintattr(f, "xlabel", detector.xlabel); - nxprintattr(f, "ylabel", detector.ylabel); - if (detector.rank > 1) { - nxprintattr(f, "zvar", detector.zvar); - nxprintattr(f, "zlabel", detector.zlabel); - } - } - - nxprintattr(f, abs(detector.rank)==1 ? - "xlimits" : - "xylimits", detector.limits); - nxprintattr(f, "variables", - strcasestr(detector.format, "list") ? detector.ylabel : detector.variables); - nxprintf(f, "distance", detector.position); - nxprintf(f, "acquisition_mode", - strcasestr(detector.format, "list") ? "event" : "summed"); - - NXclosegroup(f); - } /* NXdata (filename) */ - NXMEnableErrorReporting(); /* re-enable NeXus error messages */ - NXclosegroup(f); - } /* NXdetector (data) */ - -} /* mcdatainfo_out_nexus */ - -/******************************************************************************* -* mcdetector_out_axis_nexus: write detector axis into current NXdata -* requires: NXdata to be opened -*******************************************************************************/ -int mcdetector_out_axis_nexus(NXhandle f, char *label, char *var, int rank, long length, double min, double max) -{ - if (!f || length <= 1 || mcdisable_output_files || max == min) return(NX_OK); - else { - double axis[length]; - char valid[CHAR_BUF_LENGTH]; - int dim=(int)length; - int i; - int nprimary=1; - /* create an axis from [min:max] */ - for(i = 0; i < length; i++) - axis[i] = min+(max-min)*(i+0.5)/length; - /* create the data set */ - strcpy_valid(valid, label); - NXcompmakedata(f, valid, NX_FLOAT64, 1, &dim, NX_COMPRESSION, &dim); - /* open it */ - if (NXopendata(f, valid) != NX_OK) { - fprintf(stderr, "Warning: could not open axis rank %i '%s' (NeXus)\n", - rank, valid); - return(NX_ERROR); - } - /* put the axis and its attributes */ - NXputdata (f, axis); - nxprintattr(f, "long_name", label); - nxprintattr(f, "short_name", var); - NXputattr (f, "axis", &rank, 1, NX_INT32); - nxprintattr(f, "units", var); - NXputattr (f, "primary", &nprimary, 1, NX_INT32); - NXclosedata(f); - - return(NX_OK); - } -} /* mcdetector_out_axis_nexus */ - -/******************************************************************************* -* mcdetector_out_array_nexus: write detector array into current NXdata (1D,2D) -* requires: NXdata to be opened -*******************************************************************************/ -int mcdetector_out_array_nexus(NXhandle f, char *part, double *data, MCDETECTOR detector) -{ - - int dims[3]={detector.m,detector.n,detector.p}; /* number of elements to write */ - int fulldims[3]={detector.m,detector.n,detector.p}; - int signal=1; - int exists=0; - int current_dims[3]={0,0,0}; - int ret=NX_OK; - - if (!f || !data || !detector.m || mcdisable_output_files) return(NX_OK); - - /* when this is a list, we set 1st dimension to NX_UNLIMITED for creation */ - if (strcasestr(detector.format, "list")) fulldims[0] = NX_UNLIMITED; - - /* create the data set in NXdata group */ - NXMDisableErrorReporting(); /* unactivate NeXus error messages, as creation may fail */ - ret = NXcompmakedata(f, part, NX_FLOAT64, detector.rank, fulldims, NX_COMPRESSION, dims); - if (ret != NX_OK) { - /* failed: data set already exists */ - int datatype=0; - int rank=0; - exists=1; - /* inquire current size of data set (nb of events stored) */ - NXopendata(f, part); - NXgetinfo(f, &rank, current_dims, &datatype); - NXclosedata(f); - } - NXMEnableErrorReporting(); /* re-enable NeXus error messages */ - - /* open the data set */ - if (NXopendata(f, part) == NX_ERROR) { - fprintf(stderr, "Warning: could not open DataSet %s '%s' (NeXus)\n", - part, detector.title); - return(NX_ERROR); - } - if (strcasestr(detector.format, "list")) { - current_dims[1] = current_dims[2] = 0; /* set starting location for writing slab */ - NXputslab(f, data, current_dims, dims); - if (!exists) - printf("Events: \"%s\"\n", - strlen(detector.filename) ? detector.filename : detector.component); - else - printf("Append: \"%s\"\n", - strlen(detector.filename) ? detector.filename : detector.component); - } else { - NXputdata (f, data); - } - - if (strstr(part,"data") || strstr(part, "events")) { - NXputattr(f, "signal", &signal, 1, NX_INT32); - nxprintattr(f, "short_name", detector.filename && strlen(detector.filename) ? - detector.filename : detector.component); - } - nxprintattr(f, "long_name", "%s '%s'", part, detector.title); - NXclosedata(f); - - return(NX_OK); -} /* mcdetector_out_array_nexus */ - -/******************************************************************************* -* mcdetector_out_data_nexus: write detector axes+data into current NXdata -* The data:NXdetector is opened, then filename:NXdata -* requires: NXentry to be opened -*******************************************************************************/ -int mcdetector_out_data_nexus(NXhandle f, MCDETECTOR detector) -{ - char data_name[CHAR_BUF_LENGTH]; - - if (!f || !detector.m || mcdisable_output_files) return(NX_OK); - - strcpy_valid(data_name, - detector.filename && strlen(detector.filename) ? - detector.filename : detector.component); - - /* the NXdetector group has been created in mcinfo_out_nexus (siminfo_init) */ - if (NXopengroup(f, "data", "NXdetector") == NX_OK) { - - /* the NXdata group has been created in mcdatainfo_out_nexus */ - if (NXopengroup(f, data_name, "NXdata") == NX_OK) { - - /* write axes, for histogram data sets, not for lists */ - if (!strcasestr(detector.format, "list")) { - mcdetector_out_axis_nexus(f, detector.xlabel, detector.xvar, - 1, detector.m, detector.xmin, detector.xmax); - - mcdetector_out_axis_nexus(f, detector.ylabel, detector.yvar, - 2, detector.n, detector.ymin, detector.ymax); - - mcdetector_out_axis_nexus(f, detector.zlabel, detector.zvar, - 3, detector.p, detector.zmin, detector.zmax); - - } /* !list */ - - /* write the actual data (appended if already exists) */ - if (!strcasestr(detector.format, "list")) { - mcdetector_out_array_nexus(f, "data", detector.p1, detector); - mcdetector_out_array_nexus(f, "errors", detector.p2, detector); - mcdetector_out_array_nexus(f, "ncount", detector.p0, detector); - } else - mcdetector_out_array_nexus( f, "events", detector.p1, detector); - - NXclosegroup(f); - } /* NXdata */ - NXclosegroup(f); - } /* NXdetector */ - - return(NX_OK); -} /* mcdetector_out_array_nexus */ - -#ifdef USE_MPI -/******************************************************************************* -* mcdetector_out_list_slaves: slaves send their list data to master which writes -* requires: NXentry to be opened -* WARNING: this method has a flaw: it requires all nodes to flush the lists -* the same number of times. In case one node is just below the buffer size -* when finishing (e.g. monitor_nd), it may not trigger save but others may. -* Then the number of recv/send is not constant along nodes, and simulation stalls. -*******************************************************************************/ -MCDETECTOR mcdetector_out_list_slaves(MCDETECTOR detector) -{ - int node_i=0; - MPI_MASTER( - printf("\n** MPI master gathering slave node list data ** \n"); - ); - - if (mpi_node_rank != mpi_node_root) { - /* MPI slave: slaves send their data to master: 2 MPI_Send calls */ - /* m, n, p must be sent first, since all slaves do not have the same number of events */ - int mnp[3]={detector.m,detector.n,detector.p}; - - if (mc_MPI_Send(mnp, 3, MPI_INT, mpi_node_root)!= MPI_SUCCESS) - fprintf(stderr, "Warning: proc %i to master: MPI_Send mnp list error (mcdetector_out_list_slaves)\n", mpi_node_rank); - if (!detector.p1 - || mc_MPI_Send(detector.p1, mnp[0]*mnp[1]*mnp[2], MPI_DOUBLE, mpi_node_root) != MPI_SUCCESS) - fprintf(stderr, "Warning: proc %i to master: MPI_Send p1 list error: mnp=%i (mcdetector_out_list_slaves)\n", mpi_node_rank, abs(mnp[0]*mnp[1]*mnp[2])); - /* slaves are done: sent mnp and p1 */ - return (detector); - } /* end slaves */ - - /* MPI master: receive data from slaves sequentially: 2 MPI_Recv calls */ - - if (mpi_node_rank == mpi_node_root) { - for(node_i=0; node_i 1) { - mcdetector_out_list_slaves(detector); - } -#endif /* USE_MPI */ - - return(detector); -} /* mcdetector_out_2D_nexus */ - -#endif /* USE_NEXUS*/ - - - - - - - - -/* ========================================================================== */ - -/* Main input functions */ -/* DETECTOR_OUT_xD function calls -> ascii or NeXus */ - -/* ========================================================================== */ - -/******************************************************************************* -* siminfo_init: open SIM and write header -*******************************************************************************/ -FILE *siminfo_init(FILE *f) -{ - int exists=0; - - /* check format */ - if (!mcformat || !strlen(mcformat) - || !strcasecmp(mcformat, "MCSTAS") || !strcasecmp(mcformat, "MCXTRACE") - || !strcasecmp(mcformat, "PGPLOT") || !strcasecmp(mcformat, "GNUPLOT") || !strcasecmp(mcformat, "MCCODE") - || !strcasecmp(mcformat, "MATLAB")) { - mcformat="McCode"; -#ifdef USE_NEXUS - } else if (strcasestr(mcformat, "NeXus")) { - /* Do nothing */ -#endif - } else { - fprintf(stderr, - "Warning: You have requested the output format %s which is unsupported by this binary. Resetting to standard %s format.\n",mcformat ,"McCode"); - mcformat="McCode"; - } - - /* open the SIM file if not defined yet */ - if (siminfo_file || mcdisable_output_files) - return (siminfo_file); - -#ifdef USE_NEXUS - /* only master writes NeXus header: calls NXopen(nxhandle) */ - if (mcformat && strcasestr(mcformat, "NeXus")) { - MPI_MASTER( - siminfo_file = mcnew_file(siminfo_name, "h5", &exists); - if(!siminfo_file) - fprintf(stderr, - "Warning: could not open simulation description file '%s'\n", - siminfo_name); - else - mcinfo_out_nexus(nxhandle); - ); - return(siminfo_file); /* points to nxhandle */ - } -#endif - - /* write main description file (only MASTER) */ - MPI_MASTER( - - siminfo_file = mcnew_file(siminfo_name, "sim", &exists); - if(!siminfo_file) - fprintf(stderr, - "Warning: could not open simulation description file '%s'\n", - siminfo_name); - else - { - /* write SIM header */ - time_t t=time(NULL); - siminfo_out("%s simulation description file for %s.\n", - MCCODE_NAME, instrument_name); - siminfo_out("Date: %s", ctime(&t)); /* includes \n */ - siminfo_out("Program: %s\n\n", MCCODE_STRING); - - siminfo_out("begin instrument: %s\n", instrument_name); - mcinfo_out( " ", siminfo_file); - siminfo_out("end instrument\n"); - - siminfo_out("\nbegin simulation: %s\n", dirname); - mcruninfo_out(" ", siminfo_file); - siminfo_out("end simulation\n"); - - } - return (siminfo_file); - - ); /* MPI_MASTER */ - -} /* siminfo_init */ - -/******************************************************************************* -* siminfo_close: close SIM -*******************************************************************************/ -void siminfo_close() -{ -#ifdef USE_MPI - if(mpi_node_rank == mpi_node_root) { -#endif - if(siminfo_file && !mcdisable_output_files) { -#ifdef USE_NEXUS - if (mcformat && strcasestr(mcformat, "NeXus")) { - time_t t=time(NULL); - nxprintf(nxhandle, "end_time", ctime(&t)); - nxprintf(nxhandle, "duration", "%li", (long)t-mcstartdate); - NXclosegroup(nxhandle); /* NXentry */ - NXclose(&nxhandle); - } else { -#endif - fclose(siminfo_file); -#ifdef USE_NEXUS - } -#endif -#ifdef USE_MPI - } -#endif - siminfo_file = NULL; - } -} /* siminfo_close */ - -/******************************************************************************* -* mcdetector_out_0D: wrapper for 0D (single value). -* Output single detector/monitor data (p0, p1, p2). -* Title is t, component name is c. -*******************************************************************************/ -MCDETECTOR mcdetector_out_0D(char *t, double p0, double p1, double p2, - char *c, Coords posa) -{ - /* import and perform basic detector analysis (and handle MPI reduce) */ - MCDETECTOR detector = detector_import(mcformat, - c, (t ? t : MCCODE_STRING " data"), - 1, 1, 1, - "I", "", "", - "I", "", "", - 0, 0, 0, 0, 0, 0, "", - &p0, &p1, &p2, posa); /* write Detector: line */ - -#ifdef USE_NEXUS - if (strcasestr(detector.format, "NeXus")) - return(mcdetector_out_0D_nexus(detector)); - else -#endif - return(mcdetector_out_0D_ascii(detector)); - -} /* mcdetector_out_0D */ - - - -/******************************************************************************* -* mcdetector_out_1D: wrapper for 1D. -* Output 1d detector data (p0, p1, p2) for n bins linearly -* distributed across the range x1..x2 (x1 is lower limit of first -* bin, x2 is upper limit of last bin). Title is t, axis labels are xl -* and yl. File name is f, component name is c. -* -* t: title -* xl: x-label -* yl: y-label -* xvar: measured variable length -* x1: x axus min -* x2: x axis max -* n: 1d data vector lenght -* p0: pntr to start of data block#0 -* p1: pntr to start of data block#1 -* p2: pntr to start of data block#2 -* f: filename -* -* Not included in the macro, and here forwarded to detector_import: -* c: ? -* posa: ? -*******************************************************************************/ -MCDETECTOR mcdetector_out_1D(char *t, char *xl, char *yl, - char *xvar, double x1, double x2, - long n, - double *p0, double *p1, double *p2, char *f, - char *c, Coords posa) -{ - /* import and perform basic detector analysis (and handle MPI_Reduce) */ - // detector_import calls mcdetector_statistics, which will return different - // MCDETECTOR versions for 1-D data based on the value of mcformat. - // - MCDETECTOR detector = detector_import(mcformat, - c, (t ? t : MCCODE_STRING " 1D data"), - n, 1, 1, - xl, yl, (n > 1 ? "Signal per bin" : " Signal"), - xvar, "(I,I_err)", "I", - x1, x2, 0, 0, 0, 0, f, - p0, p1, p2, posa); /* write Detector: line */ - if (!detector.p1 || !detector.m) return(detector); - -#ifdef USE_NEXUS - if (strcasestr(detector.format, "NeXus")) - detector = mcdetector_out_1D_nexus(detector); - else -#endif - detector = mcdetector_out_1D_ascii(detector); - if (detector.p1 != p1 && detector.p1) { - // mcdetector_statistics allocated memory but it hasn't been freed. - free(detector.p1); - // plus undo the other damage done there: - detector.p0 = p0; // was set to NULL - detector.p1 = p1; // was set to this_p1 - detector.p2 = p2; // was set to NULL - detector.m = detector.n; // (e.g., labs(n)) - detector.n = 1; // not (n x n) - detector.istransposed = n < 0 ? 1 : 0; - } - return detector; - -} /* mcdetector_out_1D */ - -/******************************************************************************* -* mcdetector_out_2D: wrapper for 2D. -* Special case for list: master creates file first, then slaves append their -* blocks without header- -* -* t: title -* xl: x-label -* yl: y-label -* x1: x axus min -* x2: x axis max -* y1: y axis min -* y2: y axis max -* m: dim 1 (x) size -* n: dim 2 (y) size -* p0: pntr to start of data block#0 -* p1: pntr to start of data block#1 -* p2: pntr to start of data block#2 -* f: filename -* -* Not included in the macro, and here forwarded to detector_import: -* c: ? -* posa: ? -*******************************************************************************/ -MCDETECTOR mcdetector_out_2D(char *t, char *xl, char *yl, - double x1, double x2, double y1, double y2, - long m, long n, - double *p0, double *p1, double *p2, char *f, - char *c, Coords posa) -{ - char xvar[CHAR_BUF_LENGTH]; - char yvar[CHAR_BUF_LENGTH]; - - /* create short axes labels */ - if (xl && strlen(xl)) { strncpy(xvar, xl, CHAR_BUF_LENGTH); xvar[2]='\0'; } - else strcpy(xvar, "x"); - if (yl && strlen(yl)) { strncpy(yvar, yl, CHAR_BUF_LENGTH); yvar[2]='\0'; } - else strcpy(yvar, "y"); - - MCDETECTOR detector; - - /* import and perform basic detector analysis (and handle MPI_Reduce) */ - if (labs(m) == 1) {/* n>1 on Y, m==1 on X: 1D, no X axis*/ - detector = detector_import(mcformat, - c, (t ? t : MCCODE_STRING " 1D data"), - n, 1, 1, - yl, "", "Signal per bin", - yvar, "(I,Ierr)", "I", - y1, y2, x1, x2, 0, 0, f, - p0, p1, p2, posa); /* write Detector: line */ - } else if (labs(n)==1) {/* m>1 on X, n==1 on Y: 1D, no Y axis*/ - detector = detector_import(mcformat, - c, (t ? t : MCCODE_STRING " 1D data"), - m, 1, 1, - xl, "", "Signal per bin", - xvar, "(I,Ierr)", "I", - x1, x2, y1, y2, 0, 0, f, - p0, p1, p2, posa); /* write Detector: line */ - }else { - detector = detector_import(mcformat, - c, (t ? t : MCCODE_STRING " 2D data"), - m, n, 1, - xl, yl, "Signal per bin", - xvar, yvar, "I", - x1, x2, y1, y2, 0, 0, f, - p0, p1, p2, posa); /* write Detector: line */ - } - - if (!detector.p1 || !detector.m) return(detector); - -#ifdef USE_NEXUS - if (strcasestr(detector.format, "NeXus")) - return(mcdetector_out_2D_nexus(detector)); - else -#endif - return(mcdetector_out_2D_ascii(detector)); - -} /* mcdetector_out_2D */ - -/******************************************************************************* -* mcdetector_out_list: wrapper for list output (calls out_2D with mcformat+"list"). -* m=number of events, n=size of each event -*******************************************************************************/ -MCDETECTOR mcdetector_out_list(char *t, char *xl, char *yl, - long m, long n, - double *p1, char *f, - char *c, Coords posa) -{ - char format_new[CHAR_BUF_LENGTH]; - char *format_org; - MCDETECTOR detector; - - format_org = mcformat; - strcpy(format_new, mcformat); - strcat(format_new, " list"); - mcformat = format_new; - - detector = mcdetector_out_2D(t, xl, yl, - 1,labs(m),1,labs(n), - m,n, - NULL, p1, NULL, f, - c, posa); - - mcformat = format_org; - return(detector); -} - -/******************************************************************************* - * mcuse_dir: set data/sim storage directory and create it, - * or exit with error if exists - ******************************************************************************/ -static void -mcuse_dir(char *dir) -{ - if (!dir || !strlen(dir)) return; -#ifdef MC_PORTABLE - fprintf(stderr, "Error: " - "Directory output cannot be used with portable simulation (mcuse_dir)\n"); - exit(1); -#else /* !MC_PORTABLE */ - /* handle file://directory URL type */ - if (strncmp(dir, "file://", strlen("file://"))) - dirname = dir; - else - dirname = dir+strlen("file://"); - - -#ifdef USE_MPI - if(mpi_node_rank == mpi_node_root) { -#endif - if(mkdir(dirname, 0777)) { -#ifndef DANSE - fprintf(stderr, "Error: unable to create directory '%s' (mcuse_dir)\n", dir); - fprintf(stderr, "(Maybe the directory already exists?)\n"); -#endif -#ifdef USE_MPI - MPI_Abort(MPI_COMM_WORLD, -1); -#endif - exit(-1); - } -#ifdef USE_MPI - } -#endif - - /* remove trailing PATHSEP (if any) */ - while (strlen(dirname) && dirname[strlen(dirname) - 1] == MC_PATHSEP_C) - dirname[strlen(dirname) - 1]='\0'; -#endif /* !MC_PORTABLE */ -} /* mcuse_dir */ - -/******************************************************************************* -* mcinfo: display instrument simulation info to stdout and exit -*******************************************************************************/ -static void -mcinfo(void) -{ - fprintf(stdout, "begin instrument: %s\n", instrument_name); - mcinfo_out(" ", stdout); - fprintf(stdout, "end instrument\n"); - fprintf(stdout, "begin simulation: %s\n", dirname ? dirname : "."); - mcruninfo_out(" ", stdout); - fprintf(stdout, "end simulation\n"); - exit(0); /* includes MPI_Finalize in MPI mode */ -} /* mcinfo */ - -/******************************************************************************* -* mcparameterinfo: display instrument parameter info to stdout and exit -*******************************************************************************/ -static void -mcparameterinfo(void) -{ - mcparameterinfo_out(" ", stdout); - exit(0); /* includes MPI_Finalize in MPI mode */ -} /* mcparameterinfo */ - - - -#endif /* ndef MCCODE_R_IO_C */ - -/* end of the I/O section =================================================== */ - - - - - - - -/******************************************************************************* -* mcset_ncount: set total number of rays to generate -*******************************************************************************/ -void mcset_ncount(unsigned long long int count) -{ - mcncount = count; -} - -/* mcget_ncount: get total number of rays to generate */ -unsigned long long int mcget_ncount(void) -{ - return mcncount; -} - -/* mcget_run_num: get curent number of rays */ -/* Within the TRACE scope we are now using _particle->uid directly */ -unsigned long long int mcget_run_num() // shuld be (_class_particle* _particle) somehow -{ - /* This function only remains for the few cases outside TRACE where we need to know - the number of simulated particles */ - return mcrun_num; -} - -/* mcsetn_arg: get ncount from a string argument */ -static void -mcsetn_arg(char *arg) -{ - mcset_ncount((long long int) strtod(arg, NULL)); -} - -/* mcsetseed: set the random generator seed from a string argument */ -static void -mcsetseed(char *arg) -{ - mcseed = atol(arg); - if(!mcseed) { - // srandom(mcseed); - //} else { - fprintf(stderr, "Error: seed must not be zero (mcsetseed)\n"); - exit(1); - } -} - -/* Following part is only embedded when not redundent with mccode-r.h ========= */ - -#ifndef MCCODE_H - -/* SECTION: MCDISPLAY support. =============================================== */ - -/******************************************************************************* -* Just output MCDISPLAY keywords to be caught by an external plotter client. -*******************************************************************************/ - -void mcdis_magnify(char *what){ - // Do nothing here, better use interactive zoom from the tools -} - -void mcdis_line(double x1, double y1, double z1, - double x2, double y2, double z2){ - printf("MCDISPLAY: multiline(2,%g,%g,%g,%g,%g,%g)\n", - x1,y1,z1,x2,y2,z2); -} - -void mcdis_dashed_line(double x1, double y1, double z1, - double x2, double y2, double z2, int n){ - int i; - const double dx = (x2-x1)/(2*n+1); - const double dy = (y2-y1)/(2*n+1); - const double dz = (z2-z1)/(2*n+1); - - for(i = 0; i < n+1; i++) - mcdis_line(x1 + 2*i*dx, y1 + 2*i*dy, z1 + 2*i*dz, - x1 + (2*i+1)*dx, y1 + (2*i+1)*dy, z1 + (2*i+1)*dz); -} - -void mcdis_multiline(int count, ...){ - va_list ap; - double x,y,z; - - printf("MCDISPLAY: multiline(%d", count); - va_start(ap, count); - while(count--) - { - x = va_arg(ap, double); - y = va_arg(ap, double); - z = va_arg(ap, double); - printf(",%g,%g,%g", x, y, z); - } - va_end(ap); - printf(")\n"); -} - -void mcdis_rectangle(char* plane, double x, double y, double z, - double width, double height){ - /* draws a rectangle in the plane */ - /* x is ALWAYS width and y is ALWAYS height */ - if (strcmp("xy", plane)==0) { - mcdis_multiline(5, - x - width/2, y - height/2, z, - x + width/2, y - height/2, z, - x + width/2, y + height/2, z, - x - width/2, y + height/2, z, - x - width/2, y - height/2, z); - } else if (strcmp("xz", plane)==0) { - mcdis_multiline(5, - x - width/2, y, z - height/2, - x + width/2, y, z - height/2, - x + width/2, y, z + height/2, - x - width/2, y, z + height/2, - x - width/2, y, z - height/2); - } else if (strcmp("yz", plane)==0) { - mcdis_multiline(5, - x, y - height/2, z - width/2, - x, y - height/2, z + width/2, - x, y + height/2, z + width/2, - x, y + height/2, z - width/2, - x, y - height/2, z - width/2); - } else { - - fprintf(stderr, "Error: Definition of plane %s unknown\n", plane); - exit(1); - } -} - -/* draws a box with center at (x, y, z) and - width (deltax), height (deltay), length (deltaz) */ -void mcdis_box(double x, double y, double z, - double width, double height, double length){ - - mcdis_rectangle("xy", x, y, z-length/2, width, height); - mcdis_rectangle("xy", x, y, z+length/2, width, height); - mcdis_line(x-width/2, y-height/2, z-length/2, - x-width/2, y-height/2, z+length/2); - mcdis_line(x-width/2, y+height/2, z-length/2, - x-width/2, y+height/2, z+length/2); - mcdis_line(x+width/2, y-height/2, z-length/2, - x+width/2, y-height/2, z+length/2); - mcdis_line(x+width/2, y+height/2, z-length/2, - x+width/2, y+height/2, z+length/2); -} - -void mcdis_circle(char *plane, double x, double y, double z, double r){ - printf("MCDISPLAY: circle('%s',%g,%g,%g,%g)\n", plane, x, y, z, r); -} - -/* Draws a circle with center (x,y,z), radius (r), and in the plane - * with normal (nx,ny,nz)*/ -void mcdis_Circle(double x, double y, double z, double r, double nx, double ny, double nz){ - int i; - if(nx==0 && ny && nz==0){ - for (i=0;i<24; i++){ - mcdis_line(x+r*sin(i*2*PI/24),y,z+r*cos(i*2*PI/24), - x+r*sin((i+1)*2*PI/24),y,z+r*cos((i+1)*2*PI/24)); - } - }else{ - double mx,my,mz; - /*generate perpendicular vector using (nx,ny,nz) and (0,1,0)*/ - vec_prod(mx,my,mz, 0,1,0, nx,ny,nz); - NORM(mx,my,mz); - /*draw circle*/ - for (i=0;i<24; i++){ - double ux,uy,uz; - double wx,wy,wz; - rotate(ux,uy,uz, mx,my,mz, i*2*PI/24, nx,ny,nz); - rotate(wx,wy,wz, mx,my,mz, (i+1)*2*PI/24, nx,ny,nz); - mcdis_line(x+ux*r,y+uy*r,z+uz*r, - x+wx*r,y+wy*r,z+wz*r); - } - } -} - -/* Draws a cylinder with center at (x,y,z) with extent (r,height). - * The cylinder axis is along the vector nx,ny,nz. - * N determines how many vertical lines are drawn.*/ -void mcdis_cylinder( double x, double y, double z, - double r, double height, int N, double nx, double ny, double nz){ - int i; - /*no lines make little sense - so trigger the default*/ - if(N<=0) N=5; - - NORM(nx,ny,nz); - double h_2=height/2.0; - mcdis_Circle(x+nx*h_2,y+ny*h_2,z+nz*h_2,r,nx,ny,nz); - mcdis_Circle(x-nx*h_2,y-ny*h_2,z-nz*h_2,r,nx,ny,nz); - - double mx,my,mz; - /*generate perpendicular vector using (nx,ny,nz) and (0,1,0)*/ - if(nx==0 && ny && nz==0){ - mx=my=0;mz=1; - }else{ - vec_prod(mx,my,mz, 0,1,0, nx,ny,nz); - NORM(mx,my,mz); - } - /*draw circle*/ - for (i=0; i<24; i++){ - double ux,uy,uz; - rotate(ux,uy,uz, mx,my,mz, i*2*PI/24, nx,ny,nz); - mcdis_line(x+nx*h_2+ux*r, y+ny*h_2+uy*r, z+nz*h_2+uz*r, - x-nx*h_2+ux*r, y-ny*h_2+uy*r, z-nz*h_2+uz*r); - } -} - -/* draws a sphere with center at (x,y,z) with extent (r) - * The sphere is drawn using N longitudes and N latitudes.*/ -void mcdis_sphere(double x, double y, double z, double r, int N){ - double nx,ny,nz; - int i; - /*no lines make little sense - so trigger the default*/ - if(N<=0) N=5; - - nx=0;ny=0;nz=1; - mcdis_Circle(x,y,z,r,nx,ny,nz); - for (i=1;ix /= temp; - c->y /= temp; - c->z /= temp; -} - -/* coords_test_zero: check if zero vector*/ -int coords_test_zero(Coords a){ - return ( a.x==0 && a.y==0 && a.z==0 ); -} - -/******************************************************************************* -* The Rotation type implements a rotation transformation of a coordinate -* system in the form of a double[3][3] matrix. -* -* Contrary to the Coords type in coords.c, rotations are passed by -* reference. Functions that yield new rotations do so by writing to an -* explicit result parameter; rotations are not returned from functions. The -* reason for this is that arrays cannot by returned from functions (though -* structures can; thus an alternative would have been to wrap the -* double[3][3] array up in a struct). Such are the ways of C programming. -* -* A rotation represents the tranformation of the coordinates of a vector when -* changing between coordinate systems that are rotated with respect to each -* other. For example, suppose that coordinate system Q is rotated 45 degrees -* around the Z axis with respect to coordinate system P. Let T be the -* rotation transformation representing a 45 degree rotation around Z. Then to -* get the coordinates of a vector r in system Q, apply T to the coordinates -* of r in P. If r=(1,0,0) in P, it will be (sqrt(1/2),-sqrt(1/2),0) in -* Q. Thus we should be careful when interpreting the sign of rotation angles: -* they represent the rotation of the coordinate systems, not of the -* coordinates (which has opposite sign). -*******************************************************************************/ - -/******************************************************************************* -* rot_set_rotation: Get transformation for rotation first phx around x axis, -* then phy around y, then phz around z. -*******************************************************************************/ -void rot_set_rotation(Rotation t, double phx, double phy, double phz) -{ - if ((phx == 0) && (phy == 0) && (phz == 0)) { - t[0][0] = 1.0; - t[0][1] = 0.0; - t[0][2] = 0.0; - t[1][0] = 0.0; - t[1][1] = 1.0; - t[1][2] = 0.0; - t[2][0] = 0.0; - t[2][1] = 0.0; - t[2][2] = 1.0; - } else { - double cx = cos(phx); - double sx = sin(phx); - double cy = cos(phy); - double sy = sin(phy); - double cz = cos(phz); - double sz = sin(phz); - - t[0][0] = cy*cz; - t[0][1] = sx*sy*cz + cx*sz; - t[0][2] = sx*sz - cx*sy*cz; - t[1][0] = -cy*sz; - t[1][1] = cx*cz - sx*sy*sz; - t[1][2] = sx*cz + cx*sy*sz; - t[2][0] = sy; - t[2][1] = -sx*cy; - t[2][2] = cx*cy; - } -} - -/******************************************************************************* -* rot_test_identity: Test if rotation is identity -*******************************************************************************/ -int rot_test_identity(Rotation t) -{ - return (t[0][0] + t[1][1] + t[2][2] == 3); -} - -/******************************************************************************* -* rot_mul: Matrix multiplication of transformations (this corresponds to -* combining transformations). After rot_mul(T1, T2, T3), doing T3 is -* equal to doing first T2, then T1. -* Note that T3 must not alias (use the same array as) T1 or T2. -*******************************************************************************/ -void rot_mul(Rotation t1, Rotation t2, Rotation t3) -{ - if (rot_test_identity(t1)) { - rot_copy(t3, t2); - } else if (rot_test_identity(t2)) { - rot_copy(t3, t1); - } else { - int i,j; - for(i = 0; i < 3; i++) - for(j = 0; j < 3; j++) - t3[i][j] = t1[i][0]*t2[0][j] + t1[i][1]*t2[1][j] + t1[i][2]*t2[2][j]; - } -} - -/******************************************************************************* -* rot_copy: Copy a rotation transformation (arrays cannot be assigned in C). -*******************************************************************************/ -void rot_copy(Rotation dest, Rotation src) -{ - int i,j; - for(i = 0; i < 3; i++) - for(j = 0; j < 3; j++) - dest[i][j] = src[i][j]; -} - -/******************************************************************************* -* rot_transpose: Matrix transposition, which is inversion for Rotation matrices -*******************************************************************************/ -void rot_transpose(Rotation src, Rotation dst) -{ - dst[0][0] = src[0][0]; - dst[0][1] = src[1][0]; - dst[0][2] = src[2][0]; - dst[1][0] = src[0][1]; - dst[1][1] = src[1][1]; - dst[1][2] = src[2][1]; - dst[2][0] = src[0][2]; - dst[2][1] = src[1][2]; - dst[2][2] = src[2][2]; -} - -/******************************************************************************* -* rot_apply: returns t*a -*******************************************************************************/ -Coords rot_apply(Rotation t, Coords a) -{ - Coords b; - if (rot_test_identity(t)) { - return a; - } else { - b.x = t[0][0]*a.x + t[0][1]*a.y + t[0][2]*a.z; - b.y = t[1][0]*a.x + t[1][1]*a.y + t[1][2]*a.z; - b.z = t[2][0]*a.x + t[2][1]*a.y + t[2][2]*a.z; - return b; - } -} - -/** - * Pretty-printing of rotation matrices. - */ -void rot_print(Rotation rot) { - printf("[ %4.2f %4.2f %4.2f ]\n", - rot[0][0], rot[0][1], rot[0][2]); - printf("[ %4.2f %4.2f %4.2f ]\n", - rot[1][0], rot[1][1], rot[1][2]); - printf("[ %4.2f %4.2f %4.2f ]\n\n", - rot[2][0], rot[2][1], rot[2][2]); -} - -/** - * Vector product: used by vec_prod (mccode-r.h). Use coords_xp for Coords. - */ -void vec_prod_func(double *x, double *y, double *z, - double x1, double y1, double z1, - double x2, double y2, double z2) { - *x = (y1)*(z2) - (y2)*(z1); - *y = (z1)*(x2) - (z2)*(x1); - *z = (x1)*(y2) - (x2)*(y1); -} - -/** - * Scalar product: use coords_sp for Coords. - */ -double scalar_prod( - double x1, double y1, double z1, - double x2, double y2, double z2) { - return ((x1 * x2) + (y1 * y2) + (z1 * z2)); -} - -mcstatic void norm_func(double *x, double *y, double *z) { - double temp = (*x * *x) + (*y * *y) + (*z * *z); - if (temp != 0) { - temp = sqrt(temp); - *x /= temp; - *y /= temp; - *z /= temp; - } -} - - -/* SECTION: GPU algorithms ================================================== */ - - -/* -* Divide-and-conquer strategy for parallelizing this task: Sort absorbed -* particles last. -* -* particles: the particle array, required to checking _absorbed -* pbuffer: same-size particle buffer array required for parallel sort -* len: sorting area-of-interest size (e.g. from previous calls) -* buffer_len: total array size -* flag_split: if set, multiply live particles into absorbed slots, up to buffer_len -* multiplier: output arg, becomes the SPLIT multiplier if flag_split is set -*/ -#ifdef FUNNEL -long sort_absorb_last(_class_particle* particles, _class_particle* pbuffer, long len, long buffer_len, long flag_split, long* multiplier) { - #define SAL_THREADS 1024 // num parallel sections - if (len_absorbed)); - - // return (no SPLIT) - if (flag_split != 1) - return accumlen; - - // SPLIT - repeat the non-absorbed block N-1 times, where len % accumlen = N + R - int mult = buffer_len / accumlen; // TODO: possibly use a new arg, bufferlen, rather than len - - // not enough space for full-block split, return - if (mult <= 1) - return accumlen; - - // copy non-absorbed block - #pragma acc parallel loop present(particles) - for (long tidx = 0; tidx < accumlen; tidx++) { // tidx: thread index - unsigned long randstate[7]; - _class_particle sourcebuffer; - _class_particle targetbuffer; - // assign reduced weight to all particles - particles[tidx].p=particles[tidx].p/mult; - #pragma acc loop seq - for (long bidx = 1; bidx < mult; bidx++) { // bidx: block index - // preserve absorbed particle (for randstate) - sourcebuffer = particles[bidx*accumlen + tidx]; - // buffer full particle struct - targetbuffer = particles[tidx]; - // reassign previous randstate - targetbuffer.randstate[0] = sourcebuffer.randstate[0]; - targetbuffer.randstate[1] = sourcebuffer.randstate[1]; - targetbuffer.randstate[2] = sourcebuffer.randstate[2]; - targetbuffer.randstate[3] = sourcebuffer.randstate[3]; - targetbuffer.randstate[4] = sourcebuffer.randstate[4]; - targetbuffer.randstate[5] = sourcebuffer.randstate[5]; - targetbuffer.randstate[6] = sourcebuffer.randstate[6]; - // apply - particles[bidx*accumlen + tidx] = targetbuffer; - } - } - - // set out split multiplier value - *multiplier = mult; - - // return expanded array size - return accumlen * mult; -} - -#endif - -/* -* Fallback serial version of the one above. -*/ -long sort_absorb_last_serial(_class_particle* particles, long len) { - long i = 0; - long j = len - 1; - _class_particle pbuffer; - - // bubble - while (i < j) { - while (!particles[i]._absorbed && ix; - b.y = particle->y; - b.z = particle->z; - c = rot_apply(t, b); - b = coords_add(c, a); - particle->x = b.x; - particle->y = b.y; - particle->z = b.z; - -#if MCCODE_PARTICLE_CODE == 2112 - if (particle->vz != 0.0 || particle->vx != 0.0 || particle->vy != 0.0) - mccoordschange_polarisation(t, &(particle->vx), &(particle->vy), &(particle->vz)); - - if (particle->sz != 0.0 || particle->sx != 0.0 || particle->sy != 0.0) - mccoordschange_polarisation(t, &(particle->sx), &(particle->sy), &(particle->sz)); -#elif MCCODE_PARTICLE_CODE == 22 - if (particle->kz != 0.0 || particle->kx != 0.0 || particle->ky != 0.0) - mccoordschange_polarisation(t, &(particle->kx), &(particle->ky), &(particle->kz)); - - if (particle->Ez != 0.0 || particle->Ex != 0.0 || particle->Ey != 0.0) - mccoordschange_polarisation(t, &(particle->Ex), &(particle->Ey), &(particle->Ez)); -#endif -} - -/******************************************************************************* -* mccoordschange_polarisation: applies rotation to vector (sx sy sz) -*******************************************************************************/ -void mccoordschange_polarisation(Rotation t, double *sx, double *sy, double *sz) -{ - Coords b, c; - - b.x = *sx; - b.y = *sy; - b.z = *sz; - c = rot_apply(t, b); - *sx = c.x; - *sy = c.y; - *sz = c.z; -} - -/* SECTION: vector math ==================================================== */ - -/* normal_vec_func: Compute normal vector to (x,y,z). */ -void normal_vec(double *nx, double *ny, double *nz, - double x, double y, double z) -{ - double ax = fabs(x); - double ay = fabs(y); - double az = fabs(z); - double l; - if(x == 0 && y == 0 && z == 0) - { - *nx = 0; - *ny = 0; - *nz = 0; - return; - } - if(ax < ay) - { - if(ax < az) - { /* Use X axis */ - l = sqrt(z*z + y*y); - *nx = 0; - *ny = z/l; - *nz = -y/l; - return; - } - } - else - { - if(ay < az) - { /* Use Y axis */ - l = sqrt(z*z + x*x); - *nx = z/l; - *ny = 0; - *nz = -x/l; - return; - } - } - /* Use Z axis */ - l = sqrt(y*y + x*x); - *nx = y/l; - *ny = -x/l; - *nz = 0; -} /* normal_vec */ - -/******************************************************************************* - * solve_2nd_order: second order equation solve: A*t^2 + B*t + C = 0 - * solve_2nd_order(&t1, NULL, A,B,C) - * returns 0 if no solution was found, or set 't1' to the smallest positive - * solution. - * solve_2nd_order(&t1, &t2, A,B,C) - * same as with &t2=NULL, but also returns the second solution. - * EXAMPLE usage for intersection of a trajectory with a plane in gravitation - * field (gx,gy,gz): - * The neutron starts at point r=(x,y,z) with velocityv=(vx vy vz). The plane - * has a normal vector n=(nx,ny,nz) and contains the point W=(wx,wy,wz). - * The problem consists in solving the 2nd order equation: - * 1/2.n.g.t^2 + n.v.t + n.(r-W) = 0 - * so that A = 0.5 n.g; B = n.v; C = n.(r-W); - * Without acceleration, t=-n.(r-W)/n.v - ******************************************************************************/ -int solve_2nd_order_old(double *t1, double *t2, - double A, double B, double C) -{ - int ret=0; - - if (!t1) return 0; - *t1 = 0; - if (t2) *t2=0; - - if (fabs(A) < 1E-10) /* approximate to linear equation: A ~ 0 */ - { - if (B) { *t1 = -C/B; ret=1; if (t2) *t2=*t1; } - /* else no intersection: A=B=0 ret=0 */ - } - else - { - double D; - D = B*B - 4*A*C; - if (D >= 0) /* Delta > 0: two solutions */ - { - double sD, dt1, dt2; - sD = sqrt(D); - dt1 = (-B + sD)/2/A; - dt2 = (-B - sD)/2/A; - /* we identify very small values with zero */ - if (fabs(dt1) < 1e-10) dt1=0.0; - if (fabs(dt2) < 1e-10) dt2=0.0; - - /* now we choose the smallest positive solution */ - if (dt1<=0.0 && dt2>0.0) ret=2; /* dt2 positive */ - else if (dt2<=0.0 && dt1>0.0) ret=1; /* dt1 positive */ - else if (dt1> 0.0 && dt2>0.0) - { if (dt1 < dt2) ret=1; else ret=2; } /* all positive: min(dt1,dt2) */ - /* else two solutions are negative. ret=-1 */ - if (ret==1) { *t1 = dt1; if (t2) *t2=dt2; } - else { *t1 = dt2; if (t2) *t2=dt1; } - ret=2; /* found 2 solutions and t1 is the positive one */ - } /* else Delta <0: no intersection. ret=0 */ - } - return(ret); -} /* solve_2nd_order */ - -int solve_2nd_order(double *t0, double *t1, double A, double B, double C){ - int retval=0; - double sign=copysign(1.0,B); - double dt0,dt1; - - dt0=0; - dt1=0; - if(t1){ *t1=0;} - - /*protect against rounding errors by locally equating DBL_EPSILON with 0*/ - if (fabs(A)=0){ - dt0=(-B - sign*sqrt(B*B-4*A*C))/(2*A); - dt1=C/(A*dt0); - retval=2; - }else{ - /*no real roots*/ - retval=0; - } - } - /*sort the solutions*/ - if (retval==1){ - /*put both solutions in t0 and t1*/ - *t0=dt0; - if(t1) *t1=dt1; - }else{ - /*we have two solutions*/ - /*swap if both are positive and t1 smaller than t0 or t1 the only positive*/ - int swap=0; - if(dt1>0 && ( dt1) - * - * If height or width is zero, choose random direction in full 4PI, no target. - * - * Traditionally, this routine had the name randvec_target_rect - this is now a - * a define (see mcstas-r.h) pointing here. If you use the old rouine, you are NOT - * taking the local emmission coordinate into account. -*******************************************************************************/ -void _randvec_target_rect_real(double *xo, double *yo, double *zo, double *solid_angle, - double xi, double yi, double zi, - double width, double height, Rotation A, - double lx, double ly, double lz, int order, - _class_particle* _particle) -{ - double dx, dy, dist, dist_p, nx, ny, nz, mx, my, mz, n_norm, m_norm; - double cos_theta; - Coords tmp; - Rotation Ainverse; - - rot_transpose(A, Ainverse); - - if(height == 0.0 || width == 0.0) - { - randvec_target_circle(xo, yo, zo, solid_angle, - xi, yi, zi, 0); - return; - } - else - { - /* Now choose point uniformly on rectangle within width x height */ - dx = width*randpm1()/2.0; - dy = height*randpm1()/2.0; - - /* Determine distance to target plane*/ - dist = sqrt(xi*xi + yi*yi + zi*zi); - /* Go to global coordinate system */ - - tmp = coords_set(xi, yi, zi); - tmp = rot_apply(Ainverse, tmp); - coords_get(tmp, &xi, &yi, &zi); - - /* Determine vector normal to trajectory axis (z) and gravity [0 1 0] */ - vec_prod(nx, ny, nz, xi, yi, zi, 0, 1, 0); - - /* This now defines the x-axis, normalize: */ - n_norm=sqrt(nx*nx + ny*ny + nz*nz); - nx = nx/n_norm; - ny = ny/n_norm; - nz = nz/n_norm; - - /* Now, determine our y-axis (vertical in many cases...) */ - vec_prod(mx, my, mz, xi, yi, zi, nx, ny, nz); - m_norm=sqrt(mx*mx + my*my + mz*mz); - mx = mx/m_norm; - my = my/m_norm; - mz = mz/m_norm; - - /* Our output, random vector can now be defined by linear combination: */ - - *xo = xi + dx * nx + dy * mx; - *yo = yi + dx * ny + dy * my; - *zo = zi + dx * nz + dy * mz; - - /* Go back to local coordinate system */ - tmp = coords_set(*xo, *yo, *zo); - tmp = rot_apply(A, tmp); - coords_get(tmp, &*xo, &*yo, &*zo); - - /* Go back to local coordinate system */ - tmp = coords_set(xi, yi, zi); - tmp = rot_apply(A, tmp); - coords_get(tmp, &xi, &yi, &zi); - - if (solid_angle) { - /* Calculate vector from local point to remote random point */ - lx = *xo - lx; - ly = *yo - ly; - lz = *zo - lz; - dist_p = sqrt(lx*lx + ly*ly + lz*lz); - - /* Adjust the 'solid angle' */ - /* 1/r^2 to the chosen point times cos(\theta) between the normal */ - /* vector of the target rectangle and direction vector of the chosen point. */ - cos_theta = (xi * lx + yi * ly + zi * lz) / (dist * dist_p); - *solid_angle = width * height / (dist_p * dist_p); - int counter; - for (counter = 0; counter < order; counter++) { - *solid_angle = *solid_angle * cos_theta; - } - } - } -} -/* randvec_target_rect_real */ - - -/* SECTION: random numbers ================================================== - - How to add a new RNG: - - - Use an rng with a manegable state vector, e.g. of lengt 4 or 7. The state - will sit on the particle struct as a "randstate_t state[RANDSTATE_LEN]" - - If the rng has a long state (as MT), set an empty "srandom" and initialize - it explicitly using the appropriate define (RNG_ALG) - - Add a seed and a random function (the transforms will be reused) - - Write the proper defines in mccode-r.h, e.g. randstate_t and RANDSTATE_LEN, - srandom and random. - - Compile using -DRNG_ALG= - -============================================================================= */ - - -/* "Mersenne Twister", by Makoto Matsumoto and Takuji Nishimura. */ -/* See http://www.math.keio.ac.jp/~matumoto/emt.html for original source. */ -/* - A C-program for MT19937, with initialization improved 2002/1/26. - Coded by Takuji Nishimura and Makoto Matsumoto. - - Before using, initialize the state by using mt_srandom(seed) - or init_by_array(init_key, key_length). - - Copyright (C) 1997 - 2002, Makoto Matsumoto and Takuji Nishimura, - All rights reserved. - - Redistribution and use in source and binary forms, with or without - modification, are permitted provided that the following conditions - are met: - - 1. Redistributions of source code must retain the above copyright - notice, this list of conditions and the following disclaimer. - - 2. Redistributions in binary form must reproduce the above copyright - notice, this list of conditions and the following disclaimer in the - documentation and/or other materials provided with the distribution. - - 3. The names of its contributors may not be used to endorse or promote - products derived from this software without specific prior written - permission. - - THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR - A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR - CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, - EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, - PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR - PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF - LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING - NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS - SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. - - - Any feedback is very welcome. - http://www.math.keio.ac.jp/matumoto/emt.html - email: matumoto@math.keio.ac.jp -*/ -#include -/* Period parameters */ -#define N 624 -#define M 397 -#define MATRIX_A 0x9908b0dfUL /* constant vector a */ -#define UPPER_MASK 0x80000000UL /* most significant w-r bits */ -#define LOWER_MASK 0x7fffffffUL /* least significant r bits */ - -unsigned long mt[N]; /* the array for the state vector */ -int mti=N+1; /* mti==N+1 means mt[N] is not initialized */ - -// required for common rng alg interface (see RNG_ALG usage in mccode-r.h) -void mt_srandom_empty() {} - -// initializes mt[N] with a seed -void mt_srandom(unsigned long s) -{ - mt[0]= s & 0xffffffffUL; - for (mti=1; mti> 30)) + mti); - /* See Knuth TAOCP Vol2. 3rd Ed. P.106 for multiplier. */ - /* In the previous versions, MSBs of the seed affect */ - /* only MSBs of the array mt[]. */ - /* 2002/01/09 modified by Makoto Matsumoto */ - mt[mti] &= 0xffffffffUL; - /* for >32 bit machines */ - } -} -/* Initialize by an array with array-length. - Init_key is the array for initializing keys. - key_length is its length. */ -void init_by_array(unsigned long init_key[], unsigned long key_length) -{ - int i, j, k; - mt_srandom(19650218UL); - i=1; j=0; - k = (N>key_length ? N : key_length); - for (; k; k--) { - mt[i] = (mt[i] ^ ((mt[i-1] ^ (mt[i-1] >> 30)) * 1664525UL)) - + init_key[j] + j; /* non linear */ - mt[i] &= 0xffffffffUL; /* for WORDSIZE > 32 machines */ - i++; j++; - if (i>=N) { mt[0] = mt[N-1]; i=1; } - if (j>=key_length) j=0; - } - for (k=N-1; k; k--) { - mt[i] = (mt[i] ^ ((mt[i-1] ^ (mt[i-1] >> 30)) * 1566083941UL)) - - i; /* non linear */ - mt[i] &= 0xffffffffUL; /* for WORDSIZE > 32 machines */ - i++; - if (i>=N) { mt[0] = mt[N-1]; i=1; } - } - - mt[0] = 0x80000000UL; /* MSB is 1; assuring non-zero initial array */ -} -/* generates a random number on [0,0xffffffff]-interval */ -unsigned long mt_random(void) -{ - unsigned long y; - unsigned long mag01[2]={0x0UL, MATRIX_A}; - /* mag01[x] = x * MATRIX_A for x=0,1 */ - - if (mti >= N) { /* generate N words at one time */ - int kk; - - if (mti == N+1) /* if mt_srandom() has not been called, */ - mt_srandom(5489UL); /* a default initial seed is used */ - - for (kk=0;kk> 1) ^ mag01[y & 0x1UL]; - } - for (;kk> 1) ^ mag01[y & 0x1UL]; - } - y = (mt[N-1]&UPPER_MASK)|(mt[0]&LOWER_MASK); - mt[N-1] = mt[M-1] ^ (y >> 1) ^ mag01[y & 0x1UL]; - - mti = 0; - } - - y = mt[mti++]; - - /* Tempering */ - y ^= (y >> 11); - y ^= (y << 7) & 0x9d2c5680UL; - y ^= (y << 15) & 0xefc60000UL; - y ^= (y >> 18); - - return y; -} -#undef N -#undef M -#undef MATRIX_A -#undef UPPER_MASK -#undef LOWER_MASK -/* End of "Mersenne Twister". */ - - -/* -KISS - - From: http://www.helsbreth.org/random/rng_kiss.html - Scott Nelson 1999 - - Based on Marsaglia's KISS or (KISS+SWB) - - KISS - Keep it Simple Stupid PRNG - - the idea is to use simple, fast, individually promising - generators to get a composite that will be fast, easy to code - have a very long period and pass all the tests put to it. - The three components of KISS are - x(n)=a*x(n-1)+1 mod 2^32 - y(n)=y(n-1)(I+L^13)(I+R^17)(I+L^5), - z(n)=2*z(n-1)+z(n-2) +carry mod 2^32 - The y's are a shift register sequence on 32bit binary vectors - period 2^32-1; - The z's are a simple multiply-with-carry sequence with period - 2^63+2^32-1. The period of KISS is thus - 2^32*(2^32-1)*(2^63+2^32-1) > 2^127 -*/ - -/* the KISS state is stored as a vector of 7 unsigned long */ -/* 0 1 2 3 4 5 6 */ -/* [ x, y, z, w, carry, k, m ] */ - -unsigned long *kiss_srandom(unsigned long state[7], unsigned long seed) { - if (seed == 0) seed = 1; - state[0] = seed | 1; // x - state[1] = seed | 2; // y - state[2] = seed | 4; // z - state[3] = seed | 8; // w - state[4] = 0; // carry - return 0; -} - -unsigned long kiss_random(unsigned long state[7]) { - state[0] = state[0] * 69069 + 1; - state[1] ^= state[1] << 13; - state[1] ^= state[1] >> 17; - state[1] ^= state[1] << 5; - state[5] = (state[2] >> 2) + (state[3] >> 3) + (state[4] >> 2); - state[6] = state[3] + state[3] + state[2] + state[4]; - state[2] = state[3]; - state[3] = state[6]; - state[4] = state[5] >> 30; - return state[0] + state[1] + state[3]; -} -/* end of "KISS" rng */ - - -/* FAST KISS in another implementation (Hundt) */ - -////////////////////////////////////////////////////////////////////////////// -// fast keep it simple stupid generator -////////////////////////////////////////////////////////////////////////////// - -///////////////////////////////////////////////////////////////////////////// -// Thomas Mueller hash for initialization of rngs -// http://stackoverflow.com/questions/664014/ -// what-integer-hash-function-are-good-that-accepts-an-integer-hash-key -////////////////////////////////////////////////////////////////////////////// -randstate_t _hash(randstate_t x) { - x = ((x >> 16) ^ x) * 0x45d9f3b; - x = ((x >> 16) ^ x) * 0x45d9f3b; - x = ((x >> 16) ^ x); - return x; -} - - -// SECTION: random number transforms ========================================== - - - -// generate a random number from normal law -double _randnorm(randstate_t* state) -{ - static double v1, v2, s; /* removing static breaks comparison with McStas <= 2.5 */ - static int phase = 0; - double X, u1, u2; - - if(phase == 0) - { - do - { - u1 = _rand01(state); - u2 = _rand01(state); - v1 = 2*u1 - 1; - v2 = 2*u2 - 1; - s = v1*v1 + v2*v2; - } while(s >= 1 || s == 0); - - X = v1*sqrt(-2*log(s)/s); - } - else - { - X = v2*sqrt(-2*log(s)/s); - } - - phase = 1 - phase; - return X; -} -// another one -double _randnorm2(randstate_t* state) { - double x, y, r; - do { - x = 2.0 * _rand01(state) - 1.0; - y = 2.0 * _rand01(state) - 1.0; - r = x*x + y*y; - } while (r == 0.0 || r >= 1.0); - return x * sqrt((-2.0 * log(r)) / r); -} - -// Generate a random number from -1 to 1 with triangle distribution -double _randtriangle(randstate_t* state) { - double randnum = _rand01(state); - if (randnum>0.5) return(1-sqrt(2*(randnum-0.5))); - else return(sqrt(2*randnum)-1); -} -double _rand01(randstate_t* state) { - double randnum; - randnum = (double) _random(); - // TODO: can we mult instead of div? - randnum /= (double) MC_RAND_MAX + 1; - return randnum; -} -// Return a random number between 1 and -1 -double _randpm1(randstate_t* state) { - double randnum; - randnum = (double) _random(); - randnum /= ((double) MC_RAND_MAX + 1) / 2; - randnum -= 1; - return randnum; -} -// Return a random number between 0 and max. -double _rand0max(double max, randstate_t* state) { - double randnum; - randnum = (double) _random(); - randnum /= ((double) MC_RAND_MAX + 1) / max; - return randnum; -} -// Return a random number between min and max. -double _randminmax(double min, double max, randstate_t* state) { - return _rand0max(max - min, state) + max; -} - - -/* SECTION: main and signal handlers ======================================== */ - -/******************************************************************************* -* mchelp: displays instrument executable help with possible options -*******************************************************************************/ -static void -mchelp(char *pgmname) -{ - int i; - - fprintf(stderr, "%s (%s) instrument simulation, generated with " MCCODE_STRING " (" MCCODE_DATE ")\n", instrument_name, instrument_source); - fprintf(stderr, "Usage: %s [options] [parm=value ...]\n", pgmname); - fprintf(stderr, -"Options are:\n" -" -s SEED --seed=SEED Set random seed (must be != 0)\n" -" -n COUNT --ncount=COUNT Set number of particles to simulate.\n" -" -d DIR --dir=DIR Put all data files in directory DIR.\n" -" -t --trace Enable trace of " MCCODE_PARTICLE "s through instrument.\n" -" -g --gravitation Enable gravitation for all trajectories.\n" -" --no-output-files Do not write any data files.\n" -" -h --help Show this help message.\n" -" -i --info Detailed instrument information.\n" -" --list-parameters Print the instrument parameters to standard out\n" -" --meta-list Print names of components which defined metadata\n" -" --meta-defined COMP[:NAME] Print component defined metadata names, or (0,1) if NAME provided\n" -" --meta-type COMP:NAME Print metadata format type specified in definition\n" -" --meta-data COMP:NAME Print the metadata text\n" -" --source Show the instrument code which was compiled.\n" -#ifdef OPENACC -"\n" -" --vecsize OpenACC vector-size (default: 128)\n" -" --numgangs Number of OpenACC gangs (default: 7813)\n" -" --gpu_innerloop Maximum rays to process pr. OpenACC \n" -" kernel run (default: 2147483647)\n" -"\n" -#endif -"\n" -" --bufsiz Monitor_nD list/buffer-size (default: 1000000)\n" -" --format=FORMAT Output data files using FORMAT=" - FLAVOR_UPPER -#ifdef USE_NEXUS - " NEXUS" -#endif -"\n\n" -); -#ifdef USE_MPI - fprintf(stderr, - "This instrument has been compiled with MPI support.\n Use 'mpirun %s [options] [parm=value ...]'.\n", pgmname); -#endif -#ifdef OPENACC - fprintf(stderr, - "This instrument has been compiled with NVIDIA GPU support through OpenACC.\n Running on systems without such devices will lead to segfaults.\nFurter, fprintf, sprintf and printf have been removed from any component TRACE.\n"); -#endif - - if(numipar > 0) - { - fprintf(stderr, "Instrument parameters are:\n"); - for(i = 0; i < numipar; i++) - if (mcinputtable[i].val && strlen(mcinputtable[i].val)) - fprintf(stderr, " %-16s(%s) [default='%s']\n", mcinputtable[i].name, - (*mcinputtypes[mcinputtable[i].type].parminfo)(mcinputtable[i].name), - mcinputtable[i].val); - else - fprintf(stderr, " %-16s(%s)\n", mcinputtable[i].name, - (*mcinputtypes[mcinputtable[i].type].parminfo)(mcinputtable[i].name)); - } - -#ifndef NOSIGNALS - fprintf(stderr, "Known signals are: " -#ifdef SIGUSR1 - "USR1 (status) " -#endif -#ifdef SIGUSR2 - "USR2 (save) " -#endif -#ifdef SIGBREAK - "BREAK (save) " -#endif -#ifdef SIGTERM - "TERM (save and exit)" -#endif - "\n"); -#endif /* !NOSIGNALS */ -} /* mchelp */ - - -/* mcshowhelp: show help and exit with 0 */ -static void -mcshowhelp(char *pgmname) -{ - mchelp(pgmname); - exit(0); -} - -/* mcusage: display usage when error in input arguments and exit with 1 */ -static void -mcusage(char *pgmname) -{ - fprintf(stderr, "Error: incorrect command line arguments\n"); - mchelp(pgmname); - exit(1); -} - -/* mcenabletrace: enable trace/mcdisplay or error if requires recompile */ -static void -mcenabletrace(void) -{ - if(traceenabled) { - mcdotrace = 1; - #pragma acc update device ( mcdotrace ) - } else { - fprintf(stderr, - "Error: trace not enabled (mcenabletrace)\n" - "Please re-run the " MCCODE_NAME " compiler " - "with the --trace option, or rerun the\n" - "C compiler with the MC_TRACE_ENABLED macro defined.\n"); - exit(1); - } -} - -/******************************************************************************* -* mcreadparams: request parameters from the prompt (or use default) -*******************************************************************************/ -void -mcreadparams(void) -{ - int i,j,status; - static char buf[CHAR_BUF_LENGTH]; - char *p; - int len; - - MPI_MASTER(printf("Instrument parameters for %s (%s)\n", - instrument_name, instrument_source)); - - for(i = 0; mcinputtable[i].name != 0; i++) - { - do - { - MPI_MASTER( - if (mcinputtable[i].val && strlen(mcinputtable[i].val)) - printf("Set value of instrument parameter %s (%s) [default='%s']:\n", - mcinputtable[i].name, - (*mcinputtypes[mcinputtable[i].type].parminfo) - (mcinputtable[i].name), mcinputtable[i].val); - else - printf("Set value of instrument parameter %s (%s):\n", - mcinputtable[i].name, - (*mcinputtypes[mcinputtable[i].type].parminfo) - (mcinputtable[i].name)); - fflush(stdout); - ); -#ifdef USE_MPI - if(mpi_node_rank == mpi_node_root) - { - p = fgets(buf, CHAR_BUF_LENGTH, stdin); - if(p == NULL) - { - fprintf(stderr, "Error: empty input for paramater %s (mcreadparams)\n", mcinputtable[i].name); - exit(1); - } - } - else - p = buf; - MPI_Bcast(buf, CHAR_BUF_LENGTH, MPI_CHAR, mpi_node_root, MPI_COMM_WORLD); -#else /* !USE_MPI */ - p = fgets(buf, CHAR_BUF_LENGTH, stdin); - if(p == NULL) - { - fprintf(stderr, "Error: empty input for paramater %s (mcreadparams)\n", mcinputtable[i].name); - exit(1); - } -#endif /* USE_MPI */ - len = strlen(buf); - if (!len || (len == 1 && (buf[0] == '\n' || buf[0] == '\r'))) - { - if (mcinputtable[i].val && strlen(mcinputtable[i].val)) { - strncpy(buf, mcinputtable[i].val, CHAR_BUF_LENGTH); /* use default value */ - len = strlen(buf); - } - } - for(j = 0; j < 2; j++) - { - if(len > 0 && (buf[len - 1] == '\n' || buf[len - 1] == '\r')) - { - len--; - buf[len] = '\0'; - } - } - - status = (*mcinputtypes[mcinputtable[i].type].getparm) - (buf, mcinputtable[i].par); - if(!status) - { - (*mcinputtypes[mcinputtable[i].type].error)(mcinputtable[i].name, buf); - if (!mcinputtable[i].val || strlen(mcinputtable[i].val)) { - fprintf(stderr, " Change %s default value in instrument definition.\n", mcinputtable[i].name); - exit(1); - } - } - } while(!status); - } -} /* mcreadparams */ - -/******************************************************************************* -* mcparseoptions: parse command line arguments (options, parameters) -*******************************************************************************/ -void -mcparseoptions(int argc, char *argv[]) -{ - int i, j; - char *p; - int paramset = 0, *paramsetarray; - char *usedir=NULL; - - /* Add one to numipar to avoid allocating zero size memory block. */ - paramsetarray = (int*)malloc((numipar + 1)*sizeof(*paramsetarray)); - if(paramsetarray == NULL) - { - fprintf(stderr, "Error: insufficient memory (mcparseoptions)\n"); - exit(1); - } - for(j = 0; j < numipar; j++) - { - paramsetarray[j] = 0; - if (mcinputtable[j].val != NULL && strlen(mcinputtable[j].val)) - { - int status; - char buf[CHAR_BUF_LENGTH]; - strncpy(buf, mcinputtable[j].val, CHAR_BUF_LENGTH); - status = (*mcinputtypes[mcinputtable[j].type].getparm) - (buf, mcinputtable[j].par); - if(!status) fprintf(stderr, "Invalid '%s' default value %s in instrument definition (mcparseoptions)\n", mcinputtable[j].name, buf); - else paramsetarray[j] = 1; - } else { - (*mcinputtypes[mcinputtable[j].type].getparm) - (NULL, mcinputtable[j].par); - paramsetarray[j] = 0; - } - } - for(i = 1; i < argc; i++) - { - if(!strcmp("-s", argv[i]) && (i + 1) < argc) - mcsetseed(argv[++i]); - else if(!strncmp("-s", argv[i], 2)) - mcsetseed(&argv[i][2]); - else if(!strcmp("--seed", argv[i]) && (i + 1) < argc) - mcsetseed(argv[++i]); - else if(!strncmp("--seed=", argv[i], 7)) - mcsetseed(&argv[i][7]); - else if(!strcmp("-n", argv[i]) && (i + 1) < argc) - mcsetn_arg(argv[++i]); - else if(!strncmp("-n", argv[i], 2)) - mcsetn_arg(&argv[i][2]); - else if(!strcmp("--ncount", argv[i]) && (i + 1) < argc) - mcsetn_arg(argv[++i]); - else if(!strncmp("--ncount=", argv[i], 9)) - mcsetn_arg(&argv[i][9]); - else if(!strcmp("-d", argv[i]) && (i + 1) < argc) - usedir=argv[++i]; /* will create directory after parsing all arguments (end of this function) */ - else if(!strncmp("-d", argv[i], 2)) - usedir=&argv[i][2]; - else if(!strcmp("--dir", argv[i]) && (i + 1) < argc) - usedir=argv[++i]; - else if(!strncmp("--dir=", argv[i], 6)) - usedir=&argv[i][6]; - else if(!strcmp("-h", argv[i])) - mcshowhelp(argv[0]); - else if(!strcmp("--help", argv[i]) || !strcmp("--version", argv[i])) - mcshowhelp(argv[0]); - else if(!strcmp("-i", argv[i])) { - mcformat=FLAVOR_UPPER; - mcinfo(); - } - else if(!strcmp("--info", argv[i])) - mcinfo(); - else if (!strcmp("--list-parameters", argv[i])) - mcparameterinfo(); - else if (!strcmp("--meta-list", argv[i]) && ((i+1) >= argc || argv[i+1][0] == '-')){ - //printf("Components with metadata defined:\n"); - exit(metadata_table_print_all_components(num_metadata, metadata_table) == 0); - } - else if (!strcmp("--meta-defined", argv[i]) && (i+1) < argc){ - exit(metadata_table_print_component_keys(num_metadata, metadata_table, argv[i+1]) == 0); - } - else if (!strcmp("--meta-type", argv[i]) && (i+1) < argc){ - char * literal_type = metadata_table_type(num_metadata, metadata_table, argv[i+1]); - if (literal_type == NULL) exit(1); - printf("%s\n", literal_type); - exit(0); - } - else if (!strcmp("--meta-data", argv[i]) && (i+1) < argc){ - char * literal = metadata_table_literal(num_metadata, metadata_table, argv[i+1]); - if (literal == NULL) exit(1); - printf("%s\n", literal); - exit(0); - } - else if(!strcmp("-t", argv[i])) - mcenabletrace(); - else if(!strcmp("--trace", argv[i]) || !strcmp("--verbose", argv[i])) - mcenabletrace(); - else if(!strcmp("--gravitation", argv[i])) - mcgravitation = 1; - else if(!strcmp("-g", argv[i])) - mcgravitation = 1; - else if(!strncmp("--format=", argv[i], 9)) { - mcformat=&argv[i][9]; - } - else if(!strcmp("--format", argv[i]) && (i + 1) < argc) { - mcformat=argv[++i]; - } - else if(!strncmp("--vecsize=", argv[i], 10)) { - vecsize=atoi(&argv[i][10]); - } - else if(!strcmp("--vecsize", argv[i]) && (i + 1) < argc) { - vecsize=atoi(argv[++i]); - } - else if(!strcmp("--bufsiz", argv[i]) && (i + 1) < argc) { - MONND_BUFSIZ=atoi(argv[++i]); - } - else if(!strncmp("--numgangs=", argv[i], 11)) { - numgangs=atoi(&argv[i][11]); - } - else if(!strcmp("--numgangs", argv[i]) && (i + 1) < argc) { - numgangs=atoi(argv[++i]); - } - else if(!strncmp("--gpu_innerloop=", argv[i], 16)) { - gpu_innerloop=(long)strtod(&argv[i][16], NULL); - } - else if(!strcmp("--gpu_innerloop", argv[i]) && (i + 1) < argc) { - gpu_innerloop=(long)strtod(argv[++i], NULL); - } - - else if(!strcmp("--no-output-files", argv[i])) - mcdisable_output_files = 1; - else if(!strcmp("--source", argv[i])) { - printf("/* Source code %s from %s: */\n" - "/******************************************************************************/\n" - "%s\n" - "/******************************************************************************/\n" - "/* End of source code %s from %s */\n", - instrument_name, instrument_source, instrument_code, - instrument_name, instrument_source); - exit(1); - } - else if(argv[i][0] != '-' && (p = strchr(argv[i], '=')) != NULL) - { - *p++ = '\0'; - - for(j = 0; j < numipar; j++) - if(!strcmp(mcinputtable[j].name, argv[i])) - { - int status; - status = (*mcinputtypes[mcinputtable[j].type].getparm)(p, - mcinputtable[j].par); - if(!status || !strlen(p)) - { - (*mcinputtypes[mcinputtable[j].type].error) - (mcinputtable[j].name, p); - exit(1); - } - paramsetarray[j] = 1; - paramset = 1; - break; - } - if(j == numipar) - { /* Unrecognized parameter name */ - fprintf(stderr, "Error: unrecognized parameter %s (mcparseoptions)\n", argv[i]); - exit(1); - } - } - else if(argv[i][0] == '-') { - fprintf(stderr, "Error: unrecognized option argument %s (mcparseoptions). Ignored.\n", argv[i++]); - } - else { - fprintf(stderr, "Error: unrecognized argument %s (mcparseoptions). Aborting.\n", argv[i]); - mcusage(argv[0]); - } - } - if(!paramset) - mcreadparams(); /* Prompt for parameters if not specified. */ - else - { - for(j = 0; j < numipar; j++) - if(!paramsetarray[j]) - { - fprintf(stderr, "Error: Instrument parameter %s left unset (mcparseoptions)\n", - mcinputtable[j].name); - exit(1); - } - } - free(paramsetarray); -#ifdef USE_MPI - if (mcdotrace) mpi_node_count=1; /* disable threading when in trace mode */ -#endif - if (usedir && strlen(usedir) && !mcdisable_output_files) mcuse_dir(usedir); -} /* mcparseoptions */ - -#ifndef NOSIGNALS -/******************************************************************************* -* sighandler: signal handler that makes simulation stop, and save results -*******************************************************************************/ -void sighandler(int sig) -{ - /* MOD: E. Farhi, Sep 20th 2001: give more info */ - time_t t1, t0; -#define SIG_SAVE 0 -#define SIG_TERM 1 -#define SIG_STAT 2 -#define SIG_ABRT 3 - - printf("\n# " MCCODE_STRING ": [pid %i] Signal %i detected", getpid(), sig); -#ifdef USE_MPI - printf(" [proc %i]", mpi_node_rank); -#endif -#if defined(SIGUSR1) && defined(SIGUSR2) && defined(SIGKILL) - if (!strcmp(mcsig_message, "sighandler") && (sig != SIGUSR1) && (sig != SIGUSR2)) - { - printf("\n# Fatal : unrecoverable loop ! Suicide (naughty boy).\n"); - kill(0, SIGKILL); /* kill myself if error occurs within sighandler: loops */ - } -#endif - switch (sig) { -#ifdef SIGINT - case SIGINT : printf(" SIGINT (interrupt from terminal, Ctrl-C)"); sig = SIG_TERM; break; -#endif -#ifdef SIGILL - case SIGILL : printf(" SIGILL (Illegal instruction)"); sig = SIG_ABRT; break; -#endif -#ifdef SIGFPE - case SIGFPE : printf(" SIGFPE (Math Error)"); sig = SIG_ABRT; break; -#endif -#ifdef SIGSEGV - case SIGSEGV : printf(" SIGSEGV (Mem Error)"); sig = SIG_ABRT; break; -#endif -#ifdef SIGTERM - case SIGTERM : printf(" SIGTERM (Termination)"); sig = SIG_TERM; break; -#endif -#ifdef SIGABRT - case SIGABRT : printf(" SIGABRT (Abort)"); sig = SIG_ABRT; break; -#endif -#ifdef SIGQUIT - case SIGQUIT : printf(" SIGQUIT (Quit from terminal)"); sig = SIG_TERM; break; -#endif -#ifdef SIGTRAP - case SIGTRAP : printf(" SIGTRAP (Trace trap)"); sig = SIG_ABRT; break; -#endif -#ifdef SIGPIPE - case SIGPIPE : printf(" SIGPIPE (Broken pipe)"); sig = SIG_ABRT; break; -#endif -#ifdef SIGUSR1 - case SIGUSR1 : printf(" SIGUSR1 (Display info)"); sig = SIG_STAT; break; -#endif -#ifdef SIGUSR2 - case SIGUSR2 : printf(" SIGUSR2 (Save simulation)"); sig = SIG_SAVE; break; -#endif -#ifdef SIGHUP - case SIGHUP : printf(" SIGHUP (Hangup/update)"); sig = SIG_SAVE; break; -#endif -#ifdef SIGBUS - case SIGBUS : printf(" SIGBUS (Bus error)"); sig = SIG_ABRT; break; -#endif -#ifdef SIGURG - case SIGURG : printf(" SIGURG (Urgent socket condition)"); sig = SIG_ABRT; break; -#endif -#ifdef SIGBREAK - case SIGBREAK: printf(" SIGBREAK (Break signal, Ctrl-Break)"); sig = SIG_SAVE; break; -#endif - default : printf(" (look at signal list for signification)"); sig = SIG_ABRT; break; - } - printf("\n"); - printf("# Simulation: %s (%s) \n", instrument_name, instrument_source); - printf("# Breakpoint: %s ", mcsig_message); - if (strstr(mcsig_message, "Save") && (sig == SIG_SAVE)) - sig = SIG_STAT; - SIG_MESSAGE("sighandler"); - if (mcget_ncount() == 0) - printf("(0 %%)\n" ); - else - { - printf("%.2f %% (%10.1f/%10.1f)\n", 100.0*mcget_run_num()/mcget_ncount(), 1.0*mcget_run_num(), 1.0*mcget_ncount()); - } - t0 = (time_t)mcstartdate; - t1 = time(NULL); - printf("# Date: %s", ctime(&t1)); - printf("# Started: %s", ctime(&t0)); - - if (sig == SIG_STAT) - { - printf("# " MCCODE_STRING ": Resuming simulation (continue)\n"); - fflush(stdout); - return; - } - else - if (sig == SIG_SAVE) - { - printf("# " MCCODE_STRING ": Saving data and resume simulation (continue)\n"); - save(NULL); - fflush(stdout); - return; - } - else - if (sig == SIG_TERM) - { - printf("# " MCCODE_STRING ": Finishing simulation (save results and exit)\n"); - finally(); - exit(0); - } - else - { - fflush(stdout); - perror("# Last I/O Error"); - printf("# " MCCODE_STRING ": Simulation stop (abort).\n"); -// This portion of the signal handling only works on UNIX -#if defined(__unix__) || defined(__APPLE__) - signal(sig, SIG_DFL); /* force to use default sighandler now */ - kill(getpid(), sig); /* and trigger it with the current signal */ -#endif - exit(-1); - } -#undef SIG_SAVE -#undef SIG_TERM -#undef SIG_STAT -#undef SIG_ABRT - -} /* sighandler */ -#endif /* !NOSIGNALS */ - -#ifdef NEUTRONICS -/*Main neutronics function steers the McStas calls, initializes parameters etc */ -/* Only called in case NEUTRONICS = TRUE */ -void neutronics_main_(float *inx, float *iny, float *inz, float *invx, float *invy, float *invz, float *intime, float *insx, float *insy, float *insz, float *inw, float *outx, float *outy, float *outz, float *outvx, float *outvy, float *outvz, float *outtime, float *outsx, float *outsy, float *outsz, float *outwgt) -{ - - extern double mcnx, mcny, mcnz, mcnvx, mcnvy, mcnvz; - extern double mcnt, mcnsx, mcnsy, mcnsz, mcnp; - - /* External code governs iteration - McStas is iterated once per call to neutronics_main. I.e. below counter must be initiancated for each call to neutronics_main*/ - mcrun_num=0; - - time_t t; - t = (time_t)mcstartdate; - mcstartdate = t; /* set start date before parsing options and creating sim file */ - init(); - - /* *** parse options *** */ - SIG_MESSAGE("[" __FILE__ "] main START"); - mcformat=getenv(FLAVOR_UPPER "_FORMAT") ? - getenv(FLAVOR_UPPER "_FORMAT") : FLAVOR_UPPER; - - /* Set neutron state based on input from neutronics code */ - mcsetstate(*inx,*iny,*inz,*invx,*invy,*invz,*intime,*insx,*insy,*insz,*inw); - - /* main neutron event loop - runs only one iteration */ - - //mcstas_raytrace(&mcncount); /* prior to McStas 1.12 */ - - mcallowbackprop = 1; //avoid absorbtion from negative dt - int argc=1; - char *argv[0]; - int dummy = mccode_main(argc, argv); - - *outx = mcnx; - *outy = mcny; - *outz = mcnz; - *outvx = mcnvx; - *outvy = mcnvy; - *outvz = mcnvz; - *outtime = mcnt; - *outsx = mcnsx; - *outsy = mcnsy; - *outsz = mcnsz; - *outwgt = mcnp; - - return; -} /* neutronics_main */ - -#endif /*NEUTRONICS*/ - -#endif /* !MCCODE_H */ -/* End of file "mccode-r.c". */ -/* end of file "/home/g/.cache/mccode-libc/main/mccode-r.c" */ -/* embedding file "/home/g/.cache/mccode-libc/main/nlib/mcstas-r.c" */ -/******************************************************************************* -* -* McStas, neutron ray-tracing package -* Copyright (C) 1997-2009, All rights reserved -* Risoe National Laboratory, Roskilde, Denmark -* Institut Laue Langevin, Grenoble, France -* -* Runtime: share/mcstas-r.c -* -* %Identification -* Written by: KN -* Date: Aug 29, 1997 -* Release: McStas X.Y -* Version: $Revision$ -* -* Runtime system for McStas. -* Embedded within instrument in runtime mode. -* -* Usage: Automatically embbeded in the c code whenever required. -* -* $Id$ -* -*******************************************************************************/ - -#ifndef MCSTAS_R_H -#include "mcstas-r.h" -#endif -#ifdef DANSE -#include "mcstas-globals.h" -#endif - -/******************************************************************************* -* The I/O format definitions and functions -*******************************************************************************/ - -/*the magnet stack*/ -#ifdef MC_POL_COMPAT -void (*mcMagnetPrecession) (double, double, double, double, double, double, - double, double*, double*, double*, double, Coords, Rotation)=NULL; -Coords mcMagnetPos; -Rotation mcMagnetRot; -double* mcMagnetData = NULL; -/* mcMagneticField(x, y, z, t, Bx, By, Bz) */ -int (*mcMagneticField) (double, double, double, double, - double*, double*, double*, void *) = NULL; -#endif - -#ifndef MCSTAS_H - -/******************************************************************************* -* mcsetstate: transfer parameters into global McStas variables -*******************************************************************************/ -_class_particle mcsetstate(double x, double y, double z, double vx, double vy, double vz, - double t, double sx, double sy, double sz, double p, int mcgravitation, void *mcMagnet, int mcallowbackprop) -{ - _class_particle mcneutron; - - mcneutron.x = x; - mcneutron.y = y; - mcneutron.z = z; - mcneutron.vx = vx; - mcneutron.vy = vy; - mcneutron.vz = vz; - mcneutron.t = t; - mcneutron.sx = sx; - mcneutron.sy = sy; - mcneutron.sz = sz; - mcneutron.p = p; - mcneutron.mcgravitation = mcgravitation; - mcneutron.mcMagnet = mcMagnet; - mcneutron.allow_backprop = mcallowbackprop; - mcneutron._uid = 0; - mcneutron._index = 1; - mcneutron._absorbed = 0; - mcneutron._restore = 0; - mcneutron._scattered = 0; - - return(mcneutron); -} /* mcsetstate */ - -/******************************************************************************* -* mcgetstate: get neutron parameters from particle structure -*******************************************************************************/ -_class_particle mcgetstate(_class_particle mcneutron, double *x, double *y, double *z, - double *vx, double *vy, double *vz, double *t, - double *sx, double *sy, double *sz, double *p) -{ - *x = mcneutron.x; - *y = mcneutron.y; - *z = mcneutron.z; - *vx = mcneutron.vx; - *vy = mcneutron.vy; - *vz = mcneutron.vz; - *t = mcneutron.t; - *sx = mcneutron.sx; - *sy = mcneutron.sy; - *sz = mcneutron.sz; - *p = mcneutron.p; - - return(mcneutron); -} /* mcgetstate */ - - -/******************************************************************************* -* mcgenstate: set default neutron parameters -*******************************************************************************/ -// Moved to generated code -/* #pragma acc routine seq */ -/* _class_particle mcgenstate(void) */ -/* { */ -/* return(mcsetstate(0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, mcgravitation, mcMagnet, mcallowbackprop)); */ -/* } */ - -/******************************************************************************* -* mccoordschanges: old style rotation routine rot -> (x y z) ,(vx vy vz),(sx,sy,sz) -*******************************************************************************/ -void -mccoordschanges(Coords a, Rotation t, double *x, double *y, double *z, - double *vx, double *vy, double *vz, double *sx, double *sy, double *sz) -{ - Coords b, c; - - b.x = *x; - b.y = *y; - b.z = *z; - c = rot_apply(t, b); - b = coords_add(c, a); - *x = b.x; - *y = b.y; - *z = b.z; - - if ( (vz && vy && vx) && (*vz != 0.0 || *vx != 0.0 || *vy != 0.0) ) - mccoordschange_polarisation(t, vx, vy, vz); - - if ( (sz && sy && sx) && (*sz != 0.0 || *sx != 0.0 || *sy != 0.0) ) - mccoordschange_polarisation(t, sx, sy, sz); - -} - -/* intersection routines ==================================================== */ - -/******************************************************************************* -* inside_rectangle: Check if (x,y) is inside rectangle (xwidth, yheight) -* return 0 if outside and 1 if inside -*******************************************************************************/ -int inside_rectangle(double x, double y, double xwidth, double yheight) -{ - if (x>-xwidth/2 && x-yheight/2 && y -dy/2 && y_in < dy/2 && z_in > -dz/2 && z_in < dz/2) - t[0] = tt; - else - t[0] = 0; - - tt = (dx/2 - x)/vx; - y_in = y + tt*vy; - z_in = z + tt*vz; - if( y_in > -dy/2 && y_in < dy/2 && z_in > -dz/2 && z_in < dz/2) - t[1] = tt; - else - t[1] = 0; - } - else - t[0] = t[1] = 0; - - if(vy != 0) - { - tt = -(dy/2 + y)/vy; - x_in = x + tt*vx; - z_in = z + tt*vz; - if( x_in > -dx/2 && x_in < dx/2 && z_in > -dz/2 && z_in < dz/2) - t[2] = tt; - else - t[2] = 0; - - tt = (dy/2 - y)/vy; - x_in = x + tt*vx; - z_in = z + tt*vz; - if( x_in > -dx/2 && x_in < dx/2 && z_in > -dz/2 && z_in < dz/2) - t[3] = tt; - else - t[3] = 0; - } - else - t[2] = t[3] = 0; - - if(vz != 0) - { - tt = -(dz/2 + z)/vz; - x_in = x + tt*vx; - y_in = y + tt*vy; - if( x_in > -dx/2 && x_in < dx/2 && y_in > -dy/2 && y_in < dy/2) - t[4] = tt; - else - t[4] = 0; - - tt = (dz/2 - z)/vz; - x_in = x + tt*vx; - y_in = y + tt*vy; - if( x_in > -dx/2 && x_in < dx/2 && y_in > -dy/2 && y_in < dy/2) - t[5] = tt; - else - t[5] = 0; - } - else - t[4] = t[5] = 0; - - /* The intersection is evaluated and *dt_in and *dt_out are assigned */ - - a = b = s = 0; - count = 0; - - for( i = 0; i < 6; i = i + 1 ) - if( t[i] == 0 ) - s = s+1; - else if( count == 0 ) - { - a = t[i]; - count = 1; - } - else - { - b = t[i]; - count = 2; - } - - if ( a == 0 && b == 0 ) - return 0; - else if( a < b ) - { - *dt_in = a; - *dt_out = b; - return 1; - } - else - { - *dt_in = b; - *dt_out = a; - return 1; - } - -} /* box_intersect */ - -/******************************************************************************* - * cylinder_intersect: compute intersection with a cylinder - * returns 0 when no intersection is found - * or 2/4/8/16 bits depending on intersection, - * and resulting times t0 and t1 - * Written by: EM,NB,ABA 4.2.98 - *******************************************************************************/ -int cylinder_intersect(double *t0, double *t1, double x, double y, double z, - double vx, double vy, double vz, double r, double h) -{ - double D, t_in, t_out, y_in, y_out; - int ret=1; - - D = (2*vx*x + 2*vz*z)*(2*vx*x + 2*vz*z) - - 4*(vx*vx + vz*vz)*(x*x + z*z - r*r); - - if (D>=0) - { - if (vz*vz + vx*vx) { - t_in = (-(2*vz*z + 2*vx*x) - sqrt(D))/(2*(vz*vz + vx*vx)); - t_out = (-(2*vz*z + 2*vx*x) + sqrt(D))/(2*(vz*vz + vx*vx)); - } else if (vy) { /* trajectory parallel to cylinder axis */ - t_in = (-h/2-y)/vy; - t_out = (h/2-y)/vy; - if (t_in>t_out){ - double tmp=t_in; - t_in=t_out;t_out=tmp; - } - } else return 0; - y_in = vy*t_in + y; - y_out =vy*t_out + y; - - if ( (y_in > h/2 && y_out > h/2) || (y_in < -h/2 && y_out < -h/2) ) - return 0; - else - { - if (y_in > h/2) - { t_in = ((h/2)-y)/vy; ret += 2; } - else if (y_in < -h/2) - { t_in = ((-h/2)-y)/vy; ret += 4; } - if (y_out > h/2) - { t_out = ((h/2)-y)/vy; ret += 8; } - else if (y_out < -h/2) - { t_out = ((-h/2)-y)/vy; ret += 16; } - } - *t0 = t_in; - *t1 = t_out; - return ret; - } - else - { - *t0 = *t1 = 0; - return 0; - } -} /* cylinder_intersect */ - - -/******************************************************************************* - * sphere_intersect: Calculate intersection between a line and a sphere. - * returns 0 when no intersection is found - * or 1 in case of intersection with resulting times t0 and t1 - *******************************************************************************/ -int sphere_intersect(double *t0, double *t1, double x, double y, double z, - double vx, double vy, double vz, double r) -{ - double A, B, C, D, v; - - v = sqrt(vx*vx + vy*vy + vz*vz); - A = v*v; - B = 2*(x*vx + y*vy + z*vz); - C = x*x + y*y + z*z - r*r; - D = B*B - 4*A*C; - if(D < 0) - return 0; - D = sqrt(D); - *t0 = (-B - D) / (2*A); - *t1 = (-B + D) / (2*A); - return 1; -} /* sphere_intersect */ - -/******************************************************************************* - * plane_intersect: Calculate intersection between a plane and a line. - * returns 0 when no intersection is found (i.e. line is parallel to the plane) - * returns 1 or -1 when intersection time is positive and negative respectively - *******************************************************************************/ -int plane_intersect(double *t, double x, double y, double z, - double vx, double vy, double vz, double nx, double ny, double nz, double wx, double wy, double wz) -{ - double s; - if (fabs(s=scalar_prod(nx,ny,nz,vx,vy,vz)) -#include -#include -#include - - typedef struct struct_table - { - char filename[1024]; - long filesize; - char *header; /* text header, e.g. comments */ - double *data; /* vector { x[0], y[0], ... x[n-1], y[n-1]... } */ - double min_x; /* min value of first column */ - double max_x; /* max value of first column */ - double step_x; /* minimal step value of first column */ - long rows; /* number of rows in matrix block */ - long columns; /* number of columns in matrix block */ - - long begin; /* start fseek index of block */ - long end; /* stop fseek index of block */ - long block_number; /* block index. 0 is catenation of all */ - long array_length; /* number of elements in the t_Table array */ - char monotonic; /* true when 1st column/vector data is monotonic */ - char constantstep; /* true when 1st column/vector data has constant step */ - char method[32]; /* interpolation method: nearest, linear */ - char quiet; /*output level for messages to the console 0: print all messages, 1:only print some/including errors, 2: never print anything.*/ - } t_Table; - -/*maximum number of rows to rebin a table = 1M*/ -enum { mcread_table_rebin_maxsize = 1000000 }; - -typedef struct t_Read_table_file_item { - int ref_count; - t_Table *table_ref; -} t_Read_table_file_item; - -typedef enum enum_Read_table_file_actions {STORE,FIND,GC} t_Read_table_file_actions; - -/* read_table-lib function prototypes */ -/* ========================================================================= */ - -/* 'public' functions */ -long Table_Read (t_Table *Table, char *File, long block_number); -long Table_Read_Offset (t_Table *Table, char *File, long block_number, - long *offset, long max_lines); -long Table_Read_Offset_Binary(t_Table *Table, char *File, char *Type, - long *Offset, long Rows, long Columns); -long Table_Rebin(t_Table *Table); /* rebin table with regular 1st column and interpolate all columns 2:end */ -long Table_Info (t_Table Table); -#pragma acc routine -double Table_Index(t_Table Table, long i, long j); /* get indexed value */ -#pragma acc routine -double Table_Value(t_Table Table, double X, long j); /* search X in 1st column and return interpolated value in j-column */ -t_Table *Table_Read_Array(char *File, long *blocks); -void Table_Free_Array(t_Table *Table); -long Table_Info_Array(t_Table *Table); -int Table_SetElement(t_Table *Table, long i, long j, double value); -long Table_Init(t_Table *Table, long rows, long columns); /* create a Table */ -#pragma acc routine -double Table_Value2d(t_Table Table, double X, double Y); /* same as Table_Index with non-integer indices and 2d interpolation */ -MCDETECTOR Table_Write(t_Table Table, char*file, char*xl, char*yl, - double x1, double x2, double y1, double y2); /* write Table to disk */ -void * Table_File_List_Handler(t_Read_table_file_actions action, void *item, void *item_modifier); -t_Table *Table_File_List_find(char *name, int block, int offset); -int Table_File_List_gc(t_Table *tab); -void *Table_File_List_store(t_Table *tab); - -#define Table_ParseHeader(header, ...) \ - Table_ParseHeader_backend(header,__VA_ARGS__,NULL); - -char **Table_ParseHeader_backend(char *header, ...); - -/* private functions */ -void Table_Free(t_Table *Table); -long Table_Read_Handle(t_Table *Table, FILE *fid, long block_number, long max_lines, char *name); -static void Table_Stat(t_Table *Table); -#pragma acc routine -double Table_Interp1d(double x, double x1, double y1, double x2, double y2); -#pragma acc routine -double Table_Interp1d_nearest(double x, double x1, double y1, double x2, double y2); -#pragma acc routine -double Table_Interp2d(double x, double y, double x1, double y1, double x2, double y2, -double z11, double z12, double z21, double z22); - - -#endif - -/* end of read_table-lib.h */ -/* Contents of ref-lib.h (requested from Guide_gravity) */ -/***************************************************************************** -* -* McStas, neutron ray-tracing package -* Copyright 1997-2006, All rights reserved -* Risoe National Laboratory, Roskilde, Denmark -* Institut Laue Langevin, Grenoble, France -* -* Library: share/ref-lib.h -* -* %Identification -* Written by: Peter Christiansen -* Date: August, 2006 -* Origin: RISOE -* Release: McStas 1.10 -* Version: $Revision$ -* -* Add StdDoubleReflecFunc, ExtendedReflecFunc -* Date: October, 2022 -* Locale: ESS -* Release: McStas 2.7x, 3.1x -* -* Commonly used reflection functions are declared in this file which -* are used by some guide and mirror components. -* -* Depends on read_table-lib -* -* Usage: within SHARE -* %include "ref-lib" -* -****************************************************************************/ - -#ifndef REF_LIB_H -#define REF_LIB_H "$Revision$" - -void StdReflecFunc(double, double*, double*); -void TableReflecFunc(double, t_Table*, double*); -void StdDoubleReflecFunc(double, double*, double*); -void ExtendedReflecFunc(double, double*, double*); - -#endif - -/* end of ref-lib.h */ -/* ***************************************************************************** -* instrument 'splitRunTest' and components DECLARE -***************************************************************************** */ - -/* Instrument parameters: structure and a table for the initialisation - (Used in e.g. inputparse and I/O function (e.g. detector_out) */ - -struct _struct_instrument_parameters{ - double a1; - double a2; - double virtual_source_x; - double virtual_source_y; -}; -typedef struct _struct_instrument_parameters _class_instrument_parameters; - -struct _instrument_struct { - char _name[256]; /* the name of this instrument e.g. 'splitRunTest' */ -/* Counters per component instance */ - double counter_AbsorbProp[11]; /* absorbed events in PROP routines */ - double counter_N[11], counter_P[11], counter_P2[11]; /* event counters after each component instance */ - _class_particle _trajectory[11]; /* current trajectory for STORE/RESTORE */ -/* Components position table (absolute and relative coords) */ - Coords _position_relative[11]; /* positions of all components */ - Coords _position_absolute[11]; - _class_instrument_parameters _parameters; /* instrument parameters */ -} _instrument_var; -struct _instrument_struct *instrument = & _instrument_var; -#pragma acc declare create ( _instrument_var ) -#pragma acc declare create ( instrument ) -int numipar = 4; -struct mcinputtable_struct mcinputtable[] = { - "a1", &(_instrument_var._parameters.a1), instr_type_double, "0", "", - "a2", &(_instrument_var._parameters.a2), instr_type_double, "0", "", - "virtual_source_x", &(_instrument_var._parameters.virtual_source_x), instr_type_double, "0.05", "", - "virtual_source_y", &(_instrument_var._parameters.virtual_source_y), instr_type_double, "0.1", "", - NULL, NULL, instr_type_double, "", "" -}; -struct metadata_table_struct metadata_table[] = { - "", "", "", "" -}; -int num_metadata = 0; -/* ************************************************************************** */ -/* SHARE user declarations for all components */ -/* ************************************************************************** */ -/* Shared user declarations for all components types 'Guide_gravity'. */ - -#ifndef Gravity_guide_Version -#define Gravity_guide_Version "$Revision$" - -#ifndef PROP_GRAV_DT -#error McStas : You need PROP_GRAV_DT (McStas >= 1.4.3) to run this component -#endif - -/* -* G: (m/s^2) Gravitation acceleration along y axis [-9.81] -* Gx: (m/s^2) Gravitation acceleration along x axis [0] -* Gy: (m/s^2) Gravitation acceleration along y axis [-9.81] -* Gz: (m/s^2) Gravitation acceleration along z axis [0] -* mh: (1) m-value of material for left/right vert. mirrors -* mv: (1) m-value of material for top/bottom horz. mirrors -* mx: (1) m-value of material for left/right vert. mirrors -* my: (1) m-value of material for top/bottom horz. mirrors -*/ - - typedef struct Gravity_guide_Vars - { - double gx; - double gy; - double gz; - double nx[6], ny[6], nz[6]; - double wx[6], wy[6], wz[6]; - double A[6], norm_n2[6], norm_n[6]; - long N_reflection[7]; - double w1c, h1c; - double w2c, h2c; - double M[5]; - double Alpha[5]; - double nzC[5], norm_n2xy[5], Axy[5]; - double wav_lr, wav_tb, wav_z; - double chamfer_z, chamfer_lr, chamfer_tb; - char compcurname[256]; - double fc_freq, fc_phase; - double warnings; - } Gravity_guide_Vars_type; - - void Gravity_guide_Init(Gravity_guide_Vars_type *aVars, - MCNUM a_w1, MCNUM a_h1, MCNUM a_w2, MCNUM a_h2, MCNUM a_l, MCNUM a_R0, - MCNUM a_Qc, MCNUM a_alpha, MCNUM a_m, MCNUM a_W, MCNUM a_nslit, MCNUM a_d, - MCNUM a_Gx, MCNUM a_Gy, MCNUM a_Gz, - MCNUM a_mleft, MCNUM a_mright, MCNUM a_mtop, MCNUM a_mbottom, MCNUM a_nhslit, - MCNUM a_wavy_lr, MCNUM a_wavy_tb, MCNUM a_wavy_z, MCNUM a_wavy, - MCNUM a_chamfers_z, MCNUM a_chamfers_lr, MCNUM a_chamfers_tb, MCNUM a_chamfers, - MCNUM a_nu, MCNUM a_phase, MCNUM a_aleft, MCNUM a_aright, MCNUM a_atop, MCNUM a_abottom) - { - int i; - - for (i=0; i<7; aVars->N_reflection[i++] = 0); - for (i=0; i<5; aVars->M[i++] = 0); - for (i=0; i<5; aVars->Alpha[i++] = 0); - - aVars->gx = a_Gx; /* The gravitation vector in the current component axis system */ - aVars->gy = a_Gy; - aVars->gz = a_Gz; - aVars->warnings=0; - - if (a_nslit <= 0 || a_nhslit <= 0) { fprintf(stderr,"%s: Fatal: no channel in this guide (nhslit or nslit=0).\n", aVars->compcurname); exit(-1); } - if (a_d < 0) { fprintf(stderr,"%s: Fatal: subdividing walls have negative thickness in this guide (d<0).\n", aVars->compcurname); exit(-1); } - aVars->w1c = (a_w1 - (a_nslit-1) *a_d)/(double)a_nslit; - aVars->w2c = (a_w2 - (a_nslit-1) *a_d)/(double)a_nslit; - aVars->h1c = (a_h1 - (a_nhslit-1)*a_d)/(double)a_nhslit; - aVars->h2c = (a_h2 - (a_nhslit-1)*a_d)/(double)a_nhslit; - - for (i=0; i <= 4; aVars->M[i++]=a_m); - for (i=0; i <= 4; aVars->Alpha[i++]=a_alpha); - if (a_mleft >= 0) aVars->M[1] =a_mleft ; - if (a_mright >= 0) aVars->M[2] =a_mright ; - if (a_mtop >= 0) aVars->M[3] =a_mtop ; - if (a_mbottom >= 0) aVars->M[4] =a_mbottom; - if (a_aleft >= 0) aVars->Alpha[1] =a_aleft ; - if (a_aright >= 0) aVars->Alpha[2] =a_aright ; - if (a_atop >= 0) aVars->Alpha[3] =a_atop ; - if (a_abottom >= 0) aVars->Alpha[4] =a_abottom; - - /* n: normal vectors to surfaces */ - aVars->nx[1] = a_l; aVars->ny[1] = 0; aVars->nz[1] = 0.5*(aVars->w2c-aVars->w1c); /* 1:+X left */ - aVars->nx[2] = -a_l; aVars->ny[2] = 0; aVars->nz[2] = -aVars->nz[1]; /* 2:-X right */ - aVars->nx[3] = 0; aVars->ny[3] = a_l; aVars->nz[3] = 0.5*(aVars->h2c-aVars->h1c); /* 3:+Y top */ - aVars->nx[4] = 0; aVars->ny[4] = -a_l; aVars->nz[4] = -aVars->nz[3]; /* 4:-Y bottom */ - aVars->nx[5] = 0; aVars->ny[5] = 0; aVars->nz[5] = a_l; /* 5:+Z exit */ - aVars->nx[0] = 0; aVars->ny[0] = 0; aVars->nz[0] = -a_l; /* 0:Z0 input */ - /* w: a point on these surfaces */ - aVars->wx[1] = +(aVars->w1c)/2; aVars->wy[1] = 0; aVars->wz[1] = 0; /* 1:+X left */ - aVars->wx[2] = -(aVars->w1c)/2; aVars->wy[2] = 0; aVars->wz[2] = 0; /* 2:-X right */ - aVars->wx[3] = 0; aVars->wy[3] = +(aVars->h1c)/2; aVars->wz[3] = 0; /* 3:+Y top */ - aVars->wx[4] = 0; aVars->wy[4] = -(aVars->h1c)/2; aVars->wz[4] = 0; /* 4:-Y bottom */ - aVars->wx[5] = 0; aVars->wy[5] = 0; aVars->wz[5] = a_l; /* 5:+Z exit */ - aVars->wx[0] = 0; aVars->wy[0] = 0; aVars->wz[0] = 0; /* 0:Z0 input */ - - for (i=0; i <= 5; i++) - { - aVars->A[i] = scalar_prod(aVars->nx[i], aVars->ny[i], aVars->nz[i], aVars->gx, aVars->gy, aVars->gz)/2; - aVars->norm_n2[i] = aVars->nx[i]*aVars->nx[i] + aVars->ny[i]*aVars->ny[i] + aVars->nz[i]*aVars->nz[i]; - if (aVars->norm_n2[i] <= 0) - { fprintf(stderr,"%s: Fatal: normal vector norm %i is null/negative ! check guide dimensions.\n", aVars->compcurname, i); exit(-1); } /* should never occur */ - else - aVars->norm_n[i] = sqrt(aVars->norm_n2[i]); - } - /* partial computations for l/r/t/b sides, to save computing time */ - for (i=1; i <= 4; i++) - { /* stores nz that changes in case non box element (focus/defocus) */ - aVars->nzC[i] = aVars->nz[i]; /* partial xy terms */ - aVars->norm_n2xy[i]= aVars->nx[i]*aVars->nx[i] + aVars->ny[i]*aVars->ny[i]; - aVars->Axy[i] = (aVars->nx[i]*aVars->gx + aVars->ny[i]*aVars->gy)/2; - } - /* handle waviness init */ - if (a_wavy && (!a_wavy_tb && !a_wavy_lr && !a_wavy_z)) - { aVars->wav_tb=aVars->wav_lr=aVars->wav_z=a_wavy; } - else - { aVars->wav_tb=a_wavy_tb; aVars->wav_lr=a_wavy_lr; aVars->wav_z=a_wavy_z; } - aVars->wav_tb *= DEG2RAD/(sqrt(8*log(2))); /* Convert from deg FWHM to rad Gaussian sigma */ - aVars->wav_lr *= DEG2RAD/(sqrt(8*log(2))); - aVars->wav_z *= DEG2RAD/(sqrt(8*log(2))); - /* handle chamfers init */ - if (a_chamfers && (!a_chamfers_z && !a_chamfers_lr && !a_chamfers_tb)) - { aVars->chamfer_z=aVars->chamfer_lr=aVars->chamfer_tb=a_chamfers; } - else - { - aVars->chamfer_z=a_chamfers_z; - aVars->chamfer_lr=a_chamfers_lr; - aVars->chamfer_tb=a_chamfers_tb; - } - - aVars->fc_freq = a_nu; - aVars->fc_phase = a_phase; - } - -#pragma acc routine seq - int Gravity_guide_Trace(double *dt, - Gravity_guide_Vars_type *aVars, - double cx, double cy, double cz, - double cvx, double cvy, double cvz, - double cxnum, double cxk, double cynum, double cyk, - double *cnx, double *cny, double *cnz, _class_particle* _particle) - { - double B, C; - int ret=0; - int side=0; - double n1; - double dt0, dt_min=0; - int i; - double loc_num, loc_nslit; - int i_slope=3; - - /* look if there is a previous intersection with guide sides */ - /* A = 0.5 n.g; B = n.v; C = n.(r-W); */ - /* 5=+Z side: n=(0, 0, -l) ; W = (0, 0, l) (at z=l, guide exit)*/ - B = aVars->nz[5]*cvz; C = aVars->nz[5]*(cz - aVars->wz[5]); - ret = solve_2nd_order(&dt0, NULL, aVars->A[5], B, C); - if (ret && dt0>1e-10) { dt_min = dt0; side=5; } - - loc_num = cynum; loc_nslit = cyk; - for (i=4; i>0; i--) - { - if (i == 2) { i_slope=1; loc_num = cxnum; loc_nslit = cxk; } - - if (aVars->nzC[i_slope] != 0) { - n1 = loc_nslit - 2*(loc_num); /* slope of l/r/u/d sides depends on the channel ! */ - loc_num++; /* use partial computations to alter nz and A */ - aVars->nz[i]= aVars->nzC[i]*n1; - aVars->A[i] = aVars->Axy[i] + aVars->nz[i]*aVars->gz/2; - } - if (i < 3) - { B = aVars->nx[i]*cvx + aVars->nz[i]*cvz; C = aVars->nx[i]*(cx-aVars->wx[i]) + aVars->nz[i]*cz; } - else { B = aVars->ny[i]*cvy + aVars->nz[i]*cvz; C = aVars->ny[i]*(cy-aVars->wy[i]) + aVars->nz[i]*cz; } - ret = solve_2nd_order(&dt0, NULL, aVars->A[i], B, C); - if (ret && dt0>1e-10 && (dt0nzC[i] != 0) - { aVars->norm_n2[i] = aVars->norm_n2xy[i] + aVars->nz[i]*aVars->nz[i]; - aVars->norm_n[i] = sqrt(aVars->norm_n2[i]); } - } - } - - *dt = dt_min; - /* handles waviness: rotate n vector */ - if (side > 0 && side < 5 && (aVars->wav_z || aVars->wav_lr || aVars->wav_tb)) - { - double nt_x, nt_y, nt_z; /* transverse vector */ - double nn_x, nn_y, nn_z; /* normal vector (tmp) */ - double phi; - /* normal vector n_z = [ 0,0,1], n_t = n x n_z; */ - vec_prod(nt_x,nt_y,nt_z, aVars->nx[side],aVars->ny[side],aVars->nz[side], 0,0,1); - /* rotate n with angle wavy_z around n_t -> nn */ - if (aVars->wav_z) { - phi = aVars->wav_z; - rotate(nn_x,nn_y,nn_z, aVars->nx[side],aVars->ny[side],aVars->nz[side], aVars->wav_z*randnorm(), nt_x,nt_y,nt_z); - } else { nn_x=aVars->nx[side]; nn_y=aVars->ny[side]; nn_z=aVars->nz[side]; } - /* rotate n with angle wavy_{x|y} around n_z -> nt */ - phi = (side <=2) ? aVars->wav_lr : aVars->wav_tb; - if (phi) { - rotate(nt_x,nt_y,nt_z, nn_x,nn_y,nn_z, phi*randnorm(), 0,0,1); - } else { nt_x=nn_x; nt_y=nn_y; nt_z=nn_z; } - *cnx=nt_x; *cny=nt_y; *cnz=nt_z; - } else - { *cnx=aVars->nx[side]; *cny=aVars->ny[side]; *cnz=aVars->nz[side]; } - return (side); - } - -#endif - -/* Shared user declarations for all components types 'Slit'. */ - -void slit_print_if(int condition, char* level, char* message, char* component){ - if (condition) fprintf(stderr, "Slit: %s: %s: %s\n", component, level, message); -} -void slit_error_if(int condition, char* message, char* component){ - slit_print_if(condition, "Error", message, component); - if (condition) exit(-1); -} -void slit_warning_if(int condition, char* message, char* component){ - slit_print_if(condition, "Warning", message, component); -} - -/* Shared user declarations for all components types 'Monochromator_curved'. */ - -#pragma acc routine -double GAUSS_monocurved(double x, double mean, double rms){ - return (exp(-((x)-(mean))*((x)-(mean))/(2*(rms)*(rms)))/(sqrt(2*PI)*(rms))); -} - -/* ************************************************************************** */ -/* End of SHARE user declarations for all components */ -/* ************************************************************************** */ -/* ********************** component definition declarations. **************** */ -/* Parameter definition for component type 'Arm' */ -struct _struct_Arm_parameters { - /* Component type 'Arm' setting parameters */ -/* Component type 'Arm' private parameters */ - char Arm_has_no_parameters; -}; /* _struct_Arm_parameters */ -typedef struct _struct_Arm_parameters _class_Arm_parameters; - -/* Parameters for component type 'Arm' */ -struct _struct_Arm { - char _name[256]; /* e.g. instance of Arm name */ - char _type[4]; /* Arm */ - long _index; /* index in TRACE list */ - Coords _position_absolute; - Coords _position_relative; /* wrt PREVIOUS */ - Rotation _rotation_absolute; - Rotation _rotation_relative; /* wrt PREVIOUS */ - int _rotation_is_identity; - int _position_relative_is_zero; - _class_Arm_parameters _parameters; -}; -typedef struct _struct_Arm _class_Arm; -/* Parameter definition for component type 'Source_simple' */ -struct _struct_Source_simple_parameters { - /* Component type 'Source_simple' setting parameters */ - double radius; - double yheight; - double xwidth; - double dist; - double focus_xw; - double focus_yh; - double E0; - double dE; - double lambda0; - double dlambda; - double flux; - double gauss; - int target_index; -/* Component type 'Source_simple' private parameters */ - double pmul; /* Not initialized */ - double srcArea; /* Not initialized */ - int square; /* Not initialized */ - double tx; /* Not initialized */ - double ty; /* Not initialized */ - double tz; /* Not initialized */ -}; /* _struct_Source_simple_parameters */ -typedef struct _struct_Source_simple_parameters _class_Source_simple_parameters; - -/* Parameters for component type 'Source_simple' */ -struct _struct_Source_simple { - char _name[256]; /* e.g. instance of Source_simple name */ - char _type[14]; /* Source_simple */ - long _index; /* index in TRACE list */ - Coords _position_absolute; - Coords _position_relative; /* wrt PREVIOUS */ - Rotation _rotation_absolute; - Rotation _rotation_relative; /* wrt PREVIOUS */ - int _rotation_is_identity; - int _position_relative_is_zero; - _class_Source_simple_parameters _parameters; -}; -typedef struct _struct_Source_simple _class_Source_simple; -/* Parameter definition for component type 'Guide_gravity' */ -struct _struct_Guide_gravity_parameters { - /* Component type 'Guide_gravity' setting parameters */ - double w1; - double h1; - double w2; - double h2; - double l; - double R0; - double Qc; - double alpha; - double m; - double W; - double nslit; - double d; - double mleft; - double mright; - double mtop; - double mbottom; - double nhslit; - double G; - double aleft; - double aright; - double atop; - double abottom; - double wavy; - double wavy_z; - double wavy_tb; - double wavy_lr; - double chamfers; - double chamfers_z; - double chamfers_lr; - double chamfers_tb; - double nelements; - double nu; - double phase; - char reflect[16384]; -/* Component type 'Guide_gravity' private parameters */ - Gravity_guide_Vars_type GVars; /* Not initialized */ - t_Table pTable; /* Not initialized */ - int table_present; /* Not initialized */ -}; /* _struct_Guide_gravity_parameters */ -typedef struct _struct_Guide_gravity_parameters _class_Guide_gravity_parameters; - -/* Parameters for component type 'Guide_gravity' */ -struct _struct_Guide_gravity { - char _name[256]; /* e.g. instance of Guide_gravity name */ - char _type[14]; /* Guide_gravity */ - long _index; /* index in TRACE list */ - Coords _position_absolute; - Coords _position_relative; /* wrt PREVIOUS */ - Rotation _rotation_absolute; - Rotation _rotation_relative; /* wrt PREVIOUS */ - int _rotation_is_identity; - int _position_relative_is_zero; - _class_Guide_gravity_parameters _parameters; -}; -typedef struct _struct_Guide_gravity _class_Guide_gravity; -/* Parameter definition for component type 'Slit' */ -struct _struct_Slit_parameters { - /* Component type 'Slit' setting parameters */ - double xmin; - double xmax; - double ymin; - double ymax; - double radius; - double xwidth; - double yheight; -/* Component type 'Slit' private parameters */ - char isradial; /* Not initialized */ -}; /* _struct_Slit_parameters */ -typedef struct _struct_Slit_parameters _class_Slit_parameters; - -/* Parameters for component type 'Slit' */ -struct _struct_Slit { - char _name[256]; /* e.g. instance of Slit name */ - char _type[5]; /* Slit */ - long _index; /* index in TRACE list */ - Coords _position_absolute; - Coords _position_relative; /* wrt PREVIOUS */ - Rotation _rotation_absolute; - Rotation _rotation_relative; /* wrt PREVIOUS */ - int _rotation_is_identity; - int _position_relative_is_zero; - _class_Slit_parameters _parameters; -}; -typedef struct _struct_Slit _class_Slit; -/* Parameter definition for component type 'Monochromator_curved' */ -struct _struct_Monochromator_curved_parameters { - /* Component type 'Monochromator_curved' setting parameters */ - char reflect[16384]; - char transmit[16384]; - double zwidth; - double yheight; - double gap; - int NH; - int NV; - double mosaich; - double mosaicv; - double r0; - double t0; - double Q; - double RV; - double RH; - double DM; - double mosaic; - double width; - double height; - double verbose; - double order; -/* Component type 'Monochromator_curved' private parameters */ - double mos_rms_y; /* Not initialized */ - double mos_rms_z; /* Not initialized */ - double mos_rms_max; /* Not initialized */ - double mono_Q; /* Not initialized */ - double SlabWidth; /* Not initialized */ - double SlabHeight; /* Not initialized */ - t_Table rTable; /* Not initialized */ - t_Table tTable; /* Not initialized */ - int rTableFlag; /* Not initialized */ - int tTableFlag; /* Not initialized */ - double *tiltH; /* Not initialized */ - double *tiltV; /* Not initialized */ -}; /* _struct_Monochromator_curved_parameters */ -typedef struct _struct_Monochromator_curved_parameters _class_Monochromator_curved_parameters; - -/* Parameters for component type 'Monochromator_curved' */ -struct _struct_Monochromator_curved { - char _name[256]; /* e.g. instance of Monochromator_curved name */ - char _type[21]; /* Monochromator_curved */ - long _index; /* index in TRACE list */ - Coords _position_absolute; - Coords _position_relative; /* wrt PREVIOUS */ - Rotation _rotation_absolute; - Rotation _rotation_relative; /* wrt PREVIOUS */ - int _rotation_is_identity; - int _position_relative_is_zero; - _class_Monochromator_curved_parameters _parameters; -}; -typedef struct _struct_Monochromator_curved _class_Monochromator_curved; -/* Parameter definition for component type 'Monitor' */ -struct _struct_Monitor_parameters { - /* Component type 'Monitor' setting parameters */ - double xmin; - double xmax; - double ymin; - double ymax; - double xwidth; - double yheight; - double restore_neutron; -/* Component type 'Monitor' private parameters */ - double Nsum; /* Not initialized */ - double psum; /* Not initialized */ - double p2sum; /* Not initialized */ -}; /* _struct_Monitor_parameters */ -typedef struct _struct_Monitor_parameters _class_Monitor_parameters; - -/* Parameters for component type 'Monitor' */ -struct _struct_Monitor { - char _name[256]; /* e.g. instance of Monitor name */ - char _type[8]; /* Monitor */ - long _index; /* index in TRACE list */ - Coords _position_absolute; - Coords _position_relative; /* wrt PREVIOUS */ - Rotation _rotation_absolute; - Rotation _rotation_relative; /* wrt PREVIOUS */ - int _rotation_is_identity; - int _position_relative_is_zero; - _class_Monitor_parameters _parameters; -}; -typedef struct _struct_Monitor _class_Monitor; -/* component origin = Arm() [1] DECLARE */ -_class_Arm _origin_var; -#pragma acc declare create ( _origin_var ) -/* component source = Source_simple() [2] DECLARE */ -_class_Source_simple _source_var; -#pragma acc declare create ( _source_var ) -/* component guide = Guide_gravity() [3] DECLARE */ -_class_Guide_gravity _guide_var; -#pragma acc declare create ( _guide_var ) -/* component guide_end = Arm() [4] DECLARE */ -_class_Arm _guide_end_var; -#pragma acc declare create ( _guide_end_var ) -/* component aperture = Slit() [5] DECLARE */ -_class_Slit _aperture_var; -#pragma acc declare create ( _aperture_var ) -/* component split_at = Arm() [6] DECLARE */ -_class_Arm _split_at_var; -#pragma acc declare create ( _split_at_var ) -/* component mono_point = Arm() [7] DECLARE */ -_class_Arm _mono_point_var; -#pragma acc declare create ( _mono_point_var ) -/* component mono = Monochromator_curved() [8] DECLARE */ -_class_Monochromator_curved _mono_var; -#pragma acc declare create ( _mono_var ) -/* component sample_arm = Arm() [9] DECLARE */ -_class_Arm _sample_arm_var; -#pragma acc declare create ( _sample_arm_var ) -/* component detector = Monitor() [10] DECLARE */ -_class_Monitor _detector_var; -#pragma acc declare create ( _detector_var ) -int mcNUMCOMP = 10; -/* Contents of read_table-lib.c (requested from ref-lib) */ -/******************************************************************************* -* -* McStas, neutron ray-tracing package -* Copyright (C) 1997-2009, All rights reserved -* Risoe National Laboratory, Roskilde, Denmark -* Institut Laue Langevin, Grenoble, France -* -* Library: share/read_table-lib.c -* -* %Identification -* Written by: EF -* Date: Aug 28, 2002 -* Origin: ILL -* Release: McStas CVS_090504 -* Version: $Revision$ -* -* This file is to be imported by components that may read data from table files -* It handles some shared functions. Embedded within instrument in runtime mode. -* -* Usage: within SHARE -* %include "read_table-lib" -* -*******************************************************************************/ - -#ifndef READ_TABLE_LIB_H -#include "read_table-lib.h" -#endif - - -/******************************************************************************* - * void *Table_File_List_Handler(action, item, item_modifier) - * ACTION: handle file entries in the read_table-lib file list. If a file is read - it is supposed to be - * stored in a list such that we can avoid reading the same file many times. - * input action: FIND, STORE, GC. check if file exists in the list, store an item in the list, or check if it can be garbage collected. - * input item: depends on the action. - * FIND) item is a filename, and item_modifier is the block number - * STORE) item is the Table to store - item_modifier is ignored - * GC) item is the Table to check. If it has a ref_count >1 then this is simply decremented. - * return depends on the action - * FIND) return a reference to a table+ref_count item if found - NULL otherwise. I.e. NULL means the file has not been read before and must be read again. - * STORE) return NULL always - * GC) return NULL if no garbage collection is needed, return an adress to the t_Table which should be garbage collected. 0x1 is returned if - * the item is not found in the list -*******************************************************************************/ -void * Table_File_List_Handler(t_Read_table_file_actions action, void *item, void *item_modifier){ - - /* logic here is Read_Table should include a call to FIND. If found the return value should just be used as - * if the table had been read from disk. If not found then read the table and STORE. - * Table_Free should include a call to GC. If this returns non-NULL then we should proceed with freeing the memory - * associated with the table item - otherwise only decrement the reference counter since there are more references - * that may need it.*/ - - static t_Read_table_file_item read_table_file_list[1024]; - static int read_table_file_count=0; - - t_Read_table_file_item *tr; - switch(action){ - case FIND: - /*interpret data item as a filename, if it is found return a pointer to the table and increment refcount. - * if not found return the item itself*/ - tr=read_table_file_list; - while ( tr->table_ref!=NULL ){ - int i=*((int*) item_modifier); - int j=*( ((int*) item_modifier)+1); - if ( !strcmp(tr->table_ref->filename,(char *) item) && - tr->table_ref->block_number==i && tr->table_ref->begin==j ){ - tr->ref_count++; - return (void *) tr; - } - tr++; - } - return NULL; - case STORE: - /*find an available slot and store references to table there*/ - tr=&(read_table_file_list[read_table_file_count++]); - tr->table_ref = ((t_Table *) item); - tr->ref_count++; - return NULL; - case GC: - /* Should this item be garbage collected (freed) - if so scratch the entry and return the address of the item - - * else decrement ref_count and return NULL. - * A non-NULL return expects the item to actually be freed afterwards.*/ - tr=read_table_file_list; - while ( tr->table_ref!=NULL ){ - if ( tr->table_ref->data ==((t_Table *)item)->data && - tr->table_ref->block_number == ((t_Table *)item)->block_number){ - /*matching item found*/ - if (tr->ref_count>1){ - /*the item is found and no garbage collection needed*/ - tr->ref_count--; - return NULL; - }else{ - /* The item is found and the reference counter is 1. - * This means we should garbage collect. Move remaining list items up one slot, - * and return the table for garbage collection by caller*/ - while (tr->table_ref!=NULL){ - *tr=*(tr+1); - tr++; - } - read_table_file_count--; - return (t_Table *) item; - } - } - tr++; - } - /* item not found, and so should be garbage collected. This could be the case if freeing a - * Table that has been constructed from code - not read from file. Return 0x1 to flag it for - * collection.*/ - return (void *) 0x1 ; - } - /* If we arrive here, nothing worked, return NULL */ - return NULL; -} - -/* Access functions to the handler*/ - -/******************************************** - * t_Table *Table_File_List_find(char *name, int block, int offset) - * input name: filename to search for in the file list - * input block: data block in the file as each file may contain more than 1 data block. - * return a ref. to a table if it is found (you may use this pointer and skip reading the file), NULL otherwise (i.e. go ahead and read the file) -*********************************************/ -t_Table *Table_File_List_find(char *name, int block, int offset){ - int vars[2]={block,offset}; - t_Read_table_file_item *item = Table_File_List_Handler(FIND,name, vars); - if (item == NULL){ - return NULL; - }else{ - return item->table_ref; - } -} -/******************************************** - * int Table_File_List_gc(t_Table *tab) - * input tab: the table to check for references. - * return 0: no garbage collection needed - * 1: Table's data and header (at least) should be freed. -*********************************************/ -int Table_File_List_gc(t_Table *tab){ - void *rval=Table_File_List_Handler(GC,tab,0); - if (rval==NULL) return 0; - else return 1; -} - - -/***************************************************************************** - * void *Table_File_List_store(t_Table *tab) - * input tab: pointer to table to store. - * return None. -*******************************************************************************/ -void *Table_File_List_store(t_Table *tab){ - return Table_File_List_Handler(STORE,tab,0); -} - - -/******************************************************************************* -* FILE *Open_File(char *name, char *Mode, char *path) -* ACTION: search for a file and open it. Optionally return the opened path. -* input name: file name from which table should be extracted -* mode: "r", "w", "a" or any valid fopen mode -* path: NULL or a pointer to at least 1024 allocated chars -* return initialized file handle or NULL in case of error -*******************************************************************************/ - - FILE *Open_File(char *File, const char *Mode, char *Path) - { - char path[1024]; - FILE *hfile = NULL; - - if (!File || File[0]=='\0') return(NULL); - if (!strcmp(File,"NULL") || !strcmp(File,"0")) return(NULL); - - /* search in current or full path */ - strncpy(path, File, 1024); - hfile = fopen(path, Mode); - if(!hfile) - { - char dir[1024]; - - if (!hfile && instrument_source[0] != '\0' && strlen(instrument_source)) /* search in instrument source location */ - { - char *path_pos = NULL; - /* extract path: searches for last file separator */ - path_pos = strrchr(instrument_source, MC_PATHSEP_C); /* last PATHSEP */ - if (path_pos) { - long path_length = path_pos +1 - instrument_source; /* from start to path+sep */ - if (path_length) { - strncpy(dir, instrument_source, path_length); - dir[path_length] = '\0'; - snprintf(path, 1024, "%s%c%s", dir, MC_PATHSEP_C, File); - hfile = fopen(path, Mode); - } - } - } - if (!hfile && instrument_exe[0] != '\0' && strlen(instrument_exe)) /* search in PWD instrument executable location */ - { - char *path_pos = NULL; - /* extract path: searches for last file separator */ - path_pos = strrchr(instrument_exe, MC_PATHSEP_C); /* last PATHSEP */ - if (path_pos) { - long path_length = path_pos +1 - instrument_exe; /* from start to path+sep */ - if (path_length) { - strncpy(dir, instrument_exe, path_length); - dir[path_length] = '\0'; - snprintf(path, 1024, "%s%c%s", dir, MC_PATHSEP_C, File); - hfile = fopen(path, Mode); - } - } - } - if (!hfile) /* search in HOME or . */ - { - strcpy(dir, getenv("HOME") ? getenv("HOME") : "."); - snprintf(path, 1024, "%s%c%s", dir, MC_PATHSEP_C, File); - hfile = fopen(path, Mode); - } - if (!hfile) /* search in MCSTAS/data */ - { - strcpy(dir, getenv(FLAVOR_UPPER) ? getenv(FLAVOR_UPPER) : MCSTAS); - snprintf(path, 1024, "%s%c%s%c%s", dir, MC_PATHSEP_C, "data", MC_PATHSEP_C, File); - hfile = fopen(path, Mode); - } - if (!hfile) /* search in MVCSTAS/contrib */ - { - strcpy(dir, getenv(FLAVOR_UPPER) ? getenv(FLAVOR_UPPER) : MCSTAS); - snprintf(path, 1024, "%s%c%s%c%s", dir, MC_PATHSEP_C, "contrib", MC_PATHSEP_C, File); - hfile = fopen(path, Mode); - } - if(!hfile) - { - // fprintf(stderr, "Warning: Could not open input file '%s' (Open_File)\n", File); - return (NULL); - } - } - if (Path) strncpy(Path, path, 1024); - return(hfile); - } /* end Open_File */ - -/******************************************************************************* -* long Read_Table(t_Table *Table, char *name, int block_number) -* ACTION: read a single Table from a text file -* input Table: pointer to a t_Table structure -* name: file name from which table should be extracted -* block_number: if the file does contain more than one -* data block, then indicates which one to get (from index 1) -* a 0 value means append/catenate all -* return initialized single Table t_Table structure containing data, header, ... -* number of read elements (-1: error, 0:header only) -* The routine stores any line starting with '#', '%' and ';' into the header -* File is opened, read and closed -* Other lines are interpreted as numerical data, and stored. -* Data block should be a rectangular matrix or vector. -* Data block may be rebinned with Table_Rebin (also sort in ascending order) -*******************************************************************************/ - long Table_Read(t_Table *Table, char *File, long block_number) - { /* reads all or a single data block from 'file' and returns a Table structure */ - return(Table_Read_Offset(Table, File, block_number, NULL, 0)); - } /* end Table_Read */ - -/******************************************************************************* -* long Table_Read_Offset(t_Table *Table, char *name, int block_number, long *offset -* long max_rows) -* ACTION: read a single Table from a text file, starting at offset -* Same as Table_Read(..) except: -* input offset: pointer to an offset (*offset should be 0 at start) -* max_rows: max number of data rows to read from file (0 means all) -* return initialized single Table t_Table structure containing data, header, ... -* number of read elements (-1: error, 0:header only) -* updated *offset position (where end of reading occured) -*******************************************************************************/ - long Table_Read_Offset(t_Table *Table, char *File, - long block_number, long *offset, - long max_rows) - { /* reads all/a data block in 'file' and returns a Table structure */ - FILE *hfile; - long nelements=0; - long begin=0; - long filesize=0; - char name[1024]; - char path[1024]; - struct stat stfile; - - /*Need to be able to store the pointer*/ - if (!Table) return(-1); - - //if (offset && *offset) snprintf(name, 1024, "%s@%li", File, *offset); - //else - strncpy(name, File, 1024); - if(offset && *offset){ - begin=*offset; - } - /* Check if the table has already been read from file. - * If so just reuse the table, if not (this is flagged by returning NULL - * set up a new table and read the data into it */ - t_Table *tab_p= Table_File_List_find(name,block_number,begin); - if ( tab_p!=NULL ){ - /*table was found in the Table_File_List*/ - *Table=*tab_p; - MPI_MASTER( - if(Table->quiet<1) - printf("Reusing input file '%s' (Table_Read_Offset)\n", name); - ); - return Table->rows*Table->columns; - } - - /* open the file */ - hfile = Open_File(File, "r", path); - if (!hfile) return(-1); - else { - MPI_MASTER( - if(Table->quiet<1) - printf("Opening input file '%s' (Table_Read_Offset)\n", path); - ); - } - - /* read file state */ - stat(path,&stfile); filesize = stfile.st_size; - if (offset && *offset) fseek(hfile, *offset, SEEK_SET); - begin = ftell(hfile); - - Table_Init(Table, 0, 0); - - /* read file content and set the Table */ - nelements = Table_Read_Handle(Table, hfile, block_number, max_rows, name); - Table->begin = begin; - Table->end = ftell(hfile); - Table->filesize = (filesize>0 ? filesize : 0); - Table_Stat(Table); - - Table_File_List_store(Table); - - if (offset) *offset=Table->end; - fclose(hfile); - return(nelements); - - } /* end Table_Read_Offset */ - -/******************************************************************************* -* long Table_Read_Offset_Binary(t_Table *Table, char *File, char *type, -* long *offset, long rows, long columns) -* ACTION: read a single Table from a binary file, starting at offset -* Same as Table_Read_Offset(..) except that it handles binary files. -* input type: may be "float"/NULL or "double" -* offset: pointer to an offset (*offset should be 0 at start) -* rows : number of rows (0 means read all) -* columns: number of columns -* return initialized single Table t_Table structure containing data, header, ... -* number of read elements (-1: error, 0:header only) -* updated *offset position (where end of reading occured) -*******************************************************************************/ - long Table_Read_Offset_Binary(t_Table *Table, char *File, char *type, - long *offset, long rows, long columns) - { /* reads all/a data block in binary 'file' and returns a Table structure */ - long nelements, sizeofelement; - long filesize; - FILE *hfile; - char path[1024]; - struct stat stfile; - double *data; - long i; - long begin; - - if (!Table) return(-1); - - Table_Init(Table, 0, 0); - - /* open the file */ - hfile = Open_File(File, "r", path); - if (!hfile) return(-1); - else { - MPI_MASTER( - if(Table->quiet<1) - printf("Opening input file '%s' (Table_Read, Binary)\n", path); - ); - } - - /* read file state */ - stat(File,&stfile); - filesize = stfile.st_size; - Table->filesize=filesize; - - /* read file content */ - if (type && !strcmp(type,"double")) sizeofelement = sizeof(double); - else sizeofelement = sizeof(float); - if (offset && *offset) fseek(hfile, *offset, SEEK_SET); - begin = ftell(hfile); - if (rows && filesize > sizeofelement*columns*rows) - nelements = columns*rows; - else nelements = (long)(filesize/sizeofelement); - if (!nelements || filesize <= *offset) return(0); - data = (double*)malloc(nelements*sizeofelement); - if (!data) { - if(!(Table->quiet>1)) - fprintf(stderr,"Error: allocating %ld elements for %s file '%s'. Too big (Table_Read_Offset_Binary).\n", nelements, type, File); - exit(-1); - } - nelements = fread(data, sizeofelement, nelements, hfile); - - if (!data || !nelements) - { - if(!(Table->quiet>1)) - fprintf(stderr,"Error: reading %ld elements from %s file '%s' (Table_Read_Offset_Binary)\n", nelements, type, File); - exit(-1); - } - Table->begin = begin; - Table->end = ftell(hfile); - if (offset) *offset=Table->end; - fclose(hfile); - data = (double*)realloc(data, (double)nelements*sizeofelement); - /* copy file data into Table */ - if (type && !strcmp(type,"double")) Table->data = data; - else { - float *s; - double *dataf; - s = (float*)data; - dataf = (double*)malloc(sizeof(double)*nelements); - for (i=0; idata = dataf; - } - strncpy(Table->filename, File, 1024); - Table->rows = nelements/columns; - Table->columns = columns; - Table->array_length = 1; - Table->block_number = 1; - - Table_Stat(Table); - - return(nelements); - } /* end Table_Read_Offset_Binary */ - -/******************************************************************************* -* long Table_Read_Handle(t_Table *Table, FILE *fid, int block_number, long max_rows, char *name) -* ACTION: read a single Table from a text file handle (private) -* input Table:pointer to a t_Table structure -* fid: pointer to FILE handle -* block_number: if the file does contain more than one -* data block, then indicates which one to get (from index 1) -* a 0 value means append/catenate all -* max_rows: if non 0, only reads that number of lines -* return initialized single Table t_Table structure containing data, header, ... -* modified Table t_Table structure containing data, header, ... -* number of read elements (-1: error, 0:header only) -* The routine stores any line starting with '#', '%' and ';' into the header -* Other lines are interpreted as numerical data, and stored. -* Data block should be a rectangular matrix or vector. -* Data block may be rebined with Table_Rebin (also sort in ascending order) -*******************************************************************************/ - long Table_Read_Handle(t_Table *Table, FILE *hfile, - long block_number, long max_rows, char *name) - { /* reads all/a data block from 'file' handle and returns a Table structure */ - double *Data; - char *Header = NULL; - long malloc_size = CHAR_BUF_LENGTH; - long malloc_size_h = 4096; - long Rows = 0, Columns = 0; - long count_in_array = 0; - long count_in_header = 0; - long count_invalid = 0; - long block_Current_index = 0; - char flag_End_row_loop = 0; - - if (!Table) return(-1); - Table_Init(Table, 0, 0); - if (name && name[0]!='\0') strncpy(Table->filename, name, 1024); - - if(!hfile) { - fprintf(stderr, "Error: File handle is NULL (Table_Read_Handle).\n"); - return (-1); - } - Header = (char*) calloc(malloc_size_h, sizeof(char)); - Data = (double*)calloc(malloc_size, sizeof(double)); - if ((Header == NULL) || (Data == NULL)) { - fprintf(stderr, "Error: Could not allocate Table and Header (Table_Read_Handle).\n"); - return (-1); - } - - int flag_In_array = 0; - do { /* while (!flag_End_row_loop) */ - char line[1024*CHAR_BUF_LENGTH]; - long back_pos=0; /* ftell start of line */ - - back_pos = ftell(hfile); - if (fgets(line, 1024*CHAR_BUF_LENGTH, hfile) != NULL) { /* analyse line */ - /* first skip blank and tabulation characters */ - int i = strspn(line, " \t"); - - /* handle comments: stored in header */ - if (NULL != strchr("#%;/", line[i])) - { /* line is a comment */ - count_in_header += strlen(line); - if (count_in_header >= malloc_size_h) { - /* if succeed and in array : add (and realloc if necessary) */ - malloc_size_h = count_in_header+4096; - Header = (char*)realloc(Header, malloc_size_h*sizeof(char)); - } - strncat(Header, line, 4096); - flag_In_array=0; - /* exit line and file if passed desired block */ - if (block_number > 0 && block_number == block_Current_index) { - flag_End_row_loop = 1; - } - - /* Continue with next line */ - continue; - } - if (strstr(line, "***")) - { - count_invalid++; - /* Continue with next line */ - continue; - } - - /* get the number of columns splitting line with strtok */ - char *lexeme; - char flag_End_Line = 0; - long block_Num_Columns = 0; - const char seps[] = " ,;\t\n\r"; - - lexeme = strtok(line, seps); - while (!flag_End_Line) { - if ((lexeme != NULL) && (lexeme[0] != '\0')) { - /* reading line: the token is not empty */ - double X; - int count=1; - /* test if we have 'NaN','Inf' */ - if (!strncasecmp(lexeme,"NaN",3)) - X = 0; - else if (!strncasecmp(lexeme,"Inf",3) || !strncasecmp(lexeme,"+Inf",4)) - X = FLT_MAX; - else if (!strncasecmp(lexeme,"-Inf",4)) - X = -FLT_MAX; - else - count = sscanf(lexeme,"%lg",&X); - if (count == 1) { - /* reading line: the token is a number in the line */ - if (!flag_In_array) { - /* reading num: not already in a block: starts a new data block */ - block_Current_index++; - flag_In_array = 1; - block_Num_Columns= 0; - if (block_number > 0) { - /* initialise a new data block */ - Rows = 0; - count_in_array = 0; - } /* else append */ - } - /* reading num: all blocks or selected block */ - if (flag_In_array && (block_number == 0 || - block_number == block_Current_index)) { - /* starting block: already the desired number of rows ? */ - if (block_Num_Columns == 0 && - max_rows > 0 && Rows >= max_rows) { - flag_End_Line = 1; - flag_End_row_loop = 1; - flag_In_array = 0; - /* reposition to begining of line (ignore line) */ - fseek(hfile, back_pos, SEEK_SET); - } else { /* store into data array */ - if (count_in_array >= malloc_size) { - /* realloc data buffer if necessary */ - malloc_size = count_in_array*1.5; - Data = (double*) realloc(Data, malloc_size*sizeof(double)); - if (Data == NULL) { - fprintf(stderr, "Error: Can not re-allocate memory %li (Table_Read_Handle).\n", - malloc_size*sizeof(double)); - return (-1); - } - } - if (0 == block_Num_Columns) Rows++; - Data[count_in_array] = X; - count_in_array++; - block_Num_Columns++; - } - } /* reading num: end if flag_In_array */ - } /* end reading num: end if sscanf lexeme -> numerical */ - else { - /* reading line: the token is not numerical in that line. end block */ - if (block_Current_index == block_number) { - flag_End_Line = 1; - flag_End_row_loop = 1; - } else { - flag_In_array = 0; - flag_End_Line = 1; - } - } - } - else { - /* no more tokens in line */ - flag_End_Line = 1; - if (block_Num_Columns > 0) Columns = block_Num_Columns; - } - - // parse next token - lexeme = strtok(NULL, seps); - - } /* while (!flag_End_Line) */ - } /* end: if fgets */ - else flag_End_row_loop = 1; /* else fgets : end of file */ - - } while (!flag_End_row_loop); /* end while flag_End_row_loop */ - - Table->block_number = block_number; - Table->array_length = 1; - - // shrink header to actual size (plus terminating 0-byte) - if (count_in_header) { - Header = (char*)realloc(Header, count_in_header*sizeof(char) + 1); - } - Table->header = Header; - - if (count_in_array*Rows*Columns == 0) - { - Table->rows = 0; - Table->columns = 0; - free(Data); - return (0); - } - if (Rows * Columns != count_in_array) - { - fprintf(stderr, "Warning: Read_Table :%s %s Data has %li values that should be %li x %li\n", - (Table->filename[0] != '\0' ? Table->filename : ""), - (!block_number ? " catenated" : ""), - count_in_array, Rows, Columns); - Columns = count_in_array; Rows = 1; - } - if (count_invalid) - { - fprintf(stderr,"Warning: Read_Table :%s %s Data has %i invalid lines (*****). Ignored.\n", - (Table->filename[0] != '\0' ? Table->filename : ""), - (!block_number ? " catenated" : ""), - count_invalid); - } - Data = (double*)realloc(Data, count_in_array*sizeof(double)); - Table->data = Data; - Table->rows = Rows; - Table->columns = Columns; - - return (count_in_array); - - } /* end Table_Read_Handle */ - -/******************************************************************************* -* long Table_Rebin(t_Table *Table) -* ACTION: rebin a single Table, sorting 1st column in ascending order -* input Table: single table containing data. -* The data block is reallocated in this process -* return updated Table with increasing, evenly spaced first column (index 0) -* number of data elements (-1: error, 0:empty data) -*******************************************************************************/ - long Table_Rebin(t_Table *Table) - { - double new_step=0; - long i; - /* performs linear interpolation on X axis (0-th column) */ - - if (!Table) return(-1); - if (!Table->data - || Table->rows*Table->columns == 0 || !Table->step_x) - return(0); - Table_Stat(Table); /* recompute statitstics and minimal step */ - new_step = Table->step_x; /* minimal step in 1st column */ - - if (!(Table->constantstep)) /* not already evenly spaced */ - { - long Length_Table; - double *New_Table; - - Length_Table = ceil(fabs(Table->max_x - Table->min_x)/new_step)+1; - /*return early if the rebinned table will become too large*/ - if (Length_Table > mcread_table_rebin_maxsize){ - fprintf(stderr,"WARNING: (Table_Rebin): Rebinning table from %s would exceed 1M rows. Skipping.\n", Table->filename); - return(Table->rows*Table->columns); - } - New_Table = (double*)malloc(Length_Table*Table->columns*sizeof(double)); - - for (i=0; i < Length_Table; i++) - { - long j; - double X; - X = Table->min_x + i*new_step; - New_Table[i*Table->columns] = X; - for (j=1; j < Table->columns; j++) - New_Table[i*Table->columns+j] - = Table_Value(*Table, X, j); - } /* end for i */ - - Table->rows = Length_Table; - Table->step_x = new_step; - Table->max_x = Table->min_x + (Length_Table-1)*new_step; - /*max might not be the same anymore - * Use Length_Table -1 since the first and laset rows are the limits of the defined interval.*/ - free(Table->data); - Table->data = New_Table; - Table->constantstep=1; - } /* end else (!constantstep) */ - return (Table->rows*Table->columns); - } /* end Table_Rebin */ - -/******************************************************************************* -* double Table_Index(t_Table Table, long i, long j) -* ACTION: read an element [i,j] of a single Table -* input Table: table containing data -* i : index of row (0:Rows-1) -* j : index of column (0:Columns-1) -* return Value = data[i][j] -* Returns Value from the i-th row, j-th column of Table -* Tests are performed on indexes i,j to avoid errors -*******************************************************************************/ - -#ifndef MIN -#define MIN(a, b) (((a) < (b)) ? (a) : (b)) -#endif -#ifndef MAX -#define MAX(a, b) (((a) > (b)) ? (a) : (b)) -#endif - -double Table_Index(t_Table Table, long i, long j) -{ - long AbsIndex; - - if (Table.rows == 1 || Table.columns == 1) { - /* vector */ - j = MIN(MAX(0, i+j), Table.columns*Table.rows - 1); - i = 0; - } else { - /* matrix */ - i = MIN(MAX(0, i), Table.rows - 1); - j = MIN(MAX(0, j), Table.columns - 1); - } - - /* handle vectors specifically */ - AbsIndex = i*(Table.columns)+j; - - if (Table.data != NULL) - return (Table.data[AbsIndex]); - else - return 0; -} /* end Table_Index */ - -/******************************************************************************* -* void Table_SetElement(t_Table *Table, long i, long j, double value) -* ACTION: set an element [i,j] of a single Table -* input Table: table containing data -* i : index of row (0:Rows-1) -* j : index of column (0:Columns-1) -* value = data[i][j] -* Returns 0 in case of error -* Tests are performed on indexes i,j to avoid errors -*******************************************************************************/ -int Table_SetElement(t_Table *Table, long i, long j, - double value) -{ - long AbsIndex; - - if (Table->rows == 1 || Table->columns == 1) { - /* vector */ - j = MIN(MAX(0, i+j), Table->columns*Table->rows - 1); i=0; - } else { - /* matrix */ - i = MIN(MAX(0, i), Table->rows - 1); - j = MIN(MAX(0, j), Table->columns - 1); - } - - AbsIndex = i*(Table->columns)+j; - if (Table->data != NULL) { - Table->data[AbsIndex] = value; - return 1; - } - - return 0; -} /* end Table_SetElement */ - -/******************************************************************************* -* double Table_Value(t_Table Table, double X, long j) -* ACTION: read column [j] of a single Table at row which 1st column is X -* input Table: table containing data. -* X : data value in the first column (index 0) -* j : index of column from which is extracted the Value (0:Columns-1) -* return Value = data[index for X][j] with linear interpolation -* Returns Value from the j-th column of Table corresponding to the -* X value for the 1st column (index 0) -* Tests are performed (within Table_Index) on indexes i,j to avoid errors -* NOTE: data should rather be monotonic, and evenly sampled. -*******************************************************************************/ -double Table_Value(t_Table Table, double X, long j) -{ - long Index = -1; - double X1=0, Y1=0, X2=0, Y2=0; - double ret=0; - - if (X > Table.max_x) return Table_Index(Table,Table.rows-1 ,j); - if (X < Table.min_x) return Table_Index(Table,0 ,j); - - // Use constant-time lookup when possible - if(Table.constantstep) { - Index = (long)floor( - (X - Table.min_x) / (Table.max_x - Table.min_x) * (Table.rows-1)); - X1 = Table_Index(Table,Index-1,0); - X2 = Table_Index(Table,Index ,0); - } - // Use binary search on large, monotonic tables - else if(Table.monotonic && Table.rows > 100) { - long left = Table.min_x; - long right = Table.max_x; - - while (!((X1 <= X) && (X < X2)) && (right - left > 1)) { - Index = (left + right) / 2; - - X1 = Table_Index(Table, Index-1, 0); - X2 = Table_Index(Table, Index, 0); - - if (X < X1) { - right = Index; - } else { - left = Index; - } - } - } - - // Fall back to linear search, if no-one else has set X1, X2 correctly - if (!((X1 <= X) && (X < X2))) { - /* look for index surrounding X in the table -> Index */ - for (Index=1; Index <= Table.rows-1; Index++) { - X1 = Table_Index(Table, Index-1,0); - X2 = Table_Index(Table, Index ,0); - if ((X1 <= X) && (X < X2)) break; - } /* end for Index */ - } - - Y1 = Table_Index(Table,Index-1, j); - Y2 = Table_Index(Table,Index , j); - -#ifdef OPENACC -#define strcmp(a,b) str_comp(a,b) -#endif - - if (!strcmp(Table.method,"linear")) { - ret = Table_Interp1d(X, X1,Y1, X2,Y2); - } - else if (!strcmp(Table.method,"nearest")) { - ret = Table_Interp1d_nearest(X, X1,Y1, X2,Y2); - } - -#ifdef OPENACC -#ifdef strcmp -#undef strcmp -#endif -#endif - - return ret; -} /* end Table_Value */ - -/******************************************************************************* -* double Table_Value2d(t_Table Table, double X, double Y) -* ACTION: read element [X,Y] of a matrix Table -* input Table: table containing data. -* X : row index, may be non integer -* Y : column index, may be non integer -* return Value = data[index X][index Y] with bi-linear interpolation -* Returns Value for the indices [X,Y] -* Tests are performed (within Table_Index) on indexes i,j to avoid errors -* NOTE: data should rather be monotonic, and evenly sampled. -*******************************************************************************/ -double Table_Value2d(t_Table Table, double X, double Y) - { - long x1,x2,y1,y2; - double z11,z12,z21,z22; - double ret=0; - - x1 = (long)floor(X); - y1 = (long)floor(Y); - - if (x1 > Table.rows-1 || x1 < 0) { - x2 = x1; - } else { - x2 = x1 + 1; - } - - if (y1 > Table.columns-1 || y1 < 0) { - y2 = y1; - } else { - y2 = y1 + 1; - } - - z11 = Table_Index(Table, x1, y1); - - if (y2 != y1) z12=Table_Index(Table, x1, y2); else z12 = z11; - if (x2 != x1) z21=Table_Index(Table, x2, y1); else z21 = z11; - if (y2 != y1) z22=Table_Index(Table, x2, y2); else z22 = z21; - -#ifdef OPENACC -#define strcmp(a,b) str_comp(a,b) -#endif - - if (!strcmp(Table.method,"linear")) - ret = Table_Interp2d(X,Y, x1,y1,x2,y2, z11,z12,z21,z22); -#ifdef OPENACC -#ifdef strcmp -#undef strcmp -#endif -#endif - else { - if (fabs(X-x1) < fabs(X-x2)) { - if (fabs(Y-y1) < fabs(Y-y2)) ret = z11; else ret = z12; - } else { - if (fabs(Y-y1) < fabs(Y-y2)) ret = z21; else ret = z22; - } - } - return ret; - } /* end Table_Value2d */ - - -/******************************************************************************* -* void Table_Free(t_Table *Table) -* ACTION: free a single Table. First Call Table_File_list_gc. If this returns -* non-zero it means there are more refernces to the table, and so the table -* should not bee freed. -* return: empty Table -*******************************************************************************/ - void Table_Free(t_Table *Table) - { - if( !Table_File_List_gc(Table) ){ - return; - } - if (!Table) return; - if (Table->data != NULL) free(Table->data); - if (Table->header != NULL) free(Table->header); - Table->data = NULL; - Table->header = NULL; - } /* end Table_Free */ - -/****************************************************************************** -* void Table_Info(t_Table Table) -* ACTION: print informations about a single Table -*******************************************************************************/ - long Table_Info(t_Table Table) - { - char buffer[256]; - long ret=0; - - if (!Table.block_number) strcpy(buffer, "catenated"); - else sprintf(buffer, "block %li", Table.block_number); - printf("Table from file '%s' (%s)", - Table.filename[0] != '\0' ? Table.filename : "", buffer); - if ((Table.data != NULL) && (Table.rows*Table.columns)) - { - printf(" is %li x %li ", Table.rows, Table.columns); - if (Table.rows*Table.columns > 1) - printf("(x=%g:%g)", Table.min_x, Table.max_x); - else printf("(x=%g) ", Table.min_x); - ret = Table.rows*Table.columns; - if (Table.monotonic) printf(", monotonic"); - if (Table.constantstep) printf(", constant step"); - printf(". interpolation: %s\n", Table.method); - } - else printf(" is empty.\n"); - - if (Table.header && strlen(Table.header)) { - char *header; - int i; - header = malloc(80); - if (!header) return(ret); - for (i=0; i<80; header[i++]=0); - strncpy(header, Table.header, 75); - if (strlen(Table.header) > 75) { - strcat( header, " ..."); - } - for (i=0; iheader = NULL; - Table->filename[0]= '\0'; - Table->filesize= 0; - Table->min_x = 0; - Table->max_x = 0; - Table->step_x = 0; - Table->block_number = 0; - Table->array_length = 0; - Table->monotonic = 0; - Table->constantstep = 0; - Table->begin = 0; - Table->end = 0; - strcpy(Table->method,"linear"); - - if (rows*columns >= 1) { - data = (double*)malloc(rows*columns*sizeof(double)); - if (data) for (i=0; i < rows*columns; data[i++]=0); - else { - if(Table->quiet<2) - fprintf(stderr,"Error: allocating %ld double elements." - "Too big (Table_Init).\n", rows*columns); - rows = columns = 0; - } - } - Table->rows = (rows >= 1 ? rows : 0); - Table->columns = (columns >= 1 ? columns : 0); - Table->data = data; - return(Table->rows*Table->columns); -} /* end Table_Init */ - -/****************************************************************************** -* long Table_Write(t_Table Table, char *file, x1,x2, y1,y2) -* ACTION: write a Table to disk (ascii). -* when x1=x2=0 or y1=y2=0, the table default limits are used. -* return: 0=all is fine, non-0: error -*******************************************************************************/ -MCDETECTOR Table_Write(t_Table Table, char *file, char *xl, char *yl, - double x1, double x2, double y1, double y2) -{ - MCDETECTOR detector; - - if ((Table.data == NULL) && (Table.rows*Table.columns)) { - detector.m = 0; - return(detector); /* Table is empty - nothing to do */ - } - if (!x1 && !x2) { - x1 = Table.min_x; - x2 = Table.max_x; - } - if (!y1 && !y2) { - y1 = 1; - y2 = Table.columns; - } - - /* transfer content of the Table into a 2D detector */ - Coords coords = { 0, 0, 0}; - - if (Table.rows == 1 || Table.columns == 1) { - detector = mcdetector_out_1D(Table.filename, - xl ? xl : "", yl ? yl : "", - "x", x1, x2, - Table.rows * Table.columns, - NULL, Table.data, NULL, - file, file, coords); - } else { - detector = mcdetector_out_2D(Table.filename, - xl ? xl : "", yl ? yl : "", - x1, x2, y1, y2, - Table.rows, Table.columns, - NULL, Table.data, NULL, - file, file, coords); - } - return(detector); -} - -/****************************************************************************** -* void Table_Stat(t_Table *Table) -* ACTION: computes min/max/mean step of 1st column for a single table (private) -* return: updated Table -*******************************************************************************/ - static void Table_Stat(t_Table *Table) - { - long i; - double max_x, min_x; - double row=1; - char monotonic=1; - char constantstep=1; - double step=0; - long n; - - if (!Table) return; - if (!Table->rows || !Table->columns) return; - if (Table->rows == 1) row=0; // single row - max_x = -FLT_MAX; - min_x = FLT_MAX; - n = (row ? Table->rows : Table->columns); - /* get min and max of first column/vector */ - for (i=0; i < n; i++) - { - double X; - X = (row ? Table_Index(*Table,i ,0) - : Table_Index(*Table,0, i)); - if (X < min_x) min_x = X; - if (X > max_x) max_x = X; - } /* for */ - - /* test for monotonicity and constant step if the table is an XY or single vector */ - if (n > 1) { - /* mean step */ - step = (max_x - min_x)/(n-1); - /* now test if table is monotonic on first column, and get minimal step size */ - for (i=0; i < n-1; i++) { - double X, diff;; - X = (row ? Table_Index(*Table,i ,0) - : Table_Index(*Table,0, i)); - diff = (row ? Table_Index(*Table,i+1,0) - : Table_Index(*Table,0, i+1)) - X; - if (diff && fabs(diff) < fabs(step)) step = diff; - /* change sign ? */ - if ((max_x - min_x)*diff < 0 && monotonic) - monotonic = 0; - } /* end for */ - - /* now test if steps are constant within READ_TABLE_STEPTOL */ - if(!step){ - /*means there's a disconitnuity -> not constantstep*/ - constantstep=0; - }else if (monotonic) { - for (i=0; i < n-1; i++) { - double X, diff; - X = (row ? Table_Index(*Table,i ,0) - : Table_Index(*Table,0, i)); - diff = (row ? Table_Index(*Table,i+1,0) - : Table_Index(*Table,0, i+1)) - X; - if ( fabs(step)*(1+READ_TABLE_STEPTOL) < fabs(diff) || - fabs(diff) < fabs(step)*(1-READ_TABLE_STEPTOL) ) - { constantstep = 0; break; } - } - } - - } - Table->step_x= step; - Table->max_x = max_x; - Table->min_x = min_x; - Table->monotonic = monotonic; - Table->constantstep = constantstep; - } /* end Table_Stat */ - -/****************************************************************************** -* t_Table *Table_Read_Array(char *File, long *blocks) -* ACTION: read as many data blocks as available, iteratively from file -* return: initialized t_Table array, last element is an empty Table. -* the number of extracted blocks in non NULL pointer *blocks -*******************************************************************************/ - t_Table *Table_Read_Array(char *File, long *blocks) - { - t_Table *Table_Array=NULL; - long offset=0; - long block_number=0; - long allocated=256; - long nelements=1; - - /* first allocate an initial empty t_Table array */ - Table_Array = (t_Table *)malloc(allocated*sizeof(t_Table)); - if (!Table_Array) { - fprintf(stderr, "Error: Can not allocate memory %li (Table_Read_Array).\n", - allocated*sizeof(t_Table)); - *blocks = 0; - return (NULL); - } - - while (nelements > 0) - { - t_Table Table; - - /* if ok, set t_Table block number else exit loop */ - block_number++; - Table.block_number = block_number; - - /* access file at offset and get following block. Block number is from the set offset - * hence the hardcoded 1 - i.e. the next block counted from offset.*/ - nelements = Table_Read_Offset(&Table, File, 1, &offset,0); - /*if the block is empty - don't store it*/ - if (nelements>0){ - /* if t_Table array is not long enough, expand and realocate */ - if (block_number >= allocated-1) { - allocated += 256; - Table_Array = (t_Table *)realloc(Table_Array, - allocated*sizeof(t_Table)); - if (!Table_Array) { - fprintf(stderr, "Error: Can not re-allocate memory %li (Table_Read_Array).\n", - allocated*sizeof(t_Table)); - *blocks = 0; - return (NULL); - } - } - /* store it into t_Table array */ - //snprintf(Table.filename, 1024, "%s#%li", File, block_number-1); - Table_Array[block_number-1] = Table; - } - /* continues until we find an empty block */ - } - /* send back number of extracted blocks */ - if (blocks) *blocks = block_number-1; - - /* now store total number of elements in Table array */ - for (offset=0; offset < block_number; - Table_Array[offset++].array_length = block_number-1); - - return(Table_Array); - } /* end Table_Read_Array */ -/******************************************************************************* -* void Table_Free_Array(t_Table *Table) -* ACTION: free a Table array -*******************************************************************************/ - void Table_Free_Array(t_Table *Table) - { - long index; - if (!Table) return; - for (index=0;index < Table[0].array_length; index++){ - Table_Free(&Table[index]); - } - free(Table); - } /* end Table_Free_Array */ - -/****************************************************************************** -* long Table_Info_Array(t_Table *Table) -* ACTION: print informations about a Table array -* return: number of elements in the Table array -*******************************************************************************/ - long Table_Info_Array(t_Table *Table) - { - long index=0; - - if (!Table) return(-1); - while (index < Table[index].array_length - && (Table[index].data || Table[index].header) - && (Table[index].rows*Table[index].columns) ) { - Table_Info(Table[index]); - index++; - } - printf("This Table array contains %li elements\n", index); - return(index); - } /* end Table_Info_Array */ - -/****************************************************************************** -* char **Table_ParseHeader(char *header, symbol1, symbol2, ..., NULL) -* ACTION: search for char* symbols in header and return their value or NULL -* the search is not case sensitive. -* Last argument MUST be NULL -* return: array of char* with line following each symbol, or NULL if not found -*******************************************************************************/ -#ifndef MyNL_ARGMAX -#define MyNL_ARGMAX 50 -#endif - -char **Table_ParseHeader_backend(char *header, ...){ - va_list ap; - char exit_flag=0; - int counter =0; - char **ret =NULL; - if (!header || header[0]=='\0') return(NULL); - - ret = (char**)calloc(MyNL_ARGMAX, sizeof(char*)); - if (!ret) { - printf("Table_ParseHeader: Cannot allocate %i values array for Parser (Table_ParseHeader).\n", - MyNL_ARGMAX); - return(NULL); - } - for (counter=0; counter < MyNL_ARGMAX; ret[counter++] = NULL); - counter=0; - - va_start(ap, header); - while(!exit_flag && counter < MyNL_ARGMAX-1) - { - char *arg_char=NULL; - char *pos =NULL; - /* get variable argument value as a char */ - arg_char = va_arg(ap, char *); - if (!arg_char || arg_char[0]=='\0'){ - exit_flag = 1; break; - } - /* search for the symbol in the header */ - pos = (char*)strcasestr(header, arg_char); - if (pos) { - char *eol_pos; - eol_pos = strchr(pos+strlen(arg_char), '\n'); - if (!eol_pos) - eol_pos = strchr(pos+strlen(arg_char), '\r'); - if (!eol_pos) - eol_pos = pos+strlen(pos)-1; - ret[counter] = (char*)malloc(eol_pos - pos); - if (!ret[counter]) { - printf("Table_ParseHeader: Cannot allocate value[%i] array for Parser searching for %s (Table_ParseHeader).\n", - counter, arg_char); - exit_flag = 1; break; - } - strncpy(ret[counter], pos+strlen(arg_char), eol_pos - pos - strlen(arg_char)); - ret[counter][eol_pos - pos - strlen(arg_char)]='\0'; - } - counter++; - } - va_end(ap); - return(ret); -} /* Table_ParseHeader */ - -/****************************************************************************** -* double Table_Interp1d(x, x1, y1, x2, y2) -* ACTION: interpolates linearly at x between y1=f(x1) and y2=f(x2) -* return: y=f(x) value -*******************************************************************************/ -double Table_Interp1d(double x, - double x1, double y1, - double x2, double y2) -{ - double slope; - if (x2 == x1) return (y1+y2)/2; - if (y1 == y2) return y1; - slope = (y2 - y1)/(x2 - x1); - return y1+slope*(x - x1); -} /* Table_Interp1d */ - -/****************************************************************************** -* double Table_Interp1d_nearest(x, x1, y1, x2, y2) -* ACTION: table lookup with nearest method at x between y1=f(x1) and y2=f(x2) -* return: y=f(x) value -*******************************************************************************/ -double Table_Interp1d_nearest(double x, - double x1, double y1, - double x2, double y2) -{ - if (fabs(x-x1) < fabs(x-x2)) return (y1); - else return(y2); -} /* Table_Interp1d_nearest */ - -/****************************************************************************** -* double Table_Interp2d(x,y, x1,y1, x2,y2, z11,z12,z21,z22) -* ACTION: interpolates bi-linearly at (x,y) between z1=f(x1,y1) and z2=f(x2,y2) -* return: z=f(x,y) value -* x,y | x1 x2 -* ---------------- -* y1 | z11 z21 -* y2 | z12 z22 -*******************************************************************************/ -double Table_Interp2d(double x, double y, - double x1, double y1, - double x2, double y2, - double z11, double z12, double z21, double z22) -{ - double ratio_x, ratio_y; - if (x2 == x1) return Table_Interp1d(y, y1,z11, y2,z12); - if (y1 == y2) return Table_Interp1d(x, x1,z11, x2,z21); - - ratio_y = (y - y1)/(y2 - y1); - ratio_x = (x - x1)/(x2 - x1); - return (1-ratio_x)*(1-ratio_y)*z11 + ratio_x*(1-ratio_y)*z21 - + ratio_x*ratio_y*z22 + (1-ratio_x)*ratio_y*z12; -} /* Table_Interp2d */ - -/* end of read_table-lib.c */ -/* Contents of ref-lib.c (requested from Guide_gravity) */ -/**************************************************************************** -* -* McStas, neutron ray-tracing package -* Copyright 1997-2006, All rights reserved -* Risoe National Laboratory, Roskilde, Denmark -* Institut Laue Langevin, Grenoble, France -* -* Library: share/ref-lib.c -* -* %Identification -* Written by: Peter Christiansen -* Date: August, 2006 -* Origin: RISOE -* Release: McStas 1.10 -* Version: $Revision$ -* -* Add StdDoubleReflecFunc, ExtendedReflecFunc -* Date: October, 2022 -* Locale: ESS -* Release: McStas 2.7.x, 3.x -* -* Commonly used reflection functions are declared in this file which -* are used by some guide and mirror components. -* -* Variable names have prefix 'mc_ref_' for 'McStas Reflection' -* to avoid conflicts -* -* Usage: within SHARE -* %include "ref-lib" -* -****************************************************************************/ - -#ifndef REF_LIB_H -#include "ref-lib.h" -#endif - -#ifndef READ_TABLE_LIB_H -#include "read_table-lib.h" -#include "read_table-lib.c" -#endif - -/**************************************************************************** -* void StdReflecFunc(double q, double *par, double *r) -* -* The McStas standard analytic parametrization of the reflectivity. -* The parameters are: -* R0: [1] Low-angle reflectivity -* Qc: [AA-1] Critical scattering vector -* alpha: [AA] Slope of reflectivity -* m: [1] m-value of material. Zero means completely absorbing. -* W: [AA-1] Width of supermirror cut-off -*****************************************************************************/ -#pragma acc routine seq -void StdReflecFunc(double mc_pol_q, double *mc_pol_par, double *mc_pol_r) { - double R0 = mc_pol_par[0]; - double Qc = mc_pol_par[1]; - double alpha = mc_pol_par[2]; - double m = mc_pol_par[3]; - double W = mc_pol_par[4]; - double beta = 0; - mc_pol_q = fabs(mc_pol_q); - double arg; - double m_corr; - - /* Simpler parametrization from Henrik Jacobsen uses these values that depend on m only. - double m_value=m*0.9853+0.1978; - double W=-0.0002*m_value+0.0022; - double alpha=0.2304*m_value+5.0944; - double beta=-7.6251*m_value+68.1137; - If W and alpha are set to 0, use Henrik's approach for estimating these parameters - and apply the formulation: - arg = R0*0.5*(1-tanh(arg))*(1-alpha*(q-Qc)+beta*(q-Qc)*(q-Qc)); - */ - if (W==0 && alpha==0) { - m = m * 0.9853 + 0.1978; - m_corr = m * 0.9853 - 0.7875; - W = -0.0002 * m_corr + 0.0022; - alpha = 0.2304 * m_corr + 5.0944; - beta = -7.6251 * m_corr + 68.1137; - if (m==3) { - alpha = m_corr; - beta = 0; - } - arg = (mc_pol_q - m*Qc)/W; // <--- here m, not m_corr!! - } - - arg = W > 0 ? (mc_pol_q - m*Qc)/W : 11; - - if (arg > 10 || m <= 0 || Qc <=0 || R0 <= 0) { - *mc_pol_r = 0; - return; - } - - if (m < 1) { Qc *= m; m=1; } - - if(mc_pol_q <= Qc) { - *mc_pol_r = R0; - return; - } - - - *mc_pol_r = R0*0.5*(1 - tanh(arg))*(1 - alpha*(mc_pol_q - Qc) + beta*(mc_pol_q - Qc)*(mc_pol_q - Qc)); - - return; - } - -/**************************************************************************** -* void TableReflecFunc(double q, t_Table *par, double *r) { -* -* Looks up the reflectivity in a table using the routines in read_table-lib. -*****************************************************************************/ -#pragma acc routine seq -void TableReflecFunc(double mc_pol_q, t_Table *mc_pol_par, double *mc_pol_r) { - - *mc_pol_r = Table_Value(*mc_pol_par, mc_pol_q, 1); - if(*mc_pol_r>1) - *mc_pol_r = 1; - return; -} - - -/**************************************************************************** -* void StdDoubleReflecFunc(double q, double *par, double *r) -* -* The McStas standard analytic parametrization of the reflectivity for -* double-side coated supermirror. -* The parameters are: -* R0: [1] Low-angle reflectivity -* Qc: [AA-1] Critical scattering vector -* alpha: [AA] Slope of reflectivity -* m: [1] m-value of material. Zero means completely absorbing. -* W: [AA-1] Width of supermirror cut-off -*****************************************************************************/ -void StdDoubleReflecFunc(double mc_pol_q, double *mc_pol_par, double *mc_pol_r) { - double R0 = mc_pol_par[0]; - double Qc = mc_pol_par[1]; - double alpha = mc_pol_par[2]; - double m = mc_pol_par[3]; - double W = mc_pol_par[4]; - double beta = 0; - mc_pol_q = fabs(mc_pol_q); - double arg; - - /* Simpler parametrization from Henrik Jacobsen uses these values that depend on m only. - double m_value=m*0.9853+0.1978; - double W=-0.0002*m_value+0.0022; - double alpha=0.2304*m_value+5.0944; - double beta=-7.6251*m_value+68.1137; - If W and alpha are set to 0, use Henrik's approach for estimating these parameters - and apply the formulation: - arg = R0*0.5*(1-tanh(arg))*(1-alpha*(q-Qc)+beta*(q-Qc)*(q-Qc)); - */ - if (W==0 && alpha==0) { - m=m*0.9853+0.1978; - W=-0.0002*m+0.0022; - alpha=0.2304*m+5.0944; - beta=-7.6251*m+68.1137; - if (m<=3) { - alpha=m; - beta=0; - } - } - - arg = W > 0 ? (mc_pol_q - m*Qc)/W : 11; - - if (arg > 10 || m <= 0 || Qc <=0 || R0 <= 0) { - *mc_pol_r = 0; - return; - } - - if (m < 1) { Qc *= m; m=1; } - - /* - Reflectivity R0 = single-side coated supermirror reflectivity - double-side coated supermirror reflectivity = 1- (1-R0)^2 - */ - - if(mc_pol_q <= Qc) { - *mc_pol_r = 1- (1-R0)*(1-R0); - return; - } - - - R0 = R0*0.5*(1 - tanh(arg))*(1 - alpha*(mc_pol_q - Qc) + beta*(mc_pol_q - Qc)*(mc_pol_q - Qc)); - *mc_pol_r = 1- (1-R0)*(1-R0); - - return; - } - -void ExtendedReflecFunc(double mc_pol_q, double *mc_pol_par, double *mc_pol_r) { - double R0 = mc_pol_par[0]; - double Qc = mc_pol_par[1]; - double alpha = mc_pol_par[2]; - double m = mc_pol_par[3]; - double W = mc_pol_par[4]; - double beta = mc_pol_par[5]; - mc_pol_q = fabs(mc_pol_q); - double arg; - - /* Simpler parametrization from Henrik Jacobsen uses these values that depend on m only. - double m_value=m*0.9853+0.1978; - double W=-0.0002*m_value+0.0022; - double alpha=0.2304*m_value+5.0944; - double beta=-7.6251*m_value+68.1137; - If W and alpha are set to 0, use Henrik's approach for estimating these parameters - and apply the formulation: - arg = R0*0.5*(1-tanh(arg))*(1-alpha*(q-Qc)+beta*(q-Qc)*(q-Qc)); - */ - if (W==0 && alpha==0) { - m=m*0.9853+0.1978; - W=-0.0002*m+0.0022; - alpha=0.2304*m+5.0944; - beta=-7.6251*m+68.1137; - if (m<=3) { - alpha=m; - beta=0; - } - } - - arg = W > 0 ? (mc_pol_q - m*Qc)/W : 11; - - if (arg > 10 || m <= 0 || Qc <=0 || R0 <= 0) { - *mc_pol_r = 0; - return; - } - - if (m < 1) { Qc *= m; m=1; } - - if(mc_pol_q <= Qc) { - *mc_pol_r = R0; - return; - } - - - *mc_pol_r = R0*0.5*(1 - tanh(arg))*(1 - alpha*(mc_pol_q - Qc) + beta*(mc_pol_q - Qc)*(mc_pol_q - Qc)); - - return; - } - -/* end of ref-lib.c */ -/* User declarations from instrument definition. Can define functions. */ - -#undef compcurname -#undef compcurtype -#undef compcurindex -/* end of instrument 'splitRunTest' and components DECLARE */ -/* ***************************************************************************** -* instrument 'splitRunTest' and components INITIALISE -***************************************************************************** */ - -double index_getdistance(int first_index, int second_index) -/* Calculate the distance two components from their indexes*/ -{ - return coords_len(coords_sub(POS_A_COMP_INDEX(first_index), POS_A_COMP_INDEX(second_index))); -} - -double getdistance(char* first_component, char* second_component) -/* Calculate the distance between two named components */ -{ - int first_index = _getcomp_index(first_component); - int second_index = _getcomp_index(second_component); - return index_getdistance(first_index, second_index); -} - -double checked_setpos_getdistance(int current_index, char* first_component, char* second_component) -/* Calculate the distance between two named components at *_setpos() time, with component index checking */ -{ - int first_index = _getcomp_index(first_component); - int second_index = _getcomp_index(second_component); - if (first_index >= current_index || second_index >= current_index) { - printf("setpos_getdistance can only be used with the names of components before the current one!\n"); - return 0; - } - return index_getdistance(first_index, second_index); -} -#define setpos_getdistance(first, second) checked_setpos_getdistance(current_setpos_index, first, second) - - -/* component origin=Arm() SETTING, POSITION/ROTATION */ -int _origin_setpos(void) -{ /* sets initial component parameters, position and rotation */ - SIG_MESSAGE("[_origin_setpos] component origin=Arm() SETTING [origin:0]"); - stracpy(_origin_var._name, "origin", 7); - stracpy(_origin_var._type, "Arm", 4); - int current_setpos_index = _origin_var._index = 1; - /* component origin=Arm() AT ROTATED */ - { - Coords tc1, tc2; - tc1 = coords_set(0,0,0); - tc2 = coords_set(0,0,0); - Rotation tr1; - rot_set_rotation(tr1,0,0,0); - rot_set_rotation(_origin_var._rotation_absolute, (0)*DEG2RAD, (0)*DEG2RAD, (0)*DEG2RAD); - rot_copy(_origin_var._rotation_relative, _origin_var._rotation_absolute); - _origin_var._rotation_is_identity = rot_test_identity(_origin_var._rotation_relative); - _origin_var._position_absolute = coords_set(0, 0, 0); - tc1 = coords_neg(_origin_var._position_absolute); - _origin_var._position_relative = rot_apply(_origin_var._rotation_absolute, tc1); - } /* origin=Arm() AT ROTATED */ - DEBUG_COMPONENT("origin", _origin_var._position_absolute, _origin_var._rotation_absolute); - instrument->_position_absolute[1] = _origin_var._position_absolute; - instrument->_position_relative[1] = _origin_var._position_relative; - _origin_var._position_relative_is_zero = coords_test_zero(_origin_var._position_relative); - instrument->counter_N[1] = instrument->counter_P[1] = instrument->counter_P2[1] = 0; - instrument->counter_AbsorbProp[1] = 0; - return(0); -} /* _origin_setpos */ -/* component source=Source_simple() SETTING, POSITION/ROTATION */ -int _source_setpos(void) -{ /* sets initial component parameters, position and rotation */ - SIG_MESSAGE("[_source_setpos] component source=Source_simple() SETTING [/home/g/.cache/mccode-mcstas/main/sources/Source_simple.comp:68]"); - stracpy(_source_var._name, "source", 7); - stracpy(_source_var._type, "Source_simple", 14); - int current_setpos_index = _source_var._index = 2; - _source_var._parameters.radius = 0.1; - _source_var._parameters.yheight = 0.25; - _source_var._parameters.xwidth = 0.2; - _source_var._parameters.dist = 1.5; - _source_var._parameters.focus_xw = 0.06; - _source_var._parameters.focus_yh = 0.12; - _source_var._parameters.E0 = 5.0; - _source_var._parameters.dE = 0.1; - _source_var._parameters.lambda0 = 0; - _source_var._parameters.dlambda = 0; - _source_var._parameters.flux = 1; - _source_var._parameters.gauss = 0; - _source_var._parameters.target_index = 1; - /* component source=Source_simple() AT ROTATED */ - { - Coords tc1, tc2; - tc1 = coords_set(0,0,0); - tc2 = coords_set(0,0,0); - Rotation tr1; - rot_set_rotation(tr1,0,0,0); - rot_set_rotation(tr1, (0)*DEG2RAD, (0)*DEG2RAD, (0)*DEG2RAD); - rot_mul(tr1, _origin_var._rotation_absolute, _source_var._rotation_absolute); - rot_transpose(_origin_var._rotation_absolute, tr1); - rot_mul(_source_var._rotation_absolute, tr1, _source_var._rotation_relative); - _source_var._rotation_is_identity = rot_test_identity(_source_var._rotation_relative); - tc1 = coords_set(0, 0, 0); - rot_transpose(_origin_var._rotation_absolute, tr1); - tc2 = rot_apply(tr1, tc1); - _source_var._position_absolute = coords_add(_origin_var._position_absolute, tc2); - tc1 = coords_sub(_origin_var._position_absolute, _source_var._position_absolute); - _source_var._position_relative = rot_apply(_source_var._rotation_absolute, tc1); - } /* source=Source_simple() AT ROTATED */ - DEBUG_COMPONENT("source", _source_var._position_absolute, _source_var._rotation_absolute); - instrument->_position_absolute[2] = _source_var._position_absolute; - instrument->_position_relative[2] = _source_var._position_relative; - _source_var._position_relative_is_zero = coords_test_zero(_source_var._position_relative); - instrument->counter_N[2] = instrument->counter_P[2] = instrument->counter_P2[2] = 0; - instrument->counter_AbsorbProp[2] = 0; - return(0); -} /* _source_setpos */ -/* component guide=Guide_gravity() SETTING, POSITION/ROTATION */ -int _guide_setpos(void) -{ /* sets initial component parameters, position and rotation */ - SIG_MESSAGE("[_guide_setpos] component guide=Guide_gravity() SETTING [/home/g/.cache/mccode-mcstas/main/optics/Guide_gravity.comp:340]"); - stracpy(_guide_var._name, "guide", 6); - stracpy(_guide_var._type, "Guide_gravity", 14); - int current_setpos_index = _guide_var._index = 3; - _guide_var._parameters.w1 = 0.06; - _guide_var._parameters.h1 = 0.12; - _guide_var._parameters.w2 = 0.05; - _guide_var._parameters.h2 = 0.1; - _guide_var._parameters.l = 30; - _guide_var._parameters.R0 = 0.995; - _guide_var._parameters.Qc = 0.0218; - _guide_var._parameters.alpha = 4.38; - _guide_var._parameters.m = 4; - _guide_var._parameters.W = 0.003; - _guide_var._parameters.nslit = 1; - _guide_var._parameters.d = 0.0005; - _guide_var._parameters.mleft = -1; - _guide_var._parameters.mright = -1; - _guide_var._parameters.mtop = -1; - _guide_var._parameters.mbottom = -1; - _guide_var._parameters.nhslit = 1; - _guide_var._parameters.G = 0; - _guide_var._parameters.aleft = -1; - _guide_var._parameters.aright = -1; - _guide_var._parameters.atop = -1; - _guide_var._parameters.abottom = -1; - _guide_var._parameters.wavy = 0; - _guide_var._parameters.wavy_z = 0; - _guide_var._parameters.wavy_tb = 0; - _guide_var._parameters.wavy_lr = 0; - _guide_var._parameters.chamfers = 0; - _guide_var._parameters.chamfers_z = 0; - _guide_var._parameters.chamfers_lr = 0; - _guide_var._parameters.chamfers_tb = 0; - _guide_var._parameters.nelements = 1; - _guide_var._parameters.nu = 0; - _guide_var._parameters.phase = 0; - if ("NULL" && strlen("NULL")) - stracpy(_guide_var._parameters.reflect, "NULL" ? "NULL" : "", 16384); - else - _guide_var._parameters.reflect[0] = '\0'; - /* component guide=Guide_gravity() AT ROTATED */ - { - Coords tc1, tc2; - tc1 = coords_set(0,0,0); - tc2 = coords_set(0,0,0); - Rotation tr1; - rot_set_rotation(tr1,0,0,0); - rot_set_rotation(tr1, (0)*DEG2RAD, (0)*DEG2RAD, (0)*DEG2RAD); - rot_mul(tr1, _source_var._rotation_absolute, _guide_var._rotation_absolute); - rot_transpose(_source_var._rotation_absolute, tr1); - rot_mul(_guide_var._rotation_absolute, tr1, _guide_var._rotation_relative); - _guide_var._rotation_is_identity = rot_test_identity(_guide_var._rotation_relative); - tc1 = coords_set(0, 0, 1.5); - rot_transpose(_source_var._rotation_absolute, tr1); - tc2 = rot_apply(tr1, tc1); - _guide_var._position_absolute = coords_add(_source_var._position_absolute, tc2); - tc1 = coords_sub(_source_var._position_absolute, _guide_var._position_absolute); - _guide_var._position_relative = rot_apply(_guide_var._rotation_absolute, tc1); - } /* guide=Guide_gravity() AT ROTATED */ - DEBUG_COMPONENT("guide", _guide_var._position_absolute, _guide_var._rotation_absolute); - instrument->_position_absolute[3] = _guide_var._position_absolute; - instrument->_position_relative[3] = _guide_var._position_relative; - _guide_var._position_relative_is_zero = coords_test_zero(_guide_var._position_relative); - instrument->counter_N[3] = instrument->counter_P[3] = instrument->counter_P2[3] = 0; - instrument->counter_AbsorbProp[3] = 0; - return(0); -} /* _guide_setpos */ -/* component guide_end=Arm() SETTING, POSITION/ROTATION */ -int _guide_end_setpos(void) -{ /* sets initial component parameters, position and rotation */ - SIG_MESSAGE("[_guide_end_setpos] component guide_end=Arm() SETTING [guide_end:0]"); - stracpy(_guide_end_var._name, "guide_end", 10); - stracpy(_guide_end_var._type, "Arm", 4); - int current_setpos_index = _guide_end_var._index = 4; - /* component guide_end=Arm() AT ROTATED */ - { - Coords tc1, tc2; - tc1 = coords_set(0,0,0); - tc2 = coords_set(0,0,0); - Rotation tr1; - rot_set_rotation(tr1,0,0,0); - rot_set_rotation(tr1, (0)*DEG2RAD, (0)*DEG2RAD, (0)*DEG2RAD); - rot_mul(tr1, _guide_var._rotation_absolute, _guide_end_var._rotation_absolute); - rot_transpose(_guide_var._rotation_absolute, tr1); - rot_mul(_guide_end_var._rotation_absolute, tr1, _guide_end_var._rotation_relative); - _guide_end_var._rotation_is_identity = rot_test_identity(_guide_end_var._rotation_relative); - tc1 = coords_set(0, 0, 30); - rot_transpose(_guide_var._rotation_absolute, tr1); - tc2 = rot_apply(tr1, tc1); - _guide_end_var._position_absolute = coords_add(_guide_var._position_absolute, tc2); - tc1 = coords_sub(_guide_var._position_absolute, _guide_end_var._position_absolute); - _guide_end_var._position_relative = rot_apply(_guide_end_var._rotation_absolute, tc1); - } /* guide_end=Arm() AT ROTATED */ - DEBUG_COMPONENT("guide_end", _guide_end_var._position_absolute, _guide_end_var._rotation_absolute); - instrument->_position_absolute[4] = _guide_end_var._position_absolute; - instrument->_position_relative[4] = _guide_end_var._position_relative; - _guide_end_var._position_relative_is_zero = coords_test_zero(_guide_end_var._position_relative); - instrument->counter_N[4] = instrument->counter_P[4] = instrument->counter_P2[4] = 0; - instrument->counter_AbsorbProp[4] = 0; - return(0); -} /* _guide_end_setpos */ -/* component aperture=Slit() SETTING, POSITION/ROTATION */ -int _aperture_setpos(void) -{ /* sets initial component parameters, position and rotation */ - SIG_MESSAGE("[_aperture_setpos] component aperture=Slit() SETTING [/home/g/.cache/mccode-mcstas/main/optics/Slit.comp:66]"); - stracpy(_aperture_var._name, "aperture", 9); - stracpy(_aperture_var._type, "Slit", 5); - int current_setpos_index = _aperture_var._index = 5; - _aperture_var._parameters.xmin = UNSET; - _aperture_var._parameters.xmax = UNSET; - _aperture_var._parameters.ymin = UNSET; - _aperture_var._parameters.ymax = UNSET; - _aperture_var._parameters.radius = UNSET; - _aperture_var._parameters.xwidth = _instrument_var._parameters.virtual_source_x; - _aperture_var._parameters.yheight = _instrument_var._parameters.virtual_source_y; - /* component aperture=Slit() AT ROTATED */ - { - Coords tc1, tc2; - tc1 = coords_set(0,0,0); - tc2 = coords_set(0,0,0); - Rotation tr1; - rot_set_rotation(tr1,0,0,0); - rot_set_rotation(tr1, (0)*DEG2RAD, (0)*DEG2RAD, (0)*DEG2RAD); - rot_mul(tr1, _guide_end_var._rotation_absolute, _aperture_var._rotation_absolute); - rot_transpose(_guide_var._rotation_absolute, tr1); - rot_mul(_aperture_var._rotation_absolute, tr1, _aperture_var._rotation_relative); - _aperture_var._rotation_is_identity = rot_test_identity(_aperture_var._rotation_relative); - tc1 = coords_set(0, 0, 0.01); - rot_transpose(_guide_end_var._rotation_absolute, tr1); - tc2 = rot_apply(tr1, tc1); - _aperture_var._position_absolute = coords_add(_guide_end_var._position_absolute, tc2); - tc1 = coords_sub(_guide_var._position_absolute, _aperture_var._position_absolute); - _aperture_var._position_relative = rot_apply(_aperture_var._rotation_absolute, tc1); - } /* aperture=Slit() AT ROTATED */ - DEBUG_COMPONENT("aperture", _aperture_var._position_absolute, _aperture_var._rotation_absolute); - instrument->_position_absolute[5] = _aperture_var._position_absolute; - instrument->_position_relative[5] = _aperture_var._position_relative; - _aperture_var._position_relative_is_zero = coords_test_zero(_aperture_var._position_relative); - instrument->counter_N[5] = instrument->counter_P[5] = instrument->counter_P2[5] = 0; - instrument->counter_AbsorbProp[5] = 0; - return(0); -} /* _aperture_setpos */ -/* component split_at=Arm() SETTING, POSITION/ROTATION */ -int _split_at_setpos(void) -{ /* sets initial component parameters, position and rotation */ - SIG_MESSAGE("[_split_at_setpos] component split_at=Arm() SETTING [split_at:0]"); - stracpy(_split_at_var._name, "split_at", 9); - stracpy(_split_at_var._type, "Arm", 4); - int current_setpos_index = _split_at_var._index = 6; - /* component split_at=Arm() AT ROTATED */ - { - Coords tc1, tc2; - tc1 = coords_set(0,0,0); - tc2 = coords_set(0,0,0); - Rotation tr1; - rot_set_rotation(tr1,0,0,0); - rot_set_rotation(tr1, (0)*DEG2RAD, (0)*DEG2RAD, (0)*DEG2RAD); - rot_mul(tr1, _aperture_var._rotation_absolute, _split_at_var._rotation_absolute); - rot_transpose(_aperture_var._rotation_absolute, tr1); - rot_mul(_split_at_var._rotation_absolute, tr1, _split_at_var._rotation_relative); - _split_at_var._rotation_is_identity = rot_test_identity(_split_at_var._rotation_relative); - tc1 = coords_set(0, 0, 0.0001); - rot_transpose(_aperture_var._rotation_absolute, tr1); - tc2 = rot_apply(tr1, tc1); - _split_at_var._position_absolute = coords_add(_aperture_var._position_absolute, tc2); - tc1 = coords_sub(_aperture_var._position_absolute, _split_at_var._position_absolute); - _split_at_var._position_relative = rot_apply(_split_at_var._rotation_absolute, tc1); - } /* split_at=Arm() AT ROTATED */ - DEBUG_COMPONENT("split_at", _split_at_var._position_absolute, _split_at_var._rotation_absolute); - instrument->_position_absolute[6] = _split_at_var._position_absolute; - instrument->_position_relative[6] = _split_at_var._position_relative; - _split_at_var._position_relative_is_zero = coords_test_zero(_split_at_var._position_relative); - instrument->counter_N[6] = instrument->counter_P[6] = instrument->counter_P2[6] = 0; - instrument->counter_AbsorbProp[6] = 0; - return(0); -} /* _split_at_setpos */ -/* component mono_point=Arm() SETTING, POSITION/ROTATION */ -int _mono_point_setpos(void) -{ /* sets initial component parameters, position and rotation */ - SIG_MESSAGE("[_mono_point_setpos] component mono_point=Arm() SETTING [mono_point:0]"); - stracpy(_mono_point_var._name, "mono_point", 11); - stracpy(_mono_point_var._type, "Arm", 4); - int current_setpos_index = _mono_point_var._index = 7; - /* component mono_point=Arm() AT ROTATED */ - { - Coords tc1, tc2; - tc1 = coords_set(0,0,0); - tc2 = coords_set(0,0,0); - Rotation tr1; - rot_set_rotation(tr1,0,0,0); - rot_set_rotation(tr1, (0)*DEG2RAD, (0)*DEG2RAD, (0)*DEG2RAD); - rot_mul(tr1, _split_at_var._rotation_absolute, _mono_point_var._rotation_absolute); - rot_transpose(_aperture_var._rotation_absolute, tr1); - rot_mul(_mono_point_var._rotation_absolute, tr1, _mono_point_var._rotation_relative); - _mono_point_var._rotation_is_identity = rot_test_identity(_mono_point_var._rotation_relative); - tc1 = coords_set(0, 0, 0.8); - rot_transpose(_split_at_var._rotation_absolute, tr1); - tc2 = rot_apply(tr1, tc1); - _mono_point_var._position_absolute = coords_add(_split_at_var._position_absolute, tc2); - tc1 = coords_sub(_aperture_var._position_absolute, _mono_point_var._position_absolute); - _mono_point_var._position_relative = rot_apply(_mono_point_var._rotation_absolute, tc1); - } /* mono_point=Arm() AT ROTATED */ - DEBUG_COMPONENT("mono_point", _mono_point_var._position_absolute, _mono_point_var._rotation_absolute); - instrument->_position_absolute[7] = _mono_point_var._position_absolute; - instrument->_position_relative[7] = _mono_point_var._position_relative; - _mono_point_var._position_relative_is_zero = coords_test_zero(_mono_point_var._position_relative); - instrument->counter_N[7] = instrument->counter_P[7] = instrument->counter_P2[7] = 0; - instrument->counter_AbsorbProp[7] = 0; - return(0); -} /* _mono_point_setpos */ -/* component mono=Monochromator_curved() SETTING, POSITION/ROTATION */ -int _mono_setpos(void) -{ /* sets initial component parameters, position and rotation */ - SIG_MESSAGE("[_mono_setpos] component mono=Monochromator_curved() SETTING [/home/g/.cache/mccode-mcstas/main/optics/Monochromator_curved.comp:129]"); - stracpy(_mono_var._name, "mono", 5); - stracpy(_mono_var._type, "Monochromator_curved", 21); - int current_setpos_index = _mono_var._index = 8; - if ("NULL" && strlen("NULL")) - stracpy(_mono_var._parameters.reflect, "NULL" ? "NULL" : "", 16384); - else - _mono_var._parameters.reflect[0] = '\0'; - if ("NULL" && strlen("NULL")) - stracpy(_mono_var._parameters.transmit, "NULL" ? "NULL" : "", 16384); - else - _mono_var._parameters.transmit[0] = '\0'; - _mono_var._parameters.zwidth = 0.02; - _mono_var._parameters.yheight = 0.02; - _mono_var._parameters.gap = 0.0005; - _mono_var._parameters.NH = 13; - _mono_var._parameters.NV = 7; - _mono_var._parameters.mosaich = 30.0; - _mono_var._parameters.mosaicv = 30.0; - _mono_var._parameters.r0 = 0.7; - _mono_var._parameters.t0 = 1.0; - _mono_var._parameters.Q = 1.8734; - _mono_var._parameters.RV = 0; - _mono_var._parameters.RH = 0; - _mono_var._parameters.DM = 3.355; - _mono_var._parameters.mosaic = 0; - _mono_var._parameters.width = 0; - _mono_var._parameters.height = 0; - _mono_var._parameters.verbose = 0; - _mono_var._parameters.order = 0; - _mono_var._parameters.tiltH = NULL; - _mono_var._parameters.tiltV = NULL; - /* component mono=Monochromator_curved() AT ROTATED */ - { - Coords tc1, tc2; - tc1 = coords_set(0,0,0); - tc2 = coords_set(0,0,0); - Rotation tr1; - rot_set_rotation(tr1,0,0,0); - rot_set_rotation(tr1, (0)*DEG2RAD, (_instrument_var._parameters.a1)*DEG2RAD, (0)*DEG2RAD); - rot_mul(tr1, _mono_point_var._rotation_absolute, _mono_var._rotation_absolute); - rot_transpose(_aperture_var._rotation_absolute, tr1); - rot_mul(_mono_var._rotation_absolute, tr1, _mono_var._rotation_relative); - _mono_var._rotation_is_identity = rot_test_identity(_mono_var._rotation_relative); - tc1 = coords_set(0, 0, 0); - rot_transpose(_mono_point_var._rotation_absolute, tr1); - tc2 = rot_apply(tr1, tc1); - _mono_var._position_absolute = coords_add(_mono_point_var._position_absolute, tc2); - tc1 = coords_sub(_aperture_var._position_absolute, _mono_var._position_absolute); - _mono_var._position_relative = rot_apply(_mono_var._rotation_absolute, tc1); - } /* mono=Monochromator_curved() AT ROTATED */ - DEBUG_COMPONENT("mono", _mono_var._position_absolute, _mono_var._rotation_absolute); - instrument->_position_absolute[8] = _mono_var._position_absolute; - instrument->_position_relative[8] = _mono_var._position_relative; - _mono_var._position_relative_is_zero = coords_test_zero(_mono_var._position_relative); - instrument->counter_N[8] = instrument->counter_P[8] = instrument->counter_P2[8] = 0; - instrument->counter_AbsorbProp[8] = 0; - return(0); -} /* _mono_setpos */ -/* component sample_arm=Arm() SETTING, POSITION/ROTATION */ -int _sample_arm_setpos(void) -{ /* sets initial component parameters, position and rotation */ - SIG_MESSAGE("[_sample_arm_setpos] component sample_arm=Arm() SETTING [sample_arm:0]"); - stracpy(_sample_arm_var._name, "sample_arm", 11); - stracpy(_sample_arm_var._type, "Arm", 4); - int current_setpos_index = _sample_arm_var._index = 9; - /* component sample_arm=Arm() AT ROTATED */ - { - Coords tc1, tc2; - tc1 = coords_set(0,0,0); - tc2 = coords_set(0,0,0); - Rotation tr1; - rot_set_rotation(tr1,0,0,0); - rot_set_rotation(tr1, (0)*DEG2RAD, (_instrument_var._parameters.a2)*DEG2RAD, (0)*DEG2RAD); - rot_mul(tr1, _mono_point_var._rotation_absolute, _sample_arm_var._rotation_absolute); - rot_transpose(_mono_var._rotation_absolute, tr1); - rot_mul(_sample_arm_var._rotation_absolute, tr1, _sample_arm_var._rotation_relative); - _sample_arm_var._rotation_is_identity = rot_test_identity(_sample_arm_var._rotation_relative); - tc1 = coords_set(0, 0, 0); - rot_transpose(_mono_point_var._rotation_absolute, tr1); - tc2 = rot_apply(tr1, tc1); - _sample_arm_var._position_absolute = coords_add(_mono_point_var._position_absolute, tc2); - tc1 = coords_sub(_mono_var._position_absolute, _sample_arm_var._position_absolute); - _sample_arm_var._position_relative = rot_apply(_sample_arm_var._rotation_absolute, tc1); - } /* sample_arm=Arm() AT ROTATED */ - DEBUG_COMPONENT("sample_arm", _sample_arm_var._position_absolute, _sample_arm_var._rotation_absolute); - instrument->_position_absolute[9] = _sample_arm_var._position_absolute; - instrument->_position_relative[9] = _sample_arm_var._position_relative; - _sample_arm_var._position_relative_is_zero = coords_test_zero(_sample_arm_var._position_relative); - instrument->counter_N[9] = instrument->counter_P[9] = instrument->counter_P2[9] = 0; - instrument->counter_AbsorbProp[9] = 0; - return(0); -} /* _sample_arm_setpos */ -/* component detector=Monitor() SETTING, POSITION/ROTATION */ -int _detector_setpos(void) -{ /* sets initial component parameters, position and rotation */ - SIG_MESSAGE("[_detector_setpos] component detector=Monitor() SETTING [/home/g/.cache/mccode-mcstas/main/monitors/Monitor.comp:58]"); - stracpy(_detector_var._name, "detector", 9); - stracpy(_detector_var._type, "Monitor", 8); - int current_setpos_index = _detector_var._index = 10; - _detector_var._parameters.xmin = -0.05; - _detector_var._parameters.xmax = 0.05; - _detector_var._parameters.ymin = -0.05; - _detector_var._parameters.ymax = 0.05; - _detector_var._parameters.xwidth = 0.01; - _detector_var._parameters.yheight = 0.05; - _detector_var._parameters.restore_neutron = 0; - /* component detector=Monitor() AT ROTATED */ - { - Coords tc1, tc2; - tc1 = coords_set(0,0,0); - tc2 = coords_set(0,0,0); - Rotation tr1; - rot_set_rotation(tr1,0,0,0); - rot_set_rotation(tr1, (0)*DEG2RAD, (0)*DEG2RAD, (0)*DEG2RAD); - rot_mul(tr1, _sample_arm_var._rotation_absolute, _detector_var._rotation_absolute); - rot_transpose(_mono_var._rotation_absolute, tr1); - rot_mul(_detector_var._rotation_absolute, tr1, _detector_var._rotation_relative); - _detector_var._rotation_is_identity = rot_test_identity(_detector_var._rotation_relative); - tc1 = coords_set(0, 0, 0.8); - rot_transpose(_sample_arm_var._rotation_absolute, tr1); - tc2 = rot_apply(tr1, tc1); - _detector_var._position_absolute = coords_add(_sample_arm_var._position_absolute, tc2); - tc1 = coords_sub(_mono_var._position_absolute, _detector_var._position_absolute); - _detector_var._position_relative = rot_apply(_detector_var._rotation_absolute, tc1); - } /* detector=Monitor() AT ROTATED */ - DEBUG_COMPONENT("detector", _detector_var._position_absolute, _detector_var._rotation_absolute); - instrument->_position_absolute[10] = _detector_var._position_absolute; - instrument->_position_relative[10] = _detector_var._position_relative; - _detector_var._position_relative_is_zero = coords_test_zero(_detector_var._position_relative); - instrument->counter_N[10] = instrument->counter_P[10] = instrument->counter_P2[10] = 0; - instrument->counter_AbsorbProp[10] = 0; - return(0); -} /* _detector_setpos */ -_class_Source_simple *class_Source_simple_initialize(_class_Source_simple *_comp) { -#define radius (_comp->_parameters.radius) -#define yheight (_comp->_parameters.yheight) -#define xwidth (_comp->_parameters.xwidth) -#define dist (_comp->_parameters.dist) -#define focus_xw (_comp->_parameters.focus_xw) -#define focus_yh (_comp->_parameters.focus_yh) -#define E0 (_comp->_parameters.E0) -#define dE (_comp->_parameters.dE) -#define lambda0 (_comp->_parameters.lambda0) -#define dlambda (_comp->_parameters.dlambda) -#define flux (_comp->_parameters.flux) -#define gauss (_comp->_parameters.gauss) -#define target_index (_comp->_parameters.target_index) -#define pmul (_comp->_parameters.pmul) -#define srcArea (_comp->_parameters.srcArea) -#define square (_comp->_parameters.square) -#define tx (_comp->_parameters.tx) -#define ty (_comp->_parameters.ty) -#define tz (_comp->_parameters.tz) -SIG_MESSAGE("[_Source_simple_initialize] component NULL=Source_simple() [/home/g/.cache/mccode-mcstas/main/sources/Source_simple.comp:68]"); - -square = 0; -/* Determine source area */ -if (radius && !yheight && !xwidth ) { - square = 0; - srcArea = PI*radius*radius; - } else if(yheight && xwidth) { - square = 1; - srcArea = xwidth * yheight; - } - - if (flux) { - pmul=flux*1e4*srcArea/mcget_ncount(); - if (dlambda) - pmul *= 2*dlambda; - else if (dE) - pmul *= 2*dE; - } else { - gauss = 0; - pmul=1.0/(mcget_ncount()*4*PI); - } - - if (target_index && !dist) - { - Coords ToTarget; - ToTarget = coords_sub(POS_A_COMP_INDEX(INDEX_CURRENT_COMP+target_index),POS_A_CURRENT_COMP); - ToTarget = rot_apply(ROT_A_CURRENT_COMP, ToTarget); - coords_get(ToTarget, &tx, &ty, &tz); - dist=sqrt(tx*tx+ty*ty+tz*tz); - } else if (dist) { - tx = 0; - ty = 0; - tz = dist; - } - - if (srcArea <= 0) { - printf("Source_simple: %s: Source area is <= 0 !\n ERROR - Exiting\n", - NAME_CURRENT_COMP); - exit(0); - } - if (dist <= 0 || focus_xw <= 0 || focus_yh <= 0) { - printf("Source_simple: %s: Target area unmeaningful! (negative dist / focus_xw / focus_yh)\n ERROR - Exiting\n", - NAME_CURRENT_COMP); - exit(0); - } - - if ((!lambda0 && !E0 && !dE && !dlambda)) { - printf("Source_simple: %s: You must specify either a wavelength or energy range!\n ERROR - Exiting\n", - NAME_CURRENT_COMP); - exit(0); - } - if ((!lambda0 && !dlambda && (E0 <= 0 || dE < 0 || E0-dE <= 0)) - || (!E0 && !dE && (lambda0 <= 0 || dlambda < 0 || lambda0-dlambda <= 0))) { - printf("Source_simple: %s: Unmeaningful definition of wavelength or energy range!\n ERROR - Exiting\n", - NAME_CURRENT_COMP); - exit(0); - } - -#undef radius -#undef yheight -#undef xwidth -#undef dist -#undef focus_xw -#undef focus_yh -#undef E0 -#undef dE -#undef lambda0 -#undef dlambda -#undef flux -#undef gauss -#undef target_index -#undef pmul -#undef srcArea -#undef square -#undef tx -#undef ty -#undef tz - return(_comp); -} /* class_Source_simple_initialize */ - -_class_Guide_gravity *class_Guide_gravity_initialize(_class_Guide_gravity *_comp) { -#define w1 (_comp->_parameters.w1) -#define h1 (_comp->_parameters.h1) -#define w2 (_comp->_parameters.w2) -#define h2 (_comp->_parameters.h2) -#define l (_comp->_parameters.l) -#define R0 (_comp->_parameters.R0) -#define Qc (_comp->_parameters.Qc) -#define alpha (_comp->_parameters.alpha) -#define m (_comp->_parameters.m) -#define W (_comp->_parameters.W) -#define nslit (_comp->_parameters.nslit) -#define d (_comp->_parameters.d) -#define mleft (_comp->_parameters.mleft) -#define mright (_comp->_parameters.mright) -#define mtop (_comp->_parameters.mtop) -#define mbottom (_comp->_parameters.mbottom) -#define nhslit (_comp->_parameters.nhslit) -#define G (_comp->_parameters.G) -#define aleft (_comp->_parameters.aleft) -#define aright (_comp->_parameters.aright) -#define atop (_comp->_parameters.atop) -#define abottom (_comp->_parameters.abottom) -#define wavy (_comp->_parameters.wavy) -#define wavy_z (_comp->_parameters.wavy_z) -#define wavy_tb (_comp->_parameters.wavy_tb) -#define wavy_lr (_comp->_parameters.wavy_lr) -#define chamfers (_comp->_parameters.chamfers) -#define chamfers_z (_comp->_parameters.chamfers_z) -#define chamfers_lr (_comp->_parameters.chamfers_lr) -#define chamfers_tb (_comp->_parameters.chamfers_tb) -#define nelements (_comp->_parameters.nelements) -#define nu (_comp->_parameters.nu) -#define phase (_comp->_parameters.phase) -#define reflect (_comp->_parameters.reflect) -#define GVars (_comp->_parameters.GVars) -#define pTable (_comp->_parameters.pTable) -#define table_present (_comp->_parameters.table_present) -SIG_MESSAGE("[_Guide_gravity_initialize] component NULL=Guide_gravity() [/home/g/.cache/mccode-mcstas/main/optics/Guide_gravity.comp:340]"); - - double Gx=0, Gy=-GRAVITY, Gz=0; - Coords mcLocG; - int i; - - if (reflect && strlen(reflect) && strcmp(reflect,"NULL") && strcmp(reflect,"0")) { - if (Table_Read(&pTable, reflect, 1) <= 0) /* read 1st block data from file into pTable */ - exit(fprintf(stderr,"Guide_gravity: %s: can not read file %s\n", NAME_CURRENT_COMP, reflect)); - table_present=1; - } else { - table_present=0; - if (W < 0 || R0 < 0 || Qc < 0) - { fprintf(stderr,"Guide_gravity: %s: W R0 Qc must be >0.\n", NAME_CURRENT_COMP); - exit(-1); } - } - - if (nslit <= 0 || nhslit <= 0) - { fprintf(stderr,"Guide_gravity: %s: nslit nhslit must be >0.\n", NAME_CURRENT_COMP); - exit(-1); } - - if (!w1 || !h1) - { fprintf(stderr,"Guide_gravity: %s: input window is closed (w1=h1=0).\n", NAME_CURRENT_COMP); - exit(-1); } - - if (d*(nslit-1) > w1) exit(fprintf(stderr, "Guide_gravity: %s: absorbing walls fill input window. No space left for transmission (d*(nslit-1) > w1).\n", NAME_CURRENT_COMP)); - - if (!w2) w2=w1; - if (!h2) h2=h1; - - if (mcgravitation) G=-GRAVITY; - mcLocG = rot_apply(ROT_A_CURRENT_COMP, coords_set(0,G,0)); - coords_get(mcLocG, &Gx, &Gy, &Gz); - - strcpy(GVars.compcurname, NAME_CURRENT_COMP); - - if (l > 0 && nelements > 0) { - - Gravity_guide_Init(&GVars, - w1, h1, w2, h2, l, R0, - Qc, alpha, m, W, nslit, d, - Gx, Gy, Gz, mleft, mright, mtop, - mbottom, nhslit, wavy_lr, wavy_tb, wavy_z, wavy, - chamfers_z, chamfers_lr, chamfers_tb, chamfers,nu,phase,aleft,aright,atop,abottom); - if (!G) for (i=0; i<5; GVars.A[i++] = 0); - if (GVars.fc_freq != 0 || GVars.fc_phase != 0) { - if (w1 != w2 || h1 != h2) - exit(fprintf(stderr,"Guide_gravity: %s: rotating slit pack must be straight (w1=w2 and h1=h2).\n", NAME_CURRENT_COMP)); - printf("Guide_gravity: %s: Fermi Chopper mode: frequency=%g [Hz] phase=%g [deg]\n", - NAME_CURRENT_COMP, GVars.fc_freq, GVars.fc_phase); - } - } else printf("Guide_gravity: %s: unactivated (l=0 or nelements=0)\n", NAME_CURRENT_COMP); - - -#undef w1 -#undef h1 -#undef w2 -#undef h2 -#undef l -#undef R0 -#undef Qc -#undef alpha -#undef m -#undef W -#undef nslit -#undef d -#undef mleft -#undef mright -#undef mtop -#undef mbottom -#undef nhslit -#undef G -#undef aleft -#undef aright -#undef atop -#undef abottom -#undef wavy -#undef wavy_z -#undef wavy_tb -#undef wavy_lr -#undef chamfers -#undef chamfers_z -#undef chamfers_lr -#undef chamfers_tb -#undef nelements -#undef nu -#undef phase -#undef reflect -#undef GVars -#undef pTable -#undef table_present - return(_comp); -} /* class_Guide_gravity_initialize */ - -_class_Slit *class_Slit_initialize(_class_Slit *_comp) { -#define xmin (_comp->_parameters.xmin) -#define xmax (_comp->_parameters.xmax) -#define ymin (_comp->_parameters.ymin) -#define ymax (_comp->_parameters.ymax) -#define radius (_comp->_parameters.radius) -#define xwidth (_comp->_parameters.xwidth) -#define yheight (_comp->_parameters.yheight) -#define isradial (_comp->_parameters.isradial) -SIG_MESSAGE("[_Slit_initialize] component NULL=Slit() [/home/g/.cache/mccode-mcstas/main/optics/Slit.comp:66]"); - -if (is_unset(radius)){ - isradial=0; - if (all_set(3, xwidth, xmin, xmax)){ - slit_error_if(xwidth != xmax - xmin, "specifying xwidth, xmin and xmax requires consistent parameters", NAME_CURRENT_COMP); - } else { - slit_error_if(is_unset(xwidth) && any_unset(2, xmin, xmax), "specify either xwidth or xmin & xmax", NAME_CURRENT_COMP); - } - if (all_set(3, yheight, ymin, ymax)){ - slit_error_if(yheight != ymax - ymin, "specifying yheight, ymin and ymax requires consistent parameters", NAME_CURRENT_COMP); - } else { - slit_error_if(is_unset(yheight) && any_unset(2, ymin, ymax), "specify either yheight or ymin & ymax", NAME_CURRENT_COMP); - } - if (is_unset(xmin)) { // xmax also unset but xwidth *is* set - xmax = xwidth/2; - xmin = -xmax; - } - if (is_unset(ymin)) { // ymax also unset but yheight *is* set - ymax = yheight/2; - ymin = -ymax; - } - slit_warning_if(xmin == xmax || ymin == ymax, "Running with CLOSED rectangular slit - is this intentional?", NAME_CURRENT_COMP); -} else { - isradial=1; - slit_error_if(any_set(6, xwidth, xmin, xmax, yheight, ymin, ymax), - "specify radius OR width and height parameters", NAME_CURRENT_COMP); - slit_warning_if(radius == 0., "Running with CLOSED radial slit - is this intentional?", NAME_CURRENT_COMP); -} - - -#undef xmin -#undef xmax -#undef ymin -#undef ymax -#undef radius -#undef xwidth -#undef yheight -#undef isradial - return(_comp); -} /* class_Slit_initialize */ - -_class_Monochromator_curved *class_Monochromator_curved_initialize(_class_Monochromator_curved *_comp) { -#define reflect (_comp->_parameters.reflect) -#define transmit (_comp->_parameters.transmit) -#define zwidth (_comp->_parameters.zwidth) -#define yheight (_comp->_parameters.yheight) -#define gap (_comp->_parameters.gap) -#define NH (_comp->_parameters.NH) -#define NV (_comp->_parameters.NV) -#define mosaich (_comp->_parameters.mosaich) -#define mosaicv (_comp->_parameters.mosaicv) -#define r0 (_comp->_parameters.r0) -#define t0 (_comp->_parameters.t0) -#define Q (_comp->_parameters.Q) -#define RV (_comp->_parameters.RV) -#define RH (_comp->_parameters.RH) -#define DM (_comp->_parameters.DM) -#define mosaic (_comp->_parameters.mosaic) -#define width (_comp->_parameters.width) -#define height (_comp->_parameters.height) -#define verbose (_comp->_parameters.verbose) -#define order (_comp->_parameters.order) -#define mos_rms_y (_comp->_parameters.mos_rms_y) -#define mos_rms_z (_comp->_parameters.mos_rms_z) -#define mos_rms_max (_comp->_parameters.mos_rms_max) -#define mono_Q (_comp->_parameters.mono_Q) -#define SlabWidth (_comp->_parameters.SlabWidth) -#define SlabHeight (_comp->_parameters.SlabHeight) -#define rTable (_comp->_parameters.rTable) -#define tTable (_comp->_parameters.tTable) -#define rTableFlag (_comp->_parameters.rTableFlag) -#define tTableFlag (_comp->_parameters.tTableFlag) -#define tiltH (_comp->_parameters.tiltH) -#define tiltV (_comp->_parameters.tiltV) -SIG_MESSAGE("[_Monochromator_curved_initialize] component NULL=Monochromator_curved() [/home/g/.cache/mccode-mcstas/main/optics/Monochromator_curved.comp:129]"); - - int i; - - if (mosaic != 0) { - mos_rms_y = MIN2RAD*mosaic/sqrt(8*log(2)); - mos_rms_z = mos_rms_y; } - else { - mos_rms_y = MIN2RAD*mosaich/sqrt(8*log(2)); - mos_rms_z = MIN2RAD*mosaicv/sqrt(8*log(2)); } - mos_rms_max = mos_rms_y > mos_rms_z ? mos_rms_y : mos_rms_z; - - mono_Q = Q; - if (DM != 0) mono_Q = 2*PI/DM; - - if (mono_Q <= 0) { fprintf(stderr,"Monochromator_curved: %s: Error scattering vector Q = 0\n", NAME_CURRENT_COMP); exit(-1); } - if (r0 < 0) { fprintf(stderr,"Monochromator_curved: %s: Error reflectivity r0 is negative\n", NAME_CURRENT_COMP); exit(-1); } - if (r0 == 0) { fprintf(stderr,"Monochromator_curved: %s: Reflectivity r0 is null. Ignoring component.\n", NAME_CURRENT_COMP); } - if (NH*NV == 0) { fprintf(stderr,"Monochromator_curved: %s: no slabs ??? (NH or NV=0)\n", NAME_CURRENT_COMP); exit(-1); } - - - if (verbose && r0) - { - printf("Monochromator_curved: component %s Q=%.3g Angs-1 (DM=%.4g Angs)\n", NAME_CURRENT_COMP, mono_Q, 2*PI/mono_Q); - if (NH*NV == 1) printf(" flat.\n"); - else - { if (NH > 1) - { printf(" horizontal: %i blades", (int)NH); - if (RH != 0) printf(" focusing with RH=%.3g [m]", RH); - printf("\n"); - } - if (NV > 1) - { printf(" vertical: %i blades", (int)NV); - if (RV != 0) printf(" focusing with RV=%.3g [m]", RV); - printf("\n"); - } - } - } - - if (reflect != NULL && r0 && strlen(reflect) && strcmp(reflect,"NULL") && strcmp(reflect,"0")) - { - if (verbose) fprintf(stdout, "Monochromator_curved: %s: Reflectivity data (k, R) from %s\n", NAME_CURRENT_COMP, reflect); - Table_Read(&rTable, reflect, 1); /* read 1st block data from file into rTable */ - Table_Rebin(&rTable); /* rebin as evenly, increasing array */ - if (rTable.rows < 2) Table_Free(&rTable); - if (verbose) Table_Info(rTable); - rTableFlag = 1; - } else { - rTableFlag = 0; - } - if (transmit != NULL && strlen(transmit) && strcmp(transmit,"NULL") && strcmp(transmit,"0")) - { - if (verbose) fprintf(stdout, "Monochromator_curved: %s: Transmission data (k, T) from %s\n", NAME_CURRENT_COMP, transmit); - Table_Read(&tTable, transmit, 1); /* read 1st block data from file into rTable */ - Table_Rebin(&tTable); /* rebin as evenly, increasing array */ - if (tTable.rows < 2) Table_Free(&tTable); - if (verbose) Table_Info(tTable); - tTableFlag = 1; - } else { - tTableFlag = 0; - } - - if (width == 0) SlabWidth = zwidth; - else SlabWidth = (width+gap)/NH - gap; - if (height == 0) SlabHeight = yheight; - else SlabHeight = (height+gap)/NV - gap; - - tiltH=calloc((int)2*(NH+1),sizeof(double)); - tiltV=calloc((int)2*(NV+1),sizeof(double)); - - if (!tiltH) printf("Monochromator_curved: %s: Warning: not enough memory to allocate tilts (NH=%g).\n", NAME_CURRENT_COMP, NH); - else if (RH) { /* pre-compute tilts */ - for (i=0;i<=NH;i++){ - tiltH[i]=asin((i-(NH+1)/2.0)*(SlabWidth+gap)/RH); - } - } - if (!tiltV) printf("Monochromator_curved: %s: Warning: not enough memory to allocate tilts (NV=%g).\n", NAME_CURRENT_COMP, NV); - else if (RV) { - for (i=0;i<=NV;i++){ - tiltV[i]=-asin((i-(NV+1)/2.0)*(SlabHeight+gap)/RV); - } - } - - -#undef reflect -#undef transmit -#undef zwidth -#undef yheight -#undef gap -#undef NH -#undef NV -#undef mosaich -#undef mosaicv -#undef r0 -#undef t0 -#undef Q -#undef RV -#undef RH -#undef DM -#undef mosaic -#undef width -#undef height -#undef verbose -#undef order -#undef mos_rms_y -#undef mos_rms_z -#undef mos_rms_max -#undef mono_Q -#undef SlabWidth -#undef SlabHeight -#undef rTable -#undef tTable -#undef rTableFlag -#undef tTableFlag -#undef tiltH -#undef tiltV - return(_comp); -} /* class_Monochromator_curved_initialize */ - -_class_Monitor *class_Monitor_initialize(_class_Monitor *_comp) { -#define xmin (_comp->_parameters.xmin) -#define xmax (_comp->_parameters.xmax) -#define ymin (_comp->_parameters.ymin) -#define ymax (_comp->_parameters.ymax) -#define xwidth (_comp->_parameters.xwidth) -#define yheight (_comp->_parameters.yheight) -#define restore_neutron (_comp->_parameters.restore_neutron) -#define Nsum (_comp->_parameters.Nsum) -#define psum (_comp->_parameters.psum) -#define p2sum (_comp->_parameters.p2sum) -SIG_MESSAGE("[_Monitor_initialize] component NULL=Monitor() [/home/g/.cache/mccode-mcstas/main/monitors/Monitor.comp:58]"); - -psum = 0; -p2sum = 0; -Nsum = 0; - -if (xwidth > 0) { xmax = xwidth/2; xmin = -xmax; } - if (yheight > 0) { ymax = yheight/2; ymin = -ymax; } - - if ((xmin >= xmax) || (ymin >= ymax)) { - printf("Monitor: %s: Null detection area !\n" - "ERROR (xwidth,yheight,xmin,xmax,ymin,ymax). Exiting", - NAME_CURRENT_COMP); - exit(0); - } - -#undef xmin -#undef xmax -#undef ymin -#undef ymax -#undef xwidth -#undef yheight -#undef restore_neutron -#undef Nsum -#undef psum -#undef p2sum - return(_comp); -} /* class_Monitor_initialize */ - -int init(void) { /* called by mccode_main for splitRunTest:INITIALIZE */ - DEBUG_INSTR(); - - /* code_main/parseoptions/readparams sets instrument parameters value */ - stracpy(instrument->_name, "splitRunTest", 13); - _origin_setpos(); /* type Arm */ - _source_setpos(); /* type Source_simple */ - _guide_setpos(); /* type Guide_gravity */ - _guide_end_setpos(); /* type Arm */ - _aperture_setpos(); /* type Slit */ - _split_at_setpos(); /* type Arm */ - _mono_point_setpos(); /* type Arm */ - _mono_setpos(); /* type Monochromator_curved */ - _sample_arm_setpos(); /* type Arm */ - _detector_setpos(); /* type Monitor */ -/* call iteratively all components INITIALIZE */ - class_Source_simple_initialize(&_source_var); - class_Guide_gravity_initialize(&_guide_var); - class_Slit_initialize(&_aperture_var); - class_Monochromator_curved_initialize(&_mono_var); - class_Monitor_initialize(&_detector_var); - if (mcdotrace) display(); - DEBUG_INSTR_END(); - -#ifdef OPENACC -#include -#pragma acc update device(_origin_var) -#pragma acc update device(_source_var) -#pragma acc update device(_guide_var) -#pragma acc update device(_guide_end_var) -#pragma acc update device(_aperture_var) -#pragma acc update device(_split_at_var) -#pragma acc update device(_mono_point_var) -#pragma acc update device(_mono_var) -#pragma acc update device(_sample_arm_var) -#pragma acc update device(_detector_var) -#pragma acc update device(_instrument_var) -#endif - - return(0); -} /* initialize */ - -/******************************************************************************* -* components TRACE -*******************************************************************************/ -#define x (_particle->x) -#define y (_particle->y) -#define z (_particle->z) -#define vx (_particle->vx) -#define vy (_particle->vy) -#define vz (_particle->vz) -#define t (_particle->t) -#define sx (_particle->sx) -#define sy (_particle->sy) -#define sz (_particle->sz) -#define p (_particle->p) -#define mcgravitation (_particle->mcgravitation) -#define mcMagnet (_particle->mcMagnet) -#define allow_backprop (_particle->allow_backprop) -/* if on GPU, globally nullify sprintf,fprintf,printfs */ -/* (Similar defines are available in each comp trace but */ -/* those are not enough to handle external libs etc. ) */ -#ifdef OPENACC -#ifndef MULTICORE -#define fprintf(stderr,...) printf(__VA_ARGS__) -#define sprintf(string,...) printf(__VA_ARGS__) -#define exit(...) noprintf() -#define strcmp(a,b) str_comp(a,b) -#define strlen(a) str_len(a) -#endif -#endif -#define SCATTERED (_particle->_scattered) -#define RESTORE (_particle->_restore) -#define ABSORBED (_particle->_absorbed) -#define RESTORE_NEUTRON(_index, ...) _particle->_restore = _index; -#define mcget_run_num() _particle->_uid -#define ABSORB0 do { DEBUG_STATE(); DEBUG_ABSORB(); MAGNET_OFF; ABSORBED++; return(_comp); } while(0) -#define ABSORB ABSORB0 - -#pragma acc routine -_class_Source_simple *class_Source_simple_trace(_class_Source_simple *_comp, _class_particle *_particle) { -ABSORBED=SCATTERED=RESTORE=0; -#define radius (_comp->_parameters.radius) -#define yheight (_comp->_parameters.yheight) -#define xwidth (_comp->_parameters.xwidth) -#define dist (_comp->_parameters.dist) -#define focus_xw (_comp->_parameters.focus_xw) -#define focus_yh (_comp->_parameters.focus_yh) -#define E0 (_comp->_parameters.E0) -#define dE (_comp->_parameters.dE) -#define lambda0 (_comp->_parameters.lambda0) -#define dlambda (_comp->_parameters.dlambda) -#define flux (_comp->_parameters.flux) -#define gauss (_comp->_parameters.gauss) -#define target_index (_comp->_parameters.target_index) -#define pmul (_comp->_parameters.pmul) -#define srcArea (_comp->_parameters.srcArea) -#define square (_comp->_parameters.square) -#define tx (_comp->_parameters.tx) -#define ty (_comp->_parameters.ty) -#define tz (_comp->_parameters.tz) -SIG_MESSAGE("[_Source_simple_trace] component NULL=Source_simple() [/home/g/.cache/mccode-mcstas/main/sources/Source_simple.comp:127]"); - - double chi,E,lambda,v,r, xf, yf, rf, dx, dy, pdir; - - t=0; - z=0; - - if (square == 1) { - x = xwidth * (rand01() - 0.5); - y = yheight * (rand01() - 0.5); - } else { - chi=2*PI*rand01(); /* Choose point on source */ - r=sqrt(rand01())*radius; /* with uniform distribution. */ - x=r*cos(chi); - y=r*sin(chi); - } - randvec_target_rect_real(&xf, &yf, &rf, &pdir, - tx, ty, tz, focus_xw, focus_yh, ROT_A_CURRENT_COMP, x, y, z, 2); - - dx = xf-x; - dy = yf-y; - rf = sqrt(dx*dx+dy*dy+rf*rf); - - p = pdir*pmul; - - if(lambda0==0) { - if (!gauss) { - E=E0+dE*randpm1(); /* Choose from uniform distribution */ - } else { - E=E0+randnorm()*dE; - } - v=sqrt(E)*SE2V; - } else { - if (!gauss) { - lambda=lambda0+dlambda*randpm1(); - } else { - lambda=lambda0+randnorm()*dlambda; - } - v = K2V*(2*PI/lambda); - } - - vz=v*dist/rf; - vy=v*dy/rf; - vx=v*dx/rf; - -#ifndef NOABSORB_INF_NAN - /* Check for nan or inf particle parms */ - if(isnan(p) || isinf(p)) ABSORB; - if(isnan(t) || isinf(t)) ABSORB; - if(isnan(vx) || isinf(vx)) ABSORB; - if(isnan(vy) || isinf(vy)) ABSORB; - if(isnan(vz) || isinf(vz)) ABSORB; - if(isnan(x) || isinf(x)) ABSORB; - if(isnan(y) || isinf(y)) ABSORB; - if(isnan(z) || isinf(z)) ABSORB; -#else - if(isnan(p) || isinf(p)) printf("NAN or INF found in p, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(t) || isinf(t)) printf("NAN or INF found in t, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vx) || isinf(vx)) printf("NAN or INF found in vx, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vy) || isinf(vy)) printf("NAN or INF found in vy, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vz) || isinf(vz)) printf("NAN or INF found in vz, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(x) || isinf(x)) printf("NAN or INF found in x, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(y) || isinf(y)) printf("NAN or INF found in y, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(z) || isinf(z)) printf("NAN or INF found in z, %s (particle %lld)\n",_comp->_name,_particle->_uid); -#endif -#undef radius -#undef yheight -#undef xwidth -#undef dist -#undef focus_xw -#undef focus_yh -#undef E0 -#undef dE -#undef lambda0 -#undef dlambda -#undef flux -#undef gauss -#undef target_index -#undef pmul -#undef srcArea -#undef square -#undef tx -#undef ty -#undef tz - return(_comp); -} /* class_Source_simple_trace */ - -#pragma acc routine -_class_Guide_gravity *class_Guide_gravity_trace(_class_Guide_gravity *_comp, _class_particle *_particle) { -ABSORBED=SCATTERED=RESTORE=0; -#define w1 (_comp->_parameters.w1) -#define h1 (_comp->_parameters.h1) -#define w2 (_comp->_parameters.w2) -#define h2 (_comp->_parameters.h2) -#define l (_comp->_parameters.l) -#define R0 (_comp->_parameters.R0) -#define Qc (_comp->_parameters.Qc) -#define alpha (_comp->_parameters.alpha) -#define m (_comp->_parameters.m) -#define W (_comp->_parameters.W) -#define nslit (_comp->_parameters.nslit) -#define d (_comp->_parameters.d) -#define mleft (_comp->_parameters.mleft) -#define mright (_comp->_parameters.mright) -#define mtop (_comp->_parameters.mtop) -#define mbottom (_comp->_parameters.mbottom) -#define nhslit (_comp->_parameters.nhslit) -#define G (_comp->_parameters.G) -#define aleft (_comp->_parameters.aleft) -#define aright (_comp->_parameters.aright) -#define atop (_comp->_parameters.atop) -#define abottom (_comp->_parameters.abottom) -#define wavy (_comp->_parameters.wavy) -#define wavy_z (_comp->_parameters.wavy_z) -#define wavy_tb (_comp->_parameters.wavy_tb) -#define wavy_lr (_comp->_parameters.wavy_lr) -#define chamfers (_comp->_parameters.chamfers) -#define chamfers_z (_comp->_parameters.chamfers_z) -#define chamfers_lr (_comp->_parameters.chamfers_lr) -#define chamfers_tb (_comp->_parameters.chamfers_tb) -#define nelements (_comp->_parameters.nelements) -#define nu (_comp->_parameters.nu) -#define phase (_comp->_parameters.phase) -#define reflect (_comp->_parameters.reflect) -#define GVars (_comp->_parameters.GVars) -#define pTable (_comp->_parameters.pTable) -#define table_present (_comp->_parameters.table_present) -SIG_MESSAGE("[_Guide_gravity_trace] component NULL=Guide_gravity() [/home/g/.cache/mccode-mcstas/main/optics/Guide_gravity.comp:395]"); - - if (l > 0 && nelements > 0) { - double B, C, dt; - int ret, bounces = 0, i=0; - double this_width, this_height; - double angle=0; - - if (GVars.fc_freq != 0 || GVars.fc_phase != 0) { /* rotate neutron w/r to guide element */ - /* approximation of rotating straight Fermi Chopper */ - Coords X = coords_set(x,y,z-l/2); /* current coordinates of neutron in centered static frame */ - Rotation R; - double dt=(-z+l/2)/vz; /* time shift to each center of slit package */ - angle=fmod(360*GVars.fc_freq*(t+dt)+GVars.fc_phase, 360); /* in deg */ - /* modify angle so that Z0 guide side is always in front of incoming neutron */ - if (angle > 90 && angle < 270) { angle -= 180; } - angle *= DEG2RAD; - rot_set_rotation(R, 0, -angle, 0); /* will rotate neutron instead of comp: negative side */ - /* apply rotation to centered coordinates */ - Coords RX = rot_apply(R, X); - coords_get(RX, &x, &y, &z); - z = z+l/2; - /* rotate speed */ - X = coords_set(vx,vy,vz); - RX = rot_apply(R, X); - coords_get(RX, &vx, &vy, &vz); - } - - for (i=0; i<7; GVars.N_reflection[i++] = 0); - - /* propagate to box input (with gravitation) in comp local coords */ - /* A = 0.5 n.g; B = n.v; C = n.(r-W); */ - /* 0=Z0 side: n=(0, 0, -l) ; W = (0, 0, 0) (at z=0, guide input)*/ - B = -l*vz; C = -l*z; - - ret = solve_2nd_order(&dt, NULL, GVars.A[0], B, C); - if (ret==0) ABSORB; - - if (dt>0.0) PROP_GRAV_DT(dt, GVars.gx, GVars.gy, GVars.gz); else if (angle) ABSORB; - GVars.N_reflection[6]++; - - this_width = w1; - this_height = h1; - - /* check if we are in the box input, else absorb */ - if (fabs(x) > this_width/2 || fabs(y) > this_height/2) - ABSORB; - else - { - double w_edge, w_adj; /* Channel displacement on X */ - double h_edge, h_adj; /* Channel displacement on Y */ - double w_chnum,h_chnum; /* channel indexes */ - - SCATTER; - - /* X: Shift origin to center of channel hit (absorb if hit dividing walls) */ - x += w1/2.0; - w_chnum = floor(x/(GVars.w1c+d)); /* 0= right side, nslit+1=left side */ - w_edge = w_chnum*(GVars.w1c+d); - if(x - w_edge > GVars.w1c) - { - x -= w1/2.0; /* Re-adjust origin */ - ABSORB; - } - w_adj = w_edge + (GVars.w1c)/2.0; - x -= w_adj; w_adj -= w1/2.0; - - /* Y: Shift origin to center of channel hit (absorb if hit dividing walls) */ - y += h1/2.0; - h_chnum = floor(y/(GVars.h1c+d)); /* 0= lower side, nslit+1=upper side */ - h_edge = h_chnum*(GVars.h1c+d); - if(y - h_edge > GVars.h1c) - { - y -= h1/2.0; /* Re-adjust origin */ - ABSORB; - } - h_adj = h_edge + (GVars.h1c)/2.0; - y -= h_adj; h_adj -= h1/2.0; - - /* neutron is now in the input window of the guide */ - /* do loops on reflections in the box */ - for(;;) - { - /* get intersections for all box sides */ - double q, nx,ny,nz; - double this_length; - int side=0; - - bounces++; - /* now look for intersection with guide sides and exit */ - side = Gravity_guide_Trace(&dt, &GVars, x, y, z, - vx, vy, vz, w_chnum, nslit, h_chnum, nhslit, - &nx, &ny, &nz, _particle); - /* only positive dt are valid */ - /* exit reflection loops if no intersection (neutron is after box) */ - if (side == 0 || dt <= 0) - { - if (GVars.warnings < 100) - fprintf(stderr,"%s: warning: neutron has entered guide, but can not exit !\n", GVars.compcurname); - GVars.warnings++; - x += w_adj; y += h_adj; ABSORB; } /* should never occur */ - - /* propagate to dt */ - PROP_GRAV_DT(dt, GVars.gx, GVars.gy, GVars.gz); - - /* do reflection on speed for l/r/u/d sides */ - if (side == 5) /* neutron reaches end of guide: end loop and exit comp */ - { GVars.N_reflection[side]++; x += w_adj; y += h_adj; SCATTER; x -= w_adj; y -= h_adj; break; } - /* else reflection on a guide wall */ - if(GVars.M[side] == 0 || Qc == 0 || R0 == 0) /* walls are absorbing */ - { x += w_adj; y += h_adj; ABSORB; } - /* handle chamfers */ - this_width = w1+(w2-w1)*z/l; - this_height= h1+(h2-h1)*z/l; - this_length= fmod(z, l/nelements); - /* absorb on input/output of element parts */ - if (GVars.chamfer_z && (this_lengthl/nelements-GVars.chamfer_z)) - { x += w_adj; y += h_adj; ABSORB; } - /* absorb on l/r/t/b sides */ - if (GVars.chamfer_lr && (side==1 || side==2) && (fabs(y+h_adj)>this_height/2-GVars.chamfer_lr)) - { x += w_adj; y += h_adj; ABSORB; } - if (GVars.chamfer_tb && (side==3 || side==4) && (fabs(x+w_adj)>this_width/2- GVars.chamfer_tb)) - { x += w_adj; y += h_adj; ABSORB; } - /* change/mirror velocity: h_f = v - n.2*n.v/|n|^2 */ - GVars.N_reflection[side]++; /* GVars.norm_n2 > 0 was checked at INIT */ - /* compute n.v using current values */ - B = scalar_prod(vx,vy,vz,nx,ny,nz); - dt = 2*B/GVars.norm_n2[side]; /* 2*n.v/|n|^2 */ - vx -= nx*dt; - vy -= ny*dt; - vz -= nz*dt; - - /* compute q and modify neutron weight */ - /* scattering q=|n_i-n_f| = V2Q*|vf - v| = V2Q*2*n.v/|n| */ - q = 2*V2Q*fabs(B)/GVars.norm_n[side]; - - if (table_present==1) - TableReflecFunc(q, &pTable, &B); - else { - double par[] = {R0, Qc, GVars.Alpha[side], GVars.M[side], W}; - StdReflecFunc(q, par, &B); - } - if (B <= 0) { x += w_adj; y += h_adj; ABSORB; } - else p *= B; - x += w_adj; y += h_adj; SCATTER; x -= w_adj; y -= h_adj; - GVars.N_reflection[0]++; - /* go to the next reflection */ - if (bounces > 1000) ABSORB; - } /* end for */ - x += w_adj; y += h_adj; /* Re-adjust origin after SCATTER */ - } - - if (GVars.fc_freq != 0 || GVars.fc_phase != 0) { /* rotate back neutron w/r to guide element */ - /* approximation of rotating straight Fermi Chopper */ - Coords X = coords_set(x,y,z-l/2); /* current coordinates of neutron in centered static frame */ - Rotation R; - rot_set_rotation(R, 0, angle, 0); /* will rotate back neutron: positive side */ - /* apply rotation to centered coordinates */ - Coords RX = rot_apply(R, X); - coords_get(RX, &x, &y, &z); - z = z+l/2; - /* rotate speed */ - X = coords_set(vx,vy,vz); - RX = rot_apply(R, X); - coords_get(RX, &vx, &vy, &vz); - } - - } /* if l */ - -#ifndef NOABSORB_INF_NAN - /* Check for nan or inf particle parms */ - if(isnan(p) || isinf(p)) ABSORB; - if(isnan(t) || isinf(t)) ABSORB; - if(isnan(vx) || isinf(vx)) ABSORB; - if(isnan(vy) || isinf(vy)) ABSORB; - if(isnan(vz) || isinf(vz)) ABSORB; - if(isnan(x) || isinf(x)) ABSORB; - if(isnan(y) || isinf(y)) ABSORB; - if(isnan(z) || isinf(z)) ABSORB; -#else - if(isnan(p) || isinf(p)) printf("NAN or INF found in p, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(t) || isinf(t)) printf("NAN or INF found in t, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vx) || isinf(vx)) printf("NAN or INF found in vx, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vy) || isinf(vy)) printf("NAN or INF found in vy, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vz) || isinf(vz)) printf("NAN or INF found in vz, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(x) || isinf(x)) printf("NAN or INF found in x, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(y) || isinf(y)) printf("NAN or INF found in y, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(z) || isinf(z)) printf("NAN or INF found in z, %s (particle %lld)\n",_comp->_name,_particle->_uid); -#endif -#undef w1 -#undef h1 -#undef w2 -#undef h2 -#undef l -#undef R0 -#undef Qc -#undef alpha -#undef m -#undef W -#undef nslit -#undef d -#undef mleft -#undef mright -#undef mtop -#undef mbottom -#undef nhslit -#undef G -#undef aleft -#undef aright -#undef atop -#undef abottom -#undef wavy -#undef wavy_z -#undef wavy_tb -#undef wavy_lr -#undef chamfers -#undef chamfers_z -#undef chamfers_lr -#undef chamfers_tb -#undef nelements -#undef nu -#undef phase -#undef reflect -#undef GVars -#undef pTable -#undef table_present - return(_comp); -} /* class_Guide_gravity_trace */ - -#pragma acc routine -_class_Slit *class_Slit_trace(_class_Slit *_comp, _class_particle *_particle) { -ABSORBED=SCATTERED=RESTORE=0; -#define xmin (_comp->_parameters.xmin) -#define xmax (_comp->_parameters.xmax) -#define ymin (_comp->_parameters.ymin) -#define ymax (_comp->_parameters.ymax) -#define radius (_comp->_parameters.radius) -#define xwidth (_comp->_parameters.xwidth) -#define yheight (_comp->_parameters.yheight) -#define isradial (_comp->_parameters.isradial) -SIG_MESSAGE("[_Slit_trace] component NULL=Slit() [/home/g/.cache/mccode-mcstas/main/optics/Slit.comp:98]"); - - PROP_Z0; - if (!isradial ? (x < xmin || x > xmax || y < ymin || y > ymax) : (x * x + y * y > radius * radius)) - ABSORB; - else - SCATTER; - -#ifndef NOABSORB_INF_NAN - /* Check for nan or inf particle parms */ - if(isnan(p) || isinf(p)) ABSORB; - if(isnan(t) || isinf(t)) ABSORB; - if(isnan(vx) || isinf(vx)) ABSORB; - if(isnan(vy) || isinf(vy)) ABSORB; - if(isnan(vz) || isinf(vz)) ABSORB; - if(isnan(x) || isinf(x)) ABSORB; - if(isnan(y) || isinf(y)) ABSORB; - if(isnan(z) || isinf(z)) ABSORB; -#else - if(isnan(p) || isinf(p)) printf("NAN or INF found in p, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(t) || isinf(t)) printf("NAN or INF found in t, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vx) || isinf(vx)) printf("NAN or INF found in vx, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vy) || isinf(vy)) printf("NAN or INF found in vy, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vz) || isinf(vz)) printf("NAN or INF found in vz, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(x) || isinf(x)) printf("NAN or INF found in x, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(y) || isinf(y)) printf("NAN or INF found in y, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(z) || isinf(z)) printf("NAN or INF found in z, %s (particle %lld)\n",_comp->_name,_particle->_uid); -#endif -#undef xmin -#undef xmax -#undef ymin -#undef ymax -#undef radius -#undef xwidth -#undef yheight -#undef isradial - return(_comp); -} /* class_Slit_trace */ - -#pragma acc routine -_class_Monochromator_curved *class_Monochromator_curved_trace(_class_Monochromator_curved *_comp, _class_particle *_particle) { -ABSORBED=SCATTERED=RESTORE=0; -#define reflect (_comp->_parameters.reflect) -#define transmit (_comp->_parameters.transmit) -#define zwidth (_comp->_parameters.zwidth) -#define yheight (_comp->_parameters.yheight) -#define gap (_comp->_parameters.gap) -#define NH (_comp->_parameters.NH) -#define NV (_comp->_parameters.NV) -#define mosaich (_comp->_parameters.mosaich) -#define mosaicv (_comp->_parameters.mosaicv) -#define r0 (_comp->_parameters.r0) -#define t0 (_comp->_parameters.t0) -#define Q (_comp->_parameters.Q) -#define RV (_comp->_parameters.RV) -#define RH (_comp->_parameters.RH) -#define DM (_comp->_parameters.DM) -#define mosaic (_comp->_parameters.mosaic) -#define width (_comp->_parameters.width) -#define height (_comp->_parameters.height) -#define verbose (_comp->_parameters.verbose) -#define order (_comp->_parameters.order) -#define mos_rms_y (_comp->_parameters.mos_rms_y) -#define mos_rms_z (_comp->_parameters.mos_rms_z) -#define mos_rms_max (_comp->_parameters.mos_rms_max) -#define mono_Q (_comp->_parameters.mono_Q) -#define SlabWidth (_comp->_parameters.SlabWidth) -#define SlabHeight (_comp->_parameters.SlabHeight) -#define rTable (_comp->_parameters.rTable) -#define tTable (_comp->_parameters.tTable) -#define rTableFlag (_comp->_parameters.rTableFlag) -#define tTableFlag (_comp->_parameters.tTableFlag) -#define tiltH (_comp->_parameters.tiltH) -#define tiltV (_comp->_parameters.tiltV) -SIG_MESSAGE("[_Monochromator_curved_trace] component NULL=Monochromator_curved() [/home/g/.cache/mccode-mcstas/main/optics/Monochromator_curved.comp:214]"); - - double dt; - double Gauss_X[] = {-0.987992518020485, -0.937273392400706, -0.848206583410427, - -0.724417731360170, -0.570972172608539, -0.394151347077563, - -0.201194093997435, 0, 0.201194093997435, - 0.394151347077563, 0.570972172608539, 0.724417731360170, - 0.848206583410427, 0.937273392400706, 0.987992518020485}; - double Gauss_W[] = {0.030753241996117, 0.070366047488108, 0.107159220467172, - 0.139570677926154, 0.166269205816994, 0.186161000115562, - 0.198431485327111, 0.202578241925561, 0.198431485327111, - 0.186161000115562, 0.166269205816994, 0.139570677926154, - 0.107159220467172, 0.070366047488108, 0.030753241996117}; - - if(vx != 0.0 && (dt = -x/vx) >= 0.0 && r0) - { /* Moving towards crystal? */ - double zmin,zmax, ymin,ymax; - double yy,zz; - - zmax = ((NH*(SlabWidth+gap))-gap)/2; - zmin = -zmax; - ymax = ((NV*(SlabHeight+gap))-gap)/2; - ymin = -ymax; - - /* Test-propagate to crystal plane */ - zz=z+vz*dt; - yy=y+vy*dt; - if (zz>zmin && zzymin && yy 2*k/mono_Q) Q_order--; - if((!order && Q_order > 0) || (Q_order == fabs(order) && order)) { /* Bragg scattering possible? */ - double q0, q0x, theta, delta, p_reflect, my_r0; - - q0 = Q_order*mono_Q; - q0x = ratio < 0 ? -q0 : q0; - theta = asin(q0/(2*k)); /* Actual bragg angle */ - /* Make MC choice: reflect or transmit? */ - delta = asin(fabs(kux)) - theta; - - if (rTableFlag) - { - my_r0 = r0*Table_Value(rTable, k, 1); /* 2nd column */ - } - else my_r0 = r0; - if (my_r0 > 1) - { - if (my_r0 > 1.01 && verbose) - fprintf(stdout, "Warning: Monochromator_curved : lowered reflectivity from %f to 1 (k=%f)\n", my_r0, k); - my_r0=0.999; - } - if (my_r0 < 0) - { - if (verbose) - fprintf(stdout, "Warning: Monochromator_curved : raised reflectivity from %f to 0 (k=%f)\n", my_r0, k); - my_r0=0; - } - - p_reflect = fabs(my_r0)*exp(-kiz*kiz/(kiy*kiy + kiz*kiz)*(delta*delta)/ - (2*mos_rms_y*mos_rms_y))* - exp(-kiy*kiy/(kiy*kiy + kiz*kiz)*(delta*delta)/ - (2*mos_rms_z*mos_rms_z)); - - double rr=rand01(); - if(rr <= p_reflect) { /* Reflect */ - double bx,by,bz,ax,ay,az,phi; - double cos_2theta,k_sin_2theta,cos_phi,sin_phi,q_x,q_y,q_z; - double total,c1x,c1y,c1z,w,mos_sample; - int i=0; - - cos_2theta = cos(2*theta); - k_sin_2theta = k*sin(2*theta); - /* Get unit normal to plane containing ki and most probable kf */ - vec_prod(bx, by, bz, kix, kiy, kiz, q0x, 0, 0); - NORM(bx,by,bz); - bx = bx * k_sin_2theta; - by = by * k_sin_2theta; - bz = bz * k_sin_2theta; - /* Get unit vector normal to ki and b */ - vec_prod(ax, ay, az, bx, by, bz, kux, kuy, kuz); - /* Compute the total scattering probability at this ki */ - total = 0; - /* Choose width of Gaussian distribution to sample the angle - * phi on the Debye-Scherrer cone for the scattered neutron. - * The radius of the Debye-Scherrer cone is smaller by a - * factor 1/cos(theta) than the radius of the (partial) sphere - * describing the possible orientations of Q due to mosaicity, so we - * start with a width 1/cos(theta) greater than the largest of - * the two mosaics. */ - mos_sample = mos_rms_max/cos(theta); - c1x = kix*(cos_2theta-1); - c1y = kiy*(cos_2theta-1); - c1z = kiz*(cos_2theta-1); - /* Loop, repeatedly reducing the sample width until it is small - * enough to avoid sampling scattering directions with - * ridiculously low scattering probability. - * Use a cut-off at 5 times the gauss width for considering - * scattering probability as well as for integration limits - * when integrating the sampled distribution below. */ - for(i=0; i<100; i++) { - w = 5*mos_sample; - cos_phi = cos(w); - sin_phi = sin(w); - q_x = c1x + cos_phi*ax + sin_phi*bx; - q_y = (c1y + cos_phi*ay + sin_phi*by)/mos_rms_z; - q_z = (c1z + cos_phi*az + sin_phi*bz)/mos_rms_y; - /* Stop when we get near a factor of 25=5^2. */ - if(q_z*q_z + q_y*q_y < (25/(2.0/3.0))*(q_x*q_x)) - break; - mos_sample *= (2.0/3.0); - } - /* Now integrate the chosen sampling distribution, using a - * cut-off at five times sigma. */ - for(i = 0; i < (sizeof(Gauss_X)/sizeof(double)); i++) - { - phi = w*Gauss_X[i]; - cos_phi = cos(phi); - sin_phi = sin(phi); - q_x = c1x + cos_phi*ax + sin_phi*bx; - q_y = c1y + cos_phi*ay + sin_phi*by; - q_z = c1z + cos_phi*az + sin_phi*bz; - p_reflect = GAUSS_monocurved((q_z/q_x),0,mos_rms_y)* - GAUSS_monocurved((q_y/q_x),0,mos_rms_z); - total += Gauss_W[i]*p_reflect; - } - total *= w; - /* Choose point on Debye-Scherrer cone. Sample from a Gaussian of - * width 1/cos(theta) greater than the mosaic and correct for any - * error by adjusting the neutron weight later. */ - phi = mos_sample*randnorm(); - /* Compute final wave vector kf and scattering vector q = ki - kf */ - cos_phi = cos(phi); - sin_phi = sin(phi); - q_x = c1x + cos_phi*ax + sin_phi*bx; - q_y = c1y + cos_phi*ay + sin_phi*by; - q_z = c1z + cos_phi*az + sin_phi*bz; - p_reflect = GAUSS_monocurved((q_z/q_x),0,mos_rms_y)* - GAUSS_monocurved((q_y/q_x),0,mos_rms_z); - - vx = K2V*(kix+q_x); - vy = K2V*(kiy+q_y); - vz = K2V*(kiz+q_z); - p_reflect /= total*GAUSS_monocurved(phi,0,mos_sample); - if (p_reflect <= 0) ABSORB; - if (p_reflect > 1) p_reflect = 1; - p = p * p_reflect; - - } /* End MC choice to reflect or transmit neutron (if tmp 1) - { - if (my_t0 > 1.01 && verbose) - fprintf(stdout, "Warning: Monochromator_curved : lowered transmission from %f to 1 (k=%f)\n", my_t0, k); - my_t0=0.999; - } - if (my_t0 > 0) p = p * my_t0; - else ABSORB; - } - } /* end if not in gap */ - /* rotate back in component frame */ - Rotation TT; - rot_transpose(T, TT); - /* now make the coordinate system change */ - mccoordschange_polarisation(TT, &vx, &vy, &vz); - coords_get(rot_apply(TT,coords_set(x,y,z)),&x,&y,&z); - y= y+center_y; - z= z+center_z; - /* Visualise scattering point in proper, component frame - - but only if the neutron is reflected, that is none of: - * transmitted - * falling outside the slab material */ - if(!do_transmit) SCATTER; - - /* mccoordschange_polarisation(tt, &sx, &sy, &sz); */ - } /* End intersect the crystal (if z) */ - else { - /* restore neutron state when no interaction */ - RESTORE_NEUTRON(INDEX_CURRENT_COMP, x, y, z, vx, vy, vz, t, sx, sy, sz, p); - } - } /* End neutron moving towards crystal (if vx)*/ - -#ifndef NOABSORB_INF_NAN - /* Check for nan or inf particle parms */ - if(isnan(p) || isinf(p)) ABSORB; - if(isnan(t) || isinf(t)) ABSORB; - if(isnan(vx) || isinf(vx)) ABSORB; - if(isnan(vy) || isinf(vy)) ABSORB; - if(isnan(vz) || isinf(vz)) ABSORB; - if(isnan(x) || isinf(x)) ABSORB; - if(isnan(y) || isinf(y)) ABSORB; - if(isnan(z) || isinf(z)) ABSORB; -#else - if(isnan(p) || isinf(p)) printf("NAN or INF found in p, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(t) || isinf(t)) printf("NAN or INF found in t, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vx) || isinf(vx)) printf("NAN or INF found in vx, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vy) || isinf(vy)) printf("NAN or INF found in vy, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vz) || isinf(vz)) printf("NAN or INF found in vz, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(x) || isinf(x)) printf("NAN or INF found in x, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(y) || isinf(y)) printf("NAN or INF found in y, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(z) || isinf(z)) printf("NAN or INF found in z, %s (particle %lld)\n",_comp->_name,_particle->_uid); -#endif -#undef reflect -#undef transmit -#undef zwidth -#undef yheight -#undef gap -#undef NH -#undef NV -#undef mosaich -#undef mosaicv -#undef r0 -#undef t0 -#undef Q -#undef RV -#undef RH -#undef DM -#undef mosaic -#undef width -#undef height -#undef verbose -#undef order -#undef mos_rms_y -#undef mos_rms_z -#undef mos_rms_max -#undef mono_Q -#undef SlabWidth -#undef SlabHeight -#undef rTable -#undef tTable -#undef rTableFlag -#undef tTableFlag -#undef tiltH -#undef tiltV - return(_comp); -} /* class_Monochromator_curved_trace */ - -#pragma acc routine -_class_Monitor *class_Monitor_trace(_class_Monitor *_comp, _class_particle *_particle) { -ABSORBED=SCATTERED=RESTORE=0; -#define xmin (_comp->_parameters.xmin) -#define xmax (_comp->_parameters.xmax) -#define ymin (_comp->_parameters.ymin) -#define ymax (_comp->_parameters.ymax) -#define xwidth (_comp->_parameters.xwidth) -#define yheight (_comp->_parameters.yheight) -#define restore_neutron (_comp->_parameters.restore_neutron) -#define Nsum (_comp->_parameters.Nsum) -#define psum (_comp->_parameters.psum) -#define p2sum (_comp->_parameters.p2sum) -SIG_MESSAGE("[_Monitor_trace] component NULL=Monitor() [/home/g/.cache/mccode-mcstas/main/monitors/Monitor.comp:75]"); - - PROP_Z0; - if (x>xmin && xymin && y_name,_particle->_uid); - if(isnan(t) || isinf(t)) printf("NAN or INF found in t, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vx) || isinf(vx)) printf("NAN or INF found in vx, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vy) || isinf(vy)) printf("NAN or INF found in vy, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(vz) || isinf(vz)) printf("NAN or INF found in vz, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(x) || isinf(x)) printf("NAN or INF found in x, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(y) || isinf(y)) printf("NAN or INF found in y, %s (particle %lld)\n",_comp->_name,_particle->_uid); - if(isnan(z) || isinf(z)) printf("NAN or INF found in z, %s (particle %lld)\n",_comp->_name,_particle->_uid); -#endif -#undef xmin -#undef xmax -#undef ymin -#undef ymax -#undef xwidth -#undef yheight -#undef restore_neutron -#undef Nsum -#undef psum -#undef p2sum - return(_comp); -} /* class_Monitor_trace */ - - -/* ***************************************************************************** -* instrument 'splitRunTest' TRACE -***************************************************************************** */ - -#ifndef FUNNEL -#pragma acc routine -int raytrace(_class_particle* _particle) { /* single event propagation, called by mccode_main for splitRunTest:TRACE */ - - /* init variables and counters for TRACE */ - #undef ABSORB0 - #undef ABSORB - #define ABSORB0 do { DEBUG_ABSORB(); MAGNET_OFF; ABSORBED++; return(ABSORBED);} while(0) - #define ABSORB ABSORB0 - DEBUG_ENTER(); - DEBUG_STATE(); - _particle->flag_nocoordschange=0; /* Init */ - _class_particle _particle_save; - /* the main iteration loop for one incoming event */ - while (!ABSORBED) { /* iterate event until absorbed */ - /* send particle event to component instance, one after the other */ - /* begin component origin=Arm() [1] */ - if (!ABSORBED && _particle->_index == 1) { - _particle->flag_nocoordschange=0; /* Reset if we came here from a JUMP */ - _particle->_index++; - } /* end of component origin [1] */ - /* begin component source=Source_simple() [2] */ - if (!_particle->flag_nocoordschange) { // flag activated by JUMP to pass coords change - if (_source_var._rotation_is_identity) { - if(!_source_var._position_relative_is_zero) { - coords_get(coords_add(coords_set(x,y,z), _source_var._position_relative),&x, &y, &z); - } - } else { - mccoordschange(_source_var._position_relative, _source_var._rotation_relative, _particle); - } - } - if (!ABSORBED && _particle->_index == 2) { - _particle->flag_nocoordschange=0; /* Reset if we came here from a JUMP */ - _particle_save = *_particle; - DEBUG_COMP(_source_var._name); - DEBUG_STATE(); - class_Source_simple_trace(&_source_var, _particle); - if (_particle->_restore) - particle_restore(_particle, &_particle_save); - _particle->_index++; - if (!ABSORBED) { DEBUG_STATE(); } - } /* end of component source [2] */ - /* begin component guide=Guide_gravity() [3] */ - if (!_particle->flag_nocoordschange) { // flag activated by JUMP to pass coords change - if (_guide_var._rotation_is_identity) { - if(!_guide_var._position_relative_is_zero) { - coords_get(coords_add(coords_set(x,y,z), _guide_var._position_relative),&x, &y, &z); - } - } else { - mccoordschange(_guide_var._position_relative, _guide_var._rotation_relative, _particle); - } - } - if (!ABSORBED && _particle->_index == 3) { - _particle->flag_nocoordschange=0; /* Reset if we came here from a JUMP */ - _particle_save = *_particle; - DEBUG_COMP(_guide_var._name); - DEBUG_STATE(); - class_Guide_gravity_trace(&_guide_var, _particle); - if (_particle->_restore) - particle_restore(_particle, &_particle_save); - _particle->_index++; - if (!ABSORBED) { DEBUG_STATE(); } - } /* end of component guide [3] */ - /* begin component guide_end=Arm() [4] */ - if (!ABSORBED && _particle->_index == 4) { - _particle->flag_nocoordschange=0; /* Reset if we came here from a JUMP */ - _particle->_index++; - } /* end of component guide_end [4] */ - /* begin component aperture=Slit() [5] */ - if (!_particle->flag_nocoordschange) { // flag activated by JUMP to pass coords change - if (_aperture_var._rotation_is_identity) { - if(!_aperture_var._position_relative_is_zero) { - coords_get(coords_add(coords_set(x,y,z), _aperture_var._position_relative),&x, &y, &z); - } - } else { - mccoordschange(_aperture_var._position_relative, _aperture_var._rotation_relative, _particle); - } - } - if (!ABSORBED && _particle->_index == 5) { - _particle->flag_nocoordschange=0; /* Reset if we came here from a JUMP */ - _particle_save = *_particle; - DEBUG_COMP(_aperture_var._name); - DEBUG_STATE(); - class_Slit_trace(&_aperture_var, _particle); - if (_particle->_restore) - particle_restore(_particle, &_particle_save); - _particle->_index++; - if (!ABSORBED) { DEBUG_STATE(); } - } /* end of component aperture [5] */ - /* begin component split_at=Arm() [6] */ - if (!ABSORBED && _particle->_index == 6) { - _particle->flag_nocoordschange=0; /* Reset if we came here from a JUMP */ - _particle->_index++; - } /* end of component split_at [6] */ - /* begin component mono_point=Arm() [7] */ - if (!ABSORBED && _particle->_index == 7) { - _particle->flag_nocoordschange=0; /* Reset if we came here from a JUMP */ - _particle->_index++; - } /* end of component mono_point [7] */ - /* begin component mono=Monochromator_curved() [8] */ - if (!_particle->flag_nocoordschange) { // flag activated by JUMP to pass coords change - if (_mono_var._rotation_is_identity) { - if(!_mono_var._position_relative_is_zero) { - coords_get(coords_add(coords_set(x,y,z), _mono_var._position_relative),&x, &y, &z); - } - } else { - mccoordschange(_mono_var._position_relative, _mono_var._rotation_relative, _particle); - } - } - if (!ABSORBED && _particle->_index == 8) { - _particle->flag_nocoordschange=0; /* Reset if we came here from a JUMP */ - _particle_save = *_particle; - DEBUG_COMP(_mono_var._name); - DEBUG_STATE(); - class_Monochromator_curved_trace(&_mono_var, _particle); - if (_particle->_restore) - particle_restore(_particle, &_particle_save); - _particle->_index++; - if (!ABSORBED) { DEBUG_STATE(); } - } /* end of component mono [8] */ - /* begin component sample_arm=Arm() [9] */ - if (!ABSORBED && _particle->_index == 9) { - _particle->flag_nocoordschange=0; /* Reset if we came here from a JUMP */ - _particle->_index++; - } /* end of component sample_arm [9] */ - /* begin component detector=Monitor() [10] */ - if (!_particle->flag_nocoordschange) { // flag activated by JUMP to pass coords change - if (_detector_var._rotation_is_identity) { - if(!_detector_var._position_relative_is_zero) { - coords_get(coords_add(coords_set(x,y,z), _detector_var._position_relative),&x, &y, &z); - } - } else { - mccoordschange(_detector_var._position_relative, _detector_var._rotation_relative, _particle); - } - } - if (!ABSORBED && _particle->_index == 10) { - _particle->flag_nocoordschange=0; /* Reset if we came here from a JUMP */ - _particle_save = *_particle; - DEBUG_COMP(_detector_var._name); - DEBUG_STATE(); - class_Monitor_trace(&_detector_var, _particle); - if (_particle->_restore) - particle_restore(_particle, &_particle_save); - _particle->_index++; - if (!ABSORBED) { DEBUG_STATE(); } - } /* end of component detector [10] */ - if (_particle->_index > 10) - ABSORBED++; /* absorbed when passed all components */ - } /* while !ABSORBED */ - - DEBUG_LEAVE() - particle_restore(_particle, &_particle_save); - DEBUG_STATE() - - return(_particle->_index); -} /* raytrace */ - -/* loop to generate events and call raytrace() propagate them */ -void raytrace_all(unsigned long long ncount, unsigned long seed) { - - /* CPU-loop */ - unsigned long long loops; - loops = ceil((double)ncount/gpu_innerloop); - /* if on GPU, printf has been globally nullified, re-enable here */ - #ifdef OPENACC - #ifndef MULTICORE - #undef printf - #endif - #endif - - #ifdef OPENACC - if (ncount>gpu_innerloop) { - printf("Defining %llu CPU loops around GPU kernel and adjusting ncount\n",loops); - mcset_ncount(loops*gpu_innerloop); - } else { - #endif - loops=1; - gpu_innerloop = ncount; - #ifdef OPENACC - } - #endif - - for (unsigned long long cloop=0; cloop1) fprintf(stdout, "%d..", (int)cloop); fflush(stdout); - #endif - - /* if on GPU, re-nullify printf */ - #ifdef OPENACC - #ifndef MULTICORE - #define printf(...) noprintf() - #endif - #endif - - #pragma acc parallel loop num_gangs(numgangs) vector_length(vecsize) - for (unsigned long pidx=0 ; pidx < gpu_innerloop ; pidx++) { - _class_particle particleN = mcgenstate(); // initial particle - _class_particle* _particle = &particleN; - particleN._uid = pidx; - - srandom(_hash((pidx+1)*(seed+1))); - particle_uservar_init(_particle); - - raytrace(_particle); - } /* inner for */ - seed = seed+gpu_innerloop; - } /* CPU for */ - /* if on GPU, printf has been globally nullified, re-enable here */ - #ifdef OPENACC - #ifndef MULTICORE - #undef printf - #endif - #endif - MPI_MASTER( - printf("*** TRACE end *** \n"); - ); -} /* raytrace_all */ - -#endif //no-FUNNEL - -#ifdef FUNNEL -// Alternative raytrace algorithm which iterates all particles through -// one component at the time, can remove absorbs from the next loop and -// switch between cpu/gpu. -void raytrace_all_funnel(unsigned long long ncount, unsigned long seed) { - - // set up outer (CPU) loop / particle batches - unsigned long long loops; - - /* if on GPU, printf has been globally nullified, re-enable here */ - #ifdef OPENACC - #ifndef MULTICORE - #undef printf - #endif - #endif - - #ifdef OPENACC - loops = ceil((double)ncount/gpu_innerloop); - if (ncount>gpu_innerloop) { - printf("Defining %llu CPU loops around kernel and adjusting ncount\n",loops); - mcset_ncount(loops*gpu_innerloop); - } else { - #endif - loops=1; - gpu_innerloop = ncount; - #ifdef OPENACC - } - #endif - - // create particles struct and pointer arrays (same memory used by all batches) - _class_particle* particles = malloc(gpu_innerloop*sizeof(_class_particle)); - _class_particle* pbuffer = malloc(gpu_innerloop*sizeof(_class_particle)); - long livebatchsize = gpu_innerloop; - - #undef ABSORB0 - #undef ABSORB - #define ABSORB0 do { DEBUG_ABSORB(); MAGNET_OFF; ABSORBED++; } while(0) - #define ABSORB ABSORB0 - // outer loop / particle batches - for (unsigned long long cloop=0; cloop1) fprintf(stdout, "%d..", (int)cloop); fflush(stdout); - - // init particles - #pragma acc parallel loop present(particles) - for (unsigned long pidx=0 ; pidx < livebatchsize ; pidx++) { - // generate particle state, set loop index and seed - particles[pidx] = mcgenstate(); - _class_particle* _particle = particles + pidx; - _particle->_uid = pidx; - srandom(_hash((pidx+1)*(seed+1))); // _particle->state usage built into srandom macro - particle_uservar_init(_particle); - } - - // iterate components - - #pragma acc parallel loop present(particles) - for (unsigned long pidx=0 ; pidx < livebatchsize ; pidx++) { - _class_particle* _particle = &particles[pidx]; - _class_particle _particle_save; - // origin - if (!ABSORBED && _particle->index == 0) { - _particle->_index++; - } - // source - if (!ABSORBED && _particle->index == 1) { - if (_source_var._rotation_is_identity) { - coords_get(coords_add(coords_set(x,y,z), _source_var._position_relative),&x, &y, &z); - } else { - mccoordschange(_source_var._position_relative, _source_var._rotation_relative, _particle); - } - _particle_save = *_particle; - class_Source_simple_trace(&_source_var, _particle); - if (_particle->_restore) particle_restore(_particle, &_particle_save); - _particle->_index++; - } - // guide - if (!ABSORBED && _particle->index == 2) { - if (_guide_var._rotation_is_identity) { - coords_get(coords_add(coords_set(x,y,z), _guide_var._position_relative),&x, &y, &z); - } else { - mccoordschange(_guide_var._position_relative, _guide_var._rotation_relative, _particle); - } - _particle_save = *_particle; - class_Guide_gravity_trace(&_guide_var, _particle); - if (_particle->_restore) particle_restore(_particle, &_particle_save); - _particle->_index++; - } - // guide_end - if (!ABSORBED && _particle->index == 3) { - _particle->_index++; - } - // aperture - if (!ABSORBED && _particle->index == 4) { - if (_aperture_var._rotation_is_identity) { - coords_get(coords_add(coords_set(x,y,z), _aperture_var._position_relative),&x, &y, &z); - } else { - mccoordschange(_aperture_var._position_relative, _aperture_var._rotation_relative, _particle); - } - _particle_save = *_particle; - class_Slit_trace(&_aperture_var, _particle); - if (_particle->_restore) particle_restore(_particle, &_particle_save); - _particle->_index++; - } - // split_at - if (!ABSORBED && _particle->index == 5) { - _particle->_index++; - } - // mono_point - if (!ABSORBED && _particle->index == 6) { - _particle->_index++; - } - // mono - if (!ABSORBED && _particle->index == 7) { - if (_mono_var._rotation_is_identity) { - coords_get(coords_add(coords_set(x,y,z), _mono_var._position_relative),&x, &y, &z); - } else { - mccoordschange(_mono_var._position_relative, _mono_var._rotation_relative, _particle); - } - _particle_save = *_particle; - class_Monochromator_curved_trace(&_mono_var, _particle); - if (_particle->_restore) particle_restore(_particle, &_particle_save); - _particle->_index++; - } - // sample_arm - if (!ABSORBED && _particle->index == 8) { - _particle->_index++; - } - // detector - if (!ABSORBED && _particle->index == 9) { - if (_detector_var._rotation_is_identity) { - coords_get(coords_add(coords_set(x,y,z), _detector_var._position_relative),&x, &y, &z); - } else { - mccoordschange(_detector_var._position_relative, _detector_var._rotation_relative, _particle); - } - _particle_save = *_particle; - class_Monitor_trace(&_detector_var, _particle); - if (_particle->_restore) particle_restore(_particle, &_particle_save); - _particle->_index++; - } - } - // jump to next viable seed - seed = seed + gpu_innerloop; - } // outer loop / particle batches - - free(particles); - free(pbuffer); - - printf("\n"); -} /* raytrace_all_funnel */ -#endif // FUNNEL - - -#undef x -#undef y -#undef z -#undef vx -#undef vy -#undef vz -#undef t -#undef sx -#undef sy -#undef sz -#undef p -#undef mcgravitation -#undef mcMagnet -#undef allow_backprop -#ifdef OPENACC -#ifndef MULTICORE -#undef strlen -#undef strcmp -#undef exit -#undef printf -#undef sprintf -#undef fprintf -#endif -#endif -#undef SCATTERED -#undef RESTORE -#undef RESTORE_NEUTRON -#undef STORE_NEUTRON -#undef ABSORBED -#undef ABSORB -#undef ABSORB0 -/* ***************************************************************************** -* instrument 'splitRunTest' and components SAVE -***************************************************************************** */ -_class_Monitor *class_Monitor_save(_class_Monitor *_comp) { -#define xmin (_comp->_parameters.xmin) -#define xmax (_comp->_parameters.xmax) -#define ymin (_comp->_parameters.ymin) -#define ymax (_comp->_parameters.ymax) -#define xwidth (_comp->_parameters.xwidth) -#define yheight (_comp->_parameters.yheight) -#define restore_neutron (_comp->_parameters.restore_neutron) -#define Nsum (_comp->_parameters.Nsum) -#define psum (_comp->_parameters.psum) -#define p2sum (_comp->_parameters.p2sum) - SIG_MESSAGE("[_Monitor_save] component NULL=Monitor() [/home/g/.cache/mccode-mcstas/main/monitors/Monitor.comp:58]"); - - char title[1024]; - sprintf(title, "Single monitor %s", NAME_CURRENT_COMP); - DETECTOR_OUT_0D(title, Nsum, psum, p2sum); - -#undef xmin -#undef xmax -#undef ymin -#undef ymax -#undef xwidth -#undef yheight -#undef restore_neutron -#undef Nsum -#undef psum -#undef p2sum - return(_comp); -} /* class_Monitor_save */ - -int save(FILE *handle) { /* called by mccode_main for splitRunTest:SAVE */ - if (!handle) siminfo_init(NULL); - /* call iteratively all components SAVE */ - class_Monitor_save(&_detector_var); - if (!handle) siminfo_close(); - return(0); -} /* save */ - -/* ***************************************************************************** -* instrument splitRunTest and components FINALLY -***************************************************************************** */ -_class_Guide_gravity *class_Guide_gravity_finally(_class_Guide_gravity *_comp) { -#define w1 (_comp->_parameters.w1) -#define h1 (_comp->_parameters.h1) -#define w2 (_comp->_parameters.w2) -#define h2 (_comp->_parameters.h2) -#define l (_comp->_parameters.l) -#define R0 (_comp->_parameters.R0) -#define Qc (_comp->_parameters.Qc) -#define alpha (_comp->_parameters.alpha) -#define m (_comp->_parameters.m) -#define W (_comp->_parameters.W) -#define nslit (_comp->_parameters.nslit) -#define d (_comp->_parameters.d) -#define mleft (_comp->_parameters.mleft) -#define mright (_comp->_parameters.mright) -#define mtop (_comp->_parameters.mtop) -#define mbottom (_comp->_parameters.mbottom) -#define nhslit (_comp->_parameters.nhslit) -#define G (_comp->_parameters.G) -#define aleft (_comp->_parameters.aleft) -#define aright (_comp->_parameters.aright) -#define atop (_comp->_parameters.atop) -#define abottom (_comp->_parameters.abottom) -#define wavy (_comp->_parameters.wavy) -#define wavy_z (_comp->_parameters.wavy_z) -#define wavy_tb (_comp->_parameters.wavy_tb) -#define wavy_lr (_comp->_parameters.wavy_lr) -#define chamfers (_comp->_parameters.chamfers) -#define chamfers_z (_comp->_parameters.chamfers_z) -#define chamfers_lr (_comp->_parameters.chamfers_lr) -#define chamfers_tb (_comp->_parameters.chamfers_tb) -#define nelements (_comp->_parameters.nelements) -#define nu (_comp->_parameters.nu) -#define phase (_comp->_parameters.phase) -#define reflect (_comp->_parameters.reflect) -#define GVars (_comp->_parameters.GVars) -#define pTable (_comp->_parameters.pTable) -#define table_present (_comp->_parameters.table_present) - SIG_MESSAGE("[_Guide_gravity_finally] component NULL=Guide_gravity() [/home/g/.cache/mccode-mcstas/main/optics/Guide_gravity.comp:565]"); - -if (GVars.warnings > 100) { - fprintf(stderr,"%s: warning: neutron has entered guide, but can not exit !\n", GVars.compcurname); - fprintf(stderr,"%s: warning: This message has been repeated %g times\n", GVars.compcurname, GVars.warnings); -} - -#undef w1 -#undef h1 -#undef w2 -#undef h2 -#undef l -#undef R0 -#undef Qc -#undef alpha -#undef m -#undef W -#undef nslit -#undef d -#undef mleft -#undef mright -#undef mtop -#undef mbottom -#undef nhslit -#undef G -#undef aleft -#undef aright -#undef atop -#undef abottom -#undef wavy -#undef wavy_z -#undef wavy_tb -#undef wavy_lr -#undef chamfers -#undef chamfers_z -#undef chamfers_lr -#undef chamfers_tb -#undef nelements -#undef nu -#undef phase -#undef reflect -#undef GVars -#undef pTable -#undef table_present - return(_comp); -} /* class_Guide_gravity_finally */ - -_class_Monochromator_curved *class_Monochromator_curved_finally(_class_Monochromator_curved *_comp) { -#define reflect (_comp->_parameters.reflect) -#define transmit (_comp->_parameters.transmit) -#define zwidth (_comp->_parameters.zwidth) -#define yheight (_comp->_parameters.yheight) -#define gap (_comp->_parameters.gap) -#define NH (_comp->_parameters.NH) -#define NV (_comp->_parameters.NV) -#define mosaich (_comp->_parameters.mosaich) -#define mosaicv (_comp->_parameters.mosaicv) -#define r0 (_comp->_parameters.r0) -#define t0 (_comp->_parameters.t0) -#define Q (_comp->_parameters.Q) -#define RV (_comp->_parameters.RV) -#define RH (_comp->_parameters.RH) -#define DM (_comp->_parameters.DM) -#define mosaic (_comp->_parameters.mosaic) -#define width (_comp->_parameters.width) -#define height (_comp->_parameters.height) -#define verbose (_comp->_parameters.verbose) -#define order (_comp->_parameters.order) -#define mos_rms_y (_comp->_parameters.mos_rms_y) -#define mos_rms_z (_comp->_parameters.mos_rms_z) -#define mos_rms_max (_comp->_parameters.mos_rms_max) -#define mono_Q (_comp->_parameters.mono_Q) -#define SlabWidth (_comp->_parameters.SlabWidth) -#define SlabHeight (_comp->_parameters.SlabHeight) -#define rTable (_comp->_parameters.rTable) -#define tTable (_comp->_parameters.tTable) -#define rTableFlag (_comp->_parameters.rTableFlag) -#define tTableFlag (_comp->_parameters.tTableFlag) -#define tiltH (_comp->_parameters.tiltH) -#define tiltV (_comp->_parameters.tiltV) - SIG_MESSAGE("[_Monochromator_curved_finally] component NULL=Monochromator_curved() [/home/g/.cache/mccode-mcstas/main/optics/Monochromator_curved.comp:469]"); - - if(rTableFlag){ - Table_Free(&rTable); - } - if(tTableFlag){ - Table_Free(&tTable); - } - if (tiltH) free(tiltH); - if (tiltV) free(tiltV); - -#undef reflect -#undef transmit -#undef zwidth -#undef yheight -#undef gap -#undef NH -#undef NV -#undef mosaich -#undef mosaicv -#undef r0 -#undef t0 -#undef Q -#undef RV -#undef RH -#undef DM -#undef mosaic -#undef width -#undef height -#undef verbose -#undef order -#undef mos_rms_y -#undef mos_rms_z -#undef mos_rms_max -#undef mono_Q -#undef SlabWidth -#undef SlabHeight -#undef rTable -#undef tTable -#undef rTableFlag -#undef tTableFlag -#undef tiltH -#undef tiltV - return(_comp); -} /* class_Monochromator_curved_finally */ - -int finally(void) { /* called by mccode_main for splitRunTest:FINALLY */ -#pragma acc update host(_origin_var) -#pragma acc update host(_source_var) -#pragma acc update host(_guide_var) -#pragma acc update host(_guide_end_var) -#pragma acc update host(_aperture_var) -#pragma acc update host(_split_at_var) -#pragma acc update host(_mono_point_var) -#pragma acc update host(_mono_var) -#pragma acc update host(_sample_arm_var) -#pragma acc update host(_detector_var) -#pragma acc update host(_instrument_var) - - siminfo_init(NULL); - save(siminfo_file); /* save data when simulation ends */ - /* call iteratively all components FINALLY */ - class_Guide_gravity_finally(&_guide_var); - class_Monochromator_curved_finally(&_mono_var); - siminfo_close(); - return(0); -} /* finally */ - -/* ***************************************************************************** -* instrument 'splitRunTest' and components DISPLAY -***************************************************************************** */ - #define magnify mcdis_magnify - #define line mcdis_line - #define dashed_line mcdis_dashed_line - #define multiline mcdis_multiline - #define rectangle mcdis_rectangle - #define box mcdis_box - #define circle mcdis_circle - #define cylinder mcdis_cylinder - #define sphere mcdis_sphere -_class_Arm *class_Arm_display(_class_Arm *_comp) { - - SIG_MESSAGE("[_Arm_display] component NULL=Arm() [Arm:0]"); - printf("MCDISPLAY: component %s\n", _comp->_name); - - /* A bit ugly; hard-coded dimensions. */ - - line(0,0,0,0.2,0,0); - line(0,0,0,0,0.2,0); - line(0,0,0,0,0,0.2); - - - return(_comp); -} /* class_Arm_display */ - -_class_Source_simple *class_Source_simple_display(_class_Source_simple *_comp) { -#define radius (_comp->_parameters.radius) -#define yheight (_comp->_parameters.yheight) -#define xwidth (_comp->_parameters.xwidth) -#define dist (_comp->_parameters.dist) -#define focus_xw (_comp->_parameters.focus_xw) -#define focus_yh (_comp->_parameters.focus_yh) -#define E0 (_comp->_parameters.E0) -#define dE (_comp->_parameters.dE) -#define lambda0 (_comp->_parameters.lambda0) -#define dlambda (_comp->_parameters.dlambda) -#define flux (_comp->_parameters.flux) -#define gauss (_comp->_parameters.gauss) -#define target_index (_comp->_parameters.target_index) -#define pmul (_comp->_parameters.pmul) -#define srcArea (_comp->_parameters.srcArea) -#define square (_comp->_parameters.square) -#define tx (_comp->_parameters.tx) -#define ty (_comp->_parameters.ty) -#define tz (_comp->_parameters.tz) - SIG_MESSAGE("[_Source_simple_display] component NULL=Source_simple() [Source_simple:0]"); - printf("MCDISPLAY: component %s\n", _comp->_name); - - if (square == 1) { - - rectangle("xy",0,0,0,xwidth,yheight); - } else { - - circle("xy",0,0,0,radius); - } - if (dist) { - dashed_line(0,0,0, -focus_xw/2+tx,-focus_yh/2+ty,tz, 4); - dashed_line(0,0,0, focus_xw/2+tx,-focus_yh/2+ty,tz, 4); - dashed_line(0,0,0, focus_xw/2+tx, focus_yh/2+ty,tz, 4); - dashed_line(0,0,0, -focus_xw/2+tx, focus_yh/2+ty,tz, 4); - } - -#undef radius -#undef yheight -#undef xwidth -#undef dist -#undef focus_xw -#undef focus_yh -#undef E0 -#undef dE -#undef lambda0 -#undef dlambda -#undef flux -#undef gauss -#undef target_index -#undef pmul -#undef srcArea -#undef square -#undef tx -#undef ty -#undef tz - return(_comp); -} /* class_Source_simple_display */ - -_class_Guide_gravity *class_Guide_gravity_display(_class_Guide_gravity *_comp) { -#define w1 (_comp->_parameters.w1) -#define h1 (_comp->_parameters.h1) -#define w2 (_comp->_parameters.w2) -#define h2 (_comp->_parameters.h2) -#define l (_comp->_parameters.l) -#define R0 (_comp->_parameters.R0) -#define Qc (_comp->_parameters.Qc) -#define alpha (_comp->_parameters.alpha) -#define m (_comp->_parameters.m) -#define W (_comp->_parameters.W) -#define nslit (_comp->_parameters.nslit) -#define d (_comp->_parameters.d) -#define mleft (_comp->_parameters.mleft) -#define mright (_comp->_parameters.mright) -#define mtop (_comp->_parameters.mtop) -#define mbottom (_comp->_parameters.mbottom) -#define nhslit (_comp->_parameters.nhslit) -#define G (_comp->_parameters.G) -#define aleft (_comp->_parameters.aleft) -#define aright (_comp->_parameters.aright) -#define atop (_comp->_parameters.atop) -#define abottom (_comp->_parameters.abottom) -#define wavy (_comp->_parameters.wavy) -#define wavy_z (_comp->_parameters.wavy_z) -#define wavy_tb (_comp->_parameters.wavy_tb) -#define wavy_lr (_comp->_parameters.wavy_lr) -#define chamfers (_comp->_parameters.chamfers) -#define chamfers_z (_comp->_parameters.chamfers_z) -#define chamfers_lr (_comp->_parameters.chamfers_lr) -#define chamfers_tb (_comp->_parameters.chamfers_tb) -#define nelements (_comp->_parameters.nelements) -#define nu (_comp->_parameters.nu) -#define phase (_comp->_parameters.phase) -#define reflect (_comp->_parameters.reflect) -#define GVars (_comp->_parameters.GVars) -#define pTable (_comp->_parameters.pTable) -#define table_present (_comp->_parameters.table_present) - SIG_MESSAGE("[_Guide_gravity_display] component NULL=Guide_gravity() [/home/g/.cache/mccode-mcstas/main/optics/Guide_gravity.comp:574]"); - printf("MCDISPLAY: component %s\n", _comp->_name); - - - if (l > 0 && nelements > 0) { - int i,j,n; - double x1,x2,x3,x4; - double y1,y2,y3,y4; - double nel = (nelements > 11 ? 11 : nelements); - - - for (n=0; n_parameters.xmin) -#define xmax (_comp->_parameters.xmax) -#define ymin (_comp->_parameters.ymin) -#define ymax (_comp->_parameters.ymax) -#define radius (_comp->_parameters.radius) -#define xwidth (_comp->_parameters.xwidth) -#define yheight (_comp->_parameters.yheight) -#define isradial (_comp->_parameters.isradial) - SIG_MESSAGE("[_Slit_display] component NULL=Slit() [Slit:0]"); - printf("MCDISPLAY: component %s\n", _comp->_name); - - - if (is_unset(radius)) { - double xw, yh; - xw = (xmax - xmin)/2.0; - yh = (ymax - ymin)/2.0; - multiline(3, xmin-xw, (double)ymax, 0.0, - (double)xmin, (double)ymax, 0.0, - (double)xmin, ymax+yh, 0.0); - multiline(3, xmax+xw, (double)ymax, 0.0, - (double)xmax, (double)ymax, 0.0, - (double)xmax, ymax+yh, 0.0); - multiline(3, xmin-xw, (double)ymin, 0.0, - (double)xmin, (double)ymin, 0.0, - (double)xmin, ymin-yh, 0.0); - multiline(3, xmax+xw, (double)ymin, 0.0, - (double)xmax, (double)ymin, 0.0, - (double)xmax, ymin-yh, 0.0); - } else { - circle("xy",0,0,0,radius); - } - -#undef xmin -#undef xmax -#undef ymin -#undef ymax -#undef radius -#undef xwidth -#undef yheight -#undef isradial - return(_comp); -} /* class_Slit_display */ - -_class_Monochromator_curved *class_Monochromator_curved_display(_class_Monochromator_curved *_comp) { -#define reflect (_comp->_parameters.reflect) -#define transmit (_comp->_parameters.transmit) -#define zwidth (_comp->_parameters.zwidth) -#define yheight (_comp->_parameters.yheight) -#define gap (_comp->_parameters.gap) -#define NH (_comp->_parameters.NH) -#define NV (_comp->_parameters.NV) -#define mosaich (_comp->_parameters.mosaich) -#define mosaicv (_comp->_parameters.mosaicv) -#define r0 (_comp->_parameters.r0) -#define t0 (_comp->_parameters.t0) -#define Q (_comp->_parameters.Q) -#define RV (_comp->_parameters.RV) -#define RH (_comp->_parameters.RH) -#define DM (_comp->_parameters.DM) -#define mosaic (_comp->_parameters.mosaic) -#define width (_comp->_parameters.width) -#define height (_comp->_parameters.height) -#define verbose (_comp->_parameters.verbose) -#define order (_comp->_parameters.order) -#define mos_rms_y (_comp->_parameters.mos_rms_y) -#define mos_rms_z (_comp->_parameters.mos_rms_z) -#define mos_rms_max (_comp->_parameters.mos_rms_max) -#define mono_Q (_comp->_parameters.mono_Q) -#define SlabWidth (_comp->_parameters.SlabWidth) -#define SlabHeight (_comp->_parameters.SlabHeight) -#define rTable (_comp->_parameters.rTable) -#define tTable (_comp->_parameters.tTable) -#define rTableFlag (_comp->_parameters.rTableFlag) -#define tTableFlag (_comp->_parameters.tTableFlag) -#define tiltH (_comp->_parameters.tiltH) -#define tiltV (_comp->_parameters.tiltV) - SIG_MESSAGE("[_Monochromator_curved_display] component NULL=Monochromator_curved() [/home/g/.cache/mccode-mcstas/main/optics/Monochromator_curved.comp:481]"); - printf("MCDISPLAY: component %s\n", _comp->_name); - - int ih; - - for(ih = 0; ih < NH; ih++) - { - int iv; - for(iv = 0; iv < NV; iv++) - { - double zmin,zmax,ymin,ymax; - double xt, yt; - - zmin = (SlabWidth+gap)*(ih-NH/2.0)+gap/2; - zmax = zmin+SlabWidth; - ymin = (SlabHeight+gap)*(iv-NV/2.0)+gap/2; - ymax = ymin+SlabHeight; - - if (RH) xt = -(zmax*zmax - zmin*zmin)/RH/2; - else xt = 0; - - if (RV) yt = -(ymax*ymax - ymin*ymin)/RV/2; - else yt = 0; - multiline(5, xt+yt, (double)ymin, (double)zmin, - xt-yt, (double)ymax, (double)zmin, - -xt-yt, (double)ymax, (double)zmax, - -xt+yt, (double)ymin, (double)zmax, - xt+yt, (double)ymin, (double)zmin); - } - } - -#undef reflect -#undef transmit -#undef zwidth -#undef yheight -#undef gap -#undef NH -#undef NV -#undef mosaich -#undef mosaicv -#undef r0 -#undef t0 -#undef Q -#undef RV -#undef RH -#undef DM -#undef mosaic -#undef width -#undef height -#undef verbose -#undef order -#undef mos_rms_y -#undef mos_rms_z -#undef mos_rms_max -#undef mono_Q -#undef SlabWidth -#undef SlabHeight -#undef rTable -#undef tTable -#undef rTableFlag -#undef tTableFlag -#undef tiltH -#undef tiltV - return(_comp); -} /* class_Monochromator_curved_display */ - -_class_Monitor *class_Monitor_display(_class_Monitor *_comp) { -#define xmin (_comp->_parameters.xmin) -#define xmax (_comp->_parameters.xmax) -#define ymin (_comp->_parameters.ymin) -#define ymax (_comp->_parameters.ymax) -#define xwidth (_comp->_parameters.xwidth) -#define yheight (_comp->_parameters.yheight) -#define restore_neutron (_comp->_parameters.restore_neutron) -#define Nsum (_comp->_parameters.Nsum) -#define psum (_comp->_parameters.psum) -#define p2sum (_comp->_parameters.p2sum) - SIG_MESSAGE("[_Monitor_display] component NULL=Monitor() [Monitor:0]"); - printf("MCDISPLAY: component %s\n", _comp->_name); - - - multiline(5, (double)xmin, (double)ymin, 0.0, - (double)xmax, (double)ymin, 0.0, - (double)xmax, (double)ymax, 0.0, - (double)xmin, (double)ymax, 0.0, - (double)xmin, (double)ymin, 0.0); - -#undef xmin -#undef xmax -#undef ymin -#undef ymax -#undef xwidth -#undef yheight -#undef restore_neutron -#undef Nsum -#undef psum -#undef p2sum - return(_comp); -} /* class_Monitor_display */ - - #undef magnify - #undef line - #undef dashed_line - #undef multiline - #undef rectangle - #undef box - #undef circle - #undef cylinder - #undef sphere -int display(void) { /* called by mccode_main for splitRunTest:DISPLAY */ - printf("MCDISPLAY: start\n"); - - /* call iteratively all components DISPLAY */ - class_Arm_display(&_origin_var); - class_Source_simple_display(&_source_var); - class_Guide_gravity_display(&_guide_var); - class_Arm_display(&_guide_end_var); - class_Slit_display(&_aperture_var); - class_Arm_display(&_split_at_var); - class_Arm_display(&_mono_point_var); - class_Monochromator_curved_display(&_mono_var); - class_Arm_display(&_sample_arm_var); - class_Monitor_display(&_detector_var); - printf("MCDISPLAY: end\n"); - return(0); -} /* display */ - -void* _getvar_parameters(char* compname) -/* enables settings parameters based use of the GETPAR macro */ -{ - #ifdef OPENACC - #define strcmp(a,b) str_comp(a,b) - #endif - if (!strcmp(compname, "origin")) return (void *) &(_origin_var._parameters); - if (!strcmp(compname, "source")) return (void *) &(_source_var._parameters); - if (!strcmp(compname, "guide")) return (void *) &(_guide_var._parameters); - if (!strcmp(compname, "guide_end")) return (void *) &(_guide_end_var._parameters); - if (!strcmp(compname, "aperture")) return (void *) &(_aperture_var._parameters); - if (!strcmp(compname, "split_at")) return (void *) &(_split_at_var._parameters); - if (!strcmp(compname, "mono_point")) return (void *) &(_mono_point_var._parameters); - if (!strcmp(compname, "mono")) return (void *) &(_mono_var._parameters); - if (!strcmp(compname, "sample_arm")) return (void *) &(_sample_arm_var._parameters); - if (!strcmp(compname, "detector")) return (void *) &(_detector_var._parameters); - return 0; -} - -void* _get_particle_var(char *token, _class_particle *p) -/* enables setpars based use of GET_PARTICLE_DVAR macro and similar */ -{ - return 0; -} - -int _getcomp_index(char* compname) -/* Enables retrieving the component position & rotation when the index is not known. - * Component indexing into MACROS, e.g., POS_A_COMP_INDEX, are 1-based! */ -{ - if (!strcmp(compname, "origin")) return 1; - if (!strcmp(compname, "source")) return 2; - if (!strcmp(compname, "guide")) return 3; - if (!strcmp(compname, "guide_end")) return 4; - if (!strcmp(compname, "aperture")) return 5; - if (!strcmp(compname, "split_at")) return 6; - if (!strcmp(compname, "mono_point")) return 7; - if (!strcmp(compname, "mono")) return 8; - if (!strcmp(compname, "sample_arm")) return 9; - if (!strcmp(compname, "detector")) return 10; - return -1; -} - -/* embedding file "/home/g/.cache/mccode-libc/main/metadata-r.c" */ -/** --- Contents of metadata-r.c ---------------------------------------------------------------------------------- */ -// Created by Gregory Tucker, Data Management Software Centre, European Spallation Source ERIC on 07/07/23. -#ifndef MCCODE_NAME -#include "metadata-r.h" -#endif - -char * metadata_table_key_component(char* key){ - if (strlen(key) == 0) return NULL; - char sep[2] = ":\0"; // matches any number of repeated colons - // look for the separator in the provided key; strtok is allowed to modify the string, so copy it - char * tok = malloc((strlen(key) + 1) * sizeof(char)); - strcpy(tok, key); - char * pch = strtok(tok, sep); // this *is* the component name (if provided) -- but we need to move the pointer - char * comp = malloc((1 + strlen(pch)) * sizeof(char)); - strcpy(comp, pch); - if (tok) free(tok); - return comp; -} -char * metadata_table_key_literal(char * key){ - if (strlen(key) == 0) return NULL; - char sep[3] = ":\0"; - char * tok = malloc((strlen(key) + 1 ) * sizeof(char)); - strcpy(tok, key); - char * pch = strtok(tok, sep); // this *is* the component name (if provided) - if (pch) pch = strtok(NULL, sep); // either NULL or the literal name - char * name = NULL; - if (pch) { - name = malloc((1 + strlen(pch)) * sizeof(char)); - strcpy(name, pch); - } - if (tok) free(tok); - return name; -} -int metadata_table_defined(int no, metadata_table_t * tab, char * key){ - if (strlen(key) == 0){ - return no; - } - char * comp = metadata_table_key_component(key); - char * name = metadata_table_key_literal(key); - // look through the table for the matching component and literal names - int number = 0; - for (int i=0; i 1) { - MPI_MASTER( - printf("Simulation '%s' (%s): running on %i nodes (master is '%s', MPI version %i.%i).\n", - instrument_name, instrument_source, mpi_node_count, mpi_node_name, MPI_VERSION, MPI_SUBVERSION); - ); - /* share the same seed, then adapt random seed for each node */ - MPI_Bcast(&mcseed, 1, MPI_LONG, 0, MPI_COMM_WORLD); /* root sends its seed to slaves */ - mcseed += mpi_node_rank; /* make sure we use different seeds per noe */ - } -#endif /* USE_MPI */ - -#ifdef OPENACC -#ifdef USE_MPI - int num_devices = acc_get_num_devices(acc_device_nvidia); - if(num_devices>0){ - int my_device = mpi_node_rank % num_devices; - acc_set_device_num( my_device, acc_device_nvidia ); - printf("Have found %d GPU devices on rank %d. Will use device %d.\n", num_devices, mpi_node_rank, my_device); - }else{ - printf("There was an issue probing acc_get_num_devices, fallback to host\n"); - acc_set_device_type( acc_device_host ); - } -#endif -#endif - - /* *** parse options ******************************************************* */ - SIG_MESSAGE("[" __FILE__ "] main START"); - mcformat = getenv(FLAVOR_UPPER "_FORMAT") ? - getenv(FLAVOR_UPPER "_FORMAT") : FLAVOR_UPPER; - instrument_exe = argv[0]; /* store the executable path */ - /* read simulation parameters and options */ - mcparseoptions(argc, argv); /* sets output dir and format */ - - -#ifdef USE_MPI - if (mpi_node_count > 1) { - /* share the same seed, then adapt random seed for each node */ - MPI_Bcast(&mcseed, 1, MPI_LONG, 0, MPI_COMM_WORLD); /* root sends its seed to slaves */ - mcseed += mpi_node_rank; /* make sure we use different seeds per node */ - } -#endif - - -/* *** install sig handler, but only once !! after parameters parsing ******* */ -#ifndef NOSIGNALS -#ifdef SIGQUIT - if (signal( SIGQUIT ,sighandler) == SIG_IGN) - signal( SIGQUIT,SIG_IGN); /* quit (ASCII FS) */ -#endif -#ifdef SIGABRT - if (signal( SIGABRT ,sighandler) == SIG_IGN) - signal( SIGABRT,SIG_IGN); /* used by abort, replace SIGIOT in the future */ -#endif -#ifdef SIGTERM - if (signal( SIGTERM ,sighandler) == SIG_IGN) - signal( SIGTERM,SIG_IGN); /* software termination signal from kill */ -#endif -#ifdef SIGUSR1 - if (signal( SIGUSR1 ,sighandler) == SIG_IGN) - signal( SIGUSR1,SIG_IGN); /* display simulation status */ -#endif -#ifdef SIGUSR2 - if (signal( SIGUSR2 ,sighandler) == SIG_IGN) - signal( SIGUSR2,SIG_IGN); -#endif -#ifdef SIGHUP - if (signal( SIGHUP ,sighandler) == SIG_IGN) - signal( SIGHUP,SIG_IGN); -#endif -#ifdef SIGILL - if (signal( SIGILL ,sighandler) == SIG_IGN) - signal( SIGILL,SIG_IGN); /* illegal instruction (not reset when caught) */ -#endif -#ifdef SIGFPE - if (signal( SIGFPE ,sighandler) == SIG_IGN) - signal( SIGSEGV,SIG_IGN); /* floating point exception */ -#endif -#ifdef SIGBUS - if (signal( SIGBUS ,sighandler) == SIG_IGN) - signal( SIGSEGV,SIG_IGN); /* bus error */ -#endif -#ifdef SIGSEGV - if (signal( SIGSEGV ,sighandler) == SIG_IGN) - signal( SIGSEGV,SIG_IGN); /* segmentation violation */ -#endif -#endif /* !NOSIGNALS */ - - - // init executed by master/host - siminfo_init(NULL); /* open SIM */ - SIG_MESSAGE("[" __FILE__ "] main INITIALISE"); - init(); - - -#ifndef NOSIGNALS -#ifdef SIGINT - if (signal( SIGINT ,sighandler) == SIG_IGN) - signal( SIGINT,SIG_IGN); /* interrupt (rubout) only after INIT */ -#endif -#endif /* !NOSIGNALS */ - -/* ================ main particle generation/propagation loop ================ */ -#ifdef USE_MPI - /* sliced Ncount on each MPI node */ - mcncount = mpi_node_count > 1 ? - floor(mcncount / mpi_node_count) : - mcncount; /* number of rays per node */ -#endif - -// MT specific init, note that per-ray init is empty -#if RNG_ALG == 2 - mt_srandom(mcseed); -#endif - - -// main raytrace work loop -#ifndef FUNNEL - // legacy version - raytrace_all(mcncount, mcseed); -#else - MPI_MASTER( - // "funneled" version in which propagation is more parallelizable - printf("\nNOTE: CPU COMPONENT grammar activated:\n 1) \"FUNNEL\" raytrace algorithm enabled.\n 2) Any SPLIT's are dynamically allocated based on available buffer size. \n"); - ); - raytrace_all_funnel(mcncount, mcseed); -#endif - - -#ifdef USE_MPI - /* merge run_num from MPI nodes */ - if (mpi_node_count > 1) { - double mcrun_num_double = (double)mcrun_num; - mc_MPI_Sum(&mcrun_num_double, 1); - mcrun_num = (unsigned long long)mcrun_num_double; - } -#endif - - - // save/finally executed by master node/thread/host - finally(); - - -#ifdef USE_MPI - MPI_Finalize(); -#endif /* USE_MPI */ - - - return 0; -} /* mccode_main */ -/* end of file "/home/g/.cache/mccode-libc/main/mccode_main.c" */ -/* end of generated C code for splitRunTest */