Main Page | Modules | Namespace List | Class Hierarchy | Alphabetical List | Class List | Directories | File List | Namespace Members | Class Members | File Members | Related Pages

BfldCanvas Class Reference

#include <BfldCanvas.h>

List of all members.

Public Member Functions

 BfldCanvas (VldContext &vldc, std::string tag, int iscoil)
 BfldCanvas ()
virtual ~BfldCanvas ()
void SetTitle (std::string title)
void Fill (BField &bfld)
void Fill (BfldHandler &bhandler)
virtual void Draw ()
void SetVldContext (VldContext vldc)
void SetZ (double z)
void SetScaleMax (double scale=1.0)
void SetArrowScale (double scale=1.0)
TCanvas * GetCanvas () const
TH2F * GetScatter () const
VldContext GetVldContext () const
double GetZ () const
int GetIsCoil () const
double GetScaleMax () const
BfldInterpMethod::InterpMethod_t GetInterpMethod () const

Static Public Member Functions

BfldInterpMethod::InterpMethod_t GetDefaultInterpMethod ()
void SetDefaultInterpMethod (BfldInterpMethod::InterpMethod_t method)
void SetVerbose (int v=true)
void SetKGaussLimits (double kgmin, double kgmax)
void SetArrowGrid (int nxa=40)

Protected Member Functions

virtual void DetermineLimits (double &xmin, double &xmax, double &ymin, double &ymax, int &xpixels, int &ypixels)
virtual void DetermineColorGrid (int &nx, int &ny)
virtual void DetermineArrowGrid (int &nxa, int &nya, double &scalea)
virtual void AdjustLabels ()
virtual void AdjustMargins ()
void Fill (BField *bfld=0, BfldHandler *bhandler=0)
virtual void DrawArrows ()
virtual void DrawCheveron (double x0, double y0, double dx, double dy, double scale)
virtual std::string GetZoomInterpString ()

Protected Attributes

TCanvas * fCanvas
TH2F * fScatter
TH2F * fPhase
int fIsCoil
VldContext fVldc
double fZ
double fScaleMax
double fArrowScale
BfldInterpMethod::InterpMethod_t fInterpMethod

Static Protected Attributes

int fgVerbose = 0
 definition and initializion of class statics
double fgKGMin = 0.0
double fgKGMax = 22.0
int fgArrowGridX = 40
BfldInterpMethod::InterpMethod_t fgDefaultInterpMethod = BfldInterpMethod::kDefault

Constructor & Destructor Documentation

BfldCanvas::BfldCanvas (  VldContext &  vldc,
std::string  tag,
int  iscoil
) 
 

Definition at line 48 of file BfldCanvas.cxx.

References AdjustLabels(), DetermineArrowGrid(), DetermineColorGrid(), DetermineLimits(), fCanvas, fPhase, fScatter, and SetScaleMax().

00049   : fCanvas(0)
00050   , fScatter(0)
00051   , fPhase(0)
00052   , fIsCoil(iscoil)
00053   , fVldc(vldc)
00054   , fZ(-999)
00055   , fScaleMax(1.0)
00056   , fArrowScale(1.0)
00057   , fInterpMethod(fgDefaultInterpMethod)
00058 {
00059   // some initial defaults
00060   double xmin = -5.0*Munits::m, xmax = +5.0*Munits::m;
00061   double ymin = -5.0*Munits::m, ymax = +5.0*Munits::m;
00062   double scalea = 1.0;   // scale for arrows
00063   int  xpixels = 800, ypixels = 800;
00064   int  nx=500, ny=500;  // grid for color map
00065   int  nxa=40, nya=40;  // grid for arrows
00066 
00067   DetermineLimits(xmin,xmax,ymin,ymax,xpixels,ypixels);
00068   DetermineColorGrid(nx,ny);
00069   DetermineArrowGrid(nxa,nya,scalea);
00070   
00071   std::string cname = tag;   cname += "_canvas";
00072   std::string sname = tag;
00073   std::string pname = tag;   pname += "_phase";
00074   fCanvas  = new TCanvas(cname.c_str(),cname.c_str(),xpixels,ypixels);
00075   fScatter = new TH2F(sname.c_str(),sname.c_str(),nx,xmin,xmax,ny,ymin,ymax);
00076   fPhase   = new TH2F(pname.c_str(),pname.c_str(),nxa,xmin,xmax,nya,ymin,ymax);
00077   AdjustLabels();
00078   SetScaleMax();
00079   
00080 }

BfldCanvas::BfldCanvas (   ) 
 

Definition at line 27 of file BfldCanvas.cxx.

00028   : fCanvas(0)
00029   , fScatter(0)
00030   , fPhase(0)
00031   , fIsCoil(0)
00032   , fVldc()
00033   , fZ(-999)
00034   , fScaleMax(1.0)
00035   , fArrowScale(1.0)
00036   , fInterpMethod(fgDefaultInterpMethod)
00037 {
00038   // default ctor for ROOTCINT
00039 }

BfldCanvas::~BfldCanvas (   )  [virtual]
 

Definition at line 41 of file BfldCanvas.cxx.

00042 {
00043   if (fCanvas)  delete fCanvas;
00044   if (fScatter) delete fScatter;
00045   if (fPhase)   delete fPhase;
00046 }

Member Function Documentation

void BfldCanvas::AdjustLabels (   )  [protected, virtual]
 

Definition at line 217 of file BfldCanvas.cxx.

References fScatter.

Referenced by BfldCanvas().

00218 {
00219   fScatter->SetStats(false);
00220   fScatter->SetDirectory(0);
00221   fScatter->SetXTitle("x [ m ]");
00222   fScatter->SetYTitle("y [ m ]");
00223   fScatter->SetZTitle("B [ T ]");
00224   double tsize = 0.045, lsize = 0.035;
00225   fScatter->GetXaxis()->SetTitleSize(tsize);
00226   fScatter->GetYaxis()->SetTitleSize(tsize);
00227   fScatter->GetZaxis()->SetTitleSize(tsize);
00228   fScatter->GetXaxis()->SetLabelSize(lsize);
00229   fScatter->GetYaxis()->SetLabelSize(lsize);
00230   fScatter->GetZaxis()->SetLabelSize(lsize);
00231 }

void BfldCanvas::AdjustMargins (   )  [protected, virtual]
 

Definition at line 233 of file BfldCanvas.cxx.

References fCanvas.

Referenced by Draw().

00234 {
00235   fCanvas->SetTopMargin(0.1625); // 0.16875);   // 0.175);   // 0.075
00236   fCanvas->SetBottomMargin(0.10);   // 0.070   0.145
00237   fCanvas->SetLeftMargin(0.10);     // 0.06 
00238   fCanvas->SetRightMargin(0.1875); // 0.19375); //0.20);  // 0.110   0.170  0.025 (for z)
00239 }

void BfldCanvas::DetermineArrowGrid (  int &  nxa,
int &  nya,
double &  scalea
)  [protected, virtual]
 

Definition at line 187 of file BfldCanvas.cxx.

References DetermineLimits(), and fgArrowGridX.

Referenced by BfldCanvas(), and DrawArrows().

00188 {
00189   nxa = nya = 40;
00190   scalea = 1.0; 
00191 
00192   if ( fIsCoil ) {
00193     // change arrow size, grid
00194     nxa = nya = 25;
00195     scalea = 2.5;
00196     
00197     nxa = nya = 40;
00198     scalea = 2.0;
00199   }
00200 
00201   double xmin, xmax, ymin, ymax;
00202   int xpixels, ypixels;
00203   DetermineLimits(xmin,xmax,ymin,ymax,xpixels,ypixels);
00204 
00205   scalea = 1.5*5.0*(Munits::cm)*(xmax-xmin)/10.0*(Munits::m);
00206 
00207   scalea *= fArrowScale;     // scale for arrows size
00208 
00209   // change number of phase histogram bins in X
00210   nxa = TMath::Nint((float)(nxa) * (float)fgArrowGridX/40.0);
00211 
00212   // keep aspect ratio
00213   nya = TMath::Nint((float)(nxa) * (ymax-ymin)/(xmax-xmin));
00214 
00215 }

void BfldCanvas::DetermineColorGrid (  int &  nx,
int &  ny
)  [protected, virtual]
 

Definition at line 182 of file BfldCanvas.cxx.

Referenced by BfldCanvas().

00183 { 
00184   nx  = ny  = 500;  // grid for color map 
00185 }

void BfldCanvas::DetermineLimits (  double &  xmin,
double &  xmax,
double &  ymin,
double &  ymax,
int &  xpixels,
int &  ypixels
)  [protected, virtual]
 

Definition at line 147 of file BfldCanvas.cxx.

References fIsCoil, fVldc, and VldContext::GetDetector().

Referenced by BfldCanvas(), and DetermineArrowGrid().

00150 {
00151   xmin = -5.0*Munits::m; xmax = +5.0*Munits::m;
00152   ymin = -5.0*Munits::m; ymax = +5.0*Munits::m;
00153 
00154   if (Detector::kNear == fVldc.GetDetector()) {
00155       xmin = -2.8*Munits::m, xmax = +4.0*Munits::m;
00156       ymin = -2.4*Munits::m, ymax = +2.4*Munits::m;
00157   }
00158 
00159   if ( fIsCoil > 0 ) {
00160 
00161     xmin = -0.55*Munits::m, xmax = +0.55*Munits::m;
00162     ymin = -0.55*Munits::m, ymax = +0.55*Munits::m;
00163 
00164     if ( fIsCoil > 1 ) {
00165       xmin = -0.35*Munits::m, xmax = +0.35*Munits::m;
00166       ymin = -0.35*Munits::m, ymax = +0.35*Munits::m;
00167     }
00168 
00169       //xmin =  0.15*Munits::m, xmax =  0.23*Munits::m;
00170       //ymin = -0.12*Munits::m, ymax = -0.04*Munits::m;
00171 
00172       //xmin =  0.66*Munits::m, xmax =  0.84*Munits::m;
00173       //ymin = -0.49*Munits::m, ymax = -0.31*Munits::m;
00174       //kilogauss_min = 0.9/Munits::kilogauss;
00175       //kilogauss_max = 1.6/Munits::kilogauss;
00176   }
00177 
00178   xpixels = ypixels = 800; // canvas size 
00179   xpixels = TMath::Nint(ypixels * (xmax-xmin)/(ymax-ymin));
00180 }

void BfldCanvas::Draw (   )  [virtual]
 

Definition at line 128 of file BfldCanvas.cxx.

References AdjustMargins(), DrawArrows(), fArrowScale, fCanvas, and fScatter.

Referenced by BfldCanvasSlice::Draw().

00129 {
00130   fCanvas->Draw();
00131   fCanvas->cd();
00132   AdjustMargins();
00133 
00134   fScatter->SetContour(gStyle->GetNumberOfColors());
00135   fScatter->Draw("COLZ");
00136 
00137   if ( fArrowScale > 0 ) DrawArrows();
00138 
00139   // coil region limits
00140   //draw_border(0.298); 
00141   //draw_border(0.448); 
00142 
00143   fCanvas->Update();
00144 
00145 }

void BfldCanvas::DrawArrows (   )  [protected, virtual]
 

Definition at line 317 of file BfldCanvas.cxx.

References DetermineArrowGrid(), DrawCheveron(), and fPhase.

Referenced by Draw().

00318 {
00319 
00320   int i,j;
00321   double scalea;
00322   DetermineArrowGrid(i,j,scalea); // only use "scalea"
00323 
00324   TAxis* xaxis = fPhase->GetXaxis();
00325   int    nx    = xaxis->GetNbins();
00326   //double xmin  = xaxis->GetXmin();
00327   //double xmax  = xaxis->GetXmax();
00328 
00329   TAxis* yaxis = fPhase->GetYaxis();
00330   int    ny    = yaxis->GetNbins();
00331   //double ymin  = yaxis->GetXmin();
00332   //double ymax  = yaxis->GetXmax();
00333 
00334   double x, y;
00335   int ix, iy;
00336 
00337   //double dx = (xmax-xmin)/(double)nx;
00338   //double dy = (ymax-ymin)/(double)ny;
00339   
00340   for (ix=1; ix<=nx; ++ix) {
00341     x = xaxis->GetBinCenter(ix);
00342     for (iy=1; iy<=ny; ++iy) {
00343       y = yaxis->GetBinCenter(iy);
00344 
00345       int ihashbin = fPhase->GetBin(ix,iy);
00346       double phi = fPhase->GetBinContent(ihashbin);
00347       if ( phi > TMath::TwoPi() ) continue;  // don't draw nonsense
00348 
00349       double bx = TMath::Cos(phi);
00350       double by = TMath::Sin(phi);
00351       DrawCheveron(x,y,bx,by,scalea);
00352    
00353     } // done y
00354   } // done x
00355 
00356 }

void BfldCanvas::DrawCheveron (  double  x0,
double  y0,
double  dx,
double  dy,
double  scale
)  [protected, virtual]
 

Definition at line 358 of file BfldCanvas.cxx.

References len, and s().

Referenced by DrawArrows().

00360 {
00361   static double xp[] = { -0.16, -0.50, 0.60, -0.50, -0.16};
00362   static double yp[] = {  0.00,  0.25, 0.00, -0.25,  0.00};
00363   double x[5], y[5];
00364   double len = TMath::Sqrt(dx*dx+dy*dy);
00365   if (len<=0) return;
00366   double c = dx/len;
00367   double s = dy/len;
00368   // rotate, scale, translate                                                 
00369   for (int i=0; i<5; ++i) {
00370     x[i] = (xp[i]*c - yp[i]*s)*scale + x0;
00371     y[i] = (xp[i]*s + yp[i]*c)*scale + y0;
00372   }
00373   static TPolyLine cheveron;
00374   cheveron.SetLineWidth(1);
00375   cheveron.DrawPolyLine(5,x,y);
00376   //static TEllipse  circle;
00377   //circle.SetLineWidth(1);
00378   //circle.SetFillStyle(0);  // transparent fill
00379   //circle.DrawEllipse(x0,y0,scale,0.,0.,360.,0.,"");
00380 }

void BfldCanvas::Fill (  BField *  bfld = 0,
BfldHandler *  bhandler = 0
)  [protected]
 

Definition at line 241 of file BfldCanvas.cxx.

References fScatter, fZ, BfldHandler::GetBField(), and BField::GetBField().

00242 {
00243 
00244   TVector3 xyz, bxyz;
00245   
00246   xyz.SetZ(fZ);
00247   
00248   if (fgVerbose) cout << " start loop over grid " << endl;
00249   
00250   // fill the TH2F's, first the color grid, then the phase
00251   for (int pass=0; pass<2; ++pass) {
00252     TH2F* h2 = ((pass==0) ? fScatter : fPhase );
00253     double x, y;
00254     int ix, iy;
00255 
00256     // don't rely on functions for limit/grid info 
00257     // take it from scatter plot
00258 
00259     TAxis* xaxis = h2->GetXaxis();
00260     int    nx    = xaxis->GetNbins();
00261     double xmin  = xaxis->GetXmin();
00262     double xmax  = xaxis->GetXmax();
00263 
00264     TAxis* yaxis = h2->GetYaxis();
00265     int    ny    = yaxis->GetNbins();
00266     double ymin  = yaxis->GetXmin();
00267     double ymax  = yaxis->GetXmax();
00268 
00269     if (fgVerbose) {
00270       cout << " x " << xmin << " " << xmax << " nx " << nx << endl;
00271       cout << " y " << ymin << " " << ymax << " ny " << ny << endl;
00272     }
00273 
00274     //double dx = (xmax-xmin)/(double)nx;
00275     //double dy = (ymax-ymin)/(double)ny;
00276   
00277     for (ix=1; ix<=nx; ++ix) {
00278       x = xaxis->GetBinCenter(ix);
00279       xyz.SetX(x);
00280       for (iy=1; iy<=ny; ++iy) {
00281         y = yaxis->GetBinCenter(iy);
00282         xyz.SetY(y);
00283         
00284         //int ihashbin = h2->GetBin(ix,iy);
00285 
00286         if ( bfld ) {
00287           bxyz = bfld->GetBField(xyz);
00288         } else if ( bhandler ) {
00289           bxyz = bhandler->GetBField(xyz,false);
00290         }
00291         
00292         double bmag  = bxyz.Mag();
00293         double value = bmag;
00294         if ( pass == 1 ) {
00295           double phi   = ( (bmag > 0.0) ? bxyz.Phi() : 42 );
00296           value = phi;
00297         }
00298         h2->Fill(x,y,value);
00299       
00300 #ifdef RWHDEBUG
00301         double r2   = x*x + y*y;
00302         double r2lo = 0.09 *  0.09;
00303         double r2hi = 0.12 *  0.12;
00304         //if (TMath::Abs(x)<10.*Munits::cm && TMath::Abs(y)<10.*Munits::cm ) {
00305         if ( false && r2lo < r2 && r2 < r2hi  ) {
00306           if (fgVerbose) cout << " pos " << xyz << " B " << bxyz << endl;
00307         }
00308 #endif
00309         
00310       } // done y
00311     } // done x
00312 
00313   } // done pass
00314   
00315 }

void BfldCanvas::Fill (  BfldHandler &  bhandler  ) 
 

Definition at line 123 of file BfldCanvas.cxx.

References Fill().

00124 {
00125   Fill(0,&bhandler);
00126 }

void BfldCanvas::Fill (  BField &  bfld  ) 
 

Definition at line 119 of file BfldCanvas.cxx.

Referenced by Fill().

00120 {
00121   Fill(&bfld,0);
00122 }

TCanvas* BfldCanvas::GetCanvas (   )  const [inline]
 

Definition at line 44 of file BfldCanvas.h.

00044 { return fCanvas; }

BfldInterpMethod::InterpMethod_t BfldCanvas::GetDefaultInterpMethod (   )  [inline, static]
 

Definition at line 56 of file BfldCanvas.h.

00056 { return fgDefaultInterpMethod; }

BfldInterpMethod::InterpMethod_t BfldCanvas::GetInterpMethod (   )  const [inline]
 

Definition at line 51 of file BfldCanvas.h.

Referenced by GetZoomInterpString().

00051 { return fInterpMethod; }

int BfldCanvas::GetIsCoil (   )  const [inline]
 

Definition at line 48 of file BfldCanvas.h.

Referenced by GetZoomInterpString().

00048 { return fIsCoil; }

double BfldCanvas::GetScaleMax (   )  const [inline]
 

Definition at line 49 of file BfldCanvas.h.

00049 { return fScaleMax; }

TH2F* BfldCanvas::GetScatter (   )  const [inline]
 

Definition at line 45 of file BfldCanvas.h.

00045 { return fScatter; }

VldContext BfldCanvas::GetVldContext (   )  const [inline]
 

Definition at line 46 of file BfldCanvas.h.

00046 { return fVldc; }

double BfldCanvas::GetZ (   )  const [inline]
 

Definition at line 47 of file BfldCanvas.h.

00047 { return fZ; }

std::string BfldCanvas::GetZoomInterpString (   )  [protected, virtual]
 

Definition at line 382 of file BfldCanvas.cxx.

References Form(), GetInterpMethod(), and GetIsCoil().

00383 {
00384   std::string result = ( GetIsCoil() ? "_coil" : "_full");
00385   int im = GetInterpMethod();
00386   if ( im != (int)BfldInterpMethod::kDefault )
00387     result += Form("_im%d",im);
00388   return result;
00389 }

void BfldCanvas::SetArrowGrid (  int  nxa = 40  )  [inline, static]
 

Definition at line 62 of file BfldCanvas.h.

00062 { fgArrowGridX = nxa; }

void BfldCanvas::SetArrowScale (  double  scale = 1.0  )  [inline]
 

Definition at line 42 of file BfldCanvas.h.

00042 { fArrowScale = scale; }

void BfldCanvas::SetDefaultInterpMethod (  BfldInterpMethod::InterpMethod_t  method  )  [inline, static]
 

Definition at line 57 of file BfldCanvas.h.

00058                { fgDefaultInterpMethod = method; }

void BfldCanvas::SetKGaussLimits (  double  kgmin,
double  kgmax
)  [inline, static]
 

Definition at line 60 of file BfldCanvas.h.

00061                { fgKGMin = kgmin; fgKGMax = kgmax; }

void BfldCanvas::SetScaleMax (  double  scale = 1.0  ) 
 

Definition at line 82 of file BfldCanvas.cxx.

References fgKGMax, fgKGMin, fScaleMax, and fScatter.

Referenced by BfldCanvas().

00083 { 
00084   fScaleMax = scale;
00085   if ( fScatter ) {
00086     fScatter->SetMinimum(fgKGMin*Munits::kilogauss);
00087     fScatter->SetMaximum(fgKGMax*Munits::kilogauss*fScaleMax);
00088   }
00089 }

void BfldCanvas::SetTitle (  std::string  title  ) 
 

Definition at line 91 of file BfldCanvas.cxx.

References fCanvas, Form(), fPhase, and fScatter.

00092 {
00093   std::string name = title.substr(0,title.find(" "));
00094 
00095   //static int icnt = 0; ++icnt;
00096   //std::string nameuniq = Form("_%d",icnt);
00097   static std::map<std::string,int> namecnt;
00098   int icnt = namecnt[title]++;
00099   std::string nameuniq = "";
00100   if ( icnt > 0 ) nameuniq = Form("_%d",icnt);
00101 
00102   std::string cname = name;
00103   cname += "_canvas";
00104   fCanvas->SetTitle(cname.c_str());
00105   cname += nameuniq;
00106   fCanvas->SetName(cname.c_str());
00107 
00108   fScatter->SetTitle(""); // (title.c_str());
00109   title += "_phase";
00110   fPhase->SetTitle(title.c_str());
00111   
00112   name += nameuniq;
00113   fScatter->SetName(name.c_str());
00114   name += "_phase";
00115   fPhase->SetName(name.c_str());
00116 
00117 }

void BfldCanvas::SetVerbose (  int  v = true  )  [inline, static]
 

Definition at line 59 of file BfldCanvas.h.

00059 { fgVerbose = v; }

void BfldCanvas::SetVldContext (  VldContext  vldc  )  [inline]
 

Definition at line 39 of file BfldCanvas.h.

00039 { fVldc = vldc; }

void BfldCanvas::SetZ (  double  z  )  [inline]
 

Definition at line 40 of file BfldCanvas.h.

00040 { fZ = z; }

Member Data Documentation

double BfldCanvas::fArrowScale [protected]
 

Definition at line 86 of file BfldCanvas.h.

Referenced by Draw().

TCanvas* BfldCanvas::fCanvas [protected]
 

Definition at line 79 of file BfldCanvas.h.

Referenced by AdjustMargins(), BfldCanvas(), Draw(), and SetTitle().

int BfldCanvas::fgArrowGridX = 40 [static, protected]
 

Definition at line 21 of file BfldCanvas.cxx.

Referenced by DetermineArrowGrid().

BfldInterpMethod::InterpMethod_t BfldCanvas::fgDefaultInterpMethod = BfldInterpMethod::kDefault [static, protected]
 

Definition at line 23 of file BfldCanvas.cxx.

double BfldCanvas::fgKGMax = 22.0 [static, protected]
 

Definition at line 20 of file BfldCanvas.cxx.

Referenced by SetScaleMax().

double BfldCanvas::fgKGMin = 0.0 [static, protected]
 

Definition at line 19 of file BfldCanvas.cxx.

Referenced by SetScaleMax().

int BfldCanvas::fgVerbose = 0 [static, protected]
 

definition and initializion of class statics

Definition at line 18 of file BfldCanvas.cxx.

BfldInterpMethod::InterpMethod_t BfldCanvas::fInterpMethod [protected]
 

Definition at line 87 of file BfldCanvas.h.

int BfldCanvas::fIsCoil [protected]
 

Definition at line 82 of file BfldCanvas.h.

Referenced by DetermineLimits().

TH2F* BfldCanvas::fPhase [protected]
 

Definition at line 81 of file BfldCanvas.h.

Referenced by BfldCanvas(), DrawArrows(), and SetTitle().

double BfldCanvas::fScaleMax [protected]
 

Definition at line 85 of file BfldCanvas.h.

Referenced by SetScaleMax().

TH2F* BfldCanvas::fScatter [protected]
 

Definition at line 80 of file BfldCanvas.h.

Referenced by AdjustLabels(), BfldCanvas(), Draw(), Fill(), SetScaleMax(), and SetTitle().

VldContext BfldCanvas::fVldc [protected]
 

Definition at line 83 of file BfldCanvas.h.

Referenced by DetermineLimits().

double BfldCanvas::fZ [protected]
 

Definition at line 84 of file BfldCanvas.h.

Referenced by Fill().

The documentation for this class was generated from the following files:
Generated on Mon Feb 15 11:08:48 2010 for loon by  doxygen 1.3.9.1