BfldCanvasSlice.cxx

Go to the documentation of this file.

#include "BField/Viz/BfldCanvasSlice.h"

#include <iostream>
using namespace std;

#include "TMath.h"
#include "TString.h"
#include "TPolyLine.h"
#include "TEllipse.h"

#include "Conventions/Munits.h"
#include "Plex/PlexPlaneId.h"
#include "UgliGeometry/UgliGeomHandle.h"
#include "BField/BfldDbiPlaneMap.h"
#include "DatabaseInterface/DbiResultPtr.h"

BfldCanvasSlice::BfldCanvasSlice(VldContext& vldc, int ipln, 
                                 int iscoil, bool isgap, double zoff)
  : BfldCanvas(vldc,Form("BfldCanvasSlice%d",ipln),iscoil)
  , fPlane(ipln)
  , fIsGap(isgap)
{ 
  SetVldContext(vldc);
  if (isgap) SetScaleMax(0.1);

  UgliGeomHandle ugh(vldc);
  bool issteel = !isgap;
  double dz = zoff;
  Detector::Detector_t det = vldc.GetDetector();
  PlexPlaneId plnid(det,ipln,issteel);
  if ( ! plnid.IsValid() ) {
    if ( issteel ) {
      // no steel to reference against
      cout << "invalid steel plane " << ipln << " " << plnid << endl;
      plnid = PlexPlaneId(det,0,true);
    } else {
      // not legal scint plane
      // reference relative to steel by 0.5 of nominal pitch upstream
      cout << "invalid scint plane " << ipln << " " << plnid << endl;
      plnid = PlexPlaneId(det,ipln,true);
      dz -= 0.5 * 5.94*Munits::cm;
      if ( ! plnid.IsValid() ) plnid = PlexPlaneId(det,0,true);
    }
  }
  vector<UgliPlnHandle> uphv = ugh.GetPlnHandleVector();
  for (unsigned int i = 0; i < uphv.size(); ++i ) {
    UgliPlnHandle uph = uphv[i];
    if ( uph.GetPlexPlaneId() == plnid ) { 
      SetZ(uph.GetZ0()+dz);
      break;
    }
  }

  BField bfld(vldc);
  bfld.SetInterpMethod(GetInterpMethod());
  Fill(bfld);

  std::string name = Detector::AsString(vldc.GetDetector());
  name += ((isgap) ? "_gap" : "_pln");
  name += Form("%3.3d",ipln);
  name += GetZoomInterpString();
  name += " #scale[0.35]{#splitline";
  name += "{";
  name += GetAdmixtureString();
  name += "}";
  std::string zoffstr = "{.}";
  if ( zoff != 0.0 ) zoffstr = Form("{z0+%5.3f=%7.3f}",zoff,GetZ());
  cout << " zoffstr = " << zoffstr << " zoff " << zoff << " dz " << dz << endl;
  name += zoffstr;
  name += "}";  // finish scale

  SetTitle(name);
                    
}

BfldCanvasSlice::BfldCanvasSlice()
  : BfldCanvas()
  , fPlane(-999)
  , fIsGap(false)
{
  // default ctor for ROOTCINT
}

BfldCanvasSlice::~BfldCanvasSlice()
{ }

void BfldCanvasSlice::Draw()
{
  BfldCanvas::Draw(); // call normal one
  // no add decorations

  double radius = 0.448;  // old 0.298
  BfldDbiPlaneMap::MapShape_t shape = BfldDbiPlaneMap::kCircle;

  const BfldDbiPlaneMap* plnmap = GetPlaneMap();
  if ( plnmap ) {
    radius = plnmap->GetDetailSize();
    shape  = plnmap->GetDetailShape();
  }

  int    width = 1;
  int    color = kWhite;
  int    style = 2;
  
  switch ( shape ) {
  case BfldDbiPlaneMap::kCircle: 
    {
      static TEllipse  circle;
      circle.SetLineWidth(width);
      circle.SetLineColor(color);
      circle.SetLineStyle(style);
      circle.SetFillStyle(0);  // transparent fill
      circle.DrawEllipse(0,0,radius,radius,0.,360.,0.,"");
      break;
    }
  case BfldDbiPlaneMap::kSquare:
    {
      static TPolyLine square;
      double xs[] = { -radius, -radius,  radius,  radius, -radius };
      double ys[] = { -radius,  radius,  radius, -radius, -radius };
      square.SetLineWidth(width);
      square.SetLineColor(color);
      square.SetLineStyle(style);
      square.DrawPolyLine(5,xs,ys);
      break;
    }
  case BfldDbiPlaneMap::kDiamond:
    {
      static TPolyLine diamond;
      double xd[] = { -radius,       0,  radius     ,  0, -radius };
      double yd[] = {       0,  radius,       0, -radius,       0 };
      for (unsigned int i=0; i < sizeof(xd)/sizeof(double); ++i )
        { xd[i] *= TMath::Sqrt2(); yd[i] *= TMath::Sqrt2(); }
      diamond.SetLineWidth(width);
      diamond.SetLineColor(color);
      diamond.SetLineStyle(style);
      diamond.DrawPolyLine(5,xd,yd);
      break;
    }
  default:
    cout << "illegal detail shape " << shape << endl;
  }
}

const BfldDbiPlaneMap* BfldCanvasSlice::GetPlaneMap()
{
  DbiResultPtr<BfldDbiPlaneMap> tblrows(GetVldContext());
  return tblrows.GetRow(GetPlane());
}

std::string BfldCanvasSlice::GetAdmixtureString()
{
  std::string result;

  const BfldDbiPlaneMap* plnmap = GetPlaneMap();
  if ( plnmap ) {
    int indxpair = BfldDbiPlaneMap::kFullSteelA;
    if ( GetIsGap() ) {
      if ( GetIsCoil() ) indxpair = BfldDbiPlaneMap::kDetailGapA;
      else               indxpair = BfldDbiPlaneMap::kFullGapA;
    } else {
      if ( GetIsCoil() ) indxpair = BfldDbiPlaneMap::kDetailSteelA;
      else               indxpair = BfldDbiPlaneMap::kFullSteelA;
    }
    if ( plnmap->IsNull(indxpair+1) ) {
      // not a linear combination (assume it in the first slot!)
      result += Form("map_%d",plnmap->GetMapVariant(indxpair));
    } else {
      result += Form("%5.3f*map_%d+%5.3f*map_%d",
                     plnmap->GetScale(indxpair),
                     plnmap->GetMapVariant(indxpair),
                     plnmap->GetScale(indxpair+1),
                     plnmap->GetMapVariant(indxpair+1));
    }
  }

  return result;
}

---