BfldMap.cxx

Go to the documentation of this file.

// $Id: BfldMap.cxx,v 1.9 2007/12/11 22:53:34 rhatcher Exp $
// 
// BfldMap 
// 
// BfldMap is a structure for holding Bfield maps
// these associate field vectors with node positions but don't know
// where those nodes are in 3-space.
//
// Author:  R. Hatcher 2000.06.20
//
#include "BField/BfldMap.h"

#include "MessageService/MsgService.h"
CVSID("$Id: BfldMap.cxx,v 1.9 2007/12/11 22:53:34 rhatcher Exp $");

ClassImp(BfldMap)

//_____________________________________________________________________________
BfldMap::BfldMap() 
  : fGrid(BfldGrid::kUndefined)
  , fVariant(-1)
  , fIs3d(false)
  , fGeneratedCoilCurrent(0.0)
  , fFilledOkay(false)
  , fNumGenerators(0)
{
  // default ctor
}

//_____________________________________________________________________________
BfldMap::BfldMap(BfldGrid::Grid_t grid, Int_t variant) 
  : fGrid(grid)
  , fVariant(variant)
  , fIs3d(false)
  , fGeneratedCoilCurrent(0.0)
  , fFilledOkay(false)
  , fNumGenerators(0)
{

}

//_____________________________________________________________________________
BfldMap::~BfldMap()
{
   fGrid       = BfldGrid::kUndefined;
   fVariant    = -1;
   fFilledOkay = false;

}

//_____________________________________________________________________________

TVector3 BfldMap::GetBField(Int_t generator)
{
   // Return just a simple copy of the vector at this generator point
   // BfldMap probably isn't the place to do the slot, B-H, etc.
   // corrections anyway because:
   //  (1) we don't have the necessary info to do more than 
   //      simple linear scaling by the current ratio
   //  (2) it's expensive to do this *before* interpolating
   //      and probably no more correct
   //MAXMSG("Bfld",Msg::kDebug,5) 
   //  << "BfldMap::GetBField is same as GetRawField (no scaling)!"
   //  << endl;

   return GetRawField(generator);

}

//_____________________________________________________________________________
TVector3 BfldMap::GetRawField(Int_t generator)
{
   // Return a copy of the vector at this generator point

   if ( fFilledOkay &&
        generator >= 0 && (UInt_t)generator <= fNumGenerators ) {
     if (fIs3d) {
       return TVector3(fBx[generator],fBy[generator],fBz[generator]);
     }
     else {
       return TVector3(fBx[generator],fBy[generator],0.0);
     }
   }

   MAXMSG("Bfld",Msg::kDebug,20)
     << "BfldMap::GetRawField no raw field for generator " 
     << generator << " of " << fNumGenerators << endl;
   return TVector3(0.,0.,0.);

}

//_____________________________________________________________________________
void BfldMap::ResizeVectors(UInt_t size)
{
   // Allocate whatever's holding the data if necessary
   // Make sure we can add a generator value

   // When using vector<Double_t> or vector<Float_t> x,y,z
   // then we have to resize and make sure everything is hunky dory
   if ( size > fNumGenerators ) {
     fBx.resize(size);
     fBy.resize(size);
     if ( fIs3d ) fBz.resize(size);
     fNumGenerators = size;
   }

}

//_____________________________________________________________________________
void BfldMap::AddGenerator(UInt_t generator, 
                           Double_t bx, Double_t by, Double_t bz)
{
   // add a field vector for generator point

  // Check if the third component is non-zero
  // If so, and we haven't done so already make sure there's
  // enough space in the fBz vector
  bool hasbz = ( bz != 0.0 );
  if ( hasbz && ! fIs3d ) { fIs3d = true; fBz.resize(fNumGenerators); }
  // Make sure all arrays can hold this next entry
  // If we're expanding, be generous with an extra 128 to avoid 
  // multiple single expansions.
  if ( generator >= fNumGenerators ) ResizeVectors(generator+128);
  // Store the components of the generator's data
  fBx[generator] = bx;
  fBy[generator] = by;
  if ( hasbz ) fBz[generator] = bz;

}

//_____________________________________________________________________________
void BfldMap::AddGenerator(UInt_t generator, TVector3& bfld)
{
   // add a field vector for generator point
   AddGenerator(generator,bfld.x(),bfld.y(),bfld.z());
}

//_____________________________________________________________________________

---