Skip to content
Snippets Groups Projects
Commit 47f7fa6d authored by Vojtech Moravec's avatar Vojtech Moravec
Browse files

Rename CacheFileHeaderV1->QvcHeaderV1

parent ce5fa4e6
No related branches found
No related tags found
No related merge requests found
package cz.it4i.qcmp.cache; package cz.it4i.qcmp.cache;
import cz.it4i.qcmp.fileformat.CacheFileHeaderV1; import cz.it4i.qcmp.fileformat.QvcHeaderV1;
import java.io.DataInputStream; import java.io.DataInputStream;
import java.io.DataOutputStream; import java.io.DataOutputStream;
...@@ -12,9 +12,9 @@ public interface ICacheFile { ...@@ -12,9 +12,9 @@ public interface ICacheFile {
void readFromStream(DataInputStream inputStream) throws IOException; void readFromStream(DataInputStream inputStream) throws IOException;
void readFromStream(DataInputStream inputStream, CacheFileHeaderV1 header) throws IOException; void readFromStream(DataInputStream inputStream, QvcHeaderV1 header) throws IOException;
CacheFileHeaderV1 getHeader(); QvcHeaderV1 getHeader();
void report(StringBuilder builder); void report(StringBuilder builder);
......
...@@ -2,8 +2,8 @@ package cz.it4i.qcmp.cache; ...@@ -2,8 +2,8 @@ package cz.it4i.qcmp.cache;
import cz.it4i.qcmp.compression.CompressionOptions; import cz.it4i.qcmp.compression.CompressionOptions;
import cz.it4i.qcmp.data.V3i; import cz.it4i.qcmp.data.V3i;
import cz.it4i.qcmp.fileformat.CacheFileHeaderV1;
import cz.it4i.qcmp.fileformat.QuantizationType; import cz.it4i.qcmp.fileformat.QuantizationType;
import cz.it4i.qcmp.fileformat.QvcHeaderV1;
import cz.it4i.qcmp.quantization.scalar.SQCodebook; import cz.it4i.qcmp.quantization.scalar.SQCodebook;
import cz.it4i.qcmp.quantization.vector.VQCodebook; import cz.it4i.qcmp.quantization.vector.VQCodebook;
...@@ -68,8 +68,8 @@ public class QuantizationCacheManager { ...@@ -68,8 +68,8 @@ public class QuantizationCacheManager {
* @param codebook Final SQ codebook. * @param codebook Final SQ codebook.
* @return SQ cache file header. * @return SQ cache file header.
*/ */
private CacheFileHeaderV1 createHeaderForSQ(final String trainFile, final SQCodebook codebook) { private QvcHeaderV1 createHeaderForSQ(final String trainFile, final SQCodebook codebook) {
final CacheFileHeaderV1 header = new CacheFileHeaderV1(); final QvcHeaderV1 header = new QvcHeaderV1();
header.setQuantizationType(QuantizationType.Scalar); header.setQuantizationType(QuantizationType.Scalar);
header.setCodebookSize(codebook.getCodebookSize()); header.setCodebookSize(codebook.getCodebookSize());
header.setTrainFileName(trainFile); header.setTrainFileName(trainFile);
...@@ -105,8 +105,8 @@ public class QuantizationCacheManager { ...@@ -105,8 +105,8 @@ public class QuantizationCacheManager {
* @param codebook Final VQ codebook. * @param codebook Final VQ codebook.
* @return VQ cache file header. * @return VQ cache file header.
*/ */
private CacheFileHeaderV1 createHeaderForVQ(final String trainFile, final VQCodebook codebook) { private QvcHeaderV1 createHeaderForVQ(final String trainFile, final VQCodebook codebook) {
final CacheFileHeaderV1 header = new CacheFileHeaderV1(); final QvcHeaderV1 header = new QvcHeaderV1();
header.setQuantizationType(getQuantizationTypeFromVectorDimensions(codebook.getVectorDims())); header.setQuantizationType(getQuantizationTypeFromVectorDimensions(codebook.getVectorDims()));
header.setCodebookSize(codebook.getCodebookSize()); header.setCodebookSize(codebook.getCodebookSize());
header.setTrainFileName(trainFile); header.setTrainFileName(trainFile);
...@@ -125,7 +125,7 @@ public class QuantizationCacheManager { ...@@ -125,7 +125,7 @@ public class QuantizationCacheManager {
public String saveCodebook(final String trainFile, final SQCodebook codebook) throws IOException { public String saveCodebook(final String trainFile, final SQCodebook codebook) throws IOException {
final String fileName = getCacheFilePathForSQ(trainFile, codebook.getCodebookSize()).getAbsolutePath(); final String fileName = getCacheFilePathForSQ(trainFile, codebook.getCodebookSize()).getAbsolutePath();
final CacheFileHeaderV1 header = createHeaderForSQ(new File(trainFile).getName(), codebook); final QvcHeaderV1 header = createHeaderForSQ(new File(trainFile).getName(), codebook);
final SQCacheFile cacheFile = new SQCacheFile(header, codebook); final SQCacheFile cacheFile = new SQCacheFile(header, codebook);
try (final FileOutputStream fos = new FileOutputStream(fileName, false); try (final FileOutputStream fos = new FileOutputStream(fileName, false);
...@@ -152,7 +152,7 @@ public class QuantizationCacheManager { ...@@ -152,7 +152,7 @@ public class QuantizationCacheManager {
codebook.getCodebookSize(), codebook.getCodebookSize(),
codebook.getVectorDims()).getAbsolutePath(); codebook.getVectorDims()).getAbsolutePath();
final CacheFileHeaderV1 header = createHeaderForVQ(new File(trainFile).getName(), codebook); final QvcHeaderV1 header = createHeaderForVQ(new File(trainFile).getName(), codebook);
final VQCacheFile cacheFile = new VQCacheFile(header, codebook); final VQCacheFile cacheFile = new VQCacheFile(header, codebook);
try (final FileOutputStream fos = new FileOutputStream(fileName, false); try (final FileOutputStream fos = new FileOutputStream(fileName, false);
...@@ -357,7 +357,7 @@ public class QuantizationCacheManager { ...@@ -357,7 +357,7 @@ public class QuantizationCacheManager {
dis = new DataInputStream(inputStream); dis = new DataInputStream(inputStream);
} }
final CacheFileHeaderV1 header = new CacheFileHeaderV1(); final QvcHeaderV1 header = new QvcHeaderV1();
try { try {
header.readFromStream(dis); header.readFromStream(dis);
} catch (final IOException e) { } catch (final IOException e) {
...@@ -409,12 +409,12 @@ public class QuantizationCacheManager { ...@@ -409,12 +409,12 @@ public class QuantizationCacheManager {
* @param path Path to cache file. * @param path Path to cache file.
*/ */
public static void inspectCacheFile(final String path, final boolean verbose) { public static void inspectCacheFile(final String path, final boolean verbose) {
CacheFileHeaderV1 header = null; QvcHeaderV1 header = null;
final long fileSize; final long fileSize;
try (final FileInputStream fis = new FileInputStream(path); try (final FileInputStream fis = new FileInputStream(path);
final DataInputStream dis = new DataInputStream(fis)) { final DataInputStream dis = new DataInputStream(fis)) {
fileSize = fis.getChannel().size(); fileSize = fis.getChannel().size();
header = new CacheFileHeaderV1(); header = new QvcHeaderV1();
header.readFromStream(dis); header.readFromStream(dis);
} catch (final IOException e) { } catch (final IOException e) {
e.printStackTrace(); e.printStackTrace();
......
package cz.it4i.qcmp.cache; package cz.it4i.qcmp.cache;
import cz.it4i.qcmp.fileformat.CacheFileHeaderV1; import cz.it4i.qcmp.fileformat.QvcHeaderV1;
import cz.it4i.qcmp.quantization.scalar.SQCodebook; import cz.it4i.qcmp.quantization.scalar.SQCodebook;
import java.io.DataInputStream; import java.io.DataInputStream;
...@@ -8,13 +8,13 @@ import java.io.DataOutputStream; ...@@ -8,13 +8,13 @@ import java.io.DataOutputStream;
import java.io.IOException; import java.io.IOException;
public class SQCacheFile implements ICacheFile { public class SQCacheFile implements ICacheFile {
private CacheFileHeaderV1 header; private QvcHeaderV1 header;
private SQCodebook codebook; private SQCodebook codebook;
public SQCacheFile() { public SQCacheFile() {
} }
public SQCacheFile(final CacheFileHeaderV1 header, final SQCodebook codebook) { public SQCacheFile(final QvcHeaderV1 header, final SQCodebook codebook) {
this.header = header; this.header = header;
this.codebook = codebook; this.codebook = codebook;
assert (header.getCodebookSize() == codebook.getCodebookSize()); assert (header.getCodebookSize() == codebook.getCodebookSize());
...@@ -34,12 +34,12 @@ public class SQCacheFile implements ICacheFile { ...@@ -34,12 +34,12 @@ public class SQCacheFile implements ICacheFile {
} }
public void readFromStream(final DataInputStream inputStream) throws IOException { public void readFromStream(final DataInputStream inputStream) throws IOException {
header = new CacheFileHeaderV1(); header = new QvcHeaderV1();
header.readFromStream(inputStream); header.readFromStream(inputStream);
readFromStream(inputStream, header); readFromStream(inputStream, header);
} }
public void readFromStream(final DataInputStream inputStream, final CacheFileHeaderV1 header) throws IOException { public void readFromStream(final DataInputStream inputStream, final QvcHeaderV1 header) throws IOException {
this.header = header; this.header = header;
final int codebookSize = header.getCodebookSize(); final int codebookSize = header.getCodebookSize();
final int[] centroids = new int[codebookSize]; final int[] centroids = new int[codebookSize];
...@@ -54,7 +54,7 @@ public class SQCacheFile implements ICacheFile { ...@@ -54,7 +54,7 @@ public class SQCacheFile implements ICacheFile {
codebook = new SQCodebook(centroids, frequencies); codebook = new SQCodebook(centroids, frequencies);
} }
public CacheFileHeaderV1 getHeader() { public QvcHeaderV1 getHeader() {
return header; return header;
} }
......
package cz.it4i.qcmp.cache; package cz.it4i.qcmp.cache;
import cz.it4i.qcmp.fileformat.CacheFileHeaderV1; import cz.it4i.qcmp.fileformat.QvcHeaderV1;
import cz.it4i.qcmp.quantization.vector.VQCodebook; import cz.it4i.qcmp.quantization.vector.VQCodebook;
import java.io.DataInputStream; import java.io.DataInputStream;
...@@ -8,13 +8,13 @@ import java.io.DataOutputStream; ...@@ -8,13 +8,13 @@ import java.io.DataOutputStream;
import java.io.IOException; import java.io.IOException;
public class VQCacheFile implements ICacheFile { public class VQCacheFile implements ICacheFile {
private CacheFileHeaderV1 header; private QvcHeaderV1 header;
private VQCodebook codebook; private VQCodebook codebook;
public VQCacheFile() { public VQCacheFile() {
} }
public VQCacheFile(final CacheFileHeaderV1 header, final VQCodebook codebook) { public VQCacheFile(final QvcHeaderV1 header, final VQCodebook codebook) {
this.header = header; this.header = header;
this.codebook = codebook; this.codebook = codebook;
assert (header.getCodebookSize() == codebook.getCodebookSize()); assert (header.getCodebookSize() == codebook.getCodebookSize());
...@@ -37,13 +37,13 @@ public class VQCacheFile implements ICacheFile { ...@@ -37,13 +37,13 @@ public class VQCacheFile implements ICacheFile {
} }
public void readFromStream(final DataInputStream inputStream) throws IOException { public void readFromStream(final DataInputStream inputStream) throws IOException {
header = new CacheFileHeaderV1(); header = new QvcHeaderV1();
header.readFromStream(inputStream); header.readFromStream(inputStream);
readFromStream(inputStream, header); readFromStream(inputStream, header);
} }
@Override @Override
public void readFromStream(final DataInputStream inputStream, final CacheFileHeaderV1 header) throws IOException { public void readFromStream(final DataInputStream inputStream, final QvcHeaderV1 header) throws IOException {
this.header = header; this.header = header;
final int codebookSize = header.getCodebookSize(); final int codebookSize = header.getCodebookSize();
...@@ -64,7 +64,7 @@ public class VQCacheFile implements ICacheFile { ...@@ -64,7 +64,7 @@ public class VQCacheFile implements ICacheFile {
codebook = new VQCodebook(header.getVectorDim(), vectors, frequencies); codebook = new VQCodebook(header.getVectorDim(), vectors, frequencies);
} }
public CacheFileHeaderV1 getHeader() { public QvcHeaderV1 getHeader() {
return header; return header;
} }
......
...@@ -9,7 +9,7 @@ import java.io.DataInputStream; ...@@ -9,7 +9,7 @@ import java.io.DataInputStream;
import java.io.DataOutputStream; import java.io.DataOutputStream;
import java.io.IOException; import java.io.IOException;
public class CacheFileHeaderV1 implements IFileHeader { public class QvcHeaderV1 implements IFileHeader {
//region Constants //region Constants
private static final int VERSION = 1; private static final int VERSION = 1;
private static final String MAGIC_VALUE = "QCMPCACHE"; private static final String MAGIC_VALUE = "QCMPCACHE";
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment