20 size_t * tgt_points,
size_t count,
43 size_t const * a_ = a, * b_ = b;
45 return (*a_ > *b_) - (*b_ > *a_);
49 size_t * list_a,
size_t count_a,
size_t * list_b,
size_t count_b) {
51 if ((count_a == 0) || (count_b == 0))
return 0;
54 size_t curr_a = SIZE_MAX, curr_b = list_b[0];
57 while ((i < count_a) && (((curr_a = list_a[i++])) < curr_b));
58 if (curr_a == curr_b)
return 1;
59 while ((j < count_b) && (((curr_b = list_b[j++])) < curr_a));
60 if (curr_a == curr_b)
return 1;
61 }
while ((i < count_a) || (j < count_b));
67 size_t * list,
size_t * list_size,
size_t * insert,
size_t insert_size) {
69 if (insert_size == 0)
return;
71 size_t new_list_size = *list_size;
72 size_t old_list_size = *list_size;
74 for (
size_t i = 0, j = 0; i < insert_size; ++i) {
75 size_t curr_insert = insert[i];
76 while ((j < old_list_size) && (list[j] < curr_insert)) ++j;
77 if ((j >= old_list_size) || (list[j] != curr_insert))
78 list[new_list_size++] = curr_insert;
81 if (new_list_size != old_list_size) {
83 *list_size = new_list_size;
89 size_t tgt_start_point,
size_t * tgt_points,
size_t * count) {
91 double const * start_coord = tgt_field_coords[tgt_start_point];
93 for (
size_t i = 0, old_count = *count; i < old_count; ++i) {
95 start_coord, tgt_field_coords[tgt_points[i]]) <= max_distance) {
96 if (new_count != i) tgt_points[new_count] = tgt_points[i];
106 size_t * from_tgt_points,
size_t * to_tgt_points,
107 size_t * count,
int * flag,
size_t * temp_cell_edges,
108 size_t ** edges_buffer,
size_t * edges_buffer_array_size) {
118 size_t old_count = *
count;
119 memset(flag, 0, old_count *
sizeof(*flag));
121 for (
size_t i = 0; i < old_count; ++i) {
122 if (from_tgt_points[i] == tgt_start_point) {
128 size_t * edges = *edges_buffer;
129 size_t num_edges = 0;
130 size_t edges_array_size = *edges_buffer_array_size;
137 num_edges = curr_num_edges;
138 memcpy(edges, curr_edges, num_edges *
sizeof(*edges));
146 for (
size_t i = 0; i < old_count; ++i) {
148 if (flag[i])
continue;
154 temp_cell_edges, curr_edges, curr_num_edges *
sizeof(*temp_cell_edges));
155 qsort(temp_cell_edges, curr_num_edges,
sizeof(*edges),
compare_size_t);
158 if (
lists_overlap(edges, num_edges, temp_cell_edges, curr_num_edges)) {
159 merge_lists(edges, &num_edges, temp_cell_edges, curr_num_edges);
164 }
while (change_flag);
166 *edges_buffer = edges;
167 *edges_buffer_array_size = edges_array_size;
169 size_t new_count = 0;
170 for (
size_t i = 0; i < old_count; ++i)
171 if (flag[i]) to_tgt_points[new_count++] = from_tgt_points[i];
178 size_t num_src_points,
size_t const *
const tgt_result_points,
179 size_t * num_tgt_per_src,
size_t * spread_tgt_result_points) {
181 size_t max_num_tgt_per_src = 0;
182 for (
size_t i = 0; i < num_src_points; ++i)
183 if (num_tgt_per_src[i] > max_num_tgt_per_src)
184 max_num_tgt_per_src = num_tgt_per_src[i];
186 int * flag =
xmalloc(max_num_tgt_per_src *
sizeof(*flag));
188 size_t new_offset = 0;
189 size_t * cell_edge_buffer = NULL;
190 size_t cell_edge_buffer_array_size = 0;
191 size_t * edge_buffer = NULL;
192 size_t edge_buffer_array_size = 0;
193 size_t max_num_vertice_per_tgt = 0;
194 const int * num_vertices_per_tgt =
202 for (
size_t i = 0, old_offset = 0; i < num_src_points; ++i) {
204 size_t * old_results = spread_tgt_result_points + old_offset;
205 size_t * new_results = spread_tgt_result_points + new_offset;
206 old_offset += num_tgt_per_src[i];
211 tgt_field_coords, spread_distance, tgt_result_points[i],
212 old_results, num_tgt_per_src + i);
215 for (
size_t j = 0, curr_num_tgt_per_src = num_tgt_per_src[i];
216 j < curr_num_tgt_per_src; ++j)
217 if (num_vertices_per_tgt[old_results[j]] > max_num_vertice_per_tgt)
218 max_num_vertice_per_tgt = num_vertices_per_tgt[old_results[j]];
221 cell_edge_buffer, cell_edge_buffer_array_size,
222 max_num_vertice_per_tgt);
226 interp_grid, tgt_result_points[i],
227 old_results, new_results, num_tgt_per_src + i, flag,
228 cell_edge_buffer, &edge_buffer, &edge_buffer_array_size);
230 new_offset += num_tgt_per_src[i];
234 free(cell_edge_buffer);
242 size_t * num_src_points_,
size_t * src_points,
245 if (max_search_distance > 0.0) {
248 size_t const num_src_points = *num_src_points_;
249 size_t new_num_src_points = 0;
250 for (
size_t i = 0; i < num_src_points; ++i) {
252 src_coords[i], tgt_field_coords[tgt_result_points[i]]) <=
253 max_search_distance) {
254 if (i != new_num_src_points) {
255 src_points[new_num_src_points] = src_points[i];
256 tgt_result_points[new_num_src_points] = tgt_result_points[i];
258 new_num_src_points++;
261 *num_src_points_ = new_num_src_points;
268 size_t const *
const src_points,
size_t const *
const num_tgt_per_src,
269 size_t total_num_tgt,
size_t const *
const spread_tgt_result_points) {
271 double * weights =
xmalloc(total_num_tgt *
sizeof(*weights));
275 "ERROR(do_search_spmap): invalid weight_type")
276 switch (weight_type) {
278 for (
size_t i = 0, offset = 0; i < num_src_points; ++i) {
279 double curr_weight_data = 1.0 / (double)(num_tgt_per_src[i]);
280 for (
size_t j = 0; j < num_tgt_per_src[i]; ++j, ++offset) {
281 weights[offset] = curr_weight_data;
294 for (
size_t i = 0, offset = 0; i < num_src_points; ++i) {
296 size_t curr_num_tgt = num_tgt_per_src[i];
297 size_t const *
const curr_result_points =
298 spread_tgt_result_points + offset;
299 double const * curr_src_coord = src_field_coords[src_points[i]];
300 double * curr_weights = weights + offset;
301 offset += curr_num_tgt;
305 for (
size_t j = 0; j < curr_num_tgt; ++j) {
309 (
double*)curr_src_coord,
310 (
double*)tgt_field_coords[curr_result_points[j]]);
313 for (
size_t k = 0; k < curr_num_tgt; ++k) curr_weights[k] = 0.0;
314 curr_weights[j] = 1.0;
318 curr_weights[j] = 1.0 / distance;
324 double inv_distance_sum = 0.0;
325 for (
size_t j = 0; j < curr_num_tgt; ++j)
326 inv_distance_sum += curr_weights[j];
327 double scale = 1.0 / inv_distance_sum;
329 for (
size_t j = 0; j < curr_num_tgt; ++j) curr_weights[j] *= scale;
342 double src_area_scale,
double tgt_area_scale,
343 size_t num_src_points,
size_t const * src_points,
344 size_t const * num_tgt_per_src,
size_t const * tgt_points,
352 "ERROR(scale_weights): invalid scale_type")
365#define COMPUTE_CELL_AREA(PREFIX, IDX) \
366 yac_const_basic_grid_data_get_grid_cell( \
367 PREFIX ## _basic_grid_data, PREFIX ## _points[IDX], &grid_cell); \
368 double PREFIX ## _cell_area = yac_huiliers_area(grid_cell) * PREFIX ## _area_scale; \
370 PREFIX ## _cell_area > YAC_AREA_TOL, \
371 "ERROR(scale_weights): " \
372 "area of %s cell (global id %"XT_INT_FMT") is close to zero (%e)", \
374 PREFIX ## _basic_grid_data->ids[YAC_LOC_CELL][PREFIX ## _points[IDX]], \
375 PREFIX ## _cell_area)
376#define NO_COMPUTE_CELL_AREA(PREFIX, IDX)
378#define SCALE_WEIGHTS( \
379 COMPUTE_SRC_CELL_AREA, \
380 COMPUTE_TGT_CELL_AREA, \
381 SRC_CELL_AREA, TGT_CELL_AREA) \
383 for (size_t i = 0, offset = 0; i < num_src_points; ++i) { \
384 COMPUTE_SRC_CELL_AREA(src, i) \
385 size_t curr_num_tgt = num_tgt_per_src[i]; \
386 for (size_t j = 0; j < curr_num_tgt; ++j, ++offset) { \
387 COMPUTE_TGT_CELL_AREA(tgt, offset) \
388 weights[offset] *= SRC_CELL_AREA / TGT_CELL_AREA; \
393 switch (scale_type) {
416 double src_area_scale,
double tgt_area_scale,
417 size_t num_src_points,
size_t ** src_points_,
418 size_t ** tgt_result_points_,
double ** weights_,
419 size_t * total_num_weights_) {
421 if (spread_distance <= 0.0) {
423 *total_num_weights_ = num_src_points;
427 size_t const *
const src_points = *src_points_;
428 size_t const *
const tgt_result_points = *tgt_result_points_;
436 xmalloc(num_src_points *
sizeof(*search_bnd_circles));
437 for (
size_t i = 0; i < num_src_points; ++i) {
440 tgt_field_coords[tgt_result_points[i]],
sizeof(*tgt_field_coords));
442 search_bnd_circles[i].
sq_crd = DBL_MAX;
444 size_t * spread_tgt_result_points = NULL;
447 size_t * num_tgt_per_src =
448 xmalloc(num_src_points *
sizeof(*num_tgt_per_src));
450 interp_grid, search_bnd_circles, num_src_points,
451 &spread_tgt_result_points, num_tgt_per_src);
452 free(search_bnd_circles);
457 size_t total_num_weights =
459 interp_grid, spread_distance, num_src_points, tgt_result_points,
460 num_tgt_per_src, spread_tgt_result_points);
463 size_t * new_src_points =
464 xmalloc(total_num_weights *
sizeof(*new_src_points));
465 for (
size_t i = 0, offset = 0; i < num_src_points; ++i)
466 for (
size_t j = 0, curr_src_point = src_points[i];
467 j < num_tgt_per_src[i]; ++j, ++offset)
468 new_src_points[offset] = curr_src_point;
473 interp_grid, weight_type, num_src_points, src_points,
474 num_tgt_per_src, total_num_weights, spread_tgt_result_points);
478 interp_grid, scale_type, src_area_scale, tgt_area_scale, num_src_points,
479 src_points, num_tgt_per_src, spread_tgt_result_points, weights);
481 free(num_tgt_per_src);
482 free(*tgt_result_points_);
485 *tgt_result_points_ =
487 spread_tgt_result_points,
488 total_num_weights *
sizeof(*spread_tgt_result_points));
490 *src_points_ = new_src_points;
492 *total_num_weights_ = total_num_weights;
497 size_t * tgt_points,
size_t count,
502 "ERROR(do_search_spmap): invalid number of source fields")
505 "ERROR(do_search_spmap): "
506 "invalid source field location (has to be YAC_LOC_CELL)")
509 "ERROR(do_search_spmap): "
510 "invalid target field location (has to be YAC_LOC_CELL)")
514 size_t num_src_points;
516 interp_grid, 0, &src_points, &num_src_points);
519 interp_grid, src_points, num_src_points, 0, src_coords);
522 size_t * tgt_result_points =
523 xmalloc(num_src_points *
sizeof(*tgt_result_points));
525 interp_grid, src_coords, num_src_points, 1, tgt_result_points);
528 for (
size_t i = 0; i < num_src_points; ++i)
530 tgt_result_points[i] != SIZE_MAX,
531 "ERROR(do_search_spmap): could not find matching target point")
536 &num_src_points, src_points, src_coords, tgt_result_points);
542 double * weight_data;
543 size_t total_num_weights;
550 &src_points, &tgt_result_points, &weight_data, &total_num_weights);
554 size_t result_count = total_num_weights;
555 int to_tgt_owner = 1;
557 interp_grid, to_tgt_owner,
558 0, &src_points, &tgt_result_points, &weight_data, &result_count);
559 total_num_weights = result_count;
563 tgt_result_points, result_count, src_points, weight_data);
566 size_t * num_src_per_tgt =
xmalloc(result_count *
sizeof(*num_src_per_tgt));
567 size_t num_unique_tgt_result_points = 0;
568 for (
size_t i = 0; i < result_count;) {
570 size_t curr_tgt = tgt_result_points[i];
571 while ((i < result_count) && (curr_tgt == tgt_result_points[i])) ++i;
572 num_src_per_tgt[num_unique_tgt_result_points] = i - prev_i;
573 tgt_result_points[num_unique_tgt_result_points] = curr_tgt;
574 ++num_unique_tgt_result_points;
576 result_count = num_unique_tgt_result_points;
578 xrealloc(num_src_per_tgt, result_count *
sizeof(*num_src_per_tgt));
580 xrealloc(tgt_result_points, result_count *
sizeof(*tgt_result_points));
584 int * reorder_flag =
xmalloc(count *
sizeof(*tgt_points));
587 for (
size_t i = 0; i < result_count; ++i) {
588 size_t curr_result_tgt = tgt_result_points[i];
589 while ((j < count) && (tgt_points[j] < curr_result_tgt))
590 reorder_flag[j++] = 1;
592 (j < count) && (curr_result_tgt == tgt_points[j]),
593 "ERROR(do_search_spmap): "
594 "required target points already in use or not available")
595 reorder_flag[j++] = 0;
597 for (; j < count; ++j) reorder_flag[j] = 1;
606 interp_grid, 0, src_points, total_num_weights);
610 interp_grid, tgt_result_points, result_count),
611 .count = result_count};
612 free(tgt_result_points);
615 if (weight_data == NULL)
617 weights, &tgts, num_src_per_tgt, srcs);
620 weights, &tgts, num_src_per_tgt, srcs, weight_data);
626 free(num_src_per_tgt);
632 double spread_distance,
double max_search_distance,
635 double src_sphere_radius,
double tgt_sphere_radius) {
649 "ERROR(yac_interp_method_spmap_new): invalid spread_distance "
650 "(has to be >= 0 and <= PI/2")
654 "ERROR(yac_interp_method_spmap_new): invalid max_search_distance "
655 "(has to be >= 0 and <= PI")
658 src_sphere_radius > 0.0,
659 "ERROR(yac_interp_method_spmap_new): invalid src_sphere_radius "
663 tgt_sphere_radius > 0.0,
664 "ERROR(yac_interp_method_spmap_new): invalid tgt_sphere_radius "
Structs and interfaces for area calculations.
size_t const *const const_size_t_pointer
#define ENSURE_ARRAY_SIZE(arrayp, curr_array_size, req_size)
static double get_vector_angle(double const a[3], double const b[3])
static struct sin_cos_angle sin_cos_angle_new(double sin, double cos)
void yac_init_grid_cell(struct yac_grid_cell *cell)
void yac_free_grid_cell(struct yac_grid_cell *cell)
enum yac_location yac_interp_grid_get_tgt_field_location(struct yac_interp_grid *interp_grid)
void yac_interp_grid_relocate_src_tgt_pairs(struct yac_interp_grid *interp_grid, int to_tgt_owner, size_t src_field_idx, size_t **src_points, size_t **tgt_points, double **weights, size_t *count)
size_t yac_interp_grid_get_num_src_fields(struct yac_interp_grid *interp_grid)
void yac_interp_grid_do_nnn_search_tgt(struct yac_interp_grid *interp_grid, yac_coordinate_pointer search_coords, size_t count, size_t n, size_t *tgt_points)
void yac_interp_grid_do_bnd_circle_search_tgt(struct yac_interp_grid *interp_grid, const_bounding_circle_pointer bnd_circles, size_t count, size_t **tgt_cells, size_t *num_tgt_per_bnd_circle)
struct remote_point * yac_interp_grid_get_tgt_remote_points(struct yac_interp_grid *interp_grid, size_t *tgt_points, size_t count)
void yac_interp_grid_get_src_points(struct yac_interp_grid *interp_grid, size_t src_field_idx, size_t **src_indices, size_t *count)
enum yac_location yac_interp_grid_get_src_field_location(struct yac_interp_grid *interp_grid, size_t src_field_idx)
void yac_interp_grid_get_src_coordinates(struct yac_interp_grid *interp_grid, size_t *src_points, size_t count, size_t src_field_idx, yac_coordinate_pointer src_coordinates)
struct remote_point * yac_interp_grid_get_src_remote_points(struct yac_interp_grid *interp_grid, size_t src_field_idx, size_t *src_points, size_t count)
yac_const_coordinate_pointer yac_interp_grid_get_src_field_coords(struct yac_interp_grid *interp_grid, size_t src_field_idx)
yac_const_coordinate_pointer yac_interp_grid_get_tgt_field_coords(struct yac_interp_grid *interp_grid)
struct yac_const_basic_grid_data * yac_interp_grid_get_basic_grid_data_tgt(struct yac_interp_grid *interp_grid)
struct yac_const_basic_grid_data * yac_interp_grid_get_basic_grid_data_src(struct yac_interp_grid *interp_grid)
static int lists_overlap(size_t *list_a, size_t count_a, size_t *list_b, size_t count_b)
static void delete_spmap(struct interp_method *method)
static int compare_size_t(const void *a, const void *b)
static size_t do_search_spmap(struct interp_method *method, struct yac_interp_grid *interp_grid, size_t *tgt_points, size_t count, struct yac_interp_weights *weights)
static struct interp_method_vtable interp_method_spmap_vtable
static void check_max_search_distance(struct yac_interp_grid *interp_grid, double max_search_distance, size_t *num_src_points_, size_t *src_points, yac_coordinate_pointer src_coords, size_t *tgt_result_points)
struct interp_method * yac_interp_method_spmap_new(double spread_distance, double max_search_distance, enum yac_interp_spmap_weight_type weight_type, enum yac_interp_spmap_scale_type scale_type, double src_sphere_radius, double tgt_sphere_radius)
static void merge_lists(size_t *list, size_t *list_size, size_t *insert, size_t insert_size)
#define COMPUTE_CELL_AREA(PREFIX, IDX)
static void scale_weights(struct yac_interp_grid *interp_grid, enum yac_interp_spmap_scale_type scale_type, double src_area_scale, double tgt_area_scale, size_t num_src_points, size_t const *src_points, size_t const *num_tgt_per_src, size_t const *tgt_points, double *weights)
static double * compute_weights(struct yac_interp_grid *interp_grid, enum yac_interp_spmap_weight_type weight_type, size_t num_src_points, size_t const *const src_points, size_t const *const num_tgt_per_src, size_t total_num_tgt, size_t const *const spread_tgt_result_points)
static void remove_disconnected_points(struct yac_interp_grid *interp_grid, size_t tgt_start_point, size_t *from_tgt_points, size_t *to_tgt_points, size_t *count, int *flag, size_t *temp_cell_edges, size_t **edges_buffer, size_t *edges_buffer_array_size)
static void spread_src_data(struct yac_interp_grid *interp_grid, double spread_distance, enum yac_interp_spmap_weight_type weight_type, enum yac_interp_spmap_scale_type scale_type, double src_area_scale, double tgt_area_scale, size_t num_src_points, size_t **src_points_, size_t **tgt_result_points_, double **weights_, size_t *total_num_weights_)
static size_t check_tgt_result_points(struct yac_interp_grid *interp_grid, double spread_distance, size_t num_src_points, size_t const *const tgt_result_points, size_t *num_tgt_per_src, size_t *spread_tgt_result_points)
static void check_spread_distance(yac_const_coordinate_pointer tgt_field_coords, double max_distance, size_t tgt_start_point, size_t *tgt_points, size_t *count)
#define SCALE_WEIGHTS(COMPUTE_SRC_CELL_AREA, COMPUTE_TGT_CELL_AREA, SRC_CELL_AREA, TGT_CELL_AREA)
#define NO_COMPUTE_CELL_AREA(PREFIX, IDX)
yac_interp_spmap_scale_type
@ YAC_INTERP_SPMAP_NONE
weights are not scaled
@ YAC_INTERP_SPMAP_INVTGTAREA
@ YAC_INTERP_SPMAP_SRCAREA
@ YAC_INTERP_SPMAP_FRACAREA
yac_interp_spmap_weight_type
void yac_interp_weights_add_wsum(struct yac_interp_weights *weights, struct remote_points *tgts, size_t *num_src_per_tgt, struct remote_point *srcs, double *w)
void yac_interp_weights_add_sum(struct yac_interp_weights *weights, struct remote_points *tgts, size_t *num_src_per_tgt, struct remote_point *srcs)
#define xrealloc(ptr, size)
struct sin_cos_angle inc_angle
angle between the middle point and the boundary of the spherical cap
enum yac_interp_spmap_scale_type scale_type
enum yac_interp_spmap_weight_type weight_type
double max_search_distance
struct interp_method_vtable * vtable
size_t(* do_search)(struct interp_method *method, struct yac_interp_grid *grid, size_t *tgt_points, size_t count, struct yac_interp_weights *weights)
struct remote_point * data
const_int_pointer num_vertices_per_cell
const_size_t_pointer cell_to_edge
const_size_t_pointer cell_to_edge_offsets
void yac_quicksort_index_size_t_size_t_double(size_t *a, size_t n, size_t *b, double *c)
void yac_quicksort_index_int_size_t(int *a, size_t n, size_t *idx)
#define YAC_ASSERT(exp, msg)
double const (*const yac_const_coordinate_pointer)[3]
double(* yac_coordinate_pointer)[3]