SetDetectorHits.cpp

/*
 * SetDetectorHits.cpp
 *
 *  Created on: 27 Nov 2017
 *      Author: Valerio Formato
 */

/*
 * This function reads hits for a generic detector and lights the corresponding geometry shapes
 * depending on the hit type. shapes are color-coded and sized according to energy
 * release (w.r.t. maximum energy released in a cell for this event).
 */

#include "TEveBrowser.h"
#include "TEveScene.h"
#include "TEveTrans.h"
#include "TGButton.h"
#include "TGTab.h"
#include "TGeoBBox.h"
#include "TGeoNode.h"
#include "TSystem.h"

#include "utils/GGSSmartLog.h"

#include "application/Application.h"
#include "application/gui/MainWindow.h"

void MainWindow::SetDetectorHits(TString det, GGSTHitsReader *reader) {
  static const std::string routineName("MainWindow::SetDetectorHits");

  COUT(DEBUG) << "We are in SetDetectorHits for hit " << det << ENDL;

  EDApplication *thisApp = (EDApplication *)gApplication;

  // READING hits
  double eMax = 0;
  int nHits = reader->GetNHits(det);
  COUT(DEBUG) << nHits << " " << det << " hits " << ENDL;
  if (nHits == 0)
    return;

  std::string volName, volPath;
  Float_t *volPos;
  GGSTIntHitBase *thisHit = nullptr;

  for (int iHit = 0; iHit < nHits; iHit++) {
    thisHit = reader->GetHit(det, iHit);
    if (thisHit->eDep > eMax)
      eMax = thisHit->eDep;
  }

  COUT(DEBUG) << "_DetectorHitMap has " << _DetectorHitMap.size() << " entries" << ENDL;
  auto _dhitIter = _DetectorHitMap.find(det.Data());
  if (_dhitIter == _DetectorHitMap.end()) {
    COUT(ERROR) << "container for detector " << det << " not found!!!" << ENDL;
    return;
  }
  DetectorHit *_dhit = _dhitIter->second;

  auto thisTEveGeoShapeMap = *(_dhit->GetTEveGeoShapeMap());
  COUT(DEBUG) << "map for detector " << det << " has " << thisTEveGeoShapeMap.size() << " entries" << ENDL;
  COUT(DEBUG) << "Threshold set at " << _dhit->GetThreshold() << " GeV" << ENDL;

  for (int iHit = 0; iHit < nHits; iHit++) {
    thisHit = reader->GetHit(det, iHit);
    if (thisHit->eDep < _dhit->GetThreshold() || thisHit->eDep == 0)
      continue;
    volPos = const_cast<float *>(thisHit->GetVolumePosition());
    TGeoNode *tempnode = thisApp->GetGeoManager()->FindNode(volPos[0], volPos[1], volPos[2]);
    volName = tempnode->GetName();
    volPath = thisApp->GetGeoManager()->GetNodePath(tempnode);

    COUT(DEBUG) << volPos[0] << " " << volPos[1] << " " << volPos[2] << ENDL;
    COUT(DEBUG) << volPath << "  " << thisHit->eDep << " " << ENDL;

    TEveGeoShape *tempshape = thisTEveGeoShapeMap[volPath];

    switch (_dhit->GetType()) {
    // Hit colorcoded and transparency coded
    case DetectorHit::kColorCodedScaleHit: {
      _dhit->GetPalette()->Rebin(0, eMax); // change this

      double scale = 0.25 + 0.75 * sqrt(2 * thisHit->eDep / eMax);
      if (scale > 1)
        scale = 1;
      tempshape->RefMainTrans().SetScale(scale, scale, scale);

      double tscale = 0.5 + 0.5 * sqrt(2 * thisHit->eDep / eMax);
      if (tscale > 1)
        tscale = 1;
      tempshape->SetMainTransparency(100 * (1 - tscale) + 1);

      tempshape->SetMainColor(_dhit->GetPalette()->GetValueColor2(thisHit->eDep));
      break;
    }

    // Hit opaque and colorcoded
    case DetectorHit::kColorCodedHit: {
      _dhit->GetPalette()->Rebin(0, eMax); // change this

      double tscale = 0.5 + 0.5 * sqrt(2 * thisHit->eDep / eMax);
      if (tscale > 1)
        tscale = 1;
      tempshape->SetMainTransparency(100 * (1 - tscale) + 1);

      tempshape->RefMainTrans().SetScale(1, 1, 1);

      tempshape->SetMainColor(_dhit->GetPalette()->GetValueColor2(thisHit->eDep));
      break;
    }

    // Hit transparency coded
    case DetectorHit::kOpacityHit: {
      tempshape->SetMainColor(_dhit->GetColor());
      tempshape->SetMainTransparency(85 * (1 - thisHit->eDep / eMax));
      tempshape->RefMainTrans().SetScale(1, 1, 1);
      break;
    }

    // Draw a point, not really supported for now...
    case DetectorHit::kClusterHit: {
      tempshape->SetMainColor(_dhit->GetColor());
      break;
    }

    default: { // kFullHit
      tempshape->SetMainColor(_dhit->GetColor());
      tempshape->SetMainTransparency(0);
      tempshape->RefMainTrans().SetScale(1, 1, 1);
      break;
    }
    }

    tempshape->IncDenyDestroy();
    _dhit->GetTEveElementList()->AddElement(tempshape);
  }
}