#include "IGraphicsStressTest.h"
#include "IPlug_include_in_plug_src.h"

#include "IControls.h"

IGraphicsStressTest::IGraphicsStressTest(const InstanceInfo& info)
: Plugin(info, MakeConfig(kNumParams, 1))
{
  GetParam(0)->InitGain("Dummy");
  
#if IPLUG_EDITOR
  mMakeGraphicsFunc = [&]() {
    return MakeGraphics(*this, PLUG_WIDTH, PLUG_HEIGHT, PLUG_FPS);
  };
#endif
}

#if IPLUG_EDITOR
void IGraphicsStressTest::OnParentWindowResize(int width, int height)
{
  if(GetUI())
    GetUI()->Resize(width, height, 1.f, false);
}

void IGraphicsStressTest::LayoutUI(IGraphics* pGraphics)
{
  IRECT bounds = pGraphics->GetBounds();
  auto visualsArea = bounds.GetReducedFromBottom(50.f);
  auto labelsArea = visualsArea.GetFromBottom(50.f);
  auto buttonsArea = bounds.GetFromBRHC(bounds.W(), 50).GetPadded(-5.f);

  if(pGraphics->NControls()) {
    pGraphics->GetBackgroundControl()->SetTargetAndDrawRECTs(bounds);

    pGraphics->GetControl(1)->SetTargetAndDrawRECTs(visualsArea);
    pGraphics->GetControlWithTag(kCtrlTagNumThings)->SetTargetAndDrawRECTs(labelsArea.GetGridCell(0, 1, 2));
    pGraphics->GetControlWithTag(kCtrlTagTestNum)->SetTargetAndDrawRECTs(labelsArea.GetGridCell(1, 1, 2));
    
    for(int button=0;button<6;button++) {
      pGraphics->GetControlWithTag(kCtrlTagButton1 + button)->SetTargetAndDrawRECTs(buttonsArea.GetGridCell(button, 1, 6));
    }
    return;
  }
  
  pGraphics->AttachCornerResizer(EUIResizerMode::Size, true);
  
  enum class EFunc {Next, Prev, More, Less, Set};
  
  auto DoFunc = [&](EFunc func, int thing = 0){
    switch (func) {
      case EFunc::Next: this->mKindOfThing++; break;
      case EFunc::Prev: this->mKindOfThing--; break;
      case EFunc::More: this->mNumberOfThings++; break;
      case EFunc::Less: this->mNumberOfThings--; break;
      case EFunc::Set: this->mKindOfThing = thing; break;
      default:
        break;
    }
    
    GetUI()->GetControlWithTag(kCtrlTagNumThings)->As<ITextControl>()->SetStrFmt(64, "Number of things = %i", mNumberOfThings);
    GetUI()->GetControlWithTag(kCtrlTagTestNum)->As<ITextControl>()->SetStrFmt(64, "Test %i/%i", this->mKindOfThing, 32);
    GetUI()->SetAllControlsDirty();
  };
  
  pGraphics->SetKeyHandlerFunc([DoFunc](const IKeyPress& key, bool isUp)
  {
    if(!isUp) {
      switch (key.VK) {
        case kVK_UP: DoFunc(EFunc::More); return true;
        case kVK_DOWN: DoFunc(EFunc::Less); return true;
        case kVK_TAB: key.S ? DoFunc(EFunc::Prev) : DoFunc(EFunc::Next); return true;
        default: return false;
      }
    }
    return false;
  });
  
  pGraphics->EnableMouseOver(false);
  pGraphics->LoadFont("Roboto-Regular", ROBOTO_FN);
  pGraphics->AttachPanelBackground(COLOR_GRAY);
  pGraphics->AttachControl(new ILambdaControl(visualsArea, [&](ILambdaControl* pCaller, IGraphics& g, IRECT& r) {
    static IBitmap smiley = g.LoadBitmap(SMILEY_FN);
    static ISVG tiger = g.LoadSVG(TIGER_FN);
    
    g.FillRect(COLOR_WHITE, r);
    
    if(this->mKindOfThing == 0)
    {
      g.DrawText(IText(30), "Press tab to go to next test", r);
      g.DrawText(IText(30), "up/down to change the # of things", r.GetVShifted(40.f));
    }
    else
    //      if (!g.CheckLayer(pCaller->mLayer))
    {
      //        g.StartLayer(r);
      
      for (int i=0; i<this->mNumberOfThings; i++)
      {
        IRECT rr = r.GetRandomSubRect();
        IColor rc = IColor::GetRandomColor();
        IBlend rb = {};
        static bool dir = 0;
        static float thickness = 5.f;
        static float roundness = 5.f;
        float rrad1 = static_cast<float>(rand() % 360);
        float rrad2 = static_cast<float>(rand() % 360);
        
        switch (this->mKindOfThing)
        {
          case 1:  g.DrawRect(rc, rr, &rb); break;
          case 2:  g.FillRect(rc, rr, &rb); break;
          case 3:  g.DrawRoundRect(rc, rr, roundness, &rb); break;
          case 4:  g.FillRoundRect(rc, rr, roundness, &rb); break;
          case 5:  g.DrawEllipse(rc, rr, &rb); break;
          case 6:  g.FillEllipse(rc, rr, &rb); break;
          case 7:  g.DrawArc(rc, rr.MW(), rr.MH(), rr.W() > rr.H() ? rr.H() : rr.W(), rrad1, rrad2, &rb,thickness); break;
          case 8:  g.FillArc(rc, rr.MW(), rr.MH(), rr.W() > rr.H() ? rr.H() : rr.W(), rrad1, rrad2, &rb); break;
          case 9:  g.DrawLine(rc, dir == 0 ? rr.L : rr.R, rr.B, dir == 0 ? rr.R : rr.L, rr.T, &rb,thickness); break;
          case 10: g.DrawDottedLine(rc, dir == 0 ? rr.L : rr.R, rr.B, dir == 0 ? rr.R : rr.L, rr.T, &rb, thickness); break;
          case 11: g.DrawFittedBitmap(smiley, rr, &rb); break;
          case 12: g.DrawSVG(tiger, rr); break;
          default:
            break;
        }
        
        dir = !dir;
      }
      //        pCaller->mLayer = g.EndLayer();
    }
    
    //      g.DrawLayer(pCaller->mLayer);
    
  }, 10000, false, false));
  
  pGraphics->AttachControl(new ITextControl(labelsArea.GetGridCell(0, 1, 2), "", IText(20)), kCtrlTagNumThings);
  pGraphics->AttachControl(new ITextControl(labelsArea.GetGridCell(1, 1, 2), "", IText(20)), kCtrlTagTestNum);
  
  int button = 0;
  for (auto buttonLabel : {"Select test", "Next test", "Prev test", "Things++", "Things--"}) {
    pGraphics->AttachControl(new IVButtonControl(buttonsArea.GetGridCell(button, 1, 6), 
      SplashClickActionFunc, buttonLabel, DEFAULT_STYLE.WithLabelText(DEFAULT_TEXT.WithVAlign(EVAlign::Middle)).WithRoundness(0.2f)), kCtrlTagButton1 + button)
      ->SetAnimationEndActionFunction([button, DoFunc, pGraphics](IControl* pCaller) {
      
      switch (button){
        case 0:
        {
          static IPopupMenu menu {"Test", {"Start", "DrawRect", "FillRect", "DrawRoundRect", "FillRoundRect", "DrawEllipse", "FillEllipse", "DrawArc", "FillArc", "DrawLine", "DrawDottedLine", "DrawFittedBitmap", "DrawSVG"},
            [DoFunc](IPopupMenu* pMenu) {
              DoFunc(EFunc::Set, pMenu->GetChosenItemIdx());
            }};
          
          pGraphics->CreatePopupMenu(*pCaller, menu, pCaller->GetRECT());
          break;
        }
        case 1: DoFunc(EFunc::Next); break;
        case 2: DoFunc(EFunc::Prev); break;
        case 3: DoFunc(EFunc::More); break;
        case 4: DoFunc(EFunc::Less); break;
        default:
          break;
      }
    });
    
    button++;
  }
  
  pGraphics->AttachControl(new IVToggleControl(buttonsArea.GetGridCell(button, 1, 6), 
    SplashClickActionFunc, "", DEFAULT_STYLE.WithRoundness(0.2f), "FPS OFF", "FPS ON"), kCtrlTagButton6)
    ->SetAnimationEndActionFunction([](IControl* pCaller){
      pCaller->GetUI()->ShowFPSDisplay(pCaller->GetValue() > 0.5);
  });

}
#endif