// File: /workspace/github_cpp_makefile_star2_O3/xbai0624[P]dnn/src/ConstructLayer.cpp
// Function: ConstructLayer::UpdateImagesA(int)
void ConstructLayer::UpdateImagesA(int sample_id)
{
    // __imageA shold only store images from all active neurons, the matching info can be achieved from filter matrix
    //    drop out only happens on batch level
    //    so imageA will clear after each batch is done
    //    **** on batch level, the filter matrix stays the same, so no need to worry the change of filter matrix inside a batch
    //size_t l = __imageA.size();
    //cout<<" >>> image in lyaer id: "<<GetID()<<" size: "<<l<<endl;

    // extract the A matrices from neurons for current traning sample
    Images sample_image_A;
    Images sample_image_A_full;

    if(__type == LayerType::fullyConnected || __type == LayerType::output) 
    {
        // for fully connected layer; output layer is also a fully connected layer
        for(size_t k=0;k<__neuronDim.k;k++) // kernel
        {
            Matrix A( __neuronDim.i, __neuronDim.j, 0);
            //cout<<"A before filling"<<endl<<A<<endl;
            for(size_t i=0;i<__neuronDim.i;i++){
                for(size_t j=0;j<__neuronDim.j;j++)
                {
                    auto & a_vector = __neurons[k][i][j]->GetAVector();
                    if(__neurons[k][i][j]->IsActive())
                    {  
                        // make sure no over extract
                        //cout<<a_vector.size()<<"......"<<l<<endl;
                        //assert(a_vector.size() - 1 == l); // obsolete
                        //A[i][j] = a_vector.back();        // obsolete
                        A[i][j] = a_vector[sample_id];
                    }
                    else
                    {
                        // if neuron is inactive, set it to 0
                        A[i][j] = 0;
                    }
                }
            }
            // save full image
            sample_image_A_full.OutputImageFromKernel.push_back(A);
            //cout<<"A after filling: "<<endl<<A<<endl;

            // only save active elements
            Matrix R = filterMatrix(A, __activeFlag[k]);
            sample_image_A.OutputImageFromKernel.push_back(R);

            assert(R.Dimension().first  == __activeNeuronDim.i);
            assert(R.Dimension().second == __activeNeuronDim.j);
        }
        //__imageA.push_back(sample_image_A); // obsolete
        __imageA[sample_id] = sample_image_A;
        __imageAFull[sample_id] = sample_image_A_full;
    }
    else if(__type == LayerType::cnn || __type == LayerType::pooling ) // for cnn layer and pooling layer
    {
        // for cnn, drop out happens on kernels (weight matrix)
        // so the neurons are all active
        for(size_t k=0;k<__neuronDim.k;k++) // kernel
        {
            Matrix A( __neuronDim.i, __neuronDim.j);
            for(size_t i=0;i<__neuronDim.i;i++){
                for(size_t j=0;j<__neuronDim.j;j++)
                {
                    auto & a_vector = __neurons[k][i][j]->GetAVector();
                    if(__neurons[k][i][j]->IsActive())
                    {  
                        // make sure no over extract
                        //assert(a_vector.size() - 1 == l); // obsolete
                        //A[i][j] = a_vector.back();        // obsolete
                        A[i][j] = a_vector[sample_id];
                    }
                    else
                    {
                        A[i][j] = 0;
                    }
                }
            }
            sample_image_A.OutputImageFromKernel.push_back(A); // no need to filter
            sample_image_A_full.OutputImageFromKernel.push_back(A); // no need to filter
        }
        //__imageA.push_back(sample_image_A);
        __imageA[sample_id] = sample_image_A;
        __imageAFull[sample_id] = sample_image_A;
    }
    else // reserved for other layer types
    {
    }
}

// Assembly:
pushq %rbp
pushq %r15
pushq %r14
pushq %r13
pushq %r12
pushq %rbx
subq $0x88, %rsp
movq %rdi, %rbx
xorps %xmm0, %xmm0
movaps %xmm0, 0x30(%rsp)
xorl %eax, %eax
movq %rax, 0x40(%rsp)
movq %rax, 0x60(%rsp)
movaps %xmm0, 0x50(%rsp)
movl 0xc(%rdi), %eax
leal -0x1(%rax), %ecx
cmpl $0x2, %ecx
jb 0x7cbd
cmpl $0x4, %eax
je 0x7af7
testl %eax, %eax
jne 0x7e4b
cmpq $0x0, 0x3a8(%rbx)
movslq %esi, %rax
je 0x7c97
movq %rax, 0x8(%rsp)
xorl %r14d, %r14d
leaq 0x18(%rsp), %r15
movq %r14, (%rsp)
movq 0x398(%rbx), %rsi
movq 0x3a0(%rbx), %rdx
xorps %xmm0, %xmm0
movq %r15, %rdi
callq 0x1a892
movq 0x398(%rbx), %rax
testq %rax, %rax
je 0x7c06
movq 0x3a0(%rbx), %rcx
movq (%rsp), %rdx
leaq (%rdx,%rdx,2), %r13
xorl %ebp, %ebp
testq %rcx, %rcx
je 0x7bf8
leaq (,%rbp,2), %r12
addq %rbp, %r12
xorl %r14d, %r14d
movq 0x40(%rbx), %rax
movq (%rax,%r13,8), %rax
movq (%rax,%r12,8), %rax
movq (%rax,%r14,8), %rdi
callq 0x21db6
movq %rax, %r15
movq 0x40(%rbx), %rax
movq (%rax,%r13,8), %rax
movq (%rax,%r12,8), %rax
movq (%rax,%r14,8), %rdi
callq 0x2089a
testb %al, %al
je 0x7bc3
movq (%r15), %rax
movq 0x8(%rsp), %rcx
movsd (%rax,%rcx,8), %xmm0
movsd %xmm0, 0x10(%rsp)
leaq 0x18(%rsp), %rdi
movq %rbp, %rsi
callq 0x1adb2
movsd 0x10(%rsp), %xmm0
jmp 0x7bd3
leaq 0x18(%rsp), %rdi
movq %rbp, %rsi
callq 0x1adb2
xorps %xmm0, %xmm0
movq (%rax), %rax
movsd %xmm0, (%rax,%r14,8)
incq %r14
movq 0x3a0(%rbx), %rcx
cmpq %rcx, %r14
jb 0x7b6a
movq 0x398(%rbx), %rax
jmp 0x7bfa
xorl %ecx, %ecx
incq %rbp
cmpq %rax, %rbp
jb 0x7b53
leaq 0x50(%rsp), %rdi
leaq 0x18(%rsp), %r15
movq %r15, %rsi
callq 0x138a4
movq (%rsp), %r14
leaq (%r14,%r14,2), %rdx
shlq $0x3, %rdx
addq 0x108(%rbx), %rdx
leaq 0x70(%rsp), %r12
movq %r12, %rdi
movq %r15, %rsi
callq 0xb0f5
leaq 0x30(%rsp), %rdi
movq %r12, %rsi
callq 0x138a4
movq %r12, %rdi
callq 0x1abb8
cmpq 0x3b0(%rbx), %rax
jne 0x7e71
movq %r12, %rdi
callq 0x1abb8
cmpq 0x3b8(%rbx), %rdx
jne 0x7e90
movq %r12, %rdi
callq 0x1abf2
movq %r15, %rdi
callq 0x1abf2
incq %r14
cmpq 0x3a8(%rbx), %r14
jb 0x7b15
movq 0x8(%rsp), %rax
movq %rax, %r15
leaq (%rax,%rax,2), %rdi
shlq $0x3, %rdi
addq 0x138(%rbx), %rdi
leaq 0x30(%rsp), %rsi
callq 0x14112
leaq 0x50(%rsp), %r14
jmp 0x7e34
cmpq $0x0, 0x3a8(%rbx)
je 0x7e12
movslq %esi, %rax
movq %rax, 0x8(%rsp)
movq $0x0, (%rsp)
leaq 0x18(%rsp), %r14
movq 0x398(%rbx), %rsi
movq 0x3a0(%rbx), %rdx
movq %r14, %rdi
callq 0x1a76e
movq 0x398(%rbx), %rax
testq %rax, %rax
je 0x7dcc
movq 0x3a0(%rbx), %rcx
movq (%rsp), %rdx
leaq (%rdx,%rdx,2), %r15
xorl %r13d, %r13d
testq %rcx, %rcx
je 0x7dbe
leaq (,%r13,2), %r12
addq %r13, %r12
xorl %r14d, %r14d
movq 0x40(%rbx), %rax
movq (%rax,%r15,8), %rax
movq (%rax,%r12,8), %rax
movq (%rax,%r14,8), %rdi
callq 0x21db6
movq %rax, %rbp
movq 0x40(%rbx), %rax
movq (%rax,%r15,8), %rax
movq (%rax,%r12,8), %rax
movq (%rax,%r14,8), %rdi
callq 0x2089a
testb %al, %al
je 0x7d89
movq (%rbp), %rax
movq 0x8(%rsp), %rcx
movsd (%rax,%rcx,8), %xmm0
movsd %xmm0, 0x10(%rsp)
leaq 0x18(%rsp), %rdi
movq %r13, %rsi
callq 0x1adb2
movsd 0x10(%rsp), %xmm0
jmp 0x7d99
leaq 0x18(%rsp), %rdi
movq %r13, %rsi
callq 0x1adb2
xorps %xmm0, %xmm0
movq (%rax), %rax
movsd %xmm0, (%rax,%r14,8)
incq %r14
movq 0x3a0(%rbx), %rcx
cmpq %rcx, %r14
jb 0x7d2f
movq 0x398(%rbx), %rax
jmp 0x7dc0
xorl %ecx, %ecx
incq %r13
cmpq %rax, %r13
jb 0x7d18
leaq 0x30(%rsp), %rdi
leaq 0x18(%rsp), %r14
movq %r14, %rsi
callq 0x138a4
leaq 0x50(%rsp), %rdi
movq %r14, %rsi
callq 0x138a4
movq %r14, %rdi
callq 0x1abf2
movq (%rsp), %rcx
incq %rcx
movq %rcx, (%rsp)
cmpq 0x3a8(%rbx), %rcx
jb 0x7ce0
movq 0x8(%rsp), %rax
jmp 0x7e15
movslq %esi, %rax
movq %rax, %r15
leaq (%rax,%rax,2), %rdi
shlq $0x3, %rdi
addq 0x138(%rbx), %rdi
leaq 0x30(%rsp), %r14
movq %r14, %rsi
callq 0x14112
leaq (%r15,%r15,2), %rdi
shlq $0x3, %rdi
addq 0x198(%rbx), %rdi
movq %r14, %rsi
callq 0x14112
leaq 0x50(%rsp), %rdi
callq 0x1353a
leaq 0x30(%rsp), %rdi
callq 0x1353a
addq $0x88, %rsp
popq %rbx
popq %r12
popq %r13
popq %r14
popq %r15
popq %rbp
retq
leaq 0x1febf(%rip), %rdi     # 0x27d37
leaq 0x1f8cb(%rip), %rsi     # 0x2774a
leaq 0x1fedc(%rip), %rcx     # 0x27d62
movl $0x66a, %edx            # imm = 0x66A
callq 0x4160
leaq 0x1fef3(%rip), %rdi     # 0x27d8a
leaq 0x1f8ac(%rip), %rsi     # 0x2774a
leaq 0x1febd(%rip), %rcx     # 0x27d62
movl $0x66b, %edx            # imm = 0x66B
callq 0x4160
jmp 0x7eca
jmp 0x7eb5
jmp 0x7eb5
movq %rax, %rbx
leaq 0x70(%rsp), %rdi
callq 0x1abf2
jmp 0x7ed6
jmp 0x7ed3
jmp 0x7ed3
jmp 0x7eca
movq %rax, %rbx
jmp 0x7ee0
jmp 0x7ed3
jmp 0x7ed3
movq %rax, %rbx
leaq 0x18(%rsp), %rdi
callq 0x1abf2
leaq 0x50(%rsp), %rdi
callq 0x1353a
leaq 0x30(%rsp), %rdi
callq 0x1353a
movq %rbx, %rdi
callq 0x43e0