From 3159a4779d017883cb006964b864a720adadf584 Mon Sep 17 00:00:00 2001
From: Krisjanis Rijnieks <krisjanis.rijnieks@gmail.com>
Date: Fri, 17 Oct 2014 18:33:27 +0200
Subject: [PATCH] Show and hide source selection radio buttons depending on
 whether they contain sources or not

---
 src/Sources/ImageSource.cpp |  4 +-
 src/SourcesEditor.cpp       | 81 ++++++++++++++++++++++++++++---------
 src/ui/RadioList.cpp        |  2 +-
 3 files changed, 65 insertions(+), 22 deletions(-)

diff --git a/src/Sources/ImageSource.cpp b/src/Sources/ImageSource.cpp
index f2fd23e..0cf8dde 100644
--- a/src/Sources/ImageSource.cpp
+++ b/src/Sources/ImageSource.cpp
@@ -18,8 +18,8 @@ namespace ofx {
       //cout << "path: " << path << endl;
       image = new ofImage();
       if (!image->loadImage(filePath)) {
-        ofLogFatalError("ImageSource") << "Could not load image";
-        std::exit(EXIT_FAILURE);
+        ofLogWarning("ImageSource") << "Could not load image";
+        //std::exit(EXIT_FAILURE);
       }
       texture = &image->getTextureReference();
       loaded = true;
diff --git a/src/SourcesEditor.cpp b/src/SourcesEditor.cpp
index 9607b90..d106d67 100644
--- a/src/SourcesEditor.cpp
+++ b/src/SourcesEditor.cpp
@@ -38,15 +38,35 @@ namespace piMapper {
   void SourcesEditor::setup(ofEventArgs& args) {
     imageSelector = new RadioList();
     videoSelector = new RadioList();
-    // Get image names from media server
-    vector<string> imageNames = mediaServer->getImageNames();
-    imageSelector->setup("Images", imageNames, mediaServer->getImagePaths());
-    imageSelector->setPosition(20, 20);
-    ofAddListener(imageSelector->onRadioSelected, this, &SourcesEditor::handleImageSelected);
-    vector<string> videoNames = mediaServer->getVideoNames();
-    videoSelector->setup("Videos", videoNames, mediaServer->getVideoPaths());
-    videoSelector->setPosition(250, 20);
-    ofAddListener(videoSelector->onRadioSelected, this, &SourcesEditor::handleVideoSelected);
+    
+    // Get media count
+    int numImages = mediaServer->getNumImages();
+    int numVideos = mediaServer->getNumVideos();
+    
+    // Depending on media count, decide what to load and initialize
+    if (numImages) {
+      // Get image names from media server
+      vector<string> imageNames = mediaServer->getImageNames();
+      imageSelector->setup("Images", imageNames, mediaServer->getImagePaths());
+      ofAddListener(imageSelector->onRadioSelected, this, &SourcesEditor::handleImageSelected);
+    }
+    if (numVideos) {
+      vector<string> videoNames = mediaServer->getVideoNames();
+      videoSelector->setup("Videos", videoNames, mediaServer->getVideoPaths());
+      ofAddListener(videoSelector->onRadioSelected, this, &SourcesEditor::handleVideoSelected);
+    }
+    
+    if (numImages) {
+      imageSelector->setPosition(20, 20);
+      if (numVideos) {
+        videoSelector->setPosition(250, 20);
+      }
+    } else {
+      if (numVideos) {
+        videoSelector->setPosition(20, 20);
+      }
+    }
+    
   }
 
   void SourcesEditor::draw() {
@@ -55,13 +75,22 @@ namespace piMapper {
       ofLogNotice("SourcesEditor") << "No surface selected";
       return;
     }
-    imageSelector->draw();
-    videoSelector->draw();
+    if (imageSelector->size()) {
+      imageSelector->draw();
+    }
+    if (videoSelector->size()) {
+      videoSelector->draw();
+    }
+    
   }
 
   void SourcesEditor::disable() {
-    imageSelector->disable();
-    videoSelector->disable();
+    if (imageSelector->size()) {
+      imageSelector->disable();
+    }
+    if (videoSelector->size()) {
+      videoSelector->disable();
+    }
   }
 
   void SourcesEditor::enable() {
@@ -70,8 +99,12 @@ namespace piMapper {
       ofLogNotice("SourcesEditor") << "No surface selected. Not enabling and not showing source list.";
       return;
     }
-    imageSelector->enable();
-    videoSelector->enable();
+    if (imageSelector->size()) {
+      imageSelector->enable();
+    }
+    if (videoSelector->size()) {
+      videoSelector->enable();
+    }
     BaseSource* source = surfaceManager->getSelectedSurface()->getSource();
     selectSourceRadioButton(source->getPath());
   }
@@ -98,13 +131,23 @@ namespace piMapper {
   void SourcesEditor::selectSourceRadioButton(std::string& sourcePath) {
     if (sourcePath == "") {
       ofLogNotice("SourcesEditor") << "Path is empty";
-      imageSelector->unselectAll();
-      videoSelector->unselectAll();
+      if (imageSelector->size()) {
+        imageSelector->unselectAll();
+      }
+      if (videoSelector->size()) {
+        videoSelector->unselectAll();
+      }
       return;
     } else {
       // Check image selector first
-      bool imageRadioSelected = imageSelector->selectItemByValue(sourcePath);
-      bool videoRadioSelected = videoSelector->selectItemByValue(sourcePath);
+      bool imageRadioSelected = false;
+      bool videoRadioSelected = false;
+      if (imageSelector->size()) {
+        imageRadioSelected = imageSelector->selectItemByValue(sourcePath);
+      }
+      if (videoSelector->size()) {
+        videoRadioSelected = videoSelector->selectItemByValue(sourcePath);
+      }
       if (imageRadioSelected || videoRadioSelected) {
         return;
       }
diff --git a/src/ui/RadioList.cpp b/src/ui/RadioList.cpp
index bb7f267..d625b86 100644
--- a/src/ui/RadioList.cpp
+++ b/src/ui/RadioList.cpp
@@ -159,7 +159,7 @@ string RadioList::getItemName(int index) {
   return toggle->getName();
 }
 
-int RadioList::size() { return guiGroup.getNumControls(); }
+int RadioList::size() { return storedValues.size(); }
 
 void RadioList::onToggleClicked(bool& toggleValue) {
   unselectAll();