144 lines
4.2 KiB
C++
144 lines
4.2 KiB
C++
/*
|
|
This file is part of EqualizerAPO, a system-wide equalizer.
|
|
Copyright (C) 2015 Jonas Thedering
|
|
|
|
This program 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.
|
|
|
|
This program 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.
|
|
|
|
You should have received a copy of the GNU General Public License along
|
|
with this program; if not, write to the Free Software Foundation, Inc.,
|
|
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#include "stdafx.h"
|
|
#include <algorithm>
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
#include <Ks.h>
|
|
#include <KsMedia.h>
|
|
|
|
#include "LogHelper.h"
|
|
#include "ChannelHelper.h"
|
|
|
|
using namespace std;
|
|
|
|
unordered_map<wstring, int> ChannelHelper::channelNameToPosMap;
|
|
unordered_map<int, wstring> ChannelHelper::channelPosToNameMap;
|
|
// must come last, so that the static members are already initialized
|
|
ChannelHelper ChannelHelper::instance;
|
|
|
|
ChannelHelper::ChannelHelper()
|
|
{
|
|
channelNameToPosMap[L"L"] = SPEAKER_FRONT_LEFT;
|
|
channelNameToPosMap[L"R"] = SPEAKER_FRONT_RIGHT;
|
|
channelNameToPosMap[L"C"] = SPEAKER_FRONT_CENTER;
|
|
channelNameToPosMap[L"SUB"] = SPEAKER_LOW_FREQUENCY;
|
|
channelNameToPosMap[L"RL"] = SPEAKER_BACK_LEFT;
|
|
channelNameToPosMap[L"RR"] = SPEAKER_BACK_RIGHT;
|
|
channelNameToPosMap[L"RC"] = SPEAKER_BACK_CENTER;
|
|
channelNameToPosMap[L"SL"] = SPEAKER_SIDE_LEFT;
|
|
channelNameToPosMap[L"SR"] = SPEAKER_SIDE_RIGHT;
|
|
|
|
for (unordered_map<wstring, int>::iterator it = channelNameToPosMap.begin(); it != channelNameToPosMap.end(); it++)
|
|
channelPosToNameMap[it->second] = it->first;
|
|
}
|
|
|
|
int ChannelHelper::getDefaultChannelMask(int channelCount)
|
|
{
|
|
int channelMask;
|
|
|
|
switch (channelCount)
|
|
{
|
|
case 1:
|
|
channelMask = KSAUDIO_SPEAKER_MONO;
|
|
break;
|
|
case 2:
|
|
channelMask = KSAUDIO_SPEAKER_STEREO;
|
|
break;
|
|
case 4:
|
|
channelMask = KSAUDIO_SPEAKER_QUAD;
|
|
break;
|
|
case 6:
|
|
channelMask = KSAUDIO_SPEAKER_5POINT1_SURROUND;
|
|
break;
|
|
case 8:
|
|
channelMask = KSAUDIO_SPEAKER_7POINT1_SURROUND;
|
|
break;
|
|
default:
|
|
channelMask = 0;
|
|
}
|
|
|
|
return channelMask;
|
|
}
|
|
|
|
vector<wstring> ChannelHelper::getChannelNames(int channelCount, int channelMask)
|
|
{
|
|
vector<wstring> channelNames;
|
|
int c = 1;
|
|
for (int i = 0; i < 31; i++)
|
|
{
|
|
int channelPos = 1 << i;
|
|
if (channelMask & channelPos)
|
|
{
|
|
if (channelPosToNameMap.find(channelPos) != channelPosToNameMap.end())
|
|
channelNames.push_back(channelPosToNameMap[channelPos]);
|
|
else
|
|
channelNames.push_back(to_wstring((unsigned long long)c));
|
|
c++;
|
|
}
|
|
}
|
|
|
|
// handle channels not covered by channelMask
|
|
for (; c <= channelCount; c++)
|
|
channelNames.push_back(to_wstring((unsigned long long)c));
|
|
|
|
return channelNames;
|
|
}
|
|
|
|
int ChannelHelper::getChannelIndex(std::wstring word, const std::vector<std::wstring>& channelNames, bool allowNew)
|
|
{
|
|
int channelIndex = -1;
|
|
|
|
if (iswdigit(word[0]))
|
|
{
|
|
channelIndex = wcstol(word.c_str(), NULL, 10) - 1;
|
|
|
|
if (channelIndex < 0 || channelIndex >= (int)channelNames.size())
|
|
{
|
|
LogFStatic(L"Channel number %s out of range (1 - %d)", word.c_str(), channelNames.size());
|
|
channelIndex = -1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
vector<wstring>::const_iterator pos = find(channelNames.begin(), channelNames.end(), word);
|
|
|
|
if (pos == channelNames.end())
|
|
{
|
|
// Special handling to accept "wrong", but unambiguous positions
|
|
if (word == L"SL")
|
|
pos = find(channelNames.begin(), channelNames.end(), L"RL");
|
|
else if (word == L"SR")
|
|
pos = find(channelNames.begin(), channelNames.end(), L"RR");
|
|
else if (word == L"RL")
|
|
pos = find(channelNames.begin(), channelNames.end(), L"SL");
|
|
else if (word == L"RR")
|
|
pos = find(channelNames.begin(), channelNames.end(), L"SR");
|
|
}
|
|
|
|
if (pos != channelNames.end())
|
|
channelIndex = (int)(pos - channelNames.begin());
|
|
else if (!allowNew)
|
|
LogFStatic(L"Invalid channel position %s", word.c_str());
|
|
}
|
|
|
|
return channelIndex;
|
|
}
|