TravellingSalesmanProblem.h

Go to the documentation of this file.

#ifndef TRAVELLINGSALESMANPROBLEM_H
#define TRAVELLINGSALESMANPROBLEM_H

#include <iostream>
#include <fstream>
#include <sstream>
#include <algorithm>
#include <vector>
#include "RandomHelper.h"
#include "CoordinatesHelper.h"
#include "IData.h"

namespace operational_research
{

class TravellingSalesmanProblem : public IData
{
private:
    // Number of cities.
    unsigned int cities;

    // Coordinates of cities.
    std::vector<CoordinatesHelper> coordinates;

    bool checkSchedule(const std::vector<unsigned int>& schedule);

    void writeResultToFile(const std::vector<unsigned int>& schedule, double costFunctionValue, const std::string& filename);

public:
    TravellingSalesmanProblem(unsigned int cities, const std::vector<CoordinatesHelper>& coordinates);

    TravellingSalesmanProblem(unsigned int cities, double minCoordinate, double maxCoordinate, bool integerValues = false);

    TravellingSalesmanProblem(const std::string& filename);

    void setData(unsigned int cities, const std::vector<CoordinatesHelper>& coordinates);

    void generateData(unsigned int cities, double minCoordinate, double maxCoordinate, bool integerValues = false);

    void readDataFromFile(const std::string& filename);

    void writeDataToFile(const std::string& filename);

    void showData();

    double evaluateSTSP(const std::vector<unsigned int>& schedule);

    double evaluateSTSP(const std::vector<unsigned int>& schedule, const std::string& filename);

    unsigned int getCities() const {
        return cities;
    }
    
    const std::vector<CoordinatesHelper>& getCoordinates() const {
        return coordinates;
    }
};

} // namespace operational_research

#endif // TRAVELLINGSALESMANPROBLEM_H