//*-- AUTHOR : Ilse Koenig
//*-- Created : 10/11/03 by Ilse Koenig

///////////////////////////////////////////////////////////////////////////////
// FairGeoRootBuilder
//
// Class to create the geometry in ROOT
//
///////////////////////////////////////////////////////////////////////////////

#include "FairGeoRootBuilder.h"

#include "FairGeoMedium.h"
#include "FairGeoNode.h"
#include "FairGeoTransform.h"

#include "TGeoManager.h"
#include "TGeoMedium.h"
#include "TGeoMaterial.h"
#include "TGeoVolume.h"
#include "TGeoMatrix.h"
//#include "TGeoPgon.h"
//#include "TGeoPcon.h"
#include "TArrayD.h"

ClassImp(FairGeoRootBuilder)

FairGeoRootBuilder::FairGeoRootBuilder()
  : FairGeoBuilder(),
    geoManager(NULL)
{
  // Default constructor
}

FairGeoRootBuilder::FairGeoRootBuilder(const char* name,const char* title)
  : FairGeoBuilder(name,title),
    geoManager(NULL)
{
  // Constructor
}

Bool_t FairGeoRootBuilder::createNode(FairGeoNode* volu, Int_t hadFormat)
{
  // Creates the volume
  if (!geoManager||!volu) { return kFALSE; }
  //  volu->print();

  TString nodeName= volu->GetName();
  if ( hadFormat == 1 ) {
    volu->setHadFormat( hadFormat );
    nodeName=nodeName(0,4);
  } else {
    nodeName = (volu->GetName());
    Ssiz_t l=nodeName.Last('#');
    if (l>0) { nodeName=nodeName(0,l); }
  }

  FairGeoNode* mother=volu->getMotherNode();
  if (!mother&&!volu->isTopNode()) {
    Error("createNode","Mother volume of %s not found\n",volu->GetName());
    return kFALSE;
  }





  TGeoVolume* rv=0;
  FairGeoNode* cv=volu->getCopyNode();
//  if (cv) cout<<"Copy of "<<cv->GetName()<<endl;
  if (cv) { rv=cv->getRootVolume(); }
  if (!rv) {
    FairGeoMedium* lmedium=volu->getMedium();
    Int_t lnMed=lmedium->getMediumIndex();
    if (lnMed<=0) { lnMed=createMedium(lmedium); }
    if (lnMed<=0) { return kFALSE; }
    TArrayD* par=volu->getParameters();
    //for (Int_t k=0;k<par->GetSize();k++) cout<<par->At(k)<<" ";
    // cout<<endl;
    //cout << "Create Volume of Shape " << volu->getShape() << endl;
    if (volu->getShape().Contains("TORUS")) { // Torus is missing in the TGeoManager::Volume
      // this should solve the problem tell it is implimented in root
      TGeoMedium* medium =geoManager->GetMedium(lnMed);
      rv = geoManager->MakeTorus(nodeName.Data(),medium,par->At(0),par->At(1),par->At(2),par->At(3),par->At(4));
      // cout << "Create Torus" << nodeName.Data() << endl;
    } else if (volu->getShape().Contains("ASSEMBLY")) {
      rv=geoManager->MakeVolumeAssembly(nodeName.Data());

    } else {
      rv=geoManager->Volume(nodeName.Data(),volu->getShape().Data(),
                            lnMed,par->GetArray(),par->GetSize());
    }
    volu->setCreated();
    if (volu->isModule()&&cv) {
      cv->setCreated();
      cv->setRootVolume(rv);
      cv->getPosition();
    }
  }
  if (!rv) { return kFALSE; }
  volu->setRootVolume(rv);
  if (volu->isTopNode()) { geoManager->SetTopVolume(rv); }
  else {
    FairGeoTransform* trans=volu->getPosition();
    const FairGeoRotation& rot=trans->getRotMatrix();
    const FairGeoVector& pos=trans->getTransVector();
    TGeoMatrix* tr=0;
    if (((FairGeoRotation&)rot).isUnitMatrix()) {
      tr=new TGeoTranslation(pos.getX(),pos.getY(),pos.getZ());
    } else {
      nRot++;
      char b[10];
      sprintf(b,"R%i",nRot);
      TGeoRotation* r=new TGeoRotation(b);
      Double_t a[9];
      for (Int_t i=0; i<9; i++) { a[i]=rot(i); }
      r->SetMatrix(a);
      tr=new TGeoCombiTrans(pos.getX(),pos.getY(),pos.getZ(),r);
    }
    TGeoVolume* mo=mother->getRootVolume();
    if (!mo) { return kFALSE; }
    mo->AddNode(rv,volu->getCopyNo(),tr);
  }
  return kTRUE;
}

Int_t FairGeoRootBuilder::createMedium(FairGeoMedium* med)
{
  // Creates the medium
  if (!geoManager&&!med) { return 0; }
  Int_t nComp=med->getNComponents();
  Int_t weightFac=med->getWeightFac();
  TGeoMaterial* material=0;
  Double_t p[3];
  if (nComp==1) {
    med->getComponent(0,p);
    material=new TGeoMaterial(med->GetName(),p[0],p[1],med->getDensity(),
                              med->getRadiationLength());
    // Interaction length not defined!!!!!!
  } else {
    material=new TGeoMixture(med->GetName(),nComp,med->getDensity());
    Double_t sumWeights=0.;
    if (weightFac<0) {
      for(Int_t i=0; i<nComp; i++) {
        med->getComponent(i,p);
        sumWeights+=p[0]*p[2];
      }
    }
    for(Int_t i=0; i<nComp; i++) {
      med->getComponent(i,p);
      if (weightFac>0) {
        ((TGeoMixture*)material)->DefineElement(i,p[0],p[1],p[2]);
      } else {
        ((TGeoMixture*)material)->DefineElement(i,p[0],p[1],p[0]*p[2]/sumWeights);
      }
    }
  }
  nMed++;
  med->setMediumIndex(nMed);
  Double_t mp[10];
  med->getMediumPar(mp);
  TGeoMedium* medium=new TGeoMedium(med->GetName(),nMed,material,mp);
  if (medium) { return nMed; }
  else { return 0; }
}

void FairGeoRootBuilder::finalize()
{
  // Closes the geometry input in ROOT and draws the cave
  if (geoManager) {
    geoManager->CloseGeometry();
    geoManager->SetVisLevel(5);
    geoManager->GetTopVolume()->Draw();
  }
}

void FairGeoRootBuilder::checkOverlaps(Double_t ovlp)
{
  // Checks the geometry for overlaps and extrusions wit a default precision of 1 micron
  if (geoManager) {
    geoManager->CheckOverlaps(ovlp,"SAME");
    geoManager->PrintOverlaps();
  }
}