dmesg_init.cc

Go to the documentation of this file.

// %%%%%%%%%%
// G4 headers
// %%%%%%%%%%
#include "G4RunManager.hh"
#include "G4VisExecutive.hh"
#include "G4TrajectoryDrawByParticleID.hh"

// %%%%%%%%%%%%%
// gemc headers
// %%%%%%%%%%%%%
#include "usage.h"

vector<string> init_dmesg(gemc_opts gemcOpt)
{
    vector<string> commands;
    
    // Geant4 Process verbosity
    double PHI_VERB        = gemcOpt.args["PHY_VERBOSITY"].arg ;
    int   OVERL            = (int) gemcOpt.args["CHECK_OVERLAPS"].arg ;
    int   DAWN_N           = (int) gemcOpt.args["DAWN_N"].arg ;
    int   SAVE_ALL_MOTHERS = (int) gemcOpt.args["SAVE_ALL_MOTHERS"].arg ;
    
    char phi_verb[2];
    sprintf(phi_verb, "%d", (int) PHI_VERB);
    string proc_verb1 = "/process/setVerbose ";
    string proc_verb2 = "/process/verbose ";
    proc_verb1.append(phi_verb);
    proc_verb1.append(" all");
    proc_verb2.append(phi_verb);
    
    
    // Geant4 Tracking Verbosity
    double TRACK_VERB =  gemcOpt.args["G4TRACK_VERBOSITY"].arg ;
    char track_verb[2];
    sprintf(track_verb, "%d", (int) TRACK_VERB);
    string tra_verb = "/tracking/verbose ";
    tra_verb.append(track_verb);
    commands.push_back(tra_verb);
    
    // keep trajectories in the container if mother infos are required
    if(SAVE_ALL_MOTHERS)
    commands.push_back("/tracking/storeTrajectory 2");
  else
    commands.push_back("/tracking/storeTrajectory 0");
    
    
    // sets all verbosity to zero
    commands.push_back("/control/verbose 0");
    commands.push_back("/geometry/navigator/verbose 0");
    commands.push_back("/event/verbose 0");
    commands.push_back("/run/verbose 0");
    commands.push_back("/run/particle/verbose 0");
    commands.push_back("/material/verbose 0");
    commands.push_back(proc_verb1.c_str());
    commands.push_back(proc_verb2.c_str());
    commands.push_back("/process/eLoss/verbose 0");
    commands.push_back("/vis/verbose 0");
    commands.push_back("/vis/viewer/flush");
    commands.push_back("/process/setVerbose 0");




    if(OVERL>1)
        commands.push_back("/geometry/test/grid_test 1");

    if(OVERL>2)
        commands.push_back("/geometry/test/cylinder_test 1");
    
    
    if(DAWN_N>0)
    {
        char dawn[10];
        sprintf(dawn, "%d", DAWN_N);
        string DN = "/run/beamOn ";
        DN.append(dawn);
        
        commands.push_back("/vis/open DAWNFILE");
        commands.push_back("/vis/drawVolume");
        commands.push_back("/vis/scene/add/trajectories rich smooth");
        commands.push_back("/vis/scene/add/hits");
        commands.push_back("/vis/scene/endOfEventAction accumulate -1");
        commands.push_back(DN);
    }
        
    
    
    return commands;
}

vector<string> init_dvmesg(gemc_opts gemcOpt, G4VisManager *VM)
{
    vector<string> commands;
    commands.push_back("/vis/scene/create Hall-B");
    double use_qt   = gemcOpt.args["USE_QT"].arg;
    int   HIGH_RES  = (int) gemcOpt.args["HIGH_RES"].arg ;
    
    // OpenGL immediate mode
    if(use_qt==1)
        commands.push_back("/vis/open OGLI " + gemcOpt.args["geometry"].args);
    // openGL stored mode 
    else if(use_qt==2)
    commands.push_back("/vis/open OGLS "   + gemcOpt.args["geometry"].args);
    

    commands.push_back("/vis/viewer/set/autoRefresh 1");
//  commands.push_back("/vis/viewer/set/culling  global true");
//  commands.push_back("/vis/viewer/set/culling coveredDaughters   1");


    if(HIGH_RES==1)
    {
        commands.push_back("/vis/scene/add/trajectories rich smooth");
        commands.push_back("/vis/viewer/set/background .8 .9 .98 1");   // 205, 230, 251 = CD, E6, FA
        commands.push_back("/vis/scene/add/hits");
        //commands.push_back("/vis/viewer/set/lineSegmentsPerCircle 100");
        commands.push_back("/vis/scene/endOfEventAction accumulate -1");
    }
    else 
    {
        commands.push_back("/vis/scene/add/trajectories");
        commands.push_back("/vis/scene/add/hits");
        commands.push_back("/vis/scene/endOfEventAction accumulate -1");
    }



//  commands.push_back("/vis/viewer/zoom 1.0");

    
    // tracks colors
    // these steps come from the user manual chapter Trajectory Drawing Models
    // Create and configure a drawByParticleID model named  gemcColorIDModel
    // positive particles: RED
    // negative particles: GREEN
    // specific colors below
    G4TrajectoryDrawByParticleID* gemcColorIDModel = new G4TrajectoryDrawByParticleID("gemcColorIDModel");
    gemcColorIDModel->SetDefault("gray");
    gemcColorIDModel->Set("neutron", "black");
    gemcColorIDModel->Set("gamma",   "blue");
    gemcColorIDModel->Set("e-",      "cyan");
    gemcColorIDModel->Set("pi+",     "magenta");
    gemcColorIDModel->Set("pi-",     "yellow");
    gemcColorIDModel->Set("proton",  G4Colour(0.95, 0.6, 0.3));  // orange
    
    G4ParticleTable *particleTable = G4ParticleTable::GetParticleTable();
    for(int i=0; i<particleTable->entries(); i++)
    {
        string pname  = particleTable->GetParticleName(i);
        double charge = particleTable->FindParticle(pname)->GetPDGCharge();
        string pcolor ;
        if(pname !=  "neutron" &&
             pname !=  "gamma"   &&
             pname !=  "e-"      &&
             pname !=  "pi+"     &&
             pname !=  "pi-"     &&
             pname !=  "proton"    )
            {
                if(charge>0)
          gemcColorIDModel->Set(pname,  "red");
                if(charge==0)
          gemcColorIDModel->Set(pname,  "white");
                if(charge<0)
          gemcColorIDModel->Set(pname,  "green");
            }
    }

    VM->RegisterModel(gemcColorIDModel);
    VM->SelectTrajectoryModel(gemcColorIDModel->Name());

    // Draw by Process - interesting! But not working yet?
//  commands.push_back("/vis/modeling/trajectories/create/drawByAttribute gemcColorAttributeModel");
//  commands.push_back("/vis/modeling/trajectories/gemcColorAttributeModel/setAttribute CPN");
//  commands.push_back("/vis/modeling/trajectories/gemcColorAttributeModel/verbose true");
//  
//  commands.push_back("/vis/modeling/trajectories/gemcColorAttributeModel/addValue brem_key       eBrem");
//  commands.push_back("/vis/modeling/trajectories/gemcColorAttributeModel/addValue annihil_key annihil");
//  commands.push_back("/vis/modeling/trajectories/gemcColorAttributeModel/addValue decay_key Decay");
//  commands.push_back("/vis/modeling/trajectories/gemcColorAttributeModel/addValue muIon_key muIoni");
//  commands.push_back("/vis/modeling/trajectories/gemcColorAttributeModel/addValue eIon_key  eIoni");
//  
//  commands.push_back("/vis/modeling/trajectories/gemcColorAttributeModel/brem_key/setLineColour    yellow");
//  commands.push_back("/vis/modeling/trajectories/gemcColorAttributeModel/annihil_key/setLineColour    yellow");
//  commands.push_back("/vis/modeling/trajectories/gemcColorAttributeModel/decay_key/setLineColour    yellow");
//  commands.push_back("/vis/modeling/trajectories/gemcColorAttributeModel/eIon_key/setLineColour    yellow");
//  commands.push_back("/vis/modeling/trajectories/gemcColorAttributeModel/muIon_key/setLineColour    yellow");

    return commands;
}