// -------------------------------------------------------------------------
// -----                    CbmFieldMapData source file                -----
// -----                V. Friese                14/02/06              -----
// -------------------------------------------------------------------------


#include "TArrayF.h"

#include "CbmFieldMap.h"
#include "CbmFieldMapData.h"



// -------------   Default constructor  ----------------------------------
CbmFieldMapData::CbmFieldMapData() {
  fType  = 1;
  fXmin  = fYmin  = fZmin  = 0.;
  fXmax  = fYmax  = fZmax  = 0.;
  fNx    = fNy    = fNz    = 0;
  fBx    = fBy    = fBz    = NULL;
}
// ------------------------------------------------------------------------



// -------------   Standard constructor   ---------------------------------
CbmFieldMapData::CbmFieldMapData(const char* mapName)
  : TNamed(mapName, "CBM Field Map Data") {
  fType  = 1;
  fXmin  = fYmin  = fZmin  = 0.;
  fXmax  = fYmax  = fZmax  = 0.;
  fNx    = fNy    = fNz    = 0;
  fBx    = fBy    = fBz    = NULL;
}
// ------------------------------------------------------------------------



// -----   Constructor from CbmFieldMap   ------------------------------
CbmFieldMapData::CbmFieldMapData(const char* name,
				 const CbmFieldMap& map) 
: TNamed(name, "CBM Field Map Data") {

  fType  = map.GetType();
  fXmin  = map.GetXmin();
  fYmin  = map.GetYmin();
  fZmin  = map.GetZmin();
  fXmax  = map.GetXmax();
  fYmax  = map.GetYmax();
  fZmax  = map.GetZmax();
  fNx    = map.GetNx();
  fNy    = map.GetNy();
  fNz    = map.GetNz();
  fBx = new TArrayF(*(map.GetBx()));
  fBy = new TArrayF(*(map.GetBy()));
  fBz = new TArrayF(*(map.GetBz()));
  fUnit = map.GetUnit();

  // Take out scaling factor and convert from kG to T
  Double_t factor = map.GetScale() * 10.; 
  Int_t index = 0;
  for (Int_t ix=0; ix<fNx; ix++) {
    for (Int_t iy=0; iy<fNy; iy++) {
      for (Int_t iz=0; iz<fNz; iz++) {
	index = ix*fNy*fNz + iy*fNz + iz;
	if ( fBx ) (*fBx)[index] = (*fBx)[index] / factor;
	if ( fBy ) (*fBy)[index] = (*fBy)[index] / factor;
	if ( fBz ) (*fBz)[index] = (*fBz)[index] / factor;
      }    // z loop
    }      // y loop
  }        // x loop

}
// ------------------------------------------------------------------------



// ------------   Destructor   --------------------------------------------
CbmFieldMapData::~CbmFieldMapData() {
  if ( fBx ) delete fBx;
  if ( fBy ) delete fBy;
  if ( fBz ) delete fBz;
}
// ------------------------------------------------------------------------


ClassImp(CbmFieldMapData)