REveGeomViewer.cxx

Go to the documentation of this file.

// @(#)root/eve7:$Id$
// Author: Sergey Linev, 13.12.2018
 
/*************************************************************************
 * Copyright (C) 1995-2019, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
#include <ROOT/REveGeomViewer.hxx>
 
#include <ROOT/REveUtil.hxx> // REveLog()
#include <ROOT/RLogger.hxx>
#include <ROOT/RWebWindow.hxx>
 
#include "TSystem.h"
#include "TBase64.h"
#include "TROOT.h"
#include "TEnv.h"
#include "THttpServer.h"
#include "TBufferJSON.h"
#include "TGeoManager.h"
 
#include <fstream>
 
using namespace std::string_literals;
 
//////////////////////////////////////////////////////////////////////////////////////////////
/// constructor
 
ROOT::Experimental::REveGeomViewer::REveGeomViewer(TGeoManager *mgr, const std::string &volname)
{
   fWebWindow = RWebWindow::Create();
   fWebWindow->SetDefaultPage("file:rootui5sys/eve7/geom.html");
 
   // this is call-back, invoked when message received via websocket
   fWebWindow->SetDataCallBack([this](unsigned connid, const std::string &arg) { WebWindowCallback(connid, arg); });
   fWebWindow->SetGeometry(900, 700); // configure predefined window geometry
   fWebWindow->SetConnLimit(0); // allow any connections numbers at the same time
   fWebWindow->SetMaxQueueLength(30); // number of allowed entries in the window queue
 
   fDesc.SetPreferredOffline(gEnv->GetValue("WebGui.PreferredOffline",0) != 0);
   fDesc.SetJsonComp(gEnv->GetValue("WebGui.JsonComp", TBufferJSON::kSkipTypeInfo + TBufferJSON::kNoSpaces));
   fDesc.SetBuildShapes(gEnv->GetValue("WebGui.GeomBuildShapes", 1));
 
   if (mgr) SetGeometry(mgr, volname);
}
 
//////////////////////////////////////////////////////////////////////////////////////////////
/// destructor
 
ROOT::Experimental::REveGeomViewer::~REveGeomViewer()
{
 
}
 
//////////////////////////////////////////////////////////////////////////////////////////////
/// assign new geometry to the viewer
 
void ROOT::Experimental::REveGeomViewer::SetGeometry(TGeoManager *mgr, const std::string &volname)
{
   fGeoManager = mgr;
   fSelectedVolume = volname;
 
   fDesc.Build(mgr, volname);
 
   Update();
}
 
 
/////////////////////////////////////////////////////////////////////////////////
/// Select visible top volume, all other volumes will be disabled
 
void ROOT::Experimental::REveGeomViewer::SelectVolume(const std::string &volname)
{
   if (volname != fSelectedVolume)
      SetGeometry(fGeoManager, volname);
}
 
/////////////////////////////////////////////////////////////////////////////////
/// Show or update geometry in web window
/// If web browser already started - just refresh drawing like "reload" button does
/// If no web window exists or \param always_start_new_browser configured, starts new window
/// \param args arguments to display
 
void ROOT::Experimental::REveGeomViewer::Show(const RWebDisplayArgs &args, bool always_start_new_browser)
{
   std::string user_args = "";
   if (!GetShowHierarchy()) user_args = "{ nobrowser: true }";
   fWebWindow->SetUserArgs(user_args);
 
   if (args.GetWidgetKind().empty())
      const_cast<RWebDisplayArgs *>(&args)->SetWidgetKind("REveGeomViewer");
 
   if ((fWebWindow->NumConnections(true) == 0) || always_start_new_browser)
      fWebWindow->Show(args);
   else
      Update();
}
 
//////////////////////////////////////////////////////////////////////////////////////////////
/// Return URL address of web window used for geometry viewer
 
std::string ROOT::Experimental::REveGeomViewer::GetWindowAddr() const
{
   if (!fWebWindow) return "";
   return fWebWindow->GetAddr();
}
 
//////////////////////////////////////////////////////////////////////////////////////////////
/// Update geometry drawings in all web displays
 
void ROOT::Experimental::REveGeomViewer::Update()
{
   fWebWindow->Send(0, "RELOAD");
}
 
//////////////////////////////////////////////////////////////////////////////////////////////
/// convert JSON into stack array
 
std::vector<int> ROOT::Experimental::REveGeomViewer::GetStackFromJson(const std::string &json, bool node_ids)
{
   std::vector<int> *stack{nullptr}, res;
 
   if (TBufferJSON::FromJSON(stack, json.c_str())) {
      if (node_ids) res = fDesc.MakeStackByIds(*stack);
               else res = *stack;
      delete stack;
   } else {
      R__LOG_ERROR(REveLog()) << "Fail convert " << json << " into vector<int>";
   }
 
   return res;
}
 
//////////////////////////////////////////////////////////////////////////////////////////////
/// Send data for principal geometry draw
 
void ROOT::Experimental::REveGeomViewer::SendGeometry(unsigned connid)
{
   if (!fDesc.HasDrawData())
      fDesc.CollectVisibles();
 
   auto &json = fDesc.GetDrawJson();
 
   R__LOG_DEBUG(0, REveLog()) << "Produce geometry JSON len: " << json.length();
 
   fWebWindow->Send(connid, json);
}
 
//////////////////////////////////////////////////////////////////////////////////////////////
/// Configures draw option for geometry
/// Normally has effect before first drawing of the geometry
/// When geometry displayed, only "axis" and "rotate" options are updated
 
void ROOT::Experimental::REveGeomViewer::SetDrawOptions(const std::string &opt)
{
   fDesc.SetDrawOptions(opt);
   unsigned connid = fWebWindow->GetConnectionId();
   if (connid)
      fWebWindow->Send(connid, "DROPT:"s + opt);
}
 
//////////////////////////////////////////////////////////////////////////////////////////////
/// Produce PNG image of drawn geometry
/// Drawing should be completed at the moment
/// Executed asynchronous - method returns immediately, image stored when received from the client
 
void ROOT::Experimental::REveGeomViewer::SaveImage(const std::string &fname)
{
    unsigned connid = fWebWindow->GetConnectionId();
    if (connid)
       fWebWindow->Send(connid, "IMAGE:"s + fname);
}
 
//////////////////////////////////////////////////////////////////////////////////////////////
/// receive data from client
 
void ROOT::Experimental::REveGeomViewer::WebWindowCallback(unsigned connid, const std::string &arg)
{
   printf("Recv %s\n", arg.substr(0,100).c_str());
 
   if (arg == "GETDRAW") {
 
      SendGeometry(connid);
 
   } else if (arg == "QUIT_ROOT") {
 
      fWebWindow->TerminateROOT();
 
   } else if (arg.compare(0, 7, "SEARCH:") == 0) {
 
      std::string query = arg.substr(7);
 
      std::string hjson, json;
 
      auto nmatches = fDesc.SearchVisibles(query, hjson, json);
 
      printf("Searches %s found %d hjson %d json %d\n", query.c_str(), nmatches, (int) hjson.length(), (int) json.length());
 
      // send reply with appropriate header - NOFOUND, FOUND0:, FOUND1:
      fWebWindow->Send(connid, hjson);
 
      if (!json.empty())
         fWebWindow->Send(connid, json);
 
   } else if (arg.compare(0,4,"GET:") == 0) {
      // provide exact shape
 
      auto stack = GetStackFromJson(arg.substr(4));
 
      auto nodeid = fDesc.FindNodeId(stack);
 
      std::string json{"SHAPE:"};
 
      fDesc.ProduceDrawingFor(nodeid, json);
 
      printf("Produce shape for stack json %d\n", (int) json.length());
 
      fWebWindow->Send(connid, json);
 
   } else if (arg.compare(0, 6, "GVREQ:") == 0) {
 
      auto req = TBufferJSON::FromJSON<REveGeomRequest>(arg.substr(6));
 
      if (req && (req->oper == "HOVER")) {
         if ((req->path.size() > 0 ) && (req->path[0] != "OFF"))
            req->stack = fDesc.MakeStackByPath(req->path);
         req->path.clear();
      } else if (req && (req->oper == "HIGHL")) {
         if (req->stack.size() > 0)
            req->path = fDesc.MakePathByStack(req->stack);
         req->stack.clear();
      } else if (req && (req->oper == "INFO")) {
 
         auto info = fDesc.MakeNodeInfo(req->path);
         if (info)
            fWebWindow->Send(connid, "NINFO:"s + TBufferJSON::ToJSON(info.get(), (fDesc.GetJsonComp() % 5) + TBufferJSON::kSameSuppression).Data());
 
         // not request but different object type is send
         req.reset(nullptr);
 
      } else {
         req.reset(nullptr);
      }
 
      if (req)
         fWebWindow->Send(connid, "GVRPL:"s + TBufferJSON::ToJSON(req.get(), TBufferJSON::kSkipTypeInfo + TBufferJSON::kNoSpaces).Data());
 
   } else if ((arg.compare(0, 7, "SETVI0:") == 0) || (arg.compare(0, 7, "SETVI1:") == 0)) {
      // change visibility for specified nodeid
 
      auto nodeid = std::stoi(arg.substr(7));
 
      bool selected = (arg[5] == '1');
 
      if (fDesc.ChangeNodeVisibility(nodeid, selected)) {
 
         // send only modified entries, includes all nodes with same volume
         std::string json0 = fDesc.ProduceModifyReply(nodeid);
 
         // when visibility disabled, client will automatically remove node from drawing
         fWebWindow->Send(connid, json0);
 
         if (selected && fDesc.IsPrincipalEndNode(nodeid)) {
            // we need to send changes in drawing elements
            // there can be many elements, which reference same volume
 
            std::string json{"APPND:"};
 
            if (fDesc.ProduceDrawingFor(nodeid, json, true)) {
 
               printf("Send appending JSON %d\n", (int) json.length());
 
               fWebWindow->Send(connid, json);
            }
         } else if (selected) {
 
            // just resend full geometry
            // TODO: one can improve here and send only nodes which are not exists on client
            // TODO: for that one should remember all information send to client
 
            auto json = fDesc.ProcessBrowserRequest();
            if (json.length() > 0) fWebWindow->Send(connid, json);
 
            SendGeometry(connid);
         }
      }
   } else if (arg.compare(0,6, "BRREQ:") == 0) {
 
      // central place for processing browser requests
 
      if (!fDesc.IsBuild()) fDesc.Build(fGeoManager);
 
      auto json = fDesc.ProcessBrowserRequest(arg.substr(6));
      if (json.length() > 0) fWebWindow->Send(connid, json);
   } else if (arg.compare(0,6, "IMAGE:") == 0) {
      auto separ = arg.find("::",6);
      if (separ == std::string::npos) return;
 
      std::string fname = arg.substr(6, separ-6);
      if (fname.empty()) {
         int cnt = 0;
         do {
            fname = "geometry"s;
            if (cnt++>0) fname += std::to_string(cnt);
            fname += ".png"s;
         } while (!gSystem->AccessPathName(fname.c_str()));
      }
 
      TString binary = TBase64::Decode(arg.c_str() + separ + 2);
 
      std::ofstream ofs(fname);
      ofs.write(binary.Data(), binary.Length());
      ofs.close();
 
      printf("Image file %s size %d has been created\n", fname.c_str(), (int) binary.Length());
 
   } else if (arg.compare(0,4, "CFG:") == 0) {
 
      if (fDesc.ChangeConfiguration(arg.substr(4)))
         SendGeometry(connid);
 
   } else if (arg == "RELOAD") {
 
      SendGeometry(connid);
   }
}