<reponame>X-EcutiOnner/fileobj<gh_stars>10-100
# Copyright (c) 2013, <NAME>
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# 1. Redistributions of source code must retain the above copyright notice, this
#    list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright notice,
#    this list of conditions and the following disclaimer in the documentation
#    and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

from __future__ import division
from __future__ import with_statement
import os

from . import chunk
from . import filebytes
from . import fileobj
from . import kernel
from . import log
from . import screen
from . import setting
from . import util

class Fileobj (fileobj.Fileobj):
    _insert  = False
    _replace = False
    _delete  = False
    _truncate= False
    _enabled = True
    _partial = True

    def __init__(self, f, offset=0, length=0):
        self.cbuf = []
        self.__thresh = 0
        self.set_size(0)
        super(Fileobj, self).__init__(f, offset, length)

    def __str__(self):
        l = []
        l.append("size " + util.get_size_repr(self.get_size()))
        l.append("chunk size {0}[B]".format(self.get_chunk_size()))
        l.append("chunk total {0}\n".format(len(self.cbuf)))
        for i, o in enumerate(self.cbuf):
            l.append("[{0}] {1}".format(i, o))
        return '\n'.join(l)

    def ctr(self):
        f = self.get_path()
        assert not self.cbuf
        assert os.path.isfile(f), f
        with kernel.fopen(f) as fd:
            fd.seek(self.get_mapping_offset())
            length = self.get_mapping_length()
            if length:
                self.init_chunk(fd.read(length))
            else:
                self.init_chunk(fd.read())

    def is_dirty(self):
        return False

    def get_size(self):
        return self.__size

    def set_size(self, size):
        self.__size = size

    def get_chunk_size(self):
        ret = setting.buffer_chunk_size
        if ret != -1:
            return ret
        else:
            return kernel.get_page_size()

    def set_search_thresh(self):
        # heuristic
        if len(self.cbuf) > 10000:
            r = 0.01
        elif len(self.cbuf) > 1000:
            r = 0.02
        elif len(self.cbuf) > 20:
            r = 0.1
        else:
            r = 0.5
        self.__thresh = int(self.get_size() * r)
        log.debug("'{0}' has search thresh at {1}[B]/{2}[B]".format(
            self.get_short_path(), self.__thresh, self.get_size()))

    def alloc_chunk(self, offset, buf):
        return chunk.Chunk(offset, buf)

    def init_chunk(self, b):
        self.cbuf = []
        siz = self.get_chunk_size()
        for i in util.get_xrange(0, len(b), siz):
            bb = b[i : i + siz]
            self.cbuf.append(self.alloc_chunk(i, bb))
        self.set_size(len(b))
        self.mark_chunk()
        self.set_search_thresh()

    def mark_chunk(self):
        if not self.cbuf:
            o = self.alloc_chunk(0, filebytes.BLANK)
            self.cbuf.append(o)
        self.cbuf[-1].islast = True

    def find(self, x, s, end):
        for o in self.cbuf:
            if end != -1 and x >= end:
                break
            if x in o:
                i = self.cbuf.index(o)
                if i < len(self.cbuf) - 1:
                    pos = self.cbuf[i + 1].offset
                    siz = len(s) - 1
                    b = self.read(pos, siz)
                else:
                    b = filebytes.BLANK
                ret = o.search(x, s, b)
                if ret != fileobj.NOTFOUND:
                    return ret
                x = o.offset + len(o)
            if screen.test_signal():
                return fileobj.INTERRUPT
        return fileobj.NOTFOUND

    def rfind(self, x, s, end):
        for o in reversed(self.cbuf):
            if end != -1 and x <= end:
                break
            if x in o:
                i = self.cbuf.index(o)
                if i > 0:
                    siz = len(s) - 1
                    pos = o.offset - siz
                    if pos < 0:
                        siz = o.offset
                        pos = 0
                    b = self.read(pos, siz)
                else:
                    b = filebytes.BLANK
                ret = o.rsearch(x, s, b)
                if ret != fileobj.NOTFOUND:
                    return ret
                x = o.offset - 1
            if screen.test_signal():
                return fileobj.INTERRUPT
        return fileobj.NOTFOUND

    def iter_chunk(self, pos):
        index = self.get_chunk_index(pos)
        for i in util.get_xrange(index, len(self.cbuf)):
            yield self.cbuf[i]

    def get_chunk_index(self, pos):
        if pos <= self.__thresh:
            for o in self.cbuf:
                if pos in o:
                    return self.cbuf.index(o)
        else:
            beg = 0
            end = len(self.cbuf) - 1
            while True:
                i = (beg + end) // 2
                o = self.cbuf[i]
                if pos in o:
                    return self.cbuf.index(o)
                if pos < o.offset:
                    end = i - 1
                elif pos > o.offset + len(o) - 1:
                    beg = i + 1
                if beg > end:
                    break
        assert False, pos

    def read(self, x, n):
        if not n:
            return filebytes.BLANK
        buf = []
        for o in self.iter_chunk(x):
            b = o.read(x, n)
            if b:
                buf.append(b)
                x += len(b)
                n -= len(b)
                if n <= 0:
                    break # doesn't always come here
        return filebytes.join(buf)
