amuse/lib/Sequencer.cpp

#include "amuse/Sequencer.hpp"
#include "amuse/Submix.hpp"
#include "amuse/Voice.hpp"
#include "amuse/Engine.hpp"

namespace amuse
{

void Sequencer::ChannelState::_bringOutYourDead()
{
    for (auto it = m_chanVoxs.begin() ; it != m_chanVoxs.end() ;)
    {
        Voice* vox = it->second.get();
        vox->_bringOutYourDead();
        if (vox->_isRecursivelyDead())
        {
            it = m_chanVoxs.erase(it);
            continue;
        }
        ++it;
    }

    for (auto it = m_keyoffVoxs.begin() ; it != m_keyoffVoxs.end() ;)
    {
        Voice* vox = it->get();
        vox->_bringOutYourDead();
        if (vox->_isRecursivelyDead())
        {
            it = m_keyoffVoxs.erase(it);
            continue;
        }
        ++it;
    }
}

void Sequencer::_bringOutYourDead()
{
    for (auto& chan : m_chanStates)
        chan.second->_bringOutYourDead();

    if (!m_arrData && m_dieOnEnd && getVoiceCount() == 0)
        m_state = SequencerState::Dead;
}

void Sequencer::_destroy()
{
    Entity::_destroy();
    if (m_submix)
        m_submix->m_activeSequencers.erase(this);
}

Sequencer::~Sequencer() {}

Sequencer::Sequencer(Engine& engine, const AudioGroup& group, int groupId,
                     const SongGroupIndex& songGroup, int setupId, Submix* smx)
: Entity(engine, group, groupId), m_songGroup(songGroup), m_submix(smx)
{
    auto it = m_songGroup.m_midiSetups.find(setupId);
    if (it != m_songGroup.m_midiSetups.cend())
        m_midiSetup = it->second->data();
    if (m_submix)
        m_submix->m_activeSequencers.insert(this);
}

Sequencer::ChannelState::~ChannelState()
{
    if (m_submix)
        m_parent.m_engine.removeSubmix(m_submix);
}

Sequencer::ChannelState::ChannelState(Sequencer& parent, uint8_t chanId)
: m_parent(parent), m_chanId(chanId), m_setup(m_parent.m_midiSetup[chanId])
{
    if (chanId == 9)
    {
        auto it = m_parent.m_songGroup.m_drumPages.find(m_setup.programNo);
        if (it != m_parent.m_songGroup.m_drumPages.cend())
            m_page = it->second;
    }
    else
    {
        auto it = m_parent.m_songGroup.m_normPages.find(m_setup.programNo);
        if (it != m_parent.m_songGroup.m_normPages.cend())
            m_page = it->second;
    }

    m_submix = m_parent.m_engine.addSubmix(m_parent.m_submix);
    if (m_setup.reverb)
        m_submix->makeReverbStd(0.5f, m_setup.reverb / 127.f, 5.f, 0.5f, 0.f);
    if (m_setup.chorus)
        m_submix->makeChorus(15, m_setup.chorus * 5 / 127, 5000);
}

size_t Sequencer::ChannelState::getVoiceCount() const
{
    size_t ret = 0;
    for (const auto& vox : m_chanVoxs)
        ret += vox.second->getTotalVoices();
    for (const auto& vox : m_keyoffVoxs)
        ret += vox->getTotalVoices();
    return ret;
}

size_t Sequencer::getVoiceCount() const
{
    size_t ret = 0;
    for (const auto& chan : m_chanStates)
        ret += chan.second->getVoiceCount();
    return ret;
}

std::shared_ptr<Voice> Sequencer::ChannelState::keyOn(uint8_t note, uint8_t velocity)
{
    if (!m_page)
        return {};

    std::shared_ptr<Voice> ret = m_parent.m_engine._allocateVoice(m_parent.m_audioGroup,
                                                                  m_parent.m_groupId, 32000.0,
                                                                  true, false, m_submix);
    m_chanVoxs[note] = ret;
    ret->installCtrlValues(m_ctrlVals);
    if (!ret->loadSoundObject(SBig(m_page->objId), 0, 1000.f, note, velocity, m_ctrlVals[1]))
    {
        m_parent.m_engine._destroyVoice(ret.get());
        return {};
    }
    ret->setVolume(m_parent.m_curVol * m_setup.volume / 127.f);
    ret->setPan(m_setup.panning / 64.f - 127.f);
    ret->setPitchWheel(m_curPitchWheel);

    if (m_ctrlVals[64] > 64)
        ret->setPedal(true);

    return ret;
}

std::shared_ptr<Voice> Sequencer::keyOn(uint8_t chan, uint8_t note, uint8_t velocity)
{
    auto chanSearch = m_chanStates.find(chan);
    if (chanSearch == m_chanStates.cend())
    {
        auto it = m_chanStates.emplace(std::make_pair(chan, std::make_unique<ChannelState>(*this, chan)));
        return it.first->second->keyOn(note, velocity);
    }

    return chanSearch->second->keyOn(note, velocity);
}

void Sequencer::ChannelState::keyOff(uint8_t note, uint8_t velocity)
{
    auto keySearch = m_chanVoxs.find(note);
    if (keySearch == m_chanVoxs.cend())
        return;

    keySearch->second->keyOff();
    m_keyoffVoxs.emplace(std::move(keySearch->second));
    m_chanVoxs.erase(keySearch);
}

void Sequencer::keyOff(uint8_t chan, uint8_t note, uint8_t velocity)
{
    auto chanSearch = m_chanStates.find(chan);
    if (chanSearch == m_chanStates.cend())
        return;

    chanSearch->second->keyOff(note, velocity);
}

void Sequencer::ChannelState::setCtrlValue(uint8_t ctrl, int8_t val)
{
    m_ctrlVals[ctrl] = val;
    for (const auto& vox : m_chanVoxs)
        vox.second->notifyCtrlChange(ctrl, val);
    for (const auto& vox : m_keyoffVoxs)
        vox->notifyCtrlChange(ctrl, val);
}

bool Sequencer::ChannelState::programChange(int8_t prog)
{
    if (m_chanId == 9)
    {
        auto it = m_parent.m_songGroup.m_drumPages.find(prog);
        if (it != m_parent.m_songGroup.m_drumPages.cend())
        {
            m_page = it->second;
            m_curProgram = prog;
            return true;
        }
    }
    else
    {
        auto it = m_parent.m_songGroup.m_normPages.find(prog);
        if (it != m_parent.m_songGroup.m_normPages.cend())
        {
            m_page = it->second;
            m_curProgram = prog;
            return true;
        }
    }
    return false;
}

void Sequencer::ChannelState::nextProgram()
{
    int newProg = m_curProgram;
    while ((newProg += 1) <= 127)
        if (programChange(newProg))
            break;
}

void Sequencer::ChannelState::prevProgram()
{
    int newProg = m_curProgram;
    while ((newProg -= 1) >= 0)
        if (programChange(newProg))
            break;
}

void Sequencer::setCtrlValue(uint8_t chan, uint8_t ctrl, int8_t val)
{
    auto chanSearch = m_chanStates.find(chan);
    if (chanSearch == m_chanStates.cend())
        return;

    chanSearch->second->setCtrlValue(ctrl, val);
}

void Sequencer::ChannelState::setPitchWheel(float pitchWheel)
{
    m_curPitchWheel = pitchWheel;
    for (const auto& vox : m_chanVoxs)
        vox.second->setPitchWheel(pitchWheel);
    for (const auto& vox : m_keyoffVoxs)
        vox->setPitchWheel(pitchWheel);
}

void Sequencer::setPitchWheel(uint8_t chan, float pitchWheel)
{
    auto chanSearch = m_chanStates.find(chan);
    if (chanSearch == m_chanStates.cend())
        return;

    chanSearch->second->setPitchWheel(pitchWheel);
}

void Sequencer::setTempo(double ticksPerSec)
{
    m_ticksPerSec = ticksPerSec;
}

void Sequencer::ChannelState::allOff()
{
    for (const auto& vox : m_chanVoxs)
        vox.second->keyOff();
}

void Sequencer::allOff(bool now)
{
    if (now)
        for (auto& chan : m_chanStates)
        {
            for (const auto& vox : chan.second->m_chanVoxs)
                m_engine._destroyVoice(vox.second.get());
            for (const auto& vox : chan.second->m_keyoffVoxs)
                m_engine._destroyVoice(vox.get());
            chan.second->m_chanVoxs.clear();
            chan.second->m_keyoffVoxs.clear();
        }
    else
        for (auto& chan : m_chanStates)
            chan.second->allOff();
}

void Sequencer::ChannelState::killKeygroup(uint8_t kg, bool now)
{
    for (auto it = m_chanVoxs.begin() ; it != m_chanVoxs.end() ;)
    {
        Voice* vox = it->second.get();
        if (vox->m_keygroup == kg)
        {
            if (now)
            {
                it = m_chanVoxs.erase(it);
                continue;
            }
            vox->keyOff();
        }
        ++it;
    }

    if (now)
    {
        for (auto it = m_keyoffVoxs.begin() ; it != m_keyoffVoxs.end() ;)
        {
            Voice* vox = it->get();
            if (vox->m_keygroup == kg)
            {
                it = m_keyoffVoxs.erase(it);
                continue;
            }
            ++it;
        }
    }
}

void Sequencer::killKeygroup(uint8_t kg, bool now)
{
    for (auto& chan : m_chanStates)
        chan.second->killKeygroup(kg, now);
}

std::shared_ptr<Voice> Sequencer::ChannelState::findVoice(int vid)
{
    for (const auto& vox : m_chanVoxs)
        if (vox.second->vid() == vid)
            return vox.second;
    for (const auto& vox : m_keyoffVoxs)
        if (vox->vid() == vid)
            return vox;
    return {};
}

std::shared_ptr<Voice> Sequencer::findVoice(int vid)
{
    for (auto& chan : m_chanStates)
    {
        std::shared_ptr<Voice> ret = chan.second->findVoice(vid);
        if (ret)
            return ret;
    }
    return {};
}

void Sequencer::ChannelState::sendMacroMessage(ObjectId macroId, int32_t val)
{
    for (const auto& v : m_chanVoxs)
    {
        Voice* vox = v.second.get();
        if (vox->getObjectId() == macroId)
            vox->message(val);
    }
    for (const auto& v : m_keyoffVoxs)
    {
        Voice* vox = v.get();
        if (vox->getObjectId() == macroId)
            vox->message(val);
    }
}

void Sequencer::sendMacroMessage(ObjectId macroId, int32_t val)
{
    for (auto& chan : m_chanStates)
        chan.second->sendMacroMessage(macroId, val);
}

void Sequencer::playSong(const unsigned char* arrData, bool dieOnEnd)
{
    m_arrData = arrData;
    m_dieOnEnd = dieOnEnd;
    m_state = SequencerState::Playing;
}

void Sequencer::ChannelState::setVolume(float vol)
{
    vol = vol * m_setup.volume / 127.f;
    for (const auto& v : m_chanVoxs)
    {
        Voice* vox = v.second.get();
        vox->setVolume(vol);
    }
    for (const auto& v : m_keyoffVoxs)
    {
        Voice* vox = v.get();
        vox->setVolume(vol);
    }
}

void Sequencer::setVolume(float vol)
{
    m_curVol = vol;
    for (auto& chan : m_chanStates)
        chan.second->setVolume(vol);
}

int8_t Sequencer::getChanProgram(int8_t chanId) const
{
    auto chanSearch = m_chanStates.find(chanId);
    if (chanSearch == m_chanStates.cend())
        return 0;

    return chanSearch->second->m_curProgram;
}

bool Sequencer::setChanProgram(int8_t chanId, int8_t prog)
{
    auto chanSearch = m_chanStates.find(chanId);
    if (chanSearch == m_chanStates.cend())
        return false;

    return chanSearch->second->programChange(prog);
}

void Sequencer::nextChanProgram(int8_t chanId)
{
    auto chanSearch = m_chanStates.find(chanId);
    if (chanSearch == m_chanStates.cend())
        return;

    return chanSearch->second->nextProgram();
}

void Sequencer::prevChanProgram(int8_t chanId)
{
    auto chanSearch = m_chanStates.find(chanId);
    if (chanSearch == m_chanStates.cend())
        return;

    return chanSearch->second->prevProgram();
}

}