#include "hgeomrootbuilder.h"
#include "hgeommedium.h"
#include "hgeomnode.h"
#include "hgeomtransform.h"
#include "hgeommdcwire.h"
#include "TGeoManager.h"
#include "TGeoMedium.h"
#include "TGeoMaterial.h"
#include "TGeoVolume.h"
#include "TGeoMatrix.h"
#include "TGeoPgon.h"
#include "TGeoPcon.h"
ClassImp(HGeomRootBuilder)
HGeomRootBuilder::HGeomRootBuilder() {
geoManager=0;
}
HGeomRootBuilder::HGeomRootBuilder(const Char_t* name,const Char_t* title)
: HGeomBuilder(name,title) {
geoManager=0;
}
Bool_t HGeomRootBuilder::createNode(HGeomNode* volu) {
if (!geoManager||!volu) return kFALSE;
TString nodeName(volu->GetName(),4);
HGeomNode* mother=volu->getMotherNode();
if (!mother&&!volu->isTopNode()) {
Error("createNode","Mother volume of %s not found\n",volu->GetName());
return kFALSE;
}
TGeoVolume* rv=0;
HGeomNode* cv=volu->getCopyNode();
if (cv) rv=cv->getRootVolume();
if (!rv) {
HGeomMedium* medium=volu->getMedium();
Int_t nMed=medium->getMediumIndex();
if (nMed<=0) nMed=createMedium(medium);
if (nMed<=0) return kFALSE;
TArrayD* par=volu->getParameters();
rv=geoManager->Volume(nodeName.Data(),volu->getShape().Data(),
nMed,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 {
HGeomTransform* trans=volu->getPosition();
const HGeomRotation& rot=trans->getRotMatrix();
const HGeomVector& pos=trans->getTransVector();
TGeoMatrix* tr=0;
if (((HGeomRotation&)rot).isUnitMatrix())
tr=new TGeoTranslation(pos.getX(),pos.getY(),pos.getZ());
else {
nRot++;
Char_t 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 HGeomRootBuilder::createMedium(HGeomMedium* med) {
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());
} 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 HGeomRootBuilder::finalize() {
if (geoManager) {
geoManager->CloseGeometry();
geoManager->SetVisLevel(2);
geoManager->GetTopVolume()->Draw();
}
}
void HGeomRootBuilder::checkOverlaps(Double_t ovlp) {
if (geoManager) {
geoManager->CheckOverlaps(ovlp,"SAME");
geoManager->PrintOverlaps();
}
}
Bool_t HGeomRootBuilder::createVolume(HGeomMdcWire* pWire, Int_t mediumIndex) {
if (!pWire || mediumIndex<0) return kFALSE;
TGeoVolume* rv = NULL;
if (pWire->getCopyNumber() == 1) {
rv=geoManager->Volume(pWire->getWireName().Data(), "TUBE", mediumIndex, pWire->getParams(), pWire->getNParams());
}
if (rv) {
pWire->setRootVolume(rv);
return kTRUE;
}
return kFALSE;
}
Int_t HGeomRootBuilder::createRotation(HGeomRotation* rot) {
if (!rot) return -1;
Int_t rotInd=0;
if (rot->isUnitMatrix()==kFALSE) {
nRot++;
Char_t b[10];
sprintf(b,"R%i",nRot);
TGeoRotation* prot = new TGeoRotation(b);
Double_t a[9];
for (Int_t i=0;i<9;i++) a[i] = (*rot)(i);
prot->SetMatrix(a);
prot->RegisterYourself();
rotInd=geoManager->GetListOfMatrices()->GetEntriesFast() -1;
}
return rotInd;
}
Bool_t HGeomRootBuilder::positionNode(HGeomMdcWire* pWire, HGeomNode* pMother, Int_t rotInd) {
if (!pWire || !pMother) return kFALSE;
TGeoVolume* volu = NULL;
if (pWire->getCopyNumber() > 1 && pWire->getRootVolume() == NULL) {
HGeomMdcWire* node = pWire->getCopyNode();
if (node) volu = node->getRootVolume();
pWire->setRootVolume(volu);
} else {
volu = pWire->getRootVolume();
}
TGeoVolume* plane = pMother->getRootVolume();
if (!volu || !plane) return kFALSE;
Double_t x = 0., y = 0., z = 0.;
pWire->getPosition(x, y, z);
TGeoMatrix* tr = NULL;
if (rotInd>0) {
TGeoMatrix* prot = (TGeoMatrix*)(geoManager->GetListOfMatrices()->At(rotInd));
if (prot && prot->IsRotation()) {
tr = new TGeoCombiTrans(x, y, z, (TGeoRotation*)prot);
} else return kFALSE;
} else {
tr=new TGeoTranslation(x, y, z);
}
if (!tr) return kFALSE;
plane->AddNode(volu, pWire->getCopyNumber(), tr);
return kTRUE;
}