BCModelOutput.h

Go to the documentation of this file.

#ifndef __BCMODELOUTPUT__H
#define __BCMODELOUTPUT__H

/*
 * Copyright (C) 2008, Daniel Kollar and Kevin Kroeninger.
 * All rights reserved.
 *
 * For the licensing terms see doc/COPYING.
 */

// ---------------------------------------------------------

#include <vector>

// BAT classes
class BCModel;

// ROOT classes
class TTree;
class TFile;
class TObject;

const int MAXNPARAMETERS = 20;

// ---------------------------------------------------------

class BCModelOutput
{
   public:

      /* @{ */

      BCModelOutput();

      BCModelOutput(BCModel * model, const char * filenname);

      BCModelOutput(const BCModelOutput & modeloutput);

      virtual ~BCModelOutput();

      /* @} */

      /* @{ */

      BCModelOutput & operator = (const BCModelOutput & modeloutput);

      /* @} */

      /* @{ */

      TTree * GetAnalysisTree()
         { return fAnalysisTree; };

      TFile * GetFile()
         { return fOutputFile; };

      /* @} */

      /* @{ */

      void SetModel(BCModel * model);


      void SetFile(const char * filename);

      /* @} */

      /* @{ */

      void WriteMarkovChain(bool flag = true);

      void FillAnalysisTree();

      void WriteMarginalizedDistributions();

      void WriteErrorBand();

      void Write(TObject * o);

      void Close();

      /* @} */

   private:

      void Init();

      /*
       * Copies this BCModelOutput into another one */
      void Copy(BCModelOutput & modeloutput) const;

      void InitializeAnalysisTree();

      void InitializeMarkovChainTrees();

      void InitializeSATree(); 

      TTree * fAnalysisTree;

      /*
       * The trees containing the Markov chains. The length of the vector
       * is fMCMCNChains. */
      std::vector<TTree *> fMarkovChainTrees;

      /*
       * The tree for the simulated annealing. */ 
      TTree * fTreeSA; 

      char * fFileName;

      TFile * fOutputFile;

      BCModel * fModel;

      int fIndex;
      unsigned int fNParameters;
      double fProbability_apriori;
      double fProbability_aposteriori;
      double fMode_global[MAXNPARAMETERS];
      double fMode_marginalized[MAXNPARAMETERS];
      double fMean_marginalized[MAXNPARAMETERS];
      double fMedian_marginalized[MAXNPARAMETERS];
      double fQuantile_05[MAXNPARAMETERS];
      double fQuantile_10[MAXNPARAMETERS];
      double fQuantile_16[MAXNPARAMETERS];
      double fQuantile_84[MAXNPARAMETERS];
      double fQuantile_90[MAXNPARAMETERS];
      double fQuantile_95[MAXNPARAMETERS];

      std::vector<double> * fParameters;
      std::vector<double> * fLogLikelihood;
      std::vector <int> * fIteration;

      int * fSAIteration;
      int fSANParameters; 
      double * fSATemperature; 
      double * fSALogProb; 
      std::vector<double> * fSAParameters; 
};

// ---------------------------------------------------------

#endif

---