/* -*- mode: C++; indent-tabs-mode: nil; -*-

 *

 * This file is a part of LEMON, a generic C++ optimization library.

 *

 * Copyright (C) 2015-2017

 * EMAXA Kutato-fejleszto Kft. (EMAXA Research Ltd.)

 *

 * Permission to use, modify and distribute this software is granted

 * provided that this copyright notice appears in all copies. For

 * precise terms see the accompanying LICENSE file.

 *

 * This software is provided "AS IS" with no warranty of any kind,

 * express or implied, and with no claim as to its suitability for any

 * purpose.

 *

 */

#include <lemon/vf2.h>

#include <lemon/concepts/digraph.h>

#include <lemon/smart_graph.h>

#include <lemon/lgf_reader.h>

#include <lemon/concepts/maps.h>

#include <test/test_tools.h>

#include <sstream>

using namespace lemon;

char petersen_lgf[] =

  "@nodes\n"

  "label col1 col2\n"

  "0 1 1\n"

  "1 1 2\n"

  "2 1 3\n"

  "3 1 4\n"

  "4 2 5\n"

  "5 2 1\n"

  "6 2 2\n"

  "7 2 3\n"

  "8 2 4\n"

  "9 2 5\n"

  "@arcs\n"

  "     -\n"

  "0 1\n"

  "1 2\n"

  "2 3\n"

  "3 4\n"

  "4 0\n"

  "0 5\n"

  "1 6\n"

  "2 7\n"

  "3 8\n"

  "4 9\n"

  "5 8\n"

  "5 7\n"

  "9 6\n"

  "9 7\n"

  "6 8\n";

char c5_lgf[] =

  "@nodes\n"

  "label col\n"

  "0 1\n"

  "1 2\n"

  "2 3\n"

  "3 4\n"

  "4 5\n"

  "@arcs\n"

  "     -\n"

  "0 1\n"

  "1 2\n"

  "2 3\n"

  "3 4\n"

  "4 0\n";

char c7_lgf[] =

  "@nodes\n"

  "label\n"

  "0\n"

  "1\n"

  "2\n"

  "3\n"

  "4\n"

  "5\n"

  "6\n"

  "@arcs\n"

  "     -\n"

  "0 1\n"

  "1 2\n"

  "2 3\n"

  "3 4\n"

  "4 5\n"

  "5 6\n"

  "6 0\n";

char c10_lgf[] =

  "@nodes\n"

  "label\n"

  "0\n"

  "1\n"

  "2\n"

  "3\n"

  "4\n"

  "5\n"

  "6\n"

  "7\n"

  "8\n"

  "9\n"

  "@arcs\n"

  "     -\n"

  "0 1\n"

  "1 2\n"

  "2 3\n"

  "3 4\n"

  "4 5\n"

  "5 6\n"

  "6 7\n"

  "7 8\n"

  "8 9\n"

  "9 0\n";

char p10_lgf[] =

  "@nodes\n"

  "label\n"

  "0\n"

  "1\n"

  "2\n"

  "3\n"

  "4\n"

  "5\n"

  "6\n"

  "7\n"

  "8\n"

  "9\n"

  "@arcs\n"

  "     -\n"

  "0 1\n"

  "1 2\n"

  "2 3\n"

  "3 4\n"

  "4 5\n"

  "5 6\n"

  "6 7\n"

  "7 8\n"

  "8 9\n";

SmartGraph petersen, c5, c7, c10, p10;

SmartGraph::NodeMap<int> petersen_col1(petersen);

SmartGraph::NodeMap<int> petersen_col2(petersen);

SmartGraph::NodeMap<int> c5_col(c5);

void make_graphs() {

  std::stringstream ss(petersen_lgf);

  graphReader(petersen, ss)

    .nodeMap("col1",petersen_col1)

    .nodeMap("col2",petersen_col2)

    .run();

  ss.clear();

  ss.str("");

  ss<<c5_lgf;

  //std::stringstream ss2(c5_lgf);

  graphReader(c5, ss)

    .nodeMap("col",c5_col)

    .run();

  ss.clear();

  ss.str("");

  ss<<c7_lgf;

  graphReader(c7, ss).run();

  ss.clear();

  ss.str("");

  ss<<c10_lgf;

  graphReader(c10, ss).run();

  ss.clear();

  ss.str("");

  ss<<p10_lgf;

  graphReader(p10, ss).run();

}

class EqComparable {

public:

  bool operator==(const EqComparable&) {

    return false;

  }

};

template<class A, class B>

class EqClass {

public:

  bool operator()(A, B){

    return false;

  }

};

template<class G1,class G2>

void checkVf2Compile() {

  G1 g;

  G2 h;

  concepts::ReadWriteMap<typename G1::Node, typename G2::Node> r;

  bool succ;

  ::lemon::ignore_unused_variable_warning(succ);

  succ = vf2(g,h).run();

  succ = vf2(g,h).induced().run();

  succ = vf2(g,h).iso().run();

  succ = vf2(g,h).mapping(r).run();

  Vf2<G1,G2,concepts::ReadWriteMap<typename G1::Node, typename G2::Node>,

      EqClass<typename G1::Node,typename G2::Node> >

    myVf2(g,h,r,EqClass<typename G1::Node,typename G2::Node>());

  myVf2.find();

  succ = vf2(g,h).induced().mapping(r).run();

  succ = vf2(g,h).iso().mapping(r).run();

  concepts::ReadMap<typename G1::Node, EqComparable> l1;

  concepts::ReadMap<typename G2::Node, EqComparable> l2;

  succ = vf2(g,h).nodeLabels(l1,l2).mapping(r).run();

  succ = vf2(g,h).nodeEq(EqClass<typename G1::Node,typename G2::Node>())

    .mapping(r).run();

}

void justCompile() {

  checkVf2Compile<concepts::Graph,concepts::Graph>();

  checkVf2Compile<concepts::Graph,SmartGraph>();

  checkVf2Compile<SmartGraph,concepts::Graph>();

}

template<class G1, class G2, class I>

void checkSub(const G1 &g1, const G2 &g2, const I &i) {

  {

    std::set<typename G2::Node> image;

    for(typename G1::NodeIt n(g1);n!=INVALID;++n){

      check(i[n]!=INVALID, "Wrong isomorphism: incomplete mapping.");

      check(image.count(i[n])==0,"Wrong isomorphism: not injective.");

      image.insert(i[n]);

    }

  }

  for(typename G1::EdgeIt e(g1);e!=INVALID;++e)

    check(findEdge(g2,i[g1.u(e)],i[g1.v(e)])!=INVALID,

          "Wrong isomorphism: missing edge(checkSub).");

}

template<class G1, class G2, class I>

void checkInd(const G1 &g1, const G2 &g2, const I &i) {

  std::set<typename G2::Node> image;

  for(typename G1::NodeIt n(g1);n!=INVALID;++n) {

    check(i[n]!=INVALID, "Wrong isomorphism: incomplete mapping.");

    check(image.count(i[n])==0,"Wrong isomorphism: not injective.");

    image.insert(i[n]);

  }

  for(typename G1::NodeIt n(g1); n!=INVALID; ++n)

    for(typename G1::NodeIt m(g1); m!=INVALID; ++m)

      if((findEdge(g1,n,m)==INVALID) != (findEdge(g2,i[n],i[m])==INVALID)) {

        std::cout << "Wrong isomorphism: edge mismatch";

        exit(1);

      }

}

template<class G1,class G2>

int checkSub(const G1 &g1, const G2 &g2) {

  typename G1:: template NodeMap<typename G2::Node> iso(g1,INVALID);

  if(vf2(g1,g2).mapping(iso).run()) {

    checkSub(g1,g2,iso);

    return true;

  }

  else

    return false;

}

template<class G1,class G2>

int checkInd(const G1 &g1, const G2 &g2) {

  typename G1:: template NodeMap<typename G2::Node> iso(g1,INVALID);

  if(vf2(g1,g2).induced().mapping(iso).run()) {

    checkInd(g1,g2,iso);

    return true;

  }

  else

    return false;

}

template<class G1,class G2>

int checkIso(const G1 &g1, const G2 &g2) {

  typename G1:: template NodeMap<typename G2::Node> iso(g1,INVALID);

  if(vf2(g1,g2).iso().mapping(iso).run()) {

    check(countNodes(g1)==countNodes(g2),

          "Wrong iso alg.: they are not isomophic.");

    checkInd(g1,g2,iso);

    return true;

  }

  else

    return false;

}

template<class G1, class G2, class L1, class L2, class I>

void checkLabel(const G1 &g1, const G2 &,

                const L1 &l1, const L2 &l2,const I &i) {

  for(typename G1::NodeIt n(g1);n!=INVALID;++n)

    check(l1[n]==l2[i[n]],"Wrong isomorphism: label mismatch.");

}

template<class G1,class G2,class L1,class L2>

int checkSub(const G1 &g1, const G2 &g2, const L1 &l1, const L2 &l2) {

  typename G1:: template NodeMap<typename G2::Node> iso(g1,INVALID);

  if(vf2(g1,g2).nodeLabels(l1,l2).mapping(iso).run()){

    checkSub(g1,g2,iso);

    checkLabel(g1,g2,l1,l2,iso);

    return true;

  }

  else

    return false;

}

int main() {

  make_graphs();

  //   justCompile();

  check(checkSub(c5,petersen), "There should exist a C5->Petersen mapping.");

  check(!checkSub(c7,petersen),

        "There should not exist a C7->Petersen mapping.");

  check(checkSub(p10,petersen), "There should exist a P10->Petersen mapping.");

  check(!checkSub(c10,petersen),

        "There should not exist a C10->Petersen mapping.");

  check(checkSub(petersen,petersen),

        "There should exist a Petersen->Petersen mapping.");

  check(checkInd(c5,petersen),

        "There should exist a C5->Petersen spanned mapping.");

  check(!checkInd(c7,petersen),

        "There should exist a C7->Petersen spanned mapping.");

  check(!checkInd(p10,petersen),

        "There should not exist a P10->Petersen spanned mapping.");

  check(!checkInd(c10,petersen),

        "There should not exist a C10->Petersen spanned mapping.");

  check(checkInd(petersen,petersen),

        "There should exist a Petersen->Petersen spanned mapping.");

  check(!checkSub(petersen,c10),

        "There should not exist a Petersen->C10 mapping.");

  check(checkSub(p10,c10),

        "There should exist a P10->C10 mapping.");

  check(!checkInd(p10,c10),

        "There should not exist a P10->C10 spanned mapping.");

  check(!checkSub(c10,p10),

        "There should not exist a C10->P10 mapping.");

  check(!checkIso(p10,c10),

        "P10 and C10 are not isomorphic.");

  check(checkIso(c10,c10),

        "C10 and C10 are isomorphic.");

  check(!vf2(p10,c10).iso().run(),

        "P10 and C10 are not isomorphic.");

  check(vf2(c10,c10).iso().run(),

        "C10 and C10 are isomorphic.");

  check(!checkSub(c5,petersen,c5_col,petersen_col1),

        "There should exist a C5->Petersen mapping.");

  check(checkSub(c5,petersen,c5_col,petersen_col2),

        "There should exist a C5->Petersen mapping.");

}