Mercurial > hg > octave-nkf
diff src/oct-map.h @ 10742:604e13a89c7f
initial code for structs rewrite
author | Jaroslav Hajek <highegg@gmail.com> |
---|---|
date | Tue, 22 Jun 2010 15:22:36 +0200 |
parents | 0c42b6b7da24 |
children | cb3ed842bd30 |
line wrap: on
line diff
--- a/src/oct-map.h +++ b/src/oct-map.h @@ -2,6 +2,7 @@ Copyright (C) 1994, 1995, 1996, 1997, 2000, 2002, 2003, 2004, 2005, 2006, 2007, 2009 John W. Eaton +Copyright (C) 2010 VZLU Prague This file is part of Octave. @@ -32,6 +33,385 @@ class string_vector; +// A class holding a map field->index. Supports reference-counting. +class OCTINTERP_API +octave_fields +{ + class fields_rep : public std::map<std::string, octave_idx_type> + { + public: + fields_rep (void) : std::map<std::string, octave_idx_type> (), count (1) { } + fields_rep (const fields_rep& other) + : std::map<std::string, octave_idx_type> (other), count (1) { } + + int count; + + private: + fields_rep& operator = (const fields_rep&); // no assignment! + }; + + fields_rep *rep; + + static fields_rep nil_rep; + +public: + + octave_fields (void) : rep (&nil_rep) { rep->count++; } + octave_fields (const string_vector&); + + ~octave_fields (void) + { + if (--rep->count == 0) + delete rep; + } + + void make_unique (void) + { + if (rep->count > 1) + { + --rep->count; + rep = new fields_rep (*rep); + } + } + + octave_fields (const octave_fields& o) : rep (o.rep) { rep->count++; } + + octave_fields& + operator = (const octave_fields& o) + { + o.rep->count++; + if (--rep->count) + delete rep; + rep = o.rep; + + return *this; + } + + // constant iteration support. non-const iteration intentionally unsupported. + + typedef std::map<std::string, octave_idx_type>::const_iterator const_iterator; + typedef const_iterator iterator; + + const_iterator begin (void) const { return rep->begin (); } + const_iterator end (void) const { return rep->end (); } + + std::string key (const_iterator p) const { return p->first; } + octave_idx_type index (const_iterator p) const { return p->second; } + + const_iterator seek (const std::string& k) const + { return rep->find (k); } + + // high-level methods. + + // number of fields. + octave_idx_type nfields (void) const { return rep->size (); } + + // check whether a field exists. + bool isfield (const std::string& name) const; + + // get index of field. return -1 if not exist + octave_idx_type getfield (const std::string& name) const; + // get index of field. add if not exist + octave_idx_type getfield (const std::string& name); + // remove field and return the index. -1 if didn't exist. + octave_idx_type rmfield (const std::string& name); + + // order the fields of this map. creates a permutation + // used to order the fields. + void orderfields (Array<octave_idx_type>& perm); + + // compares two instances for equality up to order of fields. + // returns a permutation needed to bring the fields of *other* + // into the order of *this*. + bool equal_up_to_order (const octave_fields& other, + Array<octave_idx_type>& perm) const; + + bool is_same (const octave_fields& other) const + { return rep == other.rep; } + + // Returns the fields as a vector of strings. + string_vector fieldnames (void) const; + + void clear (void) + { + *this = octave_fields (); + } +}; + + +class OCTINTERP_API +octave_scalar_map +{ + octave_scalar_map (const octave_fields& k) + : keys (k), vals (k.nfields ()) { } + +public: + + octave_scalar_map (void) : keys (), vals () { } + + octave_scalar_map (const string_vector& k) + : keys (k), vals (k.length ()) { } + + octave_scalar_map (const octave_scalar_map& m) + : keys (m.keys), vals(m.vals) { } + + octave_scalar_map& operator = (const octave_scalar_map& m) + { + keys = m.keys; + vals = m.vals; + + return *this; + } + + // iteration support. note that both const and non-const iterators are the + // same. The const/non-const distinction is made by the key & contents method. + typedef octave_fields::const_iterator const_iterator; + typedef const_iterator iterator; + + const_iterator begin (void) const { return keys.begin (); } + const_iterator end (void) const { return keys.end (); } + + const_iterator seek (const std::string& k) const { return keys.seek (k); } + + std::string key (const_iterator p) const + { return keys.key (p); } + octave_idx_type index (const_iterator p) const + { return keys.index (p); } + + const octave_value& contents (const_iterator p) const + { return vals[keys.index (p)]; } + + octave_value& contents (iterator p) + { return vals[keys.index (p)]; } + + const octave_value& contents (octave_idx_type i) const + { return vals[i]; } + + octave_value& contents (octave_idx_type i) + { return vals[i]; } + + // number of fields. + octave_idx_type nfields (void) const { return keys.nfields (); } + + // check whether a field exists. + bool isfield (const std::string& name) const + { return keys.isfield (name); } + + string_vector fieldnames (void) const + { return keys.fieldnames (); } + + // get contents of a given field. empty value if not exist. + octave_value getfield (const std::string& key) const; + + // set contents of a given field. add if not exist. + void setfield (const std::string& key, const octave_value& val); + + // remove a given field. do nothing if not exist. + void rmfield (const std::string& key); + + // return a copy with fields ordered, optionally along with permutation. + octave_scalar_map orderfields (void) const; + octave_scalar_map orderfields (Array<octave_idx_type>& perm) const; + octave_scalar_map orderfields (const octave_scalar_map& other, + Array<octave_idx_type>& perm) const; + + // aka getfield/setfield, but the latter returns a reference. + octave_value contents (const std::string& k) const; + octave_value& contents (const std::string& k); + + void clear (void) + { + keys.clear (); + vals.clear (); + } + + friend class octave_map; + +private: + + octave_fields keys; + std::vector<octave_value> vals; + +}; + +class OCTINTERP_API +octave_map +{ + octave_map (const octave_fields& k) + : keys (k), vals (k.nfields ()), dimensions () { } + +public: + + octave_map (void) : keys (), vals (), dimensions () { } + + octave_map (const dim_vector& dv) : keys (), vals (), dimensions (dv) { } + + octave_map (const string_vector& k) + : keys (k), vals (k.length ()), dimensions (1, 1) { } + + octave_map (const dim_vector& dv, const string_vector& k) + : keys (k), vals (k.length ()), dimensions (dv) { } + + octave_map (const octave_map& m) + : keys (m.keys), vals (m.vals), dimensions (m.dimensions) { } + + octave_map (const octave_scalar_map& m); + + octave_map (const Octave_map& m); + + octave_map& operator = (const octave_map& m) + { + keys = m.keys; + vals = m.vals; + dimensions = m.dimensions; + + return *this; + } + + // iteration support. note that both const and non-const iterators are the + // same. The const/non-const distinction is made by the key & contents method. + typedef octave_fields::const_iterator const_iterator; + typedef const_iterator iterator; + + const_iterator begin (void) const { return keys.begin (); } + const_iterator end (void) const { return keys.end (); } + + const_iterator seek (const std::string& k) const { return keys.seek (k); } + + std::string key (const_iterator p) const + { return keys.key (p); } + octave_idx_type index (const_iterator p) const + { return keys.index (p); } + + const Cell& contents (const_iterator p) const + { return vals[keys.index (p)]; } + + Cell& contents (iterator p) + { return vals[keys.index (p)]; } + + const Cell& contents (octave_idx_type i) const + { return vals[i]; } + + Cell& contents (octave_idx_type i) + { return vals[i]; } + + // number of fields. + octave_idx_type nfields (void) const { return keys.nfields (); } + + // check whether a field exists. + bool isfield (const std::string& name) const + { return keys.isfield (name); } + + string_vector fieldnames (void) const + { return keys.fieldnames (); } + + // get contents of a given field. empty value if not exist. + Cell getfield (const std::string& key) const; + + // set contents of a given field. add if not exist. checks for + // correct dimensions. + void setfield (const std::string& key, const Cell& val); + + // remove a given field. do nothing if not exist. + void rmfield (const std::string& key); + + // return a copy with fields ordered, optionally along with permutation. + octave_map orderfields (void) const; + octave_map orderfields (Array<octave_idx_type>& perm) const; + octave_map orderfields (const octave_map& other, + Array<octave_idx_type>& perm) const; + + // aka getfield/setfield, but the latter returns a reference. + Cell contents (const std::string& k) const; + Cell& contents (const std::string& k); + + void clear (void) + { + keys.clear (); + vals.clear (); + } + + // The Array-like methods. + octave_idx_type numel (void) const { return dimensions.numel (); } + octave_idx_type length (void) const { return numel (); } + + octave_idx_type rows (void) const { return dimensions(0); } + octave_idx_type cols (void) const { return dimensions(1); } + octave_idx_type columns (void) const { return dimensions(1); } + + // Extract a scalar substructure. + octave_scalar_map checkelem (octave_idx_type n) const; + octave_scalar_map checkelem (octave_idx_type i, octave_idx_type j) const; + + octave_scalar_map + checkelem (const Array<octave_idx_type>& ra_idx) const; + + octave_map squeeze (void) const; + + octave_map permute (const Array<int>& vec, bool inv = false) const; + + dim_vector dims (void) const { return dimensions; } + + int ndims (void) const { return dimensions.length (); } + + octave_map transpose (void) const; + + octave_map reshape (const dim_vector& dv) const; + + void resize (const dim_vector& dv); + + static octave_map + cat (int dim, octave_idx_type n, const octave_scalar_map *map_list); + + static octave_map + cat (int dim, octave_idx_type n, const octave_map *map_list); + + octave_map index (const idx_vector& i, bool resize_ok) const; + + octave_map index (const idx_vector& i, const idx_vector& j, + bool resize_ok) const; + + octave_map index (const Array<idx_vector>& ia, + bool resize_ok) const; + + octave_map index (const octave_value_list&, bool resize_ok) const; + + void assign (const idx_vector& i, const octave_map& rhs); + + void assign (const idx_vector& i, const idx_vector& j, const octave_map& rhs); + + void assign (const Array<idx_vector>& ia, const octave_map& rhs); + + void assign (const octave_value_list&, const octave_map& rhs); + + void delete_elements (const idx_vector& i); + + void delete_elements (int dim, const idx_vector& i); + + void delete_elements (const Array<idx_vector>& ia); + + void delete_elements (const octave_value_list&); + + octave_map concat (const octave_map& rb, const Array<octave_idx_type>& ra_idx); + +private: + + octave_fields keys; + std::vector<Cell> vals; + dim_vector dimensions; + + void optimize_dimensions (void); + void extract_scalar (octave_scalar_map& dest, + octave_idx_type index) const; + static void do_cat (int dim, octave_idx_type n, + const octave_scalar_map *map_list, octave_map& retval); + static void do_cat (int dim, octave_idx_type n, + const octave_map *map_list, octave_map& retval); +}; + +// The original Octave_map object. Octave_map and octave_map are convertible to +// each other. + class OCTINTERP_API Octave_map @@ -85,6 +465,8 @@ Octave_map (const Octave_map& m) : map (m.map), key_list (m.key_list), dimensions (m.dimensions) { } + Octave_map (const octave_map& m); + Octave_map& operator = (const Octave_map& m) { if (this != &m)