Categories.h

Go to the documentation of this file.

#ifndef CATEGORIES_H
#define CATEGORIES_H
 
#include "Category.h"
#include <iostream>
#include <vector>
 
class Categories {
public:
  Categories(int number_of_categories = 2, std::string prefix = "cat");
 
  Categories(std::vector<std::string> vect_category_ids);
 
  Categories(const Categories &categories);
  ~Categories();
 
  // TODO looks like this could be removed as it is never used
  std::vector<int> getRankCategories();
 
  // TODO looks like this could be removed as it is never used
  void setRankCategories(std::vector<int> &set_ranks);
 
  // TODO looks like this could be removed as it is never used
  void setRankCategories();
 
  std::vector<std::string> getIdCategories();
 
  // TODO looks like this could be removed as it is never used
  void setIdCategories(std::string prefix);
 
  // TODO looks like this could be removed as it is never used
  void setIdCategories(std::vector<std::string> &set_category_ids);
 
  Category getCategoryOfRank(int rank);
 
  int getNumberCategories();
 
  friend std::ostream &operator<<(std::ostream &out,
                                  const Categories &categories);
 
  Category operator[](int index);
  Category operator[](int index) const;
 
private:
  std::vector<Category> categories_vector_;
};
 
template <typename T>
std::ostream &operator<<(std::ostream &out, const std::vector<T> &vec);
 
#endif