File size: 4,253 Bytes
35cdf53 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 | // Copyright 2024 DeepMind Technologies Limited
//
// AlphaFold 3 source code is licensed under CC BY-NC-SA 4.0. To view a copy of
// this license, visit https://creativecommons.org/licenses/by-nc-sa/4.0/
//
// To request access to the AlphaFold 3 model parameters, follow the process set
// out at https://github.com/google-deepmind/alphafold3. You may only use these
// if received directly from Google. Use is subject to terms of use available at
// https://github.com/google-deepmind/alphafold3/blob/main/WEIGHTS_TERMS_OF_USE.md
#include "alphafold3/parsers/cpp/fasta_iterator_lib.h"
#include <cstddef>
#include <optional>
#include <string>
#include <utility>
#include <vector>
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "absl/strings/ascii.h"
#include "absl/strings/str_cat.h"
#include "absl/strings/str_split.h"
#include "absl/strings/string_view.h"
#include "absl/strings/strip.h"
namespace alphafold3 {
// Parse FASTA string and return list of strings with amino acid sequences.
// Returns a list of amino acid sequences only.
std::vector<std::string> ParseFasta(absl::string_view fasta_string) {
std::vector<std::string> sequences;
std::string* sequence = nullptr;
for (absl::string_view line_raw : absl::StrSplit(fasta_string, '\n')) {
absl::string_view line = absl::StripAsciiWhitespace(line_raw);
if (absl::ConsumePrefix(&line, ">")) {
sequence = &sequences.emplace_back();
} else if (!line.empty() && sequence != nullptr) {
absl::StrAppend(sequence, line);
}
}
return sequences;
}
// Parse FASTA string and return list of strings with amino acid sequences.
// Returns two lists: The first one with amino acid sequences, the second with
// the descriptions associated with each sequence.
std::pair<std::vector<std::string>, std::vector<std::string>>
ParseFastaIncludeDescriptions(absl::string_view fasta_string) {
std::pair<std::vector<std::string>, std::vector<std::string>> result;
auto& [sequences, descriptions] = result;
std::string* sequence = nullptr;
for (absl::string_view line_raw : absl::StrSplit(fasta_string, '\n')) {
absl::string_view line = absl::StripAsciiWhitespace(line_raw);
if (absl::ConsumePrefix(&line, ">")) {
descriptions.emplace_back(line);
sequence = &sequences.emplace_back();
} else if (!line.empty() && sequence != nullptr) {
absl::StrAppend(sequence, line);
}
}
return result;
}
absl::StatusOr<std::pair<std::string, std::string>> FastaFileIterator::Next() {
std::string line_str;
while (std::getline(reader_, line_str)) {
absl::string_view line = line_str;
line = absl::StripAsciiWhitespace(line);
if (absl::ConsumePrefix(&line, ">")) {
if (!description_.has_value()) {
description_ = line;
} else {
std::pair<std::string, std::string> output(sequence_, *description_);
description_ = line;
sequence_ = "";
return output;
}
} else if (description_.has_value()) {
absl::StrAppend(&sequence_, line);
}
}
has_next_ = false;
reader_.close();
if (description_.has_value()) {
return std::pair(sequence_, *description_);
} else {
return absl::InvalidArgumentError(
absl::StrCat("Invalid FASTA file: ", filename_));
}
}
absl::StatusOr<std::pair<std::string, std::string>>
FastaStringIterator::Next() {
size_t consumed = 0;
for (absl::string_view line_raw : absl::StrSplit(fasta_string_, '\n')) {
consumed += line_raw.size() + 1; // +1 for the newline character.
absl::string_view line = absl::StripAsciiWhitespace(line_raw);
if (absl::ConsumePrefix(&line, ">")) {
if (!description_.has_value()) {
description_ = line;
} else {
std::pair<std::string, std::string> output(sequence_, *description_);
description_ = line;
sequence_ = "";
fasta_string_.remove_prefix(consumed);
return output;
}
} else if (description_.has_value()) {
absl::StrAppend(&sequence_, line);
}
}
has_next_ = false;
if (description_.has_value()) {
return std::pair(sequence_, *description_);
} else {
return absl::InvalidArgumentError("Invalid FASTA string");
}
}
} // namespace alphafold3
|