/***************************************************************************

  write_runHeader.cpp

  Author: Andreas Suter
  e-mail: andreas.suter@psi.ch

  $Id$

***************************************************************************/

/***************************************************************************
 *   Copyright (C) 2007-2011 by Andreas Suter                              *
 *   andreas.suter@psi.ch                                                  *
 *                                                                         *
 *   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.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#include <cstdlib>
#include <ctime>
#include <cstring>

#include <iostream>
using namespace std;

#include <TROOT.h>
#include <TFile.h>
#include <TFolder.h>

#include "TPsiRunHeader.h"

void write_psi_runHeader_syntax()
{
  cout << endl << "usage: write_psi_runHeader <fileName> <headerDefinition> [<strict>]";
  cout << endl << "       <fileName> is the file name including the extention root, e.g. test.root";
  cout << endl << "       <headerDefinition> is the header definition XML-file.";
  cout << endl << "       <strict> 'strict'=strict validation; otherwise=less strict validation.";
  cout << endl << endl;
}

int main(int argc, char *argv[])
{
  if ((argc != 3) && (argc != 4)) {
    write_psi_runHeader_syntax();
    return 1;
  }

  Bool_t strict = false;
  if (argc == 4) {
    if (!strcmp(argv[3], "strict"))
      strict = true;
  }

  // PSI Run Header object
  TPsiRunHeader *header = new TPsiRunHeader(argv[2]);
  TPsiRunProperty prop;

  // run info
  header->Set("RunInfo/Version", "$Id$");
  header->Set("RunInfo/Generator", "any2many");
  header->Set("RunInfo/File Name", "thisIsAFileName");
  header->Set("RunInfo/Run Title", "here comes the run title");
  header->Set("RunInfo/Run Number", 577);

  // run info - start/stop time and duration
  TString startTime("2011-04-19 14:25:22"), stopTime("2011-04-19 19:13:47");
  struct tm tm_start, tm_stop;
  header->Set("RunInfo/Run Start Time", startTime);
  header->Set("RunInfo/Run Stop Time", stopTime);
  // calculate run duration
  memset(&tm_start, 0, sizeof(tm_start));
  strptime(startTime.Data(), "%Y-%m-%d %H:%M:%S", &tm_start);
  memset(&tm_stop, 0, sizeof(tm_stop));
  strptime(stopTime.Data(), "%Y-%m-%d %H:%M:%S", &tm_stop);
  Double_t duration = difftime(mktime(&tm_stop), mktime(&tm_start));
  header->Set("RunInfo/Run Duration", (Int_t)duration);

  header->Set("RunInfo/Laboratory", "PSI");
  header->Set("RunInfo/Area", "piM3.2");
  header->Set("RunInfo/Instrument", "GPS");

  prop.Set("Muon Beam Momentum", 28.1, "MeV/c");
  header->Set("RunInfo/Muon Beam Momentum", prop);

  header->Set("RunInfo/Muon Species", "positive muon");
  header->Set("RunInfo/Setup", "a very special setup");
  header->Set("RunInfo/Comment", "nothing more to be said");
  header->Set("RunInfo/Sample Name", "the best ever");

  prop.Set("Sample Temperature", 3.2, 3.21, 0.05, "K", "CF1");
  header->Set("RunInfo/Sample Temperature", prop);

  prop.Set("Sample Magnetic Field", 350.0, 350.002, 0.005, "G", "WXY");
  header->Set("RunInfo/Sample Magnetic Field", prop);

  header->Set("RunInfo/No of Histos", 4);

  TStringVector detectorName;
  detectorName.push_back("forward");
  detectorName.push_back("top");
  detectorName.push_back("backward");
  detectorName.push_back("bottom");
  header->Set("RunInfo/Histo Names", detectorName);

  header->Set("RunInfo/Histo Length", 8192);

  prop.Set("Time Resolution", 0.1953125, "ns", "TDC 9999");
  header->Set("RunInfo/Time Resolution", prop);

  TIntVector t0;
  for (UInt_t i=0; i<4; i++) t0.push_back(215+(Int_t)(5.0*(Double_t)rand()/(Double_t)RAND_MAX));
  header->Set("RunInfo/Time Zero Bin", t0);
  for (UInt_t i=0; i<4; i++) t0[i] += 12;
  header->Set("RunInfo/First Good Bin", t0);
  for (UInt_t i=0; i<4; i++) t0[i] = 8191;
  header->Set("RunInfo/Last Good Bin", t0);

  TIntVector readGreenOffset;
  readGreenOffset.push_back(0);
  readGreenOffset.push_back(10);
  readGreenOffset.push_back(20);
  readGreenOffset.push_back(30);
  header->Set("RunInfo/Red-Green Offsets", readGreenOffset);

  TStringVector redGreenDescription;
  redGreenDescription.push_back("E-field/light off/off");
  redGreenDescription.push_back("E-field/light on/off");
  redGreenDescription.push_back("E-field/light off/on");
  redGreenDescription.push_back("E-field/light on/on");
  header->Set("RunInfo/Red-Green Description", redGreenDescription);

  TStringVector scHistoNames;
  scHistoNames.push_back("Sample Temperature");
  scHistoNames.push_back("Sample Magnetic Field");
  header->Set("RunInfo/Slow Control Histo Names", scHistoNames);

  TStringVector dummyTest;
  dummyTest.push_back("dummy1");
  dummyTest.push_back("dummy2");
  dummyTest.push_back("dummy3");
  header->Set("RunInfo/Dummy Test", dummyTest);

  // sample environment
  header->Set("SampleEnv/Cryo", "Konti-1");
  header->Set("SampleEnv/Insert", "X123");
  header->Set("SampleEnv/Orientation", "c-axis perp spin, perp field. spin perp field");

  prop.Set("CF2", 3.2, 3.22, 0.04, "K");
  header->Set("SampleEnv/CF2", prop);

  prop.Set("CF3", PRH_UNDEFINED, 3.27, 0.09, "K", "strange temperature");
  header->Set("SampleEnv/CF3", prop);

  prop.Set("CF4", 3.25, 3.28, "K");
  header->Set("SampleEnv/CF4", prop);

  prop.Set("CF5", 3.26, 3.29, "K", "another strange temperature");
  header->Set("SampleEnv/CF5", prop);

  prop.Set("Dummy Prop", -2.0, -2.001, 0.002, "SI-unit");
  header->Set("SampleEnv/Dummy Prop", prop);

  // magnetic field environment
  header->Set("MagFieldEnv/Name", "Bpar");
  prop.Set("Current", 1.34, "A");
  header->Set("MagFieldEnv/Current", prop);

  // beamline
  header->Set("Beamline/WSX61a", "DAC = 3289, ADC = 0.800");

  TIntVector dummyInt;
  for (UInt_t i=0; i<3; i++) dummyInt.push_back(i+1000);
  header->Set("Beamline/Dummy Int", dummyInt);


  // scaler
  header->Set("Scaler/Ip", 12332123);

  if (!header->IsValid(strict)) {
    cerr << endl << ">> **ERROR** run header validation failed." << endl;
    if (strict) { // clean up and quit
      delete header;
      return -1;
    }
  }

  TFile *f = new TFile(argv[1], "RECREATE", "psi_runHeader_test");
  if (f->IsZombie()) {
    delete f;
    return -1;
  }

  // root file header related things
  TFolder *runHeader = gROOT->GetRootFolder()->AddFolder("RunHeader", "PSI Run Header Info");
  gROOT->GetListOfBrowsables()->Add(runHeader, "RunHeader");

  TObjArray runInfo;
  header->GetHeaderInfo("RunInfo", runInfo);
  runHeader->Add(&runInfo);

  TObjArray sampleEnv;
  header->GetHeaderInfo("SampleEnv", sampleEnv);
  runHeader->Add(&sampleEnv);

  TObjArray magFieldEnv;
  header->GetHeaderInfo("MagFieldEnv", magFieldEnv);
  runHeader->Add(&magFieldEnv);

  TObjArray beamline;
  header->GetHeaderInfo("Beamline", beamline);
  runHeader->Add(&beamline);

  TObjArray scaler;
  header->GetHeaderInfo("Scaler", scaler);
  runHeader->Add(&scaler);

  runHeader->Write();

  f->Close();

  delete f;
  f = 0;

  return 0;
}