//Сжатие с использование квадродеревьев
public static void Compression(string filename, string quality)
{
if (!(File.Exists(filename)))
{
System.Console.WriteLine("Файла не существует");
return;
}
//Черно-белый файл или нет
if (quality == "Black")
{
colorflag = 1;
}
else
{
if (quality == "Color")
{
colorflag = 3;
}
else
{
System.Console.WriteLine("Тип выбран не правильно");
return;
}
}
Bitmap image = new Bitmap(filename);
imageWidth = image.Width;
imageHeigth = image.Height;
classImageColor = new Color[image.Width, image.Height];
for (int i = 0; i < image.Width; ++i)
{
for (int j = 0; j < image.Height; ++j)
{
classImageColor[i, j] = image.GetPixel(i, j);
}
}
brightnessImage = new double[image.Width, image.Height];
for (int i = 0; i < image.Width; ++i)
{
for (int j = 0; j < image.Height; ++j)
{
brightnessImage[i, j] = image.GetPixel(i, j).GetBrightness();
}
}
//Создаём ранговое дерево
range_block_size = 64;
range_num_width = image.Width / range_block_size;
range_num_height = image.Height / range_block_size;
rangeArray = new Object.Block[range_num_width, range_num_height];
//Изначально создаются ранговые блоки, на основе которых будут составляться деревья
for (int i = 0; i < range_num_width; ++i)
{
for (int j = 0; j < range_num_height; ++j)
{
rangeArray[i, j] = CreateBlockRange(i * range_block_size, j * range_block_size, classImageColor, range_block_size, brightnessImage);
}
}
rangeTree = new BlockTree[range_num_width, range_num_height];
for (int i = 0; i < range_num_width; ++i)
{
for (int j = 0; j < range_num_height; ++j)
{
numLock = 1;
rangeTree[i, j] = new BlockTree(rangeArray[i, j], rangeArray[i, j].BlockSize);
}
}
//Доменные блоки представлены не в виде деревьев, а в виде трехмерных массивов
domain_num_width = range_num_width - 1;
domain_num_height = range_num_height - 1;
domain_block_size = range_block_size * 2;
domainArray = new Object.Block[domain_num_width, domain_num_height];
//System.Console.WriteLine(DomainBlocks.Length);
domainBlocks = new Object.BlockArray[PostProcessing.GetPow(domain_block_size)];
for (int i = 0; i < domainBlocks.Length; ++i)
{
if (i == 0)
{
domainBlocks[i].BlockSize = domain_block_size;
}
else
{
domainBlocks[i].BlockSize = domainBlocks[i - 1].BlockSize / 2;
}
domainBlocks[i].num_width = image.Width / domainBlocks[i].BlockSize;
domainBlocks[i].num_height = image.Height / domainBlocks[i].BlockSize;
domainBlocks[i].Blocks = new Object.Block[domainBlocks[i].num_width, domainBlocks[i].num_height];
for (int j = 0; j < domainBlocks[i].num_width; ++j)
{
for (int k = 0; k < domainBlocks[i].num_height; ++k)
{
domainBlocks[i].Blocks[j, k] = CreateBlockDomain(j * domainBlocks[i].BlockSize / 2, k * domainBlocks[i].BlockSize / 2, classImageColor, domainBlocks[i].BlockSize / 2, brightnessImage);
}
}
}
listCoeff = new List <Object.Coefficients>();
//Обход всех деревьев и нахождение для нужных коэффициентов преобразования, а так же выписывание их в файл
for (int i = 0; i < range_num_width; ++i)
{
for (int j = 0; j < range_num_height; ++j)
{
blockChecker = 0;
rangeTree[i, j].RoundTree(rangeTree[i, j], domainBlocks, classImageColor, rangeTree[i, j].mainBlock.BlockSize);
}
}
bw = new BinaryWriter(File.Open(@"C:\Users\Admin\Documents\GitHub\Fractal-Compression\NewFractalCompression\NewFractalCompression\Quad Compression", FileMode.Create));
bw.Write(MyConverter.Convert(BitConverter.GetBytes(image.Width), 2));
bw.Write(MyConverter.Convert(BitConverter.GetBytes(image.Height), 2));
bw.Write(MyConverter.Convert(BitConverter.GetBytes(range_block_size), 1));
bw.Write(MyConverter.Convert(BitConverter.GetBytes(listCoeff.Count), 4));
//for (int i = 0; i < range_num_width; ++i)
//{
// for (int j = 0; j < range_num_height; ++j)
// {
// rangeTree[i, j].PrintTree(rangeTree[i, j], 0);
// }
//}
for (int i = 0; i < listCoeff.Count; ++i)
{
//Запись в файл всех нужные чисел, а так же глубину нахождения узла в дереве
Byte[] D = BitConverter.GetBytes(listCoeff[i].Depth);
Byte[] X = BitConverter.GetBytes(listCoeff[i].X);
Byte[] Y = BitConverter.GetBytes(listCoeff[i].Y);
Byte[] SR = BitConverter.GetBytes(listCoeff[i].shiftR);
Byte[] SG = BitConverter.GetBytes(listCoeff[i].shiftG);
Byte[] SB = BitConverter.GetBytes(listCoeff[i].shiftB);
//System.Console.WriteLine(listCoeff[i].Depth + " -- " + D[0]);
bw.Write(MyConverter.Convert(D, 1));
bw.Write(MyConverter.Convert(X, 1));
bw.Write(MyConverter.Convert(Y, 1));
bw.Write(MyConverter.Convert(SR, 2));
bw.Write(MyConverter.Convert(SG, 2));
bw.Write(MyConverter.Convert(SB, 2));
}
bw.Close();
}
public static void Compression(string filename, string quality)
{
if (!(File.Exists(filename)))
{
System.Console.WriteLine("Файла не существует");
return;
}
//Черно-белый файл или нет
if (quality == "Black")
{
colorflag = 1;
}
else
{
if (quality == "Color")
{
colorflag = 3;
}
else
{
System.Console.WriteLine("Тип выбран не правильно");
return;
}
}
Bitmap image = new Bitmap(filename);
classImageColor = new Color[image.Width, image.Height];
for (int i = 0; i < image.Width; ++i)
{
for (int j = 0; j < image.Height; ++j)
{
classImageColor[i, j] = image.GetPixel(i, j);
}
}
double[,] BrightnessImage = new double[image.Width, image.Height];
for (int i = 0; i < image.Width; ++i)
{
for (int j = 0; j < image.Height; ++j)
{
BrightnessImage[i, j] = image.GetPixel(i, j).GetBrightness();
}
}
//Основной параметр, отвечающий за размеры ранговых блоков
range_block_size = 128;
//Создаём ранговые блоки
range_num_width = image.Width / range_block_size;
range_num_height = image.Height / range_block_size;
rangeArray = new Object.Block[range_num_width, range_num_height];
for (int i = 0; i < range_num_width; ++i)
{
for (int j = 0; j < range_num_height; ++j)
{
rangeArray[i, j] = CreateBlockRange(i * range_block_size, j * range_block_size, classImageColor, range_block_size, BrightnessImage);
}
}
//Создаем доменные блоки
domain_num_width = range_num_width - 1;
domain_num_height = range_num_height - 1;
domain_block_size = range_block_size * 2;
domainArray = new Object.Block[domain_num_width, domain_num_height];
for (int i = 0; i < domain_num_width; ++i)
{
for (int j = 0; j < domain_num_height; ++j)
{
domainArray[i, j] = CreateBlockDomain(i * range_block_size, j * range_block_size, classImageColor, range_block_size, BrightnessImage);
}
}
//Алгоритм сжатия
count = 1;
//Общеее число преобразований
block_all_num = colorflag * range_num_width * range_num_height;
compressCoeff = new Object.Coefficients[range_num_width, range_num_height, 1];
BinaryWriter bw = new BinaryWriter(File.Open(@"C:\Users\Admin\Documents\GitHub\Fractal-Compression\NewFractalCompression\NewFractalCompression\Compression", FileMode.Create));
Parallel.For(0, range_num_width, FindCoefficients);
//Выводим коэффиценты в файл
bw.Write(MyConverter.Convert(BitConverter.GetBytes(image.Width), 2));
bw.Write(MyConverter.Convert(BitConverter.GetBytes(image.Height), 2));
bw.Write(MyConverter.Convert(BitConverter.GetBytes(range_block_size), 1));
for (int k = 0; k < 1; ++k)
{
for (int i = 0; i < range_num_width; ++i)
{
for (int j = 0; j < range_num_height; ++j)
{
Byte[] X = BitConverter.GetBytes(compressCoeff[i, j, k].X);
Byte[] Y = BitConverter.GetBytes(compressCoeff[i, j, k].Y);
Byte[] SR = BitConverter.GetBytes(compressCoeff[i, j, k].shiftR);
Byte[] SG = BitConverter.GetBytes(compressCoeff[i, j, k].shiftG);
Byte[] SB = BitConverter.GetBytes(compressCoeff[i, j, k].shiftB);
bw.Write(MyConverter.Convert(X, 1));
bw.Write(MyConverter.Convert(Y, 1));
bw.Write(MyConverter.Convert(SR, 2));
bw.Write(MyConverter.Convert(SG, 2));
bw.Write(MyConverter.Convert(SB, 2));
}
}
}
bw.Close();
}
