#include "Riostream.h"
#include "TROOT.h"
#include "TMath.h"
#include "TArrow.h"
#include "TVirtualPad.h"
Float_t TArrow::fgDefaultAngle = 60;
Float_t TArrow::fgDefaultArrowSize = 0.05;
TString TArrow::fgDefaultOption = ">";
ClassImp(TArrow)
TArrow::TArrow(): TLine(),TAttFill()
{
fAngle = fgDefaultAngle;
}
TArrow::TArrow(Double_t x1, Double_t y1,Double_t x2, Double_t y2,
Float_t arrowsize ,Option_t *option)
:TLine(x1,y1,x2,y2), TAttFill(0,1001)
{
fAngle = fgDefaultAngle;
fArrowSize = arrowsize;
fOption = option;
SetFillColor(this->GetLineColor());
}
TArrow::~TArrow()
{
}
TArrow::TArrow(const TArrow &arrow) : TLine(arrow), TAttFill(arrow)
{
((TArrow&)arrow).Copy(*this);
}
void TArrow::Copy(TObject &obj) const
{
TLine::Copy(obj);
TAttFill::Copy(((TArrow&)obj));
((TArrow&)obj).fAngle = fAngle;
((TArrow&)obj).fArrowSize = fArrowSize;
((TArrow&)obj).fOption = fOption;
}
void TArrow::Draw(Option_t *option)
{
Option_t *opt;
if (option && strlen(option)) opt = option;
else opt = (char*)GetOption();
AppendPad(opt);
}
void TArrow::DrawArrow(Double_t x1, Double_t y1,Double_t x2, Double_t y2,
Float_t arrowsize ,Option_t *option)
{
Float_t size = arrowsize;
if (size <= 0) size = fArrowSize;
if (size <= 0) size = 0.05;
const char* opt = option;
if (!opt || strlen(opt) == 0) opt = fOption.Data();
if (!opt || strlen(opt) == 0) opt = "|>";
TArrow *newarrow = new TArrow(x1,y1,x2,y2,size,opt);
newarrow->SetAngle(fAngle);
TAttLine::Copy(*newarrow);
TAttFill::Copy(*newarrow);
newarrow->SetBit(kCanDelete);
newarrow->AppendPad(opt);
}
void TArrow::Paint(Option_t *option)
{
Option_t *opt;
if (option && strlen(option)) opt = option;
else opt = (char*)GetOption();
PaintArrow(gPad->XtoPad(fX1),gPad->YtoPad(fY1),gPad->XtoPad(fX2),gPad->YtoPad(fY2), fArrowSize, opt);
}
void TArrow::PaintArrow(Double_t x1, Double_t y1, Double_t x2, Double_t y2,
Float_t arrowsize, Option_t *option)
{
Int_t i;
TString opt = option;
opt.ToLower();
TAttLine::Modify();
TAttFill::Modify();
Int_t ix1,iy1,ix2,iy2;
Int_t iw = gPad->GetWw();
Int_t ih = gPad->GetWh();
Double_t x1p,y1p,x2p,y2p;
gPad->GetPadPar(x1p,y1p,x2p,y2p);
ix1 = (Int_t)(iw*x1p);
iy1 = (Int_t)(ih*y1p);
ix2 = (Int_t)(iw*x2p);
iy2 = (Int_t)(ih*y2p);
Double_t wndc = TMath::Min(1.,(Double_t)iw/(Double_t)ih);
Double_t hndc = TMath::Min(1.,(Double_t)ih/(Double_t)iw);
Double_t rh = hndc/(Double_t)ih;
Double_t rw = wndc/(Double_t)iw;
Double_t x1ndc = (Double_t)ix1*rw;
Double_t y1ndc = (Double_t)iy1*rh;
Double_t x2ndc = (Double_t)ix2*rw;
Double_t y2ndc = (Double_t)iy2*rh;
Double_t rx1,ry1,rx2,ry2;
gPad->GetRange(rx1,ry1,rx2,ry2);
Double_t rx = (x2ndc-x1ndc)/(rx2-rx1);
Double_t ry = (y2ndc-y1ndc)/(ry2-ry1);
Double_t x1n, y1n, x2n, y2n, xm, ym;
x1n = rx*(x1-rx1)+x1ndc;
x2n = rx*(x2-rx1)+x1ndc;
y1n = ry*(y1-ry1)+y1ndc;
y2n = ry*(y2-ry1)+y1ndc;
xm = (x1n+x2n)/2;
ym = (y1n+y2n)/2;
Double_t length = TMath::Sqrt(Double_t((x2n-x1n)*(x2n-x1n)+(y2n-y1n)*(y2n-y1n)));
Double_t rSize = 0.7*arrowsize;
Double_t dSize = rSize*TMath::Tan(TMath::Pi()*fAngle/360);
Double_t cosT = 1;
Double_t sinT = 0;
if (length > 0) {
cosT = (x2n-x1n)/length;
sinT = (y2n-y1n)/length;
}
Double_t x1ar[4], y1ar[4];
Double_t x2ar[4], y2ar[4];
if (opt.BeginsWith("|-")) {
x1ar[0] = x1n-sinT*dSize;
y1ar[0] = y1n+cosT*dSize;
x1ar[1] = x1n+sinT*dSize;
y1ar[1] = y1n-cosT*dSize;
for (i=0; i<2; i++) {
x1ar[i] = (1/rx)*(x1ar[i]-x1ndc)+rx1;
y1ar[i] = (1/ry)*(y1ar[i]-y1ndc)+ry1;
}
gPad->PaintLine(x1ar[0],y1ar[0],x1ar[1],y1ar[1]);
opt(0) = ' ';
}
if (opt.EndsWith("-|")) {
x2ar[0] = x2n-sinT*dSize;
y2ar[0] = y2n+cosT*dSize;
x2ar[1] = x2n+sinT*dSize;
y2ar[1] = y2n-cosT*dSize;
for (i=0; i<2; i++) {
x2ar[i] = (1/rx)*(x2ar[i]-x1ndc)+rx1;
y2ar[i] = (1/ry)*(y2ar[i]-y1ndc)+ry1;
}
gPad->PaintLine(x2ar[0],y2ar[0],x2ar[1],y2ar[1]);
opt(opt.Length()-1) = ' ';
}
Double_t x1h = x1n;
Double_t y1h = y1n;
Double_t x2h = x2n;
Double_t y2h = y2n;
if (opt.Contains("->-") || opt.Contains("-|>-")) {
x2h = xm + cosT*rSize/2;
y2h = ym + sinT*rSize/2;
}
if (opt.Contains("-<-") || opt.Contains("-<|-")) {
x1h = xm - cosT*rSize/2;
y1h = ym - sinT*rSize/2;
}
if (opt.Contains(">")) {
x2ar[0] = x2h - rSize*cosT - sinT*dSize;
y2ar[0] = y2h - rSize*sinT + cosT*dSize;
x2ar[1] = x2h;
y2ar[1] = y2h;
x2ar[2] = x2h - rSize*cosT + sinT*dSize;
y2ar[2] = y2h - rSize*sinT - cosT*dSize;
x2ar[3] = x2ar[0];
y2ar[3] = y2ar[0];
}
if (opt.Contains("<")) {
x1ar[0] = x1h + rSize*cosT + sinT*dSize;
y1ar[0] = y1h + rSize*sinT - cosT*dSize;
x1ar[1] = x1h;
y1ar[1] = y1h;
x1ar[2] = x1h + rSize*cosT - sinT*dSize;
y1ar[2] = y1h + rSize*sinT + cosT*dSize;
x1ar[3] = x1ar[0];
y1ar[3] = y1ar[0];
}
if (opt.Contains("|>") && !opt.Contains("-|>-")) {
x2n = x2n-cosT*rSize;
y2n = y2n-sinT*rSize;
}
if (opt.Contains("<|") && !opt.Contains("-<|-")) {
x1n = x1n+cosT*rSize;
y1n = y1n+sinT*rSize;
}
x1n = (1/rx)*(x1n-x1ndc)+rx1;
y1n = (1/ry)*(y1n-y1ndc)+ry1;
x2n = (1/rx)*(x2n-x1ndc)+rx1;
y2n = (1/ry)*(y2n-y1ndc)+ry1;
gPad->PaintLine(x1n,y1n,x2n,y2n);
if (opt.Contains(">")) {
for (i=0; i<4; i++) {
x2ar[i] = (1/rx)*(x2ar[i]-x1ndc)+rx1;
y2ar[i] = (1/ry)*(y2ar[i]-y1ndc)+ry1;
}
if (opt.Contains("|>")) {
if (GetFillColor()) {
gPad->PaintFillArea(3,x2ar,y2ar);
gPad->PaintPolyLine(4,x2ar,y2ar);
} else {
gPad->PaintPolyLine(4,x2ar,y2ar);
}
} else {
gPad->PaintPolyLine(3,x2ar,y2ar);
}
}
if (opt.Contains("<")) {
for (i=0; i<4; i++) {
x1ar[i] = (1/rx)*(x1ar[i]-x1ndc)+rx1;
y1ar[i] = (1/ry)*(y1ar[i]-y1ndc)+ry1;
}
if (opt.Contains("<|")) {
if (GetFillColor()) {
gPad->PaintFillArea(3,x1ar,y1ar);
gPad->PaintPolyLine(4,x1ar,y1ar);
} else {
gPad->PaintPolyLine(4,x1ar,y1ar);
}
} else {
gPad->PaintPolyLine(3,x1ar,y1ar);
}
}
}
void TArrow::SavePrimitive(ostream &out, Option_t * )
{
char quote = '"';
if (gROOT->ClassSaved(TArrow::Class())) {
out<<" ";
} else {
out<<" TArrow *";
}
out<<"arrow = new TArrow("<<fX1<<","<<fY1<<","<<fX2<<","<<fY2
<<","<<fArrowSize<<","<<quote<<GetDrawOption()<<quote<<");"<<endl;
SaveFillAttributes(out,"arrow",0,1);
SaveLineAttributes(out,"arrow",1,1,1);
if (fAngle !=60) {
out << " arrow->SetAngle(" << GetAngle() << ");" << endl;
}
out<<" arrow->Draw();"<<endl;
}
void TArrow::SetDefaultAngle(Float_t Angle)
{
fgDefaultAngle = Angle;
}
void TArrow::SetDefaultArrowSize (Float_t ArrowSize)
{
fgDefaultArrowSize = ArrowSize;
}
void TArrow::SetDefaultOption(Option_t *Option)
{
fgDefaultOption = Option;
}
Float_t TArrow::GetDefaultAngle()
{
return fgDefaultAngle;
}
Float_t TArrow::GetDefaultArrowSize()
{
return fgDefaultArrowSize;
}
Option_t *TArrow::GetDefaultOption()
{
return fgDefaultOption.Data();
}
Last change: Wed Jun 25 08:34:47 2008
Last generated: 2008-06-25 08:34
This page has been automatically generated. If you have any comments or suggestions about the page layout send a mail to ROOT support, or contact the developers with any questions or problems regarding ROOT.