#include <iostream>
#include <string>
#include <vector>
#include <utility> // pair
#include <cstdlib> // rand

using namespace std;

#include "Huffman.h"

int main() {
      
   vector< pair<char, float> > datos;
      
   datos.push_back(make_pair('a', 80));
   datos.push_back(make_pair('b', 30));
   datos.push_back(make_pair('c', 40));
   datos.push_back(make_pair('d', 20));
   datos.push_back(make_pair('e', 70));
   datos.push_back(make_pair('f', 60));
   datos.push_back(make_pair('g', 10));
   datos.push_back(make_pair('h', 50));
      
   Huffman miCodificador(datos);
	 
   vector<string> mensajes = {"a", "b", "c", "d", "e", "f", "g", "h", "baba", "edad", "cagada", "bache"};
	 
   for (string mensaje : mensajes) {
      string mensajeCodificado   = miCodificador.codificar(mensaje);
      string mensajeDecodificado = miCodificador.decodificar(mensajeCodificado);
      if (mensajeDecodificado != mensaje)
	 cout << "MAL: ";
      else
	 cout << "BIEN: ";
      cout << mensaje << "--->" << mensajeCodificado << "--->" << mensajeDecodificado << endl;
   }

}