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
|
//
// designmap.h
// Mothur
//
// Created by SarahsWork on 6/17/13.
// Copyright (c) 2013 Schloss Lab. All rights reserved.
//
#ifndef __Mothur__designmap__
#define __Mothur__designmap__
#include "mothurout.h"
#include "utils.hpp"
/* This class is a representation of the design file.
group treatment sex age
F000142 Early female young
F000132 Late male old
F000138 Mid male old
*/
class DesignMap {
public:
DesignMap() { m = MothurOut::getInstance(); defaultClass = "not found"; }
DesignMap(string); //reads file as well
~DesignMap() = default;
//read designfile name
int read(string);
//like groupMap getGroup
string get(string, string); //groupName, category returns value. example F000132, sex -> male
string get(string); //groupName, returns default categories value. example F000132, -> late
//like groupMap getNamesOfGroups
vector<string> getCategory(string); //categoryName, returns values. example treatment, -> early,late,mid
vector<string> getCategory(); //returns default categories values. example treatment, -> early,late,mid
int setValues(string, map<string, string>); //groupName, map<category, value>
int push_back(string, vector<string>); //groupName, vector<value> - assumes you put values in order of getNamesOfCategories
//refers to header labels
vector<string> getNamesOfCategories() {
sort(namesOfCategories.begin(), namesOfCategories.end());
return namesOfCategories;
}
//set deault treatment, mothur sets this to column 2.
void setDefaultClass(string);
string getDefaultClass() { return defaultClass; }
//number of treatments / columns in file
int getNumCategories() { return (int)namesOfCategories.size(); }
//number of groups / rows in file
int getNumGroups() { return (int)designMap.size(); }
//options to select groups based on values
vector<string> getNamesGroups() { return groups; }
vector<string> getNamesGroups(string, string); //get names groups with category and value.
vector<string> getNamesGroups(vector<string>); //assume default category and get names groups that match any values in vector passed in. <early, late> = F000142, F000132.
//options to selects - may want to expand on these
int getNumUnique(map<string, vector<string> >); //get number of groups belonging to a category or set of categories, with value or a set of values. Must have all categories and values. Example:
// map<treatment - > early, late>, <sex -> male> would return 1. Only one group is male and from early or late.
int getNumShared(map<string, vector<string> >); //get number of groups belonging to a category or set of categories, with value or a set of values. Must have at least one categories and values. Example:
// map<treatment - > early, late>, <sex -> male> would return 3. All three group have are either male or from early or late.
vector<string> getNamesUnique(map<string, vector<string> >); //get names of groups belonging to a category or set of categories, with value or a set of values. Must have all categories and values. Example:
// map<treatment - > early, late>, <sex -> male> would return F000132. F000132 is the only group which is male and from early or late.
vector<string> getNamesShared(map<string, vector<string> >); //get names of groups belonging to a category or set of categories, with value or a set of values. Must have at least one categories and values. Example:
// map<treatment - > early, late>, <sex -> male> would return F000132, F000142, F000138. All three group have are either male or from early or late.
int print(ofstream&);
int printCategories(ofstream&, vector<string>); //print certain categories
int printGroups(ofstream&, vector<string>); //print certain Groups
private:
string defaultClass;
MothurOut* m;
vector< map<string, int> > totalCategories; //for each category, total groups assigned to it. vector[0] early -> 1, vector[1] male -> 2
vector<string> groups;
vector<string> namesOfCategories;
vector< vector<string> > designMap;
map<string, int> indexGroupNameMap; //maps groupName to row in values
map<string, int> indexCategoryMap; //maps category to column in values
Utils util;
};
#endif /* defined(__Mothur__designmap__) */
|
