tst_scene_render.py

Go to the documentation of this file.

#!/usr/bin/env python
 
# *****************************************************************************
# *                            cloudFPGA
# *                Copyright 2016 -- 2022 IBM Corporation
# * 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.
# *----------------------------------------------------------------------------
 
# *****************************************************************************
# *                            OpenCV
# *                Copyright 2021 -- OpenCV team
# * 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.
# *----------------------------------------------------------------------------
 
 
# Python 2/3 compatibility
from __future__ import print_function
 
import numpy as np
import cv2 as cv
 
from numpy import pi, sin, cos
 
 
defaultSize = 512
 
class TestSceneRender():
 
    def __init__(self, bgImg = None, fgImg = None,
        deformation = False, speed = 0.25, **params):
        self.timetime = 0.0
        self.timeSteptimeStep = 1.0 / 30.0
        self.foregroundforeground = fgImg
        self.deformationdeformation = deformation
        self.speedspeed = speed
 
        if bgImg is not None:
            self.sceneBgsceneBg = bgImg.copy()
        else:
            self.sceneBgsceneBg = np.zeros(defaultSize, defaultSize, np.uint8)
 
        self.ww = self.sceneBgsceneBg.shape[0]
        self.hh = self.sceneBgsceneBg.shape[1]
 
        if fgImg is not None:
            self.foregroundforeground = fgImg.copy()
            self.centercenter = self.currentCentercurrentCenter = (int(self.ww/2 - fgImg.shape[0]/2), int(self.hh/2 - fgImg.shape[1]/2))
 
            self.xAmplxAmpl = self.sceneBgsceneBg.shape[0] - (self.centercenter[0] + fgImg.shape[0])
            self.yAmplyAmpl = self.sceneBgsceneBg.shape[1] - (self.centercenter[1] + fgImg.shape[1])
 
        self.initialRectinitialRect = np.array([ (self.hh/2, self.ww/2), (self.hh/2, self.ww/2 + self.ww/10),
         (self.hh/2 + self.hh/10, self.ww/2 + self.ww/10), (self.hh/2 + self.hh/10, self.ww/2)]).astype(int)
        self.currentRectcurrentRect = self.initialRectinitialRect
 
    def getXOffset(self, time):
        return int( self.xAmplxAmpl*cos(time*self.speedspeed))
 
 
    def getYOffset(self, time):
        return int(self.yAmplyAmpl*sin(time*self.speedspeed))
 
    def setInitialRect(self, rect):
        self.initialRectinitialRect = rect
 
    def getRectInTime(self, time):
 
        if self.foregroundforeground is not None:
            tmp = np.array(self.centercenter) + np.array((self.getXOffsetgetXOffset(time), self.getYOffsetgetYOffset(time)))
            x0, y0 = tmp
            x1, y1 = tmp + self.foregroundforeground.shape[0:2]
            return np.array([y0, x0, y1, x1])
        else:
            x0, y0 = self.initialRectinitialRect[0] + np.array((self.getXOffsetgetXOffset(time), self.getYOffsetgetYOffset(time)))
            x1, y1 = self.initialRectinitialRect[2] + np.array((self.getXOffsetgetXOffset(time), self.getYOffsetgetYOffset(time)))
            return np.array([y0, x0, y1, x1])
 
    def getCurrentRect(self):
 
        if self.foregroundforeground is not None:
 
            x0 = self.currentCentercurrentCenter[0]
            y0 = self.currentCentercurrentCenter[1]
            x1 = self.currentCentercurrentCenter[0] + self.foregroundforeground.shape[0]
            y1 = self.currentCentercurrentCenter[1] + self.foregroundforeground.shape[1]
            return np.array([y0, x0, y1, x1])
        else:
            x0, y0 = self.currentRectcurrentRect[0]
            x1, y1 = self.currentRectcurrentRect[2]
            return np.array([x0, y0, x1, y1])
 
    def getNextFrame(self):
        img = self.sceneBgsceneBg.copy()
 
        if self.foregroundforeground is not None:
            self.currentCentercurrentCenter = (self.centercenter[0] + self.getXOffsetgetXOffset(self.timetime), self.centercenter[1] + self.getYOffsetgetYOffset(self.timetime))
            img[self.currentCentercurrentCenter[0]:self.currentCentercurrentCenter[0]+self.foregroundforeground.shape[0],
             self.currentCentercurrentCenter[1]:self.currentCentercurrentCenter[1]+self.foregroundforeground.shape[1]] = self.foregroundforeground
        else:
            self.currentRectcurrentRect = self.initialRectinitialRect + np.int( 30*cos(self.timetime*self.speedspeed) + 50*sin(self.timetime*self.speedspeed))
            if self.deformationdeformation:
                self.currentRectcurrentRect[1:3] += int(self.hh/20*cos(self.timetime))
            cv.fillConvexPoly(img, self.currentRectcurrentRect, (0, 0, 255))
 
        self.timetime += self.timeSteptimeStep
        return img
 
    def resetTime(self):
        self.timetime = 0.0
 
 
def main():
    backGr = cv.imread(cv.samples.findFile('graf1.png'))
    fgr = cv.imread(cv.samples.findFile('box.png'))
 
    render = TestSceneRender(backGr, fgr)
 
    while True:
 
        img = render.getNextFrame()
        cv.imshow('img', img)
 
        ch = cv.waitKey(3)
        if  ch == 27:
            break
 
    print('Done')
 
 
if __name__ == '__main__':
    print(__doc__)
    main()
    cv.destroyAllWindows()