AecObjects.h

Go to the documentation of this file.

/*
 * This file is part of the Wave Arts VQE library.
 * Copyright (C) 2011 Wave Arts, Inc. (http://wavearts.com)
 *
 * VQE is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * VQE is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * A copy of the GNU General Public License is included with VQE,
 * and can be found online at <http://www.gnu.org/licenses/>.
 *
 * If you'd like to distribute a closed-source product which uses VQE,
 * you must obtain a commercial VQE license from Wave Arts.
 */
//
// AEC class objects and calibrator. The calibrator could be put in
// a separate file which would eliminate this file's dependency on WaAudioDev.h.
//


#ifndef _AEC_OBJECTS_H
#define _AEC_OBJECTS_H

#include "WaAudioDev.h"
#include "AecCal.h"
#include "AecLatCal.h"
#include "Aes.h"
#include "Aec.h"
#include "WaList.h"

class AecProcessor {
public:
        virtual ~AecProcessor();

        virtual void Process(float *refBuf, float *recBuf, int n) = 0;

        virtual void Reset(bool isHard) = 0;

        virtual void SetLatency(int latency);
};


//=============================================================================

typedef struct {
        char inDevName[128];    
        char outDevName[128];   
        int sampRate;                   
        int latency;                    
} AecLatData;

typedef struct AecLatDataItem_tag {
        struct AecLatDataItem_tag *next;        
        AecLatData data;                                        
} AecLatDataItem;

class AecLatencyCache {
public:
        static AecLatencyCache *GetInstance();
        bool GetData(AecLatData *data);
        void SetData(AecLatData *data);
private:
        AecLatencyCache();
        bool EqualData(AecLatData *a, AecLatData *b);
        AecLatData *FindData(AecLatData *data);
        LIST list;
};

//=============================================================================

class AecLatCalObject : public AecProcessor {
public:
        AecLatCalObject(AecLatCalParam *param, AecProcessor *aec);
        ~AecLatCalObject();
        void Process(float *refBuf, float *recBuf, int n);
        void Reset(bool isHard);
        AecLatCal *GetLatCal();
        bool GetLatency(int *pLatency);
protected:
        AecLatCal *latCal;      
        AecProcessor *aec;      
        bool isComplete;        
        int latency;            
};

//=============================================================================

class AesObject : public AecProcessor {
public:
        AesObject(AesParam *param);
        ~AesObject();
        void Process(float *refBuf, float *recBuf, int n);
        void Reset(bool isHard);
        void SetLatency(int latency);
        Aes *GetAes();

protected:
        Aes *aes;       
};


//=============================================================================


class AecObject : public AecProcessor {
public:
        AecObject(AecParam *param);
        ~AecObject();
        void Process(float *refBuf, float *recBuf, int n);
        void Reset(bool isHard);
        void SetLatency(int latency);
        Aec *GetAec();  

protected:
        Aec *aec;       
};

//=============================================================================

typedef struct {
        CalParam calParam;              
        bool tryUpsampling;             
        double syncThresh;              
} AecCalibratorParam;

typedef void AecCalibratorRecFn(void *arg, float *buf, int n, int sampRate);

class AecCalibrator {
public:
        AecCalibrator(WaAudioDev *dev, AecCalibratorParam *calibParam);

        ~AecCalibrator();

        AecCalErr Calibrate(WadParam *param, AecSyncParam *syncParam, float inVol, float outVol);

        int InCallback(float *buf, int numFrames, int numChan);
        int OutCallback(float *buf, int numFrames, int numChan);

        void GetLastResults(CalResults *pResults);

        void SetRecCallback(AecCalibratorRecFn *fn, void *arg);

        static void GetDefaultParam(AecCalibratorParam *calibParam);

protected:
        WaAudioDev *dev;        
        float inVol;            
        float outVol;           
        float *playBuf;         
        int playLen;            
        int playIndex;          
        float *recBuf;          
        int recLen;                     
        int recIndex;           
        AecCalibratorParam calibParam;  
        CalResults results;     
        AecCalibratorRecFn *recFn;      
        void *arg;                      

        bool TryUpsampling();
        bool IsSynchronized(double ratio);
        AecCalErr CalDeviceAtRateBufs(WadParam *param, int devSampRate, CalResults *results);
        AecCalErr CalDeviceAtRate(WadParam *param, int devSampRate, CalResults *results);
};

#endif