#include <fstream>
#include <iostream>
#include <iomanip>
using namespace std;
#include "hgeantinput.h"

#define NKEY 14

// Function to read filenames from initialization-file and
// to store them in the arrays geoFiles, hitFiles, fieldFiles and tupFiles 
int HGeantInput::readFileNames() {
  ifstream fin;
  fin.open(sourceFile,ios::in);
  if(fin.fail()) {
    cerr << "failed to open file " << sourceFile << endl;
    return 0;
  }
  TString tmp(256), s(256); 

  TString geaIoKey[NKEY]={geoKey1,geoKey2,hitKey,paramKey,tupKey,evKey,
                          fieldKey,rootKey,detSetupKey,
                          refRunKey,historyKey,debugKey,dbSupportKey,
                          refRunIdKey};
  Int_t k;
  while(!fin.eof()) {
    tmp.ReadLine(fin);
    tmp=tmp.Strip(tmp.kBoth);
    if (!tmp.IsNull() && tmp(0,2)!="//" && tmp(0,1)!="*") {
      for (int i=0;i<NKEY;i++){
	if (tmp.Contains(geaIoKey[i])){
	  switch(i){ 
	  case 0:
	    geoFiles[geoNoFiles++]=tmp;
	    break;
	  case 1:
	    geoFiles[geoNoFiles++]=tmp;
	    break;
	  case 2:
	    hitFiles[hitNoFiles++]=tmp;
	    break;
	  case 3:
            k=tmp.First(' ')+1;
            if (k) {
              paramFile=tmp(k,tmp.Length()-k);
              paramFile=paramFile.Strip(paramFile.kBoth);
            }
            break;
	  case 4:
	    tupFiles[tupNoFiles++]=tmp;
	    break;
	  case 5:
	    evFiles[evNoFiles++]=tmp;
	    break;
	  case 6:
            fieldFiles[fieldNoFiles++]=tmp;
            break;
	  case 7:
            if (!tmp.Contains(geaIoKey[3])) rootFiles[rootNoFiles++]=tmp;
            break;
	  case 8:
            detSetupFile=tmp;
            break;
          case 9:
            k=tmp.Last(' ')+1;
            if (k) simulRefRun=tmp(k,tmp.Length()-k);    
            break;
          case 10:
            k=tmp.First(' ')+1;
            if (k) {
              historyDate=tmp(k,tmp.Length()-k);
              historyDate=historyDate.Strip(historyDate.kBoth);
            }
            break;
          case 11:
            k=tmp.First(' ')+1;
            if (k) {
              debugFile=tmp(k,tmp.Length()-k);
              debugFile=debugFile.Strip(debugFile.kBoth);
            }
            break;
          case 12:
            k=tmp.First(' ')+1;
            if (k) {
              dbSupport=tmp(k,tmp.Length()-k);
              dbSupport=dbSupport.Strip(dbSupport.kBoth);
              dbSupport.ToUpper();
            }
            break;
          case 13:
            k=tmp.Last(' ')+1;
            if (k) {
              s=tmp(k,tmp.Length()-k);
              simulRefRunId=s.Atoi();
            }    
            break;
	  default:
	    cerr << "no container for" << tmp << "defined";
	    return 0;
	  }
	}
      }
    }
  }
  
  fin.close();

  if (geoNoFiles>0||simulRefRun.Length()>0) return 1;
  else {
    cerr<<"*----------------------------------------------------------------*"<<endl;
    cerr<<endl<< "*     NO GEOMETRY FILES FOUND IN " <<sourceFile<<"!       *"<<endl;
    cerr<<endl<< "*----------------------------------------------------------------*"<<endl; 
    return 0;
  }
  return 1;
}


// Function to show all filenames
void HGeantInput::showFileNames() {
  if (simulRefRun.Length()>0&&dbSupport.CompareTo("ON")==0) {
    cout <<"*---------------------------------*\n";
    cout <<"| Initialization from database:   |\n";
    cout <<"*---------------------------------*\n";
    cout <<"     Simulation Reference Run: " << simulRefRun << '\n';
    cout <<"     History Date:             " << historyDate << '\n';
  } else {
    if (simulRefRunId>0) {
      cout <<"*---------------------------------*\n";
      cout <<"| Simulation Reference Run:       |\n";
      cout <<"*---------------------------------*\n";
      cout <<"     Run Id: " << simulRefRunId << '\n';
    }
  }
  cout <<"*---------------------------------*\n";
  cout <<"| input files for geometry:       |\n";
  cout <<"*---------------------------------*\n";
  for(int i=0;i<geoNoFiles;i++) { cout << "     " << getGeoFileName(i) << '\n'; }
  cout <<"*---------------------------------*\n";
  if (detSetupFile.Length()>0) { 
    cout <<"*---------------------------------*\n";
    cout <<"| Detector setup file for geometry:       |\n";
    cout <<"*---------------------------------*\n";
    cout << "     " << detSetupFile << '\n';
    cout <<"*---------------------------------*\n";
  }
  cout <<"| input files for hit definition: |\n";
  cout <<"*---------------------------------*\n";
  for(int i=0;i<hitNoFiles;i++) { cout << "     " << getHitFileName(i) << '\n'; }
  cout <<"*---------------------------------*\n";
  cout <<"| magnetic field-map file(s):     |\n";
  cout <<"*---------------------------------*\n";
  if (paramFile.Length()>0) { 
    cout <<"*---------------------------------*\n";
    cout <<"| Parameter file:       |\n";
    cout <<"*---------------------------------*\n";
    cout << "     " << paramFile << '\n';
    cout <<"*---------------------------------*\n";
  }
  for(int i=0;i<fieldNoFiles;i++) { cout << "     " << getFieldFileName(i) << '\n'; }
  cout <<"*---------------------------------*\n";
  cout <<"| ntuples requested:              |\n";
  cout <<"*---------------------------------*\n";
  for(int i=0;i<tupNoFiles;i++) { cout << "     " << getTupFileName(i) << '\n'; }
  if (rootNoFiles > 0) {
     cout <<"*---------------------------------*\n";
     cout <<"| ROOT file(s):                   |\n";
     cout <<"*---------------------------------*\n";
     for(int i=0;i<rootNoFiles;i++) { cout << "     " << getRootFileName(i) << '\n'; }
  }
  cout <<"*---------------------------------*\n";
  cout <<"| event input files :             |\n";
  cout <<"*---------------------------------*\n";
  for(int i=0;i<evNoFiles;i++) { cout << "     " << getEvFileName(i) << '\n'; }
  cout <<"*------------------------------------------------*"<<endl;
  if (debugFile.Length()>0) { 
    cout <<"*---------------------------------*\n";
    cout <<"| Debug file for geometry:       |\n";
    cout <<"*---------------------------------*\n";
    cout << "     " << debugFile << '\n';
    cout <<"*---------------------------------*\n";
  }
}