xcoll 0.3.5__py3-none-any.whl → 0.4.0__py3-none-any.whl

xcoll/scattering_routines/everest/scatter.h

@@ -1,169 +0,0 @@
-// copyright ############################### #
-// This file is part of the Xcoll Package.   #
-// Copyright (c) CERN, 2023.                 #
-// ######################################### #
-
-#ifndef XCOLL_EVEREST_SCAT_H
-#define XCOLL_EVEREST_SCAT_H
-#include <math.h>
-#include <stdlib.h>
-#include <stdio.h>
-
-
-/*gpufun*/
-void scatter(EverestData restrict everest, LocalParticle* part, double length){
-
-    // geometry values
-    double aperture = everest->coll->aperture;
-    double offset   = everest->coll->offset;
-    double tilt_L   = everest->coll->tilt_L;
-    double tilt_R   = everest->coll->tilt_R;
-    double side     = everest->coll->side;
-
-    // Store initial coordinates for updating later
-    double const rpp_in  = LocalParticle_get_rpp(part);
-    double const rvv_in  = LocalParticle_get_rvv(part);
-    double const e0      = LocalParticle_get_energy0(part) / 1e9; // Reference energy in GeV
-    double const beta0   = LocalParticle_get_beta0(part);
-    double const ptau_in = LocalParticle_get_ptau(part);
-    double const x_in    = LocalParticle_get_x(part);
-    double const px_in   = LocalParticle_get_px(part);
-    double const y_in    = LocalParticle_get_y(part);
-    double const py_in   = LocalParticle_get_py(part);
-    double p0 = LocalParticle_get_p0c(part) / 1e9;
-
-    // TODO: missing correction due to m/m0 (but also wrong in xpart...)
-    double energy = p0*ptau_in + e0; // energy, not momentum, in GeV
-
-    // Status flags
-    int is_hit = 0;
-    int is_abs = 0;
-
-    double x = LocalParticle_get_x(part);
-
-    // For one-sided collimators consider only positive X. For negative X jump to the next particle
-    if (side==0 || (side==1 && x>=0.) || (side==2 && x<=0.)) {
-
-        double mirror = 1;
-        double tiltangle = tilt_L;
-        if (x < 0) {
-            mirror = -1;
-            tiltangle = -tilt_R;
-            LocalParticle_scale_x(part, mirror);
-            LocalParticle_scale_px(part, mirror);
-        }
-
-        // Shift with opening and offset
-        LocalParticle_add_to_x(part, -aperture/2. - mirror*offset);
-
-        // Include collimator tilt
-        double rot_shift = YRotation_single_particle_rotate_only(part, 0., tiltangle);
-        Drift_single_particle_4d(part, -rot_shift);
-
-        // Check if/where particle hits collimator
-        double zlm; 
-        x  = LocalParticle_get_x(part);
-        double xp = LocalParticle_get_px(part)*rpp_in;
-        if (x >= 0.) { // hit at front
-            zlm = length;
-            // val_part_impact = x;
-            // store these in impact table, as interaction type 'hitting collimator' or something like that.
-            // val_part_indiv  = xp;
-        } else if (xp <= 0.) { // no hit
-            zlm = 0.;
-            Drift_single_particle_4d(part, length);
-        } else { // hit from side
-            double s = -x/xp;
-            if (s < length) {
-                zlm = length - s;
-                Drift_single_particle_4d(part, s);
-                // val_part_impact = 0.;
-                // val_part_indiv  = xp;
-            } else {
-                zlm = 0.;
-                Drift_single_particle_4d(part, length);
-            }
-        }
-
-        // Now do the scattering part
-        if (zlm > 0.) {
-            is_hit = 1;
-
-            double* jaw_result = jaw(everest, part, energy, zlm, 1);
-
-            energy = jaw_result[0];
-            if (jaw_result[1] == 1){
-                is_abs = 1;
-            }
-            double s_out = jaw_result[2];
-            free(jaw_result);
-
-            if (is_abs != 1) {
-               // Do the rest drift, if particle left collimator early
-                Drift_single_particle_4d(part, zlm-s_out);
-            }
-        }
-
-        // Transform back to particle coordinates with opening and offset
-        // Include collimator tilt
-        rot_shift = YRotation_single_particle_rotate_only(part, length, -tiltangle);
-        Drift_single_particle_4d(part, length-rot_shift);
-
-        // Transform back to particle coordinates with opening and offset
-        LocalParticle_add_to_x(part, aperture/2. + mirror*offset);
-
-        // Now mirror at the horizontal axis for negative X offset
-        if (mirror < 0) {
-            LocalParticle_scale_x(part, mirror);
-            LocalParticle_scale_px(part, mirror);
-        }
-
-    } else {
-        Drift_single_particle_4d(part, length);
-    }
-
-    // Cannot assign energy directly to LocalParticle as it would update dependent variables, but needs to be corrected first!
-
-    // Update energy    ---------------------------------------------------
-    // Only particles that hit the jaw and survived need to be updated
-    if (is_hit>0 && is_abs==0){
-        double ptau_out = (energy - e0) / (e0 * beta0);
-        LocalParticle_update_ptau(part, ptau_out);
-    }
-
-    // Update 4D coordinates    -------------------------------------------
-    // Absorbed particles get their coordinates set to the entrance of collimator
-    if (is_abs>0){
-        LocalParticle_set_x(part, x_in);
-        LocalParticle_set_px(part, px_in);
-        LocalParticle_set_y(part, y_in);
-        LocalParticle_set_py(part, py_in);
-    }
-
-    // Update longitudinal coordinate zeta    -----------------------------
-    // Absorbed particles keep coordinates at the entrance of collimator, others need correcting:
-    // Non-hit particles are just drifting (zeta not yet drifted in K2, so do here)
-    if (is_hit==0){
-        LocalParticle_add_to_zeta(part, drift_zeta_single(rvv_in, px_in*rpp_in, py_in*rpp_in, length) );
-    }
-    // Hit and survived particles need correcting:
-    if (is_hit>0 && is_abs==0){
-        double px  = LocalParticle_get_px(part);
-        double py  = LocalParticle_get_py(part);
-        double rvv = LocalParticle_get_rvv(part);
-        double rpp = LocalParticle_get_rpp(part);
-        // First we drift half the length with the old angles:
-        LocalParticle_add_to_zeta(part, drift_zeta_single(rvv_in, px_in*rpp_in, py_in*rpp_in, length/2) );
-        // then half the length with the new angles:
-        LocalParticle_add_to_zeta(part, drift_zeta_single(rvv, px*rpp, py*rpp, length/2) );
-    }
-
-    // Update state    ----------------------------------------------------
-    if (is_abs > 0){
-        LocalParticle_set_state(part, XC_LOST_ON_EVEREST_COLL);
-    }
-
-    return;
-}
-
-#endif /* XCOLL_EVEREST_SCAT_H */

xcoll/scattering_routines/everest/scatter_crystal.h

@@ -1,260 +0,0 @@
-// copyright ############################### #
-// This file is part of the Xcoll Package.   #
-// Copyright (c) CERN, 2023.                 #
-// ######################################### #
-
-#ifndef XCOLL_EVEREST_SCAT_CRY_H
-#define XCOLL_EVEREST_SCAT_CRY_H
-#include <math.h>
-#include <stdlib.h>
-#include <stdio.h>
-
-
-/*gpufun*/
-void scatter_cry(LocalParticle* part, double length, CrystalMaterialData material, RandomRutherfordData rng,
-                 double cRot, double sRot, double c_aperture, double c_offset, int side, double cry_tilt,
-                 double cry_rcurv, double cry_bend, double cry_alayer, double cry_xmax, double cry_ymax, 
-                 double cry_orient, double cry_miscut, CollimatorImpactsData record, RecordIndex record_index){
-
-    // Store initial coordinates for updating later
-    double const rpp_in  = LocalParticle_get_rpp(part);
-    double const rvv_in  = LocalParticle_get_rvv(part);
-    double const e0      = LocalParticle_get_energy0(part) / 1e9; // Reference energy in GeV
-    double const beta0   = LocalParticle_get_beta0(part);
-    double const ptau_in = LocalParticle_get_ptau(part);
-    double const x_in2    = LocalParticle_get_x(part);
-    double const px_in2   = LocalParticle_get_px(part);
-    double const y_in2    = LocalParticle_get_y(part);
-    double const py_in2   = LocalParticle_get_py(part);
-    double p0 = LocalParticle_get_p0c(part) / 1e9;
-
-
-    double x_in  = LocalParticle_get_x(part);
-    double xp_in = LocalParticle_get_px(part)*rpp_in;
-    double y_in  = LocalParticle_get_y(part);
-    double yp_in = LocalParticle_get_py(part)*rpp_in;
-    double s_in = 0;   // s at start of collimator
-
-    // TODO: missing correction due to m/m0 (but also wrong in xpart...)
-    double p_in = p0*ptau_in + e0; // energy, not momentum, in GeV
-
-    // Status flags
-    int val_part_hit = 0;
-    int val_part_abs = 0;
-    int val_part_impact = -1;
-    double val_part_indiv = -1.;
-    // double val_part_linteract = -1.;
-
-    p0 = e0;
-    double x0     = 0;
-    double xp0    = 0;
-    double nhit   = 0;
-    double nabs   = 0;
-    double nnuc0  = 0;
-    double ien0   = 0;
-    double nnuc1  = 0;
-    double ien1   = 0;
-    double iProc  = 0;
-    double n_chan = 0;
-    double n_VR   = 0;
-    double n_amorphous = 0;
-    double s_imp  = 0;
-
-    double x = x_in;
-    double xp = xp_in;
-    double z = y_in;
-    double zp = yp_in;
-    double p = p_in;
-
-    double mirror = 1.;
-
-    // TODO: use xtrack C-code for rotation element
-
-    // Transform particle coordinates to get into collimator coordinate  system
-    // First do rotation into collimator frame
-    double const cRRot = cRot;
-    double const sRRot = -sRot;
-    x  =  x_in*cRot + sRot*y_in;
-    z  =  y_in*cRot - sRot*x_in;
-    xp = xp_in*cRot + sRot*yp_in;
-    zp = yp_in*cRot - sRot*xp_in;
-
-    // Output variables
-    double x_out = x_in;
-    double y_out = y_in;
-    double xp_out = xp_in;
-    double yp_out = yp_in;
-    double p_out = p_in;
-
-
-    // For one-sided collimators consider only positive X. For negative X jump to the next particle
-    if (side==0 || (side==1 && x>=0.) || (side==2 && x<=0.)) {
-        // Log input energy + nucleons as per the FLUKA coupling
-        nnuc0 = nnuc0 + 1.;
-        ien0 = ien0 + p_in * 1.0e3;
-
-        // Now mirror at the horizontal axis for negative X offset
-        if (x < 0) {
-            mirror    = -1;
-        } else {
-            mirror    = 1;
-        }
-        x  = mirror*x;
-        xp = mirror*xp;
-
-        // Shift with opening and offset
-        x = (x - c_aperture/2.) - mirror*c_offset;
-
-        // particle passing above the jaw are discarded => take new event
-        // entering by the face, shorten the length (zlm) and keep track of
-        // entrance longitudinal coordinate (keeps) for histograms
-
-        // The definition is that the collimator jaw is at x>=0.
-
-        // 1) Check whether particle hits the collimator
-        int isimp = 0;
-        double s = 0.;
-        double zlm = -1*length;
-
-        double* crystal_result = crystal(rng, part, x,
-                                xp,
-                                z,
-                                zp,
-                                s,
-                                p,
-                                x0,
-                                xp0,
-                                zlm,
-                                s_imp,
-                                isimp,
-                                val_part_hit,
-                                val_part_abs,
-                                val_part_impact,
-                                val_part_indiv,
-                                length,
-                                material,
-                                nhit,
-                                nabs,
-                                cry_tilt,
-                                cry_rcurv,
-                                cry_bend,
-                                cry_alayer,
-                                cry_xmax,
-                                cry_ymax,
-                                cry_orient,
-                                cry_miscut,
-                                iProc,
-                                n_chan,
-                                n_VR,
-                                n_amorphous
-                                );
-
-        val_part_hit = crystal_result[0];
-        val_part_abs = crystal_result[1];
-        val_part_impact = crystal_result[2];
-        val_part_indiv = crystal_result[3];
-        nhit = crystal_result[4];
-        nabs = crystal_result[5];
-        s_imp = crystal_result[6];
-        isimp = crystal_result[7];
-        s = crystal_result[8];
-        zlm = crystal_result[9];
-        x = crystal_result[10];
-        xp = crystal_result[11];
-        x0 = crystal_result[12];
-        xp0 = crystal_result[13];
-        z = crystal_result[14];
-        zp = crystal_result[15];
-        p = crystal_result[16];
-        iProc = crystal_result[17];
-        n_chan = crystal_result[18];
-        n_VR = crystal_result[19];
-        n_amorphous = crystal_result[20];
-
-        free(crystal_result);
-
-        if (nabs != 0.) {
-            val_part_abs = 1.;
-            // val_part_linteract = zlm;
-        }
-        s_imp  = (s - length) + s_imp;
-
-        // Transform back to particle coordinates with opening and offset
-        x   = (x + c_aperture/2) + mirror*c_offset;
-
-        // Now mirror at the horizontal axis for negative X offset
-        x  = mirror * x;
-        xp = mirror * xp;
-
-        // Last do rotation into collimator frame
-        x_out  =  x*cRRot +  z*sRRot;
-        y_out  =  z*cRRot -  x*sRRot;
-        xp_out = xp*cRRot + zp*sRRot;
-        yp_out = zp*cRRot - xp*sRRot;
-
-        // Log output energy + nucleons as per the FLUKA coupling
-        // Do not log dead particles
-        nnuc1       = nnuc1 + 1;                           // outcoming nucleons
-        ien1        = ien1  + p_out * 1e3;                 // outcoming energy
-
-        p_out = p;
-        s_in = s_in + s;
-    }
-
-    LocalParticle_set_x(part, x_out);
-    LocalParticle_set_px(part, xp_out/rpp_in);
-    LocalParticle_set_y(part, y_out);
-    LocalParticle_set_py(part, yp_out/rpp_in);
-
-    double energy_out = p_out;       //  Cannot assign energy directly to LocalParticle as it would update dependent variables, but needs to be corrected first!
-
-    // Update energy    ---------------------------------------------------
-    // Only particles that hit the jaw and survived need to be updated
-    if (val_part_hit>0 && val_part_abs==0){
-        double ptau_out = (energy_out - e0) / (e0 * beta0);
-        LocalParticle_update_ptau(part, ptau_out);
-    }
-
-    // Update 4D coordinates    -------------------------------------------
-    // Absorbed particles get their coordinates set to the entrance of collimator
-    if (val_part_abs>0){
-        LocalParticle_set_x(part, x_in2);
-        LocalParticle_set_px(part, px_in2);
-        LocalParticle_set_y(part, y_in2);
-        LocalParticle_set_py(part, py_in2);
-    }
-
-    // Update longitudinal coordinate zeta    -----------------------------
-    // Absorbed particles keep coordinates at the entrance of collimator, others need correcting:
-    // Non-hit particles are just drifting (zeta not yet drifted in K2, so do here)
-    if (val_part_hit==0){
-        LocalParticle_add_to_zeta(part, drift_zeta_single(rvv_in, px_in2*rpp_in, py_in2*rpp_in, length) );
-    }
-    // Hit and survived particles need correcting:
-    if (val_part_hit>0 && val_part_abs==0){
-        double px  = LocalParticle_get_px(part);
-        double py  = LocalParticle_get_py(part);
-        double rvv = LocalParticle_get_rvv(part);
-        double rpp = LocalParticle_get_rpp(part);
-        // First we drift half the length with the old angles:
-        LocalParticle_add_to_zeta(part, drift_zeta_single(rvv_in, px_in2*rpp_in, py_in2*rpp_in, length/2) );
-        // then half the length with the new angles:
-        LocalParticle_add_to_zeta(part, drift_zeta_single(rvv, px*rpp, py*rpp, length/2) );
-    }
-
-    // Update s    --------------------------------------------------------
-    // TODO: move absorbed particles to last impact location
-    if (val_part_abs==0){
-        LocalParticle_add_to_s(part, length);
-    }
-
-    // Update state    ----------------------------------------------------
-    if (val_part_abs > 0){
-        LocalParticle_set_state(part, XC_LOST_ON_EVEREST_CRYSTAL);
-    }
-
-    return;
-
-}
-
-#endif /* XCOLL_EVEREST_SCAT_CRY_H */