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
114
115
116
117
118
119
120
121
122
123
|
/* ScummVM - Graphic Adventure Engine
*
* ScummVM is the legal property of its developers, whose names
* are too numerous to list here. Please refer to the COPYRIGHT
* file distributed with this source distribution.
*
* 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 3 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, see <http://www.gnu.org/licenses/>.
*
*/
#include "common/formats/disk_image.h"
#include "common/memstream.h"
#include "common/path.h"
#include "common/textconsole.h"
#include "agi/disk_image.h"
namespace Agi {
/**
* DiskImageStream is a stream wrapper around Common::DiskImage.
*
* This allows DiskImage to lazily decode tracks as a stream is used.
* This is important for detection, because the .woz format is noticeably
* expensive to decode all tracks at once, and detection has to read
* INITDIR to discover which track to read LOGDIR from.
*/
class DiskImageStream : virtual public Common::SeekableReadStream {
public:
DiskImageStream(Common::DiskImage *diskImage) : _diskImage(diskImage), _stream(_diskImage->getDiskStream()) {}
~DiskImageStream() {
delete _diskImage;
}
uint32 read(void *dataPtr, uint32 dataSize) override {
return _diskImage->read(dataPtr, pos(), dataSize);
}
bool eos() const override { return _stream->eos(); }
void clearErr() override { _stream->clearErr(); }
int64 pos() const override{ return _stream->pos(); }
int64 size() const override { return _stream->size(); }
bool seek(int64 offs, int whence = SEEK_SET) override { return _stream->seek(offs, whence); }
private:
Common::DiskImage *_diskImage;
Common::SeekableReadStream *_stream;
};
Common::SeekableReadStream *openPCDiskImage(const Common::Path &path, const Common::FSNode &node) {
Common::SeekableReadStream *stream = node.createReadStream();
if (stream == nullptr) {
warning("unable to open disk image: %s", path.baseName().c_str());
return nullptr;
}
// validate disk size
if (stream->size() != PC_DISK_SIZE) {
delete stream;
return nullptr;
}
return stream;
}
Common::SeekableReadStream *openA2DiskImage(const Common::Path &path, const Common::FSNode &node, bool loadAllTracks) {
Common::String name = path.baseName();
// Open the image with Common::DiskImage, unless the file extension is ".img".
// DiskImage expects ".img" to be a PC disk image, but it also gets used as
// an Apple II raw sector disk image, so just open it and and read it.
Common::SeekableReadStream *stream = nullptr;
if (name.hasSuffixIgnoreCase(".img")) {
stream = node.createReadStream();
} else {
if (loadAllTracks) {
// when loading all tracks, open with DiskImage and take the stream.
Common::DiskImage diskImage;
if (diskImage.open(node)) {
stream = diskImage.releaseStream();
}
} else {
// when loading tracks as they're used, create a DiskImage with lazy
// decoding and wrap it in a stream.
Common::DiskImage *diskImage = new Common::DiskImage();
diskImage->setLazyDecoding(true);
if (diskImage->open(node)) {
stream = new DiskImageStream(diskImage);
} else {
delete diskImage;
}
}
}
if (stream == nullptr) {
warning("unable to open disk image: %s", path.baseName().c_str());
return nullptr;
}
// validate disk size
if (stream->size() != A2_DISK_SIZE) {
delete stream;
return nullptr;
}
return stream;
}
} // End of namespace Agi
|
