//
// ********************************************************************
// * 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 "G4RayTracerSceneHandler.hh"

#include "G4VisManager.hh"

#define G4warn G4cout

G4int G4RayTracerSceneHandler::fSceneIdCount = 0;

G4RayTracerSceneHandler::G4RayTracerSceneHandler(G4VGraphicsSystem& system,
						 const G4String& name)
: G4VSceneHandler(system, fSceneIdCount++, name)
{
  // Keep vis manager happy when someone opens a ray tracer with "/vis/open
  // RayTracer" but uses the ray tracer with "/vis/rayTracer" commands
  // before creating any scenes, for example, instead of using
  // "/vis/drawVolume"...
  G4VisManager* visManager = G4VisManager::GetInstance();
  if(visManager) {
    G4Scene* pScene = visManager->GetCurrentScene();
    if (!pScene) {
      // Create new scene like /vis/scene/create...
      fpScene = new G4Scene("dummy-ray-tracer-scene");
      // Add dummy run-duration model to avoid world being added and
      // notifyHandler being invoked...
      fpScene->AddWorldIfEmpty();
      // Add to vis manager list; ownership thereby passes to vis manager...
      visManager->SetSceneList().push_back(fpScene);
      // ...and make current...
      visManager->SetCurrentScene(fpScene);
    }
  }
}

G4RayTracerSceneHandler::~G4RayTracerSceneHandler()
{}

void G4RayTracerSceneHandler::ClearStore()
{
  fSceneVisAttsMap.clear();
}

G4bool G4RayTracerSceneHandler::PathLessThan::operator()
  (const G4ModelingParameters::PVPointerCopyNoPath& a,
   const G4ModelingParameters::PVPointerCopyNoPath& b) const
{
  if (a.size() != b.size()) return a.size() < b.size();
  auto ia = a.begin();
  auto ib = b.begin();
  for (; ia != a.end(); ++ia, ++ib) {
    if (ia->GetPVPointer() < ib->GetPVPointer()) return true;
    if (ia->GetPVPointer() > ib->GetPVPointer()) return false;
    // Pointers equal
    if (ia->GetCopyNo() < ib->GetCopyNo()) return true;
    if (ia->GetCopyNo() > ib->GetCopyNo()) return false;
    // Both pointers and copy no are equal - continue
  }
  // Equality
  return false;
}

void G4RayTracerSceneHandler::BuildVisAttsMap (const G4VSolid&)
{
  // Build map of vis attributes

  G4PhysicalVolumeModel* fpPVModel = dynamic_cast<G4PhysicalVolumeModel*>(fpModel);
  if (fpPVModel) {
    G4ModelingParameters::PVPointerCopyNoPath temp;
    for (const auto& nodeID: fpPVModel->GetFullPVPath()) {
      // Build an element from the nodeid.
      temp.push_back(G4ModelingParameters::PVPointerCopyNo(nodeID.GetPhysicalVolume(),nodeID.GetCopyNo()));
    }
    const G4VisAttributes* pVisAtts = fpVisAttribs;
    if (!pVisAtts) {
      // Shouldn't happen.
      if (G4VisManager::GetInstance()->GetVerbosity() >= G4VisManager::warnings) {
        G4warn <<
        "WARNING: G4RayTracerSceneHandler::BuildVisAttsMap: null vis atts pointer."
        "\n  Using a default vis atts."
        << G4endl;
      }
      static const G4VisAttributes defaultVisAtts;
      pVisAtts = &defaultVisAtts;
    }
    // Copy vis atts into the vis atts map
    fSceneVisAttsMap[temp] = *pVisAtts;
  }
}