/* Last edited on 2009-07-24 14:22:22 by stolfilocal */
/***************************************************************************
 *   Copyright (C) 2009 by Douglas Castro   *
 *   douglas@ime.unicamp.br   *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/


#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include "arvore.h"
#include "bool.h"
#include "definicoes.h"
#include "inicializa.h"
#include "imprime.h"
#include "timestep.h"
#include "interp_integ.h"
#include "gnuplot_i.h"

#define SLEEP_LGTH  1
#define SLEEP_LGTH1  2

#define OUT_PREFIX "out"

int main(int argc, char *argv[])
{
  int d = 1;
  int niv = 7, profund = 0, indice = 1;
  double eps = 0.0;
  double xmin[] = {-1.0};
  double xmax[] = {1.0};
  
//   DoEPSTests(OUT_PREFIX);
  
  /* velocidade */
  double c = 1.0;
  
  /* niv diz quantos niveis de resolucao tem a arvore mas nao 
     que no ultimo nivel  tenha 2^niv celulas, corrigindo... */
  /* Nivel de resolucao */ 
  int L = niv+1;/* 7, 8, 9, 10 */
  
  Funcao *f = &condicaoinicial;
  /* inicializacao */
  int tm = pow(5,d);
  double fval[tm];
  int kx, ky=0;
  for (kx = 0; kx < 5; kx++)
  { 
    double xlo[] = {xmin[0] + (-2.0 + kx)*(xmax[0] - xmin[0])};
    double xhi[] = {xmin[0] + (-1.0 + kx)*(xmax[0] - xmin[0])};
    double fint = integra(d, xlo, xhi, condicaoinicial);
    fval[5*ky + kx] = fint/((xhi[0]-xlo[0]));
  }
  No *u = malloc(sizeof(No));
  u->fil[0] = NULL;
  u->fil[1] = NULL;
  u->deletable = FALSE;u->leaf = FALSE; u->virtual_leaf = FALSE;u->fval = fval[(int)tm/2];
//   No *u = completa(d, L, profund, indice, xmin, xmax, f, integra);
  
//   No *v = completa(d, niv, profund, indice, xmin, xmax, NULL, integra);
//   No *pc[] = {NULL,NULL, u, NULL, NULL};
//   No *pc[] = {v, v, u, v, v}; // roubando pra podar os galohs da fronteira
  /* COMO SERA UMA SOLUCAO INTELIGENTE PRA IMPLEMENTAR A CONDICAO DE CONTORNO? */
  No *pc[] = {NULL, u, u, u, NULL}; // teste pra ver se da a condicao de contorno periodica
  VNo pac[5];
  int i;
  for(i=0;i<5;i++)
  {
    pac[i].p = pc[i];
    pac[i].fv = fval[i];
  }
  int lim;
  double esq[] = {1.0, 1.0}; // so vai interferir p profund >= 4
  double dir[] = {1.0, 1.0};
  printf("podando\n");
//   poda_arvore(d, L, pac, xmin, xmax, profund, interpola_quad, /*fpoda*/ 1.0, eps);

  for(lim=1;lim<L;lim++)
    inicia2(d, pac, xmin, xmax, profund, interpola_quad, L, lim, condicaoinicial, integra, eps);

//   DoMesh(OUT_PREFIX, u);
//   MakeTree(OUT_PREFIX, u);
  //   return EXIT_SUCCESS;
  /* solucao analitica */
  int conta = 0;
  conta_folhas(u, &conta);
  /* vetor das slucoes(tempo n, old, new, analitica) */
  double uu[conta], uo[conta], un[conta], ua[conta];
  int pos = 0;
  copia_folhas(u, ua, &pos);

  /* inicia gnuplot */
  gnuplot_ctrl    *   h1;
  h1 = gnuplot_init() ;
  /* plota sol inicial */
  double xp = xmin[0];
  FILE *disco = fopen("solucao.txt", "w");
    escrevearquivo1d(disco, u, profund, &xp);
  fclose(disco);
  printf("controle...t=0\n");
  gnuplot_resetplot(h1) ;
  gnuplot_cmd(h1, "set xrange [-1:1]");
  gnuplot_cmd(h1, "set yrange [-0.2:1.2]");
  gnuplot_cmd(h1, "plot \"solucao.txt\" using 1:2 with lines") ;
  sleep(SLEEP_LGTH) ;
  /* evolucao no tempo */
  int n;
  /* calculo de Dt, Dx e tempo final N */
  /* CFL = 0.5 = c*Dt/Dx */
  int tam = pow(2,L);
  double Dx = 2.0/tam;
  double Dt = 0.5*Dx/c;
  int N = 1/*2/Dt*/;
  double fpoda = 1.0;
  for(n=1;n<N;n++)
  {
    /* primeiro passo de Runge-Kutta 2 */
    avancaTempoMeio( d, profund, pac, xmin, xmax, Dt, c, interpola_quad);

    /* segundo passo de Runge-Kutta */
    avancaTempoUm( d, profund, pac, xmin, xmax, Dt, c, interpola_quad);

    /* Remesh */
    printf("n=%d\n",n);
    atualiza_folhas_virtuais(d, profund, pac, interpola_quad);
    atualiza_arvore(u);
    
//     marca_nos( d, pac, niv, profund, interpola_quad, fpoda, eps);
//     teste(u,0);
//     adapta_arvore( d, pac, xmin, xmax, profund, L, interpola_quad, eps);
    adapta_arvore2( d, pac, xmin, xmax, profund, L, interpola_quad,1.0, eps);
    
    if(n%20==0)
    {
      /* plot */
      xp = -1.0; // LEMBRETE ESSE VALOR E REFERENTE A 1D
      FILE *disco = fopen("solucao.txt", "w");
        escrevearquivo1d(disco, u, profund, &xp);
      fclose(disco);
      printf("controle...n=%d\n",n);
      gnuplot_resetplot(h1) ;
      gnuplot_cmd(h1, "set xrange [-1:1]");
      gnuplot_cmd(h1, "set yrange [-0.2:1.2]");
      gnuplot_cmd(h1, "plot \"solucao.txt\" using 1:2 with lines") ;
      sleep(SLEEP_LGTH) ;
    }
  }
//   poda_arvore(d, L, pac, xmin, xmax, profund, interpola_quad, /*fpoda*/ 1.0, 0.1);
  gnuplot_close(h1) ;
  MakeTree(OUT_PREFIX, u, L);
  /* calcular erros norma L_infty e L_1 */
//   printf("Erro L_infty = %f\n", errmax( u, ua, L));
//   printf("Erro L_1 = %f\n", errmid( u, ua, L));

  /* plota sol final */
  /* inicia gnuplot */
  gnuplot_ctrl    *   h2;
  h2 = gnuplot_init() ;
  gnuplot_resetplot(h2) ;
  gnuplot_cmd(h2, "set xrange [-1:1]");
  gnuplot_cmd(h2, "set yrange [-0.2:1.2]");
  gnuplot_cmd(h2, "plot \"solucao.txt\" using 1:2 with lines") ;
  sleep(SLEEP_LGTH1) ;
  gnuplot_close(h2) ;

  printf("Hello, world!\n");

  return EXIT_SUCCESS;
}