mirror of
https://github.com/polybar/polybar.git
synced 2026-03-01 21:58:53 +00:00
init(git): Base commit
This commit is contained in:
Michael Carlberg
230
src/modules/volume.cpp
Normal file
230
src/modules/volume.cpp
Normal file
@ -0,0 +1,230 @@
|
||||
#include <thread>
|
||||
|
||||
#include "lemonbuddy.hpp"
|
||||
#include "bar.hpp"
|
||||
#include "services/builder.hpp"
|
||||
#include "utils/config.hpp"
|
||||
#include "utils/math.hpp"
|
||||
#include "modules/volume.hpp"
|
||||
|
||||
using namespace modules;
|
||||
|
||||
VolumeModule::VolumeModule(const std::string& name_) throw(ModuleError) : EventModule(name_)
|
||||
{
|
||||
auto speaker_mixer = config::get<std::string>(name(), "speaker_mixer", "");
|
||||
auto headphone_mixer = config::get<std::string>(name(), "headphone_mixer", "");
|
||||
|
||||
this->headphone_ctrl_numid = config::get<int>(name(), "headphone_control_numid", -1);
|
||||
|
||||
if (!headphone_mixer.empty() && this->headphone_ctrl_numid == -1)
|
||||
throw ModuleError("[VolumeModule] Missing required property value for \"headphone_control_numid\"...");
|
||||
else if (headphone_mixer.empty())
|
||||
throw ModuleError("[VolumeModule] Missing required property value for \"headphone_mixer\"...");
|
||||
|
||||
if (string::lower(speaker_mixer) == "master")
|
||||
throw ModuleError("[VolumeModule] The \"Master\" mixer is already processed internally. Specify another mixer or comment out the \"speaker_mixer\" parameter...");
|
||||
if (string::lower(headphone_mixer) == "master")
|
||||
throw ModuleError("[VolumeModule] The \"Master\" mixer is already processed internally. Specify another mixer or comment out the \"headphone_mixer\" parameter...");
|
||||
|
||||
auto create_mixer = [](std::string mixer_name)
|
||||
{
|
||||
std::unique_ptr<alsa::Mixer> mixer;
|
||||
|
||||
try {
|
||||
mixer = std::make_unique<alsa::Mixer>(mixer_name);
|
||||
} catch (alsa::MixerError &e) {
|
||||
log_error("Failed to open \""+ mixer_name +"\" mixer => "+ STR(e.what()));
|
||||
mixer.reset();
|
||||
}
|
||||
|
||||
return mixer;
|
||||
};
|
||||
|
||||
this->master_mixer = create_mixer("Master");
|
||||
|
||||
if (!speaker_mixer.empty())
|
||||
this->speaker_mixer = create_mixer(speaker_mixer);
|
||||
if (!headphone_mixer.empty())
|
||||
this->headphone_mixer = create_mixer(headphone_mixer);
|
||||
|
||||
if (!this->master_mixer && !this->speaker_mixer && !this->headphone_mixer) {
|
||||
this->stop();
|
||||
return;
|
||||
}
|
||||
|
||||
if (this->headphone_mixer && this->headphone_ctrl_numid > -1) {
|
||||
try {
|
||||
this->headphone_ctrl = std::make_unique<alsa::ControlInterface>(this->headphone_ctrl_numid);
|
||||
} catch (alsa::ControlInterfaceError &e) {
|
||||
log_error("Failed to open headphone control interface => "+ STR(e.what()));
|
||||
this->headphone_ctrl.reset();
|
||||
}
|
||||
}
|
||||
|
||||
this->builder = std::make_unique<Builder>();
|
||||
|
||||
this->formatter->add(FORMAT_VOLUME, TAG_LABEL_VOLUME,
|
||||
{ TAG_RAMP_VOLUME, TAG_LABEL_VOLUME, TAG_BAR_VOLUME });
|
||||
this->formatter->add(FORMAT_MUTED, TAG_LABEL_MUTED,
|
||||
{ TAG_RAMP_VOLUME, TAG_LABEL_MUTED, TAG_BAR_VOLUME });
|
||||
|
||||
if (this->formatter->has(TAG_BAR_VOLUME))
|
||||
this->bar_volume = drawtypes::get_config_bar(name(), get_tag_name(TAG_BAR_VOLUME));
|
||||
if (this->formatter->has(TAG_RAMP_VOLUME))
|
||||
this->ramp_volume = drawtypes::get_config_ramp(name(), get_tag_name(TAG_RAMP_VOLUME));
|
||||
if (this->formatter->has(TAG_LABEL_VOLUME, FORMAT_VOLUME))
|
||||
this->label_volume = drawtypes::get_optional_config_label(name(), get_tag_name(TAG_LABEL_VOLUME), "%percentage%");
|
||||
if (this->formatter->has(TAG_LABEL_MUTED, FORMAT_MUTED))
|
||||
this->label_muted = drawtypes::get_optional_config_label(name(), get_tag_name(TAG_LABEL_MUTED), "%percentage%");
|
||||
|
||||
register_command_handler(name());
|
||||
}
|
||||
|
||||
VolumeModule::~VolumeModule()
|
||||
{
|
||||
std::lock_guard<concurrency::SpinLock> lck(this->update_lock);
|
||||
|
||||
this->master_mixer.reset();
|
||||
this->speaker_mixer.reset();
|
||||
this->headphone_mixer.reset();
|
||||
this->headphone_ctrl.reset();
|
||||
}
|
||||
|
||||
bool VolumeModule::has_event()
|
||||
{
|
||||
bool has_event = false;
|
||||
|
||||
try {
|
||||
if (this->master_mixer)
|
||||
has_event |= this->master_mixer->wait(25);
|
||||
if (this->speaker_mixer)
|
||||
has_event |= this->speaker_mixer->wait(25);
|
||||
if (this->headphone_mixer)
|
||||
has_event |= this->headphone_mixer->wait(25);
|
||||
if (this->headphone_ctrl)
|
||||
has_event |= this->headphone_ctrl->wait(25);
|
||||
} catch (alsa::Exception &e) {
|
||||
log_error(e.what());
|
||||
}
|
||||
|
||||
return has_event;
|
||||
}
|
||||
|
||||
bool VolumeModule::update()
|
||||
{
|
||||
int volume = 0;
|
||||
bool muted = false;
|
||||
auto headphones_connected = false;
|
||||
|
||||
if (this->master_mixer) {
|
||||
volume = this->master_mixer->get_volume();
|
||||
muted |= this->master_mixer->is_muted();
|
||||
} else {
|
||||
volume = 100;
|
||||
}
|
||||
|
||||
if (this->headphone_ctrl && this->headphone_mixer)
|
||||
headphones_connected = this->headphone_ctrl->test_device_plugged();
|
||||
|
||||
if (headphones_connected) {
|
||||
volume *= this->headphone_mixer->get_volume() / 100.0f;
|
||||
muted |= this->headphone_mixer->is_muted();
|
||||
} else if (this->speaker_mixer) {
|
||||
volume *= this->speaker_mixer->get_volume() / 100.0f;
|
||||
muted |= this->speaker_mixer->is_muted();
|
||||
}
|
||||
|
||||
this->volume = volume;
|
||||
this->muted = muted;
|
||||
|
||||
if (!this->label_volume_tokenized)
|
||||
this->label_volume_tokenized = this->label_volume->clone();
|
||||
if (!this->label_muted_tokenized)
|
||||
this->label_muted_tokenized = this->label_muted->clone();
|
||||
|
||||
this->label_volume_tokenized->text = this->label_volume->text;
|
||||
this->label_volume_tokenized->replace_token("%percentage%", std::to_string(this->volume) +"%");
|
||||
|
||||
this->label_muted_tokenized->text = this->label_muted->text;
|
||||
this->label_muted_tokenized->replace_token("%percentage%", std::to_string(this->volume) +"%");
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
std::string VolumeModule::get_format() {
|
||||
return this->muted ? FORMAT_MUTED : FORMAT_VOLUME;
|
||||
}
|
||||
|
||||
std::string VolumeModule::get_output()
|
||||
{
|
||||
this->builder->cmd(Cmd::LEFT_CLICK, EVENT_TOGGLE_MUTE);
|
||||
|
||||
if (volume < 100)
|
||||
this->builder->cmd(Cmd::SCROLL_UP, EVENT_VOLUME_UP, volume < 100);
|
||||
if (volume > 0)
|
||||
this->builder->cmd(Cmd::SCROLL_DOWN, EVENT_VOLUME_DOWN);
|
||||
|
||||
this->builder->node(this->Module::get_output());
|
||||
|
||||
return this->builder->flush();
|
||||
}
|
||||
|
||||
bool VolumeModule::build(Builder *builder, const std::string& tag)
|
||||
{
|
||||
bool built = true;
|
||||
|
||||
if (tag == TAG_BAR_VOLUME)
|
||||
builder->node(this->bar_volume, volume);
|
||||
else if (tag == TAG_RAMP_VOLUME)
|
||||
builder->node(this->ramp_volume, volume);
|
||||
else if (tag == TAG_LABEL_VOLUME)
|
||||
builder->node(this->label_volume_tokenized);
|
||||
else if (tag == TAG_LABEL_MUTED)
|
||||
builder->node(this->label_muted_tokenized);
|
||||
else
|
||||
built = false;
|
||||
|
||||
return built;
|
||||
}
|
||||
|
||||
bool VolumeModule::handle_command(const std::string& cmd)
|
||||
{
|
||||
if (cmd.length() < 3 || cmd.substr(0, 3) != "vol")
|
||||
return false;
|
||||
|
||||
alsa::Mixer *master_mixer = nullptr;
|
||||
alsa::Mixer *other_mixer = nullptr;
|
||||
|
||||
bool headphones_connected = false;
|
||||
|
||||
if (this->headphone_ctrl && this->headphone_mixer)
|
||||
headphones_connected = this->headphone_ctrl->test_device_plugged();
|
||||
|
||||
if (headphones_connected)
|
||||
other_mixer = this->headphone_mixer.get();
|
||||
else if (this->speaker_mixer)
|
||||
other_mixer = this->speaker_mixer.get();
|
||||
|
||||
if (this->master_mixer)
|
||||
master_mixer = this->master_mixer.get();
|
||||
|
||||
if (cmd == EVENT_VOLUME_UP) {
|
||||
auto vol = math::cap<float>(this->master_mixer->get_volume() + 5, 0, 100);
|
||||
if (master_mixer != nullptr)
|
||||
master_mixer->set_volume(vol);
|
||||
} else if (cmd == EVENT_VOLUME_DOWN) {
|
||||
auto vol = math::cap<float>(this->master_mixer->get_volume() - 5, 0, 100);
|
||||
if (master_mixer != nullptr)
|
||||
master_mixer->set_volume(vol);
|
||||
} else if (cmd == EVENT_TOGGLE_MUTE) {
|
||||
bool mute = master_mixer->is_muted();
|
||||
if (master_mixer != nullptr)
|
||||
master_mixer->toggle_mute();
|
||||
if (other_mixer != nullptr)
|
||||
other_mixer->set_mute(mute);
|
||||
} else {
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
Reference in New Issue
Block a user