#include <iostream>
#include <cstdlib>

#include <TCanvas.h>
#include <TROOT.h>
#include <TApplication.h>

#include "Garfield/MediumMagboltz.hh"
#include "Garfield/ViewMedium.hh"

using namespace Garfield;

int main(int argc, char * argv[]) {

  TApplication app("app", &argc, argv);
 
  // Setup the gas.
  MediumMagboltz gas;
  gas.LoadGasFile("C3H8_1Torr_313K_0.1_to_1800.gas");
  gas.LoadIonMobility("IonMobility_C3H8+_C3H8.txt");
  gas.PrintGas();

  ViewMedium view;
  view.SetMedium(&gas);
  view.SetMagneticField(0.);
  
  TCanvas cV("cV", "", 600, 600);
  view.SetCanvas(&cV);
  view.PlotElectronVelocity();

  TCanvas cD("cD", "", 600, 600);
  view.SetCanvas(&cD);
  view.PlotElectronDiffusion();

  TCanvas cI("cI", "", 600, 600);
  view.SetCanvas(&cI);
  view.PlotIonVelocity();

  TCanvas cID("cID", "", 600, 600);
  view.SetCanvas(&cID);
  view.PlotIonDiffusion();

  app.Run(true);

}