# indie:lang_version = 5 # ============================================================================= # Trend Line Methods Indicator v1.0.0 # ============================================================================= # # METHOD 1: PIVOT SPAN - Two-point pivot trendlines (oldest <-> newest pivot) # METHOD 2: 5-POINT CHANNEL - OLS linear regression through 5 segment extremes # # REPAINT: Pivot lines confirm after pivot_right bars. This is expected behavior. # LINE STYLES: 0 = Solid, 1 = Dashed, 2 = Dotted # ============================================================================= from indie import indicator, param, color, MainContext, Optional from indie.algorithms import PivotHighLow from indie.drawings import LineSegment, AbsolutePosition, line_segment_style from math import isnan, nan, floor @indicator('Trend Line Methods', overlay_main_pane=True) @param.bool('enable_pivot_span', default=True, title='Pivot Span') @param.color('pivot_high_color', default=color.ORANGE, title='High Color') @param.color('pivot_low_color', default=color.ORANGE, title='Low Color') @param.int('pivot_left', default=5, min=1, title='Pivot Left') @param.int('pivot_right', default=5, min=1, title='Pivot Right') @param.int('pivot_count', default=5, min=2, title='Pivot Count') @param.int('pivot_lookback', default=150, min=10, title='Lookback') @param.int('pivot_line_width', default=2, min=1, max=10, title='Pivot Line Width') @param.int('pivot_line_style', default=1, min=0, max=2, title='Pivot Style 0/1/2') @param.bool('enable_five_point', default=False, title='5-Point Channel') @param.color('five_high_color', default=color.FUCHSIA, title='5pt High') @param.color('five_low_color', default=color.FUCHSIA, title='5pt Low') @param.int('five_lookback', default=100, min=10, title='5pt Lookback') @param.int('five_line_width', default=3, min=1, max=10, title='5pt Line Width') @param.int('five_line_style', default=0, min=0, max=2, title='5pt Style 0/1/2') class Main(MainContext): def __init__(self): empty_int_list: list[int] = [] empty_float_list: list[float] = [] self._high_idx_points = self.new_var(empty_int_list) self._high_val_points = self.new_var(empty_float_list) self._low_idx_points = self.new_var(empty_int_list) self._low_val_points = self.new_var(empty_float_list) none_line: Optional[LineSegment] = None self._high_trend_line = self.new_var(none_line) self._low_trend_line = self.new_var(none_line) self._five_high_line = self.new_var(none_line) self._five_low_line = self.new_var(none_line) def calc(self, enable_pivot_span, pivot_high_color, pivot_low_color, pivot_left, pivot_right, pivot_count, pivot_lookback, pivot_line_width, pivot_line_style, enable_five_point, five_high_color, five_low_color, five_lookback, five_line_width, five_line_style): # Get line styles p_style = line_segment_style.SOLID if pivot_line_style == 1: p_style = line_segment_style.DASHED elif pivot_line_style == 2: p_style = line_segment_style.DOTTED f_style = line_segment_style.SOLID if five_line_style == 1: f_style = line_segment_style.DASHED elif five_line_style == 2: f_style = line_segment_style.DOTTED # ===================================================================== # METHOD 1: PIVOT SPAN # ===================================================================== if enable_pivot_span: ph, _ = PivotHighLow.new(self.high, left_bars=pivot_left, right_bars=pivot_right) _, pl = PivotHighLow.new(self.low, left_bars=pivot_left, right_bars=pivot_right) # Collect pivot highs if not isnan(ph[0]): piv_hi_idx = self.bar_index - pivot_right piv_hi_price = self.high[pivot_right] self._high_idx_points.get().append(piv_hi_idx) self._high_val_points.get().append(piv_hi_price) while len(self._high_idx_points.get()) > pivot_count: self._high_idx_points.get().pop(0) self._high_val_points.get().pop(0) # Collect pivot lows if not isnan(pl[0]): piv_lo_idx = self.bar_index - pivot_right piv_lo_price = self.low[pivot_right] self._low_idx_points.get().append(piv_lo_idx) self._low_val_points.get().append(piv_lo_price) while len(self._low_idx_points.get()) > pivot_count: self._low_idx_points.get().pop(0) self._low_val_points.get().pop(0) # Draw high trend line if len(self._high_idx_points.get()) >= 2: far_hi_idx = self._high_idx_points.get()[0] far_hi_val = self._high_val_points.get()[0] near_hi_idx = self._high_idx_points.get()[-1] near_hi_val = self._high_val_points.get()[-1] hi_bar_diff = near_hi_idx - far_hi_idx # Declare slope before conditional (Indie scoping) hi_slope = 0.0 if hi_bar_diff != 0: hi_slope = (near_hi_val - far_hi_val) / hi_bar_diff hi_intercept = far_hi_val - hi_slope * far_hi_idx x1_hi = self.bar_index - (pivot_lookback - 1) x2_hi = self.bar_index y1_hi = hi_intercept + hi_slope * x1_hi y2_hi = hi_intercept + hi_slope * x2_hi if self._high_trend_line.get() is not None: self.chart.erase(self._high_trend_line.get().value()) line = LineSegment( AbsolutePosition(self.time[pivot_lookback - 1], y1_hi), AbsolutePosition(self.time[0], y2_hi), color=pivot_high_color, line_style=p_style, line_width=pivot_line_width ) self._high_trend_line.set(line) self.chart.draw(line) # Draw low trend line if len(self._low_idx_points.get()) >= 2: far_lo_idx = self._low_idx_points.get()[0] far_lo_val = self._low_val_points.get()[0] near_lo_idx = self._low_idx_points.get()[-1] near_lo_val = self._low_val_points.get()[-1] lo_bar_diff = near_lo_idx - far_lo_idx # Declare slope before conditional (Indie scoping) lo_slope = 0.0 if lo_bar_diff != 0: lo_slope = (near_lo_val - far_lo_val) / lo_bar_diff lo_intercept = far_lo_val - lo_slope * far_lo_idx x1_lo = self.bar_index - (pivot_lookback - 1) x2_lo = self.bar_index y1_lo = lo_intercept + lo_slope * x1_lo y2_lo = lo_intercept + lo_slope * x2_lo if self._low_trend_line.get() is not None: self.chart.erase(self._low_trend_line.get().value()) line = LineSegment( AbsolutePosition(self.time[pivot_lookback - 1], y1_lo), AbsolutePosition(self.time[0], y2_lo), color=pivot_low_color, line_style=p_style, line_width=pivot_line_width ) self._low_trend_line.set(line) self.chart.draw(line) # ===================================================================== # METHOD 2: 5-POINT CHANNEL # ===================================================================== if enable_five_point: if self.bar_index >= five_lookback: sum_x_hi = 0.0 sum_y_hi = 0.0 sum_xy_hi = 0.0 sum_x2_hi = 0.0 n_hi = 0 sum_x_lo = 0.0 sum_y_lo = 0.0 sum_xy_lo = 0.0 sum_x2_lo = 0.0 n_lo = 0 seg_len_base = max(1, floor(five_lookback / 5)) for k in range(5): seg_start = k * seg_len_base remaining = five_lookback - seg_start if remaining <= 0: break seg_len_k = min(seg_len_base, remaining) if k < 4 else remaining max_hi = nan barsAgo_hi = -1 for i in range(seg_len_k): sh = seg_start + i v = self.high[sh] if not isnan(v) and (isnan(max_hi) or v > max_hi): max_hi = v barsAgo_hi = sh if barsAgo_hi >= 0: x_hi = float(self.bar_index - barsAgo_hi) y_hi = self.high[barsAgo_hi] sum_x_hi += x_hi sum_y_hi += y_hi sum_xy_hi += x_hi * y_hi sum_x2_hi += x_hi * x_hi n_hi += 1 min_lo = nan barsAgo_lo = -1 for i in range(seg_len_k): sl = seg_start + i v2 = self.low[sl] if not isnan(v2) and (isnan(min_lo) or v2 < min_lo): min_lo = v2 barsAgo_lo = sl if barsAgo_lo >= 0: x_lo = float(self.bar_index - barsAgo_lo) y_lo = self.low[barsAgo_lo] sum_x_lo += x_lo sum_y_lo += y_lo sum_xy_lo += x_lo * y_lo sum_x2_lo += x_lo * x_lo n_lo += 1 # Draw five-point high line if n_hi >= 2: nf_hi = float(n_hi) denom_hi = nf_hi * sum_x2_hi - sum_x_hi * sum_x_hi # Declare slope before conditional slope_hi = 0.0 if denom_hi != 0.0: slope_hi = (nf_hi * sum_xy_hi - sum_x_hi * sum_y_hi) / denom_hi intercept_hi = (sum_y_hi - slope_hi * sum_x_hi) / nf_hi x1_5pt = self.bar_index - five_lookback + 1 x2_5pt = self.bar_index y1_hi_5pt = intercept_hi + slope_hi * float(x1_5pt) y2_hi_5pt = intercept_hi + slope_hi * float(x2_5pt) if self._five_high_line.get() is not None: self.chart.erase(self._five_high_line.get().value()) line = LineSegment( AbsolutePosition(self.time[five_lookback - 1], y1_hi_5pt), AbsolutePosition(self.time[0], y2_hi_5pt), color=five_high_color, line_style=f_style, line_width=five_line_width ) self._five_high_line.set(line) self.chart.draw(line) # Draw five-point low line if n_lo >= 2: nf_lo = float(n_lo) denom_lo = nf_lo * sum_x2_lo - sum_x_lo * sum_x_lo # Declare slope before conditional slope_lo = 0.0 if denom_lo != 0.0: slope_lo = (nf_lo * sum_xy_lo - sum_x_lo * sum_y_lo) / denom_lo intercept_lo = (sum_y_lo - slope_lo * sum_x_lo) / nf_lo x1_5pt = self.bar_index - five_lookback + 1 x2_5pt = self.bar_index y1_lo_5pt = intercept_lo + slope_lo * float(x1_5pt) y2_lo_5pt = intercept_lo + slope_lo * float(x2_5pt) if self._five_low_line.get() is not None: self.chart.erase(self._five_low_line.get().value()) line = LineSegment( AbsolutePosition(self.time[five_lookback - 1], y1_lo_5pt), AbsolutePosition(self.time[0], y2_lo_5pt), color=five_low_color, line_style=f_style, line_width=five_line_width ) self._five_low_line.set(line) self.chart.draw(line) return