// process_cbnt.cxx
//
// David Adams
// April 2003
//
// Function that applies a PAW kumac to an ATLAS combined ntuple dataset.
//

#include "dataset_CBNT/process_cbnt.h"
#include <map>
#include <string>
#include <iostream>
#include <fstream>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include "dataset_util/FileStatus.h"
#include "dataset_util/Path.h"
#include "dataset_util/CommandLine.h"
#include "dataset_util/WorkingDirectory.h"
#include "dataset_util/Text.h"
#include "dataset_CBNT/HbookFile.h"
#include "dataset_CBNT/CbntDataset.h"

using std::string;
using std::cout;
using std::endl;
using std::ofstream;
using dset::CbntDataset;

typedef Text::Line Line;

//**********************************************************************
// Local definitions.
//**********************************************************************

namespace {

// Function which handles either initialization case.
// For initialization from file, init has only one line
// which is the file name.

int process_cbnt(const CbntDataset& dst, bool init_from_script,
                 const Text& inittxt, const Text& usercode,
                 string resfile, int verbose) {
  WorkingDirectory wd;
  // Find paw.
  Path exepath;
  exepath.getenv("PATH");
  if ( exepath.size() == 0 ) return 1;
  Path pawpath = exepath.find("paw");
  if ( pawpath.size() == 0 ) return 2;
  string pawexe = pawpath.front() + "/paw";
  if ( ! FileStatus(pawexe).is_executable() ) return 3;

  // Create the include file from the dataset.
  // Extract the HBNAME calls as we parse the file.
  // We do not expect to find any HBNAME calls.
  Text hbcalls;
  string incfile = "cbnt.inc";
  int id = 3333;
  HbookFile& hfile = dst.hbook_file();
  if ( ! hfile.is_valid() ) return 5;
  if ( hfile.open() != 0 ) return 6;
  if ( hfile.cd("cbnt") != 0 ) return 7;
  if ( hfile.evcode(id) != 0 ) return 8;
  if ( hfile.close() != 0 ) return 9;
  Text oldcode("evcode.f");
  Text newcode;
  newcode.append("* Created in process_cbnt");
  newcode.append("");
  bool found_end = false;
  for ( Text::size_type ilin=0; ilin<oldcode.size(); ++ilin ) {
    const Line& line = oldcode.line(ilin);
    // Skip the function  or subroutine declaration.
    if ( line.find("REAL FUNCTION") != Line::npos ) continue;
    if ( line.find("SUBROUTINE") != Line::npos ) continue;
    // Extract HBNAME calls.
    if ( line.find("HBNAME") != Line::npos ) {
      if ( line.find("$CLEAR") ) {
	hbcalls.append(line);
      } else {
	Line::size_type ip1 = line.find("'");
	Line::size_type ip2 = line.find("'", ip1);
	string key = line.substr(ip1+1, ip2-1);
	cout << "    " << key << endl;
	ContentId cid(CbntDataset::content_id_name(), key);
	cout << key << endl;
      }
      continue;
    }
    // Exit when we reach the user code section.
    if ( line.find("Enter user code") != Line::npos ) {
      found_end = true;
      break;
    }
    newcode.append(line);
  }
  if ( hbcalls.size() != 0 ) return 10;
  if ( ! found_end ) return 11;
  newcode.write(incfile);
  if ( ! FileStatus(incfile).is_readable() ) return 12;

  // Extract the included events from the dataset.
  string flagfile = "evflags.dat";
  const CbntDataset::FlagList evflags = dst.event_flags();
  if ( evflags.size() == 0 ) return 13;
  {
    string com = "rm -rf " + flagfile;
    system(com.c_str());
  }
  {
    ofstream flagfile(flagfile.c_str());
    flagfile << evflags.size() << '\n';
    for ( CbntDataset::FlagList::size_type ient=0;
          ient<evflags.size(); ++ient ) {
      flagfile << int(evflags[ient]) << '\n';
    }
  }

  // Initial histograms.
  string inithist = "inithist.hbk";

  // Write the user initilization kumac.
  string initfile = "init.kumac";
  system("rm -rf init.kumac");
  if ( init_from_script ) {
    inittxt.write(initfile);
  } else {
    if ( inittxt.size() != 1 ) return 14;
    inithist = inittxt.line(0);
  }

  // Create link to input ntuple file.
  if ( hfile.open(HbookFile::FORPAW) != 0 ) return 15;
  if ( hfile.paw_open_command().size() == 0 ) return 16;

  // Write the user code.
  string codefile = "evproc.f";
  Text evcode;
  evcode.append("* Code generated by process_cbnt");
  evcode.append("");
  evcode.append("      integer function evproc");
  evcode.append("      include 'cbnt.inc'");
  evcode.append("      logical evkeep");
//  evcode.append("      integer idnevt");
//  evcode.append("      common /pawidn/ idnevt");
  evcode.append("");
  evcode.append(hbcalls);
  evcode.append("");
  evcode.append("      if ( evkeep(idnevt, run, event) ) then");
  evcode.append("        call cbnt");
  evcode.append("        evproc = 1");
  evcode.append("      else");
  evcode.append("        evproc = 0");
  evcode.append("      endif");
  evcode.append("      end");
  evcode.append("");
  evcode.append("      logical function evkeep(idnevt, run, event)");
  evcode.append("      implicit none");
  evcode.append("      integer idnevt, run, event");
  evcode.append("      integer flgcnt, icnt");
  evcode.append("      logical flstat(1000)");
  evcode.append("      common /evkept/ flgcnt, flstat");
  evcode.append("");
  evcode.append("      if ( idnevt .eq. 1 ) then");
  evcode.append("        open(23, file='evflags.dat')");
  evcode.append("        read(23,*) flgcnt");
  evcode.append("        do icnt = 1, flgcnt");
  evcode.append("          read(23,*) flstat(icnt)");
  evcode.append("        enddo");
  evcode.append("        close(23)");
  evcode.append("      endif");
  evcode.append("");
  evcode.append("      evkeep = flstat(idnevt)");
  evcode.append("      end");
  evcode.append("");
  evcode.append("* User code follows");
  evcode.append("");
  evcode.append(usercode);
  evcode.write(codefile);

  // Create kumac.
  Text kum;
  if ( init_from_script ) {
    kum.append("macro/exec " + initfile);
    kum.append("hi/file 22 " + inithist + " CHOPT=N");
    kum.append("hrout *");
  } else {
    kum.append("hi/file 22 " + inithist + " CHOPT=R");
  }
  kum.append("hrin *");
  kum.append("close 22");
  kum.append("message Begin processing commands from process_cbnt");
  kum.append(hfile.paw_open_command());
  kum.append("cd cbnt");
  kum.append("ntuple/list");
  kum.append("ntuple/loop 3333 " + codefile);
  kum.append("histogram/file 26 " + resfile + " CHOPT=N");
  kum.append("hrout *");
  kum.append("close 26");
  if ( verbose > 4 ) cout << kum << endl;

  // Write kumac.
  string kumfile = "process_cbnt.kumac";
  {
    string com = "rm -rf ";
    com += kumfile;
    system(com.c_str());
  } 
  if ( FileStatus(kumfile).exists() ) return 17;
  kum.write(kumfile);

  // Create command line.
  CommandLine::Name comname = pawexe;
  CommandLine::ArgumentList comargs;
  comargs.push_back("-b");
  comargs.push_back(kumfile);
  CommandLine comline(comname, comargs);

  // Execute paw subprocess.
  int fstat = fork();
  // Fork error.
  if ( fstat == -1 ) {
    return 18;
  // Child process.
  } else if ( fstat == 0 ) {
    if ( verbose > 3 ) cout << comline << endl;
    execv( comline.cname(), comline.unsafe_argv() );
  // Parent process -- wait for child to finish.
  } else {
    int rstat;
    pid_t pid = wait(&rstat);
    assert( pid = fstat );
    // Normal exit.
    if ( WIFEXITED(rstat) ) {
      int stat = WEXITSTATUS(rstat);
      if ( stat != 0 ) return stat + 100;
    // Other exit.
    } else {
      return 20;
    }
  }

  return 0;
}

}  // end unnamed namespace

//**********************************************************************
// Functions.
//**********************************************************************

// Initialization from script.

int process_cbnt(const CbntDataset& dst, const Text& inittxt,
                 const Text& usercode, string resfile, int verbose) {
  return process_cbnt(dst, true, inittxt, usercode, resfile, verbose);
}

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

// Initialization from HBOOK file.

int process_cbnt(const CbntDataset& dst, string hbkfile,
                 const Text& usercode, string resfile, int verbose) {
  Text inittxt;
  inittxt.append(hbkfile);
  return process_cbnt(dst, false, inittxt, usercode, resfile, verbose);
}

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