Make EnumWriter exceptions a bit more informative.

[ardour.git] / libs / pbd / epa.cc

diff --git a/libs/pbd/epa.cc b/libs/pbd/epa.cc
index 8665823d771872fc0de7973986f3d21da501da18..3d3f7477d7dfe56151a809b52398fed86698b1b1 100644 (file)
--- a/libs/pbd/epa.cc
+++ b/libs/pbd/epa.cc
@@ -21,6 +21,7 @@
 #include <iostream>
 
 #include "pbd/epa.h"
+#include "pbd/strsplit.h"
 
 extern char** environ;
 
@@ -29,8 +30,9 @@ using namespace std;
 
 EnvironmentalProtectionAgency* EnvironmentalProtectionAgency::_global_epa = 0;
 
-EnvironmentalProtectionAgency::EnvironmentalProtectionAgency (bool arm)
+EnvironmentalProtectionAgency::EnvironmentalProtectionAgency (bool arm, const std::string& envname)
         : _armed (arm)
+        , _envname (envname)
 {
         if (_armed) {
                 save ();
@@ -55,29 +57,65 @@ EnvironmentalProtectionAgency::save ()
 {
         e.clear ();
 
-        for (size_t i = 0; environ[i]; ++i) {
+        if (!_envname.empty()) {
+                
+                /* fetch environment from named environment variable, rather than "environ"
+                 */
 
-                string estring = environ[i];
-                string::size_type equal = estring.find_first_of ('=');
+                const char* estr = getenv (_envname.c_str());
 
-                if (equal == string::npos) {
-                        /* say what? an environ value without = ? */
-                        continue;
+                if (!estr) {
+                        return;
+                }
+                
+                /* parse line by line, and save into "e" 
+                 */
+
+                vector<string> lines;
+                split (estr, lines, '\n');
+
+                for (vector<string>::iterator i = lines.begin(); i != lines.end(); ++i) {
+
+                        string estring = *i;
+                        string::size_type equal = estring.find_first_of ('=');
+                        
+                        if (equal == string::npos) {
+                                /* say what? an environ value without = ? */
+                                continue;
+                        }
+                        
+                        string before = estring.substr (0, equal);
+                        string after = estring.substr (equal+1);
+                        
+                        e.insert (pair<string,string>(before,after));
+                }
+                
+        } else {
+
+                /* fetch environment from "environ"
+                 */
+
+                for (size_t i = 0; environ[i]; ++i) {
+                        
+                        string estring = environ[i];
+                        string::size_type equal = estring.find_first_of ('=');
+                        
+                        if (equal == string::npos) {
+                                /* say what? an environ value without = ? */
+                                continue;
+                        }
+                        
+                        string before = estring.substr (0, equal);
+                        string after = estring.substr (equal+1);
+                        
+                        e.insert (pair<string,string>(before,after));
                 }
-
-                string before = estring.substr (0, equal);
-                string after = estring.substr (equal+1);
-
-                cerr << "Save [" << before << "] = " << after << endl;
-
-                e.insert (pair<string,string>(before,after));
         }
 }                         
 void
 EnvironmentalProtectionAgency::restore () const
 {
         for (map<string,string>::const_iterator i = e.begin(); i != e.end(); ++i) {
-                cerr << "Restore [" << i->first << "] = " << i->second << endl;
                 setenv (i->first.c_str(), i->second.c_str(), 1);
         }
 }