| 517 | } |
| 518 | |
| 519 | int32_t calcIndex(SampleBuffer& buffer, int32_t c, int32_t h, int32_t w) |
| 520 | { |
| 521 | int32_t index; |
| 522 | |
| 523 | if (!buffer.desc.channelPivot) |
| 524 | { |
| 525 | index = c / buffer.desc.dims[4] * buffer.desc.dims[2] * buffer.desc.dims[3] * buffer.desc.dims[4] |
| 526 | + h * buffer.desc.dims[3] * buffer.desc.dims[4] + w * buffer.desc.dims[4] + c % buffer.desc.dims[4]; |
| 527 | } |
| 528 | else |
| 529 | { |
| 530 | index = h * buffer.desc.dims[3] * buffer.desc.dims[2] + w * buffer.desc.dims[3] + c; |
| 531 | } |
| 532 | |
| 533 | return index; |
| 534 | } |
| 535 | |
| 536 | //! |
| 537 | //! \brief Reformats the buffer. Src and dst buffers should be of same datatype and dims. |