1 // Common/StdInStream.cpp

     2 

     3 #include "StdAfx.h"

     4 

     5 #include <tchar.h>

     6 #include "StdInStream.h"

     7 

     8 #ifdef _MSC_VER

     9 // "was declared deprecated" disabling

    10 #pragma warning(disable : 4996 )

    11 #endif

    12 

    13 static const char kIllegalChar = '\0';

    14 static const char kNewLineChar = '\n';

    15 

    16 static const char *kEOFMessage = "Unexpected end of input stream";

    17 static const char *kReadErrorMessage  ="Error reading input stream"; 

    18 static const char *kIllegalCharMessage = "Illegal character in input stream";

    19 

    20 static LPCTSTR kFileOpenMode = TEXT("r");

    21 

    22 CStdInStream g_StdIn(stdin);

    23 

    24 bool CStdInStream::Open(LPCTSTR fileName)

    25 {

    26   Close();

    27   _stream = _tfopen(fileName, kFileOpenMode);

    28   _streamIsOpen = (_stream != 0);

    29   return _streamIsOpen;

    30 }

    31 

    32 bool CStdInStream::Close()

    33 {

    34   if(!_streamIsOpen)

    35     return true;

    36   _streamIsOpen = (fclose(_stream) != 0);

    37   return !_streamIsOpen;

    38 }

    39 

    40 CStdInStream::~CStdInStream()

    41 {

    42   Close();

    43 }

    44 

    45 AString CStdInStream::ScanStringUntilNewLine()

    46 {

    47   AString s;

    48   for (;;)

    49   {

    50     int intChar = GetChar();

    51     if(intChar == EOF)

    52       throw kEOFMessage;

    53     char c = char(intChar);

    54     if (c == kIllegalChar)

    55       throw kIllegalCharMessage;

    56     if(c == kNewLineChar)

    57       break;

    58     s += c;

    59   }

    60   return s;

    61 }

    62 

    63 void CStdInStream::ReadToString(AString &resultString)

    64 {

    65   resultString.Empty();

    66   int c;

    67   while((c = GetChar()) != EOF)

    68     resultString += char(c);

    69 }

    70 

    71 bool CStdInStream::Eof()

    72 {

    73   return (feof(_stream) != 0);

    74 }

    75 

    76 int CStdInStream::GetChar()

    77 {

    78   int c = fgetc(_stream); // getc() doesn't work in BeOS?

    79   if(c == EOF && !Eof())

    80     throw kReadErrorMessage;

    81   return c;

    82 }

    83 

    84