Web   ·   Wiki   ·   Activities   ·   Blog   ·   Lists   ·   Chat   ·   Meeting   ·   Bugs   ·   Git   ·   Translate   ·   Archive   ·   People   ·   Donate
summaryrefslogtreecommitdiffstats
path: root/document.py
blob: 99f1a043a36ad03f612e4b08ef26722a59618809 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

import os
import gtk
import logging

import toolkit.json as json

import theme
from sound import *
from ground import *
from utils import *
from char import *

from toolkit.tarball import Tarball

logger = logging.getLogger('cartoon-builder')


class Document:
    tape = []
    ground = None
    sound = None

    for i in range(theme.TAPE_COUNT):
        tape.append(EmptyFrame())

def clean(index):
    from char import Frame
    Document.tape[index] = EmptyFrame()

def save(filepath):
    tar = Tarball(filepath, 'w')

    cfg = { 'ground': {},
            'sound' : {},
            'frames': {},
            'tape'  : [] }

    def _save(node, arcname, value):
        if value.custom():
            node['custom'] = True
            node['filename'] = arcname
            tar.write(arcname, value.serialize())
        else:
            node['custom'] = False
        node['name'] = unicode(value.name)
        node['id'] = value.id

    _save(cfg['ground'], 'ground.png', Document.ground)
    _save(cfg['sound'], 'sound', Document.sound)

    for i, frame in enumerate(
            [i for i in set(Document.tape) if not i.empty() and i.custom()]):
        arcname = 'frame%03d.png' % i
        cfg['frames'][frame.id] = arcname
        tar.write(arcname, frame.serialize())

    for i, frame in enumerate(Document.tape):
        if not frame.empty():
            node = {}
            node['custom'] = frame.custom()
            node['id'] = frame.id
            node['index'] = i
            cfg['tape'].append(node)

    tar.write('MANIFEST', json.dumps(cfg))
    tar.close()

def load(filepath):
    try:
        tar = Tarball(filepath)
        cfg = json.loads(tar.read('MANIFEST'))

        def _load(node, restored_class, preinstalled_class):
            if node['custom']:
                return restored_class(node['name'], node['id'],
                        tar.read(node['filename']))
            else:
                return preinstalled_class(node['name'], node['id'])

        Document.ground = _load(cfg['ground'], RestoredGround, PreinstalledGround)
        Document.sound = _load(cfg['sound'], RestoredSound, PreinstalledSound)

        frames = {}

        for id, arcname in cfg['frames'].items():
            frames[id] = RestoredFrame(id, tar.read(arcname))

        for node in cfg['tape']:
            i = node['index']
            if i < theme.TAPE_COUNT:
                if node['custom']:
                    Document.tape[i] = frames[node['id']]
                else:
                    Document.tape[i] = PreinstalledFrame(node['id'])

        tar.close()

    except Exception, e:
        logger.error('Cannot load jobject: %s' % e)