src/mdParser/SimplePreprocessor.hpp

/*
 * Copyright (c) 2005 The University of Notre Dame. All Rights Reserved.
 *
 * The University of Notre Dame grants you ("Licensee") a
 * non-exclusive, royalty free, license to use, modify and
 * redistribute this software in source and binary code form, provided
 * that the following conditions are met:
 *
 * 1. Acknowledgement of the program authors must be made in any
 *    publication of scientific results based in part on use of the
 *    program.  An acceptable form of acknowledgement is citation of
 *    the article in which the program was described (Matthew
 *    A. Meineke, Charles F. Vardeman II, Teng Lin, Christopher
 *    J. Fennell and J. Daniel Gezelter, "OOPSE: An Object-Oriented
 *    Parallel Simulation Engine for Molecular Dynamics,"
 *    J. Comput. Chem. 26, pp. 252-271 (2005))
 *
 * 2. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 3. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the
 *    distribution.
 *
 * This software is provided "AS IS," without a warranty of any
 * kind. All express or implied conditions, representations and
 * warranties, including any implied warranty of merchantability,
 * fitness for a particular purpose or non-infringement, are hereby
 * excluded.  The University of Notre Dame and its licensors shall not
 * be liable for any damages suffered by licensee as a result of
 * using, modifying or distributing the software or its
 * derivatives. In no event will the University of Notre Dame or its
 * licensors be liable for any lost revenue, profit or data, or for
 * direct, indirect, special, consequential, incidental or punitive
 * damages, however caused and regardless of the theory of liability,
 * arising out of the use of or inability to use software, even if the
 * University of Notre Dame has been advised of the possibility of
 * such damages.
 */

#ifndef MDPARSER_SIMPLEPREPROCESSOR_HPP
#define MDPARSER_SIMPLEPREPROCESSOR_HPP
#include <iostream>
#include <set>
#include <fstream>
#include <sstream>
#include "utils/StringTokenizer.hpp"
#include "utils/Trim.hpp"
#include "utils/OOPSEException.hpp"
#include "utils/simError.h"


/**
 * @class SimplePreprocessor
 * @brief A simple preprocessor.
 * @note only supports #include #ifdef, #ifndef, #endif, #define and #undef, c-like multiple line
 *  comment is not supported, macro substitute is not supported.
 */
namespace oopse { 
class SimplePreprocessor {
    public:
        bool preprocess(std::istream& myStream, const std::string& filename, int startingLine, ostream& os) {
            std::set<std::string> defineSet;
            std::stack<bool> ifStates;

            ifStates.push(true);
            return doPreprocess(myStream, filename, startingLine, os, defineSet, ifStates);
        }
        
    private:
        bool doPreprocess(std::istream& myStream, const std::string& filename, int startingLine, ostream& os, std::set<std::string>& defineSet, std::stack<bool>& ifStates) {
            //std::ifstream input(filename.c_str());
            //if (!input.is_open()) {
            //    std::stringstream ss;
            //    ss << "Can not open " << filename << " for preprocessing\n";
            //    
            //    sprintf(painCave.errMsg,
            //            "Can not open (%s) for processing. \n"
            //            "\tPlease check md file name syntax.\n", filename.c_str());
            //    
            //    painCave.isFatal = 1;
            //    simError();
            //    
            //    throw OOPSEException(ss.str());                
            //}
            int lineNo = startingLine;
            os << "#line " << lineNo << " \"" << filename << "\"\n";
            const int bufferSize = 1024;
            char buffer[bufferSize];
            while(myStream.getline(buffer, bufferSize)) {
              ++lineNo;
              std::string line = trimLeftCopy(buffer);
              if (!line.empty() && line[0] == '#') {
                    StringTokenizer tokenizer(line.substr(1, line.length()));
                    std::vector<std::string> tokens = tokenizer.getAllTokens();
                    if (tokens.size() < 1 ) {
                        return false;
                    }
                    std::string keyword = tokens[0];
                    if (tokens[0] == "endif") {
                        ifStates.pop();
                        if (ifStates.empty()) {
                            std::cout << "Error in preprocessing: endif \n";
                            return false;
                        }
                        os << std::endl;                        
                    } else if (tokens.size() == 2) {
                        if (tokens[0] == "include") {
                            SimplePreprocessor subPreprocessor;
                            std::string includeFilename = tokens[1];
                            includeFilename = includeFilename.substr(1, includeFilename.length() -2);
                            std::ifstream includeStream(includeFilename.c_str());
                            if (!includeStream.is_open()) {
                                std::stringstream ss;
                                ss << "Can not open " << includeFilename << " for preprocessing\n";
                                throw OOPSEException(ss.str()); 
                            }
                            
                            bool ret = subPreprocessor.doPreprocess(includeStream, includeFilename, 1, os, defineSet, ifStates);
                            if (!ret) {
                                std::cout << "Error in preprocessing\n";
                                return false;
                            }
                            os << "#line " << lineNo << " \"" << filename << "\"\n";
                        } else if (tokens[0] == "define") {
                           defineSet.insert(tokens[1]);
                           os << std::endl;
                        } else if (tokens[0] == "undef") {
                           defineSet.erase(tokens[1]);
                           os << std::endl;
                        } else if (tokens[0] == "ifdef") {
                           if (defineSet.find(tokens[1]) != defineSet.end() ) {
                                ifStates.push(true);
                           } else {
                              ifStates.push(false);
                           }
                           os << std::endl;
                        } else if (tokens[0] == "ifndef") {
                           if (defineSet.find(tokens[1]) == defineSet.end() ) {
                                ifStates.push(true);
                           } else {
                              ifStates.push(false);
                           }
                           os << std::endl;
                        } else {
                            std::cout << tokens[0] << " is not supported (yet)." << std::endl;
                            return false;
                        }
                    }else {
                        return false;
                    }
                    
              }else if (ifStates.top()){
                os << buffer << std::endl;
              }
              
            }

            return true;
        }
    private:
        
};

}
#endif
Revision:	1245
Committed:	Tue May 27 16:39:06 2008 UTC (16 years, 11 months ago) by chuckv
File size:	7521 byte(s)
Log Message:	Checking in changes for Hefland moment calculations
#	Content
1	/*
2	* Copyright (c) 2005 The University of Notre Dame. All Rights Reserved.
3	*
4	* The University of Notre Dame grants you ("Licensee") a
5	* non-exclusive, royalty free, license to use, modify and
6	* redistribute this software in source and binary code form, provided
7	* that the following conditions are met:
8	*
9	* 1. Acknowledgement of the program authors must be made in any
10	* publication of scientific results based in part on use of the
11	* program. An acceptable form of acknowledgement is citation of
12	* the article in which the program was described (Matthew
13	* A. Meineke, Charles F. Vardeman II, Teng Lin, Christopher
14	* J. Fennell and J. Daniel Gezelter, "OOPSE: An Object-Oriented
15	* Parallel Simulation Engine for Molecular Dynamics,"
16	* J. Comput. Chem. 26, pp. 252-271 (2005))
17	*
18	* 2. Redistributions of source code must retain the above copyright
19	* notice, this list of conditions and the following disclaimer.
20	*
21	* 3. Redistributions in binary form must reproduce the above copyright
22	* notice, this list of conditions and the following disclaimer in the
23	* documentation and/or other materials provided with the
24	* distribution.
25	*
26	* This software is provided "AS IS," without a warranty of any
27	* kind. All express or implied conditions, representations and
28	* warranties, including any implied warranty of merchantability,
29	* fitness for a particular purpose or non-infringement, are hereby
30	* excluded. The University of Notre Dame and its licensors shall not
31	* be liable for any damages suffered by licensee as a result of
32	* using, modifying or distributing the software or its
33	* derivatives. In no event will the University of Notre Dame or its
34	* licensors be liable for any lost revenue, profit or data, or for
35	* direct, indirect, special, consequential, incidental or punitive
36	* damages, however caused and regardless of the theory of liability,
37	* arising out of the use of or inability to use software, even if the
38	* University of Notre Dame has been advised of the possibility of
39	* such damages.
40	*/
41
42	#ifndef MDPARSER_SIMPLEPREPROCESSOR_HPP
43	#define MDPARSER_SIMPLEPREPROCESSOR_HPP
44	#include <iostream>
45	#include <set>
46	#include <fstream>
47	#include <sstream>
48	#include "utils/StringTokenizer.hpp"
49	#include "utils/Trim.hpp"
50	#include "utils/OOPSEException.hpp"
51	#include "utils/simError.h"
52
53
54	/**
55	* @class SimplePreprocessor
56	* @brief A simple preprocessor.
57	* @note only supports #include #ifdef, #ifndef, #endif, #define and #undef, c-like multiple line
58	* comment is not supported, macro substitute is not supported.
59	*/
60	namespace oopse {
61	class SimplePreprocessor {
62	public:
63	bool preprocess(std::istream& myStream, const std::string& filename, int startingLine, ostream& os) {
64	std::set<std::string> defineSet;
65	std::stack<bool> ifStates;
66
67	ifStates.push(true);
68	return doPreprocess(myStream, filename, startingLine, os, defineSet, ifStates);
69	}
70
71	private:
72	bool doPreprocess(std::istream& myStream, const std::string& filename, int startingLine, ostream& os, std::set<std::string>& defineSet, std::stack<bool>& ifStates) {
73	//std::ifstream input(filename.c_str());
74	//if (!input.is_open()) {
75	// std::stringstream ss;
76	// ss << "Can not open " << filename << " for preprocessing\n";
77	//
78	// sprintf(painCave.errMsg,
79	// "Can not open (%s) for processing. \n"
80	// "\tPlease check md file name syntax.\n", filename.c_str());
81	//
82	// painCave.isFatal = 1;
83	// simError();
84	//
85	// throw OOPSEException(ss.str());
86	//}
87	int lineNo = startingLine;
88	os << "#line " << lineNo << " \"" << filename << "\"\n";
89	const int bufferSize = 1024;
90	char buffer[bufferSize];
91	while(myStream.getline(buffer, bufferSize)) {
92	++lineNo;
93	std::string line = trimLeftCopy(buffer);
94	if (!line.empty() && line[0] == '#') {
95	StringTokenizer tokenizer(line.substr(1, line.length()));
96	std::vector<std::string> tokens = tokenizer.getAllTokens();
97	if (tokens.size() < 1 ) {
98	return false;
99	}
100	std::string keyword = tokens[0];
101	if (tokens[0] == "endif") {
102	ifStates.pop();
103	if (ifStates.empty()) {
104	std::cout << "Error in preprocessing: endif \n";
105	return false;
106	}
107	os << std::endl;
108	} else if (tokens.size() == 2) {
109	if (tokens[0] == "include") {
110	SimplePreprocessor subPreprocessor;
111	std::string includeFilename = tokens[1];
112	includeFilename = includeFilename.substr(1, includeFilename.length() -2);
113	std::ifstream includeStream(includeFilename.c_str());
114	if (!includeStream.is_open()) {
115	std::stringstream ss;
116	ss << "Can not open " << includeFilename << " for preprocessing\n";
117	throw OOPSEException(ss.str());
118	}
119
120	bool ret = subPreprocessor.doPreprocess(includeStream, includeFilename, 1, os, defineSet, ifStates);
121	if (!ret) {
122	std::cout << "Error in preprocessing\n";
123	return false;
124	}
125	os << "#line " << lineNo << " \"" << filename << "\"\n";
126	} else if (tokens[0] == "define") {
127	defineSet.insert(tokens[1]);
128	os << std::endl;
129	} else if (tokens[0] == "undef") {
130	defineSet.erase(tokens[1]);
131	os << std::endl;
132	} else if (tokens[0] == "ifdef") {
133	if (defineSet.find(tokens[1]) != defineSet.end() ) {
134	ifStates.push(true);
135	} else {
136	ifStates.push(false);
137	}
138	os << std::endl;
139	} else if (tokens[0] == "ifndef") {
140	if (defineSet.find(tokens[1]) == defineSet.end() ) {
141	ifStates.push(true);
142	} else {
143	ifStates.push(false);
144	}
145	os << std::endl;
146	} else {
147	std::cout << tokens[0] << " is not supported (yet)." << std::endl;
148	return false;
149	}
150	}else {
151	return false;
152	}
153
154	}else if (ifStates.top()){
155	os << buffer << std::endl;
156	}
157
158	}
159
160	return true;
161	}
162	private:
163
164	};
165
166	}
167	#endif