feat: add useRealValue option

hongfaqiu · Jan 12, 2024 · 567861a · 567861a
1 parent 64ef6ef
commit 567861a
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 3 deletions.
diff --git a/packages/TIFFImageryProvider/README.md b/packages/TIFFImageryProvider/README.md
@@ -184,6 +184,9 @@ interface SingleBandRenderOptions {
   */
   colors?: [number, string][] | string[];
 
+  /** Determine whether to use the true value range for custom color ranges, defaults to false */
+  useRealValue?: boolean;
+
   /** defaults to continuous */
   type?: 'continuous' | 'discrete';
 

diff --git a/packages/TIFFImageryProvider/README_CN.md b/packages/TIFFImageryProvider/README_CN.md
@@ -182,6 +182,9 @@ interface SingleBandRenderOptions {
    * [[0, 'red'], [0.6, 'green'], [1, 'blue']]
   */
   colors?: [number, string][] | string[];
+
+  /** 控制是否使用真实值生成插值色带，colors参数的stopValue将使用真实值，默认为否 */
+  useRealValue?: boolean;
 
   /** 默认为连续 */
   type?: 'continuous' | 'discrete';

diff --git a/packages/TIFFImageryProvider/src/TIFFImageryProvider.ts b/packages/TIFFImageryProvider/src/TIFFImageryProvider.ts
@@ -34,6 +34,9 @@ export interface SingleBandRenderOptions {
   */
   colors?: [number, string][] | string[];
 
+  /** Determine whether to use the true value range for custom color ranges */
+  useRealValue?: boolean;
+
   /** defaults to continuous */
   type?: 'continuous' | 'discrete';
 
@@ -386,7 +389,7 @@ export class TIFFImageryProvider {
         const { expression, colors, colorScaleImage } = single;
         this.plot.setExpression(expression);
         if (colors) {
-          const colorScale = generateColorScale(colors)
+          const colorScale = generateColorScale(colors, single?.useRealValue ? [band.min, band.max] : [0, 1])
           addColorScale('temp', colorScale.colors, colorScale.positions);
           this.plot.setColorScale('temp' as any);
         } else if (!colorScaleImage) {

diff --git a/packages/TIFFImageryProvider/src/helpers/utils.ts b/packages/TIFFImageryProvider/src/helpers/utils.ts
@@ -42,13 +42,14 @@ export function getRange(bands: Record<number, {
   return { min, max, range };
 }
 
-export function generateColorScale(colors: [number, string][] | string[]) {
+export function generateColorScale(colors: [number, string][] | string[], minMax: number[]) {
   let stops: [number, string][];
 
   if (typeof colors[0] === 'string') {
     stops = (colors as string[]).map((color, index) => [index / colors.length, color])
   } else {
-    stops = (colors as [number, string][])
+    const [min, max] = minMax;
+    stops = (colors as [number, string][]).map(item => [((item[0] - min) / (max - min)), item[1]])
   }
 
   stops.sort((a, b) => a[0] - b[0]);