| 1 | // This may look like C code, but it is really -*- C++ -*-
 | 
|---|
| 2 | 
 | 
|---|
| 3 | #ifndef PPERSIST_H_SEEN
 | 
|---|
| 4 | #define PPERSIST_H_SEEN
 | 
|---|
| 5 | 
 | 
|---|
| 6 | // Flat file persistance, similar to Java serialization
 | 
|---|
| 7 | //
 | 
|---|
| 8 | // E. Aubourg     CEA DAPNIA/SPP  1999
 | 
|---|
| 9 | 
 | 
|---|
| 10 | 
 | 
|---|
| 11 | #include "machdefs.h"
 | 
|---|
| 12 | #include "pexceptions.h"
 | 
|---|
| 13 | #include "md5.h"
 | 
|---|
| 14 | 
 | 
|---|
| 15 | #include <time.h>
 | 
|---|
| 16 | 
 | 
|---|
| 17 | #include <string>
 | 
|---|
| 18 | #include <map>
 | 
|---|
| 19 | #include <vector>
 | 
|---|
| 20 | #include <typeinfo>
 | 
|---|
| 21 | 
 | 
|---|
| 22 | 
 | 
|---|
| 23 | // Classe de base pour les objets qui peuvent devenir PPersist
 | 
|---|
| 24 | 
 | 
|---|
| 25 | namespace SOPHYA {
 | 
|---|
| 26 |   
 | 
|---|
| 27 |   class AnyDataObj;
 | 
|---|
| 28 | 
 | 
|---|
| 29 |   class PIOPersist;
 | 
|---|
| 30 |   class PInPersist;
 | 
|---|
| 31 |   class POutPersist;
 | 
|---|
| 32 |   class PPersist;
 | 
|---|
| 33 | 
 | 
|---|
| 34 |   /* Persistant (delegate or mixin) object */
 | 
|---|
| 35 | 
 | 
|---|
| 36 |   class PPersist  {
 | 
|---|
| 37 |   public:
 | 
|---|
| 38 |     virtual           ~PPersist() {}
 | 
|---|
| 39 | // J'ajoute cette fonction pour assurer la compatibilite 
 | 
|---|
| 40 | // avec l'ancien PPersist d'Eros     (Reza 23/04/99)
 | 
|---|
| 41 |     virtual int_4      ClassId() const { return(0); }
 | 
|---|
| 42 | 
 | 
|---|
| 43 |     void               Write(string const& fn) const;
 | 
|---|
| 44 |     void               Read(string const& fn);
 | 
|---|
| 45 | 
 | 
|---|
| 46 |     virtual void       Write(POutPersist&) const;
 | 
|---|
| 47 |     void               Read(PInPersist& s);               // Reads the type tag and the object
 | 
|---|
| 48 |     void               Write(POutPersist&, string const& tag) const; 
 | 
|---|
| 49 |     void               ReadAtTag(PInPersist& s, string const& tag);
 | 
|---|
| 50 | 
 | 
|---|
| 51 |     virtual AnyDataObj* DataObj()       // Retourne l'objet reel $CHECK$ - Reza 
 | 
|---|
| 52 |                   { return(NULL); }     // Devrait etre virtuelle pure
 | 
|---|
| 53 |   protected:  
 | 
|---|
| 54 |     virtual void       ReadSelf(PInPersist&)=0;           
 | 
|---|
| 55 |     virtual void       WriteSelf(POutPersist&) const=0;  
 | 
|---|
| 56 | 
 | 
|---|
| 57 |     friend class       PInPersist;
 | 
|---|
| 58 |     friend class       POutPersist;
 | 
|---|
| 59 |   };
 | 
|---|
| 60 | 
 | 
|---|
| 61 | 
 | 
|---|
| 62 | 
 | 
|---|
| 63 |   // Ancestor for PInPersist and POutPersist
 | 
|---|
| 64 |   // Handles (statically) the registration of classes.
 | 
|---|
| 65 | 
 | 
|---|
| 66 |   class PIOPersist {
 | 
|---|
| 67 |   public:
 | 
|---|
| 68 |     enum                         {PPS_NATIVE = -1, PPS_LITTLE_ENDIAN = 0, PPS_BIG_ENDIAN = 1};
 | 
|---|
| 69 |     typedef PPersist*            (*ClassCreatorFunc)();
 | 
|---|
| 70 |   
 | 
|---|
| 71 |     static void                  RegisterClass(uint_8 classId, string typname, ClassCreatorFunc f);
 | 
|---|
| 72 |     static ClassCreatorFunc      FindCreatorFunc(uint_8 classId);
 | 
|---|
| 73 |     static uint_8 Hash(string const& typname) {
 | 
|---|
| 74 |         MD5Init(&ctx);
 | 
|---|
| 75 |         MD5Update(&ctx, (unsigned char*) typname.c_str(), typname.size());
 | 
|---|
| 76 |         MD5Final(&ctx);
 | 
|---|
| 77 |         return ( *((uint_8*) ctx.digest) + *((uint_8*) (ctx.digest+8)));
 | 
|---|
| 78 |     }
 | 
|---|
| 79 |     static MD5_CTX ctx;
 | 
|---|
| 80 |   
 | 
|---|
| 81 |     static void Initialize();   // Pour initialiser classList
 | 
|---|
| 82 |     static uint_8 getTypeId(string s) {return (*typeids)[s];}
 | 
|---|
| 83 |   private:
 | 
|---|
| 84 | 
 | 
|---|
| 85 |     typedef map<uint_8, ClassCreatorFunc, less<uint_8> > ClassList;  
 | 
|---|
| 86 |     // Pas de createur appele pour objets statiques sur Linux - $CHECK$ Reza 26/04/99
 | 
|---|
| 87 |     static ClassList *           classList;  
 | 
|---|
| 88 |     static map<string, uint_8> * typeids;
 | 
|---|
| 89 | 
 | 
|---|
| 90 |   protected:
 | 
|---|
| 91 | 
 | 
|---|
| 92 |     enum {PPS_NULL = 0,             // this is a null object
 | 
|---|
| 93 |           PPS_STRING = 1,           // string, length (2b) + data
 | 
|---|
| 94 |           PPS_OBJECT = 2,           // classId, data...
 | 
|---|
| 95 |           PPS_REFERENCE = 3,        // objectId
 | 
|---|
| 96 |           PPS_TAG = 4,              // tag entries
 | 
|---|
| 97 |           PPS_EOF = 5,              // Just before tag infomation, offset to PPS_TAG
 | 
|---|
| 98 |           PPS_LINE = 6,             // '\n'-terminated, deprecated ?
 | 
|---|
| 99 |           PPS_TAG_MARK = 7,         // to reset object count when skipped
 | 
|---|
| 100 |           PPS_SIMPLE = 16,          // 16 + number of bytes, up to 8 bytes
 | 
|---|
| 101 |           PPS_SIMPLE_ARRAY = 32,    // 32 + number of bytes, up to 8 bytes, then 2 bytes of length
 | 
|---|
| 102 |           PPS_SIMPLE_ARRAY4 = 64,   // 64 + number of bytes, up to 8 bytes, then 4 bytes of length
 | 
|---|
| 103 |           PPS_SIMPLE_ARRAY8 = 128   // 64 + number of bytes, up to 8 bytes, then 8 bytes of length
 | 
|---|
| 104 |    };
 | 
|---|
| 105 | 
 | 
|---|
| 106 |     map<string, int_8> tags;
 | 
|---|
| 107 |   };
 | 
|---|
| 108 | 
 | 
|---|
| 109 | 
 | 
|---|
| 110 |   // TBD : use hash tables instead of maps. Check hashtbl status in STL.
 | 
|---|
| 111 | 
 | 
|---|
| 112 |   class PInPersist : public PIOPersist {
 | 
|---|
| 113 |   public:
 | 
|---|
| 114 |     PInPersist(string const& flnm, bool scan=true);
 | 
|---|
| 115 |     ~PInPersist();
 | 
|---|
| 116 | 
 | 
|---|
| 117 |     bool   GotoTag(string const& name);
 | 
|---|
| 118 |     int    NbTags();
 | 
|---|
| 119 |     bool   GotoTagNum(int itag);  // 0..NbTags-1
 | 
|---|
| 120 |     string GetTagName(int itag);  // 0..NbTags-1
 | 
|---|
| 121 |     vector<string> const &  GetTagNames();  
 | 
|---|
| 122 | 
 | 
|---|
| 123 |     void   GetByte (char& c);
 | 
|---|
| 124 |     void   GetBytes(void* ptr, size_t bytes);
 | 
|---|
| 125 |     void   GetR4   (r_4&);
 | 
|---|
| 126 |     void   GetR4s  (r_4*, size_t);
 | 
|---|
| 127 |     void   GetR8   (r_8&);
 | 
|---|
| 128 |     void   GetR8s  (r_8*, size_t);
 | 
|---|
| 129 |     void   GetI2   (int_2&);
 | 
|---|
| 130 |     void   GetI2s  (int_2*, size_t);
 | 
|---|
| 131 |     void   GetU2   (uint_2&);
 | 
|---|
| 132 |     void   GetU2s  (uint_2*, size_t);
 | 
|---|
| 133 |     void   GetI4   (int_4&);
 | 
|---|
| 134 |     void   GetI4s  (int_4*, size_t);
 | 
|---|
| 135 |     void   GetU4   (uint_4&);
 | 
|---|
| 136 |     void   GetU4s  (uint_4*, size_t);
 | 
|---|
| 137 |     void   GetI8   (int_8&);
 | 
|---|
| 138 |     void   GetI8s  (int_8*, size_t);
 | 
|---|
| 139 |     void   GetU8   (uint_8&);
 | 
|---|
| 140 |     void   GetU8s  (uint_8*, size_t);
 | 
|---|
| 141 |     void   GetLine (char* ptr, size_t len);
 | 
|---|
| 142 |     void   GetStr  (string&);
 | 
|---|
| 143 | 
 | 
|---|
| 144 |     void   Get(char&   c) {GetByte(c);}
 | 
|---|
| 145 |     void   Get(r_4&    x) {GetR4(x);}
 | 
|---|
| 146 |     void   Get(r_8&    x) {GetR8(x);}
 | 
|---|
| 147 |     void   Get(uint_2& x) {GetU2(x);}
 | 
|---|
| 148 |     void   Get(int_2&  x) {GetI2(x);}
 | 
|---|
| 149 |     void   Get(uint_4& x) {GetU4(x);}
 | 
|---|
| 150 |     void   Get(int_4&  x) {GetI4(x);}
 | 
|---|
| 151 |     void   Get(uint_8& x) {GetU8(x);}
 | 
|---|
| 152 |     void   Get(int_8&  x) {GetI8(x);}
 | 
|---|
| 153 |     void   Get(r_4*    x, size_t n) {GetR4s(x,n);}
 | 
|---|
| 154 |     void   Get(r_8*    x, size_t n) {GetR8s(x,n);}
 | 
|---|
| 155 |     void   Get(uint_2* x, size_t n) {GetU2s(x,n);}
 | 
|---|
| 156 |     void   Get(int_2*  x, size_t n) {GetI2s(x,n);}
 | 
|---|
| 157 |     void   Get(uint_4* x, size_t n) {GetU4s(x,n);}
 | 
|---|
| 158 |     void   Get(int_4*  x, size_t n) {GetI4s(x,n);}
 | 
|---|
| 159 |     void   Get(uint_8* x, size_t n) {GetU8s(x,n);}
 | 
|---|
| 160 |     void   Get(int_8*  x, size_t n) {GetI8s(x,n);}
 | 
|---|
| 161 |     void   Get(string& x) {GetStr(x);}
 | 
|---|
| 162 |   
 | 
|---|
| 163 |     PPersist*  ReadObject();
 | 
|---|
| 164 | 
 | 
|---|
| 165 | 
 | 
|---|
| 166 |     int    Version() {return version;}
 | 
|---|
| 167 |     time_t CreationDate() { return creationdate; }
 | 
|---|
| 168 | 
 | 
|---|
| 169 |   protected:
 | 
|---|
| 170 |     void   CheckTag   (short datasz);
 | 
|---|
| 171 |     void   CheckArrayTag(short datasz, size_t sz);
 | 
|---|
| 172 |     void   GetTypeTag (char& c);
 | 
|---|
| 173 |     void   GetRawByte (char& c);
 | 
|---|
| 174 |     void   GetRawBytes(void* ptr, size_t bytes);
 | 
|---|
| 175 |     void   GetRawI2   (int_2&);
 | 
|---|
| 176 |     void   GetRawI4   (int_4&);
 | 
|---|
| 177 |     void   GetRawI8   (int_8&);
 | 
|---|
| 178 |     void   GetRawU8   (uint_8&);
 | 
|---|
| 179 |     void   GetObject(PPersist*);               // Object has been allocated with correct type
 | 
|---|
| 180 |     int_4  assignObjectId(PPersist* x);
 | 
|---|
| 181 |     void   Scan();
 | 
|---|
| 182 |     char*  GetCStr(uint_2 l);
 | 
|---|
| 183 | 
 | 
|---|
| 184 |     istream* s;
 | 
|---|
| 185 | 
 | 
|---|
| 186 |     bool bigEndian;
 | 
|---|
| 187 |     int version;
 | 
|---|
| 188 | 
 | 
|---|
| 189 |     time_t creationdate;
 | 
|---|
| 190 | 
 | 
|---|
| 191 |     // already read objects, id = order in array 
 | 
|---|
| 192 |     typedef vector<PPersist*> ObjList;
 | 
|---|
| 193 |     ObjList objList;
 | 
|---|
| 194 | 
 | 
|---|
| 195 |     friend class PPersist;
 | 
|---|
| 196 |   };
 | 
|---|
| 197 | 
 | 
|---|
| 198 |   class POutPersist : public PIOPersist {
 | 
|---|
| 199 |   public:
 | 
|---|
| 200 |     POutPersist(string const& flnm, int endianness = PPS_NATIVE);
 | 
|---|
| 201 |     ~POutPersist();
 | 
|---|
| 202 | 
 | 
|---|
| 203 |     void WriteTag(string const& name);
 | 
|---|
| 204 |   
 | 
|---|
| 205 |     void PutByte (char c);
 | 
|---|
| 206 |     void PutBytes(void const* ptr, size_t bytes);
 | 
|---|
| 207 |     void PutR4   (r_4);
 | 
|---|
| 208 |     void PutR4s  (r_4 const*, size_t);
 | 
|---|
| 209 |     void PutR8   (r_8);
 | 
|---|
| 210 |     void PutR8s  (r_8 const*, size_t);
 | 
|---|
| 211 |     void PutI2   (int_2);
 | 
|---|
| 212 |     void PutI2s  (int_2 const*, size_t);
 | 
|---|
| 213 |     void PutU2   (uint_2);
 | 
|---|
| 214 |     void PutU2s  (uint_2 const*, size_t);
 | 
|---|
| 215 |     void PutI4   (int_4);
 | 
|---|
| 216 |     void PutI4s  (int_4 const*, size_t);
 | 
|---|
| 217 |     void PutU4   (uint_4);
 | 
|---|
| 218 |     void PutU4s  (uint_4 const*, size_t);
 | 
|---|
| 219 |     void PutI8   (int_8);
 | 
|---|
| 220 |     void PutI8s  (int_8 const*, size_t);
 | 
|---|
| 221 |     void PutU8   (uint_8);
 | 
|---|
| 222 |     void PutU8s  (uint_8 const*, size_t);
 | 
|---|
| 223 |     void PutLine (char const* ptr, size_t len=0); // deprecated ?
 | 
|---|
| 224 |     void PutStr  (string const&);
 | 
|---|
| 225 |     void PutObject (PPersist const*); // Like doing Write(stream) on object
 | 
|---|
| 226 | 
 | 
|---|
| 227 |     void   Put(char   c) {PutByte(c);}
 | 
|---|
| 228 |     void   Put(r_4    x) {PutR4(x);}
 | 
|---|
| 229 |     void   Put(r_8    x) {PutR8(x);}
 | 
|---|
| 230 |     void   Put(uint_2 x) {PutU2(x);}
 | 
|---|
| 231 |     void   Put(int_2  x) {PutI2(x);}
 | 
|---|
| 232 |     void   Put(uint_4 x) {PutU4(x);}
 | 
|---|
| 233 |     void   Put(int_4  x) {PutI4(x);}
 | 
|---|
| 234 |     void   Put(uint_8 x) {PutU8(x);}
 | 
|---|
| 235 |     void   Put(int_8  x) {PutI8(x);}
 | 
|---|
| 236 |     void   Put(r_4 const*    x, size_t n) {PutR4s(x,n);}
 | 
|---|
| 237 |     void   Put(r_8 const*    x, size_t n) {PutR8s(x,n);}
 | 
|---|
| 238 |     void   Put(uint_2 const* x, size_t n) {PutU2s(x,n);}
 | 
|---|
| 239 |     void   Put(int_2 const*  x, size_t n) {PutI2s(x,n);}
 | 
|---|
| 240 |     void   Put(uint_4 const* x, size_t n) {PutU4s(x,n);}
 | 
|---|
| 241 |     void   Put(int_4 const*  x, size_t n) {PutI4s(x,n);}
 | 
|---|
| 242 |     void   Put(uint_8 const* x, size_t n) {PutU8s(x,n);}
 | 
|---|
| 243 |     void   Put(int_8 const*  x, size_t n) {PutI8s(x,n);}
 | 
|---|
| 244 |     void   Put(string const& s) {PutStr(s);}
 | 
|---|
| 245 |     void   Put(PPersist const* x) {PutObject(x);}
 | 
|---|
| 246 | 
 | 
|---|
| 247 | 
 | 
|---|
| 248 |   protected:
 | 
|---|
| 249 |     ostream* s;
 | 
|---|
| 250 |     bool bigEndian;
 | 
|---|
| 251 | 
 | 
|---|
| 252 |     void     PutArrayTag(short datasz, size_t sz);
 | 
|---|
| 253 |     void     PutRawByte (char);
 | 
|---|
| 254 |     void     PutRawI2   (int_2);
 | 
|---|
| 255 |     void     PutRawI4   (int_4);
 | 
|---|
| 256 |     void     PutRawI8   (int_8);
 | 
|---|
| 257 |     void     PutRawU8   (uint_8);
 | 
|---|
| 258 |     void     PutRawBytes(void const* ptr, size_t bytes);
 | 
|---|
| 259 |     bool     serializeNullAndRepeat(PPersist const* x);
 | 
|---|
| 260 |     int_4    findObjectId(PPersist const* x);
 | 
|---|
| 261 |     int_4    assignObjectId(PPersist const* x);
 | 
|---|
| 262 | 
 | 
|---|
| 263 |     // already serialized objects
 | 
|---|
| 264 |     typedef map<PPersist const*, int_4, less<PPersist const*> >  ObjList;
 | 
|---|
| 265 |     ObjList objList;
 | 
|---|
| 266 |   };
 | 
|---|
| 267 |   
 | 
|---|
| 268 |   
 | 
|---|
| 269 | #define RAWPERSISTIO(_Type_,_xtyp_)                                             \
 | 
|---|
| 270 |   inline POutPersist& operator << (POutPersist& c, _Type_ const& data)            \
 | 
|---|
| 271 |   {                                                                               \
 | 
|---|
| 272 |     c.Put##_xtyp_(data);                                                          \
 | 
|---|
| 273 |     return c;                                                                     \
 | 
|---|
| 274 |   }                                                                               \
 | 
|---|
| 275 |                                                                                 \
 | 
|---|
| 276 |   inline PInPersist& operator >> (PInPersist& c, _Type_& data)                    \
 | 
|---|
| 277 |   {                                                                               \
 | 
|---|
| 278 |     c.Get##_xtyp_(data);                                                          \
 | 
|---|
| 279 |     return c;                                                                     \
 | 
|---|
| 280 |   }                                                                               
 | 
|---|
| 281 | 
 | 
|---|
| 282 |   RAWPERSISTIO(int_4,I4);
 | 
|---|
| 283 |   RAWPERSISTIO(uint_4,U4);
 | 
|---|
| 284 |   RAWPERSISTIO(int_2,I2);
 | 
|---|
| 285 |   RAWPERSISTIO(uint_2,U2);
 | 
|---|
| 286 |   RAWPERSISTIO(char,Byte);
 | 
|---|
| 287 |   RAWPERSISTIO(r_4,R4);
 | 
|---|
| 288 |   RAWPERSISTIO(r_8,R8);
 | 
|---|
| 289 |   
 | 
|---|
| 290 | #if 0
 | 
|---|
| 291 | #define STRUCTPERSISTIO(_Type_, _field_, _size_)                                \
 | 
|---|
| 292 |   inline POutPersist& operator << (POutPersist& c, _Type_ const& data)            \
 | 
|---|
| 293 |   {                                                                               \
 | 
|---|
| 294 |     c.PutBytes(&data._field_, _size_);                                            \
 | 
|---|
| 295 |     return c;                                                                     \
 | 
|---|
| 296 |   }                                                                               \
 | 
|---|
| 297 |                                                                                 \
 | 
|---|
| 298 |   inline PInPersist& operator >> (PInPersist& c, _Type_& data)                    \
 | 
|---|
| 299 |   {                                                                               \
 | 
|---|
| 300 |     c.GetBytes(&data._field_, _size_);                                            \
 | 
|---|
| 301 |     return c;                                                                     \
 | 
|---|
| 302 |   }                                                                               
 | 
|---|
| 303 | 
 | 
|---|
| 304 | #endif
 | 
|---|
| 305 |   
 | 
|---|
| 306 |   inline POutPersist& operator << (POutPersist& c, PPersist const& obj)
 | 
|---|
| 307 |     {
 | 
|---|
| 308 |       obj.Write(c);
 | 
|---|
| 309 |       return c;
 | 
|---|
| 310 |     }
 | 
|---|
| 311 |   
 | 
|---|
| 312 |   inline PInPersist& operator >> (PInPersist& c, PPersist& obj)
 | 
|---|
| 313 |     {
 | 
|---|
| 314 |       obj.Read(c);
 | 
|---|
| 315 |       return c;
 | 
|---|
| 316 |     }
 | 
|---|
| 317 |   
 | 
|---|
| 318 |   // Utility class to
 | 
|---|
| 319 |   //   - compute the class ID from a MD5 hash of the class name
 | 
|---|
| 320 |   //   - register classes with PIOPersist, through PPRegister macro
 | 
|---|
| 321 |   
 | 
|---|
| 322 |   template <class T>
 | 
|---|
| 323 |     class PPersistRegistrar {
 | 
|---|
| 324 |     public:
 | 
|---|
| 325 |       static PPersist* Create() {return new T();}
 | 
|---|
| 326 |       static void Register(string id) {PIOPersist::RegisterClass(Hash(id),typeid(T).name(),Create);}
 | 
|---|
| 327 |       static uint_8 Hash(string id) {
 | 
|---|
| 328 |         return PIOPersist::Hash(id);
 | 
|---|
| 329 |       }
 | 
|---|
| 330 |     };
 | 
|---|
| 331 |   
 | 
|---|
| 332 | #define PPRegister(className) PPersistRegistrar<className>::Register(#className);
 | 
|---|
| 333 |   
 | 
|---|
| 334 | } // namespace
 | 
|---|
| 335 | 
 | 
|---|
| 336 | #endif
 | 
|---|