// File: myudelson[P]hmm-scalable/HMMProblemSlicedAB.cpp
// Function: HMMProblemSlicedAB::BaumWelchBit(FitBitSlicedAB*)
FitResult HMMProblemSlicedAB::BaumWelchBit(FitBitSlicedAB *fb) {
    FitResult fr;
    fr.iter = 1;
    fr.pO0  = 0.0;
    fr.pO   = 0.0;
    fr.conv = 0; // converged
    fr.ndat = 0;
    NCAT xndat = fb->xndat;
    struct data **x_data = fb->x_data;
    
    while( !fr.conv && fr.iter<=this->p->maxiter ) {
        fr.ndat = -1; // no accounting so far
        if(fr.iter==1) {
            computeAlphaAndPOParam(xndat, x_data);
            fr.pO0 = HMMProblemSlicedAB::getSumLogPOPara(xndat, x_data);
            fr.pOmid = fr.pO0;
        }
        fb->copy(FBS_PAR, FBS_PARm1);
        fr.pO = doBaumWelchStep(fb);// PI, A, B);
        
        // check convergence
        fr.conv = fb->checkConvergence(&fr);
        
        if( fr.conv || fr.iter==this->p->maxiter ) {
            //computeAlphaAndPOParam(fb->xndat, fb->x_data);
            fr.pO = HMMProblemSlicedAB::getSumLogPOPara(xndat, x_data);
        }
        fr.iter ++;
        fr.pOmid = fr.pO;
    } // main solver loop
    // recycle memory (Alpha, Beta, p_O_param, Xi, Gamma)
    RecycleFitData(fb->xndat, fb->x_data, this->p);
    fr.iter--;
    return fr;
    
}

// Assembly:
pushq %rbp
movq %rsp, %rbp
subq $0x40, %rsp
movq %rdi, -0x38(%rbp)
movq %rdi, %rax
movq %rax, -0x30(%rbp)
movq %rsi, -0x8(%rbp)
movq %rdx, -0x10(%rbp)
movq -0x8(%rbp), %rax
movq %rax, -0x28(%rbp)
movl $0x1, (%rdi)
xorps %xmm0, %xmm0
movsd %xmm0, 0x8(%rdi)
xorps %xmm0, %xmm0
movsd %xmm0, 0x18(%rdi)
movl $0x0, 0x20(%rdi)
movl $0x0, 0x24(%rdi)
movq -0x10(%rbp), %rax
movl 0xd0(%rax), %eax
movl %eax, -0x14(%rbp)
movq -0x10(%rbp), %rax
movq 0xd8(%rax), %rax
movq %rax, -0x20(%rbp)
movq -0x38(%rbp), %rcx
xorl %eax, %eax
cmpl $0x0, 0x20(%rcx)
movb %al, -0x39(%rbp)
jne 0x3fbfc
movq -0x28(%rbp), %rcx
movq -0x38(%rbp), %rax
movl (%rax), %eax
movq 0x88(%rcx), %rcx
cmpl 0x38(%rcx), %eax
setle %al
movb %al, -0x39(%rbp)
movb -0x39(%rbp), %al
testb $0x1, %al
jne 0x3fc08
jmp 0x3fcdb
movq -0x38(%rbp), %rax
movl $0xffffffff, 0x24(%rax) # imm = 0xFFFFFFFF
cmpl $0x1, (%rax)
jne 0x3fc4b
movq -0x28(%rbp), %rdi
movl -0x14(%rbp), %esi
movq -0x20(%rbp), %rdx
callq 0x39fb0
movq -0x28(%rbp), %rdi
movl -0x14(%rbp), %esi
movq -0x20(%rbp), %rdx
callq 0x1a9f0
movq -0x38(%rbp), %rax
movsd %xmm0, 0x8(%rax)
movsd 0x8(%rax), %xmm0
movsd %xmm0, 0x10(%rax)
movq -0x10(%rbp), %rdi
movl $0x1, %esi
movl $0x2, %edx
callq 0x12890
movq -0x28(%rbp), %rdi
movq -0x10(%rbp), %rsi
callq 0x3fd20
movq -0x38(%rbp), %rsi
movsd %xmm0, 0x18(%rsi)
movq -0x10(%rbp), %rdi
callq 0x129f0
movb %al, %cl
movq -0x38(%rbp), %rax
andb $0x1, %cl
movzbl %cl, %ecx
movl %ecx, 0x20(%rax)
cmpl $0x0, 0x20(%rax)
jne 0x3fca8
movq -0x28(%rbp), %rcx
movq -0x38(%rbp), %rax
movl (%rax), %eax
movq 0x88(%rcx), %rcx
cmpl 0x38(%rcx), %eax
jne 0x3fcc1
movq -0x28(%rbp), %rdi
movl -0x14(%rbp), %esi
movq -0x20(%rbp), %rdx
callq 0x1a9f0
movq -0x38(%rbp), %rax
movsd %xmm0, 0x18(%rax)
movq -0x38(%rbp), %rax
movl (%rax), %ecx
addl $0x1, %ecx
movl %ecx, (%rax)
movsd 0x18(%rax), %xmm0
movsd %xmm0, 0x10(%rax)
jmp 0x3fbd3
movq -0x28(%rbp), %rax
movq -0x10(%rbp), %rcx
movl 0xd0(%rcx), %edi
movq -0x10(%rbp), %rcx
movq 0xd8(%rcx), %rsi
movq 0x88(%rax), %rdx
callq 0xbfd0
movq -0x38(%rbp), %rcx
movq -0x30(%rbp), %rax
movl (%rcx), %edx
addl $-0x1, %edx
movl %edx, (%rcx)
addq $0x40, %rsp
popq %rbp
retq
nopw %cs:(%rax,%rax)