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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
17
"""MemoryTree object.
19
19
See MemoryTree for more details.
22
from __future__ import absolute_import
23
from copy import deepcopy
26
26
from bzrlib import (
29
30
revision as _mod_revision,
31
from bzrlib.decorators import needs_read_lock
32
from bzrlib.inventory import Inventory
32
from bzrlib.decorators import needs_read_lock, needs_write_lock
33
33
from bzrlib.osutils import sha_file
34
34
from bzrlib.mutabletree import needs_tree_write_lock
35
35
from bzrlib.transport.memory import MemoryTransport
38
class MemoryTree(mutabletree.MutableInventoryTree):
38
class MemoryTree(mutabletree.MutableTree):
39
39
"""A MemoryTree is a specialisation of MutableTree.
41
41
It maintains nearly no state outside of read_lock and write_lock
42
42
transactions. (it keeps a reference to the branch, and its last-revision
52
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
61
54
@needs_tree_write_lock
62
55
def _add(self, files, ids, kinds):
63
56
"""See MutableTree._add."""
82
75
def _gather_kinds(self, files, kinds):
83
76
"""See MutableTree._gather_kinds.
85
78
This implementation does not care about the file kind of
86
79
missing files, so is a no-op.
220
213
def _populate_from_branch(self):
221
214
"""Populate the in-tree state from the branch."""
215
self._basis_tree = self.branch.repository.revision_tree(
216
self._branch_revision_id)
223
217
if self._branch_revision_id == _mod_revision.NULL_REVISION:
224
218
self._parent_ids = []
226
220
self._parent_ids = [self._branch_revision_id]
227
self._inventory = Inventory(None, self._basis_tree.get_revision_id())
221
self._inventory = deepcopy(self._basis_tree._inventory)
228
222
self._file_transport = MemoryTransport()
229
223
# TODO copy the revision trees content, or do it lazy, or something.
230
inventory_entries = self._basis_tree.iter_entries_by_dir()
224
inventory_entries = self._inventory.iter_entries()
231
225
for path, entry in inventory_entries:
232
self._inventory.add(entry.copy())
235
228
if entry.kind == 'directory':
287
280
_mod_revision.check_not_reserved_id(revision_id)
288
281
if len(revision_ids) == 0:
289
282
self._parent_ids = []
290
self._branch_revision_id = _mod_revision.NULL_REVISION
283
self._basis_tree = self.branch.repository.revision_tree(
284
_mod_revision.NULL_REVISION)
292
286
self._parent_ids = revision_ids
287
self._basis_tree = self.branch.repository.revision_tree(
293
289
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)
308
291
def set_parent_trees(self, parents_list, allow_leftmost_as_ghost=False):
309
292
"""See MutableTree.set_parent_trees()."""