Global.cc

#include "Global.h"
#include "Parameters.h"
#include "Element.h"
#include "Line.h"
#include "Triangle.h"
#include "Tetrahedron.h"


#include <stdarg.h>
#include <stdio.h>

namespace AMDiS {

  const char *funcName = NULL;
  const char *Msg::oldFuncName = NULL;
  ::std::ostream* Msg::out = NULL;
  ::std::ostream* Msg::error = NULL;
  int Global::dimOfWorld = 0;
  ::std::vector< ::std::vector< int > > Global::geoIndexTable;
  int Msg::msgInfo = 10;
  bool Msg::msgWait = true;

  Element *Global::referenceElement[4] = 
    { NULL, 
      NEW Line(NULL), 
      NEW Triangle(NULL), 
      NEW Tetrahedron(NULL) 
    };

  void Msg::wait(bool w)
  {
    FUNCNAME("Msg::wait()");

    if (w) {
      char line[10];
      MSG("wait for <enter> ...");
      fgets(line, 9, stdin);
    }
    return;
  }

  void Msg::change_out(::std::ostream  *fp)
  {
    FUNCNAME("Msg::change_out()");
  
    if (fp) {
      if (out && *out != ::std::cout && *out != ::std::cerr) {
	dynamic_cast< ::std::ofstream*>(out)->close();
	delete out;
      }

      out = fp;
    } else {
      ERROR("file pointer is pointer to nil;\n");
      ERROR("use previous stream for errors furthermore\n");
    }
  
    return;
  }

  void Msg::open_file(const char *filename, OPENMODE type)
  {
    FUNCNAME("Msg::open_file()");
    ::std::ofstream *fp;

 
    if (filename  &&  (fp = new ::std::ofstream(filename, type))) {
      if (out && *out != ::std::cout && *out != ::std::cerr) {
	dynamic_cast< ::std::ofstream*>(out)->close();
	delete out;
      }
      
      out = fp;
    } else {
      if (filename)
	ERROR("can not open %s;\n", filename);
      else
	ERROR("no filename specified;\n");
    
      ERROR("use previous stream for messages furthermore\n");
    }
    return;
  }

  void Msg::change_error_out(::std::ofstream *fp)
  {
    FUNCNAME("Msg::change_error_out()");

    if (fp) {
      if (error  &&  *error != ::std::cout  &&  *error != ::std::cerr) {
	dynamic_cast< ::std::ofstream*>(error)->close();
	delete error;
      }
      
      error = fp;
    } else {
      ERROR("file pointer is pointer to nil;\n");
      ERROR("use previous stream for errors furthermore\n");
    }

    return;
  }

  void Msg::open_error_file(const char *filename, OPENMODE type)
  {
    FUNCNAME("Msg::open_error_file()");
    ::std::ofstream *fp;

    if (filename  &&  (fp = new ::std::ofstream(filename, type))) {
      if (error  &&  *error != ::std::cout  &&  *error != ::std::cerr) {
	dynamic_cast< ::std::ofstream*>(error)->close();
	delete error;
      }

      error = fp;
    } else {
      if (filename)
	ERROR("can not open %s;\n", filename);
      else
	ERROR("no filename specified;\n");
      ERROR("use previous stream for errors furthermore\n");
    }

    return;
  }

  const char *generate_filename(const char * path, const char * fn, int ntime)
  {
    static char name[256];
    char *cp;
  
    if (path == NULL || path[0] == '\0') {
      sprintf(name, "./%s", fn);
    } else {
      const char* ccp = path;
      while (*ccp)
	ccp++;
      ccp--;
      if (*ccp == '/')
	sprintf(name, "%s%s", path, fn);
      else
	sprintf(name, "%s/%s", path, fn);
    }
    cp = name;
    while (*cp)
      cp++;
    sprintf(cp, "%d", 100000 + ntime);
    if (ntime < 100000)  *cp = '0';

    return(const_cast<const char *>(name));
  }

  void Msg::print_funcname(const char *funcName)
  {
    if (!out) 
      out = &::std::cout;

    if (funcName &&  oldFuncName != funcName) {
      (*out)<< funcName << ":" << ::std::endl;
    } else if (!funcName) {
      (*out)<<"*unknown function*" << ::std::endl;
    }
    (*out) << "               ";
    oldFuncName = funcName;
  }

  void Msg::print_error_funcname(const char *funcName, 
				 const char *file, 
				 int line)
  {
    static int old_line = -1;

    if (!error) 
      error = &::std::cerr;

    if (funcName &&  oldFuncName != funcName) {
      (*error)<<funcName<< ": ";
    }
    else if (!funcName) {
      if (line-old_line > 5) 
	(*error) << "*unknown function*";
    }
    if (oldFuncName != funcName) {
      (*error) << "ERROR in " << file << ", line " << line << "\n";
      oldFuncName = funcName;
    } else if (line - old_line > 5)
      (*error) << "ERROR in " << file << ", line " << line << "\n";

    old_line = line;
  }

  void Msg::print_error_exit(const char *format, ...)
  {
    va_list arg;
    char buff[255];

    if (!error) 
      error = &::std::cerr;

    va_start(arg, format);
    vsprintf(buff, format, arg);
    (*error)<<buff;
    va_end(arg);

    exit(1);
  }

  void Msg::print_error(const char *format, ...)
  {
    va_list arg;
    char buff[255];


    if (!error) 
      error = &::std::cerr;

    va_start(arg, format);
    vsprintf(buff, format, arg);
    (*error)<<buff;
    va_end(arg);

    return;
  }

  void Msg::print_warn_funcname(const char *funcName,
				const char *file, 
				int line)
  {
    static int old_line = -1;

    if (!out) 
      out = &::std::cout;

    if (funcName  &&  oldFuncName != funcName) {
      (*out) << funcName << ": ";
    }
    else if (!funcName){
      (*out) << "*unknown function*";
    }
    if (oldFuncName != funcName) {
      (*out) << "WARNING in " << file << ", line " << line << "\n";
      oldFuncName = funcName;
    } else if (line - old_line > 5)
      (*out) << "WARNING in " << file << ", line " << line << "\n";

    old_line = line;
  }

  void Msg::print_warn(const char *format, ...)
  {
    va_list arg;
    char buff[255];

    if (!out) 
      out = &::std::cout;

    va_start(arg, format);
    vsprintf(buff, format, arg);
    (*out)<<buff;
    va_end(arg);

    return;
  }


  void Msg::print(const char *format, ...)
  {
    va_list arg;
    char buff[255];

    if (!out) 
      out = &::std::cout;

    va_start(arg, format);
    vsprintf(buff, format, arg);
    (*out) << buff;
    va_end(arg);
  }

  void Global::init()
  {
    int d = -1;

    // get dimension
    TEST_EXIT(Parameters::initialized())("Parameters not initialized!\n");
    Parameters::getGlobalParameter(0, "dimension of world","%d",&d);
    TEST_EXIT(d > 0)("Cannot initialize dimension!\n");
    TEST_EXIT((d == 1) || (d == 2) || (d == 3))("Invalid world dimension %d!\n",d);

    // set dimension
    dimOfWorld = d;

    // prepare geoIndex-Table
    int geoTableSize = abs(static_cast<int>(MINPART)) + MAXPART + 1;
    geoIndexTable.resize(4);
    for (int i = 0; i < 4; i++) {
      geoIndexTable[i].resize(geoTableSize);
      for (int j = 0; j < geoTableSize; j++) {
	geoIndexTable[i][j] = 0;
      }
    }

    geoIndexTable[0][PARTS - MINPART] = 1;
    geoIndexTable[0][VERTEX - MINPART] = 1;
    geoIndexTable[0][EDGE - MINPART] = 0;
    geoIndexTable[0][FACE - MINPART] = 0;
    geoIndexTable[0][WORLD - MINPART] = dimOfWorld;

    for (int i = 1; i < 4; i++) {
      geoIndexTable[i][CENTER - MINPART] = referenceElement[i]->getGeo(CENTER);
      geoIndexTable[i][VERTEX - MINPART] = referenceElement[i]->getGeo(VERTEX);
      geoIndexTable[i][EDGE - MINPART] = referenceElement[i]->getGeo(EDGE);
      geoIndexTable[i][FACE - MINPART] = referenceElement[i]->getGeo(FACE);
      geoIndexTable[i][DIMEN - MINPART] = referenceElement[i]->getGeo(DIMEN);
      geoIndexTable[i][PARTS - MINPART] = referenceElement[i]->getGeo(PARTS);
      geoIndexTable[i][NEIGH - MINPART] = referenceElement[i]->getGeo(NEIGH);
      geoIndexTable[i][WORLD - MINPART] = dimOfWorld;
      geoIndexTable[i][BOUNDARY - MINPART] = referenceElement[i]->getGeo(BOUNDARY);
      geoIndexTable[i][PROJECTION - MINPART] = referenceElement[i]->getGeo(PROJECTION);
    }

    // set msgWait
    Parameters::getGlobalParameter(0, "WAIT", "%d", &d);  
    Msg::setMsgWait(!(d == 0));
  };

  int fac(int i)
  {
    if (i <= 1) 
      return 1;
    else 
      return i * fac(i - 1);
  }

  ::std::string memSizeStr(int size)
  {
    ::std::string result;
//     if (size > 1024) {
//       if (size > 1024 * 1024) {
// 	result << size / (1024 * 1024) << " MByte";
//       } else {
// 	result << size / 1024 << " KByte";
//       }
//     } else {
//       result << size << " Byte";
//     }

    return result;
  }
}