G3Node.cxx

Go to the documentation of this file.

/* *************************************************************************
 * Copyright(c) 1998-1999, 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.                  *
 **************************************************************************/

/*
Revision 1.3  2004/01/28 08:17:52  brun
Reintroduce the Geant3 graphics classes (thanks Andreas Morsch)

Revision 1.1.1.1  2002/07/24 15:56:26  rdm
initial import into CVS

*/

//
// G3 Node Class for the G3 GUI
// Author: Andreas Morsch
// andreas.morsch@cern.ch
//


#include "G3Node.h"
#include "TList.h"
#include "TShape.h"
#include "TTUBE.h"
#include "TBRIK.h"
#include "TTRD1.h"
#include "TTRD2.h"
#include "TTRAP.h"
#include "TTUBS.h"
#include "TCONE.h"
#include "TCONS.h"
#include "TSPHE.h"
#include "TPARA.h"
#include "TPGON.h"
#include "TPCON.h"
#include "TTUBS.h"
#include "TELTU.h"
#include "THYPE.h"
#include "TGTRA.h"
#include "TCTUB.h"

ClassImp(G3Node)
    G3Node::G3Node(const char* name, const char* title, const char* shapename,
                     Double_t x, Double_t y, Double_t z, const char* matrixname,
                     Option_t* option) : 
        TNode(name, title, shapename, x, y, z, matrixname, option)
{
    fNDivision   = -1;
    fAxis   =  0;
    fStartC = 0.;
    fStep   = 0.;
}

  G3Node::G3Node(const char* name, const char* title, TShape* shape,
                     Double_t x, Double_t y, Double_t z, TRotMatrix* matrix,
                     Option_t* option) : 
        TNode(name, title, shape, x, y, z, matrix, option)
{
    fNDivision   = -1;
    fAxis   =  0;
    fStartC = 0.;
    fStep   = 0.;
}

void   G3Node::SetDivision(Int_t ndiv, Int_t axis, Float_t start, Float_t step)
{
    fNDivision   = ndiv;
    fAxis   = axis;
    fStartC = start;
    fStep   = step;
}

void G3Node::ExpandDivisions()
{
    Int_t i;
    char vName[20];
    char nName[20];
    
    char tmp[4];
    G3Node* node;
    TShape*  parent =  fParent->GetShape();
    TShape*  newsh;
    
    strcpy(tmp, parent->GetTitle());
    Int_t ndiv = fNDivision;

//                                      TUBE
    
    if (strcmp(tmp, "TUBE")==0) {
        TTUBE * shape = (TTUBE*) parent;

        Float_t dZ   = shape->GetDz();
        Float_t rMin = shape->GetRmin();
        Float_t rMax = shape->GetRmax();
        
        if (fAxis == 1) {
//  radial-division
            Float_t dr = (rMax-rMin)/Float_t(fNDivision);
            Float_t r1, r2;
            for (i=0; i<ndiv; i++) {
                r1 = rMin+Float_t(i)*dr;
                r2 = r1+dr;
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TTUBE(vName, "TUBE", "void", r1, r2, dZ);
                fParent->cd();
                node = new G3Node(nName,"", newsh, 0., 0., 0.);
                node->AddSons(fNodes);
                cd();
            }
    
        } else if (fAxis == 2) {
//  phi-division
            Float_t dPhi = 360./Float_t(fNDivision);
            Float_t phi1, phi2;
            for (i=0; i<ndiv; i++) {
                phi1 = Float_t(i)*dPhi;
                phi2 = phi1+dPhi;
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TTUBS(vName, "TUBS", "void", rMin, rMax, dZ, phi1, phi2);
                fParent->cd();
                node = new G3Node(nName, "", newsh, 0., 0., 0.);
                node->AddSons(fNodes);
                cd();
            }
        } else {
//  z-division
            Float_t delZ = dZ/Float_t(fNDivision);
            for (i=0; i<ndiv; i++) {
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TTUBE(vName, "TUBE", "void", rMin, rMax, delZ);
                fParent->cd();
                Float_t zpos = -dZ+delZ*(2.*Float_t(i)+1.);
                node = new G3Node(nName, "",newsh, 0., 0., zpos);
                node->AddSons(fNodes);
                cd();
            }
        }
//
//                            TUBS
//
    } else if (strcmp(tmp, "TUBS")==0) {
        TTUBS * shape = (TTUBS*) parent;
        Float_t dZ   = shape->GetDz();
        Float_t rMin = shape->GetRmin();
        Float_t rMax = shape->GetRmax();
        Float_t phi1 = shape->GetPhi1();
        Float_t phi2 = shape->GetPhi2();
        
        if (fAxis == 1) {
//  radial-division
            Float_t dr = (rMax-rMin)/Float_t(fNDivision);
            Float_t r1, r2;
            Int_t ndiv = fNDivision;
            for (i=0; i<ndiv; i++) {
                r1 = rMin+Float_t(i)*dr;
                r2 = r1+dr;
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TTUBS(vName, "TUBS", "void", r1, r2, dZ, phi1, phi2);
                fParent->cd();
                node = new G3Node(nName,"", newsh, 0., 0., 0.);
                node->AddSons(fNodes);
                cd();
            }
            
        } else if (fAxis == 2) {
//  phi-division
            Float_t dPhi = (phi2-phi1)/Float_t(fNDivision);
            Float_t nphi1, nphi2;
            
            for (i=0; i<fNDivision; i++) {
                nphi1 = phi1+Float_t(i)*dPhi;
                nphi2 = nphi1+dPhi;
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);

                newsh = new TTUBS(vName, "TUBS", "void", rMin, rMax, dZ, nphi1, nphi2);
                fParent->cd();
                node = new G3Node(nName, "", newsh, 0., 0., 0.);
                node->AddSons(fNodes);
                cd();
            }
        } else {
//  z-division
            Float_t delZ = dZ/Float_t(fNDivision);
            for (i=0; i<ndiv; i++) {
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TTUBS(vName, "TUBS", "void", rMin, rMax, delZ, phi1, phi2);
                fParent->cd();
                Float_t zpos = -dZ+delZ*(2.*Float_t(i)+1.);
                node = new G3Node(nName, "",newsh, 0., 0., zpos);
                node->AddSons(fNodes);
                cd();
            }
        }
        
    } else if (strcmp(tmp, "CONE")==0) {
        TCONE * shape = (TCONE*) parent;

        Float_t dZ   = shape->GetDz();
        Float_t rMin1 = shape->GetRmin();
        Float_t rMax1 = shape->GetRmax();
        Float_t rMin2 = shape->GetRmin2();
        Float_t rMax2 = shape->GetRmax2();
        
        if (fAxis == 1) {
//  radial-division
            Float_t dr1 = (rMax1-rMin1)/Float_t(fNDivision);
            Float_t dr2 = (rMax2-rMin2)/Float_t(fNDivision);
            Float_t r11, r12, r21, r22;
            for (i=0; i<ndiv; i++) {
                r11 = rMin1+Float_t(i)*dr1;
                r12 = r11+dr1;
                r21 = rMin2+Float_t(i)*dr2;
                r22 = r21+dr2;

                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TCONE(vName, "CONE", "void", dZ, r11, r12, r21, r22);
                fParent->cd();
                node = new G3Node(nName,"", newsh, 0., 0., 0.);
                node->AddSons(fNodes);
                cd();
            }
        } else if (fAxis == 2) {
//  phi-division
            Float_t dPhi = 360./Float_t(fNDivision);
            Float_t phi1, phi2;
            for (i=0; i<ndiv; i++) {
                phi1 = Float_t(i)*dPhi;
                phi2 = phi1+dPhi;
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TCONS(vName, "CONS", "void", dZ, rMin1, rMax1, 
                          rMin2, rMax2, phi1, phi2);
                fParent->cd();
                node = new G3Node(nName, "",newsh, 0., 0., 0.);
                node->AddSons(fNodes);
                cd();
            }
        } else {
//  z-division
            Float_t delZ = dZ/Float_t(fNDivision);
            for (i=0; i<ndiv; i++) {
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TCONE(vName, "CONE", "void", delZ, rMin1, rMax1, rMin2, rMax2);
                fParent->cd();
                Float_t zpos = -dZ+delZ*(2.*Float_t(i)+1.);
                node = new G3Node(nName, "",newsh, 0., 0., zpos);
                node->AddSons(fNodes);
                cd();
            }
        }
        
    } else if (strcmp(tmp, "CONS")==0) {
        TCONS * shape = (TCONS*) parent;
        Float_t dZ   = shape->GetDz();
        Float_t rMin1 = shape->GetRmin();
        Float_t rMax1 = shape->GetRmax();
        Float_t rMin2 = shape->GetRmin2();
        Float_t rMax2 = shape->GetRmax2();
        Float_t phi1 = shape->GetPhi1();
        Float_t phi2 = shape->GetPhi2();
        if (fAxis == 1) {
//  radial-division
            Float_t dr1 = (rMax1-rMin1)/Float_t(fNDivision);
            Float_t dr2 = (rMax2-rMin2)/Float_t(fNDivision);
            Float_t r11, r12, r21, r22;
            for (i=0; i<ndiv; i++) {
                r11 = rMin1+Float_t(i)*dr1;
                r12 = r11+dr1;
                r21 = rMin2+Float_t(i)*dr2;
                r22 = r21+dr2;

                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TCONS(vName, "CONS", "void", dZ, r11, r12, r21, r22, phi1, phi2);
                fParent->cd();
                node = new G3Node(nName,"", newsh, 0., 0., 0.);
                node->AddSons(fNodes);
                cd();
            }
            
        } else if (fAxis == 2) {
//  phi-division
            Float_t dPhi = (phi2-phi1)/Float_t(fNDivision);
            Float_t nphi1, nphi2;
            
            for (i=0; i<fNDivision; i++) {
                nphi1 = phi1+Float_t(i)*dPhi;
                nphi2 = nphi1+dPhi;
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);

                newsh = new TCONS(vName, "CONS", "void", dZ, rMin1, rMax1, rMin2, rMax2, nphi1, nphi2);
                fParent->cd();
                node = new G3Node(nName, "", newsh, 0., 0., 0.);
                node->AddSons(fNodes);
                cd();
            }
        } else {
//  z-division
            Float_t delZ = dZ/Float_t(fNDivision);
            for (i=0; i<ndiv; i++) {
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TCONS(vName, "CONS", "void", delZ, rMin1, rMax1, rMin2, rMax2, phi1, phi2);
                fParent->cd();
                Float_t zpos = -dZ+delZ*(2.*Float_t(i)+1.);
                node = new G3Node(nName,"",newsh, 0., 0., zpos);
                node->AddSons(fNodes);
                cd();
            }
        }
    } else if (strcmp(tmp, "BRIK")==0) {
//
//                          BRIK
//
        TBRIK * shape = (TBRIK*) parent;
        Float_t dX   = shape->GetDx();
        Float_t dY   = shape->GetDy();
        Float_t dZ   = shape->GetDz();

        if (fAxis == 1) {
// division in x
            Float_t delX = dX/Float_t(fNDivision);
            for (i=0; i<ndiv; i++) {
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TBRIK(vName, "BRIK", "void", delX, dY, dZ);
                fParent->cd();
                Float_t xpos = -dX+delX*(2.*Float_t(i)+1.);
                node = new G3Node(nName,"",newsh, xpos, 0., 0.);
                node->AddSons(fNodes);
                cd();
            }
        } else if (fAxis == 2) {
// division in y
            Float_t delY = dY/Float_t(fNDivision);
            for (i=0; i<ndiv; i++) {
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TBRIK(vName, "BRIK", "void", dX, delY, dZ);
                fParent->cd();
                Float_t ypos = -dY+delY*(2.*Float_t(i)+1.);
                node = new G3Node(nName,"",newsh, 0., ypos, 0.);
                node->AddSons(fNodes);
                cd();
            }
        } else {
// division in z
            Float_t delZ = dZ/Float_t(fNDivision);
            for (i=0; i<ndiv; i++) {
                sprintf(vName, "%sD%d", fShape->GetName(), i);
                sprintf(nName, "%sD%d", GetName(), i);
                newsh = new TBRIK(vName, "BRIK", "void", dX, dY, delZ);
                fParent->cd();
                Float_t zpos = -dZ+delZ*(2.*Float_t(i)+1.);
                node = new G3Node(nName,"",newsh, 0., 0., zpos);
                node->AddSons(fNodes);
                cd();
            }
        }
    }
}

void G3Node::AddSons(TList* list)
{
    if (!list) return;
    if (!fNodes) fNodes = new TList();
    
    TIter next(list);
    G3Node* node;
    
    while((node = (G3Node*)next())) {
        G3Node* newNode = new G3Node(*node, this);
        fNodes->Add(newNode);
        newNode->SetParent(this);
        newNode->AddSons(node->GetListOfNodes());
    }
}


G3Node::G3Node(const G3Node &node, G3Node* parent)
{
    fNDivision   = node.Ndiv();
    fAxis        = node.Axis();
    fStartC      = node.StartC();
    fStep        = node.Step();
    fName        = node.GetName();
    fTitle       = node.GetTitle();
    fX           = node.GetX();
    fY           = node.GetY();
    fZ           = node.GetZ();
    fMatrix      = node.GetMatrix();
    fNodes       = 0;
    fParent      = parent;
    
    
    if (fNDivision > 0) {
        fShape = new TShape(*node.GetShape());
    } else {
        fShape = (TShape*) node.GetShape()->Clone();
    }
    
    
//    fShape = (TShape*) shape->Clone();
/*
    char tmp[4];        
    strcpy(tmp, shape->ClassName());

    if (strcmp(tmp, "TTUBE")==0) 
        fShape =  (TTUBE*)shape->Clone();
    else if (strcmp(tmp, "TBRIK")==0) 
        fShape =  (TBRIK*)shape->Clone();
    else if (strcmp(tmp, "TCONE")==0) 
        fShape =  (TCONE*)shape->Clone();
    else if (strcmp(tmp, "TCONS")==0) 
        fShape =  (TCONS*)shape->Clone();
    else if (strcmp(tmp, "TTUBS")==0) 
        fShape =  (TTUBS*)shape->Clone();
    else if (strcmp(tmp, "TTRAP")==0) 
        fShape =  (TTRAP*)shape->Clone();
    else if (strcmp(tmp, "TTRD1")==0) 
        fShape =  (TTRD1*)shape->Clone();
    else if (strcmp(tmp, "TTRD2")==0)
        fShape =  (TTRD2*)shape->Clone();
    else if (strcmp(tmp, "TSPHE")==0) 
        fShape =  (TSPHE*)shape->Clone();
    else if (strcmp(tmp, "TPGON")==0) 
        fShape =  (TPGON*)shape->Clone();
    else if (strcmp(tmp, "TPCON")==0) 
        fShape =  (TPCON*)shape->Clone();
*/
}

void G3Node::AddSon(G3Node* node)
{
    fNodes->Add(node);
}

---