/* This file is part of Lemma, a geophysical modelling and inversion API.
 * More information is available at http://lemmasoftware.org
 */

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

/**
 * @file
 * @date      02/01/19 22:34:11
 * @version   $Id$
 * @author    Trevor Irons (ti)
 * @email     Trevor.Irons@utah.edu
 * @copyright Copyright (c) 2019, University of Utah
 * @copyright Copyright (c) 2019, Lemma Software, LLC
 */

#include <cxxtest/TestSuite.h>
#include <FDEM1D>
#include <random>
#include "timer.h"

#define EPSILON 1e-10
using namespace Lemma;

class MyAlgorithmTest : public CxxTest::TestSuite {
   // Put in your usual, quick unit tests here
};

class MyAlgorithmTestPerformance : public CxxTest::TestSuite {

public:

    std::shared_ptr< DipoleSource > dipole;
    std::shared_ptr< LayeredEarthEM > earth;
    std::shared_ptr< FieldPoints > receivers;
    std::shared_ptr< EMEarth1D > EmEarth;

	jsw_timer timer;

    Real Delta;

    void setUp() {
        // Put in any code needed to set up your test,
        // but that should NOT be counted in the execution time.
	    dipole = DipoleSource::NewSP();

        //dipole->SetType(GROUNDEDELECTRICDIPOLE);
		//dipole->SetPolarisation(XPOLARISATION);


		dipole->SetNumberOfFrequencies(1);
		dipole->SetMoment(1);
		dipole->SetFrequency(0, 4400.1000);
		//dipole->SetPhase(0);
		//dipole->SetLocation( (VectorXr(3) << 49, 49, -1e-4).finished() );
		dipole->SetLocation( 0, 0, -1e-1  );

	    // Define model
	    VectorXcr sigma(8);
		sigma << 0., 1e-2, .1, .01, .001, .1, .05, .2;
	    VectorXr  thick(6);
		thick << 10, 10, 10, 10, 10;

	    earth = LayeredEarthEM::NewSP();
        earth->SetNumberOfLayers(8);
		earth->SetLayerConductivity(sigma);
		earth->SetLayerThickness(thick);

	    receivers = FieldPoints::NewSP();
		Vector3r loc;

        Real ox = 250.;
		Real oy = 250.;
		Real oz = -250.;

		Real dx = 20;
		Real dy = 20;
		Real dz = 20;

		int  nx = 11;  //13
		int  ny = 11;  //13
		int  nz = 11;  //13
        Delta = nx*ny*nz*1e-9;

		receivers->SetNumberOfPoints(nx*ny*nz);
		int ir = 0;
		for (int ix=0; ix<nx; ++ix) {
		for (int iy=0; iy<ny; ++iy) {
		for (int iz=0; iz<nz; ++iz) {
			loc << ox+ix*dx, oy+iy*dy, oz+iz*dz;
            //std::cout << "Receiver location " << loc.transpose() << std::endl;
			receivers->SetLocation(ir, loc);
			//oz += dz;
			++ ir;
		}
        }
        }

        EmEarth = EMEarth1D::NewSP();
        EmEarth->SetHankelTransformMethod(CHAVE);
        EmEarth->SetFieldsToCalculate(BOTH); // Fortran needs this
		EmEarth->AttachDipoleSource(dipole);
		EmEarth->AttachLayeredEarthEM(earth);
		EmEarth->AttachFieldPoints(receivers);

   }

   void tearDown() {
      // Clean-up code, also NOT counted in execution time.
   }

   void test_Hz() {

        std::cout.precision(4);

        dipole->SetType(MAGNETICDIPOLE);
		dipole->SetPolarisation(ZPOLARISATION);

        // Put in a unit test that will be slow.
        std::cout << "MAGNETICDIPOLE Z polarisation" << std::endl;
        std::cout << "=====================================\n";
	    std::cout << std::setw(18) << "Lemma/C++: ";
        std::cout.flush();

  	    timer.begin();
	    EmEarth->MakeCalc3();
	    Real lemmaTime = timer.end();
        std::cout << std::setw(14) << lemmaTime << std::setw(6) << " [s]" << std::endl;

        auto lc = receivers->GetEfield( 0 );

        #ifdef KIHALEE_EM1D
	    receivers->ClearFields();
        std::cout << std::setw(18) << "KiHa/Fortran: ";
        std::cout.flush();

  	    timer.begin();
 	    EmEarth->MakeCalc();
	    Real kihaTime = timer.end();
        std::cout << std::setw(14) << kihaTime << std::setw(6) << " [s]" << std::endl;

        auto fc = receivers->GetEfield( 0 ); //0,0);

        //std::cout.precision(16);
        std::cout << std::setw(18) << "Lemma norm: "      << std::setw(14) << (lc).norm() << std::endl;
        std::cout << std::setw(18) << "KiHa norm: "       << std::setw(14) <<  (fc).norm() << std::endl;
        std::cout << std::setw(18) << "Difference norm: " << std::setw(14) << (lc - fc).norm() << "\n";
        std::cout << std::setw(18) << "Speedup: "         << std::setw(14) << kihaTime/lemmaTime << "\n" << std::endl;

        TS_ASSERT_DELTA((lc-fc).norm(), 0.0, Delta);
        #endif
   }

   void test_Hx() {

        std::cout.precision(4);

        dipole->SetType(MAGNETICDIPOLE);
		dipole->SetPolarisation(XPOLARISATION);

        // Put in a unit test that will be slow.
        std::cout << "MAGNETICDIPOLE X polarisation" << std::endl;
        std::cout << "=====================================\n";
	    std::cout << std::setw(18) << "Lemma/C++: ";
        std::cout.flush();

  	    timer.begin();
	    EmEarth->MakeCalc3();
	    Real lemmaTime = timer.end();
        std::cout << std::setw(14) << lemmaTime << std::setw(6) << " [s]" << std::endl;

        auto lc = receivers->GetEfield( 0 );

        #ifdef KIHALEE_EM1D
	    receivers->ClearFields();
        std::cout << std::setw(18) << "KiHa/Fortran: ";
  	    timer.begin();
 	    EmEarth->MakeCalc();
	    Real kihaTime = timer.end();
        std::cout << std::setw(14) << kihaTime << std::setw(6) << " [s]" << std::endl;

        auto fc = receivers->GetEfield( 0 ); //0,0);

        //std::cout.precision(16);
        std::cout << std::setw(18) << "Lemma norm: "      << std::setw(14) << (lc).norm() << std::endl;
        std::cout << std::setw(18) << "KiHa norm: "       << std::setw(14) <<  (fc).norm() << std::endl;
        std::cout << std::setw(18) << "Difference norm: " << std::setw(14) << (lc - fc).norm() << "\n";
        std::cout << std::setw(18) << "Speedup: "         << std::setw(14) << kihaTime/lemmaTime << "\n" << std::endl;

        TS_ASSERT_DELTA((lc-fc).norm(), 0.0, Delta);
        #endif
   }

   void test_Hy() {

        std::cout.precision(4);

        dipole->SetType(MAGNETICDIPOLE);
		dipole->SetPolarisation(YPOLARISATION);

        // Put in a unit test that will be slow.
        std::cout << "MAGNETICDIPOLE Y polarisation" << std::endl;
        std::cout << "=====================================\n";
	    std::cout << std::setw(18) << "Lemma/C++: ";
        std::cout.flush();

  	    timer.begin();
	    EmEarth->MakeCalc3();
	    Real lemmaTime = timer.end();
        std::cout << std::setw(14) << lemmaTime << std::setw(6) << " [s]" << std::endl;

        auto lc = receivers->GetEfield( 0 );

        #ifdef KIHALEE_EM1D
	    receivers->ClearFields();
        std::cout << std::setw(18) << "KiHa/Fortran: ";
        std::cout.flush();
  	    timer.begin();
 	    EmEarth->MakeCalc();
	    Real kihaTime = timer.end();
        std::cout << std::setw(14) << kihaTime << std::setw(6) << " [s]" << std::endl;

        auto fc = receivers->GetEfield( 0 ); //0,0);

        //std::cout.precision(16);
        std::cout << std::setw(18) << "Lemma norm: "      << std::setw(14) << (lc).norm() << std::endl;
        std::cout << std::setw(18) << "KiHa norm: "       << std::setw(14) <<  (fc).norm() << std::endl;
        std::cout << std::setw(18) << "Difference norm: " << std::setw(14) << (lc - fc).norm() << "\n";
        std::cout << std::setw(18) << "Speedup: "         << std::setw(14) << kihaTime/lemmaTime << "\n" << std::endl;

        TS_ASSERT_DELTA((lc-fc).norm(), 0.0, Delta);
        #endif
   }

   void test_Ex() {

        std::cout.precision(4);

        dipole->SetType(GROUNDEDELECTRICDIPOLE);
		dipole->SetPolarisation(XPOLARISATION);

        // Put in a unit test that will be slow.
        std::cout << "GROUNDEDELECTRICDIPOLE X polarisation" << std::endl;
        std::cout << "=====================================\n";
	    std::cout << std::setw(18) << "Lemma/C++: ";
        std::cout.flush();

  	    timer.begin();
	    EmEarth->MakeCalc3();
	    Real lemmaTime = timer.end();
        std::cout << std::setw(14) << lemmaTime << std::setw(6) << " [s]" << std::endl;

        auto lc = receivers->GetEfield( 0 );

        #ifdef KIHALEE_EM1D
	    receivers->ClearFields();
        std::cout << std::setw(18) << "KiHa/Fortran: ";
        std::cout.flush();
  	    timer.begin();
 	    EmEarth->MakeCalc();
	    Real kihaTime = timer.end();
        std::cout << std::setw(14) << kihaTime << std::setw(6) << " [s]" << std::endl;

        auto fc = receivers->GetEfield( 0 ); //0,0);

        //std::cout.precision(16);
        std::cout << std::setw(18) << "Lemma norm: "      << std::setw(14) << (lc).norm() << std::endl;
        std::cout << std::setw(18) << "KiHa norm: "       << std::setw(14) <<  (fc).norm() << std::endl;
        std::cout << std::setw(18) << "Difference norm: " << std::setw(14) << (lc - fc).norm() << "\n";
        std::cout << std::setw(18) << "Speedup: "         << std::setw(14) << kihaTime/lemmaTime << "\n" << std::endl;

        TS_ASSERT_DELTA((lc-fc).norm(), 0.0, Delta);
        #endif
   }

   void test_Ey() {

        std::cout.precision(4);

        dipole->SetType(GROUNDEDELECTRICDIPOLE);
		dipole->SetPolarisation(YPOLARISATION);

        // Put in a unit test that will be slow.
        std::cout << "GROUNDEDELECTRICDIPOLE Y polarisation" << std::endl;
        std::cout << "=====================================\n";
	    std::cout << std::setw(18) << "Lemma/C++: ";
        std::cout.flush();

  	    timer.begin();
	    EmEarth->MakeCalc3();
	    Real lemmaTime = timer.end();
        std::cout << std::setw(14) << lemmaTime << std::setw(6) << " [s]" << std::endl;

        auto lc = receivers->GetEfield( 0 );

        #ifdef KIHALEE_EM1D
	    receivers->ClearFields();
        std::cout << std::setw(18) << "KiHa/Fortran: ";
        std::cout.flush();

        timer.begin();
 	    EmEarth->MakeCalc();
	    Real kihaTime = timer.end();
        std::cout << std::setw(14) << kihaTime << std::setw(6) << " [s]" << std::endl;

        auto fc = receivers->GetEfield( 0 ); //0,0);

        //std::cout.precision(16);
        std::cout << std::setw(18) << "Lemma norm: "      << std::setw(14) << (lc).norm() << std::endl;
        std::cout << std::setw(18) << "KiHa norm: "       << std::setw(14) <<  (fc).norm() << std::endl;
        std::cout << std::setw(18) << "Difference norm: " << std::setw(14) << (lc - fc).norm() << "\n";
        std::cout << std::setw(18) << "Speedup: "         << std::setw(14) << kihaTime/lemmaTime << "\n" << std::endl;

        TS_ASSERT_DELTA((lc-fc).norm(), 0.0, Delta);
        #endif
   }

   void test_Ez() {

        std::cout.precision(4);

        dipole->SetType(GROUNDEDELECTRICDIPOLE);
		dipole->SetPolarisation(ZPOLARISATION);

        // Put in a unit test that will be slow.
        std::cout << "GROUNDEDELECTRICDIPOLE Z polarisation" << std::endl;
        std::cout << "=====================================\n";
	    std::cout << std::setw(18) << "Lemma/C++: ";
        std::cout.flush();

  	    timer.begin();
	    EmEarth->MakeCalc3();
	    Real lemmaTime = timer.end();
        std::cout << std::setw(14) << lemmaTime << std::setw(6) << " [s]" << std::endl;

        auto lc = receivers->GetEfield( 0 );

        #ifdef KIHALEE_EM1D
	    receivers->ClearFields();
        std::cout << std::setw(18) << "KiHa/Fortran: ";
        std::cout.flush();

        timer.begin();
 	    EmEarth->MakeCalc();
	    Real kihaTime = timer.end();
        std::cout << std::setw(14) << kihaTime << std::setw(6) << " [s]" << std::endl;

        auto fc = receivers->GetEfield( 0 ); //0,0);

        //std::cout.precision(16);
        std::cout << std::setw(18) << "Lemma norm: "      << std::setw(14) << (lc).norm() << std::endl;
        std::cout << std::setw(18) << "KiHa norm: "       << std::setw(14) <<  (fc).norm() << std::endl;
        std::cout << std::setw(18) << "Difference norm: " << std::setw(14) << (lc - fc).norm() << "\n";
        std::cout << std::setw(18) << "Speedup: "         << std::setw(14) << kihaTime/lemmaTime << "\n" << std::endl;

        TS_ASSERT_DELTA((lc-fc).norm(), 0.0, Delta);
        #endif
   }

};