import math
import all_constants as _K


ISLAND_COLORS = {
    'A': ('rgb(21, 96, 130)',  'white'),
    'B': ('rgb(233, 113, 50)', 'white'),
    'C': ('rgb(25, 107, 36)',  'white'),
    'D': ('rgb(15, 158, 213)', 'white'),
    'E': ('rgb(160, 43, 147)', 'white'),
    'F': ('rgb(209, 209, 209)', '#222'),
}


def generate_candidate_graph(payoffs=[100, 200, 300, 250, 120, 150],
                             priorities=[100, 100, 100, 100, 100, 100],
                             treatment='random',
                             hide_priorities=False,
                             num_players=None):
    """Generate HTML for candidate payoff table and number lines."""
    if num_players is None:
        num_players = 150

    bar_colors = ['rgb(21, 96, 130)', 'rgb(233, 113, 50)',
                  'rgb(25, 107, 36)', 'rgb(15, 158, 213)',
                  'rgb(160, 43, 147)', 'rgb(209, 209, 209)']
    islands = ['A', 'B', 'C', 'D', 'E', 'F']

    pct_lower_vals = []
    for priority_now in priorities:
        pct_lower = 100 * (priority_now - 1) / (num_players - 1)
        if pct_lower != 100:
            pct_lower_vals.append(f'{pct_lower:3.1f}%')
        else:
            pct_lower_vals.append(f'{pct_lower:3.0f}%')

    def _pop_style(i):
        parts = []
        if treatment == 'aligned' and i in (0, 1):
            parts.append('background-color:#f0f0f0')
        if treatment == 'aligned' and i == 1:
            parts.append('border-right: 3px solid black')
        return (f' style="{"; ".join(parts)}"') if parts else ''

    html = '<div class="info-table-container">'
    html += '<table class="info-table">'

    if treatment == 'aligned':
        html += ('<tr style="line-height:1">'
                 '<th class="row-label" style="border:none;padding-top:1px;padding-bottom:1px;background-color:transparent"></th>'
                 '<th colspan="2" style="font-style:italic;text-align:center;font-weight:normal;'
                 'font-size:0.675em;color:#555;border:none;background-color:transparent;'
                 'padding-top:1px;padding-bottom:1px">Popular Islands</th>'
                 '<th colspan="4" style="border:none;padding-top:1px;padding-bottom:1px;background-color:transparent"></th></tr>')

    html += '<tr>'
    html += '<th class="row-label">Island</th>'
    for i, island in enumerate(islands):
        txt_color = '#222' if island == 'F' else 'white'
        html += (f'<th data-island="{island}"{_pop_style(i)}><span class="island-badge" '
                 f'style="background-color:{bar_colors[i]};color:{txt_color}">'
                 f'{island}</span></th>')
    html += '</tr>'

    html += '<tr>'
    html += '<th class="row-label">Points</th>'
    for i, pf in enumerate(payoffs):
        html += f'<td data-island="{islands[i]}"{_pop_style(i)}>{pf:.0f}</td>'
    html += '</tr>'

    html += '<tr>'
    html += '<th class="row-label">Priority Score</th>'
    for i, pr in enumerate(priorities):
        cell_val = '?' if hide_priorities else pr
        html += f'<td data-island="{islands[i]}"{_pop_style(i)}>{cell_val}</td>'
    html += '</tr>'

    html += '<tr>'
    html += '<th class="row-label">% Lower Priority</th>'
    for i, pl in enumerate(pct_lower_vals):
        cell_val = '?' if hide_priorities else pl
        html += f'<td data-island="{islands[i]}"{_pop_style(i)}>{cell_val}</td>'
    html += '</tr>'

    html += '</table>'
    html += '</div>'

    svg_width = 600
    pad = 50
    line_width = svg_width - 2 * pad
    spacing = 20
    r = 9

    def make_svg(label, values, v_min, v_max):
        items = []
        for i, val in enumerate(values):
            x = pad + (val - v_min) / (v_max - v_min) * line_width
            items.append({'x': x, 'color': bar_colors[i], 'island': islands[i]})

        groups = {}
        for item in items:
            key = round(item['x'])
            groups.setdefault(key, []).append(item)

        for key, group in groups.items():
            n = len(group)
            group.sort(key=lambda p: p['island'])
            for j, p in enumerate(group):
                p['y'] = -(n - 1) / 2 * spacing + j * spacing

        min_sep = 2 * r + 2
        for _ in range(300):
            moved = False
            for idx_a in range(len(items)):
                for idx_b in range(idx_a + 1, len(items)):
                    a, b = items[idx_a], items[idx_b]
                    dx = abs(a['x'] - b['x'])
                    if dx >= min_sep:
                        continue
                    needed_dy = math.sqrt(max(0.0, min_sep ** 2 - dx ** 2))
                    dy = b['y'] - a['y']
                    if abs(dy) < needed_dy - 0.01:
                        extra = (needed_dy - abs(dy)) / 2
                        if dy >= 0:
                            a['y'] -= extra
                            b['y'] += extra
                        else:
                            a['y'] += extra
                            b['y'] -= extra
                        moved = True
            if not moved:
                break

        min_y = min(item['y'] for item in items)
        offset = (r + 8) - min_y
        for item in items:
            item['y'] += offset
        y_axis_local = int(offset)

        max_y = max(item['y'] for item in items)
        svg_h = max(70, int(max_y) + r + 20)

        s = f'<div class="number-line-section">'
        s += f'<div class="number-line-label">{label}</div>'
        s += f'<svg width="{svg_width}" height="{svg_h}" class="number-line-svg" overflow="visible">'

        if treatment == 'aligned' and label == 'Points':
            x_mid_pop = pad + (_K.MID_PTS - _K.PTS_STEP / 2 - v_min) / (v_max - v_min) * line_width
            x_right_pop = pad + line_width + r + 2
            mid_x_pop = (x_mid_pop + pad + line_width) / 2
            pop_h = (y_axis_local + 6) * 2
            pop_y = y_axis_local - pop_h / 2
            s += (f'<rect x="{x_mid_pop:.1f}" y="{pop_y:.1f}" width="{x_right_pop - x_mid_pop:.1f}" '
                  f'height="{pop_h}" fill="rgba(180,180,180,0.18)" stroke="none"/>')
            s += (f'<text x="{mid_x_pop:.1f}" y="{pop_y + 12:.1f}" text-anchor="middle" '
                  f'font-size="10" font-style="italic" fill="#888">Popular Islands</text>')

        s += (f'<line x1="{pad}" y1="{y_axis_local}" x2="{pad + line_width}" '
              f'y2="{y_axis_local}" stroke="#888" stroke-width="2"/>')

        s += (f'<line x1="{pad}" y1="{y_axis_local - 5}" x2="{pad}" '
              f'y2="{y_axis_local + 5}" stroke="#888" stroke-width="2"/>')
        s += (f'<text x="{pad - 10}" y="{y_axis_local + 5}" text-anchor="end" '
              f'font-size="16" fill="#444">{v_min}</text>')
        s += (f'<text x="{pad - 10}" y="{y_axis_local + 18}" text-anchor="end" '
              f'font-size="11" fill="#bbb">(Lowest)</text>')

        x_max = pad + line_width
        s += (f'<line x1="{x_max}" y1="{y_axis_local - 5}" x2="{x_max}" '
              f'y2="{y_axis_local + 5}" stroke="#888" stroke-width="2"/>')
        s += (f'<text x="{x_max + 10}" y="{y_axis_local + 5}" text-anchor="start" '
              f'font-size="16" fill="#444">{v_max}</text>')
        s += (f'<text x="{x_max + 10}" y="{y_axis_local + 18}" text-anchor="start" '
              f'font-size="11" fill="#bbb">(Highest)</text>')

        for item in items:
            cx = item['x']
            cy = item['y']
            color = item['color']
            island = item['island']
            text_color = '#222' if island == 'F' else 'white'

            s += (f'<g class="island-dot" data-island="{island}" style="cursor:pointer">')
            s += (f'<circle cx="{cx:.1f}" cy="{cy:.1f}" r="{r}" fill="{color}" '
                  f'stroke="white" stroke-width="1"/>')
            s += (f'<text x="{cx:.1f}" y="{cy + 4:.1f}" text-anchor="middle" '
                  f'font-size="13" font-weight="bold" fill="{text_color}" '
                  f'style="pointer-events:none">{island}</text>')
            s += '</g>'

        s += '</svg></div>'
        return s

    html += make_svg('Points', payoffs, min(payoffs), max(payoffs))

    if hide_priorities:
        svg_width_h = 600
        pad_h = 50
        lw_h = svg_width_h - 2 * pad_h
        y_ax_h = 17
        svg_h_h = y_ax_h + 60
        s = '<div class="number-line-section">'
        s += '<div class="number-line-label">Priority Scores</div>'
        s += f'<svg width="{svg_width_h}" height="{svg_h_h}" class="number-line-svg" overflow="visible">'
        s += (f'<line x1="{pad_h}" y1="{y_ax_h}" x2="{pad_h + lw_h}" '
              f'y2="{y_ax_h}" stroke="#888" stroke-width="2"/>')
        s += (f'<line x1="{pad_h}" y1="{y_ax_h - 5}" x2="{pad_h}" '
              f'y2="{y_ax_h + 5}" stroke="#888" stroke-width="2"/>')
        s += (f'<text x="{pad_h - 10}" y="{y_ax_h + 5}" text-anchor="end" '
              f'font-size="16" fill="#444">1</text>')
        s += (f'<text x="{pad_h - 10}" y="{y_ax_h + 18}" text-anchor="end" '
              f'font-size="11" fill="#bbb">(Lowest)</text>')
        x_max_h = pad_h + lw_h
        s += (f'<line x1="{x_max_h}" y1="{y_ax_h - 5}" x2="{x_max_h}" '
              f'y2="{y_ax_h + 5}" stroke="#888" stroke-width="2"/>')
        s += (f'<text x="{x_max_h + 10}" y="{y_ax_h + 5}" text-anchor="start" '
              f'font-size="16" fill="#444">{num_players}</text>')
        s += (f'<text x="{x_max_h + 10}" y="{y_ax_h + 18}" text-anchor="start" '
              f'font-size="11" fill="#bbb">(Highest)</text>')
        cx_mid = pad_h + lw_h / 2
        s += (f'<text x="{cx_mid:.1f}" y="{y_ax_h + 48}" text-anchor="middle" '
              f'font-size="28" font-weight="bold" fill="#999">?</text>')
        s += '</svg></div>'
        html += s
    else:
        html += make_svg('Priority Scores', priorities, 1, num_players)

    return html


def get_treatment(player):
    """Return 'nocap' or 'cap' for this player's current round.

    Odd uid  → rounds 1-5 nocap, rounds 6-10 cap.
    Even uid → rounds 1-5 cap,   rounds 6-10 nocap.
    """
    uid = player.participant.uid
    first_block = player.round_number <= _K.NUM_ROUNDS_PER_BLOCK
    is_odd = (uid % 2 == 1)
    if first_block:
        return 'nocap' if is_odd else 'cap'
    else:
        return 'cap' if is_odd else 'nocap'


def get_num_cap(player):
    return _K.NUM_CAP_NOC if get_treatment(player) == 'nocap' else _K.NUM_CAP_CAP


def _recycle_uid(player):
    uid = player.participant.uid
    recycled = player.session.recycled_uids
    if uid != 1000 and uid not in recycled:
        player.session.recycled_uids = recycled + [uid]