diff options
author | toma <toma@283d02a7-25f6-0310-bc7c-ecb5cbfe19da> | 2009-11-25 17:56:58 +0000 |
---|---|---|
committer | toma <toma@283d02a7-25f6-0310-bc7c-ecb5cbfe19da> | 2009-11-25 17:56:58 +0000 |
commit | e2de64d6f1beb9e492daf5b886e19933c1fa41dd (patch) | |
tree | 9047cf9e6b5c43878d5bf82660adae77ceee097a /noatun/library/vequalizer.cpp | |
download | tdemultimedia-e2de64d6f1beb9e492daf5b886e19933c1fa41dd.tar.gz tdemultimedia-e2de64d6f1beb9e492daf5b886e19933c1fa41dd.zip |
Copy the KDE 3.5 branch to branches/trinity for new KDE 3.5 features.
BUG:215923
git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/kdemultimedia@1054174 283d02a7-25f6-0310-bc7c-ecb5cbfe19da
Diffstat (limited to 'noatun/library/vequalizer.cpp')
-rw-r--r-- | noatun/library/vequalizer.cpp | 936 |
1 files changed, 936 insertions, 0 deletions
diff --git a/noatun/library/vequalizer.cpp b/noatun/library/vequalizer.cpp new file mode 100644 index 00000000..4b131e8a --- /dev/null +++ b/noatun/library/vequalizer.cpp @@ -0,0 +1,936 @@ +/* + * Copyright (c) 2003 Charles Samuels <charles@kde.org> + * + * This file is hereby licensed under the GNU General Public License version + * 2 or later at your option. + * + * This file is licensed under the Qt Public License version 1 with the + * condition that the licensed will be governed under the Laws of California + * (USA) instead of Norway. Disputes will be settled in Santa Clara county + * courts. + * + * This file is licensed under the following additional license. Be aware + * that it is identical to the BSD license, except for the added clause 3: + + Redistribution and use in source and binary forms, with or without + modification, are permitted provided that the following conditions + are met: + + 1. Redistributions of source code must retain the above copyright + notice, this list of conditions and the following disclaimer. + 2. Redistributions in binary form must reproduce the above copyright + notice, this list of conditions and the following disclaimer in the + documentation and/or other materials provided with the distribution. + 3. By integrating this software into any other software codebase, you waive + all rights to any patents associated with the stated codebase. + + THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR + IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES + OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. + IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, + INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT + NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF + THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +*/ + +#include "vequalizer.h" +#include "engine.h" +#include "spline.h" +#include "ksaver.h" + +#include <noatunarts.h> +#include <app.h> +#include <player.h> + +#include <common.h> +#include <dynamicrequest.h> +#include <artsflow.h> +#include <soundserver.h> + +#include <ktempfile.h> +#include <kio/netaccess.h> +#include <kstandarddirs.h> +#include <kconfig.h> +#include <klocale.h> + +#include <qdom.h> +#include <qtextstream.h> +#include <qfile.h> + +#include <math.h> + +#define EQ (napp->vequalizer()) +#define EQBACK (napp->player()->engine()->equalizer()) + +using std::vector; + +static const double splineTension = 4.0; + +VBandsInterface::VBandsInterface() +{ + +} + +VBandsInterface::~VBandsInterface() +{ + +} + + +VBand VBandsInterface::operator [] (int num) +{ + return band(num); +} + +struct VBand::Private +{ + int refs; + + int index; + int start, end; + VBandsInterface *bands; +}; + +VBand::VBand(VBandsInterface *bands, int index, int start, int end) +{ + d = new Private; + d->refs=1; + d->index = index; + d->start = start; + d->end = end; + d->bands = bands; +} + +VBand::~VBand() +{ + if (--d->refs == 0) + { + delete d; + } +} + +VBand::VBand(const VBand ©) +{ + d=0; + operator=(copy); +} + +VBand & VBand::operator =(const VBand ©) +{ + if (d && --d->refs == 0) + { + delete d; + } + + d= copy.d; + d->refs++; + return *this; +} + +int VBand::level() const +{ + return d->bands->level(d->index); +} + +void VBand::setLevel(int l) +{ + d->bands->setLevel(d->index, l); +} + +int VBand::start() const +{ + return d->start; +} + +int VBand::end() const +{ + return d->end; +} + +int VBand::center() const +{ + return (d->start + d->end)/2; +} + +static QString formatFreq(int f, bool withHz) +{ + QString format; + if (f<991) + format=QString::number(f); + else + format=QString::number((int)((f+500)/1000.0))+"k"; + + if (withHz) + format+="Hz"; + + return format; +} + +QString VBand::formatStart(bool withHz) const +{ + return formatFreq(d->start, withHz); +} + +QString VBand::formatEnd(bool withHz) const +{ + return formatFreq(d->end, withHz); +} + +QString VBand::format(bool withHz) const +{ + return formatFreq(center(), withHz); +} + + + + +struct VInterpolation::Private +{ + int bands; + Spline spline; + +}; + +VInterpolation::VInterpolation(int bands) +{ + d = new Private; + d->bands = bands; +} + +VInterpolation::~VInterpolation() +{ + delete d; +} + +int VInterpolation::bands() const +{ + return d->bands; +} + +void VInterpolation::getFrequencies(int num, int *low, int *high) const +{ + QValueList<int> fs = VEqualizer::frequencies(bands()); + + if (num == 0) *low = 1; + else *low = fs[num-1]+1; + *high=fs[num]; +} + + +VBand VInterpolation::band(int num) +{ + int low, high; + getFrequencies(num, &low, &high); + return VBand(this, num, low, high); +} + +int VInterpolation::level(int index) const +{ + const_cast<VInterpolation*>(this)->refresh(); + double x = onSpline(index); + + return int(d->spline[x*splineTension]); +} + +void VInterpolation::setLevel(int index, int level) +{ + refresh(); + + double numbands = double(bands()); + Spline spline; + + for (int i=0; i < numbands; ++i) + { + VBand b = band(i); + spline.add(i*splineTension, double(index == i ? level : b.level())); + } + + int realbands = EQ->bands(); + QValueList<int> values; + for (int i=0; i < realbands; ++i) + { + // i + // realbands numbands + double x = double(i)*numbands/double(realbands)*splineTension; + double value = spline[x]; + values.append(int(value)); + } + EQ->setLevels(values); +} + +double VInterpolation::onSpline(int bandNum) const +{ + double maxReal(EQ->bands()); // 2 + double maxInter(bands()); // 4 + double offset(bandNum); // 4 + + return maxReal/maxInter*offset; +} + +void VInterpolation::refresh() +{ + d->spline.clear(); + + VEqualizer *eq = EQ; + + for (int i=0; i < eq->bands(); ++i) + { + VBand band = eq->band(i); + d->spline.add(double(i*splineTension), double(band.level())); + } + +} + +struct VEqualizer::Private +{ + struct BandInfo + { + int level; + int start; + int end; + }; + + std::vector<BandInfo> bands; + int preamp; +}; + +/* rate 4 +27 54 0-108 108 +81 163 109-217 108 +243 514 218-810 269 +729 1621 811-2431 1620 +2187 4661 2432-7290 4858 +6561 13645 7291+ 12708 +*/ +VEqualizer::VEqualizer() +{ + d = new Private; + d->preamp=1; + setBands(6, false); +} + +void VEqualizer::init() +{ + KURL url; + url.setPath(kapp->dirs()->localkdedir()+"/share/apps/noatun/equalizer"); + if(!load(url)) + { + setPreamp(0); + disable(); + } + else + { + KConfig *config=kapp->config(); + config->setGroup("Equalizer"); + setEnabled(config->readBoolEntry("enabled", false)); + } +} + + +VEqualizer::~VEqualizer() +{ + KURL url; + url.setPath(kapp->dirs()->localkdedir()+"/share/apps/noatun/equalizer"); + save(url, "auto"); + + delete d; +} + +int VEqualizer::start() +{ + return 20; +} + +int VEqualizer::end() +{ + return 20000; +} + +int VEqualizer::maxBands() const +{ + return 14; +} + +int VEqualizer::minBands() const +{ + return 2; +} + + +QValueList<int> VEqualizer::frequencies(int _num) +{ +#if 0 + QValueList<int> fs; + fs += 108; + fs += 217; + fs += 810; + fs += 2431; + fs += 7290; + fs += 19999; + return fs; + +#else + + + + // we're looking for + // ?^_num = end()-start() + // so log[?] (end()-start()) = _num + // log(end()-start()) / log(?) = _num + // log(end()-start()) = _num * log(?) + // log(end()-start()) / _num = log(?) + // ? = 10^(log(end()-start()) / _num) + + double num = double(_num); + double vstart = double(start()); + double vend = double(end()); + const double base = ::pow(10.0, ::log10(vend-vstart)/num); + + QValueList<int> fs; + + for (double i=1.0; i <= num; i++) + { + double f = ::pow(base, i); + f += vstart; + fs.append(int(f)); + } + + return fs; +#endif +} + +void VEqualizer::setBands(int num, bool interpolate) +{ + if (interpolate) + { + setBands(num); + return; + } + + if (num > maxBands()) + num=maxBands(); + else if (num < minBands()) num = minBands(); + + if (num == bands()) return; + + QValueList<int> fs = VEqualizer::frequencies(num); + std::vector<Private::BandInfo> modified; + + int bstart=0; + for (QValueList<int>::Iterator i(fs.begin()); i != fs.end(); ++i) + { + Private::BandInfo info; + info.start = bstart+1; + info.level = 0; + info.end = *i; + bstart = info.end; + + modified.push_back(info); + } + d->bands = modified; + update(true); + + emit changedBands(); + emit changed(); +} + +void VEqualizer::setPreamp(int preamp) +{ + d->preamp = preamp; + EQBACK->preamp(pow(2,float(preamp)/100.0)); + emit changed(); + emit preampChanged(); + emit preampChanged(preamp); +} + + +void VEqualizer::setBands(int num) +{ + if (num == bands()) return; + VInterpolation ip(num); + + std::vector<Private::BandInfo> modified; + + for (int i=0; i < num; ++i) + { + Private::BandInfo info; + VBand b = ip[i]; + info.level = b.level(); + info.start = b.start(); + info.end = b.end(); + + modified.push_back(info); + } + d->bands = modified; + update(true); + + emit changedBands(); + emit changed(); +} + +VBand VEqualizer::band(int num) +{ + return VBand(this, num, d->bands[num].start, d->bands[num].end); +} + +int VEqualizer::bands() const +{ + return d->bands.size(); +} + +bool VEqualizer::isEnabled() const +{ + return bool(EQBACK->enabled()); +} + +int VEqualizer::preamp() const +{ + return d->preamp; +} + +void VEqualizer::enable() +{ + setEnabled(true); +} + +void VEqualizer::disable() +{ + setEnabled(false); +} + +int VEqualizer::level(int index) const +{ + return d->bands[index].level; +} + +void VEqualizer::setLevel(int index, int level) +{ + d->bands[index].level = level; + update(); + emit changed(); + emit modified(); +} + +void VEqualizer::setLevels(const QValueList<int> &levels) +{ + int index=0; + for ( + QValueList<int>::ConstIterator i(levels.begin()); + i != levels.end(); ++i + ) + { + d->bands[index].level = *i; + index++; + } + update(); + emit changed(); + emit modified(); +} + + +void VEqualizer::setEnabled(bool e) +{ + update(true); // just in case + EQBACK->enabled((long)e); + KConfig *config=kapp->config(); + config->setGroup("Equalizer"); + config->writeEntry("enabled", e); + config->sync(); + + emit enabled(e); + if (e) + emit enabled(); + else + emit disabled(); +} + +void VEqualizer::update(bool full) +{ + std::vector<float> levels; + std::vector<float> mids; + std::vector<float> widths; + + for (unsigned int i=0; i < d->bands.size(); ++i) + { + Private::BandInfo &info = d->bands[i]; + levels.push_back(::pow(2.0, float(info.level)/50.0)); + if (full) + { + int mid = info.start + info.end; + mids.push_back(float(mid)*0.5); + widths.push_back(float(info.end - info.start)); + } + } + if (full) + EQBACK->set(levels, mids, widths); + else + EQBACK->levels(levels); +} + + + +bool VEqualizer::save(const KURL &filename, const QString &friendly) const +{ + Noatun::KSaver saver(filename); + if(!saver.open()) return false; + + saver.textStream() << toString(friendly); + saver.close(); + + return saver.close(); +} + +bool VEqualizer::load(const KURL &filename) +{ + QString dest; + if(KIO::NetAccess::download(filename, dest, 0L)) + { + QFile file(dest); + if (!file.open(IO_ReadOnly)) + return false; + + QTextStream t(&file); + QString str = t.read(); + fromString(str); + return true; + } + return false; +} + +QString VEqualizer::toString(const QString &name) const +{ + QDomDocument doc("noatunequalizer"); + doc.setContent(QString("<!DOCTYPE NoatunEqualizer><noatunequalizer/>")); + QDomElement docElem = doc.documentElement(); + + { + docElem.setAttribute("level", preamp()); + docElem.setAttribute("name", name); + docElem.setAttribute("version", napp->version()); + } + + int bandc = bands(); + for (int i=0; i < bandc; ++i) + { + VBand band = const_cast<VEqualizer*>(this)->operator[](i); + QDomElement elem = doc.createElement("band"); + elem.setAttribute("start", band.start()); + elem.setAttribute("end", band.end()); + elem.setAttribute("level", band.level()); + + docElem.appendChild(elem); + } + + return doc.toString(); +} + +bool VEqualizer::fromString(const QString &str) +{ + QDomDocument doc("noatunequalizer"); + if (!doc.setContent(str)) + return false; + + QDomElement docElem = doc.documentElement(); + if (docElem.tagName()!="noatunequalizer") + return false; + + setPreamp(docElem.attribute("level", "0").toInt()); + + std::vector<Private::BandInfo> modified; + for (QDomNode n = docElem.firstChild(); !n.isNull(); n = n.nextSibling()) + { + QDomElement e = n.toElement(); + if(e.isNull()) continue; + if (e.tagName().lower() != "band") continue; + + Private::BandInfo data; + + data.level = e.attribute("level", "0").toInt(); + data.start = e.attribute("start", "1").toInt(); + data.end = e.attribute("end", "19999").toInt(); + + modified.push_back(data); + } + d->bands = modified; + update(true); + + emit changedBands(); + emit changed(); + return true; +} + +static QString makePresetFile() +{ + QString basedir=kapp->dirs()->localkdedir()+"/share/apps/noatun/eq.preset/"; + // now append a filename that doesn't exist + KStandardDirs::makeDir(basedir); + QString fullpath; + int num=0; + do + { + if (num) + fullpath=basedir+"preset."+QString::number(num); + else + fullpath=basedir+"preset"; + num++; + } + while (QFile(fullpath).exists()); + return fullpath; +} + +VPreset VEqualizer::createPreset(const QString &name, bool smart) +{ + if (presetExists(name) && !smart) return VPreset(); + QString nameReal=name; + { + int number=1; + while (presetExists(nameReal)) + { + nameReal=name+" ("+QString::number(number)+')'; + number++; + } + } + + VPreset preset(makePresetFile()); + preset.setName(nameReal); + + save(preset.file(), nameReal); + KConfig *config=kapp->config(); + config->setGroup("Equalizer"); + QStringList list = config->readListEntry("presets"); + list += preset.file(); + config->writeEntry("presets", list); + config->sync(); + + emit created(preset); + return preset; +} + + +QValueList<VPreset> VEqualizer::presets() const +{ + KConfig *conf=KGlobal::config(); + conf->setGroup("Equalizer"); + + QStringList list; + if (conf->hasKey("presets")) + { + list=conf->readListEntry("presets"); + } + else + { + list=kapp->dirs()->findAllResources("data", "noatun/eq.preset/*"); + conf->writeEntry("presets", list); + conf->sync(); + } + + QValueList<VPreset> presets; + + for (QStringList::Iterator i = list.begin(); i!=list.end(); ++i) + { + QFile file(*i); + if (!file.open(IO_ReadOnly)) continue; + + QDomDocument doc("noatunequalizer"); + if (!doc.setContent(&file)) continue; + + QDomElement docElem = doc.documentElement(); + if (docElem.tagName()!="noatunequalizer") continue; + + presets.append(VPreset(*i)); + } + return presets; +} + +VPreset VEqualizer::presetByName(const QString &name) +{ + QValueList<VPreset> pr = presets(); + for ( + QValueList<VPreset>::Iterator i(pr.begin()); + i != pr.end(); ++i + ) + { + if ((*i).name() == name) + return *i; + } + return VPreset(); +} + +VPreset VEqualizer::presetByFile(const QString &file) +{ + KConfig *conf=KGlobal::config(); + conf->setGroup("Equalizer"); + QStringList list=kapp->config()->readListEntry("presets"); + if (list.contains(file)) + return VPreset(file); + return VPreset(); +} + +bool VEqualizer::presetExists(const QString &name) const +{ + QValueList<VPreset> list=presets(); + for ( + QValueList<VPreset>::Iterator i(list.begin()); + i != list.end(); ++i + ) + { + if ((*i).name() == name) + return true; + } + return false; +} + + + +struct VPreset::Private +{ + QString file; +}; + + + +VPreset::VPreset(const QString &file) +{ + d = new Private; + d->file = file; + +} + + +VPreset::VPreset() +{ + d = new Private; +} + +VPreset::VPreset(const VPreset ©) +{ + d = new Private; + operator =(copy); +} + +VPreset::~VPreset() +{ + delete d; +} + +bool VPreset::operator ==(const VPreset &other) const +{ + return name() == other.name(); +} + +VPreset & VPreset::operator=(const VPreset ©) +{ + d->file = copy.file(); + return *this; +} + +QString VPreset::name() const +{ + QFile file(d->file); + if (!file.open(IO_ReadOnly)) return 0; + + QDomDocument doc("noatunequalizer"); + if (!doc.setContent(&file)) return 0; + + QDomElement docElem = doc.documentElement(); + if (docElem.tagName()!="noatunequalizer") return 0; + bool standard=docElem.attribute("default", "0")=="0"; + QString n=docElem.attribute("name", 0); + + { // All the translations for the presets +# ifdef I18N_STUFF + I18N_NOOP("Trance"); + I18N_NOOP("Dance"); + I18N_NOOP("Metal"); + I18N_NOOP("Jazz"); + I18N_NOOP("Zero"); + I18N_NOOP("Eclectic Guitar"); +# endif + } + + if (standard) + n=i18n(n.local8Bit()); + + return n; +} + +bool VPreset::setName(const QString &name) +{ + QFile file(d->file); + if (!file.open(IO_ReadOnly)) return false; + + QDomDocument doc("noatunequalizer"); + if (!doc.setContent(&file)) return false; + + QDomElement docElem = doc.documentElement(); + if (docElem.tagName()!="noatunequalizer") return false; + + if (docElem.attribute("name") == name) return true; + if (EQ->presetByName(name)) return false; + + docElem.setAttribute("name", name); + file.close(); + + if (!file.open(IO_ReadWrite | IO_Truncate)) return false; + + QTextStream s(&file); + s << doc.toString(); + file.close(); + + emit EQ->renamed(*this); + + return true; +} + +bool VPreset::isValid() const +{ + return d->file.length(); +} + +void VPreset::save() +{ + KURL url; + url.setPath(d->file); + EQ->load(url); +} + +void VPreset::load() const +{ + KURL url; + url.setPath(d->file); + EQ->load(url); +} + +QString VPreset::file() const +{ + return d->file; +} + +void VPreset::remove() +{ + KConfig *config=kapp->config(); + config->setGroup("Equalizer"); + QStringList items=config->readListEntry("presets"); + items.remove(file()); + config->writeEntry("presets", items); + config->sync(); + + emit EQ->removed(*this); + + if (file().find(kapp->dirs()->localkdedir())==0) + { + QFile f(file()); + f.remove(); + } + d->file = ""; +} + + +#undef EQ +#undef EQBACK + +#include "vequalizer.moc" + |