1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
/*******************************************************************************
* Copyright 2015-2016 Juan Francisco Crespo Galán
*
* 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 "fx/HRTF.h"
#include "Exception.h"
#include <cmath>
AUD_NAMESPACE_BEGIN
HRTF::HRTF() :
HRTF(std::make_shared<FFTPlan>(0.0))
{
}
HRTF::HRTF(std::shared_ptr<FFTPlan> plan) :
m_plan(plan)
{
m_specs.channels = CHANNELS_INVALID;
m_specs.rate = 0;
m_empty = true;
}
bool HRTF::addImpulseResponse(std::shared_ptr<StreamBuffer> impulseResponse, float azimuth, float elevation)
{
Specs spec = impulseResponse->getSpecs();
azimuth = std::fmod(azimuth, 360);
if(azimuth < 0)
azimuth += 360;
if((spec.channels != CHANNELS_MONO) || (spec.rate != m_specs.rate && m_specs.rate > 0.0))
return false;
m_hrtfs[elevation][azimuth] = std::make_shared<ImpulseResponse>(impulseResponse, m_plan);
m_specs.channels = CHANNELS_MONO;
m_specs.rate = spec.rate;
m_empty = false;
return true;
}
std::pair<std::shared_ptr<ImpulseResponse>, std::shared_ptr<ImpulseResponse>> HRTF::getImpulseResponse(float &azimuth, float &elevation)
{
if(m_hrtfs.empty())
return std::make_pair(nullptr, nullptr);
azimuth = std::fmod(azimuth, 360);
if(azimuth < 0)
azimuth += 360;
std::shared_ptr<ImpulseResponse> R, L;
float az = 0, el = 0, dif=0, minDif=360;
for(auto elem : m_hrtfs)
{
dif = std::fabs(elevation - elem.first);
if(dif < minDif)
{
minDif = dif;
el = elem.first;
}
}
elevation = el;
dif = 0;
minDif = 360;
for(auto elem : m_hrtfs[elevation])
{
dif = std::fabs(azimuth - elem.first);
if(dif < minDif)
{
minDif = dif;
az = elem.first;
R = elem.second;
}
}
azimuth = az;
float azL = 360 - azimuth;
if(azL == 360)
azL = 0;
auto iter = m_hrtfs[elevation].find(azL);
if(iter != m_hrtfs[elevation].end())
L = iter->second;
else
{
dif = 0;
minDif = 360;
for(auto elem : m_hrtfs[elevation])
{
dif = std::fabs(azL - elem.first);
if(dif < minDif)
{
minDif = dif;
L = elem.second;
}
}
}
return std::make_pair(L, R);
}
Specs HRTF::getSpecs()
{
return m_specs;
}
bool HRTF::isEmpty()
{
return m_empty;
}
AUD_NAMESPACE_END
|