/**
 * Furnace Tracker - multi-system chiptune tracker
 * Copyright (C) 2021-2023 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 <imgui.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
  );

  ImVec4 portBorderColorH=uiColors[GUI_COLOR_PATCHBAY_PORT_HIDDEN];
  ImVec4 portColorH=ImVec4(
    portBorderColorH.x*0.75f,
    portBorderColorH.y*0.75f,
    portBorderColorH.z*0.75f,
    portBorderColorH.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);
  bool visible=ImGui::ItemAdd(rect,ImGui::GetID(portID.c_str()));
  bool hovered=false;
  bool active=false;
  if (visible) {
    hovered=ImGui::ItemHoverable(rect,ImGui::GetID(portID.c_str()),0);
    active=(hovered && ImGui::IsMouseClicked(ImGuiMouseButton_Left));

    if (hovered && ImGui::IsMouseClicked(ImGuiMouseButton_Right)) {
      ImGui::OpenPopup("SubPortOptions");
      selectedPortSet=portSetID;
      clickedPort=-1;
    }

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

    // label
    dl->AddRectFilled(minArea,maxArea,ImGui::GetColorU32(portSetColor),0.0f);
    dl->AddRect(minArea,maxArea,ImGui::GetColorU32((selectedPortSet==portSetID)?uiColors[GUI_COLOR_TEXT]:portSetBorderColor),0.0f,0,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 (visible) {
      if (ImGui::ItemAdd(portRect,ImGui::GetID(subPortID.c_str()))) {
        dl->AddRectFilled(portMin,portMax,ImGui::GetColorU32((i<activeIns)?portColor:portColorH),0.0f);
        dl->AddRect(portMin,portMax,ImGui::GetColorU32((i<activeIns)?portBorderColor:portBorderColorH),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 (visible) {
      if (ImGui::ItemAdd(portRect,ImGui::GetID(subPortID.c_str()))) {
        dl->AddRectFilled(portMin,portMax,ImGui::GetColorU32((i<activeOuts)?portColor:portColorH),0.0f);
        dl->AddRect(portMin,portMax,ImGui::GetColorU32((i<activeOuts)?portBorderColor:portBorderColorH),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 (visible && 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))) {
    if (ImGui::BeginTabBar("MixerView")) {
      if (ImGui::BeginTabItem("Mixer")) {
        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 (ImGui::BeginTable("CurPortSet",2)) {
          ImGui::TableSetupColumn("c0",ImGuiTableColumnFlags_WidthStretch);
          ImGui::TableSetupColumn("c1",ImGuiTableColumnFlags_WidthFixed);

          for (int i=0; i<e->song.systemLen; i++) {
            bool doInvert=e->song.systemVol[i]<0;
            float vol=fabs(e->song.systemVol[i]);

            ImGui::PushID(i);

            ImGui::TableNextRow();
            ImGui::TableNextColumn();
            ImGui::AlignTextToFramePadding();
            ImGui::Text("%d. %s",i+1,getSystemName(e->song.system[i]));
            ImGui::TableNextColumn();
            if (ImGui::Checkbox("Invert",&doInvert)) {
              e->song.systemVol[i]=-e->song.systemVol[i];
              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[i]=(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[i],-1.0f,1.0f)) {
              if (e->song.systemPan[i]<-1.0f) e->song.systemPan[i]=-1.0f;
              if (e->song.systemPan[i]>1.0f) e->song.systemPan[i]=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[i],-1.0f,1.0f)) {
              if (e->song.systemPanFR[i]<-1.0f) e->song.systemPanFR[i]=-1.0f;
              if (e->song.systemPanFR[i]>1.0f) e->song.systemPanFR[i]=1.0f;
              MARK_MODIFIED;
            } rightClickable
            ImGui::TableNextColumn();
            ImGui::Text("Front/Rear");

            ImGui::PopID();
          }

          ImGui::EndTable();
        }
        ImGui::EndTabItem();
      }
      if (!basicMode) if (ImGui::BeginTabItem("Patchbay")) {
        std::map<unsigned int,ImVec2> portPos;

        if (ImGui::BeginTable("PatchbayOptions",3)) {
          ImGui::TableNextRow();
          ImGui::TableNextColumn();
          if (ImGui::Checkbox("Automatic patchbay",&e->song.patchbayAuto)) {
            if (e->song.patchbayAuto) e->autoPatchbayP();
            MARK_MODIFIED;
          }
          ImGui::TableNextColumn();
          ImGui::Checkbox("Display hidden ports",&displayHiddenPorts);
          ImGui::TableNextColumn();
          ImGui::Checkbox("Display internal",&displayInternalPorts);
          ImGui::EndTable();
        }

        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;
          if (ImGui::GetContentRegionAvail().y>sysSize.y) 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();

                auto subPortI=portPos.find((selectedPortSet<<4)|selectedSubPort);
                if (subPortI!=portPos.cend()) {
                  subPortPos=subPortI->second;
                } else {
                  portDragActive=false;
                }
              }
            }
          }

          // metronome/sample preview
          if (displayInternalPorts) {
            if (portSet("Sample Preview",0xffd,0,1,0,1,selectedSubPort,portPos)) {
              selectedPortSet=0xffd;
              if (selectedSubPort>=0) {
                portDragActive=true;
                ImGui::InhibitInertialScroll();
                auto subPortI=portPos.find((selectedPortSet<<4)|selectedSubPort);
                if (subPortI!=portPos.cend()) {
                  subPortPos=subPortI->second;
                } else {
                  portDragActive=false;
                }
              }
            }
            if (portSet("Metronome",0xffe,0,1,0,1,selectedSubPort,portPos)) {
              selectedPortSet=0xffe;
              if (selectedSubPort>=0) {
                portDragActive=true;
                ImGui::InhibitInertialScroll();
                auto subPortI=portPos.find((selectedPortSet<<4)|selectedSubPort);
                if (subPortI!=portPos.cend()) {
                  subPortPos=subPortI->second;
                } else {
                  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();
              auto subPortI=portPos.find((selectedPortSet<<4)|selectedSubPort);
              if (subPortI!=portPos.cend()) {
                subPortPos=subPortI->second;
              } else {
                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);
                }
                MARK_MODIFIED;
              }
            }
          }

          // draw connections
          for (unsigned int i: e->song.patchbay) {
            if ((i>>20)==selectedPortSet) continue;
            auto portSrcI=portPos.find(i>>16);
            auto portDestI=portPos.find(0x10000|(i&0xffff));
            if (portSrcI!=portPos.cend() && portDestI!=portPos.cend()) {
              ImVec2 portSrc=portSrcI->second;
              ImVec2 portDest=portDestI->second;
              dl->AddLine(portSrc,portDest,ImGui::GetColorU32(uiColors[GUI_COLOR_PATCHBAY_CONNECTION_BG]),2.0f*dpiScale);
            }
          }

          // foreground
          for (unsigned int i: e->song.patchbay) {
            if ((i>>20)!=selectedPortSet) continue;
            auto portSrcI=portPos.find(i>>16);
            auto portDestI=portPos.find(0x10000|(i&0xffff));
            if (portSrcI!=portPos.cend() && portDestI!=portPos.cend()) {
              ImVec2 portSrc=portSrcI->second;
              ImVec2 portDest=portDestI->second;
              dl->AddLine(portSrc,portDest,ImGui::GetColorU32(uiColors[GUI_COLOR_PATCHBAY_CONNECTION]),2.0f*dpiScale);
            }
          }
        }
        if (ImGui::BeginPopup("SubPortOptions",ImGuiWindowFlags_NoMove|ImGuiWindowFlags_AlwaysAutoResize|ImGuiWindowFlags_NoTitleBar|ImGuiWindowFlags_NoSavedSettings)) {
          if (ImGui::MenuItem("disconnect all")) {
            e->patchDisconnectAll(selectedPortSet);
            MARK_MODIFIED;
          }
          ImGui::EndPopup();
        }
        ImGui::EndChild();
        ImGui::EndTabItem();
      }
      ImGui::EndTabBar();
    }

  }
  if (ImGui::IsWindowFocused(ImGuiFocusedFlags_ChildWindows)) curWindow=GUI_WINDOW_MIXER;
  ImGui::End();
}