]> git.donarmstrong.com Git - mothur.git/blob - sharedrabundvector.h
Merge remote-tracking branch 'mothur/master'
[mothur.git] / sharedrabundvector.h
1 #ifndef SHAREDRABUNDVECTOR_H
2 #define SHAREDRABUNDVECTOR_H
3
4 /*
5  *  sharedrabundvector.h
6  *  Dotur
7  *
8  *  Created by Sarah Westcott on 12/5/08.
9  *  Copyright 2008 Schloss Lab UMASS Amherst. All rights reserved.
10  *
11  */
12
13 #include "datavector.hpp"
14 #include "sharedordervector.h"
15 #include "sharedsabundvector.h"
16 #include "sharedrabundfloatvector.h"
17 #include "rabundvector.hpp"
18 //#include "groupmap.h"
19
20 /*  DataStructure for a shared file.
21         This class is a child to datavector.  It represents OTU information at a certain distance. 
22         It is similiar to an rabundvector except each member of data knows which group it belongs to.
23         Each member of the internal container "data" is a struct of type individual. 
24         An individual which knows the OTU from which it came, 
25         the group it is in and its abundance.  */
26
27
28 class SharedRAbundVector : public DataVector {
29         
30 public:
31         SharedRAbundVector();
32         SharedRAbundVector(int);
33         //SharedRAbundVector(string, vector<int>);
34         SharedRAbundVector(const SharedRAbundVector& bv) : DataVector(bv), data(bv.data), maxRank(bv.maxRank), numBins(bv.numBins), numSeqs(bv.numSeqs), group(bv.group), index(bv.index){};
35     SharedRAbundVector(ifstream&);
36         ~SharedRAbundVector();
37
38         int getNumBins();               
39         int getNumSeqs();                                                       
40         int getMaxRank();
41         string getGroup();
42         void setGroup(string);
43         string getBinLabel();
44         void setBinLabel(string);
45         int getGroupIndex();
46         void setGroupIndex(int);                                                                
47
48         void set(int, int, string);                     //OTU, abundance, groupname
49         void setData(vector <individual>);
50         individual get(int);
51         vector <individual> getData();
52         int getAbundance(int);
53         int numNZ();
54         void sortD();  //Sorts the data in descending order.
55         void push_front(int, int, string); //abundance, otu, groupname
56         void insert(int, int, string); //abundance, otu, groupname
57         void push_back(int, string);  //abundance, groupname
58         void pop_back();
59         void resize(int);
60         int size();
61         void clear();
62         vector<individual>::reverse_iterator rbegin();
63         vector<individual>::reverse_iterator rend();
64         
65         void print(ostream&);
66         void printHeaders(ostream&);
67                 
68         RAbundVector getRAbundVector();
69         RAbundVector getRAbundVector2();
70         SAbundVector getSAbundVector();
71         OrderVector getOrderVector(map<string,int>*);
72         SharedOrderVector getSharedOrderVector();
73         SharedSAbundVector getSharedSAbundVector();
74         SharedRAbundVector getSharedRAbundVector();
75         vector<SharedRAbundVector*> getSharedRAbundVectors();
76         vector<SharedRAbundFloatVector*> getSharedRAbundFloatVectors(vector<SharedRAbundVector*>);
77         
78 private:
79         vector<individual>  data; 
80         vector<SharedRAbundVector*> lookup;
81         //GlobalData* globaldata;
82         //GroupMap* groupmap;
83         int maxRank;
84         int numBins;
85         int numSeqs;
86         string group;
87         int index;      
88         
89         int eliminateZeroOTUS(vector<SharedRAbundVector*>&);
90 };
91
92
93 #endif
94