---

DAGDBFile.h

Go to the documentation of this file.

#ifndef __DBFile_H__
#define __DBFile_H__

#include <fstream.h>
#include "String.h"
#include "SmartArray.h"
#include "DAG.h"

#define GROW_FACTOR     20

using namespace std;

class DAGDBFile : public fstream
{
        struct DAGINFO
        {
                int nType;
                streampos nOffset;

                friend istream& operator>>(istream& is, DAGINFO& x)       
                { is.read((char*)&x, sizeof(x)); }

                friend ostream& operator<<(ostream& os, const DAGINFO& x)
                { os.write((char*)&x, sizeof(x)); }
        };

        class DBINFO
        {
                int m_nErrorCheck;

        public:
                int m_nMaxTSVDimension;
                streampos m_nTailOffset; // Leave it as the last field if possible.

                DBINFO() { Clear(); }

                void Clear()
                {
                        m_nMaxTSVDimension = 0;
                        m_nTailOffset = 0;
                        m_nErrorCheck = -1;
                }

                bool IsValid() 
                { 
                        return m_nErrorCheck == (m_nMaxTSVDimension + m_nTailOffset);
                }

                void MakeValid()
                {
                        m_nErrorCheck = m_nMaxTSVDimension + m_nTailOffset;
                }
        };

        int m_nDBSize;
        String m_strFileName;

        SmartArray<DAGINFO> m_dagInfo;
        DBINFO* m_pDBInfo;
        bool* m_pbModified;

public:
        DAGDBFile() : m_dagInfo(0, GROW_FACTOR)
        {
                m_nDBSize = 0;
                m_pbModified = new bool(false);
                m_pDBInfo = new DBINFO;
        }

        ~DAGDBFile() { close(); delete m_pDBInfo; delete m_pbModified; }

        int GetDAGCount() const { return m_dagInfo.GetSize(); }
        String GetName() const { return m_strFileName; }
        int GetClassID(const String& strClassName) const;
        DAG* CreateDAGObj(int nClassID) const;

        void MoveTo(std::streamoff off) { clear(); seekg(off, ios::beg);        }
        void MoveToBeg() { MoveTo(0); }
        void MoveToEnd() { clear(); seekg(0, ios::end); }

        void open(const char* szName, ios_base::openmode nMode);

        void close();

        bool ReadTailInfo();
        void WriteTailInfo();
        DAGPtr ReadDAG(int nDagId, bool bOnlyDataForMatching = false);
        DAGPtr ReadDAG(std::streamoff nDagOffset, const String& strClassName);
        bool AddDAG(const DAG* pDag, bool bReadSaveTailInfo = false);
        bool RebuildTailInfo();
        
        bool IsDBModified() const { return *m_pbModified; }
        bool SetDBModified(bool bNewVal) { return *m_pbModified = bNewVal; }
        
        const DBINFO& DBInfo() const { return *m_pDBInfo; }
        DBINFO& DBInfo() { return *m_pDBInfo; }

        int GetMaxTSVDimension() const { return DBInfo().m_nMaxTSVDimension; }
};

#endif //__DBFile_H__

---