mirror of
https://github.com/davidgiven/fluxengine.git
synced 2025-10-31 11:17:01 -07:00
Add writing IMD images functionality
This commit is contained in:
wybren1971
420
lib/imagewriter/imdimagewriter.cc
Normal file
420
lib/imagewriter/imdimagewriter.cc
Normal file
@@ -0,0 +1,420 @@
|
||||
#include "globals.h"
|
||||
#include "flags.h"
|
||||
#include "sector.h"
|
||||
#include "imagewriter/imagewriter.h"
|
||||
#include "image.h"
|
||||
#include "lib/config.pb.h"
|
||||
#include "imginputoutpututils.h"
|
||||
#include "fmt/format.h"
|
||||
#include "logger.h"
|
||||
#include <algorithm>
|
||||
#include <iostream>
|
||||
#include <fstream>
|
||||
|
||||
/*
|
||||
* Where to get the type of encoding?? FM or MFM??
|
||||
*
|
||||
*/
|
||||
static const char LABEL[] = "IMD file by fluxengine"; //22 karakters
|
||||
static uint8_t getModulationandSpeed(int flags)
|
||||
{
|
||||
if (flags == 0)
|
||||
{
|
||||
Error() << fmt::format("Can't write IMD files with this speed {}, and modulation {}. Try another format.", flags, false);
|
||||
}
|
||||
else{
|
||||
flags = 1000000.0 / flags;
|
||||
}
|
||||
|
||||
|
||||
if ((flags>950) and (flags<1050)) //HD disk 5% discrepency is ok 1000*5% = 50
|
||||
{
|
||||
/* 500 kbps */
|
||||
// if (IbmDecoder::getuseFm() == true)
|
||||
if (false)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
return 3;
|
||||
}
|
||||
} else if ((flags>570) and (flags<630))//dont know exactly what the clock is for 300kbps assuming twice just as 500kbps en 250kbps
|
||||
/* 300 kbps*/
|
||||
{
|
||||
// if (IbmDecoder::getuseFm() == true)
|
||||
if (false)
|
||||
{
|
||||
return 1;
|
||||
}
|
||||
else
|
||||
{
|
||||
return 4;
|
||||
}
|
||||
} else if ((flags>475) and (flags<525)) //DD disk
|
||||
/* 250 kbps */
|
||||
{
|
||||
// if (IbmDecoder::getuseFm() == true)
|
||||
if (false)
|
||||
{
|
||||
return 2;
|
||||
}
|
||||
else
|
||||
{
|
||||
return 5;
|
||||
}
|
||||
} else {
|
||||
// Error() << fmt::format("Can't write IMD files with this speed {}, and modulation {}. Try another format.", flags, IbmDecoder::getuseFm());
|
||||
Error() << fmt::format("Can't write IMD files with this speed {}, and modulation {}. Try another format.", flags, false);
|
||||
}
|
||||
}
|
||||
|
||||
struct TrackHeader
|
||||
{
|
||||
uint8_t ModeValue;
|
||||
uint8_t track;
|
||||
uint8_t Head;
|
||||
uint8_t numSectors;
|
||||
uint8_t SectorSize;
|
||||
};
|
||||
|
||||
static uint8_t setSectorSize(int flags)
|
||||
{
|
||||
switch (flags)
|
||||
{
|
||||
case 128: return 0;
|
||||
case 256: return 1;
|
||||
case 512: return 2;
|
||||
case 1024: return 3;
|
||||
case 2048: return 4;
|
||||
case 4096: return 5;
|
||||
case 8192: return 6;
|
||||
}
|
||||
Error() << fmt::format("Sector size {} not in standard range (128, 256, 512, 1024, 2048, 4096, 8192).", flags);
|
||||
}
|
||||
|
||||
|
||||
#define SEC_CYL_MAP_FLAG 0x80
|
||||
#define SEC_HEAD_MAP_FLAG 0x40
|
||||
#define HEAD_MASK 0x3F
|
||||
#define END_OF_FILE 0x1A
|
||||
|
||||
/*
|
||||
*IMAGE FILE FORMAT
|
||||
*The overall layout of an ImageDisk .IMD image file is:
|
||||
*IMD v.vv: dd/mm/yyyy hh:mm:ss
|
||||
*Comment (ASCII only - unlimited size)
|
||||
*1A byte - ASCII EOF character
|
||||
*- For each track on the disk:
|
||||
*1 byte Mode value (0-5) see getModulationspeed for definition
|
||||
*1 byte Cylinder (0-n)
|
||||
*1 byte Head (0-1)
|
||||
*1 byte number of sectors in track (1-n)
|
||||
*1 byte sector size (0-6) see getsectorsize for definition
|
||||
*sector numbering map IMD start numbering sectors with 1.
|
||||
*sector cylinder map (optional) definied in high byte of head (since head is 0 or 1)
|
||||
*sector head map (optional) definied in high byte of head (since head is 0 or 1)
|
||||
*sector data records For each data record:
|
||||
* 1 byte Sector status
|
||||
* 0: Sector data unavailable - could not be read
|
||||
* 1: Normal data: (Sector Size) bytes follow
|
||||
* 2: Compressed: All bytes in sector have same value (xx)
|
||||
* 3: Normal data with "Deleted-Data address mark"
|
||||
* 4: Compressed with "Deleted-Data address mark"
|
||||
* 5: Normal data read with data error
|
||||
* 6: Compressed read with data error"
|
||||
* 7: Deleted data read with data error"
|
||||
* 8: Compressed, Deleted read with data error"
|
||||
* sector size of Sector data
|
||||
*<End of file>
|
||||
*/
|
||||
|
||||
class ImdImageWriter : public ImageWriter
|
||||
{
|
||||
public:
|
||||
ImdImageWriter(const ImageWriterProto& config):
|
||||
ImageWriter(config)
|
||||
{}
|
||||
|
||||
void writeImage(const Image& image)
|
||||
{
|
||||
const Geometry& geometry = image.getGeometry();
|
||||
unsigned numCylinders;
|
||||
unsigned numHeads;
|
||||
unsigned numSectors;
|
||||
unsigned numBytes;
|
||||
//const Geometry& geometry = image.getGeometry();
|
||||
//image.calculateSize(numCylinders, numHeads, numSectors, numBytes); //this gets the maximum size of each..
|
||||
// _config.PrintDebugString();
|
||||
std::ofstream outputFile(_config.filename(), std::ios::out | std::ios::binary);
|
||||
if (!outputFile.is_open())
|
||||
Error() << "cannot open output file";
|
||||
//IbmDecoder::setuseFm(false);
|
||||
uint32_t offset = 0;
|
||||
unsigned numSectorsinTrack = 0;
|
||||
|
||||
numHeads = geometry.numSides;
|
||||
numSectors = geometry.numSectors;
|
||||
numBytes = geometry.sectorSize;
|
||||
|
||||
size_t headSize = geometry.numSectors * geometry.sectorSize;
|
||||
size_t trackSize = headSize * geometry.numSides;
|
||||
|
||||
|
||||
Bytes imagenew;
|
||||
ByteWriter bw(imagenew);
|
||||
|
||||
//Give the user a option to give a comment in the IMD file for archive purposes.
|
||||
std::string comment;
|
||||
// std::cout << ("\nWrite a comment to store in de IMD file.\nIf no comment is given the default (IMD file by fluxengine) will be put in the file.\nPush Enter to store comment: \n");
|
||||
// getline (std::cin, comment);
|
||||
if (comment.size() == 0)
|
||||
{
|
||||
comment = LABEL;
|
||||
} else
|
||||
{
|
||||
comment.insert(0,"IMD ");
|
||||
}
|
||||
bw.seek(0);
|
||||
|
||||
// bw.append(comment);
|
||||
bw.write_8(END_OF_FILE);
|
||||
std::string sector_skew;
|
||||
sector_skew.clear();
|
||||
unsigned Status_Sector = 1;
|
||||
bool blnOptionalCylinderMap = false;
|
||||
bool blnOptionalHeadMap = false;
|
||||
// int emptyCylinders = 0;
|
||||
|
||||
|
||||
/* Write the actual sector data. */
|
||||
for (int track = 0; track < geometry.numTracks; track++)
|
||||
{
|
||||
for (int head = 0; head < numHeads; head++)
|
||||
{
|
||||
unsigned sectorIdBase = 1; //IMD starts sector numbering with 1;
|
||||
unsigned sectorId = 0;
|
||||
TrackHeader header = {0, 0, 0, 0, 0}; //define something to hold the header values
|
||||
const auto& sector = image.get(track, head, sectorId+1);
|
||||
if (!sector)
|
||||
{//sector 0 doesnt exist exit with error
|
||||
//this track, head has no sectors
|
||||
Status_Sector = 0;
|
||||
Logger() << fmt::format("sector {} not found on track {}, head {}\n", sectorId+1, track, head);
|
||||
// emptyCylinders++;
|
||||
break;
|
||||
} else
|
||||
{
|
||||
/* Get the header information */
|
||||
numBytes = sector->data.size(); //number of bytes can change per sector per track
|
||||
header.track = track;
|
||||
header.Head = head;
|
||||
header.SectorSize = setSectorSize(numBytes);
|
||||
sector_skew.clear();
|
||||
numSectorsinTrack = 0;
|
||||
// Logger() << fmt::format("clock {} \n", sector->clock);
|
||||
header.ModeValue = getModulationandSpeed(sector->clock);
|
||||
|
||||
|
||||
}
|
||||
//determine number of sectors in track
|
||||
for (int sectorId = 0; sectorId < numSectors; sectorId++)
|
||||
{
|
||||
const auto& sector = image.get(track, head, sectorId+1);
|
||||
if (!sector)
|
||||
{
|
||||
break;
|
||||
} else
|
||||
{
|
||||
numSectorsinTrack++;
|
||||
}
|
||||
}
|
||||
//determine sector skew and if there are optional cylindermaps or headermaps
|
||||
for (int sectorId = 0; sectorId < numSectorsinTrack; sectorId++)
|
||||
{
|
||||
const auto& sector = image.get(track, head, sectorId+1);
|
||||
if (!sector)
|
||||
{
|
||||
break;
|
||||
} else
|
||||
{
|
||||
sector_skew.push_back((sectorId + sectorIdBase) + '0'); //fill sectorskew start with 1
|
||||
if ((sector->physicalTrack) != (sector->logicalTrack)) //different physicaltrack fromn logicaltrack
|
||||
{
|
||||
blnOptionalCylinderMap = true;
|
||||
}
|
||||
if (sector->logicalSide != sector->physicalHead) //different physicalside fromn logicalside
|
||||
{
|
||||
blnOptionalHeadMap = true;
|
||||
}
|
||||
}
|
||||
}
|
||||
bw.write_8(header.ModeValue); //1 byte ModeValue
|
||||
bw.write_8(track); //1 byte Cylinder
|
||||
//are there optional cylinder or head maps?
|
||||
if (blnOptionalCylinderMap)
|
||||
{
|
||||
header.Head = header.Head^SEC_CYL_MAP_FLAG; //if head was 0 (00000000) it becomes (10000000)
|
||||
}
|
||||
if (blnOptionalHeadMap)
|
||||
{
|
||||
header.Head = header.Head^SEC_HEAD_MAP_FLAG; //if head was 1 (00000001) it becomes (01000001)
|
||||
}
|
||||
bw.write_8(head); //1 byte Head
|
||||
bw.write_8(numSectorsinTrack); //1 byte number of sectors in track
|
||||
bw.write_8(header.SectorSize); //1 byte sector size
|
||||
for (int sectorId = 0; sectorId < numSectorsinTrack; sectorId++)
|
||||
{
|
||||
bw.write_8((sectorId+ sectorIdBase)); //sector numbering map
|
||||
}
|
||||
//Write optional cylinder map
|
||||
//The Sector Cylinder Map has one entry for each sector, and contains the logical Cylinder ID for the corresponding sector in the Sector Numbering Map.
|
||||
if (blnOptionalCylinderMap)
|
||||
{
|
||||
//determine how the optional cylinder map looks like
|
||||
//write the corresponding logical ID
|
||||
for (int sectorId = 0; sectorId < numSectorsinTrack; sectorId++)
|
||||
{
|
||||
//const auto& sector = sectors.get(track, head, sectorId);
|
||||
bw.write_8(sector->logicalTrack); //1 byte logical track
|
||||
}
|
||||
}
|
||||
|
||||
//Write optional sector head map
|
||||
//The Sector Head Map has one entry for each sector, and contains the logical Head ID for the corresponding sector in the Sector Numbering Map.
|
||||
if (blnOptionalHeadMap)
|
||||
{
|
||||
//determine how the optional head map looks like
|
||||
//write the corresponding logical ID
|
||||
for (int sectorId = 0; sectorId < numSectorsinTrack; sectorId++)
|
||||
{
|
||||
// const auto& sector = sectors.get(track, head, sectorId);
|
||||
bw.write_8(sector->logicalSide); //1 byte logical side
|
||||
}
|
||||
}
|
||||
//Now read data and write to file
|
||||
for (int sectorId = 0; sectorId < numSectorsinTrack; sectorId++)
|
||||
{
|
||||
/* For each data record:
|
||||
* 1 byte Sector status
|
||||
* 0: Sector data unavailable - could not be read
|
||||
* 1: Normal data: (Sector Size) bytes follow
|
||||
* 2: Compressed: All bytes in sector have same value (xx)
|
||||
* 3: Normal data with "Deleted-Data address mark"
|
||||
* 4: Compressed with "Deleted-Data address mark"
|
||||
* 5: Normal data read with data error
|
||||
* 6: Compressed read with data error"
|
||||
* 7: Deleted data read with data error"
|
||||
* 8: Compressed, Deleted read with data error"
|
||||
* sector size of Sector data
|
||||
*/
|
||||
//read sector
|
||||
const auto& sector = image.get(track, head, sectorId+1);
|
||||
bool blnCompressable = false; //Consists the sector of 1 value? if yes then compresses IMD this to 1 value
|
||||
Bytes sectordata(numBytes); //define the sectordata with the size of the sectorsize
|
||||
Bytes compressed(1); //reserve 1 byte for comressed sectordata
|
||||
uint8_t byte; //value read
|
||||
uint8_t byte_previous; //previous value read (to determine if all bytes are equel in this sector)
|
||||
if (!sector)
|
||||
{
|
||||
Status_Sector = 0;
|
||||
break;
|
||||
} else
|
||||
{
|
||||
ByteReader br(sector->data); //read the sector data
|
||||
int i;
|
||||
//determine if all bytes are the same -> compress and sector status = 2
|
||||
for (i=0 ; i<numBytes ; i++)
|
||||
{
|
||||
byte = br.read_8();
|
||||
if (i == 0)
|
||||
{
|
||||
byte_previous = byte;
|
||||
}
|
||||
if (byte_previous == byte)
|
||||
{
|
||||
blnCompressable = true;
|
||||
} else
|
||||
{
|
||||
blnCompressable = false;
|
||||
break;
|
||||
}
|
||||
}
|
||||
switch (sector->status)
|
||||
{
|
||||
/*fluxengine knows of a few sector statussen but not all of the statussen in IMD.
|
||||
* // the statussen are in sector.h. Translation to fluxengine is as follows:
|
||||
* Statussen fluxengine | Status IMD
|
||||
*--------------------------------------------------------------------------------------------------------------------
|
||||
* OK, | 1, 2 (Normal data: (Sector Size) of (compressed) bytes follow)
|
||||
* BAD_CHECKSUM, | 5, 6, 7, 8
|
||||
* MISSING, sector not found | 0 (Sector data unavailable - could not be read)
|
||||
* DATA_MISSING, sector present but no data found | 3, 4
|
||||
* CONFLICT, |
|
||||
* INTERNAL_ERROR |
|
||||
*/
|
||||
case Sector::MISSING: /* Sector data unavailable - could not be read */
|
||||
|
||||
Status_Sector = 0;
|
||||
break;
|
||||
|
||||
case Sector::OK: /* Normal data: (Sector Size) bytes follow */
|
||||
if (blnCompressable) //data is compressable
|
||||
{
|
||||
Status_Sector = 2;
|
||||
} else
|
||||
{
|
||||
Status_Sector = 1;
|
||||
}
|
||||
break;
|
||||
case Sector::DATA_MISSING:
|
||||
Status_Sector = 3; //we misuse normal data with deleted-data addres mark for this missing data option
|
||||
break;
|
||||
// IMD recognizes all of these cases. but fluxengine doesnt support them.
|
||||
// case 2: /* Compressed: All bytes in sector have same value (xx) */
|
||||
// case 3: /* Normal data with "Deleted-Data address mark" */
|
||||
// case 4: /* Compressed with "Deleted-Data address mark"*/
|
||||
case Sector::BAD_CHECKSUM:
|
||||
// case 5: /* Normal data read with data error - could not be read*/
|
||||
Status_Sector = 5;
|
||||
break;
|
||||
// case 6: /* Compressed read with data error - could not be read */
|
||||
// case 7: /* Deleted data read with data error - could not be read */
|
||||
// case 8: /* Compressed, Deleted read with data error - could not be read */
|
||||
|
||||
default:
|
||||
Error() << fmt::format("Don't understand IMD files with sector status {}", Status_Sector);
|
||||
}
|
||||
bw.write_8(Status_Sector); //1 byte status sector
|
||||
if (blnCompressable)
|
||||
{
|
||||
bw.write_8(byte);
|
||||
blnCompressable = false;
|
||||
} else
|
||||
{
|
||||
bw.append(sector->data);
|
||||
}
|
||||
numSectors = numSectorsinTrack;
|
||||
}
|
||||
blnOptionalCylinderMap = false;
|
||||
blnOptionalHeadMap = false;
|
||||
}
|
||||
}
|
||||
}
|
||||
numCylinders = geometry.numTracks; //-emptyCylinders; //calculate the cylinders with sectors and hence data. Otherwise the calculation of kB goes wrong
|
||||
imagenew.writeTo(outputFile);
|
||||
Logger() << fmt::format("Writing {} tracks, {} heads, {} sectors, {} bytes per sector, {} kB total",
|
||||
numCylinders, numHeads,
|
||||
numSectors, numBytes,
|
||||
outputFile.tellp() / 1024);
|
||||
//<< std::endl;
|
||||
}
|
||||
};
|
||||
|
||||
std::unique_ptr<ImageWriter> ImageWriter::createImdImageWriter(
|
||||
const ImageWriterProto& config)
|
||||
{
|
||||
return std::unique_ptr<ImageWriter>(new ImdImageWriter(config));
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user