13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
17
"""MemoryTree object.
19
19
See MemoryTree for more details.
22
from __future__ import absolute_import
23
from copy import deepcopy
25
26
from bzrlib import (
28
29
revision as _mod_revision,
30
from bzrlib.decorators import needs_read_lock, needs_write_lock
31
from bzrlib.decorators import needs_read_lock
32
from bzrlib.inventory import Inventory
31
33
from bzrlib.osutils import sha_file
32
34
from bzrlib.mutabletree import needs_tree_write_lock
33
35
from bzrlib.transport.memory import MemoryTransport
36
class MemoryTree(mutabletree.MutableTree):
38
class MemoryTree(mutabletree.MutableInventoryTree):
37
39
"""A MemoryTree is a specialisation of MutableTree.
39
41
It maintains nearly no state outside of read_lock and write_lock
40
42
transactions. (it keeps a reference to the branch, and its last-revision
50
52
self._lock_mode = None
54
def get_config_stack(self):
55
return self.branch.get_config_stack()
57
def is_control_filename(self, filename):
58
# Memory tree doesn't have any control filenames
52
61
@needs_tree_write_lock
53
62
def _add(self, files, ids, kinds):
54
63
"""See MutableTree._add."""
73
82
def _gather_kinds(self, files, kinds):
74
83
"""See MutableTree._gather_kinds.
76
85
This implementation does not care about the file kind of
77
86
missing files, so is a no-op.
99
108
return None, False, None
100
109
return entry.kind, entry.executable, None
111
@needs_tree_write_lock
112
def rename_one(self, from_rel, to_rel):
113
file_id = self.path2id(from_rel)
114
to_dir, to_tail = os.path.split(to_rel)
115
to_parent_id = self.path2id(to_dir)
116
self._file_transport.move(from_rel, to_rel)
117
self._inventory.rename(file_id, to_parent_id, to_tail)
102
119
def path_content_summary(self, path):
103
120
"""See Tree.path_content_summary."""
104
121
id = self.path2id(path)
203
220
def _populate_from_branch(self):
204
221
"""Populate the in-tree state from the branch."""
205
self._basis_tree = self.branch.repository.revision_tree(
206
self._branch_revision_id)
207
223
if self._branch_revision_id == _mod_revision.NULL_REVISION:
208
224
self._parent_ids = []
210
226
self._parent_ids = [self._branch_revision_id]
211
self._inventory = deepcopy(self._basis_tree._inventory)
227
self._inventory = Inventory(None, self._basis_tree.get_revision_id())
212
228
self._file_transport = MemoryTransport()
213
229
# TODO copy the revision trees content, or do it lazy, or something.
214
inventory_entries = self._inventory.iter_entries()
230
inventory_entries = self._basis_tree.iter_entries_by_dir()
215
231
for path, entry in inventory_entries:
232
self._inventory.add(entry.copy())
218
235
if entry.kind == 'directory':
270
287
_mod_revision.check_not_reserved_id(revision_id)
271
288
if len(revision_ids) == 0:
272
289
self._parent_ids = []
273
self._basis_tree = self.branch.repository.revision_tree(
274
_mod_revision.NULL_REVISION)
290
self._branch_revision_id = _mod_revision.NULL_REVISION
276
292
self._parent_ids = revision_ids
277
self._basis_tree = self.branch.repository.revision_tree(
279
293
self._branch_revision_id = revision_ids[0]
294
self._allow_leftmost_as_ghost = allow_leftmost_as_ghost
297
def _set_basis(self):
299
self._basis_tree = self.branch.repository.revision_tree(
300
self._branch_revision_id)
301
except errors.NoSuchRevision:
302
if self._allow_leftmost_as_ghost:
303
self._basis_tree = self.branch.repository.revision_tree(
304
_mod_revision.NULL_REVISION)
281
308
def set_parent_trees(self, parents_list, allow_leftmost_as_ghost=False):
282
309
"""See MutableTree.set_parent_trees()."""