include/mcld/Script/ScriptScanner.h

//===- ScriptScanner.h ----------------------------------------------------===//
//
//                     The MCLinker Project
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#ifndef MCLD_SCRIPT_SCRIPTSCANNER_H
#define MCLD_SCRIPT_SCRIPTSCANNER_H

#ifndef __FLEX_LEXER_H
#include "FlexLexer.h"
#endif

#ifndef YY_DECL
#define YY_DECL                                                       \
  mcld::ScriptParser::token_type                                      \
  mcld::ScriptScanner::lex(mcld::ScriptParser::semantic_type* yylval, \
                           mcld::ScriptParser::location_type* yylloc, \
                           const mcld::ScriptFile& pScriptFile)
#endif

#include <mcld/Script/ScriptFile.h>
#include "ScriptParser.h"
#include <stack>

namespace mcld {

/** \class ScriptScanner
 *
 */
class ScriptScanner : public yyFlexLexer
{
public:
  ScriptScanner(std::istream* yyin = NULL, std::ostream* yyout = NULL);

  virtual ~ScriptScanner();

  virtual ScriptParser::token_type lex(ScriptParser::semantic_type* yylval,
                                       ScriptParser::location_type* yylloc,
                                       const ScriptFile& pScriptFile);

  void setLexState(ScriptFile::Kind pKind);

  void popLexState();

private:
  void enterComments(ScriptParser::location_type& pLocation);

private:
  ScriptFile::Kind m_Kind;
  std::stack<ScriptFile::Kind> m_StateStack;
};

} // namespace of mcld

#endif