x3d-tidy/src/infer.js

X3D Converter, Beautifier and Minimizer

"use strict";

const
   X3D      = require ("x_ite-node"),
   Traverse = require ("x3d-traverse") (X3D);

module .exports = function inferProfileAndComponents (scene)
{
   const
      browser              = scene .getBrowser (),
      usedComponents       = getUsedComponents (scene),
      profileAndComponents = getProfileAndComponentsFromUsedComponents (browser, usedComponents);

   setProfile    (scene, profileAndComponents .profile);
   setComponents (scene, profileAndComponents .components);
}

function getUsedComponents (scene)
{
   const components = new Set ();

   for (const object of scene .traverse (Traverse .PROTO_DECLARATIONS | Traverse .PROTO_DECLARATION_BODY | Traverse .ROOT_NODES | Traverse .PROTOTYPE_INSTANCES))
   {
      if (!(object instanceof X3D .SFNode))
         continue;

      const node = object .getValue ();

      components .add (node .getComponentInfo () .name);
   }

   return components;
}

function getProfileAndComponentsFromUsedComponents (browser, usedComponents)
{
   const profiles = ["Interactive", "Interchange", "Immersive"] .map (name =>
   {
      return { profile: browser .getProfile (name), components: new Set (usedComponents) };
   });

   profiles .forEach (object =>
   {
      for (const component of object .profile .components)
         object .components .delete (component .name);
   });

   const min = profiles .reduce ((min, object) =>
   {
      const count = object .profile .components .length + object .components .size;

      return min .count < count ? min : {
         count: count,
         object: object,
      };
   },
   { count: Number .POSITIVE_INFINITY });

   return {
      profile: min .object .profile,
      components: Array .from (min .object .components) .sort () .map (name => browser .getSupportedComponents () .get (name)),
   };
}

function setProfile (scene, profile)
{
   scene .setProfile (profile);
}

function setComponents (scene, components)
{
   const oldComponents = Array .from (scene .getComponents ());

   for (const component of oldComponents)
      scene .removeComponent (component .name);

   for (const component of components)
      scene .addComponent (component);
}