#include "ofxPiMapper.h"

ofxPiMapper::ofxPiMapper():
bShowInfo(false),
isSetUp(false){
  ofAddListener(ofEvents().keyPressed, this, &ofxPiMapper::keyPressed);
}

ofxPiMapper::~ofxPiMapper() {
  ofRemoveListener(ofEvents().keyPressed, this, &ofxPiMapper::keyPressed);
}

void ofxPiMapper::setup() {
  ofLogNotice("ofxPiMapper") << "Setting up...";
  
  // Assign media server to other pi mapper components
  surfaceManager.setMediaServer(&mediaServer);
  gui.setMediaServer(&mediaServer);
  
  // Check if we have user surfaces defined, if not - load default
  if (ofFile::doesFileExist(PIMAPPER_USER_SURFACES_XML_FILE)) {
    ofLogNotice("ofxPiMapper") << "Loading user surfaces from " << PIMAPPER_USER_SURFACES_XML_FILE;
    surfaceManager.loadXmlSettings(PIMAPPER_USER_SURFACES_XML_FILE);
  } else {
    ofLogNotice("ofxPiMapper") << "Loading default surfaces from " << PIMAPPER_DEF_SURFACES_XML_FILE;
    surfaceManager.loadXmlSettings(PIMAPPER_DEF_SURFACES_XML_FILE);
  }
  
  // The GUI needs something to interface with
  gui.setSurfaceManager(&surfaceManager);
  
  isSetUp = true;
  
  ofLogNotice("ofxPiMapper") << "Done setting up";
}

void ofxPiMapper::draw() {
  if (!isSetUp) {
    return;
  }
  
  // Draw the piMapper GUI
  gui.draw();
  
  if (bShowInfo) {
    // Draw instructions
    stringstream ss;
    ss << "There are 4 modes:\n\n";
    ss << " 1. Presentation mode\n";
    ss << " 2. Texture mapping mode\n";
    ss << " 3. Projection mapping mode\n";
    ss << " 4. Source selection mode\n\n";
    ss << "You can switch between the modes by using <1>, <2>, <3> and <4> "
    "keys on the keyboard.\n\n";
    ss << "Press <t> to add new triangle surface\n";
    ss << "Press <q> to add new quad surface\n";
    ss << "Press <s> to save the composition\n";
    ss << "Press <f> to toggle fullscreen\n";
    ss << "Press <i> to hide this message";
    
    ofDrawBitmapStringHighlight(ss.str(), 10, 20, ofColor(0, 0, 0, 100),
                                ofColor(255, 255, 255, 200));
  }
  
} // draw

void ofxPiMapper::keyPressed(ofKeyEventArgs &args) {
  ofLogNotice("ofxPiMapper") << "Key pressed: " << static_cast<char>(args.key);
  
  switch (args.key) {
    case '1':
      gui.setMode(ofx::piMapper::GuiMode::NONE);
      break;
    case '2':
      gui.setMode(ofx::piMapper::GuiMode::TEXTURE_MAPPING);
      break;
    case '3':
      gui.setMode(ofx::piMapper::GuiMode::PROJECTION_MAPPING);
      break;
    case '4':
      gui.setMode(ofx::piMapper::GuiMode::SOURCE_SELECTION);
      break;
    case 'i':
      bShowInfo = !bShowInfo;
      break;
    case 'q':
      addQuadSurface();
      break;
    case 't':
      addTriangleSurface();
      break;
    case 'f':
      ofToggleFullscreen();
      break;
    case 's':
      surfaceManager.saveXmlSettings(PIMAPPER_USER_SURFACES_XML_FILE);
      break;
    case OF_KEY_BACKSPACE:
      surfaceManager.removeSelectedSurface();
      break;
    default:
      break;
  }
} // keyPressed

void ofxPiMapper::addFboSource(ofx::piMapper::FboSource &fboSource) {
  mediaServer.addFboSource(fboSource);
} // addFboSource

void ofxPiMapper::addTriangleSurface() {
  int surfaceType = ofx::piMapper::SurfaceType::TRIANGLE_SURFACE;
  
  vector<ofVec2f> vertices;
  float margin = 50.0f;
  vertices.push_back(ofVec2f((float)ofGetWidth() / 2.0f, margin));
  vertices.push_back(ofVec2f((float)ofGetWidth() - margin, (float)ofGetHeight() - margin));
  vertices.push_back(ofVec2f(margin, (float)ofGetHeight() - margin));
  
  vector<ofVec2f> texCoords;
  texCoords.push_back(ofVec2f(0.5f, 0.0f));
  texCoords.push_back(ofVec2f(1.0f, 1.0f));
  texCoords.push_back(ofVec2f(0.0f, 1.0f));
  surfaceManager.addSurface(surfaceType, vertices, texCoords);
  
  // Select this surface right away
  surfaceManager.selectSurface(surfaceManager.size() - 1);
  
} // addTriangleSurface

void ofxPiMapper::addQuadSurface() {
  
  int surfaceType = ofx::piMapper::SurfaceType::QUAD_SURFACE;
  
  vector<ofVec2f> vertices;
  float margin = 50.0f;
  vertices.push_back(ofVec2f(margin, margin));
  vertices.push_back(ofVec2f((float)ofGetWidth() - margin, margin));
  vertices.push_back(ofVec2f((float)ofGetWidth() - margin, (float)ofGetHeight() - margin));
  vertices.push_back(ofVec2f(margin, (float)ofGetHeight() - margin));
  
  vector<ofVec2f> texCoords;
  texCoords.push_back(ofVec2f(ofVec2f(0.0f, 0.0f)));
  texCoords.push_back(ofVec2f(ofVec2f(1.0f, 0.0f)));
  texCoords.push_back(ofVec2f(ofVec2f(1.0f, 1.0f)));
  texCoords.push_back(ofVec2f(ofVec2f(0.0f, 1.0f)));
  
  surfaceManager.addSurface(surfaceType, vertices, texCoords);
  
  // select this surface right away
  surfaceManager.selectSurface(surfaceManager.size() - 1);
  
} // addQuadSurface

ofx::piMapper::MediaServer& ofxPiMapper::getMediaServer() {
  return mediaServer;
}

ofx::piMapper::SurfaceManager& ofxPiMapper::getSurfaceManager() {
  return surfaceManager;
}