// File: myudelson[P]hmm-scalable/HMMProblem.cpp
// Function: HMMProblem::doLagrangeStep(FitBit*)
NUMBER HMMProblem::doLagrangeStep(FitBit *fb) {
	NPAR nS = this->p->nS, nO = this->p->nO;
	NPAR i,j,m;
    
    NUMBER  ll = 0;
	NUMBER * b_PI = init1D<NUMBER>((NDAT)this->p->nS);
    NUMBER   b_PI_den = 0;
	NUMBER ** b_A     = init2D<NUMBER>((NDAT)this->p->nS, (NDAT)this->p->nS);
	NUMBER *  b_A_den = init1D<NUMBER>((NDAT)this->p->nS);
	NUMBER ** b_B     = init2D<NUMBER>((NDAT)this->p->nS, (NDAT)this->p->nO);
	NUMBER *  b_B_den = init1D<NUMBER>((NDAT)this->p->nS);
	// compute sums PI
    
    NUMBER buf = 0;
    // collapse
    for(i=0; i<nS; i++) {
        if(fb->pi != NULL)
            buf = -fb->pi[i]*fb->gradPI[i]; /*negatie since were going against gradient*/
        b_PI_den += buf;
        b_PI[i] += buf;
        if(fb->A != NULL)
            for(j=0; j<nS; j++){
                buf = -fb->A[i][j]*fb->gradA[i][j]; /*negatie since were going against gradient*/
                b_A_den[i] += buf;
                b_A[i][j] += buf;
            }
        if(fb->B != NULL)
            for(m=0; m<nO; m++) {
                buf = -fb->B[i][m]*fb->gradB[i][m]; /*negatie since were going against gradient*/
                b_B_den[i] += buf;
                b_B[i][m] += buf;
            }
    }
    // divide
    for(i=0; i<nS; i++) {
        if(fb->pi != NULL)
            fb->pi[i] = (b_PI_den>0) ? (b_PI[i] / b_PI_den) : fb->pi[i];
        if(fb->A != NULL)
            for(j=0; j<nS; j++)
                fb->A[i][j] = (b_A_den[i]>0) ? (b_A[i][j] / b_A_den[i]) : fb->A[i][j];
        if(fb->B != NULL)
            for(m=0; m<nO; m++)
                fb->B[i][m] = (b_B_den[i]>0) ? (b_B[i][m] / b_B_den[i]) : fb->B[i][m];
    }
    // scale
    if( !this->hasNon01Constraints() || fb->observe_non01_boundaries==0 ) {
        if(fb->pi != NULL) projectsimplex(fb->pi, nS);
        for(i=0; i<nS; i++) {
            if(fb->A != NULL) projectsimplex(fb->A[i], nS);
            if(fb->B != NULL) projectsimplex(fb->B[i], nO);
        }
    } else {
        if(fb->pi != NULL) projectsimplexbounded(fb->pi, this->getLbPI(), this->getUbPI(), nS);
        for(i=0; i<nS; i++) {
            if(fb->A != NULL) projectsimplexbounded(fb->A[i], this->getLbA()[i], this->getUbA()[i], nS);
            if(fb->B != NULL) projectsimplexbounded(fb->B[i], this->getLbB()[i], this->getUbB()[i], nO);
        }
    }
    // compute LL
    computeAlphaAndPOParam(fb->xndat, fb->x_data);
    ll = HMMProblem::getSumLogPOPara(fb->xndat, fb->x_data);

	free(b_PI);
	free2D<NUMBER>(b_A, nS);
	free(b_A_den);
	free2D<NUMBER>(b_B, nS);
	free(b_B_den);
    
    return ll;
}

// Assembly:
pushq %rbp
movq %rsp, %rbp
subq $0xb0, %rsp
movq %rdi, -0x8(%rbp)
movq %rsi, -0x10(%rbp)
movq -0x8(%rbp), %rax
movq %rax, -0x60(%rbp)
movq 0x88(%rax), %rcx
movb 0x8b8(%rcx), %cl
movb %cl, -0x11(%rbp)
movq 0x88(%rax), %rcx
movb 0x8b9(%rcx), %cl
movb %cl, -0x12(%rbp)
xorps %xmm0, %xmm0
movsd %xmm0, -0x20(%rbp)
movq 0x88(%rax), %rax
movsbl 0x8b8(%rax), %edi
callq 0xc410
movq %rax, %rcx
movq -0x60(%rbp), %rax
movq %rcx, -0x28(%rbp)
xorps %xmm0, %xmm0
movsd %xmm0, -0x30(%rbp)
movq 0x88(%rax), %rcx
movsbl 0x8b8(%rcx), %edi
movq 0x88(%rax), %rax
movsbl 0x8b8(%rax), %esi
callq 0xf580
movq %rax, %rcx
movq -0x60(%rbp), %rax
movq %rcx, -0x38(%rbp)
movq 0x88(%rax), %rax
movsbl 0x8b8(%rax), %edi
callq 0xc410
movq %rax, %rcx
movq -0x60(%rbp), %rax
movq %rcx, -0x40(%rbp)
movq 0x88(%rax), %rcx
movsbl 0x8b8(%rcx), %edi
movq 0x88(%rax), %rax
movsbl 0x8b9(%rax), %esi
callq 0xf580
movq %rax, %rcx
movq -0x60(%rbp), %rax
movq %rcx, -0x48(%rbp)
movq 0x88(%rax), %rax
movsbl 0x8b8(%rax), %edi
callq 0xc410
movq %rax, -0x50(%rbp)
xorps %xmm0, %xmm0
movsd %xmm0, -0x58(%rbp)
movb $0x0, -0x13(%rbp)
movsbl -0x13(%rbp), %eax
movsbl -0x11(%rbp), %ecx
cmpl %ecx, %eax
jge 0x26b1d
movq -0x10(%rbp), %rax
cmpq $0x0, 0x10(%rax)
je 0x2696d
movq -0x10(%rbp), %rax
movq 0x10(%rax), %rax
movsbq -0x13(%rbp), %rcx
movsd (%rax,%rcx,8), %xmm0
movq %xmm0, %rax
movabsq $-0x8000000000000000, %rcx # imm = 0x8000000000000000
xorq %rcx, %rax
movq %rax, %xmm0
movq -0x10(%rbp), %rax
movq 0x58(%rax), %rax
movsbq -0x13(%rbp), %rcx
mulsd (%rax,%rcx,8), %xmm0
movsd %xmm0, -0x58(%rbp)
movsd -0x58(%rbp), %xmm0
addsd -0x30(%rbp), %xmm0
movsd %xmm0, -0x30(%rbp)
movsd -0x58(%rbp), %xmm0
movq -0x28(%rbp), %rax
movsbq -0x13(%rbp), %rcx
addsd (%rax,%rcx,8), %xmm0
movsd %xmm0, (%rax,%rcx,8)
movq -0x10(%rbp), %rax
cmpq $0x0, 0x18(%rax)
je 0x26a51
movb $0x0, -0x14(%rbp)
movsbl -0x14(%rbp), %eax
movsbl -0x11(%rbp), %ecx
cmpl %ecx, %eax
jge 0x26a4f
movq -0x10(%rbp), %rax
movq 0x18(%rax), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movsbq -0x14(%rbp), %rcx
movsd (%rax,%rcx,8), %xmm0
movq %xmm0, %rax
movabsq $-0x8000000000000000, %rcx # imm = 0x8000000000000000
xorq %rcx, %rax
movq %rax, %xmm0
movq -0x10(%rbp), %rax
movq 0x60(%rax), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movsbq -0x14(%rbp), %rcx
mulsd (%rax,%rcx,8), %xmm0
movsd %xmm0, -0x58(%rbp)
movsd -0x58(%rbp), %xmm0
movq -0x40(%rbp), %rax
movsbq -0x13(%rbp), %rcx
addsd (%rax,%rcx,8), %xmm0
movsd %xmm0, (%rax,%rcx,8)
movsd -0x58(%rbp), %xmm0
movq -0x38(%rbp), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movsbq -0x14(%rbp), %rcx
addsd (%rax,%rcx,8), %xmm0
movsd %xmm0, (%rax,%rcx,8)
movb -0x14(%rbp), %al
addb $0x1, %al
movb %al, -0x14(%rbp)
jmp 0x269a7
jmp 0x26a51
movq -0x10(%rbp), %rax
cmpq $0x0, 0x20(%rax)
je 0x26b0e
movb $0x0, -0x15(%rbp)
movsbl -0x15(%rbp), %eax
movsbl -0x12(%rbp), %ecx
cmpl %ecx, %eax
jge 0x26b0c
movq -0x10(%rbp), %rax
movq 0x20(%rax), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movsbq -0x15(%rbp), %rcx
movsd (%rax,%rcx,8), %xmm0
movq %xmm0, %rax
movabsq $-0x8000000000000000, %rcx # imm = 0x8000000000000000
xorq %rcx, %rax
movq %rax, %xmm0
movq -0x10(%rbp), %rax
movq 0x68(%rax), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movsbq -0x15(%rbp), %rcx
mulsd (%rax,%rcx,8), %xmm0
movsd %xmm0, -0x58(%rbp)
movsd -0x58(%rbp), %xmm0
movq -0x50(%rbp), %rax
movsbq -0x13(%rbp), %rcx
addsd (%rax,%rcx,8), %xmm0
movsd %xmm0, (%rax,%rcx,8)
movsd -0x58(%rbp), %xmm0
movq -0x48(%rbp), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movsbq -0x15(%rbp), %rcx
addsd (%rax,%rcx,8), %xmm0
movsd %xmm0, (%rax,%rcx,8)
movb -0x15(%rbp), %al
addb $0x1, %al
movb %al, -0x15(%rbp)
jmp 0x26a64
jmp 0x26b0e
jmp 0x26b10
movb -0x13(%rbp), %al
addb $0x1, %al
movb %al, -0x13(%rbp)
jmp 0x26912
movb $0x0, -0x13(%rbp)
movsbl -0x13(%rbp), %eax
movsbl -0x11(%rbp), %ecx
cmpl %ecx, %eax
jge 0x26d0b
movq -0x10(%rbp), %rax
cmpq $0x0, 0x10(%rax)
je 0x26b92
movsd -0x30(%rbp), %xmm0
xorps %xmm1, %xmm1
ucomisd %xmm1, %xmm0
jbe 0x26b64
movq -0x28(%rbp), %rax
movsbq -0x13(%rbp), %rcx
movsd (%rax,%rcx,8), %xmm0
divsd -0x30(%rbp), %xmm0
movsd %xmm0, -0x68(%rbp)
jmp 0x26b7b
movq -0x10(%rbp), %rax
movq 0x10(%rax), %rax
movsbq -0x13(%rbp), %rcx
movsd (%rax,%rcx,8), %xmm0
movsd %xmm0, -0x68(%rbp)
movsd -0x68(%rbp), %xmm0
movq -0x10(%rbp), %rax
movq 0x10(%rax), %rax
movsbq -0x13(%rbp), %rcx
movsd %xmm0, (%rax,%rcx,8)
movq -0x10(%rbp), %rax
cmpq $0x0, 0x18(%rax)
je 0x26c47
movb $0x0, -0x14(%rbp)
movsbl -0x14(%rbp), %eax
movsbl -0x11(%rbp), %ecx
cmpl %ecx, %eax
jge 0x26c45
movq -0x40(%rbp), %rax
movsbq -0x13(%rbp), %rcx
movsd (%rax,%rcx,8), %xmm0
xorps %xmm1, %xmm1
ucomisd %xmm1, %xmm0
jbe 0x26bf8
movq -0x38(%rbp), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movsbq -0x14(%rbp), %rcx
movsd (%rax,%rcx,8), %xmm0
movq -0x40(%rbp), %rax
movsbq -0x13(%rbp), %rcx
divsd (%rax,%rcx,8), %xmm0
movsd %xmm0, -0x70(%rbp)
jmp 0x26c18
movq -0x10(%rbp), %rax
movq 0x18(%rax), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movsbq -0x14(%rbp), %rcx
movsd (%rax,%rcx,8), %xmm0
movsd %xmm0, -0x70(%rbp)
movsd -0x70(%rbp), %xmm0
movq -0x10(%rbp), %rax
movq 0x18(%rax), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movsbq -0x14(%rbp), %rcx
movsd %xmm0, (%rax,%rcx,8)
movb -0x14(%rbp), %al
addb $0x1, %al
movb %al, -0x14(%rbp)
jmp 0x26ba5
jmp 0x26c47
movq -0x10(%rbp), %rax
cmpq $0x0, 0x20(%rax)
je 0x26cfc
movb $0x0, -0x15(%rbp)
movsbl -0x15(%rbp), %eax
movsbl -0x12(%rbp), %ecx
cmpl %ecx, %eax
jge 0x26cfa
movq -0x50(%rbp), %rax
movsbq -0x13(%rbp), %rcx
movsd (%rax,%rcx,8), %xmm0
xorps %xmm1, %xmm1
ucomisd %xmm1, %xmm0
jbe 0x26cad
movq -0x48(%rbp), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movsbq -0x15(%rbp), %rcx
movsd (%rax,%rcx,8), %xmm0
movq -0x50(%rbp), %rax
movsbq -0x13(%rbp), %rcx
divsd (%rax,%rcx,8), %xmm0
movsd %xmm0, -0x78(%rbp)
jmp 0x26ccd
movq -0x10(%rbp), %rax
movq 0x20(%rax), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movsbq -0x15(%rbp), %rcx
movsd (%rax,%rcx,8), %xmm0
movsd %xmm0, -0x78(%rbp)
movsd -0x78(%rbp), %xmm0
movq -0x10(%rbp), %rax
movq 0x20(%rax), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movsbq -0x15(%rbp), %rcx
movsd %xmm0, (%rax,%rcx,8)
movb -0x15(%rbp), %al
addb $0x1, %al
movb %al, -0x15(%rbp)
jmp 0x26c5a
jmp 0x26cfc
jmp 0x26cfe
movb -0x13(%rbp), %al
addb $0x1, %al
movb %al, -0x13(%rbp)
jmp 0x26b21
movq -0x60(%rbp), %rdi
callq 0x1a4e0
testb $0x1, %al
jne 0x26d1a
jmp 0x26d2e
movq -0x10(%rbp), %rax
movsbl 0xf9(%rax), %eax
cmpl $0x0, %eax
jne 0x26db5
movq -0x10(%rbp), %rax
cmpq $0x0, 0x10(%rax)
je 0x26d4a
movq -0x10(%rbp), %rax
movq 0x10(%rax), %rdi
movsbl -0x11(%rbp), %esi
callq 0x7980
movb $0x0, -0x13(%rbp)
movsbl -0x13(%rbp), %eax
movsbl -0x11(%rbp), %ecx
cmpl %ecx, %eax
jge 0x26db0
movq -0x10(%rbp), %rax
cmpq $0x0, 0x18(%rax)
je 0x26d7f
movq -0x10(%rbp), %rax
movq 0x18(%rax), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rdi
movsbl -0x11(%rbp), %esi
callq 0x7980
movq -0x10(%rbp), %rax
cmpq $0x0, 0x20(%rax)
je 0x26da4
movq -0x10(%rbp), %rax
movq 0x20(%rax), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rdi
movsbl -0x12(%rbp), %esi
callq 0x7980
jmp 0x26da6
movb -0x13(%rbp), %al
addb $0x1, %al
movb %al, -0x13(%rbp)
jmp 0x26d4e
jmp 0x26eeb
movq -0x10(%rbp), %rax
cmpq $0x0, 0x10(%rax)
je 0x26dfc
movq -0x60(%rbp), %rdi
movq -0x10(%rbp), %rax
movq 0x10(%rax), %rax
movq %rax, -0x88(%rbp)
callq 0x1a6b0
movq -0x60(%rbp), %rdi
movq %rax, -0x80(%rbp)
callq 0x1a770
movq -0x88(%rbp), %rdi
movq -0x80(%rbp), %rsi
movq %rax, %rdx
movsbl -0x11(%rbp), %ecx
callq 0x87e0
movb $0x0, -0x13(%rbp)
movsbl -0x13(%rbp), %eax
movsbl -0x11(%rbp), %ecx
cmpl %ecx, %eax
jge 0x26ee9
movq -0x10(%rbp), %rax
cmpq $0x0, 0x18(%rax)
je 0x26e75
movq -0x60(%rbp), %rdi
movq -0x10(%rbp), %rax
movq 0x18(%rax), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movq %rax, -0x98(%rbp)
callq 0x1a6f0
movq -0x60(%rbp), %rdi
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movq %rax, -0x90(%rbp)
callq 0x1a7b0
movq -0x98(%rbp), %rdi
movq -0x90(%rbp), %rsi
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rdx
movsbl -0x11(%rbp), %ecx
callq 0x87e0
movq -0x10(%rbp), %rax
cmpq $0x0, 0x20(%rax)
je 0x26eda
movq -0x60(%rbp), %rdi
movq -0x10(%rbp), %rax
movq 0x20(%rax), %rax
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movq %rax, -0xa8(%rbp)
callq 0x1a730
movq -0x60(%rbp), %rdi
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rax
movq %rax, -0xa0(%rbp)
callq 0x1a7f0
movq -0xa8(%rbp), %rdi
movq -0xa0(%rbp), %rsi
movsbq -0x13(%rbp), %rcx
movq (%rax,%rcx,8), %rdx
movsbl -0x12(%rbp), %ecx
callq 0x87e0
jmp 0x26edc
movb -0x13(%rbp), %al
addb $0x1, %al
movb %al, -0x13(%rbp)
jmp 0x26e00
jmp 0x26eeb
movq -0x60(%rbp), %rdi
movq -0x10(%rbp), %rax
movl 0xe8(%rax), %esi
movq -0x10(%rbp), %rax
movq 0xf0(%rax), %rdx
callq 0x1b0c0
movq -0x60(%rbp), %rdi
movq -0x10(%rbp), %rax
movl 0xe8(%rax), %esi
movq -0x10(%rbp), %rax
movq 0xf0(%rax), %rdx
callq 0x1a9f0
movsd %xmm0, -0x20(%rbp)
movq -0x28(%rbp), %rdi
callq 0x5380
movq -0x38(%rbp), %rdi
movsbl -0x11(%rbp), %esi
callq 0xc7f0
movq -0x40(%rbp), %rdi
callq 0x5380
movq -0x48(%rbp), %rdi
movsbl -0x11(%rbp), %esi
callq 0xc7f0
movq -0x50(%rbp), %rdi
callq 0x5380
movsd -0x20(%rbp), %xmm0
addq $0xb0, %rsp
popq %rbp
retq
nop