#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', 0.15));
   datos.push_back(make_pair('b', 0.30));
   datos.push_back(make_pair('c', 0.20));
   datos.push_back(make_pair('d', 0.05));
   datos.push_back(make_pair('e', 0.15));
   datos.push_back(make_pair('f', 0.05));
   datos.push_back(make_pair('g', 0.10));
      
   Huffman miCodificador(datos);
	 
   vector<string> mensajes = {"a", "b", "c", "d", "e", "f", "g", "baba", "edad", "cagada"};
	 
   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;
   }

}