ObjectARX. Получение/изменение из AutoLisp свойств чертежа (Summary Info) (Страница 2 из 3)

//////////////////////////////////////////////////////////////////////////
// Реализация DwgProps для AutoCAD 2004...2007 (возможно и дальше,
// если сохранится структура dwg-файла).
//////////////////////////////////////////////////////////////////////////
#include <io.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <malloc.h>
#include <stdlib.h>
#include <vector>
#include <string>
//
// Имена стандартных свойств чертежа
//
const char *stdPropNames[] = {
  "title",
  "subject",
  "author",
  "keywords",
  "comments",
  "lastSavedBy",
  "revisionNumber",
  "hyperlinkBase"
};
class DwgProps
{
  struct record {
    WORD len;
    union {
      char    c[32767];
      wchar_t w[32767];
    };
  };
public:
    DwgProps() {
      for (int i=0; i < sizeof(stdPropNames)/sizeof(stdPropNames[0]); i++)  {
        m_stdProps.push_back("");
      }
      stimes[0]=stimes[1]=stimes[2]=0;
    };
    ~DwgProps() {}
    int load(const char *fname);
    //  data items
    const char *title()          const    {return m_stdProps[0].c_str(); }
    const char *subject()        const    {return m_stdProps[1].c_str(); }
    const char *author()         const    {return m_stdProps[2].c_str(); }
    const char *keywords()       const    {return m_stdProps[3].c_str(); }
    const char *comments()       const    {return m_stdProps[4].c_str(); }
    const char *lastSavedBy()    const    {return m_stdProps[5].c_str(); }
    const char *revisionNumber() const    {return m_stdProps[6].c_str(); }
    const char *hyperlinkBase()  const    {return m_stdProps[7].c_str(); }
    const char *customName(int idx) const  {
      return ((idx >= 0 && idx < (int)m_custPropNames.size())?m_custPropNames[idx].c_str():0);
    }
    const char *customValue(int idx) const {
      return ((idx >= 0 && idx < (int)m_custPropValues.size())?m_custPropValues[idx].c_str():0);
    }
    const char *customValue(const char *name) const  {
      for (int i=0; i < (int)m_custPropNames.size(); i++) {
        if (m_custPropNames[i] == name) return m_custPropValues[i].c_str();
      }
      return 0;
    }
    int    customNumber() const { return (int) m_custPropNames.size(); }
    const __int64 inDwg()   const { return stimes[0];} // Не поддерживается
    const __int64 created() const { return stimes[1];} // Не поддерживается
    const __int64 updated() const { return stimes[2];} // Не поддерживается
private:
    //  data
    std::vector<std::string> m_stdProps;
    std::vector<std::string> m_custPropNames;
    std::vector<std::string> m_custPropValues;
    __int64 stimes[3];
};
inline int DwgProps::load(const char * fname)
{
  bool unicode = false;
  int h;
  char buf[32767]; memset(buf,0,sizeof(buf));
  if ((h = _open(fname,_O_BINARY|_O_RDONLY)) == -1)
    return 0;
  char ACADVER[7];
  if (_read(h,ACADVER,sizeof(ACADVER)) <= 0) {
    _close(h); return 0;
  }
  if (atoi(ACADVER+2) < 1018) {  // AutoCAD R14,2000...2002 обрабатывать не будем
    _close(h); return 0;
  }
  if (atoi(ACADVER+2) >= 1021) { // AutoCAD 2007... строки в Unicode
    unicode = true;
  }
  // Указатель перемещаем на WORD, содержащий смещение DwgProps
  if (_lseek(h,0x20,SEEK_SET) == -1) {
    _close(h); return 0;
  }
  // Считываем смещение DwgProps
  WORD offset_dwgprops = 0;
  if (_read(h,&offset_dwgprops,sizeof(offset_dwgprops)) <= 0) {
    _close(h); return 0;
  }
  long begin_offset = offset_dwgprops, cur_offset = offset_dwgprops;
  // Устанавливаем указатель на начало DwgProps
  if (_lseek(h,offset_dwgprops,SEEK_SET) == -1) {
    _close(h); return 0;
  }
  // Теперь можно читать длины и сами строки:
  record rec;
  for (int i=0; i < sizeof(stdPropNames)/sizeof(stdPropNames[0]); i++) {
    if (_read(h,&rec,sizeof(record)) == -1) {
      _close(h); return 0;
    }
    if (unicode) {
      WideCharToMultiByte(CP_ACP,0,rec.w,rec.len,buf,rec.len,0,NULL);
      m_stdProps[i] = buf;
    } else {
      m_stdProps[i] = rec.c;
    }
    cur_offset += (rec.len*((unicode)?2:1) + sizeof(rec.len));
    if (_lseek(h,cur_offset,SEEK_SET) == -1) {
      _close(h); return 0;
    }
  }
  // Пропускаем 24 байта - три поля по 8 байт - это даты в непонятном формате
  for (int i=0; i < 3; i++) {
    if (_read(h,&(stimes[i]),sizeof(__int64)) == -1) {
      _close(h); return 0;
    }
  }
  cur_offset += 3*sizeof(__int64);
  if (_lseek(h,cur_offset,SEEK_SET) == -1) {
    _close(h); return 0;
  }
  // Читаем количество собственных (custom) переменных
  WORD nCust = 0;
  if (_read(h,&nCust,sizeof(nCust)) <= 0) { _close(h); return 0; }
  cur_offset += sizeof(WORD);
  for (int i=0; i < nCust; i++) {
    // Считываем имя переменной
    if (_read(h,&rec,sizeof(record)) == -1) {_close(h); return 0;}
    if (unicode) {
      WideCharToMultiByte(CP_ACP,0,rec.w,rec.len,buf,rec.len,0,NULL);
      m_custPropNames.push_back(buf);
    } else {
      m_custPropNames.push_back(rec.c);
    }
    cur_offset += (rec.len*((unicode)?2:1) + sizeof(rec.len));
    if (_lseek(h,cur_offset,SEEK_SET) == -1) {
      _close(h); return 0;
    }
    if (_read(h,&rec,sizeof(record)) == -1) {_close(h); return 0;}
    if (unicode) {
      WideCharToMultiByte(CP_ACP,0,rec.w,rec.len,buf,rec.len,0,NULL);
      m_custPropValues.push_back(buf);
    } else {
      m_custPropValues.push_back(rec.c);
    }
    cur_offset += (rec.len*((unicode)?2:1) + sizeof(rec.len));
    if (_lseek(h,cur_offset,SEEK_SET) == -1) {_close(h); return 0;}
  }
  _close(h);
  return 1;
}
//-----------------------------------------
// Пример использования класса DwgProps
//-----------------------------------------
// const char * valid(const char *s) { return s ? s : ""; }
//
// void main(int argc, char *argv[])
// {
//   DwgProps p;
//
//   if ( argc < 2 )  {
//     fprintf(stderr, "Usage: pscan file [file ...]\n");
//     return;
//   }
//
//   //  loop over files
//   for ( int f = 1; f < argc; f++ )  {
//     printf("%s\n", argv[f]);
//
//     if ( p.load(argv[f]) )     {
//       //  print list of tags
//       printf("  Title:         [%s]\n", valid(p.title()));
//       printf("  Subject:       [%s]\n", valid(p.subject()));
//       printf("  Author:        [%s]\n", valid(p.author()));
//       printf("  Comments:      [%s]\n", valid(p.comments()));
//       printf("  Keywords:      [%s]\n", valid(p.keywords()));
//       printf("  HyperlinkBase: [%s]\n", valid(p.hyperlinkBase()));
//       printf("  LastSavedBy:   [%s]\n", valid(p.lastSavedBy()));
//       printf("  RevisionNo.:   [%s]\n", valid(p.revisionNumber()));
//       for ( int i = 0; i < p.customNumber(); i++ ) {
//         printf("  Custom Name[%d]:   [%s]\n", i, valid(p.customName(i)));
//         printf("  Custom Value[%d]:  [%s]\n", i, valid(p.customValue(i)));
//       }
//     }
//   }
// }

  // Считываем смещение DwgProps
  WORD offset_dwgprops = 0;

  // Считываем смещение DwgProps
  DWORD offset_dwgprops = 0;

(setdwgprops '(("Title" "Тестовый чертеж") ("Subject" "Чертеж (\"Название\")") ("Параметр" "а это его значение")) T)