]> git.donarmstrong.com Git - bamtools.git/blob - src/api/SamReadGroupDictionary.h
Brought API up to compliance with recent SAM Format Spec (v1.4-r962)
[bamtools.git] / src / api / SamReadGroupDictionary.h
1 // ***************************************************************************
2 // SamReadGroupDictionary.h (c) 2010 Derek Barnett
3 // Marth Lab, Department of Biology, Boston College
4 // All rights reserved.
5 // ---------------------------------------------------------------------------
6 // Last modified: 18 April 2011 (DB)
7 // ---------------------------------------------------------------------------
8 // Provides methods for operating on a collection of SamReadGroup entries.
9 // ***************************************************************************
10
11 #ifndef SAM_READGROUP_DICTIONARY_H
12 #define SAM_READGROUP_DICTIONARY_H
13
14 #include <api/api_global.h>
15 #include <api/SamReadGroup.h>
16 #include <string>
17 #include <vector>
18
19 namespace BamTools {
20
21 typedef std::vector<SamReadGroup>             SamReadGroupContainer;
22 typedef SamReadGroupContainer::iterator       SamReadGroupIterator;
23 typedef SamReadGroupContainer::const_iterator SamReadGroupConstIterator;
24
25 class API_EXPORT SamReadGroupDictionary {
26
27     // ctor & dtor
28     public:
29         SamReadGroupDictionary(void);
30         SamReadGroupDictionary(const SamReadGroupDictionary& other);
31         ~SamReadGroupDictionary(void);
32
33     // query/modify read group data
34     public:
35         // adds a read group
36         void Add(const SamReadGroup& readGroup);
37         void Add(const std::string& readGroupId);
38
39         // adds multiple read groups
40         void Add(const std::vector<SamReadGroup>& readGroups);
41         void Add(const std::vector<std::string>& readGroupIds);
42
43         // clears all read group entries
44         void Clear(void);
45
46         // returns true if dictionary contains this read group
47         bool Contains(const SamReadGroup& readGroup) const;
48         bool Contains(const std::string& readGroupId) const;
49
50         // returns true if dictionary is empty
51         bool IsEmpty(void) const;
52
53         // removes read group, if found
54         void Remove(const SamReadGroup& readGroup);
55         void Remove(const std::string& readGroupId);
56
57         // removes multiple read groups
58         void Remove(const std::vector<SamReadGroup>& readGroups);
59         void Remove(const std::vector<std::string>& readGroupIds);
60
61         // returns number of read groups in dictionary
62         int Size(void) const;
63
64         // retrieves a modifiable reference to the SamReadGroup object associated with this ID
65         SamReadGroup& operator[](const std::string& readGroupId);
66
67     // retrieve STL-compatible iterators
68     public:
69         SamReadGroupIterator      Begin(void);              // returns iterator to begin()
70         SamReadGroupConstIterator Begin(void) const;        // returns const_iterator to begin()
71         SamReadGroupConstIterator ConstBegin(void) const;   // returns const_iterator to begin()
72         SamReadGroupIterator      End(void);                // returns iterator to end()
73         SamReadGroupConstIterator End(void) const;          // returns const_iterator to end()
74         SamReadGroupConstIterator ConstEnd(void) const;     // returns const_iterator to end()
75
76     // internal methods
77     private:
78         int IndexOf(const std::string& readGroupId) const;
79
80     // data members
81     private:
82         SamReadGroupContainer m_data;
83 };
84
85 } // namespace BamTools
86
87 #endif // SAM_READGROUP_DICTIONARY_H