jack/chanlun_1
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add files to chanlun_1

Browse Source
This commit is contained in:
jackyu66git
2025-05-23 19:09:55 +08:00
parent c0e218d23d
commit 4481c47c34
20 changed files with 5574 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/visualization.py
+445
View File
@@ -0,0 +1,445 @@
"""
缠论可视化模块：使用Plotly生成交互式图表
"""
import plotly.graph_objects as go
import plotly.express as px
from plotly.subplots import make_subplots
import pandas as pd
import numpy as np
from typing import Dict, List, Optional
import logging
logger = logging.getLogger(__name__)
class ChanVisualizer:
"""缠论可视化器"""
def __init__(self):
"""初始化可视化器"""
self.colors = {
'up_candle': '#26a69a',
'down_candle': '#ef5350',
'fractal_top': '#ff6b6b',
'fractal_bottom': '#4ecdc4',
'stroke_up': '#2e86de',
'stroke_down': '#f39c12',
'segment_up': '#0984e3',
'segment_down': '#e17055',
'central_bank': 'rgba(155, 89, 182, 0.3)',
'buy_signal': '#00b894',
'sell_signal': '#d63031'
}
def create_comprehensive_chart(self, data: Dict) -> go.Figure:
"""
创建综合缠论分析图表
Args:
data: 包含所有分析数据的字典
Returns:
Plotly图表对象
"""
if not data or 'klines' not in data:
return go.Figure()
# 创建子图
fig = make_subplots(
rows=2, cols=1,
shared_xaxes=True,
vertical_spacing=0.1,
subplot_titles=('缠论分析图', '成交量'),
row_heights=[0.8, 0.2]
)
# 添加K线图
self._add_candlestick(fig, data['klines'])
# 添加分型
if 'fractals' in data and data['fractals']:
self._add_fractals(fig, data['fractals'])
# 添加笔
if 'strokes' in data and data['strokes']:
self._add_strokes(fig, data['strokes'])
# 添加线段
if 'segments' in data and data['segments']:
self._add_segments(fig, data['segments'])
# 添加中枢
if 'central_banks' in data and data['central_banks']:
self._add_central_banks(fig, data['central_banks'])
# 添加买卖点
if 'trading_points' in data and data['trading_points']:
self._add_trading_signals(fig, data['trading_points'])
# 添加成交量
self._add_volume(fig, data['klines'])
# 更新布局
self._update_layout(fig)
return fig
def _add_candlestick(self, fig: go.Figure, klines: pd.DataFrame):
"""添加K线图"""
fig.add_trace(
go.Candlestick(
x=klines.index,
open=klines['open'],
high=klines['high'],
low=klines['low'],
close=klines['close'],
name='K线',
increasing=dict(line=dict(color=self.colors['up_candle'])),
decreasing=dict(line=dict(color=self.colors['down_candle']))
),
row=1, col=1
)
def _add_fractals(self, fig: go.Figure, fractals: List):
"""添加分型标记"""
top_fractals = [f for f in fractals if f.fractal_type == 'top']
bottom_fractals = [f for f in fractals if f.fractal_type == 'bottom']
if top_fractals:
fig.add_trace(
go.Scatter(
x=[f.timestamp for f in top_fractals],
y=[f.price for f in top_fractals],
mode='markers',
marker=dict(
symbol='triangle-down',
size=8,
color=self.colors['fractal_top']
),
name='顶分型',
hovertemplate='顶分型<br>时间: %{x}<br>价格: %{y}<br>强度: %{customdata}<extra></extra>',
customdata=[f.strength for f in top_fractals]
),
row=1, col=1
)
if bottom_fractals:
fig.add_trace(
go.Scatter(
x=[f.timestamp for f in bottom_fractals],
y=[f.price for f in bottom_fractals],
mode='markers',
marker=dict(
symbol='triangle-up',
size=8,
color=self.colors['fractal_bottom']
),
name='底分型',
hovertemplate='底分型<br>时间: %{x}<br>价格: %{y}<br>强度: %{customdata}<extra></extra>',
customdata=[f.strength for f in bottom_fractals]
),
row=1, col=1
)
def _add_strokes(self, fig: go.Figure, strokes: List):
"""添加笔"""
for stroke in strokes:
color = self.colors['stroke_up'] if stroke.direction == 1 else self.colors['stroke_down']
fig.add_trace(
go.Scatter(
x=[stroke.start_fractal.timestamp, stroke.end_fractal.timestamp],
y=[stroke.start_fractal.price, stroke.end_fractal.price],
mode='lines',
line=dict(color=color, width=2),
name='笔' if stroke == strokes[0] else '',
showlegend=stroke == strokes[0],
hovertemplate=f'笔<br>方向: {"上升" if stroke.direction == 1 else "下降"}<br>长度: {stroke.length:.2f}<br>强度: {stroke.strength:.2f}<extra></extra>'
),
row=1, col=1
)
def _add_segments(self, fig: go.Figure, segments: List):
"""添加线段"""
for segment in segments:
color = self.colors['segment_up'] if segment.direction == 1 else self.colors['segment_down']
fig.add_trace(
go.Scatter(
x=[segment.start_time, segment.end_time],
y=[segment.start_price, segment.end_price],
mode='lines',
line=dict(color=color, width=4, dash='dash'),
name='线段' if segment == segments[0] else '',
showlegend=segment == segments[0],
hovertemplate=f'线段<br>方向: {"上升" if segment.direction == 1 else "下降"}<br>长度: {segment.length:.2f}<br>强度: {segment.strength:.2f}<extra></extra>'
),
row=1, col=1
)
def _add_central_banks(self, fig: go.Figure, central_banks: List):
"""添加中枢"""
for cb in central_banks:
# 添加中枢矩形区域
fig.add_shape(
type="rect",
x0=cb.start_time,
y0=cb.low_price,
x1=cb.end_time,
y1=cb.high_price,
fillcolor=self.colors['central_bank'],
opacity=0.3,
line=dict(color="rgba(155, 89, 182, 0.8)", width=1),
row=1, col=1
)
# 添加中枢标签
fig.add_annotation(
x=cb.start_time + (cb.end_time - cb.start_time) / 2,
y=cb.center_price,
text=f"中枢({cb.level})",
showarrow=False,
font=dict(size=10, color="purple"),
bgcolor="rgba(255,255,255,0.8)",
row=1, col=1
)
def _add_trading_signals(self, fig: go.Figure, trading_points: List):
"""添加买卖点信号"""
buy_points = [p for p in trading_points if p.signal_type == 'buy']
sell_points = [p for p in trading_points if p.signal_type == 'sell']
if buy_points:
fig.add_trace(
go.Scatter(
x=[p.timestamp for p in buy_points],
y=[p.price for p in buy_points],
mode='markers',
marker=dict(
symbol='triangle-up',
size=12,
color=self.colors['buy_signal']
),
name='买点',
hovertemplate='%{customdata}<br>时间: %{x}<br>价格: %{y}<br>强度: %{text}<extra></extra>',
customdata=[p.description for p in buy_points],
text=[f"{p.strength:.3f}" for p in buy_points]
),
row=1, col=1
)
if sell_points:
fig.add_trace(
go.Scatter(
x=[p.timestamp for p in sell_points],
y=[p.price for p in sell_points],
mode='markers',
marker=dict(
symbol='triangle-down',
size=12,
color=self.colors['sell_signal']
),
name='卖点',
hovertemplate='%{customdata}<br>时间: %{x}<br>价格: %{y}<br>强度: %{text}<extra></extra>',
customdata=[p.description for p in sell_points],
text=[f"{p.strength:.3f}" for p in sell_points]
),
row=1, col=1
)
def _add_volume(self, fig: go.Figure, klines: pd.DataFrame):
"""添加成交量"""
colors = [self.colors['up_candle'] if close >= open_ else self.colors['down_candle']
for close, open_ in zip(klines['close'], klines['open'])]
fig.add_trace(
go.Bar(
x=klines.index,
y=klines['volume'],
name='成交量',
marker_color=colors,
showlegend=False
),
row=2, col=1
)
def _update_layout(self, fig: go.Figure):
"""更新图表布局"""
fig.update_layout(
title=dict(
text="缠论分析图表",
x=0.5,
font=dict(size=20)
),
xaxis_rangeslider_visible=False,
height=800,
showlegend=True,
legend=dict(
orientation="h",
yanchor="bottom",
y=1.02,
xanchor="right",
x=1
),
margin=dict(l=50, r=50, t=100, b=50),
plot_bgcolor='white',
paper_bgcolor='white'
)
# 更新X轴
fig.update_xaxes(
title_text="时间",
showgrid=True,
gridwidth=1,
gridcolor='lightgray'
)
# 更新Y轴
fig.update_yaxes(
title_text="价格",
showgrid=True,
gridwidth=1,
gridcolor='lightgray',
row=1, col=1
)
fig.update_yaxes(
title_text="成交量",
row=2, col=1
)
def create_statistics_charts(self, data: Dict) -> List[go.Figure]:
"""
创建统计分析图表
Args:
data: 分析数据
Returns:
统计图表列表
"""
charts = []
# 分型强度分布
if 'fractals' in data and data['fractals']:
fractal_chart = self._create_fractal_strength_chart(data['fractals'])
charts.append(fractal_chart)
# 买卖点类别分布
if 'trading_points' in data and data['trading_points']:
signal_chart = self._create_signal_distribution_chart(data['trading_points'])
charts.append(signal_chart)
# 中枢级别分布
if 'central_banks' in data and data['central_banks']:
cb_chart = self._create_central_bank_chart(data['central_banks'])
charts.append(cb_chart)
return charts
def _create_fractal_strength_chart(self, fractals: List) -> go.Figure:
"""创建分型强度分布图"""
strengths = [f.strength for f in fractals]
types = [f.fractal_type for f in fractals]
df = pd.DataFrame({'strength': strengths, 'type': types})
fig = px.histogram(
df,
x='strength',
color='type',
title='分型强度分布',
labels={'strength': '强度', 'type': '类型'},
color_discrete_map={'top': self.colors['fractal_top'], 'bottom': self.colors['fractal_bottom']}
)
return fig
def _create_signal_distribution_chart(self, trading_points: List) -> go.Figure:
"""创建买卖点分布图"""
classes = [f"{p.point_class}类{p.signal_type}点" for p in trading_points]
fig = px.pie(
values=[classes.count(c) for c in set(classes)],
names=list(set(classes)),
title='买卖点类别分布'
)
return fig
def _create_central_bank_chart(self, central_banks: List) -> go.Figure:
"""创建中枢分析图"""
levels = [cb.level for cb in central_banks]
strengths = [cb.strength for cb in central_banks]
fig = go.Figure()
for level in set(levels):
level_strengths = [s for l, s in zip(levels, strengths) if l == level]
fig.add_trace(go.Box(
y=level_strengths,
name=level,
boxpoints='all'
))
fig.update_layout(
title='中枢强度分布（按级别）',
xaxis_title='中枢级别',
yaxis_title='强度'
)
return fig
def create_market_structure_chart(self, market_structure: Dict) -> go.Figure:
"""
创建市场结构图
Args:
market_structure: 市场结构数据
Returns:
市场结构图表
"""
fig = go.Figure()
# 当前价格线
if 'current_price' in market_structure:
fig.add_hline(
y=market_structure['current_price'],
line_dash="dash",
line_color="black",
annotation_text=f"当前价格: {market_structure['current_price']:.2f}"
)
# 支撑阻力位
if 'support_resistance' in market_structure:
sr = market_structure['support_resistance']
# 支撑位
if 'support_levels' in sr:
for i, support in enumerate(sr['support_levels']):
fig.add_hline(
y=support,
line_dash="dot",
line_color=self.colors['buy_signal'],
annotation_text=f"支撑{i+1}: {support:.2f}"
)
# 阻力位
if 'resistance_levels' in sr:
for i, resistance in enumerate(sr['resistance_levels']):
fig.add_hline(
y=resistance,
line_dash="dot",
line_color=self.colors['sell_signal'],
annotation_text=f"阻力{i+1}: {resistance:.2f}"
)
fig.update_layout(
title="市场结构分析",
yaxis_title="价格",
height=400
)
return fig