//*-- AUTHOR : Ilse Koenig
//*-- Modified : 22/06/2003 by Ilse Koenig

///////////////////////////////////////////////////////////////////////////////
//
// FairGeoShapes
//
// Class to manage the GEANT/ROOT geometry shapes
//
// The 3D geometry in ROOT uses the ROOT library geom.
//
// This class holds a list of shape classes (type FairGeoBasicShape). They are
// used by all detectors when reading or writing geometry parameter containers
// to ascii files or to create an event display. This class is instantiated in
// the class HSpectrometer inside the function createDetGeomPar(Text_t* name).
// Every geometry parameter container gets a pointer to it.
//
// The individual shape classes are accessible with the selectShape(...)
// functions. The functions readPoints(...) and writePoints(...) use it
// internally.
//
// In the current version the following shapes are implemented:
//     BOX  TRAP  TRD1  PGON  PCON TUBE TUBS CONE CONS SPHE  ELTU
//     TORUS is added 06.11.06  m.al-turany
//
///////////////////////////////////////////////////////////////////////////////

#include"FairGeoShapes.h"

#include "FairGeoVolume.h"
#include "FairGeoBasicShape.h"
#include "FairGeoBrik.h"
#include "FairGeoTrap.h"
#include "FairGeoTrd1.h"
#include "FairGeoPgon.h"
#include "FairGeoPcon.h"
#include "FairGeoTube.h"
#include "FairGeoTubs.h"
#include "FairGeoCone.h"
#include "FairGeoCons.h"
#include "FairGeoSphe.h"
#include "FairGeoEltu.h"
#include "FairGeoTorus.h"
#include "FairGeoAssembly.h"

#include "TList.h"

ClassImp(FairGeoShapes)

FairGeoShapes::FairGeoShapes()
  : TObject(),
    shapes(new TList())
{
  // constructor creates empty list of shapes

}


FairGeoShapes::~FairGeoShapes()
{
  // destructor deletes all shapes
  shapes->Delete();
  delete shapes;
}


FairGeoBasicShape* FairGeoShapes::selectShape(FairGeoVolume* volu)
{
  // returns a pointer to the shape used in the given volume
  // calls internally selectShape(TString&) with the name of the shape
  // returns NULL if the corresponding shape class is not implemented
  const TString& name(volu->getShape());
  return selectShape(name);
}


FairGeoBasicShape* FairGeoShapes::selectShape(const TString& name)
{
  // returns a pointer to the shape given by name
  // creates a shape object and adds it to the list of shapes if
  // not existing
  // returns NULL if the corresponding shape class is not implemented
  TString allShapes[13]= {"BOX ","TRAP","TRD1","PGON","PCON","TUBE","TUBS",
                          "CONE","CONS","SPHE","ELTU","TORUS", "ASSEMBLY"
                         };
  TString sName(name);
  if (sName.Length()==3) { sName+=" "; }
  FairGeoBasicShape* s=(FairGeoBasicShape*)shapes->FindObject(sName);
  if (s) { return s; }
  Int_t no=-1;
  for(Int_t i=0; i<13; i++) {if (sName.CompareTo(allShapes[i])==0) { no=i; }}
  switch(no) {
  case 0:
  {s= new FairGeoBrik();  break;}
  case 1:
  {s= new FairGeoTrap();  break;}
  case 2:
  {s= new FairGeoTrd1();  break;}
  case 3:
  {s= new FairGeoPgon();  break;}
  case 4:
  {s= new FairGeoPcon();  break;}
  case 5:
  {s= new FairGeoTube();  break;}
  case 6:
  {s= new FairGeoTubs();  break;}
  case 7:
  {s= new FairGeoCone();  break;}
  case 8:
  {s= new FairGeoCons();  break;}
  case 9:
  {s= new FairGeoSphe();  break;}
  case 10:
  {s= new FairGeoEltu();  break;}
  case 11:
  {s= new FairGeoTorus();  break;}
  case 12:
  {s= new FairGeoAssembly();  break;}
  default: {
    Error("selectShape","shape %s not implemented",name.Data());
  }
  }
  if (s) { shapes->Add(s); }
  return s;
}


Int_t FairGeoShapes::readPoints(fstream* pFile,FairGeoVolume* volu)
{
  // reads the points of the given volume from the Ascii file
  // returns the number of points read
  // returns 0 if if the corresponding shape class is not implemented
  FairGeoBasicShape* s=selectShape(volu);
  if (s) { return s->readPoints(pFile,volu); }
  else { return 0; }
}


Bool_t FairGeoShapes::writePoints(fstream* pFile,FairGeoVolume* volu)
{
  // writes the points of the given volume to the Ascii file
  // return kFALSE if the corresponding shape class is not implemented
  FairGeoBasicShape* s=selectShape(volu);
  if (s) { return s->writePoints(pFile,volu); }
  else { return kFALSE; }
}


void FairGeoShapes::printPoints(FairGeoVolume* volu)
{
  // writes the points of the given volume to the Ascii file
  // return kFALSE if the corresponding shape class is not implemented
  FairGeoBasicShape* s=selectShape(volu);
  if (s) { return s->printPoints(volu); }
}