QuadraticAssignmentProblem.h

Go to the documentation of this file.

#ifndef QUADRATICASSIGNMENTPROBLEM_H
#define QUADRATICASSIGNMENTPROBLEM_H

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

namespace operational_research
{

class QuadraticAssignmentProblem : public IData
{
private:
    // Number of objects.
    unsigned int objects;

    // Flows.
    std::vector<std::vector<double> > flows;

    // Distances.
    std::vector<std::vector<double> > distances;

    // Costs of assignment.
    std::vector<std::vector<double> > costs;

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

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

public:
    QuadraticAssignmentProblem(unsigned int objects, const std::vector<std::vector<double> >& flows, const std::vector<std::vector<double> >& distances, const std::vector<std::vector<double> >& costs);

    QuadraticAssignmentProblem(unsigned int objects, double minFlow, double maxFlow, double minDistance, double maxDistance, double minCost, double maxCost, bool integerValues = false);

    QuadraticAssignmentProblem(const std::string& filename);

    void setData(unsigned int objects, const std::vector<std::vector<double> >& flows, const std::vector<std::vector<double> >& distances, const std::vector<std::vector<double> >& costs);

    void generateData(unsigned int objects, double minFlow, double maxFlow, double minDistance, double maxDistance, double minCost, double maxCost, bool integerValues = false);

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

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

    void showData();

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

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

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

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

    const std::vector<std::vector<double> >& getCosts() const {
        return costs;
    }
    
    const std::vector<std::vector<double> >& getDistances() const {
        return distances;
    }
    
    const std::vector<std::vector<double> >& getFlows() const {
        return flows;
    }
    
    unsigned int getObjects() const {
        return objects;
    }
};

} // namespace operational_research

#endif // QUADRATICASSIGNMENTPROBLEM_H