#include "TROOT.h"
#include "TStyle.h"
#include "TCanvas.h"
#include "TView.h"
#include "TPolyLine3D.h"
#include "TLatex.h"

void right_tri_solid_angle_cal(){
  gROOT->SetStyle("Plain");
  gStyle->SetTextFont(132);

  Double_t cwidth  = 400;
  Double_t cheight = 400;
  TCanvas* c1 = new TCanvas("c1", "c1", cwidth, cheight);
  c1->SetWindowSize(cwidth + (cwidth - c1->GetWw()), cheight + (cheight - c1->GetWh()));

  c1->SetTheta(25.);
  c1->SetPhi(-125.);
  TView *view = TView::CreateView(1);
  Double_t view_size = 1.5;
  view->SetRange(-view_size*1.0,-view_size*1.0,-view_size*1.0,
		  view_size*1.0, view_size*1.0, view_size*1.0);

  const Int_t nPolyLine3D = 20;
  TPolyLine3D *guide[nPolyLine3D];
  for (Int_t i = 0; i<nPolyLine3D; i++){
    guide[i]=new TPolyLine3D();
  }

  Double_t pi = acos(-1.);
  Int_t    n = 60;
  Double_t theta0 = pi/5.0;
  Double_t phi0   = pi/4.;
  Double_t psi0   = asin(sqrt(1-pow(cos(theta0)*cos(phi0),2)));
  
  Double_t r0 =  2.4;
  Double_t x0 =  r0*sin(theta0);
  Double_t y0 =  r0*sin(phi0);

  Double_t gtheta = -c1->GetTheta()*pi/180.;
  Double_t gphi   = -c1->GetPhi()*pi/180.;  
  for (Int_t i=0;i<=n;i++) {
    Double_t r = r0;
    Double_t theta = i*2.*pi/n;
    Double_t x = r*cos(theta);
    Double_t z = r*sin(theta);
    guide[1]->SetNextPoint(x*cos(gphi)+z*sin(gtheta)*sin(gphi),
			   x*sin(gphi)-z*sin(gtheta)*cos(gphi),
			   z*cos(gtheta));
  }

  for (Int_t i=0;i<=n;i++) {
    Double_t r = 0.6;
    Double_t theta = -i*theta0/n+pi;
    Double_t phi   = -i*phi0/n+pi;
    guide[2]->SetNextPoint(r*cos(theta),r*sin(theta),0);
    guide[3]->SetNextPoint(r*cos(phi),0,r*sin(phi));
  }

  for (Int_t i=0;i<=n;i++) {
    Double_t r = r0;
    Double_t theta = -i*theta0/n+pi;
    Double_t phi   = -i*phi0/n+pi;
    guide[4]->SetNextPoint(r*cos(theta),r*sin(theta),0);
    guide[5]->SetNextPoint(r*cos(phi),0,r*sin(phi));
  }

  /*
    [ cos(t)  sin(t) 0] [1   0      0     ] [x]
    [-sin(t)  cos(t) 0] [0   cos(p) sin(p)] [y]
    [0        0      1] [0  -sin(p) cos(p)] [0]

    [ cos(t) sin(t) 0] [x]
    [-sin(t) cos(t) 0] [cos(p)*y]
    [ 0      0      1] [-sin(p)*y]

    [ cos(t)*x +sin(t)*cos(p)*y]
    [-sin(t)*x +cos(t)*cos(p)*y]
    [ sin(p)*y]
   */

  for (Int_t i=0;i<=n;i++) {
    Double_t r = r0;
    Double_t psi = i*psi0/n+pi;
    Double_t phi = -asin(sin(phi0)/
			  sqrt(1-pow(cos(phi0)*cos(theta0),2)));
    Double_t x = r*cos(psi);
    Double_t y = r*sin(psi);
    guide[6]->SetNextPoint( x*cos(theta0)+y*sin(theta0)*cos(phi),
			    -x*sin(theta0)+y*cos(theta0)*cos(phi),
			     y*sin(phi));
  }

  guide[7]->SetNextPoint(-1.4,1.4*tan(theta0),0);
  guide[7]->SetNextPoint(-1.4,0,0);
  guide[7]->SetNextPoint(-1.4,0,1.4*tan(phi0));
  guide[7]->SetNextPoint(-1.4,1.4*tan(theta0),0);

  guide[4]->SetLineWidth(5);
  guide[5]->SetLineWidth(5);
  guide[6]->SetLineWidth(5);
  guide[7]->SetLineWidth(5);

  for (Int_t i=0;i<=n;i++) {
    Double_t r = r0;
    Double_t theta = i*2.*pi/n;
    guide[8]->SetNextPoint(r*cos(theta),r*sin(theta),0);
    guide[9]->SetNextPoint(r*cos(theta),0,r*sin(theta));

    Double_t phi   = -asin(sin(phi0)/
			  sqrt(1-pow(cos(phi0)*cos(theta0),2)));
    Double_t x = r*cos(theta);
    Double_t y = r*sin(theta);
    guide[10]->SetNextPoint( x*cos(theta0)+y*sin(theta0)*cos(phi),
			    -x*sin(theta0)+y*cos(theta0)*cos(phi),
			     y*sin(phi));
  }
  guide[8]->SetLineStyle(2);
  guide[9]->SetLineStyle(2);
  guide[10]->SetLineStyle(2);

  guide[11]->SetNextPoint(-r0*cos(theta0),x0,0);
  guide[11]->SetNextPoint(0,0,0);
  guide[11]->SetNextPoint(-r0,0,0);
  guide[12]->SetNextPoint(-r0*cos(phi0),0,y0);
  guide[12]->SetNextPoint(0,0,0);


  Double_t virlen = 0.2;
  guide[13]->SetNextPoint(-r0,virlen,0);
  guide[13]->SetNextPoint(-r0,virlen,virlen);
  guide[13]->SetNextPoint(-r0,0,virlen);
  guide[14]->SetNextPoint(r0,virlen,0);
  guide[14]->SetNextPoint(r0,virlen,virlen);
  guide[14]->SetNextPoint(r0,0,virlen);

  guide[15]->SetNextPoint(-1.4,virlen,0);
  guide[15]->SetNextPoint(-1.4,virlen,virlen);
  guide[15]->SetNextPoint(-1.4,0,virlen);
  guide[16]->SetNextPoint(-1.4,virlen,0);
  guide[16]->SetNextPoint(-1.4+virlen,virlen,0);
  guide[16]->SetNextPoint(-1.4+virlen,0,0);
  guide[17]->SetNextPoint(-1.4,0,virlen);
  guide[17]->SetNextPoint(-1.4+virlen,0,virlen);
  guide[17]->SetNextPoint(-1.4+virlen,0,0);


  for (Int_t i = 0; i<nPolyLine3D; i++){
    guide[i]->Draw();
  }

  Double_t xx[3],xxt[3];
  TLatex Tl;
  Tl.SetTextSize(0.06);
  xx[0]=0.1; xx[1]=0.0; xx[2]=-0.25; view->WCtoNDC(xx,xxt);
  Tl.DrawLatex(xxt[0],xxt[1],"O");
  xx[0]=-0.6; xx[1]=0.3; xx[2]=0.00; view->WCtoNDC(xx,xxt);
  Tl.DrawLatex(xxt[0],xxt[1],"#it{#theta}");
  xx[0]=-0.6; xx[1]=0.0; xx[2]=0.2; view->WCtoNDC(xx,xxt);
  Tl.DrawLatex(xxt[0],xxt[1],"#it{#varphi}");
}