Added support for serializing complex matrices in Yaml

Modules/FDEM1D/examples/CircularLoop.cpp

@@ -0,0 +1,55 @@
+/* This file is part of Lemma, a geophysical modelling and inversion API.
+ * More information is available at http://lemmasoftware.org
+ */
+
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+/**
+ * @file
+ * @date      12/01/2017 09:38:57 PM
+ * @version   $Id$
+ * @author    Trevor Irons (ti)
+ * @email     tirons@egi.utah.edu
+ * @copyright Copyright (c) 2017, University of Utah
+ * @copyright Copyright (c) 2017, Trevor Irons & Lemma Software, LLC
+ */
+
+
+#include <FDEM1D>
+using namespace Lemma;
+
+int main(int argc, char** argv) {
+
+    if (argc < 6) {
+        std::cout << "Usage:\n";
+        std::cout << "CircularLoop <radius> <centreNorthing>  <centreEasting>"
+                  <<  "  <numTurns>  <npoints>\n"  ;
+        exit(EXIT_SUCCESS);
+    }
+
+    Real rad = atof(argv[1]);
+    Real cx = atof(argv[2]);
+    Real cy = atof(argv[3]);
+    int nt = atoi(argv[4]);
+    int np = atoi(argv[5]);
+    std::cout << rad << std::endl;
+
+    auto wire = PolygonalWireAntenna::NewSP();
+        wire->SetNumberOfPoints(np);
+        wire->SetNumberOfTurns(nt);
+        Real da = 2.*PI / (Real)(np-1);
+        for (int ip=0; ip<np-1; ++ip) {
+            wire->SetPoint(ip, cx+rad*std::sin(da*static_cast<Real>(ip)),
+                               cy+rad*std::cos(da*static_cast<Real>(ip)), -1e-3);
+        }
+        wire->SetPoint(np-1, cx, cy+rad, -1e-3);
+
+    std::cout << *wire << std::endl;
+
+    //auto wire2 = wire->Clone();
+    //std::cout << "copy\n" << *wire2 << std::endl;
+}
+

Modules/LemmaCore/include/helper.h

@@ -321,6 +321,42 @@ struct convert<Lemma::MatrixXr> {
 
 };
 
+template<>
+struct convert<Lemma::MatrixXcr> {
+  static Node encode(const Lemma::MatrixXcr& rhs) {
+    Node node;
+    node["rows"] = rhs.rows();
+    node["cols"] = rhs.cols();
+    for (int ir=0; ir<rhs.rows(); ++ir) {
+        for (int ic=0; ic<rhs.cols(); ++ic) {
+            node["data"][ir][ic] = rhs(ir,ic);
+        }
+        node["data"][ir].SetStyle(YAML::EmitterStyle::Flow);
+    }
+    //node.SetStyle(YAML::EmitterStyle::Block);
+    node.SetTag( "MatrixXcr" );
+    return node;
+  }
+
+  static bool decode(const Node& node, Lemma::MatrixXcr& rhs) {
+    if( node.Tag() != "MatrixXcr" ) {
+        return false;
+    }
+    int nc = node["cols"].as<int>();
+    int nr = node["rows"].as<int>();
+    rhs.resize(nr, nc);
+    for (int ir=0; ir<nr; ++ir) {
+        int ic=0;
+        for(YAML::const_iterator it=node["data"][ir].begin(); it!=node["data"][ir].end(); ++it) {
+            rhs(ir,ic) = it->as<Lemma::Complex>();
+            ++ic;
+        }
+    }
+    return true;
+  }
+
+};
+
 }
 
 #endif   // ----- #ifndef HELPER_INC  -----