/**************************************************************************
 * Copyright(c) 1998-2010, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */

//_________________________________________________________________________
// Main class for TRD1 geometry of Shish-Kebab case.
// Author: Alexei Pavlinov(WSU).
// Sep 20004 - Nov 2006; Apr 2010
// See web page with description of Shish-Kebab geometries:
// http://pdsfweb01.nersc.gov/~pavlinov/ALICE/SHISHKEBAB/RES/shishkebabALICE.html
// Nov 9,2006 - added case of 3X3
//_________________________________________________________________________

#include "AliLog.h"
#include "AliEMCALShishKebabTrd1Module.h"
#include "AliEMCALEMCGeometry.h"

#include <cassert>

ClassImp(AliEMCALShishKebabTrd1Module)

  //AliEMCALEMCGeometry *AliEMCALShishKebabTrd1Module::fgGeometry=0; 
  Double_t AliEMCALShishKebabTrd1Module::fga=0.; 
  Double_t AliEMCALShishKebabTrd1Module::fga2=0.; 
  Double_t AliEMCALShishKebabTrd1Module::fgb=0.; 
  Double_t AliEMCALShishKebabTrd1Module::fgr=0.; 
  Double_t AliEMCALShishKebabTrd1Module::fgangle=0.;   // around one degree 
  Double_t AliEMCALShishKebabTrd1Module::fgtanBetta=0; //

//_____________________________________________________________________________
AliEMCALShishKebabTrd1Module::AliEMCALShishKebabTrd1Module(Double_t theta, AliEMCALEMCGeometry *g) 
  : TNamed(),
    fGeometry(g),
    fOK(),
    fA(0.),
    fB(0.),
    fThetaA(0.),
    fTheta(theta),
    fOK1(),
    fOK2(),
    fOB(),
    fOB1(),
    fOB2(),
    fThetaOB1(0.),
    fThetaOB2(0.),
    fOK3X3(),
    fORB(), 
    fORT()
{ 
  TString snam(g->GetName());
  Int_t key=0;
  if(snam.Contains("FIRSTYEARv1",TString::kIgnoreCase)) key=1;
  if(snam.Contains("COMPLETEv1",TString::kIgnoreCase)) key=1;
  if(GetParameters()) {
    DefineFirstModule(key);
  }
  DefineName(fTheta);
  AliDebug(10,Form("AliEMCALShishKebabTrd1Module - first module key=%i:  theta %1.4f geometry %s\n",key,fTheta, g->GetName()));
}

//_____________________________________________________________________________
AliEMCALShishKebabTrd1Module::AliEMCALShishKebabTrd1Module(AliEMCALShishKebabTrd1Module &leftNeighbor) 
  : TNamed(),
    fGeometry(leftNeighbor.fGeometry),
    fOK(),
    fA(0.),
    fB(0.),
    fThetaA(0.),
    fTheta(0.),
    fOK1(),
    fOK2(),
    fOB(),
    fOB1(),
    fOB2(),
    fThetaOB1(0.),
    fThetaOB2(0.),
    fOK3X3(),
    fORB(), 
    fORT()
{ 
  //  printf("** Left Neighbor : %s **\n", leftNeighbor.GetName());
  fTheta  = leftNeighbor.GetTheta() - fgangle; 

  TObject::SetUniqueID(leftNeighbor.GetUniqueID()+1);

  Init(leftNeighbor.GetA(),leftNeighbor.GetB());
}

//________________________________________________________________
AliEMCALShishKebabTrd1Module::AliEMCALShishKebabTrd1Module(const AliEMCALShishKebabTrd1Module& mod) 
  : TNamed(mod.GetName(),mod.GetTitle()),
    fGeometry(mod.fGeometry),
    fOK(mod.fOK),
    fA(mod.fA),
    fB(mod.fB),
    fThetaA(mod.fThetaA),
    fTheta(mod.fTheta),
    fOK1(mod.fOK1),
    fOK2(mod.fOK2),
    fOB(mod.fOB),
    fOB1(mod.fOB1),
    fOB2(mod.fOB2),
    fThetaOB1(mod.fThetaOB1),
    fThetaOB2(mod.fThetaOB2),
    fORB(mod.fORB),
    fORT(mod.fORT)
{
  //copy ctor
  for (Int_t i=0; i<3; i++) fOK3X3[i] = mod.fOK3X3[i];
}

//________________________________________________________________
void AliEMCALShishKebabTrd1Module::Init(Double_t A, Double_t B)
{ 
  // Define parameter module from parameters A,B from previos.
  Double_t yl = (fgb/2)*TMath::Sin(fTheta) + (fga/2)*TMath::Cos(fTheta) + fgr, y = yl;
  Double_t xl = (yl - B) / A;     // y=A*x+B

  //  Double_t xp1 = (fga/2. + fgb/2.*fgtanBetta)/(TMath::Sin(fTheta) + fgtanBetta*TMath::Cos(fTheta));
  //  printf(" xp1 %9.3f \n ", xp1);
  // xp1 == xp => both methods give the same results - 3-feb-05
  Double_t alpha = TMath::Pi()/2. + fgangle/2;
  Double_t xt = (fga+fga2)*TMath::Tan(fTheta)*TMath::Tan(alpha)/(4.*(1.-TMath::Tan(fTheta)*TMath::Tan(alpha)));
  Double_t yt = xt / TMath::Tan(fTheta), xp = TMath::Sqrt(xt*xt + yt*yt);
  Double_t x  = xl + xp;
  fOK.Set(x, y);
  //  printf(" yl %9.3f | xl %9.3f | xp %9.3f \n", yl, xl, xp);

  // have to define A and B; 
  Double_t yCprev = fgr + fga*TMath::Cos(fTheta);
  Double_t xCprev = (yCprev - B) / A;
  Double_t xA     = xCprev + fga*TMath::Sin(fTheta), yA = fgr;

  fThetaA = fTheta - fgangle/2.;
  fA = TMath::Tan(fThetaA); // !!
  fB = yA - fA*xA;

  DefineAllStaff();
}

void AliEMCALShishKebabTrd1Module::DefineAllStaff()
{ 
  //Define some parameters	
  DefineName(fTheta);
  // Centers of cells - 2X2 case
  Double_t kk1 = (fga+fga2)/(2.*4.); // kk1=kk2 

  Double_t xk1 = fOK.X() - kk1*TMath::Sin(fTheta);
  Double_t yk1 = fOK.Y() + kk1*TMath::Cos(fTheta) - fgr;
  fOK1.Set(xk1,yk1);

  Double_t xk2 = fOK.X() + kk1*TMath::Sin(fTheta);
  Double_t yk2 = fOK.Y() - kk1*TMath::Cos(fTheta) - fgr;
  fOK2.Set(xk2,yk2);

  // Centers of cells - 3X3 case; Nov 9,2006
  fOK3X3[1].Set(fOK.X(), fOK.Y()-fgr); // coincide with module center

  kk1 = ((fga+fga2)/4. + fga/6.)/2.; 

  xk1 = fOK.X() - kk1*TMath::Sin(fTheta);
  yk1 = fOK.Y() + kk1*TMath::Cos(fTheta) - fgr;
  fOK3X3[0].Set(xk1,yk1);

  xk2 = fOK.X() + kk1*TMath::Sin(fTheta);
  yk2 = fOK.Y() - kk1*TMath::Cos(fTheta) - fgr;
  fOK3X3[2].Set(xk2,yk2);

  // May 15, 2006; position of module(cells) center face 
  fOB.Set(fOK.X()-fgb/2.*TMath::Cos(fTheta),  fOK.Y()-fgb/2.*TMath::Sin(fTheta)-fgr);
  fOB1.Set(fOB.X()-fga/4.*TMath::Sin(fTheta), fOB.Y()+fga/4.*TMath::Cos(fTheta));
  fOB2.Set(fOB.X()+fga/4.*TMath::Sin(fTheta), fOB.Y()-fga/4.*TMath::Cos(fTheta));
  // Jul 30, 2007 - for taking into account a position of shower maximum
  fThetaOB1 = fTheta - fgangle/4.; // ??
  fThetaOB2 = fTheta + fgangle/4.;

  // Position of right/top point of module
  // Gives the posibility to estimate SM size in z direction
  Double_t xBottom     = (fgr - fB) / fA;
  Double_t yBottom     = fgr;
  fORB.Set(xBottom, yBottom);

  Double_t l = fgb / TMath::Cos(fgangle/2.); // length of lateral module side
  Double_t xTop = xBottom + l*TMath::Cos(TMath::ATan(fA));
  Double_t yTop = fA*xTop + fB;
  fORT.Set(xTop, yTop);
}

//_____________________________________________________________________________
void AliEMCALShishKebabTrd1Module::DefineFirstModule(const Int_t key)
{
  //    Oct 23-25, 2010
  //  key=0 - zero tilt of first module;
  //  key=1 - angle=fgangle/2 = 0.75 degree.
  //  This is what we have in produced SM. 

  // Define first module
  if(key==0) {
    // theta in radians ; first object theta=pi/2.
    fTheta = TMath::PiOver2();
    fOK.Set(fga2/2., fgr + fgb/2.); // position the center of module vs o

    // parameters of right line : y = A*z + B in system where zero point is IP.
    fThetaA = fTheta - fgangle/2.;
    fA      = TMath::Tan(fThetaA);
    Double_t xA = fga/2. + fga2/2.;
    Double_t yA = fgr;
    fB = yA - fA*xA;

  } else if(key==1) {
    // theta in radians ; first object theta = 90-0.75 = 89.25 degree
    fTheta = 89.25*TMath::DegToRad();
    Double_t al1 = fgangle/2.;
    Double_t x = 0.5*(fga*TMath::Cos(al1) + fgb*TMath::Sin(al1));
    Double_t y = 0.5*(fgb + fga*TMath::Sin(al1))*TMath::Cos(al1);
    fOK.Set(x, fgr + y);
    // parameters of right line : y = A*z + B in system where zero point is IP.
    fThetaA = fTheta - fgangle/2.;
    fA      = TMath::Tan(fThetaA);
    Double_t xA = fga*TMath::Cos(al1);
    Double_t yA = fgr;
    fB = yA - fA*xA;

  } else {
    printf("<E> key=%i : wrong case \n",key);
    assert(0);
  }
  TObject::SetUniqueID(1); //

  DefineAllStaff();
}

//_____________________________________________________________________________
void AliEMCALShishKebabTrd1Module::DefineName(Double_t theta)
{
  // Define name of object
  SetName(Form("%2i(%5.2f)", TObject::GetUniqueID(), theta*TMath::RadToDeg()));
}

//_____________________________________________________________________________
Bool_t AliEMCALShishKebabTrd1Module::GetParameters()
{
  
  // Get needing module parameters from EMCAL geometry
  
  if(!fGeometry) {
    Warning("GetParameters()"," No geometry ");
    return kFALSE; 
  }  
  
  TString sn(fGeometry->GetName()); // 2-Feb-05
  sn.ToUpper();
  
  fga        = (Double_t)fGeometry->GetEtaModuleSize();
  fgb        = (Double_t)fGeometry->GetLongModuleSize();
  fgangle    = Double_t(fGeometry->GetTrd1Angle())*TMath::DegToRad();
  fgtanBetta = TMath::Tan(fgangle/2.);
  fgr        = (Double_t)fGeometry->GetIPDistance();

  fgr += fGeometry->GetSteelFrontThickness();

  fga2       = Double_t(fGeometry->Get2Trd1Dx2());
  //PH  PrintShish(0);
  return kTRUE;
}

// service methods
//_____________________________________________________________________________
void AliEMCALShishKebabTrd1Module::PrintShish(int pri) const
{
  // service method
  if(pri>=0) { 
    if(pri >= 1) {
      printf("PrintShish() \n a %7.3f:%7.3f | b %7.2f | r %7.2f \n TRD1 angle %7.6f(%5.2f) | tanBetta %7.6f", 
      fga, fga2, fgb, fgr, fgangle, fgangle*TMath::RadToDeg(), fgtanBetta);
      printf(" fTheta %f : %5.2f : cos(theta) %f\n", 
      fTheta, GetThetaInDegree(),TMath::Cos(fTheta)); 
      printf(" OK : %i |%s| theta %f :  phi = %f(%5.2f) \n", 
      GetUniqueID(), GetName(), fTheta, fOK.Phi(),fOK.Phi()*TMath::RadToDeg());
    }
    printf(" y %9.3f x %9.3f xrb %9.3f (right bottom on r=%9.3f ) \n", 
	   fOK.X(), fOK.Y(), fORB.X(),fORB.Y());
    if(pri>=2) {
      printf(" A %f B %f | fThetaA %7.6f(%5.2f)\n", fA,fB, fThetaA,fThetaA*TMath::RadToDeg());
      printf(" fOK  : X %9.4f: Y %9.4f : eta  %5.3f\n",  fOK.X(), fOK.Y(), GetEtaOfCenterOfModule());
      printf(" fOK1 : X %9.4f: Y %9.4f :   (local, ieta=2)\n", fOK1.X(), fOK1.Y());
      printf(" fOK2 : X %9.4f: Y %9.4f :   (local, ieta=1)\n\n", fOK2.X(), fOK2.Y());
      printf(" fOB  : X %9.4f: Y %9.4f \n", fOB.X(), fOB.Y());
      printf(" fOB1 : X %9.4f: Y %9.4f (local, ieta=2)\n", fOB1.X(), fOB1.Y());
      printf(" fOB2 : X %9.4f: Y %9.4f (local, ieta=1)\n", fOB2.X(), fOB2.Y());
      // 3X3 
      printf(" 3X3 \n");
      for(int ieta=0; ieta<3; ieta++) {
        printf(" fOK3X3[%i] : X %9.4f: Y %9.4f (local) \n", ieta, fOK3X3[ieta].X(), fOK3X3[ieta].Y());
      }
      //      fOK.Dump();
      GetMaxEtaOfModule(pri);
    }
  }
}

//_____________________________________________________________________________
Double_t  AliEMCALShishKebabTrd1Module::GetThetaInDegree() const 
{
  return fTheta*TMath::RadToDeg();
}

//_____________________________________________________________________________
Double_t  AliEMCALShishKebabTrd1Module::GetEtaOfCenterOfModule() const 
{ 
  return -TMath::Log(TMath::Tan(fOK.Phi()/2.));
}

//_____________________________________________________________________________
void AliEMCALShishKebabTrd1Module::GetPositionAtCenterCellLine(Int_t ieta, Double_t dist, TVector2 &v)
{
  // Jul 30, 2007
  static Double_t theta=0., x=0., y=0.;
  if(ieta==0) {    
    v     = fOB2;
    theta = fTheta;
  } else if(ieta==1) {
    v     = fOB1;
    theta = fTheta;
  } else {
    assert(0);
  }
  
  x = v.X() + TMath::Cos(theta) * dist;
  y = v.Y() + TMath::Sin(theta) * dist;
  //  printf(" GetPositionAtCenterCellLine() %s : dist %f : ieta %i : x %f %f v.X() | y %f %f v.Y() : cos %f sin %f \n", 
  //GetName(), dist, ieta, v.X(),x, y,v.Y(),TMath::Cos(theta),TMath::Sin(theta));
  v.Set(x,y);
}  


//_____________________________________________________________________________
Double_t  AliEMCALShishKebabTrd1Module::GetMaxEtaOfModule(int pri) const 
{ 
  // Right bottom point of module
  Double_t thetaBottom = TMath::ATan2(fORB.Y(),fORB.X());
  Double_t etaBottom   = ThetaToEta(thetaBottom);
  // Right top point of module
  Double_t thetaTop = TMath::ATan2(fORT.Y(),fORT.X());
  Double_t etaTop   = ThetaToEta(thetaTop);

  if(pri) { 
    printf(" Right bottom point of module : eta %5.4f : theta %6.4f (%6.2f) : x(zglob) %7.2f y(phi) %5.2f \n", 
	   etaBottom, thetaBottom, thetaBottom * TMath::RadToDeg(), fORB.X(), fORB.Y());
    printf(" Right    top point of module : eta %5.4f : theta %6.4f (%6.2f) : x(zglob) %7.2f y(phi) %5.2f \n", 
	   etaTop, thetaTop, thetaTop * TMath::RadToDeg(), fORT.X(), fORT.Y());
  }
  return etaBottom>etaTop ? etaBottom : etaTop;
}