Added SetTransform Function (#5)

* Added SetTransform Function * Added requested changes to PR * Added Requested Changes to PR * Fixed issues --------- Co-authored-by: venkatesh-sivaraman <[email protected]>
cmudig · Oct 2, 2024 · e431dac · e431dac
1 parent 841ed67
commit e431dac
Show file tree

Hide file tree

Showing 5 changed files with 67 additions and 4 deletions.
diff --git a/counterpoint/dist/counterpoint-vis.es.js b/counterpoint/dist/counterpoint-vis.es.js
@@ -307,6 +307,15 @@ class b {
   freeze() {
     return this.animation && this._cleanUpAnimation(!0), this.value = this.last(), this.valueFn = void 0, this;
   }
+  /**
+   * Sets transform of an attribute to a new function, can be applied to an Attribute,
+   * a Mark, or a MarkRenderGroup
+   * @param attrToModify attribute to modify
+   * @param newFunc new function to set transform to
+   */
+  setTransform(t) {
+    this.transform = t, this.updateTransform();
+  }
   /**
    * @returns whether or not this attribute changed value (due to animation or
    * other updates) on the last call to `advance`
@@ -3216,6 +3225,15 @@ class ra {
   count() {
     return this.markSet.size;
   }
+  /**
+   * Sets transform of an attribute to a new function, can be applied to an Attribute,
+   * a Mark, or a MarkRenderGroup
+   * @param attrToModify attribute to modify
+   * @param newFunc new function to set transform to
+   */
+  setTransform(t, e) {
+    this.forEach((r) => r.setTransform(t, e));
+  }
   /**
    * @param attrNames the attributes to check for changes in (if none provided,
    *  checks all attributes)
@@ -3562,6 +3580,15 @@ class Z {
   _removeEdgeFrom(t) {
     return this._reverseAdjacency.delete(t), this;
   }
+  /**
+   * Sets transform of an attribute to a new function, can be applied to an Attribute,
+   * a Mark, or a MarkRenderGroup
+   * @param attrToModify attribute to modify
+   * @param newFunc new function to set transform to
+   */
+  setTransform(t, e) {
+    this.attributes[t].setTransform(e);
+  }
 }
 function ca(i) {
   let t;

diff --git a/counterpoint/dist/counterpoint-vis.umd.js b/counterpoint/dist/counterpoint-vis.umd.js
diff --git a/counterpoint/lib/attribute.ts b/counterpoint/lib/attribute.ts
@@ -684,7 +684,21 @@ export class Attribute<
     this.valueFn = undefined;
     return this;
   }
-
+  /**
+   * Sets transform of an attribute to a new function, can be applied to an Attribute,
+   * a Mark, or a MarkRenderGroup
+   * @param attrToModify attribute to modify
+   * @param newFunc new function to set transform to
+   */
+  setTransform(
+    newFunc: (
+      raw: ValueType,
+      computeArg: ComputeArgumentType
+    ) => TransformedValueType
+  ): void {
+    this.transform = newFunc;
+    this.updateTransform();
+  }
   /**
    * @returns whether or not this attribute changed value (due to animation or
    * other updates) on the last call to `advance`

diff --git a/counterpoint/lib/mark.ts b/counterpoint/lib/mark.ts
@@ -669,7 +669,6 @@ export class Mark<AttributeSet extends AttributeSetBase = MarkAttributes>
     this._reverseAdjacency.add(sourceMark);
     return this;
   }
-
   /**
    * Tells the mark that it no longer has an edge from the given mark.
    * @param sourceMark the mark that has no longer has an edge to this mark
@@ -678,6 +677,18 @@ export class Mark<AttributeSet extends AttributeSetBase = MarkAttributes>
     this._reverseAdjacency.delete(sourceMark);
     return this;
   }
+  /**
+   * Sets transform of an attribute to a new function, can be applied to an Attribute,
+   * a Mark, or a MarkRenderGroup
+   * @param attrToModify attribute to modify
+   * @param newFunc new function to set transform to
+   */
+  setTransform(
+    attrToModify: string,
+    newFunc: (raw: any, computeArg: any) => any
+  ): void {
+    this.attributes[attrToModify].setTransform(newFunc);
+  }
 }
 
 type AttributeConstructorShorthand<

diff --git a/counterpoint/lib/rendergroup.ts b/counterpoint/lib/rendergroup.ts
@@ -708,7 +708,18 @@ export class MarkRenderGroup<
   count(): number {
     return this.markSet.size;
   }
-
+  /**
+   * Sets transform of an attribute to a new function, can be applied to an Attribute,
+   * a Mark, or a MarkRenderGroup
+   * @param attrToModify attribute to modify
+   * @param newFunc new function to set transform to
+   */
+  setTransform(
+    attrToModify: string,
+    newFunc: (raw: any, computeArg: any) => any
+  ): void {
+    this.forEach((m) => m.setTransform(attrToModify, newFunc));
+  }
   /**
    * @param attrNames the attributes to check for changes in (if none provided,
    *  checks all attributes)