.. _program_listing_file__tmp_ws_src_vitis_common_include_aie_imgproc_xf_accumulateweighted_aie.hpp:

Program Listing for File xf_accumulateweighted_aie.hpp
======================================================

|exhale_lsh| :ref:`Return to documentation for file <file__tmp_ws_src_vitis_common_include_aie_imgproc_xf_accumulateweighted_aie.hpp>` (``/tmp/ws/src/vitis_common/include/aie/imgproc/xf_accumulateweighted_aie.hpp``)

.. |exhale_lsh| unicode:: U+021B0 .. UPWARDS ARROW WITH TIP LEFTWARDS

.. code-block:: cpp

   /*
    * Copyright 2021 Xilinx, Inc.
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *     http://www.apache.org/licenses/LICENSE-2.0
    *
    * Unless required by applicable law or agreed to in writing, software
    * distributed under the License is distributed on an "AS IS" BASIS,
    * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    * See the License for the specific language governing permissions and
    * limitations under the License.
    */
   
   #include <adf.h>
   #include <aie_api/aie.hpp>
   #include <common/xf_aie_utils.hpp>
   
   #ifndef _AIE_ACCUMULATEWEIGHTED_H_
   #define _AIE_ACCUMULATEWEIGHTED_H_
   
   #define PARALLEL_FACTOR_16b 16
   
   #define SRS_SHIFT 8
   
   namespace xf {
   namespace cv {
   namespace aie {
   
   template <typename T, unsigned int N>
   __attribute__((noinline)) void accumulateweighted(const T* restrict img_in1,
                                                     const T* restrict img_in2,
                                                     T* restrict img_out,
                                                     const T& img_width,
                                                     const T& img_height,
                                                     const float& scale) {
       //    int16_t fix_scale=float2fix(scale,SRS_SHIFT) ;//(float)scale*(1<<8);
       //    int16_t fix_scale1=float2fix((1-scale),SRS_SHIFT) ;// (1-scale)*(1<<8);
       T fix_scale = ::aie::to_fixed<float>(scale, SRS_SHIFT);        //(float)scale*(1<<8);
       T fix_scale1 = ::aie::to_fixed<float>((1 - scale), SRS_SHIFT); //(float)scale*(1<<8);
   
       ::aie::vector<T, N> weight(fix_scale, fix_scale1);
   
       for (int j = 0; j < (img_height * img_width); j += N) // 16x samples per loop
           chess_prepare_for_pipelining chess_loop_range(14, ) {
               ::aie::vector<T, N> data_buf1 = ::aie::load_v<16>(img_in1);
               img_in1 += N;
               ::aie::vector<T, N> data_buf2 = ::aie::load_v<16>(img_in2);
               img_in2 += N;
   
               ::aie::accum<acc32, N> acc =
                   ::aie::accumulate<16>(weight, data_buf1, data_buf2); // weight[0] * data_buf1 + weight[1] * data_buf2
   
               ::aie::store_v(img_out, acc.template to_vector<T>(SRS_SHIFT));
   
               img_out += 16;
           }
   }
   
   __attribute__((noinline)) void accumulateweighted_api(input_window_int16* img_in1,
                                                         input_window_int16* img_in2,
                                                         output_window_int16* img_out,
                                                         const float& alpha) {
       int16* restrict img_in_ptr = (int16*)img_in1->ptr;
       int16* restrict img_in_ptr1 = (int16*)img_in2->ptr;
       int16* restrict img_out_ptr = (int16*)img_out->ptr;
   
       const int16_t img_width = xfcvGetTileWidth(img_in_ptr);
       const int16_t img_height = xfcvGetTileHeight(img_in_ptr);
   
       xfcvCopyMetaData(img_in_ptr, img_out_ptr);
       xfcvUnsignedSaturation(img_out_ptr);
   
       int16_t* restrict ptr0 = xfcvGetImgDataPtr(img_in_ptr);
       int16_t* restrict ptr1 = xfcvGetImgDataPtr(img_in_ptr1);
       int16_t* restrict ptr_out = xfcvGetImgDataPtr(img_out_ptr);
   
       accumulateweighted<int16_t, PARALLEL_FACTOR_16b>(ptr0, ptr1, ptr_out, img_width, img_height, alpha);
   }
   
   } // aie
   } // cv
   } // xf
   #endif