/*
<ubqpval.h>
Copyright (C) DOLPHIN Project-Team, INRIA Lille - Nord Europe, 2006-2010

Sebastien Verel

This software is governed by the CeCILL license under French law and
abiding by the rules of distribution of free software.  You can  ue,
modify and/ or redistribute the software under the terms of the CeCILL
license as circulated by CEA, CNRS and INRIA at the following URL
"http://www.cecill.info".

In this respect, the user's attention is drawn to the risks associated
with loading,  using,  modifying and/or developing or reproducing the
software by the user in light of its specific status of free software,
that may mean  that it is complicated to manipulate,  and  that  also
therefore means  that it is reserved for developers  and  experienced
professionals having in-depth computer knowledge. Users are therefore
encouraged to load and test the software's suitability as regards their
requirements in conditions enabling the security of their systems and/or
data to be ensured and,  more generally, to use and operate it in the
same conditions as regards security.
The fact that you are presently reading this means that you have had
knowledge of the CeCILL license and that you accept its terms.

ParadisEO WebSite : http://paradiseo.gforge.inria.fr
Contact: paradiseo-help@lists.gforge.inria.fr
*/

#ifndef _ubqpEval_h
#define _ubqpEval_h

#include <vector>
#include <eoEvalFunc.h>

/**
 * Full evaluation Function for unconstrainted binary quadratic programming problem
 */
template< class EOT >
class UbqpEval : public eoEvalFunc<EOT>
{
public:

  /**
   * Constructor
   * instance is given in the ORLIB format: 
   * The format of these data files is:
   * number of test problem in the serie
   * for each test problem in turn:
   *    number of variables (n), number of non-zero elements in the q(i,j) matrix
   *       for each non-zero element in turn: 
   *       i, j, q(i,j) {=q(j,i) as the matrix is symmetric}
   *
   * @param _fileName file name of the instance in ORLIB format
   * @param _numInstance the number of the given instance to solve
   */
  UbqpEval(std::string & _fileName, unsigned int _numInstance = 0) {
    std::fstream file(_fileName.c_str(), std::ios::in);

    if (!file) {
      std::string str = "UbqpEval: Could not open file [" + _fileName + "]." ;
      throw std::runtime_error(str);
    }

    unsigned int nbInstances;
    file >> nbInstances;

    unsigned int i, j;
    int v;

    for(unsigned k = 0; k < _numInstance; k++) {
      file >> nbVar >> nbNonZero;
      for(unsigned kk = 0; kk < nbNonZero; kk++)
	file >> i >> j >> v;
    }

    // the chosen instance
    file >> nbVar >> nbNonZero;

    // creation of the matrix
    Q = new int*[nbVar];

    for(unsigned int i = 0; i < nbVar; i++) {
      Q[i] = new int[nbVar];
      for(unsigned int j = 0; j < nbVar; j++)
	Q[i][j] = 0;
    }

    // read the matrix
    for(unsigned int k = 0; k < nbNonZero; k++) {
      file >> i >> j >> v;
      Q[i][j] = v;
    }
    
    file.close();

    // put the matrix in lower triangular form
    for(unsigned i = 1; i < nbVar; i++)
      for(unsigned int j = 0; j < i; j++) {
	Q[i][j] = Q[i][j] + Q[j][i];
	Q[j][i] = 0;
      }
  }

  /**
   * Destructor
   */
  ~UbqpEval() {
    if (Q != NULL) {
      for(unsigned i = 0; i < nbVar; i++)
	delete[] Q[i];

      // delete the matrix
      delete[] Q;
    }
  }

  /**
   * fitness evaluation of the solution
   *
   * @param _solution the solution to evaluation
   */
  virtual void operator()(EOT & _solution) {
    int fit = 0;
    unsigned int j;

    for(unsigned i = 0; i < nbVar; i++)
      if (_solution[i] == 1) 
	for(j = 0; j < i; j++)
	  if (_solution[j] == 1) 
	    fit += Q[i][j];

    _solution.fitness(fit);
  }

  /*
   * to get the matrix Q
   *
   * @return matrix Q
   */
  int** getQ() {
    return Q;
  }

  /*
   * to get the number of variable (bit string length)
   *
   * @return bit string length
   */
  int getNbVar() {
    return nbVar;
  }

private:
  /**
   * variables (used in incremental evaluation)
   */

  // number of variable
  unsigned int nbVar; 

  // number of non zero in the matrix
  unsigned int nbNonZero;
  
  // matrix of flux:
  //   the matrix is put in lower triangular form: for i<j Q[i][j] = 0
  int ** Q;
};

#endif