import ChemPlugin.*;

import java.io.*;
import java.util.Scanner;

public class HeatTransfer1 {

   public static void main(String[] args) throws IOException {

      System.out.println("Model heat transfer in one dimension");
      System.out.println("");

      //Simulation parameters
      int nx = 400; // number of instances along x
      double length = 100; // m
      double deltax = length / nx; // m
      double deltay= 1.0, deltaz = 1.0;
      double porosity = 0.25;  // volume fraction

      double tcond = 2.0; // W/m/K
      double trans = deltay * deltaz * tcond / deltax; // W/K

      System.out.print("Please enter fluid velocity in m/yr: ");
      Scanner input = new Scanner(System.in);
      double veloc_in = input.nextDouble();
      input.nextLine();

      double velocity = veloc_in / 31557600;  // m/s
      double flow = deltay * deltaz * porosity * velocity; // m3/s

      double time_end = 10.0; // years
      double delta_years = time_end / 5.0; // years
      double next_output = 0.0; //years

      // Open output file and write instance positions on the first line
      String file = "HeatTransfer.txt";
      FileWriter fileWriter = new FileWriter(file);
      BufferedWriter bufferedWriter =  new BufferedWriter(fileWriter);
      bufferedWriter.write("years");
      for (int i=0; i<nx; i++){
         bufferedWriter.write(String.format("\t%8.4f", (i+0.5) * deltax));
      }
      bufferedWriter.newLine();

      // Configure and initialize the inlet instance
      ChemPlugin cp_inlet = new ChemPlugin();
      cp_inlet.Config("T=100 C; Na+ = 0.001 mmol/kg; Cl- = 0.001 mmol/kg");
      if (cp_inlet.Initialize() != 0)
         System.out.println("Inlet failed to initialize");

      // configure and initialize the interior instances

      String cmd = "T = 20 C; span 20 C to 100 C; "
	     + "Na+ = 0.001 mmol/kg; Cl- = 0.001 mmol/kg; "
		 + "volume = "  + (deltax * deltay * deltaz) + " m3; "
		 + "porosity = " + porosity + "; time end = " + time_end + " years;";


      ChemPlugin cp[] = new ChemPlugin[nx];
      for (int i=0; i<nx; i++){
         cp[i] = new ChemPlugin();
         cp[i].Config(cmd);
      }

      cp[0].Console("stdout");
      cp[0].Config("pluses = banner");

      for (int i=0; i<nx; i++){
         if (cp[i].Initialize() != 0)
            System.out.println("Interior Instance" + i + "failed to initialize");
      }

      next_output = write_line(bufferedWriter, cp, nx, delta_years, next_output);

      // Link the instances
      CpiLink link = cp[0].Link(cp_inlet);
      link.HeatTrans(trans, "W/K");
      link.FlowRate(flow, "m3/s");

      for (int i=1; i<nx; i++){
         link = cp[i].Link(cp[i-1]);
         link.HeatTrans(trans, "W/K");
         link.FlowRate(flow, "m3/s");
      }

      link = cp[nx-1].Link();
      link.FlowRate(-flow, "m3/s");

      double deltat;
      while (true) {
         deltat = 1e99;
         for(int i=0; i<nx; i++)
            deltat = Math.min(deltat, cp[i].ReportTimeStep());

         for(int i=0; i<nx; i++){
            if (cp[i].AdvanceTimeStep(deltat) != 0){
               bufferedWriter.close();
               String pause = input.nextLine();
               System.exit(0);
            }
         }

         for(int i=0; i<nx; i++){
            if (cp[i].AdvanceHeatTransport() != 0){
               System.exit(1);
            }
         }

         for(int i=0; i<nx; i++){
            if (cp[i].AdvanceChemical() != 0){
               System.exit(1);
            }
         }
         next_output = write_line(bufferedWriter, cp, nx, delta_years, next_output);
      } 

   }

   public static double write_line(BufferedWriter bufferedWriter, ChemPlugin cp[],
         int nx, double gap, double then ) throws IOException
   {
      double now = cp[0].Report1("Time", "years");
      if ((then - now) <= gap / 1e4) {
         bufferedWriter.write(String.format("%8.4f",now));
         for (int i =0;i<nx;i++){
            bufferedWriter.write(String.format("\t%8.4f", cp[i].Report1("temperature", "C")));
         }
         then += gap;
         bufferedWriter.newLine();
      }
      return then;

   }
}