mercury2pvd.cpp File Reference

#include "DPMBase.h"

Go to the source code of this file.

Classes
class	Mercury2PVD

Functions
int	main (int argc UNUSED, char *argv[] UNUSED)

Function Documentation

int main	(	int argc	UNUSED,
		char *argv[]	UNUSED
	)

Todo:

TW maybe this should be part of readRestartFile

Definition at line 37 of file mercury2pvd.cpp.

References BaseHandler< T >::clear(), Files::dataFile, FATAL, Files::fStatFile, File::getName(), Files::getName(), BaseHandler< T >::getNumberOfObjects(), DPMBase::getTime(), INFO, DPMBase::interactionHandler, logger, MULTIPLE_FILES, File::open(), DPMBase::particleHandler, DPMBase::readNextDataFile(), DPMBase::readRestartFile(), Files::restartFile, File::setCounter(), File::setName(), Files::setName(), InteractionHandler::setWriteVTK(), DPMBase::wallHandler, WARN, WallHandler::writeVTK(), and ParticleHandler::writeVTK().

{
    Mercury2PVD m;

    //Check to see if we received one argument
    if (argc == 2) {
        std::string prefix = argv[1];
        std::string::size_type pos = prefix.find(".restart");

        if (pos == std::string::npos) {
            m.setName(argv[1]);
        } else {
            m.setName(prefix.substr(0,pos-1));
            m.restartFile.setName(prefix);
            logger(INFO, "restart file % given; prefix is %",m.restartFile.getName(),m.getName());
        }

    } else {
        m.setName("../Drivers/MercuryCG/NewtonsCradleSelfTest");
        //We didn't. Print a usage and exit the program.
        logger(WARN, "Usage: mercury2pvd [prefix] \n"
         "  This program converts MercuryDPM files into ParaView data files,\n"
         "  which can then be used to visualize your particles.\n"
         "\n"
         "   prefix: Prefix to prepend to the i/o files.\n"
         "\n"
         "   The following files will be generated:\n"
         "     - prefix.pvd\n"
         "     - prefix_0.vtu\n"
         "     - prefix_1.vtu\n"
         "         ( ... )\n"
         "     - prefix_987654321.vtu\n"
         "     depending on the amount of timesteps.", argv[0]);
    }

    //Store path as it is overwritten by readRestartFile
    std::string::size_type pos = m.getName().find_last_of("/");
    std::string path = m.getName().substr(0,pos+1);
    logger(INFO,"Path %",path);

    //Read restart file, and re-add path to data and fstat file name
    logger(INFO,"Reading restart file %",m.restartFile.getName());
    bool success = m.readRestartFile();
    if (!success) logger(FATAL,"Reading restart file % failed",m.restartFile.getName());
    m.dataFile.setName(path+m.dataFile.getName());
    m.fStatFile.setName(path+m.fStatFile.getName());

    //prepare dpm class for reading in
    m.dataFile.open(std::fstream::in);
    m.fStatFile.open(std::fstream::in);
    m.interactionHandler.clear();
    m.dataFile.setCounter(0);
    m.fStatFile.setCounter(0);
    m.interactionHandler.setWriteVTK(FileType::MULTIPLE_FILES);

    //read in
    logger(INFO,"Reading data/fstat files %, %",m.dataFile.getName(), m.fStatFile.getName());
    while (m.readNextDataFile())
    {
        //m.readNextFStatFile();
        logger(INFO,"t % N % C %",m.getTime(),m.particleHandler.getNumberOfObjects(),m.interactionHandler.getNumberOfObjects());
        //m.writeOutputFiles()
        m.particleHandler.writeVTK();
        //m.interactionHandler.writeVTK();
        m.wallHandler.writeVTK();
    }
}