furnace/src/gui/mixer.cpp

/**
 * Furnace Tracker - multi-system chiptune tracker
 * Copyright (C) 2021-2022 tildearrow and contributors
 *
 * 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 "gui.h"
#include "intConst.h"
#include <fmt/printf.h>
#include "../ta-log.h"
#include "imgui_internal.h"

const char* portNamesStereo[2]={
  "left",
  "right"
};

ImVec2 FurnaceGUI::calcPortSetSize(String label, int ins, int outs) {
  ImGuiStyle& style=ImGui::GetStyle();

  ImVec2 labelSize=ImGui::CalcTextSize(label.c_str(),NULL,false,ImGui::GetWindowSize().x*0.6f);

  ImVec2 size=labelSize;

  // pad
  size.x+=style.FramePadding.x*2.0f;
  size.y+=style.FramePadding.y*2.0f;

  // space for ports
  size.y+=MAX(ins,outs)*(labelSize.y+style.FramePadding.y+style.ItemSpacing.y);

  return size;
}

bool FurnaceGUI::portSet(String label, unsigned int portSetID, int ins, int outs, int activeIns, int activeOuts, int& clickedPort, std::map<unsigned int,ImVec2>& portPos) {
  String portID=fmt::sprintf("portSet%.4x",portSetID);

  ImDrawList* dl=ImGui::GetWindowDrawList();
  ImGuiWindow* window=ImGui::GetCurrentWindow();
  ImGuiStyle& style=ImGui::GetStyle();

  ImVec2 labelSize=ImGui::CalcTextSize(label.c_str(),NULL,false,ImGui::GetWindowSize().x*0.6f);

  ImVec2 size=labelSize;

  // pad
  size.x+=style.FramePadding.x*2.0f;
  size.y+=style.FramePadding.y*2.0f;

  // space for ports
  size.y+=MAX(ins,outs)*(ImGui::GetFontSize()+style.FramePadding.y+style.ItemSpacing.y);

  ImVec4 portSetBorderColor=uiColors[GUI_COLOR_PATCHBAY_PORTSET];
  ImVec4 portSetColor=ImVec4(
    portSetBorderColor.x*0.75f,
    portSetBorderColor.y*0.75f,
    portSetBorderColor.z*0.75f,
    portSetBorderColor.w
  );

  ImVec4 portBorderColor=uiColors[GUI_COLOR_PATCHBAY_PORT];
  ImVec4 portColor=ImVec4(
    portBorderColor.x*0.75f,
    portBorderColor.y*0.75f,
    portBorderColor.z*0.75f,
    portBorderColor.w
  );

  ImVec2 minArea=window->DC.CursorPos;
  ImVec2 maxArea=ImVec2(
    minArea.x+size.x,
    minArea.y+size.y
  );
  ImRect rect=ImRect(minArea,maxArea);

  ImVec2 textPos=ImVec2(
    minArea.x+style.FramePadding.x,
    minArea.y+style.FramePadding.y
  );

  ImGui::ItemSize(size,style.FramePadding.y);
  if (ImGui::ItemAdd(rect,ImGui::GetID(portID.c_str()))) {
    bool hovered=ImGui::ItemHoverable(rect,ImGui::GetID(portID.c_str()));
    bool active=(hovered && ImGui::IsMouseClicked(ImGuiMouseButton_Left));


    if (hovered) hoveredPortSet=portSetID;
    if (active) clickedPort=-1;

    // label
    dl->AddRectFilled(minArea,maxArea,ImGui::GetColorU32(portSetColor),0.0f);
    dl->AddRect(minArea,maxArea,ImGui::GetColorU32(portSetBorderColor),0.0f,dpiScale);
    dl->AddText(ImGui::GetFont(),ImGui::GetFontSize(),textPos,ImGui::GetColorU32(uiColors[GUI_COLOR_TEXT]),label.c_str(),NULL,ImGui::GetWindowSize().x*0.6f);

    // input ports
    for (int i=0; i<ins; i++) {
      String portLabel="input";
      String subPortID=fmt::sprintf("subPort%.5x",(portSetID<<4)|i);
      if (ins==2) {
        portLabel=portNamesStereo[i&1];
      } else if (ins>2) {
        portLabel=fmt::sprintf("%d",i+1);
      }
      ImVec2 portLabelSize=ImGui::CalcTextSize(portLabel.c_str());

      ImVec2 portMin=ImVec2(
        minArea.x,
        minArea.y+style.FramePadding.y+labelSize.y+style.ItemSpacing.y+(style.ItemSpacing.y+portLabelSize.y+style.FramePadding.y)*i
      );
      ImVec2 portMax=ImVec2(
        minArea.x+portLabelSize.x+style.FramePadding.x,
        portMin.y+style.FramePadding.y+portLabelSize.y
      );
      ImRect portRect=ImRect(portMin,portMax);
      ImVec2 portLabelPos=portMin;
      portLabelPos.x+=style.FramePadding.x*0.5;
      portLabelPos.y+=style.FramePadding.y*0.5;

      portPos[(portSetID<<4)|i]=ImLerp(portMin,portMax,ImVec2(0.0f,0.5f));

      if (ImGui::ItemAdd(portRect,ImGui::GetID(subPortID.c_str()))) {
        dl->AddRectFilled(portMin,portMax,ImGui::GetColorU32(portColor),0.0f);
        dl->AddRect(portMin,portMax,ImGui::GetColorU32(portBorderColor),0.0f,dpiScale);
        dl->AddText(portLabelPos,ImGui::GetColorU32(uiColors[GUI_COLOR_TEXT]),portLabel.c_str());
      }

      if (ImGui::IsMouseHoveringRect(portMin,portMax)) {
        hoveredSubPort=i;
        if (active) clickedPort=i;
      }
    }

    // output ports
    for (int i=0; i<outs; i++) {
      String portLabel="output";
      String subPortID=fmt::sprintf("subPort%.5x",(portSetID<<4)|i);
      if (outs==2) {
        portLabel=portNamesStereo[i&1];
      } else if (outs>2) {
        portLabel=fmt::sprintf("%d",i+1);
      }
      ImVec2 portLabelSize=ImGui::CalcTextSize(portLabel.c_str());

      ImVec2 portMin=ImVec2(
        maxArea.x-portLabelSize.x-style.FramePadding.x,
        minArea.y+style.FramePadding.y+labelSize.y+style.ItemSpacing.y+(style.ItemSpacing.y+portLabelSize.y+style.FramePadding.y)*i
      );
      ImVec2 portMax=ImVec2(
        maxArea.x,
        portMin.y+style.FramePadding.y+portLabelSize.y
      );
      ImRect portRect=ImRect(portMin,portMax);
      ImVec2 portLabelPos=portMin;
      portLabelPos.x+=style.FramePadding.x*0.5;
      portLabelPos.y+=style.FramePadding.y*0.5;

      portPos[(portSetID<<4)|i]=ImLerp(portMin,portMax,ImVec2(1.0f,0.5f));

      if (ImGui::ItemAdd(portRect,ImGui::GetID(subPortID.c_str()))) {
        dl->AddRectFilled(portMin,portMax,ImGui::GetColorU32(portColor),0.0f);
        dl->AddRect(portMin,portMax,ImGui::GetColorU32(portBorderColor),0.0f,dpiScale);
        dl->AddText(portLabelPos,ImGui::GetColorU32(uiColors[GUI_COLOR_TEXT]),portLabel.c_str());
      }

      if (ImGui::IsMouseHoveringRect(portMin,portMax)) {
        if (active) clickedPort=i;
        hoveredSubPort=i;
      }
    }

    if (hovered && ImGui::IsMouseClicked(ImGuiMouseButton_Left)) return true;
  }

  return false;
}

void FurnaceGUI::drawMixer() {
  if (nextWindow==GUI_WINDOW_MIXER) {
    mixerOpen=true;
    ImGui::SetNextWindowFocus();
    nextWindow=GUI_WINDOW_NOTHING;
  }
  if (!mixerOpen) return;
  if (mobileUI) {
    patWindowPos=(portrait?ImVec2(0.0f,(mobileMenuPos*-0.65*canvasH)):ImVec2((0.16*canvasH)+0.5*canvasW*mobileMenuPos,0.0f));
    patWindowSize=(portrait?ImVec2(canvasW,canvasH-(0.16*canvasW)):ImVec2(canvasW-(0.16*canvasH),canvasH));
    ImGui::SetNextWindowPos(patWindowPos);
    ImGui::SetNextWindowSize(patWindowSize);
  } else {
    ImGui::SetNextWindowSizeConstraints(ImVec2(400.0f*dpiScale,200.0f*dpiScale),ImVec2(canvasW,canvasH));
  }
  if (ImGui::Begin("Mixer",&mixerOpen,globalWinFlags|(settings.allowEditDocking?0:ImGuiWindowFlags_NoDocking))) {
    std::map<unsigned int,ImVec2> portPos;
    if (ImGui::SliderFloat("Master Volume",&e->song.masterVol,0,3,"%.2fx")) {
      if (e->song.masterVol<0) e->song.masterVol=0;
      if (e->song.masterVol>3) e->song.masterVol=3;
      MARK_MODIFIED;
    } rightClickable
    if (selectedPortSet<e->song.systemLen) {
      if (ImGui::BeginTable("CurPortSet",2)) {
        ImGui::TableSetupColumn("c0",ImGuiTableColumnFlags_WidthStretch);
        ImGui::TableSetupColumn("c1",ImGuiTableColumnFlags_WidthFixed);

        bool doInvert=e->song.systemVol[selectedPortSet]<0;
        float vol=fabs(e->song.systemVol[selectedPortSet]);

        ImGui::TableNextRow();
        ImGui::TableNextColumn();
        ImGui::Text("%d. %s",selectedPortSet+1,getSystemName(e->song.system[selectedPortSet]));
        ImGui::TableNextColumn();
        if (ImGui::Checkbox("Invert",&doInvert)) {
          e->song.systemVol[selectedPortSet]=-e->song.systemVol[selectedPortSet];
          MARK_MODIFIED;
        }

        ImGui::TableNextRow();
        ImGui::TableNextColumn();
        ImGui::SetNextItemWidth(ImGui::GetContentRegionAvail().x);
        if (CWSliderFloat("##Volume",&vol,0,2)) {
          if (doInvert) {
            if (vol<0.0001) vol=0.0001;
          }
          if (vol<0) vol=0;
          if (vol>10) vol=10;
          e->song.systemVol[selectedPortSet]=(doInvert)?-vol:vol;
          MARK_MODIFIED;
        } rightClickable
        ImGui::TableNextColumn();
        ImGui::Text("Volume");

        ImGui::TableNextRow();
        ImGui::TableNextColumn();
        ImGui::SetNextItemWidth(ImGui::GetContentRegionAvail().x);
        if (CWSliderFloat("##Panning",&e->song.systemPan[selectedPortSet],-1.0f,1.0f)) {
          if (e->song.systemPan[selectedPortSet]<-1.0f) e->song.systemPan[selectedPortSet]=-1.0f;
          if (e->song.systemPan[selectedPortSet]>1.0f) e->song.systemPan[selectedPortSet]=1.0f;
          MARK_MODIFIED;
        } rightClickable
        ImGui::TableNextColumn();
        ImGui::Text("Panning");

        ImGui::TableNextRow();
        ImGui::TableNextColumn();
        ImGui::SetNextItemWidth(ImGui::GetContentRegionAvail().x);
        if (CWSliderFloat("##FrontRear",&e->song.systemPanFR[selectedPortSet],-1.0f,1.0f)) {
          if (e->song.systemPanFR[selectedPortSet]<-1.0f) e->song.systemPanFR[selectedPortSet]=-1.0f;
          if (e->song.systemPanFR[selectedPortSet]>1.0f) e->song.systemPanFR[selectedPortSet]=1.0f;
          MARK_MODIFIED;
        } rightClickable
        ImGui::TableNextColumn();
        ImGui::Text("Front/Rear");

        ImGui::EndTable();
      }
    } else {
      if (ImGui::Checkbox("Automatic patchbay",&e->song.patchbayAuto)) {
        if (e->song.patchbayAuto) e->autoPatchbayP();
        MARK_MODIFIED;
      }
      ImGui::Checkbox("Display hidden ports",&displayHiddenPorts);
      ImGui::Checkbox("Display internal",&displayInternalPorts);
      ImGui::Dummy(ImVec2(1.0f,ImGui::GetFrameHeightWithSpacing()));
    }

    hoveredPortSet=0x1fff;
    hoveredSubPort=-1;

    if (ImGui::BeginChild("Patchbay",ImVec2(0,0),true)) {
      ImDrawList* dl=ImGui::GetWindowDrawList();
      ImVec2 topPos=ImGui::GetCursorPos();
      ImVec2 sysSize=calcPortSetSize("System",displayHiddenPorts?DIV_MAX_OUTPUTS:e->getAudioDescGot().outChans,0);
      topPos.x+=ImGui::GetContentRegionAvail().x-sysSize.x;
      topPos.y+=(ImGui::GetContentRegionAvail().y-sysSize.y)*0.5+ImGui::GetScrollY();

      if (ImGui::IsWindowHovered() && ImGui::IsMouseClicked(ImGuiMouseButton_Left)) selectedPortSet=0x1fff;

      if (portDragActive) {
        dl->AddLine(subPortPos,ImGui::GetMousePos(),ImGui::GetColorU32(uiColors[GUI_COLOR_PATCHBAY_CONNECTION]),2.0f*dpiScale);
      }

      for (int i=0; i<e->song.systemLen; i++) {
        DivDispatch* dispatch=e->getDispatch(i);
        if (dispatch==NULL) continue;
        int outputs=dispatch->getOutputCount();
        if (portSet(fmt::sprintf("%d. %s",i+1,getSystemName(e->song.system[i])),i,0,outputs,0,outputs,selectedSubPort,portPos)) {
          selectedPortSet=i;
          if (selectedSubPort>=0) {
            portDragActive=true;
            ImGui::InhibitInertialScroll();
            try {
              subPortPos=portPos.at((selectedPortSet<<4)|selectedSubPort);
            } catch (std::out_of_range& e) {
              portDragActive=false;
            }
          }
        }
      }

      // metronome/sample preview
      if (displayInternalPorts) {
        if (portSet("Sample Preview",0xffd,0,1,0,1,selectedSubPort,portPos)) {
          selectedPortSet=0xffe;
          if (selectedSubPort>=0) {
            portDragActive=true;
            ImGui::InhibitInertialScroll();
            try {
              subPortPos=portPos.at((selectedPortSet<<4)|selectedSubPort);
            } catch (std::out_of_range& e) {
              portDragActive=false;
            }
          }
        }
        if (portSet("Metronome",0xffe,0,1,0,1,selectedSubPort,portPos)) {
          selectedPortSet=0xffe;
          if (selectedSubPort>=0) {
            portDragActive=true;
            ImGui::InhibitInertialScroll();
            try {
              subPortPos=portPos.at((selectedPortSet<<4)|selectedSubPort);
            } catch (std::out_of_range& e) {
              portDragActive=false;
            }
          }
        }
      }

      ImGui::SetCursorPos(topPos);
      if (portSet("System",0x1000,displayHiddenPorts?DIV_MAX_OUTPUTS:e->getAudioDescGot().outChans,0,e->getAudioDescGot().outChans,0,selectedSubPort,portPos)) {
        selectedPortSet=0x1000;
        if (selectedSubPort>=0) {
          portDragActive=true;
          ImGui::InhibitInertialScroll();
          try {
            subPortPos=portPos.at((selectedPortSet<<4)|selectedSubPort);
          } catch (std::out_of_range& e) {
            portDragActive=false;
          }
        }
      }

      if (portDragActive) {
        if (ImGui::IsMouseReleased(ImGuiMouseButton_Left)) {
          portDragActive=false;
          if (hoveredPortSet!=0x1fff && hoveredSubPort>=0 && selectedPortSet!=hoveredPortSet) {
            unsigned int src=(selectedPortSet<<4)|selectedSubPort;
            unsigned int dest=(hoveredPortSet<<4)|hoveredSubPort;

            if (src&0x10000) {
              src^=dest;
              dest^=src;
              src^=dest;
            }

            src&=0xffff;
            dest&=0xffff;

            if (!e->patchConnect(src,dest)) {
              e->patchDisconnect(src,dest);
            }
          }
        }
      }

      // draw connections
      for (unsigned int i: e->song.patchbay) {
        try {
          ImVec2 portSrc=portPos.at(i>>16);
          ImVec2 portDest=portPos.at(0x10000|(i&0xffff));
          dl->AddLine(portSrc,portDest,ImGui::GetColorU32(uiColors[GUI_COLOR_PATCHBAY_CONNECTION]),2.0f*dpiScale);
        } catch (std::out_of_range& e) {
        }
      }
    }
    ImGui::EndChild();
  }
  if (ImGui::IsWindowFocused(ImGuiFocusedFlags_ChildWindows)) curWindow=GUI_WINDOW_MIXER;
  ImGui::End();
}