ohlcv-indicators 1.6.6 → 1.6.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ohlcv-indicators",
-  "version": "1.6.6",
+  "version": "1.6.8",
   "description": "A JavaScript library featuring optimized trading indicators for OHLCV arrays [{open, high, low, close, value}]. This library is optimized for backtesting and trading bot development.",
   "main": "index.js",
   "type": "module",

package/src/studies/candleStudies.js

@@ -12,6 +12,10 @@ const getCandlesStudies = (inputOhlcv, period = 20, len) => {
 
     let getSize = (a, b) => Math.abs(a - b)
 
+    //candle_label
+    let candle_label = new Array(len).fill(null)
+
+    //close > open
     let candle_direction = new Array(len).fill(null)
 
     //body
@@ -76,19 +80,45 @@ const getCandlesStudies = (inputOhlcv, period = 20, len) => {
         candle_direction[x] = candleDirection
 
         //body
-        candle_body_size[x] = bodySize //classifySize(bodySize, bodySizeMean, 0.25)
+        candle_body_size[x] = classifySize(bodySize, bodySizeMean, 0.25)
         
         //top
-        candle_top_size[x] = topSize //classifySize(topSize, bodySizeMean, 0.25)
+        candle_top_size[x] = classifySize(topSize, bodySizeMean, 0.25)
         
         //bottom
-        candle_bottom_size[x] = bottomSize //classifySize(bottomSize, bodySizeMean, 0.25)
+        candle_bottom_size[x] = classifySize(bottomSize, bodySizeMean, 0.25)
 
         //gap
-        candle_gap_size[x] = gapSize //classifySize(gapSize, gapSizeMean, 0.25)
+        candle_gap_size[x] = classifySize(gapSize, gapSizeMean, 0.25)
+
+
+        //label
+
+        const labelArr = [candle_direction[x], candle_body_size[x], candle_top_size[x], candle_bottom_size[x]]
+
+        if(labelArr.every(v => typeof v === 'number'))
+        {
+            candle_label[x] = labelArr.join('_')
+        }
     }
 
+
+    // Create a Map for unique labels with indices
+    const labelMap = new Map();
+    let labelIndex = 0;
+
+    // Populate the label map and assign indices only once
+    for (const label of candle_label) {
+        if (label !== null && !labelMap.has(label)) {
+            labelMap.set(label, labelIndex++);
+        }
+    }
+
+    // Transform candle_label in place using the labelMap for quick lookup
+    candle_label = candle_label.map(v => (v !== null && labelMap.has(v)) ? labelMap.get(v) : null);
+
     return {
+        candle_label,
         candle_direction,
         candle_gap_size,
         candle_body_size,