---

Gismo.cxx

Go to the documentation of this file.

//     $Id: Gismo.cxx,v 1.5 2001/01/23 03:41:32 burnett Exp $
// Project: Gismo
//  Author: Toby Burnett
//

#include "gismo/Gismo.h"

#include "gui/DisplayControl.h"
#include "gui/DisplayRep.h"
#include "gui/PrintControl.h"

#include "geometry/Track.h"
#include "geomrep/TrackRep.h"


#include "gismo/World.h"
#include "gismo/Generator.h"
#include "gismo/MCParticle.h"
#include "gismo/Material.h"
#include "gismo/PDT.h"
#include <cassert>

#include <stdio.h>   // for sprintf()

// classes for display, print of the gismo generator
class ParticleRep : public gui::DisplayRep
{
    // Represent the particle trajectory and subsequent shower
public:
    ParticleRep(const MCParticle & part);
    void update(){} // did it in the constructor
};

// can no inline because if recursion
ParticleRep::ParticleRep(const MCParticle & part) {
  if(part.track() ) {
    append( TrackRep(*part.track()) );
  }
  for(int i=0; i< part.numChildren(); i++)
    append(ParticleRep( *(MCParticle*) part.child(i)));
}

class Gismo::DisplayEvent : public gui::DisplayRep {friend class Gismo;

    DisplayEvent(Gismo* gismo): m_gismo(gismo){}
    void update() {
        Generator* gen = m_gismo->event();
        if( gen ) {
            markerAt(gen->position());
            append(ParticleRep(*gen));
        }
    }
    Gismo* m_gismo;
};

class Gismo::PrintEvent : public gui::Command {friend class Gismo;
    PrintEvent(std::ostream* out):m_out(out){}
    void execute() { Gismo::instance()->event()->printOn(*m_out);}
    std::ostream* m_out;
};
class Gismo::PrintPDT : public gui::Command { friend class Gismo;
    PrintPDT(std::ostream* out):m_out(out){}
    void execute() { Gismo::instance()->event()->thePDT()->printOn(*m_out);}
    std::ostream* m_out;
};
class Gismo::PrintMaterials : public gui::Command { friend class Gismo;
    PrintMaterials(std::ostream* out):m_out(out){}
    void execute() { Material::printAll(*m_out);}
    std::ostream* m_out;
};
//  Constructor: setup everything!

static float ke_cutoff = 0.0005f;   // 0.5 MeV
static float max_step_size = 100; // 1000 cm

Gismo::Gismo(float size, Generator* gen, gui::DisplayControl* dc)
: m_world(new World(size))
, m_generator(gen)
{
    s_instance = this;


    // generic environment for interactive interface
    // ---------------------------------------------

    MCParticle::saveTrack = 10;   // number of generations that will be displayed


    // Set up display of detector, response, event

    dc->add( m_world->detectorViewer(),"Detector",  -1 );
    dc->add( m_world->responseViewer(), "Response",1 );
    dc->add( new DisplayEvent(this) ,  "Event"    ,1);

    // setup commands to print event, response
    gui::PrintControl* pc= gui::PrintControl::instance();
    pc->add( new PrintEvent( pc->out() ),  "Event"   );
    pc->add(m_world->responsePrinter(pc->out()), "Response"  );

    setKEcutoff(ke_cutoff);
    setMaxStep(max_step_size);
    addMaterialPath("PEGS4");

}

Gismo& Gismo::addMaterialPath(const char* path)
{
    Material::addPath(path);
    return *this;
}       
Gismo& Gismo::addInteractor(Interactor* inter)
{
    MCParticle::addInteractor(inter);
    return *this;
}
Gismo& Gismo::setKEcutoff(float e)
{
    m_world->setKECutOff(e);
    return *this;
}
Gismo& Gismo::setMaxStep(float step)
{   m_world->setMaxStep(step);
    return *this;
}
Gismo& Gismo::setGenerator(Generator* gen)
{
    m_generator = gen;
    return *this;
}

Gismo& Gismo::setField(Field* field)
{
    m_world->setField(field);
    return *this;
}


void
Gismo::finishSetup()
{
    gui::PrintControl* pc= gui::PrintControl::instance();
    pc->add(new PrintMaterials(pc->out()) , "Materials" );
    pc->add(new PrintPDT(pc->out()), "PDT"    );
}

//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//         singleton implementation
Gismo* Gismo::s_instance=0;
Gismo* Gismo::instance()
{
    assert( s_instance );
    return s_instance;
}

//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//       create an event
void Gismo::generate()
{
    m_world->processEvent(*m_generator);
}

static char* itoa(int i)  //little local thing to format integers
{   static char buf[20];
    sprintf(buf,"%d",i);
    return buf;
}
const char * Gismo::eventTitle()const
{
    static std::string title;
    title = std::string("Generated  ");
    title+= m_generator->name();
    title+= ": event ";
    title+= itoa(m_generator->eventNumber());
    return title.c_str();
}



Generator*
Gismo::event()const
{
    return m_generator;
}

Gismo::~Gismo()
// destructor: delete all Gismo stuff
{
    delete GParticle::thePDT();
    delete m_world;
    Material::deleteAll();

}

---