wb-file.h

Go to the documentation of this file.

// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// Copyright 2014-2015 Tsinghua University
// Author: wb.th08@gmail.com (Bin Wang), ozj@tsinghua.edu.cn (Zhijian Ou)
//
// All h, cpp, cc, and script files (e.g. bat, sh, pl, py) should include the above
// license declaration. Different coding language may use different comment styles.




#pragma once
#include "iostream"
#include "fstream"
using namespace std;

#include <stdio.h>
#include <time.h>
#include <cstdarg>  //include va_list

#include "wb-win.h"
#include "wb-log.h"

#define SAFE_FOPEN(pfile, path, mode) {\
    if ( !path || !(pfile = fopen(path, mode)) ) {cout<<"File Open Failed: "<<path<<endl; exit(0);}}
#define SAFE_FMOPEN(file, path, mode) {\
    file.open(path, mode); if (!file.good()) {cout<<"File Open Failed: "<<path<<endl; exit(0);}}
#define SAFE_FSOPEN(file, path) {\
    file.open(path); if (!file.good()) {cout<<"File Open Failed: "<<path<<endl; exit(0);}}
#define SAFE_FCLOSE(fp) { if(fp) {fclose(fp); fp=NULL;} }

#define MAX_SENTENCE_LEN    32*1024

#define WRITE_VALUE(x,fp) fwrite(&(x),sizeof(x),1,fp)
#define READ_VALUE(x,fp) fread(&(x),sizeof(x),1,fp)
#define FPRINT_VALUE(x, fp) fprintf(fp, #x"=%d\n", (x))
#define FSCANF_VALUE(x, fp) fscanf(fp, #x"=%d\n", &(x))

namespace wb
{
    class File
    {
    public:
        FILE *fp;  
        int nLine; 

        char *pStrLine; 
        string strFileName; 

        bool bOver; 
        short nBuf; 

    public:
        File() :fp(NULL), pStrLine(NULL) { fp = NULL; bOver = false; nBuf = 1; }
        File(const char *path, const char *mode, bool bHardOpen = true) :fp(NULL), pStrLine(NULL) { Open(path, mode, bHardOpen); }
        ~File() { SAFE_FCLOSE(fp); delete[]pStrLine; }

        virtual bool Open(const char *path, const char *mode, bool bHardOpen = true);
        virtual bool Reopen(const char* model);
        virtual void Close() { SAFE_FCLOSE(fp); }
        operator FILE* () { return fp; }

        virtual char *GetLine(bool bPrecent = false);
        virtual bool GetLine(char *str, int maxLen = MAX_SENTENCE_LEN);
        virtual void Print(const char* p_pMessage, ...);
        virtual int Scanf(const char* p_pMessage, ...);
        void Flush()  { fflush(fp); }

        void Reset() { _fseeki64(fp, 0, SEEK_SET); nLine = 0; }
        bool Good() const { return !(fp == NULL); }
        template <typename TYPE>
        void PrintArray(const char* pformat, TYPE* pbuf, int num)
        {
            for (int i = 0; i<num; i++) {
                Print(pformat, *pbuf);
                putc(' ', fp);
                pbuf++;
            }
            putc('\n', fp);
        }
    };

    class IO_Obj
    {
    public:
        virtual void WriteT(File &file) {};
        virtual void ReadT(File &file) {};
        virtual void WriteB(File &file) {};
        virtual void ReadB(File &file) {};
    };

    class ObjFile : public File
    {
    public:
        int m_nTotalNum; 
        int m_nCurNum;   
        IO_Obj *m_pObj; 
    public:
        ObjFile() :File(), m_nTotalNum(0), m_nCurNum(-1){}
        ObjFile(const char *path, const char *mode) :File(path, mode), m_nTotalNum(0), m_nCurNum(-1){}
        void WriteHeadT();
        void ReadHeadT();
        void WriteObjT();
        bool ReadObjT();
        void WriteHeadB();
        void ReadHeadB();
        void WriteObjB();
        bool ReadObjB();
    };

}