// File: ekg[P]sgd2/src/sgd2.hpp
// Function: sgd2::sgd(double*, std::vector<sgd2::term_sparse, std::allocator<sgd2::term_sparse>>&, std::vector<double, std::allocator<double>> const&)
void sgd(double* X, std::vector<term_sparse> &terms, const std::vector<double> &etas)
{
    std::random_device rd;
    std::mt19937 rg(rd());

    // iterate through step sizes
    for (double eta : etas)
    {
        // shuffle terms
        std::shuffle(terms.begin(), terms.end(), rg);

        for (const term_sparse& t : terms)
        {
            // cap step size
            double mu_i = eta * t.w_ij;
            if (mu_i > 1)
                mu_i = 1;

            // cap step size
            double mu_j = eta * t.w_ji;
            if (mu_j > 1)
                mu_j = 1;

            double d_ij = t.d;
            uint64_t i = t.i, j = t.j;

            double dx = X[i*2]-X[j*2], dy = X[i*2+1]-X[j*2+1];
            double mag = sqrt(dx*dx + dy*dy);

            double r = (mag-d_ij) / (2*mag);
            double r_x = r * dx;
            double r_y = r * dy;

            X[i*2] -= mu_i * r_x;
            X[i*2+1] -= mu_i * r_y;
            X[j*2] += mu_j * r_x;
            X[j*2+1] += mu_j * r_y;
        }
    }
}

// Assembly:
pushq %rbp
movq %rsp, %rbp
subq $0x27f0, %rsp           # imm = 0x27F0
movq %rdi, -0x8(%rbp)
movq %rsi, -0x10(%rbp)
movq %rdx, -0x18(%rbp)
leaq -0x13a0(%rbp), %rdi
movq %rdi, -0x27f0(%rbp)
callq 0x93e0
movq -0x27f0(%rbp), %rdi
callq 0x9480
movl %eax, -0x27e4(%rbp)
jmp 0x489e
movl -0x27e4(%rbp), %eax
movl %eax, %eax
movl %eax, %esi
leaq -0x2728(%rbp), %rdi
callq 0x94a0
jmp 0x48b6
movq -0x18(%rbp), %rax
movq %rax, -0x2740(%rbp)
movq -0x2740(%rbp), %rdi
callq 0x94d0
movq %rax, -0x2748(%rbp)
movq -0x2740(%rbp), %rdi
callq 0x9500
movq %rax, -0x2750(%rbp)
leaq -0x2748(%rbp), %rdi
leaq -0x2750(%rbp), %rsi
callq 0x9530
testb $0x1, %al
jne 0x4903
jmp 0x4c9c
leaq -0x2748(%rbp), %rdi
callq 0x9570
movsd (%rax), %xmm0
movsd %xmm0, -0x2758(%rbp)
movq -0x10(%rbp), %rdi
callq 0xaf70
movq %rax, -0x2760(%rbp)
movq -0x10(%rbp), %rdi
callq 0xafa0
movq %rax, -0x2768(%rbp)
movq -0x2760(%rbp), %rdi
movq -0x2768(%rbp), %rsi
leaq -0x2728(%rbp), %rdx
callq 0xad00
jmp 0x4957
movq -0x10(%rbp), %rax
movq %rax, -0x2770(%rbp)
movq -0x2770(%rbp), %rdi
callq 0xaf70
movq %rax, -0x2778(%rbp)
movq -0x2770(%rbp), %rdi
callq 0xafa0
movq %rax, -0x2780(%rbp)
leaq -0x2778(%rbp), %rdi
leaq -0x2780(%rbp), %rsi
callq 0xafd0
testb $0x1, %al
jne 0x49a4
jmp 0x4c89
leaq -0x2778(%rbp), %rdi
callq 0xb010
movq %rax, -0x2788(%rbp)
movsd -0x2758(%rbp), %xmm0
movq -0x2788(%rbp), %rax
mulsd 0x18(%rax), %xmm0
movsd %xmm0, -0x2790(%rbp)
movsd -0x2790(%rbp), %xmm0
movsd 0x19625(%rip), %xmm1    # 0x1e008
ucomisd %xmm1, %xmm0
jbe 0x4a1e
movsd 0x19617(%rip), %xmm0    # 0x1e008
movsd %xmm0, -0x2790(%rbp)
jmp 0x4a1e
movq %rax, %rcx
movl %edx, %eax
movq %rcx, -0x2730(%rbp)
movl %eax, -0x2734(%rbp)
leaq -0x13a0(%rbp), %rdi
callq 0x9900
jmp 0x4cb1
movsd -0x2758(%rbp), %xmm0
movq -0x2788(%rbp), %rax
mulsd 0x20(%rax), %xmm0
movsd %xmm0, -0x2798(%rbp)
movsd -0x2798(%rbp), %xmm0
movsd 0x195be(%rip), %xmm1    # 0x1e008
ucomisd %xmm1, %xmm0
jbe 0x4a60
movsd 0x195b0(%rip), %xmm0    # 0x1e008
movsd %xmm0, -0x2798(%rbp)
movq -0x2788(%rbp), %rax
movsd 0x10(%rax), %xmm0
movsd %xmm0, -0x27a0(%rbp)
movq -0x2788(%rbp), %rax
movq (%rax), %rax
movq %rax, -0x27a8(%rbp)
movq -0x2788(%rbp), %rax
movq 0x8(%rax), %rax
movq %rax, -0x27b0(%rbp)
movq -0x8(%rbp), %rax
movq -0x27a8(%rbp), %rcx
shlq %rcx
movsd (%rax,%rcx,8), %xmm0
movq -0x8(%rbp), %rax
movq -0x27b0(%rbp), %rcx
shlq %rcx
subsd (%rax,%rcx,8), %xmm0
movsd %xmm0, -0x27b8(%rbp)
movq -0x8(%rbp), %rax
movq -0x27a8(%rbp), %rcx
shlq %rcx
movsd 0x8(%rax,%rcx,8), %xmm0
movq -0x8(%rbp), %rax
movq -0x27b0(%rbp), %rcx
shlq %rcx
subsd 0x8(%rax,%rcx,8), %xmm0
movsd %xmm0, -0x27c0(%rbp)
movsd -0x27b8(%rbp), %xmm0
movsd -0x27b8(%rbp), %xmm2
movsd -0x27c0(%rbp), %xmm1
mulsd -0x27c0(%rbp), %xmm1
mulsd %xmm2, %xmm0
addsd %xmm1, %xmm0
callq 0x2110
movsd %xmm0, -0x27c8(%rbp)
movsd -0x27c8(%rbp), %xmm0
subsd -0x27a0(%rbp), %xmm0
movsd 0x194ce(%rip), %xmm1    # 0x1e010
mulsd -0x27c8(%rbp), %xmm1
divsd %xmm1, %xmm0
movsd %xmm0, -0x27d0(%rbp)
movsd -0x27d0(%rbp), %xmm0
mulsd -0x27b8(%rbp), %xmm0
movsd %xmm0, -0x27d8(%rbp)
movsd -0x27d0(%rbp), %xmm0
mulsd -0x27c0(%rbp), %xmm0
movsd %xmm0, -0x27e0(%rbp)
movsd -0x2790(%rbp), %xmm0
movsd -0x27d8(%rbp), %xmm2
movq -0x8(%rbp), %rax
movq -0x27a8(%rbp), %rcx
shlq %rcx
movsd (%rax,%rcx,8), %xmm1
movq %xmm0, %rdx
movabsq $-0x8000000000000000, %rsi # imm = 0x8000000000000000
xorq %rsi, %rdx
movq %rdx, %xmm0
mulsd %xmm2, %xmm0
addsd %xmm1, %xmm0
movsd %xmm0, (%rax,%rcx,8)
movsd -0x2790(%rbp), %xmm0
movsd -0x27e0(%rbp), %xmm2
movq -0x8(%rbp), %rax
movq -0x27a8(%rbp), %rcx
shlq %rcx
movsd 0x8(%rax,%rcx,8), %xmm1
movq %xmm0, %rdx
movabsq $-0x8000000000000000, %rsi # imm = 0x8000000000000000
xorq %rsi, %rdx
movq %rdx, %xmm0
mulsd %xmm2, %xmm0
addsd %xmm1, %xmm0
movsd %xmm0, 0x8(%rax,%rcx,8)
movsd -0x2798(%rbp), %xmm0
movsd -0x27d8(%rbp), %xmm2
movq -0x8(%rbp), %rax
movq -0x27b0(%rbp), %rcx
shlq %rcx
movsd (%rax,%rcx,8), %xmm1
mulsd %xmm2, %xmm0
addsd %xmm1, %xmm0
movsd %xmm0, (%rax,%rcx,8)
movsd -0x2798(%rbp), %xmm0
movsd -0x27e0(%rbp), %xmm2
movq -0x8(%rbp), %rax
movq -0x27b0(%rbp), %rcx
shlq %rcx
movsd 0x8(%rax,%rcx,8), %xmm1
mulsd %xmm2, %xmm0
addsd %xmm1, %xmm0
movsd %xmm0, 0x8(%rax,%rcx,8)
leaq -0x2778(%rbp), %rdi
callq 0xb030
jmp 0x4988
jmp 0x4c8b
leaq -0x2748(%rbp), %rdi
callq 0x98e0
jmp 0x48e7
leaq -0x13a0(%rbp), %rdi
callq 0x9900
addq $0x27f0, %rsp           # imm = 0x27F0
popq %rbp
retq
movq -0x2730(%rbp), %rdi
callq 0x22a0
nopl (%rax)