X-Git-Url: https://git.donarmstrong.com/?p=mothur.git;a=blobdiff_plain;f=groupmap.h;h=5397e02e7c158e5074a7f8e626d813561cc6d9b6;hp=54085a19bab041d0caf17981c8c6c81f05a0c8b9;hb=615301e57c25e241356a9c2380648d117709458d;hpb=0746c58a680458c0bc874b3a5fc1334b12ed2f18 diff --git a/groupmap.h b/groupmap.h index 54085a1..5397e02 100644 --- a/groupmap.h +++ b/groupmap.h @@ -2,7 +2,7 @@ #define GROUPMAP_H /* * groupmap.h - * Dotur + * Mothur * * Created by Sarah Westcott on 12/1/08. * Copyright 2008 Schloss Lab UMASS Amherst. All rights reserved. @@ -17,27 +17,42 @@ class GroupMap { public: - GroupMap() {}; + GroupMap() { m = MothurOut::getInstance(); } GroupMap(string); ~GroupMap(); int readMap(); + int readMap(string); int readDesignMap(); + int readDesignMap(string); int getNumGroups(); bool isValidGroup(string); //return true if string is a valid group string getGroup(string); void setGroup(string, string); - vector namesOfGroups; - map groupIndex; //groupname, vectorIndex in namesOfGroups. - used by collectdisplays and libshuff commands. + vector getNamesOfGroups() { + sort(namesOfGroups.begin(), namesOfGroups.end()); + groupIndex.clear(); + for (int i = 0; i < namesOfGroups.size(); i++) { groupIndex[namesOfGroups[i]] = i; } + return namesOfGroups; + } + vector getNamesSeqs(); + void setNamesOfGroups(vector sn) { namesOfGroups = sn; } int getNumSeqs() { return groupmap.size(); } - vector getNamesSeqs(); vector getNamesSeqs(vector); //get names of seqs belonging to a group or set of groups int getNumSeqs(string); //return the number of seqs in a given group - + int getCopy(GroupMap*); + int renameSeq(string, string); + int print(ofstream&); + int print(ofstream&, vector); //print certain groups + + + map groupIndex; //groupname, vectorIndex in namesOfGroups. - used by collectdisplays and libshuff commands. + private: + vector namesOfGroups; MothurOut* m; ifstream fileHandle; string groupFileName; - int index; + int index; map::iterator it; void setNamesOfGroups(string); map groupmap; //sequence name and groupname