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."""
68
77
def create_on_branch(branch):
69
78
"""Create a MemoryTree for branch, using the last-revision of branch."""
70
79
revision_id = _mod_revision.ensure_null(branch.last_revision())
71
if _mod_revision.is_null(revision_id):
73
80
return MemoryTree(branch, revision_id)
75
82
def _gather_kinds(self, files, kinds):
76
83
"""See MutableTree._gather_kinds.
78
85
This implementation does not care about the file kind of
79
86
missing files, so is a no-op.
101
108
return None, False, None
102
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)
104
119
def path_content_summary(self, path):
105
120
"""See Tree.path_content_summary."""
106
121
id = self.path2id(path)
205
220
def _populate_from_branch(self):
206
221
"""Populate the in-tree state from the branch."""
207
self._basis_tree = self.branch.repository.revision_tree(
208
self._branch_revision_id)
209
if self._branch_revision_id is None:
223
if self._branch_revision_id == _mod_revision.NULL_REVISION:
210
224
self._parent_ids = []
212
226
self._parent_ids = [self._branch_revision_id]
213
self._inventory = deepcopy(self._basis_tree._inventory)
227
self._inventory = Inventory(None, self._basis_tree.get_revision_id())
214
228
self._file_transport = MemoryTransport()
215
229
# TODO copy the revision trees content, or do it lazy, or something.
216
inventory_entries = self._inventory.iter_entries()
230
inventory_entries = self._basis_tree.iter_entries_by_dir()
217
231
for path, entry in inventory_entries:
232
self._inventory.add(entry.copy())
220
235
if entry.kind == 'directory':
272
287
_mod_revision.check_not_reserved_id(revision_id)
273
288
if len(revision_ids) == 0:
274
289
self._parent_ids = []
275
self._basis_tree = self.branch.repository.revision_tree(None)
290
self._branch_revision_id = _mod_revision.NULL_REVISION
277
292
self._parent_ids = revision_ids
278
self._basis_tree = self.branch.repository.revision_tree(
280
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)
282
308
def set_parent_trees(self, parents_list, allow_leftmost_as_ghost=False):
283
309
"""See MutableTree.set_parent_trees()."""
284
310
if len(parents_list) == 0:
285
311
self._parent_ids = []
286
self._basis_tree = self.branch.repository.revision_tree(None)
312
self._basis_tree = self.branch.repository.revision_tree(
313
_mod_revision.NULL_REVISION)
288
315
if parents_list[0][1] is None and not allow_leftmost_as_ghost:
289
316
# a ghost in the left most parent
290
317
raise errors.GhostRevisionUnusableHere(parents_list[0][0])
291
318
self._parent_ids = [parent_id for parent_id, tree in parents_list]
292
319
if parents_list[0][1] is None or parents_list[0][1] == 'null:':
293
import pdb; pdb.set_trace()
294
self._basis_tree = self.branch.repository.revision_tree(None)
320
self._basis_tree = self.branch.repository.revision_tree(
321
_mod_revision.NULL_REVISION)
296
323
self._basis_tree = parents_list[0][1]
297
324
self._branch_revision_id = parents_list[0][0]