frozenleaves/alphafold3
1
0
Fork 0
mirror of https://github.com/google-deepmind/alphafold3.git synced 2025-10-20 13:23:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

Validate the number of tokens in a CIF loop

Browse Source 
PiperOrigin-RevId: 797771843
Change-Id: Ib34796c2e9f402cd2cad79d67127ddb32f0803ae
This commit is contained in:
Augustin Zidek
2025-08-21 08:17:08 -07:00
committed by Copybara-Service
parent 2e2ffc10ab
commit a2b03dab51
1 changed files with 21 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/alphafold3/parsers/cpp/cif_dict_lib.cc
View File

@ -340,6 +340,16 @@ struct GroupedKeys {
int value_size;
};
absl::Status CheckLoopColumnSizes(int num_loop_keys, int num_loop_values) {
if ((num_loop_keys > 0) && (num_loop_values % num_loop_keys != 0)) {
return absl::InvalidArgumentError(absl::StrFormat(
"The number of values (%d) in a loop is not a multiple of the "
"number of the loop's columns (%d)",
num_loop_values, num_loop_keys));
}
return absl::OkStatus();
}
} // namespace
absl::StatusOr<CifDict> CifDict::FromString(absl::string_view cif_string) {
@ -380,7 +390,12 @@ absl::StatusOr<CifDict> CifDict::FromString(absl::string_view cif_string) {
++token_itr) {
auto token = *token_itr;
if (absl::EqualsIgnoreCase(token, "loop_")) {
// A new loop started, get rid of old loop's data.
// A new loop started, check the previous loop and get rid of its data.
absl::Status loop_status =
CheckLoopColumnSizes(num_loop_keys, loop_token_index);
if (!loop_status.ok()) {
return loop_status;
}
loop_flag = true;
loop_column_values.clear();
loop_token_index = 0;
@ -433,6 +448,11 @@ absl::StatusOr<CifDict> CifDict::FromString(absl::string_view cif_string) {
key = "";
}
}
absl::Status loop_status =
CheckLoopColumnSizes(num_loop_keys, loop_token_index);
if (!loop_status.ok()) {
return loop_status;
}
return CifDict(std::move(cif));
}