core/main.c

Go to the documentation of this file.

#define _MAIN_C

#include <stdlib.h>
#include <string.h>
#ifdef WIN32
#include <direct.h>
#include <process.h>
#else
#include <unistd.h>
#endif

#include "globals.h"
#include "legal.h"
#include "cmdarg.h"
#include "module.h"
#include "output.h"
#include "environment.h"
#include "test.h"
#include "random.h"
#include "realtime.h"
#include "save.h"
#include "local.h"
#include "exec.h"
#include "kml.h"

#if defined WIN32 && _DEBUG 

void pause_at_exit(void) 
{
    if (global_pauseatexit)
        system("pause");
}
#endif

void delete_pidfile(void)
{
    unlink(global_pidfile);
}

int main(int argc, 
         char *argv[]) 
{
#if defined WIN32 && _DEBUG 
    atexit(pause_at_exit);
#endif
    /* main initialization */
    if (!exec_init())
        exit(6);

    /* process command line arguments */
    if (cmdarg_load(argc,argv)==FAILED)
    {
        output_fatal("shutdown after command line rejected");
        exit(1);
    }

    /* pidfile */
    if (strcmp(global_pidfile,"")!=0)
    {
        FILE *fp = fopen(global_pidfile,"w");
        if (fp==NULL)
        {
            output_fatal("unable to create pidfile '%s'", global_pidfile);
            exit(1);
        }
#ifdef WIN32
#define getpid _getpid
#endif
        fprintf(fp,"%d\n",getpid());
        output_verbose("process id %d written to %s", getpid(), global_pidfile);
        fclose(fp);
        atexit(delete_pidfile);
    }

    /* do legal stuff */
    if (strcmp(global_pidfile,"")==0 && legal_notice()==FAILED)
        exit(4);

    /* set up the test */
    if (global_test_mode)
    {
#ifndef _NO_CPPUNIT
        output_message("Entering test mode");
        if (test_start(argc,argv)==FAILED)
        {
            output_fatal("shutdown after startup test failed");
            exit(3);
        }
        exit(0); /* There is no environment to speak of, so exit. */
#else
        output_message("Unit Tests not enabled.  Recompile with _NO_CPPUNIT unset");
#endif
    }
    
    /* start the processing environment */
    output_verbose("starting up %s environment", global_environment);
    if (environment_start(argc,argv)==FAILED)
    {
        output_fatal("environment startup failed: %s", strerror(errno));
        exit(2);
    }

    /* post process the test */
    if (global_test_mode)
    {
#ifndef _NO_CPPUNIT
        output_message("Exiting test mode");
        if (test_end(argc,argv)==FAILED)
        {
            output_error("shutdown after end test failed");
            exit(3);
        }
#endif
    }

    /* save the model */
    if (strcmp(global_savefile,"")!=0)
    {
        if (saveall(global_savefile)==FAILED)
            output_error("save to '%s' failed", global_savefile);
    }

    /* do module dumps */
    if (global_dumpall!=FALSE)
    {
        output_verbose("dumping module data");
        module_dumpall();
    }

    /* KML output */
    if (strcmp(global_kmlfile,"")!=0)
        kml_dump(global_kmlfile);

    /* wrap up */
    output_verbose("shutdown complete");

    /* profile results */
    if (global_profiler)
        class_profiles();

    /* restore locale */
    locale_pop();

    /* compute elapsed runtime */
    output_verbose("elapsed runtime %d seconds", realtime_runtime());

    exit(0);
}

---

GridLAB-D^TM Version 1.0

An open-source project initiated by the US Department of Energy