//===- SubtargetEmitter.cpp - Generate subtarget enumerations -------------===//
//
//                     The LLVM Compiler Infrastructure
//
// This file was developed by James M. Laskey and is distributed under
// the University of Illinois Open Source License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This tablegen backend emits subtarget enumerations.
//
//===----------------------------------------------------------------------===//

#include "SubtargetEmitter.h"
#include "CodeGenTarget.h"
#include "Record.h"
#include "llvm/ADT/StringExtras.h"
#include "llvm/Support/Debug.h"
#include <algorithm>
#include <set>
using namespace llvm;

//
// Convenience types.
//
typedef std::vector<Record*> RecordList;
typedef std::vector<Record*>::iterator RecordListIter;

//
// Record sort by name function.
//
struct LessRecord {
  bool operator()(const Record *Rec1, const Record *Rec2) const {
    return Rec1->getName() < Rec2->getName();
  }
};

//
// Record sort by field "Name" function.
//
struct LessRecordFieldName {
  bool operator()(const Record *Rec1, const Record *Rec2) const {
    return Rec1->getValueAsString("Name") < Rec2->getValueAsString("Name");
  }
};


// 
// SubtargetEmitter::run - Main subtarget enumeration emitter.
//
void SubtargetEmitter::run(std::ostream &OS) {
  EmitSourceFileHeader("Subtarget Enumeration Source Fragment", OS);
  
  RecordList Features = Records.getAllDerivedDefinitions("SubtargetFeature");
  sort(Features.begin(), Features.end(), LessRecord());
  
  RecordList Processors = Records.getAllDerivedDefinitions("Processor");
  sort(Processors.begin(), Processors.end(), LessRecordFieldName());

  OS << "namespace llvm {\n\n";
  
  { // Feature enumeration
    int i = 0;
    
    OS << "enum {\n";
    
    for (RecordListIter RI = Features.begin(), E = Features.end(); RI != E;){
      Record *R = *RI++;
      std::string Instance = R->getName();
      OS << "  "
         << Instance
         << " = "
         << " 1 << " << i++
         << ((RI != E) ? ",\n" : "\n");
    }
    
    OS << "};\n";
  }
  
  { // Feature key values
    OS << "\n\n"
       << "/// Sorted (by key) array of values for CPU features.\n"
       << "static SubtargetFeatureKV FeatureKV[] = {\n";
    for (RecordListIter RI = Features.begin(), E = Features.end(); RI != E;) {
      Record *R = *RI++;
      std::string Instance = R->getName();
      std::string Name = R->getValueAsString("Name");
      std::string Desc = R->getValueAsString("Desc");
      OS << "  { "
         << "\"" << Name << "\", "
         << "\"" << Desc << "\", "
         << Instance
         << ((RI != E) ? " },\n" : " }\n");
    }
    OS << "};\n";
  }
  
  { // CPU key values
    OS << "\n\n"
       << "/// Sorted (by key) array of values for CPU subtype.\n"
       << "static const SubtargetFeatureKV SubTypeKV[] = {\n";
    for (RecordListIter RI = Processors.begin(), E = Processors.end();
         RI != E;) {
      Record *R = *RI++;
      std::string Name = R->getValueAsString("Name");
      Record *ProcItin = R->getValueAsDef("ProcItin");
      ListInit *Features = R->getValueAsListInit("Features");
      unsigned N = Features->getSize();
      OS << "  { "
         << "\"" << Name << "\", "
         << "\"Select the " << Name << " processor\", ";
         
      
      if (N == 0) {
        OS << "0";
      } else {
        for (unsigned i = 0; i < N; ) {
          if (DefInit *DI = dynamic_cast<DefInit*>(Features->getElement(i++))) {
            Record *Feature = DI->getDef();
            std::string Name = Feature->getName();
            OS << Name;
            if (i != N) OS << " | ";
          } else {
            throw "Feature: " + Name +
                  " expected feature in processor feature list!";
          }
        }
      }
      
      OS << ((RI != E) ? " },\n" : " }\n");
    }
    OS << "};\n";
  }

  OS << "\n} // End llvm namespace \n";
}
