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
|
/*
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
/*
* Fasta.hpp
*
* Created on: May 27, 2009
* Author: caa
*/
#ifndef FASTA_HPP_
#define FASTA_HPP_
#include <string>
#include "bam.h"
#include "faidx.h"
namespace fasta {
using namespace std;
class Fasta {
public:
Fasta()
{
fai=NULL;
}
Fasta(const string & fileName)
{
fai=NULL;
fai = fai_load(fileName.c_str());
if (!fai) {
throw string("Fasta: cannot open reference file.");
}
}
string getSequence(const string & tid, int start, int end)
{
char *str;
char *ref;
str = (char*)calloc(strlen(tid.c_str()) + 30, 1);
sprintf(str, "%s:%d-%d", tid.c_str(), start, end);
int len;
ref = fai_fetch(fai, str, &len);
if (len==0) throw string("faidx error, len==0");
string result(ref);
transform(result.begin(), result.end(), result.begin(), ::toupper);
free(str);
free(ref);
return result;
}
~Fasta()
{
if (fai) fai_destroy(fai);
}
protected:
faidx_t *fai;
};
}
#endif /* FASTA_HPP_ */
|