chore: add dielectric tensor to tree and maps

Exabyte-io · Sep 12, 2023 · 635d6e8 · 635d6e8
1 parent 23a9642
commit 635d6e8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/src/classmap.js b/src/classmap.js
@@ -6,6 +6,7 @@ import { BandGapsProperty } from "./include/non-scalar/band_gaps";
 import { BandStructureProperty } from "./include/non-scalar/band_structure";
 import { ChargeDensityProfileProperty } from "./include/non-scalar/charge_density_profile";
 import { DensityOfStatesProperty } from "./include/non-scalar/density_of_states";
+import { DielectricTensorProperty } from "./include/non-scalar/dielectric_tensor";
 import { PhononDispersionsProperty } from "./include/non-scalar/phonon_dispersions";
 import { PhononDOSProperty } from "./include/non-scalar/phonon_dos";
 import { PotentialProfileProperty } from "./include/non-scalar/potential_profile";
@@ -54,6 +55,7 @@ export const PROPERTY_CLASS_MAP = {
     [PROPERTIES.valence_band_offset]: null,
     [PROPERTIES.pseudopotential]: Pseudopotential,
     [PROPERTIES.boundary_conditions]: null,
+    [PROPERTIES.dielectric_tensor]: DielectricTensorProperty,
 };
 
 export const PROPERTY_BRANCH_MAP = {

diff --git a/src/settings.js b/src/settings.js
@@ -52,4 +52,5 @@ export const PROPERTIES = {
     valence_band_offset: "valence_band_offset",
     pseudopotential: "pseudopotential",
     boundary_conditions: "boundary_conditions",
+    dielectric_tensor: "dielectric_tensor",
 };
diff --git a/src/tree.js b/src/tree.js
@@ -132,4 +132,7 @@ export default {
     [PROPERTIES.boundary_conditions]: {
         type: PROPERTY_TYPES.non_scalar,
     },
+    [PROPERTIES.dielectric_tensor]: {
+        type: PROPERTY_TYPES.non_scalar,
+    },
 };