diff --git a/DBEngine.cpp b/DBEngine.cpp index eaa3e49..16b6f7a 100755 --- a/DBEngine.cpp +++ b/DBEngine.cpp @@ -52,13 +52,19 @@ void DBEngine::saveToFile(vector cmds){ //assumes that all attribute titles are unique Relation DBEngine::projection(vector input, Relation r){ + vector v; + string new_name = r.getTableName() + " Projection"; + for(int i = 0; i < input.size(); ++i) { for(int j = 0; j < r.getSize(); ++j) { if((r.getAttributes())[j].getName() == input[i]) - (r.getAttributes())[j].display(); + v.push_back((r.getAttributes())[j]); } } + + Relation temp(new_name, v); + return temp; } //ASAP: TEST ALL OF THIS