ragel/inputdata.cpp

   1 /*
   2  *  Copyright 2008 Adrian Thurston <thurston@complang.org>
   3  */
   4
   5 /*  This file is part of Ragel.
   6  *
   7  *  Ragel is free software; you can redistribute it and/or modify
   8  *  it under the terms of the GNU General Public License as published by
   9  *  the Free Software Foundation; either version 2 of the License, or
  10  *  (at your option) any later version.
  11  *
  12  *  Ragel is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  *  GNU General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU General Public License
  18  *  along with Ragel; if not, write to the Free Software
  19  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20  */
  21
  22 #include "ragel.h"
  23 #include "common.h"
  24 #include "inputdata.h"
  25 #include "parsedata.h"
  26 #include "rlparse.h"
  27 #include <iostream>
  28
  29 using std::cout;
  30 using std::cerr;
  31 using std::endl;
  32 using std::ios;
  33
  34 void InputData::generateSpecificReduced()
  35 {
  36         if ( parserDict.length() > 0 ) {
  37                 /* There is either a machine spec or machine name given. */
  38                 ParseData *parseData = 0;
  39                 GraphDictEl *graphDictEl = 0;
  40
  41                 /* Traverse the sections, break out when we find a section/machine
  42                  * that matches the one specified. */
  43                 for ( ParserDict::Iter parser = parserDict; parser.lte(); parser++ ) {
  44                         ParseData *checkPd = parser->value->pd;
  45                         if ( machineSpec == 0 || strcmp( checkPd->sectionName, machineSpec ) == 0 ) {
  46                                 GraphDictEl *checkGdEl = 0;
  47                                 if ( machineName == 0 || (checkGdEl =
  48                                                 checkPd->graphDict.find( machineName )) != 0 )
  49                                 {
  50                                         /* Have a machine spec and/or machine name that matches
  51                                          * the -M/-S options. */
  52                                         parseData = checkPd;
  53                                         graphDictEl = checkGdEl;
  54                                         break;
  55                                 }
  56                         }
  57                 }
  58
  59                 if ( parseData == 0 )
  60                         error() << "could not locate machine specified with -S and/or -M" << endl;
  61                 else {
  62                         /* Section/Machine to emit was found. Prepare and emit it. */
  63                         parseData->prepareMachineGen( graphDictEl );
  64                         if ( gblErrorCount == 0 )
  65                                 parseData->generateReduced( *this );
  66                 }
  67         }
  68
  69         writeOutput();
  70 }
  71
  72
  73 /* Invoked by the parser when the root element is opened. */
  74 void InputData::cdDefaultFileName( const char *inputFile )
  75 {
  76         /* If the output format is code and no output file name is given, then
  77          * make a default. */
  78         if ( outputFileName == 0 ) {
  79                 const char *ext = findFileExtension( inputFile );
  80                 if ( ext != 0 && strcmp( ext, ".rh" ) == 0 )
  81                         outputFileName = fileNameFromStem( inputFile, ".h" );
  82                 else {
  83                         const char *defExtension = 0;
  84                         switch ( hostLang->lang ) {
  85                                 case HostLang::C: defExtension = ".c"; break;
  86                                 case HostLang::D: defExtension = ".d"; break;
  87                                 default: break;
  88                         }
  89                         outputFileName = fileNameFromStem( inputFile, defExtension );
  90                 }
  91         }
  92 }
  93
  94 /* Invoked by the parser when the root element is opened. */
  95 void InputData::javaDefaultFileName( const char *inputFile )
  96 {
  97         /* If the output format is code and no output file name is given, then
  98          * make a default. */
  99         if ( outputFileName == 0 )
 100                 outputFileName = fileNameFromStem( inputFile, ".java" );
 101 }
 102
 103 /* Invoked by the parser when the root element is opened. */
 104 void InputData::rubyDefaultFileName( const char *inputFile )
 105 {
 106         /* If the output format is code and no output file name is given, then
 107          * make a default. */
 108         if ( outputFileName == 0 )
 109                 outputFileName = fileNameFromStem( inputFile, ".rb" );
 110 }
 111
 112 /* Invoked by the parser when the root element is opened. */
 113 void InputData::csharpDefaultFileName( const char *inputFile )
 114 {
 115         /* If the output format is code and no output file name is given, then
 116          * make a default. */
 117         if ( outputFileName == 0 ) {
 118                 const char *ext = findFileExtension( inputFile );
 119                 if ( ext != 0 && strcmp( ext, ".rh" ) == 0 )
 120                         outputFileName = fileNameFromStem( inputFile, ".h" );
 121                 else
 122                         outputFileName = fileNameFromStem( inputFile, ".cs" );
 123         }
 124 }
 125
 126 void InputData::openOutput()
 127 {
 128         if ( ! generateDot ) {
 129                 switch ( hostLang->lang ) {
 130                         case HostLang::C:
 131                         case HostLang::D:
 132                                 cdDefaultFileName( inputFileName );
 133                                 break;
 134                         case HostLang::Java:
 135                                 javaDefaultFileName( inputFileName );
 136                                 break;
 137                         case HostLang::Ruby:
 138                                 rubyDefaultFileName( inputFileName );
 139                                 break;
 140                         case HostLang::CSharp:
 141                                 csharpDefaultFileName( inputFileName );
 142                                 break;
 143                 }
 144         }
 145
 146         /* Make sure we are not writing to the same file as the input file. */
 147         if ( outputFileName != 0 ) {
 148                 if ( strcmp( inputFileName, outputFileName  ) == 0 ) {
 149                         error() << "output file \"" << outputFileName  <<
 150                                         "\" is the same as the input file" << endl;
 151                 }
 152
 153                 /* Create the filter on the output and open it. */
 154                 outFilter = new output_filter( outputFileName );
 155
 156                 /* Open the output stream, attaching it to the filter. */
 157                 outStream = new ostream( outFilter );
 158         }
 159         else {
 160                 /* Writing out ot std out. */
 161                 outStream = &cout;
 162         }
 163 }
 164
 165 void InputData::openOutput2()
 166 {
 167         if ( outFilter != 0 ) {
 168                 outFilter->open( outputFileName, ios::out|ios::trunc );
 169                 if ( !outFilter->is_open() ) {
 170                         error() << "error opening " << outputFileName << " for writing" << endl;
 171                         exit(1);
 172                 }
 173         }
 174 }
 175
 176 void InputData::prepareMachineGen()
 177 {
 178         /* No machine spec or machine name given. Generate everything. */
 179         for ( ParserDict::Iter parser = parserDict; parser.lte(); parser++ ) {
 180                 ParseData *pd = parser->value->pd;
 181                 if ( pd->instanceList.length() > 0 )
 182                         pd->prepareMachineGen( 0 );
 183         }
 184 }
 185
 186 void InputData::generateReduced()
 187 {
 188         for ( ParserDict::Iter parser = parserDict; parser.lte(); parser++ ) {
 189                 ParseData *pd = parser->value->pd;
 190                 if ( pd->instanceList.length() > 0 )
 191                         pd->generateReduced( *this );
 192         }
 193 }
 194
 195 void InputData::writeOutput()
 196 {
 197         for ( InputItemList::Iter ii = inputItems; ii.lte(); ii++ ) {
 198                 if ( ii->type == InputItem::Write ) {
 199                         CodeGenData *cgd = ii->pd->cgd;
 200                         ::keyOps = &cgd->thisKeyOps;
 201
 202                         cgd->writeStatement( ii->loc, ii->writeArgs.length()-1, ii->writeArgs.data );
 203                 }
 204                 else /*if ( /!generateDot )*/ {
 205                         *outStream << '\n';
 206                         lineDirective( *outStream, inputFileName, ii->loc.line );
 207                         *outStream << ii->data.str();
 208                 }
 209         }
 210 }
 211