CifSchemaMap.h

Go to the documentation of this file.

/*$$FILE$$*/
/*$$VERSION$$*/
/*$$DATE$$*/
/*$$LICENSE$$*/


#ifndef CIFSCHEMAMAP_H
#define CIFSCHEMAMAP_H


#include <string>
#include <ostream>
#include <fstream>

#include "CifFile.h"
#include "SchemaMap.h"
#include "Db.h"
#include "DbOutput.h"



class DbOracle : public Db
{

  // Oracle 8.1.6 SQLLDR

  public:
    DbOracle(SchemaMap& schemaMapping,
      const string& dbName = DB_DEFAULT_NAME);
    ~DbOracle();

    void WriteSchemaStart(std::ostream& io);

    void WriteLoadingStart(std::ostream& io);
    void WriteLoadingTable(std::ostream& io, const string& tableName,
      const string& path);

    void GetDate(string& dType);
    void GetText(string& dType, const unsigned int width);

    void WriteNull(std::ostream& io, const int iNull,
      const unsigned int curr, const unsigned int attSize);
    void WriteTableIndex(std::ostream& io, const string& tableNameDb,
      const vector<string>& indexList);

    void WriteNewLine(std::ostream& io, bool special = false);
};


class DbDb2 : public Db
{

  public:
    DbDb2(SchemaMap& schemaMapping,
      const string& dbName = DB_DEFAULT_NAME);
    ~DbDb2();


    void GetStart(string& start);

    void WriteSchemaStart(std::ostream& io);

    void WriteLoadingStart(std::ostream& io);
    void WriteLoadingEnd(std::ostream& io);
    void WriteLoadingTable(std::ostream& io, const string& tableName,
      const string& path);

    void GetFloat(string& dType);
    void GetDate(string& dType);
    void GetText(string& dType, const unsigned int width);
    void WriteNull(std::ostream& io, const int iNull,
      const unsigned int curr, const unsigned int attSize);

    void WriteTableIndex(std::ostream& io, const string& tableNameDb,
      const vector<string>& indexList);

    void WriteBcpDoubleQuotes(std::ostream& io);
};


class DbMySql : public Db
{

  public:
    DbMySql(SchemaMap& schemaMapping,
      const string& dbName = DB_DEFAULT_NAME);
    ~DbMySql();

    void GetStart(string& start);

    void DropTableSql(std::ostream& io, const string& tableNameDb);

    void WriteLoad(std::ostream& io);
    void WriteLoadingStart(std::ostream& io);
    void WriteLoadingTable(std::ostream& io, const string& tableName,
      const string& path);

    void WriteTableIndex(std::ostream& io, const string& tableNameDb,
      const vector<string>& indexList);

    void WriteNull(std::ostream& io, const int iNull,
      const unsigned int curr, const unsigned int attSize);

    void WriteNewLine(std::ostream& io, bool special = false);

  private:
    static const string _SQL_LOADING_FILE;

};


class DbSybase : public Db
{
  public:
    DbSybase(SchemaMap& schemaMapping,
      const string& dbName = DB_DEFAULT_NAME);
    ~DbSybase();

    void GetStart(string& start);

    void WriteLoadingStart(std::ostream& io);
    void WriteLoadingTable(std::ostream& io, const string& tableName,
      const string& path);

    void WritePrint(std::ostream& io, const string& tableNameDb);
    void WriteNull(std::ostream& io, const int iNull,
      const unsigned int curr, const unsigned int attSize);

    void WriteTableIndex(std::ostream& io, const string& tableNameDb,
      const vector<string>& indexList);

    void GetDateAndTime(string& dateAndTime);

#ifdef VLAD_DATE_OBSOLETE
    void ConvertDate(string& dbDate, const string& cifDate);
#endif
};


class BcpOutput : public DbOutput
{
  public:
    BcpOutput(Db& db);
    virtual ~BcpOutput();

    void WriteDataLoadingScripts(const string& path = std::string());
    void WriteData(Block& block, const string& path = std::string());

  private:
    static const string _DATA_DELETE_FILE;

    void WriteDataLoadingScript(const string& path);
    void WriteDataLoadingFile(const string& path = std::string());

    void WriteDelete(std::ostream& io);

    void WriteEmptyString(std::ostream& io);

    void WriteSpecialDateChar(std::ostream& io, const char& specDateChar);
};
 

class SqlOutput : public DbOutput
{

  public:
    SqlOutput(Db& db);
    virtual ~SqlOutput();

    void WriteSchema(const string& path = std::string());
    void WriteDataLoadingScripts(const string& path = std::string());
    void WriteData(Block& block, const string& path = std::string());

  protected:
    void WriteEmptyNumeric(std::ostream& io);
    bool IsFirstTextNewLineSpecial();
    void WriteNewLine(std::ostream& io, bool special = false);
    void GetTableStart(string& tableStart, const string& tableName);
    void GetTableEnd(string& tableEnd);

  private:
    static const unsigned int _MAX_SQL_NAME_LENGTH = 60;
    static const string _SCHEMA_LOADING_SCRIPT;
    static const string _SCHEMA_DELETE_FILE;
    static const string _DATA_FILE;

    void WriteSqlScriptSchemaInfo(std::ostream& io);
    void WriteDataLoadingScript(const string& path);

    void CreateTableSql(std::ostream& io, const string& tableName);

    void WriteAuxTables(std::ostream& io, ISTable* infoP,
      const vector<string>& tableNames);

    void WriteEmptyString(std::ostream& io);
    void WriteEmptyDate(std::ostream& io);
};


class DbLoader
{
  public:
    enum eConvOpt
    {
        // Generate only data loading files
        eDATA_ONLY = 0,

        // Generate both data loading files and loading shell scripts
        eDATA_WITH_SCRIPTS,

        // Generate only loading shell scripts
        eSCRIPTS_ONLY
    };

    DbLoader(SchemaMap& schemaMapping, DbOutput& dbOutput,
      bool verbose = false, const string& workDir = std::string());

    virtual ~DbLoader();

    void SetWorkDir(const string& workDir);

    void AsciiFileToDb(const string& asciiFile, const eConvOpt convOpt);

    void SerFileToDb(const string& serFile, const eConvOpt convOpt);

    void FileObjToDb(CifFile& cifFile, const eConvOpt convOpt);

#ifdef DB_HASH_ID
    void SetHashMode(int mode);
#endif

  private:
    static const string _LOG_FILE;

    string _workDir; // Working directory for all generated files.
    string _INPUT_FILE;

#ifdef DB_HASH_ID
    string _HASH_ID;
    int _hashMode;
#endif

    // Block name of loadable data in mmCIF format
    string _blockName;

    bool _verbose;

    std::ofstream _log;

    SchemaMap& _schemaMapping;
    DbOutput& _dbOutput;

    void _LoadBlock(Block& rBlock, Block& wBlock);

    bool _Search(vector<vector<string> >& dMap, const unsigned int iAttr,
      ISTable* isTableP, const string& blockName,
      const vector<string>& cNameMap, const string& sItem,
      const string& sCnd, const string& sFnct);
 
    void _DoFunc(vector<string>& s, const vector<string>& r,
      const string& sFnct);

    void _OpenLog(const string& logName);

    int _GetMapColumnIndex(const vector<string>& cNameMap, const string& vOf);
    void _GetMapColumnValue(string& p, vector<string>& dMapVec,
      unsigned int irow);

    void CreateTables(CifFile& writeCifFile, CifFile& readCifFile);

#ifdef VLAD_DATE_OBSOLETE
    void _dformat_1(const char *date, char *odate);
    void _dformat_2(const char *date, char *odate);
    void _dformat_3(const char *date, char *odate, int shortFlag);
    void _dformat_4(const char *date, char *odate);
    void _dformat_5(const char *date, char *odate);
#endif

    void _ReorderName(string& res, char *string, int mode);
    void _ToUpperString(string& aString);
    void _StripString(string& aString, int mode);

#ifdef DB_HASH_ID
    long long pdbIdHash(const string& id);
#endif

    static void CleanString(string& aString);

    void Clear();
};


#endif

---