/**
 * \file CbmLitFieldGridCreator.cxx
 *
 * \author Andrey Lebedev <andrey.lebedev@gsi.de>
 * \date 2011
 *
 */
#include "CbmLitFieldGridCreator.h"
#include "../../parallel/LitFieldGrid.h"

#include "FairField.h"
#include "FairRunAna.h"

CbmLitFieldGridCreator::CbmLitFieldGridCreator():
   fField(FairRunAna::Instance()->GetField()),
   fXangle(35.),
   fYangle(35.),
   fCellSizeX(2.),
   fCellSizeY(2.)
{

}

CbmLitFieldGridCreator::~CbmLitFieldGridCreator()
{

}

void CbmLitFieldGridCreator::CreateGrid(
   fscal Z,
   lit::parallel::LitFieldGrid& grid)
{
   float tanXangle = std::tan(fXangle * 3.14159265 / 180.); //
   float tanYangle = std::tan(fYangle * 3.14159265 / 180.); //
   float Xmax = Z * tanXangle;
   float Ymax = Z * tanYangle;
   int nofCellsX = int(Xmax / fCellSizeX);
   int nofCellsY = int(Ymax / fCellSizeY);
   Xmax = nofCellsX * fCellSizeX;
   Ymax = nofCellsY * fCellSizeY;
   nofCellsX += nofCellsX;
   nofCellsY += nofCellsY;

   std::vector<std::vector<lit::parallel::LitFieldValue<fscal> > > field;
   field.resize(nofCellsX + 1);
   for (unsigned int i = 0; i < nofCellsX + 1; i++) {
      field[i].resize(nofCellsY + 1);
   }

   for (int j = 0; j < nofCellsX + 1; j++) { // loop over x position
      double X = -Xmax + j * fCellSizeX;
      for (int k = 0; k < nofCellsY + 1; k++) { // loop over y position
         double Y = -Ymax + k * fCellSizeY;

         // get field value
         double pos[3] = {X, Y, Z};
         double B[3];
         fField->GetFieldValue(pos, B);

         lit::parallel::LitFieldValue<fscal> v;
         v.Bx = B[0];
         v.By = B[1];
         v.Bz = B[2];
         field[j][k] = v;
      }
   }
   grid.SetZ(Z);
   grid.SetField(field, -Xmax, Xmax, -Ymax, Ymax, nofCellsX, nofCellsY);
   std::cout << "Grid created: " << grid << "\n";
}