fastica_main.cc

Go to the documentation of this file.

/* MLPACK 0.2
 *
 * Copyright (c) 2008, 2009 Alexander Gray,
 *                          Garry Boyer,
 *                          Ryan Riegel,
 *                          Nikolaos Vasiloglou,
 *                          Dongryeol Lee,
 *                          Chip Mappus, 
 *                          Nishant Mehta,
 *                          Hua Ouyang,
 *                          Parikshit Ram,
 *                          Long Tran,
 *                          Wee Chin Wong
 *
 * Copyright (c) 2008, 2009 Georgia Institute of Technology
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 */
#include "fastica.h"

const fx_entry_doc fastica_main_entries[] = {
  {"data", FX_REQUIRED, FX_STR, NULL,
   "  A file containing data.\n"},
  {"ic_filename", FX_PARAM, FX_STR, NULL,
   "  Filename to which independent components are written.\n"},
  {"unmixing_filename", FX_PARAM, FX_STR, NULL,
   "  Filename to which unmixing matrix is written.\n"},
  FX_ENTRY_DOC_DONE
};

const fx_submodule_doc fastica_main_submodules[] = {
  {"fastica", &fastica_doc,
   " Responsible for performing fastica.\n"},
  FX_SUBMODULE_DOC_DONE
};

const fx_module_doc fastica_main_doc = {
  fastica_main_entries, fastica_main_submodules,
  "This program performs fastica.\n"
};



int main(int argc, char *argv[]) {
  fx_module* root = fx_init(argc, argv, &fastica_main_doc);

  Matrix X;
  const char* data = fx_param_str_req(NULL, "data");
  data::Load(data, &X);

  const char* ic_filename = fx_param_str(NULL, "ic_filename", "ic.dat");
  const char* unmixing_filename =
    fx_param_str(NULL, "unmixing_filename", "unmixing.dat");
  struct datanode* fastica_module =
    fx_submodule(root, "fastica");

  FastICA fastica;

  int success_status = SUCCESS_FAIL;
  if(fastica.Init(X, fastica_module) == SUCCESS_PASS) {
    Matrix W, Y;
    if(fastica.DoFastICA(&W, &Y) == SUCCESS_PASS) {
      SaveCorrectly(unmixing_filename, W);
      data::Save(ic_filename, Y);
      success_status = SUCCESS_PASS;
      VERBOSE_ONLY( W.PrintDebug("W") );
    }
  }
  

  if(success_status == SUCCESS_FAIL) {
    VERBOSE_ONLY( printf("FAILED!\n") );
  }

  fx_done(root);

  return success_status;
}