#include <vector>
#include <iostream>
#include <cmath>

#include <TApplication.h>
#include <TCanvas.h>
#include <TGraph.h>
#include <TLegend.h>

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

using namespace Garfield;

int main(int argc, char* argv[]) {
  TApplication app("app", &argc, argv);

  MediumMagboltz gas1;
  gas1.LoadGasFile("customP10_testrun.gas");
  gas1.SetPressure(760);
  gas1.SetTemperature(295);

  MediumMagboltz gas2;
  gas2.LoadGasFile("ar_90_ch4_10.gas");
  gas2.SetPressure(760);
  gas2.SetTemperature(295);

  //MediumMagboltz gas3;
  //gas3.LoadGasFile("customP10_fromgasfile.gas");
  //gas3.SetPressure(760);
  //gas3.SetTemperature(295);

    
  ViewMedium view;
  TCanvas c1;
  view.SetCanvas(&c1);
  view.SetMedium(&gas1);
  view.PlotElectronVelocity('e');
  std::vector<std::string> labels = {"P10 Bolsig+","P10 magboltz"};
  view.SetLabels(labels);
  view.SetMedium(&gas2);
  view.PlotElectronVelocity('e',true);
  //view.SetMedium(&gas3);
  //view.PlotElectronVelocity('e',true);




  app.Run();
  return 0;
}