camserver_config/camserver_sf
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
850baa581663be649cd835c8d35a3899b5cfe199
camserver_sf/configuration/user_scripts/single_photon.c

284 lines
9.6 KiB
C
Raw Blame History

#include "module.c"
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <math.h>
const char *CHANNEL_NAMES[] = {"EVENT_NUM", "EVENT_I", "EVENT_J", "EVENT_CHARGE", "EVENT_ETA_X", "EVENT_ETA_Y", "EVENT_I_INTERP", "EVENT_J_INTERP"};
// max number of events per frame
#define MAX_NUM_EVENTS 100
#define EVENT_CHANNELS 7
double evt_p[EVENT_CHANNELS][MAX_NUM_EVENTS];
int func_ph_1d_double( double *frame, int i_dim, int j_dim, double *th_m);
void setArrayToValue(double array[], int size, double value) {
for (int i = 0; i < size; i++) {
array[i] = value;
}
}
//Initialization: Threshold & background
initialized = 0;
//const char *THRESHOLD_FILE = "/configuration/user_scripts/lib/Thresh_2D_start600_size800.npy";
const char *THRESHOLD_FILE = "/configuration/user_scripts/lib/threshold_2d_start600_800.txt";
int threshold_num_elements=-1;
double *threshold= NULL;
double *background= NULL;
const int MAX_LINE_LENGTH = 50000;
int initialize(int size_x, int size_y, PyObject *pars){
//background (all matrices are indexed in 1d)
background = malloc(size_x*size_y*sizeof(double));
for(int i=0; i<size_y;i++) {
for(int j=0;j<size_x;j++) {
background[i*size_x+j]=0.0;
}
}
threshold = (double *)malloc(size_x*size_y*sizeof(double));
int ret = 1;
if (THRESHOLD_FILE!=NULL){
FILE *file = fopen(THRESHOLD_FILE, "rb");
if (file == NULL) {
printf("Failed to open file.\n");
return -1;
}
char line[MAX_LINE_LENGTH];
int x = 0;
int y = 0;
while (fgets(line, sizeof(line), file) != NULL) {
char* token = strtok(line, " ");
while (token != NULL) {
threshold[y * size_x + x] = atof(token);
token = strtok(NULL, " ");
x++;
}
y++;
if (x != size_x){
printf("Invalid threshold file: wrong number of columns\n");
ret = -2;
break;
}
x = 0;
}
if (y != size_y){
printf("Invalid threshold file: wrong number of rows\n");
ret = -3;
}
fclose(file);
if (ret<0){
free(threshold); threshold = NULL;
}
} else {
PyObject* threshold_obj = PyDict_GetItemString(pars, "threshold");
double threshold_val = 60000.0;
if (threshold_obj!=NULL){
if (PyFloat_Check(threshold_obj)) {
threshold_val = PyFloat_AsDouble(threshold_obj);
} else if (PyLong_Check(threshold_obj)) {
threshold_val = PyLong_AsDouble(threshold_obj);
}
}
for(int i=0; i<size_y;i++) {
for(int j=0;j<size_x;j++) {
threshold[i*size_x+j]=100.0;//threshold_val; #FIX ME
}
}
}
return ret;
}
//def process_image(image, pulse_id, timestamp, x_axis, y_axis, parameters, bsdata=None):
PyObject *process(PyObject *self, PyObject *args)
{
PyArrayObject *image;
long pulse_id;
PyObject /*double*/ *timestamp;
long seconds, nanos;
PyArrayObject *x_axis;
PyArrayObject *y_axis;
PyObject *pars;
PyObject *bsdata;
//if (!PyArg_ParseTuple(args, "OldOOO|O", &image, &pulse_id, &timestamp, &x_axis, &y_axis, &pars, &bsdata))
if ( !PyArg_ParseTuple(args, "OlOOOO|O", &image, &pulse_id, &timestamp, &x_axis, &y_axis, &pars, &bsdata) ||
!PyArg_ParseTuple(timestamp, "ll", &seconds, &nanos) )
return NULL;
if (pulse_id < 0) {
PyErr_SetString(moduleErr, "Invalid Pulse ID");
return NULL;
}
//Acessing image
int element_size = image->descr->elsize;
int dims = image->nd;
int size_x = image->dimensions[1];
int size_y = image->dimensions[0];
unsigned short* img_data = (unsigned short*)image->data;
//Initialization
if (initialized==0){
initialized = initialize(size_x, size_y, pars);
}
int i,j,l;
int i_dim=size_y;
int j_dim=size_x;
double *frameBKsub = malloc(i_dim*j_dim*sizeof(double));
for(i=0; i<i_dim;i++) {
for(j=0;j<j_dim;j++) {
frameBKsub[i*j_dim+j]=(double)img_data[i*j_dim+j] - background[i*j_dim+j];
}
}
int evnt_num = func_ph_1d_double( frameBKsub, i_dim, j_dim, threshold);
// int evnt_num = func_ph_1d_double( (double)img_data, i_dim, j_dim, threshold);
//Create return dictionary
PyObject *ret = PyDict_New();
PyObject* camera_name = PyDict_GetItemString(pars, "camera_name");
const char * camera_name_str = PyUnicode_AsUTF8(camera_name);
char channel_name[200];
//PyDict_SetItemString(ret, "camera_name", camera_name);
sprintf(channel_name, "%s:%s",camera_name_str, CHANNEL_NAMES[0]);
PyDict_SetItemString(ret, channel_name, PyLong_FromLong(evnt_num));
for (int i=0; i<EVENT_CHANNELS; i++){
sprintf(channel_name, "%s:%s",camera_name_str, CHANNEL_NAMES[i+1]);
npy_intp arr_dims[1] = {MAX_NUM_EVENTS};
//setArrayToValue(evt_p[i],MAX_NUM_EVENTS, i+10.0);
PyObject *parr =PyArray_SimpleNewFromData(1, arr_dims, NPY_DOUBLE, evt_p[i]);
PyDict_SetItemString(ret, channel_name, parr);
}
//for (l = 0; l < EVENT_CHANNELS; l++) {
// free(evns1d.evnt_ijc[l]);
//}
//free(evns1d.evnt_ijc);
free(frameBKsub);
return ret;
}
int func_ph_1d_double( double *frame, int i_dim, int j_dim, double *th_m)
{
//int th= 50;
int i, j, l=0, m=0, n=0, evt_m_i=0, evt_i=0 ;
int dist_i, dist_j;
double charge_evt;
int th;
//double **evt_p= (double**)malloc(EVENT_CHANNELS * sizeof(double*));
//for (l = 0; l < EVENT_CHANNELS; l++) {
// evt_p[l] = (double*)malloc(MAX_NUM_EVENTS * sizeof(double));}
/*Counter variables for the loop*/
double charge = 0;
double eta_x = 0;
double eta_y = 0;
double i_interp = 0;
double j_interp = 0;
double C[]= { 4.37805097e-03, 2.43266401e-01, -7.81479328e+00, 7.70533057e+01, -2.37906124e+02, 3.42988113e+02, -2.35979731e+02, 6.24074562e+01};
double D[]= {-2.05180773e-04, 4.35314696e-01, -1.32557223e+01, 1.19990373e+02, -3.80696758e+02, 5.81427096e+02, -4.30789104e+02, 1.23894761e+02};
for(i=0; i<i_dim-1; i++) {
if (evt_i>=MAX_NUM_EVENTS){
break;
}
for(j=0;j<j_dim-1;j++) {
if (evt_i>=MAX_NUM_EVENTS){
break;
}
// 2x2 version
charge = frame[i*j_dim+j]+frame[(i+1)*j_dim+j] + frame[i*j_dim+(j+1)]+frame[(i+1)*j_dim+j+1];
//pixel by pixel threshold
th = th_m[i*j_dim +j];
//check if charge above threshold
if(charge>th) {
eta_x = (frame[(i+1)*j_dim + j ]+frame[(i+1)*j_dim + (j+1)])/charge;
eta_y = (frame[ i*j_dim + (j+1)]+frame[(i+1)*j_dim + (j+1)])/charge;
i_interp = i + (C[0] + C[1]*eta_x+ C[2]*pow(eta_x,2) + C[3]*pow(eta_x,3) + C[4]*pow(eta_x,4)+ C[5]*pow(eta_x,5) + C[6]*pow(eta_x,6) + C[7]*pow(eta_x,7));
j_interp = j + (D[0] + D[1]*eta_y+ D[2]*pow(eta_y,2) + D[3]*pow(eta_y,3) + D[4]*pow(eta_y,4)+ D[5]*pow(eta_y,5) + D[6]*pow(eta_y,6) + D[7]*pow(eta_y,7));
// 1st case: first event
if(evt_i==0){
evt_p[0][evt_i] = i;
evt_p[1][evt_i] = j;
evt_p[2][evt_i] = charge;
evt_p[3][evt_i] = eta_x;
evt_p[4][evt_i] = eta_y;
evt_p[5][evt_i] = i_interp; //
evt_p[6][evt_i] = j_interp; //
evt_i++;
} else {
// 2nd case: not 1st event. we check if it is a neighbourg of the previos events and if charge is larger
n=0;
evt_m_i = evt_i;
for(m=0; m<evt_m_i; m++) {
dist_i = abs(evt_p[0][evt_i-1-m] - i); //fix here
dist_j = abs(evt_p[1][evt_i-1-m] - j);
charge_evt = evt_p[2][evt_i-1-m];
if( dist_i< 2 && dist_j<2) {
if(charge_evt < charge) {
//2nd case: not 1st event, but neigboor of previos event, this event has more charge
evt_p[0][evt_i-1-m] = i;
evt_p[1][evt_i-1-m] = j;
evt_p[2][evt_i-1-m] = charge;
evt_p[3][evt_i-1-m] = eta_x;
evt_p[4][evt_i-1-m] = eta_y;
evt_p[5][evt_i-1-m] = i_interp;
evt_p[6][evt_i-1-m] = j_interp;
} else {
//3d case not 1st event, but neigboor of previos event, this event has less charge
}
} else {
// not a neighbor of the m event
n++;
//now we check if it is not a neighborg of any previous event
if((n)==evt_i) {
evt_p[0][evt_i] = i;
evt_p[1][evt_i] = j;
evt_p[2][evt_i] = charge;
evt_p[3][evt_i] = eta_x;
evt_p[4][evt_i] = eta_y;
evt_p[5][evt_i] = i_interp; //
evt_p[6][evt_i] = j_interp; //
evt_i++;
}
}
}
}
}
}
}
return evt_i;
}
Reference in New Issue View Git Blame Copy Permalink