G4VScoreColorMap.cc

Go to the documentation of this file.

//
// ********************************************************************
// * License and Disclaimer                                           *
// *                                                                  *
// * The  Geant4 software  is  copyright of the Copyright Holders  of *
// * the Geant4 Collaboration.  It is provided  under  the terms  and *
// * conditions of the Geant4 Software License,  included in the file *
// * LICENSE and available at  http://cern.ch/geant4/license .  These *
// * include a list of copyright holders.                             *
// *                                                                  *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work  make  any representation or  warranty, express or implied, *
// * regarding  this  software system or assume any liability for its *
// * use.  Please see the license in the file  LICENSE  and URL above *
// * for the full disclaimer and the limitation of liability.         *
// *                                                                  *
// * This  code  implementation is the result of  the  scientific and *
// * technical work of the GEANT4 collaboration.                      *
// * By using,  copying,  modifying or  distributing the software (or *
// * any work based  on the software)  you  agree  to acknowledge its *
// * use  in  resulting  scientific  publications,  and indicate your *
// * acceptance of all terms of the Geant4 Software license.          *
// ********************************************************************
//
//
//
 
#include "G4VScoreColorMap.hh"
#include <string>
#include <sstream>
#include <iomanip>
 
#include "G4VVisManager.hh"
#include "G4VisAttributes.hh"
#include "G4Text.hh"
#include "G4Circle.hh"
#include "G4Polyline.hh"
#include "G4Colour.hh"
#include "G4Polyhedron.hh"
 
G4VScoreColorMap::G4VScoreColorMap(G4String mName)
  : fName(mName)
  , ifFloat(true)
  , fMinVal(0.)
  , fMaxVal(DBL_MAX)
  , fVisManager(nullptr)
{
  ;
}
 
G4VScoreColorMap::~G4VScoreColorMap() { ; }
 
void G4VScoreColorMap::DrawColorChart(G4int _nPoint)
{
  fVisManager = G4VVisManager::GetConcreteInstance();
  if(!fVisManager)
  {
    G4cerr << "G4VScoringMesh::DrawColorChart(): no visualization system"
           << G4endl;
    return;
  }
 
  DrawColorChartBar(_nPoint);
  DrawColorChartText(_nPoint);
}
 
void G4VScoreColorMap::DrawColorChartBar(G4int _nPoint)
{
  G4double min  = this->GetMin();
  G4double max  = this->GetMax();
  G4double smin = -0.89, smax = smin + 0.05 * (_nPoint) *0.83, step = 0.001;
  G4double c[4];
 
  fVisManager->BeginDraw2D();
 
  for(G4double y = smin; y < smax; y += step)
  {
    G4double ra = (y - smin) / (smax - smin), rb = 1. - ra;
    G4Polyline line;
    line.push_back(G4Point3D(-0.96, y, 0.));
    line.push_back(G4Point3D(-0.91, y, 0.));
    this->GetMapColor((ra * max + rb * min) / (ra + rb), c);
    G4Colour col(c[0], c[1], c[2]);
    G4VisAttributes att(col);
    line.SetVisAttributes(&att);
    fVisManager->Draw2D(line);
  }
 
  fVisManager->EndDraw2D();
}
 
void G4VScoreColorMap::DrawColorChartText(G4int _nPoint)
{
  G4double min = this->GetMin();
  G4double max = this->GetMax();
  G4double c[4];
  G4Colour black(0.1, 0.1, 0.1, 0.8);
 
  fVisManager->BeginDraw2D();
 
  for(int n = 0; n < _nPoint; n++)
  {
    G4double a = n / (_nPoint - 1.), b = 1. - a;
    G4double v = (a * max + b * min) / (a + b);
    // background color
 
    for(int l = 0; l < 21; l++)
    {
      G4Polyline line;
      line.push_back(G4Point3D(-0.9, -0.905 + 0.05 * n + 0.002 * l, 0.));
      line.push_back(G4Point3D(-0.75, -0.905 + 0.05 * n + 0.002 * l, 0.));
      G4VisAttributes attblack(black);
      line.SetVisAttributes(&attblack);
      fVisManager->Draw2D(line);
    }
 
    // text
    // char cstring[80];
    // std::sprintf(cstring, "%8.2e", v);
    // G4String value(cstring);
    std::ostringstream oss;
    oss << std::setw(8) << std::setprecision(1) << std::scientific << v;
    std::string str = oss.str();
    G4String value  = str;  //.c_str());
 
    G4Text text(value, G4Point3D(-0.9, -0.9 + 0.05 * n, 0.4));
    G4double size = 12.;
    text.SetScreenSize(size);
    this->GetMapColor(v, c);
    G4Colour color(c[0], c[1], c[2], 1.);
    G4VisAttributes att(color);
    text.SetVisAttributes(&att);
 
    fVisManager->Draw2D(text);
  }
 
  // draw ps name
  // background
  G4double lpsname = 2. + fPSName.size() * 0.95;
  if(lpsname > 0)
  {
    for(int l = 0; l < 22; l++)
    {
      G4Polyline line;
      line.push_back(G4Point3D(-0.92, -0.965 + 0.002 * l, 0.));
      line.push_back(
        G4Point3D(-0.92 + 0.025 * lpsname, -0.965 + 0.002 * l, 0.));
      G4VisAttributes attblack(black);
      // G4VisAttributes attblack(G4Colour(.5, .5, 1.));
      line.SetVisAttributes(&attblack);
      fVisManager->Draw2D(line);
    }
 
    // ps name
    G4Text txtpsname(fPSName, G4Point3D(-0.9, -0.96, 0.1));
    G4double size = 12.;
    txtpsname.SetScreenSize(size);
    G4Colour color(1., 1., 1.);
    G4VisAttributes att(color);
    txtpsname.SetVisAttributes(&att);
    fVisManager->Draw2D(txtpsname);
  }
 
  // draw unit
  // background
  G4double len = 2. + fPSUnit.size();
  if(len > 0)
  {
    for(int l = 0; l < 21; l++)
    {
      G4Polyline line;
      line.push_back(G4Point3D(-0.7, -0.9 + 0.002 * l, 0.));
      line.push_back(G4Point3D(-0.7 + 0.025 * len, -0.9 + 0.002 * l, 0.));
      G4VisAttributes attblack(black);
      // G4VisAttributes attblack(G4Colour(.5, .5, .5));
      line.SetVisAttributes(&attblack);
      fVisManager->Draw2D(line);
    }
 
    // unit
    G4String psunit = "[" + fPSUnit + "]";
    G4Text txtunit(psunit, G4Point3D(-0.69, -0.9, 0.1));
    G4double size = 12.;
    txtunit.SetScreenSize(size);
    G4Colour color(1., 1., 1.);
    G4VisAttributes att(color);
    txtunit.SetVisAttributes(&att);
    fVisManager->Draw2D(txtunit);
  }
 
  fVisManager->EndDraw2D();
}