handy-diffusion 1.0.7 → 1.0.9

package/ADI.js

@@ -45,7 +45,7 @@ export const setADIProperties = (
 
 export const updateSinksAndSources = (sinks, sources) => {
     for (let i = 0; i < WIDTH * HEIGHT; i++) {
-        gamma[i] = (sinks[i] * halfDeltaT) / 4;
+        gamma[i] = (sinks[i] * halfDeltaT) / 2;
         scaledSources[i] = sources[i] * halfDeltaT;
     }
 }

package/bessel.js

@@ -0,0 +1,52 @@
+import BESSEL from 'bessel'
+
+export const BesselSolution = (
+    concentrationData,
+    width,
+    height,
+    sources,
+    decayRate,
+) => {
+    if (width*height !== concentrationData.length) {
+        throw new Error("Width and height do not match the size of concentration data");
+    }
+    if (width*height !== sources.length) {
+        throw new Error("Width and height do not match the size of sources data");
+    }
+
+    const result = new Float64Array(concentrationData.length);
+    const sourcePositions = [];
+    // first get source positions
+    for (let y = 0; y < height; y++) {
+        for (let x = 0; x < width; x++) {
+            const index = y * width + x;
+            if (sources[index] !== 0) {
+                sourcePositions.push({ x, y, strength: sources[index] });
+            }
+        }
+    }
+
+    // now calculate the contribution from each source using the Greens function for the screened Poisson equation, which involves the modified Bessel function of the second kind
+    for (let y = 0; y < height; y++) {
+        for (let x = 0; x < width; x++) {
+            const index = y * width + x;
+            let value = 0;
+            for (const source of sourcePositions) {
+                const dx = x - source.x;
+                const dy = y - source.y;
+                const r = Math.sqrt(dx * dx + dy * dy);
+                if (r === 0) {
+                    value += source.strength; // handle singularity at r=0
+                } else {
+                    value +=  source.strength * BESSEL.besselk( Math.sqrt(decayRate) * r, 0) / (2 * Math.PI);
+                }
+            }
+            result[index] = value;
+        }
+    }
+
+    return result;
+
+}
+
+ 

package/index.d.ts

@@ -21,13 +21,11 @@ export function setADIProperties(
 /**
  * Solve 2D diffusion equation using ADI method
  * @param concentrationData - Concentration array (width * height)
- * @param sources - Source terms array
  * @param totalNumberOfIterations - Number of time steps
  * @param allowNegativeValues - Whether to allow negative concentrations (default: false)
  */
 export function ADI(
     concentrationData: Float64Array | number[],
-    sources: Float64Array | number[],
     totalNumberOfIterations: number,
     allowNegativeValues?: boolean
 ): void;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "handy-diffusion",
-  "version": "1.0.7",
+  "version": "1.0.9",
   "description": "algorithms to simulate diffusion",
   "main": "index.js",
   "types": "index.d.ts",
@@ -21,5 +21,8 @@
   "bugs": {
     "url": "https://github.com/CritalMediumBlue/diffusionForBacteria/issues"
   },
-  "homepage": "https://github.com/CritalMediumBlue/diffusionForBacteria#readme"
+  "homepage": "https://github.com/CritalMediumBlue/diffusionForBacteria#readme",
+  "dependencies": {
+    "bessel": "^1.0.2"
+  }
 }