analysis.h

Go to the documentation of this file.

/*
  @copyright Russell Standish 2000-2013
  @author Russell Standish
  This file is part of EcoLab

  Open source licensed under the MIT license. See LICENSE for details.
*/

#ifndef ANALYSIS_H
#define ANALYSIS_H
#include "arrays.h"
#include "graph.h"
#include "TCL_obj_stl.h"
#include <vector>
#include <string>
#include <algorithm>

namespace ecolab
{

  class palette_class
  { 
  public:
    static int size;
    static std::vector<std::string> table;
    const char * operator[](int i){return table[i%size].c_str();}
    palette_class();
  };

  struct Stats: public array_ns::array<float>
  {
    double sum, sumsq; 
    float max, min;    
    Stats(): sum(0), sumsq(0), max(-std::numeric_limits<float>::max()), 
             min(std::numeric_limits<float>::max()) {}
    void clear() {
      resize(0); sum=sumsq=0; 
      max=-std::numeric_limits<float>::max();
      min=std::numeric_limits<float>::max();
    }
    double av() {return sum/size();} 
    double median() {
      std::nth_element(begin(),begin()+size()/2,end());
      return *(begin()+size()/2);
    }
    double stddev() {double d=sumsq - sum*sum/size();
      if (d<0) return 0; // deal with roundoff error
      else return sqrt( d/size());}

    Stats& operator<<=(float x);
    Stats& operator<<=(const array_ns::array<float>& x);
    Stats& operator<<=(const array_ns::array<double>& x)
    {operator<<=(array_ns::array<float>(x)); return *this;}

    void add_data(TCL_args args) {
      array_ns::array<float> t;
      args>>t;
      operator<<=(t);
    }
    bool operator<(const Stats& x) const 
    {return std::lexicographical_compare(begin(),end(),x.begin(),x.end());}
  };

  template<class charT, class Traits>
  std::basic_ostream<charT,Traits>& operator<<
    (std::basic_ostream<charT,Traits>& o, const Stats& x)
  {return o<<static_cast<const array_ns::array<float>&>(x);}

  struct HistoStats: public Stats
  {
    unsigned nbins; 
    bool logbins;  
    float logmin; 
    HistoStats(): nbins(100), logbins(false) {}
    array_ns::array<double> histogram(); 
    array_ns::array<double> bins();  

    double loglikelihood(TCL_args args); 
    array_ns::array<double> fitPowerLaw(TCL_args); 

    double fitExponential() {return av();} 

    array_ns::array<double> fitNormal() { 
      return array_ns::array<double>()<<av()<<stddev();
    }

    array_ns::array<double> fitLogNormal();   
  };

  struct NetworkFromTimeSeries: public std::vector<Stats>
  {
    ConcreteGraph<DiGraph> net;
    std::vector<unsigned> n;
    void constructNet();
    std::vector<int> columnIdx;
    void loadData(const char* filename);
  };
}
#include "analysis.cd"

#ifdef BLT
#include "analysisBLT.h"
#endif
#ifdef CAIRO
#include "analysisCairo.h"
#endif
#endif