chore: add dielectric tensor property class

Exabyte-io · Sep 12, 2023 · 23a9642 · 23a9642
1 parent d23a7fe
commit 23a9642
Showing 1 changed file with 166 additions and 0 deletions.
diff --git a/src/include/non-scalar/dielectric_tensor.js b/src/include/non-scalar/dielectric_tensor.js
@@ -0,0 +1,166 @@
+/* eslint-disable class-methods-use-this */
+/* eslint-disable max-classes-per-file */
+import zip from "lodash/zip";
+
+import { Property } from "../../property";
+import { TwoDimensionalHighChartConfigMixin } from "../mixins/2d_plot";
+
+export class DielectricTensorProperty extends Property {
+    /**
+     * @returns {{part: string, spin: number, frequencies: number[], components: number[][]}[]}
+     */
+    get values() {
+        return this.prop("values");
+    }
+
+    get subtitle() {
+        return "Dielectric Tensor";
+    }
+
+    get yAxisTitle() {
+        return "Dielectric Tensor Component";
+    }
+
+    get xAxisTitle() {
+        return "Frequency (eV)";
+    }
+
+    get legend() {
+        return this.prop("legend");
+    }
+
+    get chartConfig() {
+        return this.getAllChartConfigs().map((chartConfig) => {
+            // eslint-disable-next-line no-use-before-define
+            const cfg = new DielectricTensorConfig(chartConfig);
+            return cfg.config;
+        });
+    }
+
+    /**
+     *
+     * @param {[number, number, number][]} matrix
+     */
+    rowMajorToColumnMajor(matrix) {
+        return matrix.reduce(
+            (accumulator, item) => {
+                const [x, y, z] = item;
+                accumulator[0].push(x);
+                accumulator[1].push(y);
+                accumulator[2].push(z);
+                return accumulator;
+            },
+            [[], [], []],
+        );
+    }
+
+    /**
+     * @returns {{part: "real" | "imaginary", spin: number, frequencies: number[], components: number[][]}[][]}
+     */
+    getComplementaryPairs(precision = 3) {
+        const groupedBySpin = {};
+
+        this.values.forEach((item) => {
+            // Round the spin value to mitigate floating-point precision issues
+            const spinValue = item.spin !== undefined ? item.spin.toFixed(precision) : "undefined";
+            groupedBySpin[spinValue] = groupedBySpin[spinValue] || [];
+            groupedBySpin[spinValue].push(item);
+        });
+
+        return Object.values(groupedBySpin).filter(
+            (group) =>
+                group.length === 2 &&
+                group.find((item) => item.part === "real") &&
+                group.find((item) => item.part === "imaginary"),
+        );
+    }
+
+    getAllChartConfigs() {
+        const complementaryPairs = this.getComplementaryPairs();
+
+        return complementaryPairs.flatMap((pair) => {
+            const xDataArray = pair[0].frequencies;
+            const components_first = this.rowMajorToColumnMajor(pair[0].components);
+            const components_second = this.rowMajorToColumnMajor(pair[1].components);
+            return components_first.map((epsilon, index) => {
+                return {
+                    subtitle: this.subtitle,
+                    xAxisTitle: this.xAxisTitle,
+                    yAxisTitle: this.yAxisTitle,
+                    yAxisType: "linear",
+                    xDataArray,
+                    yDataSeries: [epsilon, components_second[index]],
+                    legend: pair.map((p) => p.part),
+                };
+            });
+        });
+    }
+}
+
+export class DielectricTensorConfig extends TwoDimensionalHighChartConfigMixin {
+    constructor(config) {
+        super(config);
+        this.legend = config.legend;
+    }
+
+    /**
+     * @returns {{animation: boolean, name: string, data: number[][]}[]}
+     */
+    get series() {
+        return this.yDataSeries.map((item, index) => {
+            return {
+                animation: false,
+                name: this.legend[index],
+                data: zip(this.xDataArray, item),
+            };
+        });
+    }
+
+    // eslint-disable-next-line no-unused-vars
+    tooltipFormatter(xDataArray, yAxisName = "frequency") {
+        // eslint-disable-next-line func-names
+        return function () {
+            return (
+                "<b>state:</b> " +
+                this.series.name +
+                "<br>" +
+                "<b>frequency:</b> " +
+                this.key.toFixed(4) +
+                "<br>" +
+                "<b>epsilon: </b>  " +
+                this.y.toFixed(4)
+            );
+        };
+    }
+
+    get overrideConfig() {
+        return {
+            ...super.overrideConfig,
+            // colors: [
+            //     "#7cb5ec",
+            //     "#90ed7d",
+            //     "#f7a35c",
+            //     "#8085e9",
+            //     "#f15c80",
+            //     "#e4d354",
+            //     "#2b908f",
+            //     "#f45b5b",
+            //     "#91e8e1",
+            // ],
+            credits: {
+                enabled: false,
+            },
+            chart: {
+                type: "spline",
+                zoomType: "xy",
+                animation: false,
+            },
+            legend: {
+                layout: "horizontal",
+                align: "center",
+                verticalAlign: "bottom",
+                borderWidth: 0,
+            },
+        };
+    }
+}