opencv/samples/cpp/travelsalesman.cpp

#include <opencv2/opencv.hpp>

using namespace cv;

class TravelSalesman : public ml::SimulatedAnnealingSolverSystem
{
private :
    const std::vector<Point>& posCity;
    std::vector<int>& next;
    RNG rng;
    int d0,d1,d2,d3;

public:
    TravelSalesman(std::vector<Point> &p, std::vector<int> &n) :
        posCity(p), next(n)
    {
        rng = theRNG();
    }
    /** Give energy value for  a state of system.*/
    /*virtual*/ double energy() const;
    /** Function which change the state of system (random pertubation).*/
    /*virtual*/ void changeState();
    /** Function to reverse to the previous state.*/
    /*virtual*/ void reverseState();

};

void TravelSalesman::changeState()
{
    d0 = rng.uniform(0,static_cast<int>(posCity.size()));
    d1 = next[d0];
    d2 = next[d1];
    d3 = next[d2];

    next[d0] = d2;
    next[d2] = d1;
    next[d1] = d3;
}


void TravelSalesman::reverseState()
{
    next[d0] = d1;
    next[d1] = d2;
    next[d2] = d3;
}

double TravelSalesman::energy() const
{
    double e = 0;
    for (size_t i = 0; i < next.size(); i++)
    {
        e += norm(posCity[i]-posCity[next[i]]);
    }
    return e;
}


static void DrawTravelMap(Mat &img, std::vector<Point> &p, std::vector<int> &n)
{
    for (size_t i = 0; i < n.size(); i++)
    {
        circle(img,p[i],5,Scalar(0,0,255),2);
        line(img,p[i],p[n[i]],Scalar(0,255,0),2);
    }
}
int main(void)
{
    int nbCity=40;
    Mat img(500,500,CV_8UC3,Scalar::all(0));
    RNG rng(123456);
    int radius=static_cast<int>(img.cols*0.45);
    Point center(img.cols/2,img.rows/2);

    std::vector<Point> posCity(nbCity);
    std::vector<int> next(nbCity);
    for (size_t i = 0; i < posCity.size(); i++)
    {
        double theta = rng.uniform(0., 2 * CV_PI);
        posCity[i].x = static_cast<int>(radius*cos(theta)) + center.x;
        posCity[i].y = static_cast<int>(radius*sin(theta)) + center.y;
        next[i]=(i+1)%nbCity;
    }
    Ptr<TravelSalesman> ts_system(new TravelSalesman(posCity, next));
    ml::SimulatedAnnealingSolver ts(ts_system);

    ts.setIterPerStep(10000*nbCity);
    ts.setCoolingRatio(0.99);
    ts.setInitialTemperature(100);
    ts.setFinalTemperature(100*0.97);
    DrawTravelMap(img,posCity,next);
    imshow("Map",img);
    waitKey(10);
    for (int i = 0, zeroChanges = 0; zeroChanges < 10; i++)
    {
        int changesApplied = ts.run();
        img = Mat::zeros(img.size(),CV_8UC3);
        DrawTravelMap(img, posCity, next);
        imshow("Map", img);
        int k = waitKey(10);
        double ti=ts.getFinalTemperature();
        std::cout << "i=" << i << " changesApplied=" << changesApplied << " temp=" << ti << " result=" << ts_system->energy() << std::endl;
        if (k == 27 || k == 'q' || k == 'Q')
            return 0;
        if (changesApplied == 0)
            zeroChanges++;
        ts.setInitialTemperature(ti);
        ts.setFinalTemperature(ti*0.97);
    }
    std::cout << "Done" << std::endl;
    waitKey(0);
    return 0;
}