Convert URDF definitions to typescript

Author: ahoenerBE

src/urdf/UrdfBox.ts

@@ -4,32 +4,29 @@
  * @author Russell Toris - [email protected]
  */
 
-import Vector3 from '../math/Vector3.js';
-import * as UrdfTypes from './UrdfTypes.js';
+import { Vector3 } from '../math/index.js';
+import { UrdfAttrs, UrdfDefaultOptions, UrdfType } from './UrdfTypes.js';
 
 /**
  * A Box element in a URDF.
  */
 export default class UrdfBox {
-  /** @type {Vector3 | null} */
-  dimension;
-  /**
-   * @param {Object} options
-   * @param {Element} options.xml - The XML element to parse.
-   */
-  constructor(options) {
-    this.type = UrdfTypes.URDF_BOX;
+  type: UrdfType;
+  dimension: Vector3 | null = null;
+
+  constructor({xml}: UrdfDefaultOptions) {
+    this.type = UrdfType.BOX;
 
     // Parse the xml string
-    var xyz = options.xml.getAttribute('size')?.split(' ');
-    if (xyz) {
-      this.dimension = new Vector3({
-        x: parseFloat(xyz[0]),
-        y: parseFloat(xyz[1]),
-        z: parseFloat(xyz[2])
-      });
-    } else {
-      this.dimension = null;
+    const size: string[] | undefined = xml.getAttribute(UrdfAttrs.Size)?.split(' ');
+    if (!size || size.length !== 3) {
+      return;
     }
+
+    this.dimension = new Vector3({
+      x: parseFloat(size[0]),
+      y: parseFloat(size[1]),
+      z: parseFloat(size[2])
+    });
   }
 }

src/urdf/UrdfColor.ts

@@ -4,22 +4,40 @@
  * @author Russell Toris - [email protected]
  */
 
+import { UrdfAttrs, UrdfDefaultOptions } from './UrdfTypes.js';
+
 /**
  * A Color element in a URDF.
  */
 export default class UrdfColor {
+
+  /**
+   * Color Red, [0, 1]
+   */
+  r: number = 0.0;
+  /**
+   * Color Green, [0, 1]
+   */
+  g: number = 0.0;
+  /**
+   * Color Blue, [0, 1]
+   */
+  b: number = 0.0;
   /**
-   * @param {Object} options
-   * @param {Element} options.xml - The XML element to parse.
+   * Alpha/Opacity, [0, 1]
    */
-  constructor(options) {
+  a: number = 1.0;
+
+  constructor({xml}: UrdfDefaultOptions) {
     // Parse the xml string
-    var rgba = options.xml.getAttribute('rgba')?.split(' ');
-    if (rgba) {
-      this.r = parseFloat(rgba[0]);
-      this.g = parseFloat(rgba[1]);
-      this.b = parseFloat(rgba[2]);
-      this.a = parseFloat(rgba[3]);
+    const rgba: string[] | undefined = xml.getAttribute(UrdfAttrs.Rgba)?.split(' ');
+    if (!rgba || rgba.length !== 4) {
+      return;
     }
+
+    this.r = parseFloat(rgba[0]);
+    this.g = parseFloat(rgba[1]);
+    this.b = parseFloat(rgba[2]);
+    this.a = parseFloat(rgba[3]);
   }
 }

src/urdf/UrdfCylinder.ts

@@ -4,21 +4,21 @@
  * @author Russell Toris - [email protected]
  */
 
-import * as UrdfTypes from './UrdfTypes.js';
+import { UrdfDefaultOptions, UrdfType, UrdfAttrs } from './UrdfTypes.js';
 
 /**
  * A Cylinder element in a URDF.
  */
 export default class UrdfCylinder {
-  /**
-   * @param {Object} options
-   * @param {Element} options.xml - The XML element to parse.
-   */
-  constructor(options) {
-    this.type = UrdfTypes.URDF_CYLINDER;
-    // @ts-expect-error -- possibly null
-    this.length = parseFloat(options.xml.getAttribute('length'));
-    // @ts-expect-error -- possibly null
-    this.radius = parseFloat(options.xml.getAttribute('radius'));
+
+  type: UrdfType;
+  length: number;
+  radius: number;
+
+  constructor({xml}: UrdfDefaultOptions) {
+    this.type = UrdfType.CYLINDER;
+
+    this.length = parseFloat(xml.getAttribute(UrdfAttrs.Length) ?? 'NaN');
+    this.radius = parseFloat(xml.getAttribute(UrdfAttrs.Radius) ?? 'NaN');
   }
 }

src/urdf/UrdfJoint.ts

@@ -3,93 +3,48 @@
  * @author David V. Lu!! - [email protected]
  */
 
-import Pose from '../math/Pose.js';
-import Vector3 from '../math/Vector3.js';
-import Quaternion from '../math/Quaternion.js';
+import { UrdfAttrs, UrdfDefaultOptions } from './UrdfTypes.js';
+import { Pose } from '../math/index.js';
+import { parseUrdfOrigin } from './UrdfUtils.js';
 
 /**
  * A Joint element in a URDF.
  */
 export default class UrdfJoint {
-  /**
-   * @param {Object} options
-   * @param {Element} options.xml - The XML element to parse.
-   */
-  constructor(options) {
-    this.name = options.xml.getAttribute('name');
-    this.type = options.xml.getAttribute('type');
 
-    var parents = options.xml.getElementsByTagName('parent');
+  name: string;
+  type: string | null;
+  parent: string | null = null;
+  child: string | null = null;
+  minval: number = NaN;
+  maxval: number = NaN;
+  origin: Pose = new Pose();
+
+
+  constructor({xml}: UrdfDefaultOptions) {
+    this.name = xml.getAttribute(UrdfAttrs.Name) ?? 'unknown_name';
+    this.type = xml.getAttribute(UrdfAttrs.Type);
+
+    const parents = xml.getElementsByTagName(UrdfAttrs.Parent);
     if (parents.length > 0) {
-      this.parent = parents[0].getAttribute('link');
+      this.parent = parents[0].getAttribute(UrdfAttrs.Link);
     }
 
-    var children = options.xml.getElementsByTagName('child');
+    const children = xml.getElementsByTagName(UrdfAttrs.Child);
     if (children.length > 0) {
-      this.child = children[0].getAttribute('link');
+      this.child = children[0].getAttribute(UrdfAttrs.Link);
     }
 
-    var limits = options.xml.getElementsByTagName('limit');
+    const limits = xml.getElementsByTagName(UrdfAttrs.Limit);
     if (limits.length > 0) {
-      this.minval = parseFloat(limits[0].getAttribute('lower') || 'NaN');
-      this.maxval = parseFloat(limits[0].getAttribute('upper') || 'NaN');
+      this.minval = parseFloat(limits[0].getAttribute(UrdfAttrs.Lower) ?? 'NaN');
+      this.maxval = parseFloat(limits[0].getAttribute(UrdfAttrs.Upper) ?? 'NaN');
     }
 
     // Origin
-    var origins = options.xml.getElementsByTagName('origin');
-    if (origins.length === 0) {
-      // use the identity as the default
-      this.origin = new Pose();
-    } else {
-      // Check the XYZ
-      var xyzValue = origins[0].getAttribute('xyz');
-      var position = new Vector3();
-      if (xyzValue) {
-        var xyz = xyzValue.split(' ');
-        position = new Vector3({
-          x: parseFloat(xyz[0]),
-          y: parseFloat(xyz[1]),
-          z: parseFloat(xyz[2])
-        });
-      }
-
-      // Check the RPY
-      var rpyValue = origins[0].getAttribute('rpy');
-      var orientation = new Quaternion();
-      if (rpyValue) {
-        var rpy = rpyValue.split(' ');
-        // Convert from RPY
-        var roll = parseFloat(rpy[0]);
-        var pitch = parseFloat(rpy[1]);
-        var yaw = parseFloat(rpy[2]);
-        var phi = roll / 2.0;
-        var the = pitch / 2.0;
-        var psi = yaw / 2.0;
-        var x =
-          Math.sin(phi) * Math.cos(the) * Math.cos(psi) -
-          Math.cos(phi) * Math.sin(the) * Math.sin(psi);
-        var y =
-          Math.cos(phi) * Math.sin(the) * Math.cos(psi) +
-          Math.sin(phi) * Math.cos(the) * Math.sin(psi);
-        var z =
-          Math.cos(phi) * Math.cos(the) * Math.sin(psi) -
-          Math.sin(phi) * Math.sin(the) * Math.cos(psi);
-        var w =
-          Math.cos(phi) * Math.cos(the) * Math.cos(psi) +
-          Math.sin(phi) * Math.sin(the) * Math.sin(psi);
-
-        orientation = new Quaternion({
-          x: x,
-          y: y,
-          z: z,
-          w: w
-        });
-        orientation.normalize();
-      }
-      this.origin = new Pose({
-        position: position,
-        orientation: orientation
-      });
+    const origins = xml.getElementsByTagName(UrdfAttrs.Origin);
+    if (origins.length > 0) {
+      this.origin = parseUrdfOrigin(origins[0]);
     }
   }
 }

src/urdf/UrdfLink.ts

@@ -5,21 +5,21 @@
  */
 
 import UrdfVisual from './UrdfVisual.js';
+import { UrdfAttrs, UrdfDefaultOptions } from './UrdfTypes.js';
 
 /**
  * A Link element in a URDF.
  */
 export default class UrdfLink {
-  /**
-   * @param {Object} options
-   * @param {Element} options.xml - The XML element to parse.
-   */
-  constructor(options) {
-    this.name = options.xml.getAttribute('name');
-    this.visuals = [];
-    var visuals = options.xml.getElementsByTagName('visual');
 
-    for (var i = 0; i < visuals.length; i++) {
+  name: string;
+  visuals: UrdfVisual[] = [];
+
+  constructor({xml}: UrdfDefaultOptions) {
+    this.name = xml.getAttribute(UrdfAttrs.Name) ?? 'unknown_name';
+    const visuals = xml.getElementsByTagName(UrdfAttrs.Visuals);
+
+    for (let i = 0; i < visuals.length; i++) {
       this.visuals.push(
         new UrdfVisual({
           xml: visuals[i]

src/urdf/UrdfMaterial.ts

@@ -5,42 +5,42 @@
  */
 
 import UrdfColor from './UrdfColor.js';
+import { UrdfAttrs, UrdfDefaultOptions } from './UrdfTypes.js';
 
 /**
  * A Material element in a URDF.
  */
 export default class UrdfMaterial {
-  /** @type {string | null} */
-  textureFilename = null;
-  /** @type {UrdfColor | null} */
-  color = null;
-  /**
-   * @param {Object} options
-   * @param {Element} options.xml - The XML element to parse.
-   */
-  constructor(options) {
-
-    this.name = options.xml.getAttribute('name');
+
+  name: string;
+  textureFilename: string | null = null;
+  color: UrdfColor | null = null;
+
+  constructor({xml}: UrdfDefaultOptions) {
+
+    this.name = xml.getAttribute(UrdfAttrs.Name) ?? 'unknown_name';
 
     // Texture
-    var textures = options.xml.getElementsByTagName('texture');
+    const textures = xml.getElementsByTagName(UrdfAttrs.Texture);
     if (textures.length > 0) {
-      this.textureFilename = textures[0].getAttribute('filename');
+      this.textureFilename = textures[0].getAttribute(UrdfAttrs.Filename);
     }
 
     // Color
-    var colors = options.xml.getElementsByTagName('color');
+    const colors = xml.getElementsByTagName(UrdfAttrs.Color);
     if (colors.length > 0) {
       // Parse the RBGA string
       this.color = new UrdfColor({
         xml: colors[0]
       });
     }
   }
+
   isLink() {
     return this.color === null && this.textureFilename === null;
   }
-  assign(obj) {
+
+  assign(obj: UrdfMaterial): this & UrdfMaterial {
     return Object.assign(this, obj);
   }
 }

src/urdf/UrdfMesh.ts

@@ -4,33 +4,35 @@
  * @author Russell Toris - [email protected]
  */
 
-import Vector3 from '../math/Vector3.js';
-import * as UrdfTypes from './UrdfTypes.js';
+import { Vector3 } from '../math/index.js';
+import { UrdfAttrs, UrdfDefaultOptions, UrdfType } from './UrdfTypes.js';
 
 /**
  * A Mesh element in a URDF.
  */
 export default class UrdfMesh {
-  /** @type {Vector3 | null} */
-  scale = null;
+  type: UrdfType;
+  scale: Vector3 | null = null;
+  filename: string | null;
+
   /**
    * @param {Object} options
    * @param {Element} options.xml - The XML element to parse.
    */
-  constructor(options) {
-    this.type = UrdfTypes.URDF_MESH;
-    this.filename = options.xml.getAttribute('filename');
+  constructor({xml}: UrdfDefaultOptions) {
+    this.type = UrdfType.MESH;
+    this.filename = xml.getAttribute('filename');
 
     // Check for a scale
-    var scale = options.xml.getAttribute('scale');
-    if (scale) {
-      // Get the XYZ
-      var xyz = scale.split(' ');
-      this.scale = new Vector3({
-        x: parseFloat(xyz[0]),
-        y: parseFloat(xyz[1]),
-        z: parseFloat(xyz[2])
-      });
+    const scale: string[] | undefined = xml.getAttribute(UrdfAttrs.Scale)?.split(' ');
+    if (!scale || scale.length !== 3) {
+      return;
     }
+
+    this.scale = new Vector3({
+      x: parseFloat(scale[0]),
+      y: parseFloat(scale[1]),
+      z: parseFloat(scale[2])
+    });
   }
 }