//--- C/C++ includes ---
#include <iostream.h>

//--- Friend includes ---
#include "AMoleculeBody.hh"
#include "ACommandLineParser.hh"

extern "C"
#ifdef MSWIN
__declspec(dllexport)
#endif
void function1(AMoleculeBody *bodyList,
	       ACommandLineParser *parser,
	       std::ostream &fout);

extern "C"
#ifdef MSWIN
__declspec(dllexport)
#endif
void function2(AMoleculeBody *bodyList,
	       ACommandLineParser *parser,
	       std::ostream &fout);


// The function calculates number of chains, residues and atoms
// for each molecules. At the end it printe number of molecules.
// Use command "lib test1 function1" to run it.
void function1(AMoleculeBody *bodyList,
	       ACommandLineParser *parser,
	       std::ostream &fout)
{
  int n_mol = 0; // Counter for molecules

  for (AMoleculeBody *body = bodyList;body;body = body->getNext()) {

    AMolecule *mol = body->getMolecule();
    if (!mol) continue;

    n_mol++;

    int n_chains = 0; // Counter for chains
    int n_groups = 0; // Counter for groups
    int n_atoms  = 0; // Counter for atoms

    for (AChain *chain = mol->getFirstChain();chain;chain = chain->getNext()) {

      n_chains++;

      for (AGroup *group = chain->getFirstGroup();group;
	   group = group->getNext()) {

	n_groups++;

	for (AAtom *atom = group->getFirstAtom();atom;atom = atom->getNext()) {

	  n_atoms++;
	    
	}
      }
    }

    // Printing output to "Friend: command line" window
    fout<<"Molecule '"<<body->getMolname()<<"':"<<std::endl;
    fout<<"Chains...."<<n_chains<<std::endl;
    fout<<"Groups...."<<n_groups<<std::endl;
    fout<<"Atoms....."<<n_atoms<<std::endl<<std::endl;

    // Printing output to terminal
    cout<<"Molecule '"<<body->getMolname()<<"':"<<endl;
    cout<<"Chains...."<<n_chains<<endl;
    cout<<"Groups...."<<n_groups<<endl;
    cout<<"Atoms....."<<n_atoms<<endl<<endl;
  }

  // Printing output to "Friend: command line" window
  fout<<"Total "<<n_mol<<" molecules."<<std::endl;

  // Printing output to terminal
  cout<<"Total "<<n_mol<<" molecules."<<endl;
}


// The function parses command line arguments.
// It assumes the input of the form: bodyNum1 chain1 bodyNum2 chain2
// Use command "lib test1 function2 @0 A @1 B" to run it.
void function2(AMoleculeBody *bodyList,
	       ACommandLineParser *parser,
	       std::ostream &fout)
{
  std::string error_message("Bad parameter string.");

  fout<<std::endl;

  int bodyNum1 = parser->readBodyNum();
  if (parser->errorHappend()) {
    fout<<error_message.c_str()<<" "<<parser->getErrorMessage()<<std::endl;
    cerr<<error_message.c_str()<<" "<<parser->getErrorMessage()<<endl;
    return;
  }

  char chain1 = parser->readChar();
  if (parser->errorHappend()) {
    fout<<error_message.c_str()<<" "<<parser->getErrorMessage()<<std::endl;
    cerr<<error_message.c_str()<<" "<<parser->getErrorMessage()<<endl;
    return;
  }
  
  int bodyNum2 = parser->readBodyNum();
  if (parser->errorHappend()) {
    fout<<error_message.c_str()<<" "<<parser->getErrorMessage()<<std::endl;
    cerr<<error_message.c_str()<<" "<<parser->getErrorMessage()<<endl;
    return;
  }

  char chain2 = parser->readChar();
  if (parser->errorHappend()) {
    fout<<error_message.c_str()<<" "<<parser->getErrorMessage()<<std::endl;
    cerr<<error_message.c_str()<<" "<<parser->getErrorMessage()<<endl;
    return;
  }

  fout<<"bodyNum1 = "<<bodyNum1<<std::endl;
  fout<<"chain1   = "<<chain1<<std::endl;
  fout<<"bodyNum2 = "<<bodyNum2<<std::endl;
  fout<<"chain2   = "<<chain2<<std::endl;

  cout<<"bodyNum1 = "<<bodyNum1<<endl;
  cout<<"chain1   = "<<chain1<<endl;
  cout<<"bodyNum2 = "<<bodyNum2<<endl;
  cout<<"chain2   = "<<chain2<<endl;
}