#!/usr/bin/python

# Copyright (C) 2012 The Android Open Source Project
#
# 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.

from consts import *
import numpy as np
import scipy as sp

# Example python script for signal processing in CTS audio
# There should be a function with the same name as the script
# Here, function example in example.py

# inputData : list of inputs with different types like int64, double,
#             mono or stereo audio data
# inputTypes: list of types for each input. Types are defined as TYPE_XXX
#             consts from consts.py
# return value: 3 elements list
#     element 0 : execution result value as defined as RESULT_XXX in consts.py
#     element 1 : outputData
#     element 2 : outputTypes
#
# This example takes 2 stereo data, 2 mono data, 2 i64, and 2 doubles
# and returns average as 1 stereo data, 1 mono data, 1 i64, and 1 double
# inputTypes for this function is expected to be
#   [ TYPE_STEREO, TYPE_STEREO, TYPE_MONO, TYPE_MONO, TYPE_I64, TYPE_I64,
#     TYPE_DOUBLE, TYPE_DOUBLE ]
# outputTypes will be [ TYPE_STEREO, TYPE_MONO, TYPE_I64, TYPE_DOUBLE ]
def example(inputData, inputTypes):
    output = []
    outputData = []
    outputTypes = []
    stereoInt = (inputData[0].astype(int) + inputData[1].astype(int))/2
    stereo = stereoInt.astype(np.int16)
    #print len(inputData[0]), len(inputData[1]), len(stereoInt), len(stereo)
    monoInt = (inputData[2].astype(int) + inputData[3].astype(int))/2
    mono = monoInt.astype(np.int16)
    #print len(inputData[2]), len(inputData[3]), len(monoInt), len(mono)
    i64Val = (inputData[4] + inputData[5])/2
    doubleVal = (inputData[6] + inputData[7])/2
    outputData.append(stereo)
    outputTypes.append(TYPE_STEREO)
    outputData.append(mono)
    outputTypes.append(TYPE_MONO)
    outputData.append(i64Val)
    outputTypes.append(TYPE_I64)
    outputData.append(doubleVal)
    outputTypes.append(TYPE_DOUBLE)
    output.append(RESULT_OK)
    output.append(outputData)
    output.append(outputTypes)

    return output