diff options
| -rw-r--r-- | include/tng_io.h | 20 | ||||
| -rw-r--r-- | src/lib/tng_io.c | 74 |
2 files changed, 94 insertions, 0 deletions
diff --git a/include/tng_io.h b/include/tng_io.h index 62bcef7..eddfb5c 100644 --- a/include/tng_io.h +++ b/include/tng_io.h @@ -2689,6 +2689,26 @@ tng_function_status DECLSPECDLLEXPORT tng_data_block_name_get char *name, int max_len); +/** @brief Get the dependency of a data block of a specific ID. + * @param tng_data is the trajectory data container. + * @param block_id is the ID of the data block of which to get the name. + * @param block_dependency is a pointer to the depency of the data block. + * If the block is frame dependent it will be set to TNG_FRAME_DEPENDENT, + * if it is particle dependent it will be set to TNG_PARTICLE_DEPENDENT and + * if it is both it will be set to TNG_FRAME_DEPENDENT & TNG_PARTICLE_DEPENDENT. + * @pre \code tng_data != 0 \endcode The trajectory container (tng_data) + * must be initialised before using it. + * @pre \code block_dependency != 0 \endcode The pointer to the block dependency + * must not be a NULL pointer. + * @return TNG_SUCCESS (0) if the data block is found, TNG_FAILURE (1) + * if a minor error has occured or the data block is not found or + * TNG_CRITICAL (2) if a major error has occured. + */ +tng_function_status DECLSPECDLLEXPORT tng_data_block_dependency_get + (tng_trajectory_t tng_data, + int64_t block_id, + int *block_dependency); + /** * @brief Write data of one trajectory frame to the output_file of tng_data. * @param tng_data is a trajectory data container. tng_data->output_file_path diff --git a/src/lib/tng_io.c b/src/lib/tng_io.c index b313c83..5abce04 100644 --- a/src/lib/tng_io.c +++ b/src/lib/tng_io.c @@ -12850,6 +12850,80 @@ tng_function_status DECLSPECDLLEXPORT tng_data_block_name_get return(TNG_FAILURE); } +tng_function_status DECLSPECDLLEXPORT tng_data_block_dependency_get + (tng_trajectory_t tng_data, + int64_t block_id, + int *block_dependency) +{ + int64_t i; + tng_function_status stat; + tng_particle_data_t p_data; + tng_non_particle_data_t np_data; + + TNG_ASSERT(tng_data, "TNG library: Trajectory container not properly setup."); + TNG_ASSERT(block_dependency, "TNG library: block_dependency must not be a NULL pointer."); + + for(i = 0; i < tng_data->n_particle_data_blocks; i++) + { + p_data = &tng_data->non_tr_particle_data[i]; + if(p_data->block_id == block_id) + { + *block_dependency = TNG_PARTICLE_DEPENDENT; + return(TNG_SUCCESS); + } + } + for(i = 0; i < tng_data->n_data_blocks; i++) + { + np_data = &tng_data->non_tr_data[i]; + if(np_data->block_id == block_id) + { + *block_dependency = 0; + return(TNG_SUCCESS); + } + } + + stat = tng_particle_data_find(tng_data, block_id, &p_data); + if(stat == TNG_SUCCESS) + { + *block_dependency = TNG_PARTICLE_DEPENDENT & TNG_FRAME_DEPENDENT; + return(TNG_SUCCESS); + } + else + { + stat = tng_data_find(tng_data, block_id, &np_data); + if(stat == TNG_SUCCESS) + { + *block_dependency = TNG_FRAME_DEPENDENT; + return(TNG_SUCCESS); + } + else + { + stat = tng_frame_set_read_current_only_data_from_block_id(tng_data, TNG_USE_HASH, block_id); + if(stat != TNG_SUCCESS) + { + return(stat); + } + stat = tng_particle_data_find(tng_data, block_id, &p_data); + if(stat == TNG_SUCCESS) + { + *block_dependency = TNG_PARTICLE_DEPENDENT & TNG_FRAME_DEPENDENT; + return(TNG_SUCCESS); + } + else + { + stat = tng_data_find(tng_data, block_id, &np_data); + if(stat == TNG_SUCCESS) + { + *block_dependency = TNG_FRAME_DEPENDENT; + return(TNG_SUCCESS); + } + } + } + } + + return(TNG_FAILURE); +} + tng_function_status DECLSPECDLLEXPORT tng_frame_data_write (tng_trajectory_t tng_data, const int64_t frame_nr, |