ep_reader.cpp

#include <iostream>
#include <fstream>
#include <vector>
#include <bitset>
#include <iomanip>
#include <algorithm>
#include <numeric>

int MAX_WEFT_FEEDERS = 12;
int REGULATOR_HOOK = 22;
int PICK_VARIATION_HOOK = 17;

class JacquardDesignReader {
private:
    std::string filename;
    int16_t totalPicks;        // Total Picks
    int16_t totalHooks;        // Total Design Hooks
    std::vector<std::vector<bool>> designData;  // Store hooks state for each card

    int calculateCardSizeBytes() const {
        return (totalHooks + 32 + 7) / 8;  // Rounds up to the nearest byte
    }

    std::vector<uint8_t> convertCardToByteVector(const std::vector<bool>& cardHooks) const {
        int cardSizeBytes = calculateCardSizeBytes();
        std::vector<uint8_t> cardData(cardSizeBytes, 0);

        for (int bitPos = 0; bitPos < cardHooks.size(); ++bitPos) {
            if (cardHooks[bitPos]) {
                int byteIndex = bitPos / 8;
                int bitIndex = (bitPos % 8);
                cardData[byteIndex] |= (1 << bitIndex);
            }
        }
        return cardData;
    }

public:
    explicit JacquardDesignReader(const std::string &file) : filename(file), totalPicks(0), totalHooks(0) {}

    bool loadFile() {
        std::ifstream file(filename, std::ios::binary);
        if (!file.is_open()) {
            std::cerr << "Error: Could not open file " << filename << std::endl;
            return false;
        }

        // Read Total Picks (at address 0x03)
        file.seekg(0x03, std::ios::beg);
        file.read(reinterpret_cast<char *>(&totalPicks), sizeof(totalPicks));

        // Read Total Hooks (at address 0x20)
        file.seekg(0x20, std::ios::beg);
        file.read(reinterpret_cast<char *>(&totalHooks), sizeof(totalHooks));

        if (totalPicks <= 0 || totalHooks <= 0) {
            std::cerr << "Error: Invalid picks or hooks count in file." << std::endl;
            return false;
        }

        file.seekg(0x22, std::ios::beg);

        int cardSizeBytes = (totalHooks + 32 + 7) / 8;

        for (int i = 0; i < totalPicks; ++i) {
            std::vector<bool> cardHooks(totalHooks + 32, false);
            std::vector<uint8_t> cardData(cardSizeBytes, 0);
            file.read(reinterpret_cast<char *>(cardData.data()), cardSizeBytes);

            // Convert each byte to individual bits and store as hook states
            for (int bitPos = 0; bitPos < totalHooks + 32; ++bitPos) {
                int byteIndex = bitPos / 8;
                int bitIndex = (bitPos % 8);  // Bit order within byte
                cardHooks[bitPos] = (cardData[byteIndex] & (1 << bitIndex)) != 0;
            }

            designData.push_back(cardHooks);
        }
        
        file.close();
        return true;
    }

    void printDesignData() const {
        std::cout << "Total Picks: " << totalPicks << std::endl;
        std::cout << "Total Hooks: " << totalHooks << std::endl;
    }

    std::vector<bool> getCard(int cardNumber) const {
        if (cardNumber < 1 || cardNumber > totalPicks) {
            throw std::out_of_range("Error: Card number out of range.");
        }
        return designData[cardNumber - 1];  // Card numbers are 1-based, vector is 0-based
    }

    void displayCard(int cardNumber) const {
        try {
            auto cardHooks = getCard(cardNumber);
            std::cout << "Hooks for Card " << cardNumber << ": ";
            for (bool hook : cardHooks) {
                std::cout << (hook ? '1' : '0');
            }
            std::cout << std::endl;
        } catch (const std::out_of_range &e) {
            std::cerr << e.what() << std::endl;
        }
    }

    int getTotalPicks() const { return totalPicks; }
    int getTotalHooks() const { return totalHooks; }

    bool getHookState(int cardNumber, int hookNumber) const {// Current state of hook (0,1  )
        return designData[cardNumber-1][hookNumber-1];

    }

    bool getNextHookState(int cardNumber, int hookNumber) const {
        if (hookNumber > getTotalPicks()) {
            return 0;
        }

        return designData[cardNumber][hookNumber - 1];
    }

    void setHookState(int cardNumber, int hookNumber, int value) {
        designData[cardNumber-1][hookNumber-1] = value;
    }

    std::vector<int> getHookActiveCards(int hookNumber) {// Card Numbers for which hook is active
        std::vector<int> hookActiveCards;
        
        for (int cardNum = 1 ; cardNum <= totalPicks; cardNum++) {
            if (designData[cardNum - 1][hookNumber - 1]) {
                hookActiveCards.push_back(cardNum);
            }
        }

        return hookActiveCards;

    }

    std::vector<bool> getActiveWeft() const {
        std::vector<bool> activeWeft(12,0);
        for(int card_no = 1; card_no < getTotalPicks() + 1; card_no++) {
            std::vector<bool> card = getCard(card_no);

            for ( int i = 1;i<=MAX_WEFT_FEEDERS ; i++) {
                if (getHookState(card_no, i)) {
                    activeWeft[i-1] = 1;
                }
            }
        }
        return activeWeft;
    }

    int getActiveWeftCount() const {
        std::vector<bool> active = getActiveWeft();
        int count = 0;

        count = std::accumulate(active.begin(), active.end(), 0);
        return count;
    }


    bool writeToFile(const std::string &outputFilename) const {
        std::ifstream inputFile(filename, std::ios::binary);
        if (!inputFile.is_open()) {
            std::cerr << "Error: Could not open file " << filename << " for reading." << std::endl;
            return false;
        }

        std::ofstream outputFile(outputFilename, std::ios::binary);
        if (!outputFile.is_open()) {
            std::cerr << "Error: Could not open file " << outputFilename << " for writing." << std::endl;
            return false;
        }

        inputFile.seekg(0, std::ios::end);               // Move to the end of the file
        std::streampos originalFileSize = inputFile.tellg();  // Get the file size
        inputFile.seekg(0, std::ios::beg);               // Move back to the beginning for further processing


        // Copy everything before the design data (header and part of the file before card data)
        char buffer[1024];
        inputFile.read(buffer, 0x22);  // Read up to the start of the design data (address 0x22)
        outputFile.write(buffer, inputFile.gcount());

        int cardSizeBytes = calculateCardSizeBytes();
        for (int i = 0; i < totalPicks; ++i) {
            std::vector<uint8_t> cardData = convertCardToByteVector(designData[i]);
            outputFile.write(reinterpret_cast<const char*>(cardData.data()), cardData.size());
        }

        // Copy everything after the card data (if there's any trailing data in the file)
        inputFile.seekg(0x22 + totalPicks * cardSizeBytes, std::ios::beg);
        while (inputFile.read(buffer, sizeof(buffer))) {
            outputFile.write(buffer, inputFile.gcount());
        }

        std::streampos currentOutputFileSize = outputFile.tellp();
        std::streampos paddingNeeded = originalFileSize - currentOutputFileSize;

        if (paddingNeeded > 0) {
            // Write exactly the remaining padding bytes as zero to match the original file size
            std::vector<char> paddingBuffer(static_cast<size_t>(paddingNeeded), 0);
            outputFile.write(paddingBuffer.data(), paddingBuffer.size());
        }

        inputFile.close();
        outputFile.close();
        return true;
    }
 

};

void loadOrCreateConfig(const std::string& configFile) {
    // Check if the file exists by attempting to open it
    std::ifstream file(configFile);
    if (!file.is_open()) {
        // File does not exist, so create it with default values
        std::ofstream outFile(configFile);
        if (!outFile.is_open()) {
            std::cerr << "Error: Could not create config file: " << configFile << std::endl;
            return;
        }
        
        // Write default values to the new config file
        outFile << "MAX_WEFT_FEEDERS=" << MAX_WEFT_FEEDERS << std::endl;
        outFile << "REGULATOR_HOOK=" << REGULATOR_HOOK << std::endl;
        outFile << "PICK_VARIATION_HOOK=" << PICK_VARIATION_HOOK << std::endl;
        outFile.close();
        std::cout << "Config file created with default values: " << configFile << std::endl;
    } else {
        // Config file exists, so read values from it
        std::string line;
        while (std::getline(file, line)) {
            std::istringstream lineStream(line);
            std::string key;
            if (std::getline(lineStream, key, '=')) {
                std::string valueStr;
                if (std::getline(lineStream, valueStr)) {
                    int value = std::stoi(valueStr);

                    if (key == "MAX_WEFT_FEEDERS") {
                        MAX_WEFT_FEEDERS = value;
                    } else if(key == "REGULATOR_HOOK") {
                        REGULATOR_HOOK = value;
                    } else if (key == "PICK_VARIATION_HOOK") {
                        PICK_VARIATION_HOOK = value;
                    }
                }
            }
        }
        file.close();
        std::cout << "Config file loaded: " << configFile << std::endl;
    }
}


int main() {
    loadOrCreateConfig("config.txt");
    std::string filename = ".\\ep_files\\loom-020.EP";  // Adjust this filename as needed
    JacquardDesignReader reader(filename);
    
    std::cout << "Reading file " << filename << std::endl;
    if (reader.loadFile()) {
        reader.writeToFile("output.ep");
    } else {
        std::cerr << "Failed to load design file." << std::endl;
        return 1;
    }

    return 0;
}