Changeset 12594

Timestamp:

2006-08-30T10:20:19+12:00 (18 years ago)

Author:

shaoqun

Message:

added code that uses MARCXML mapping file

Location:

trunk/gsdl

Files:

• bin/java/org/nzdl/gsdl/ApplyXSLT.class (modified) ( previous)
• src/java/org/nzdl/gsdl/ApplyXSLT.java (modified) (5 diffs)

trunk/gsdl/src/java/org/nzdl/gsdl/ApplyXSLT.java

@@ -10 +10 @@
 import javax.xml.transform.stream.StreamSource;
 
+import javax.xml.parsers.*;
+import javax.xml.transform.dom.*;
+import org.w3c.dom.*;
+
+
+
 /**
  *  Use the TraX interface to perform a transformation in the simplest manner possible
@@ -21 +27 @@
     public static final String INPUT_END = new String ("<?Done?>"); 
 
+    private static final String RECORD_ELEMENT = "record";
+    private static final String CONTROLFIELD_ELEMENT = "controlfield";
+    private static final String SUBFIELD_ELEMENT = "subfield";
+    private static final String LEADER_ELEMENT = "leader";
+
     private final int BEFORE_READING = 0;
     private final int IS_READING = 1;
     private String xsl_file;
+    private String mapping_file;
 
     public ApplyXSLT(String xsl_file)
     {
     this.xsl_file = xsl_file; 
+    }
+
+    public ApplyXSLT(String xsl_file,String mapping_file){
+    this.xsl_file = xsl_file; 
+    this.mapping_file = mapping_file;
     }
 
@@ -41 +58 @@
 
     try{
-        while (br.ready())
-        {
-                    
-            String this_line = br.readLine();
-                     
-                        
-            if(system_status == BEFORE_READING)
-            {
-                
-                if(this_line.compareTo(DOC_START) == 0)
-                {
-                    output_file = br.readLine(); // read the next line as the output file name
-                    system_status = IS_READING;
-                    a_doc = "";
-                }
-                else if(this_line.compareTo(INPUT_END) == 0)
-                {
-                    return true;
-                }
-                else
-                {
-                     
-                    System.err.println("Undefined process status:" + this_line);
-                    system_status = BEFORE_READING;
-                }
-                
+        while (br.ready()) {
+                           
+        String this_line = br.readLine();
+        if(system_status == BEFORE_READING){
+            if(this_line.compareTo(DOC_START) == 0){
+            output_file = br.readLine(); // read the next line as the output file name
+            system_status = IS_READING;
+            a_doc = "";
+            }
+            else if(this_line.compareTo(INPUT_END) == 0){
+            return true;
+            }
+            else{
+            System.err.println("Undefined process status:" + this_line);
+            system_status = BEFORE_READING;
+            }
+            
+        }
+        else if(system_status == IS_READING){
+            if(this_line.compareTo(DOC_END) == 0){
+            boolean result = false;
+            if (mapping_file !=null && !mapping_file.equals("")){
+                result = translateXMLWithMapping(a_doc,output_file);                
             }
-            else if(system_status == IS_READING)
-            {
-                if(this_line.compareTo(DOC_END) == 0)
-                {
-                   
-                    if(translateXML(a_doc,xsl_file,output_file) == false)
-                    {
-                        System.err.println("Translation Failed!!");
-                        return false;
-                    }
-                    system_status = BEFORE_READING;
-                   
-                }
-                else
-                {
-                       a_doc = a_doc + this_line + "\n";
-                                       
-                }
+            else{
+                result = translateXML(a_doc,output_file); 
             }
-            else
-            {
-              
-                System.err.println ("Undefined system status in ApplyXSLT.java main().");
-                System.exit(-1);
+            
+            if (!result){
+                System.err.println("Translation Failed!!");
+                return false;
             }
-                
+            
+            system_status = BEFORE_READING;
+            
+            }
+            else{
+            a_doc = a_doc + this_line + "\n";
+            
+            }
         }
+        else{         
+            System.err.println ("Undefined system status in ApplyXSLT.java main().");
+            System.exit(-1);
+            }
+        
+        }
     }catch (Exception e)
         {
@@ -102 +112 @@
     return false;
     }
-
-
-    private boolean translateXML(String full_doc, String xsl_file, String output_file) 
+    
+    
+    private boolean translateXML(String full_doc, String output_file) 
     throws IOException,TransformerException, TransformerConfigurationException, FileNotFoundException
     {
-       
+    
     StringReader str = new StringReader(full_doc) ;
     
-    TransformerFactory tFactory = TransformerFactory.newInstance();
+    TransformerFactory tFactory = TransformerFactory.newInstance();
     Transformer transformer = tFactory.newTransformer(new StreamSource(xsl_file));
     transformer.transform(new StreamSource(str), new StreamResult(new FileOutputStream(output_file)));
@@ -116 +126 @@
     }
 
+    private boolean translateXMLWithMapping(String full_doc, String output_file) 
+    throws IOException,TransformerException, TransformerConfigurationException, FileNotFoundException
+    {   
+    StringReader str = new StringReader(full_doc) ;
+
+    try{
+        TransformerFactory tFactory = TransformerFactory.newInstance();
+        Transformer transformer = tFactory.newTransformer(new StreamSource(xsl_file));
+        
+        Document mapping_doc = DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(mapping_file);
+        Element mapping =mapping_doc.getDocumentElement(); 
+        
+        transformer.setParameter("mapping",mapping);
+        
+        Document output_doc = DocumentBuilderFactory.newInstance().newDocumentBuilder().newDocument();
+        
+        transformer.transform(new StreamSource(str), new DOMResult(output_doc));
+        
+        calculateRecordsLength(output_doc);
+       
+        transformer = tFactory.newTransformer();
+       
+        transformer.transform(new DOMSource(output_doc), new StreamResult(new FileOutputStream(output_file)));
+
+    }
+    catch(Exception e){
+        e.printStackTrace();
+        return false;
+    }
+
+    return true;
+    }
+
+    private void calculateRecordsLength(Document output_doc){
+    NodeList records = output_doc.getDocumentElement().getElementsByTagName(RECORD_ELEMENT);
+
+    for(int i=0;i<records.getLength();i++){
+        Element record = (Element)records.item(i);
+        calculateRecordLength(record);
+    }
+    }
+
+    private void calculateRecordLength(Element record){
+    int total_length =0;    
+    NodeList controlfileds = record.getElementsByTagName(CONTROLFIELD_ELEMENT);
+    for(int i=0;i<controlfileds.getLength();i++){
+        Element controlfiled = (Element)controlfileds.item(i);
+        total_length +=getElementTextValue(controlfiled).length();
+    }
+    
+    NodeList subfileds = record.getElementsByTagName(SUBFIELD_ELEMENT);
+    for(int i=0;i<subfileds.getLength();i++){
+        Element subfiled = (Element)subfileds.item(i);
+        total_length +=getElementTextValue(subfiled).length();
+    }
+
+    String record_length = total_length+"";
+    //fill in a extra digit as record length needs to be five characters long
+    if (total_length < 10000){
+        record_length = "0"+record_length;
+        if (total_length < 1000){
+        record_length = "0"+record_length;  
+        }
+        if (total_length < 100){
+        record_length = "0"+record_length;  
+        }
+        if (total_length < 10){
+         record_length = "0"+record_length; 
+        }
+       
+    }
+
+    NodeList leaders = record.getElementsByTagName(LEADER_ELEMENT); 
+    
+    //only one leader element
+        if (leaders.getLength() >0){
+        Element leader_element = (Element)leaders.item(0);
+        removeFirstTextNode(leader_element);
+        leader_element.insertBefore(leader_element.getOwnerDocument().createTextNode(record_length),leader_element.getFirstChild()); 
+    }
+
+    }
+
+    private void removeFirstTextNode(Element element){
+    //remove the first text node
+    NodeList children_nodelist = element.getChildNodes();
+    for (int i = 0; i < children_nodelist.getLength(); i++) {
+        Node child_node = children_nodelist.item(i);
+        if (child_node.getNodeType() == Node.TEXT_NODE) {
+        element.removeChild(child_node);
+        return;
+        }
+    }
+
+    }
+
+    private String getElementTextValue(Element element)
+    {
+    String text ="";    
+
+    // Find the node child
+    NodeList children_nodelist = element.getChildNodes();
+    for (int i = 0; i < children_nodelist.getLength(); i++) {
+        Node child_node = children_nodelist.item(i);
+        if (child_node.getNodeType() == Node.TEXT_NODE) {
+        text +=child_node.getNodeValue();
+        }
+    }
+
+        return text;
+    }
+ 
+
+    private void setMappingVariable(Document style_doc){
+    Node child = style_doc.getDocumentElement().getFirstChild();
+    while(child != null) {
+        String name = child.getNodeName();
+        if (name.equals("xsl:variable")) {
+        Element variable_element = (Element)child;
+        if ( variable_element.getAttribute("name").trim().equals("mapping")){
+            variable_element.setAttribute("select","document('"+mapping_file+"')/Mapping");
+        }
+        }
+        child = child.getNextSibling();
+    }
+    
+    } 
+
+
     public static void main(String[] args) 
     { 
     // Checking Arguments
-    if(args.length != 1)
+    if(args.length < 1)
         {
         System.err.println("Must provide a XSL file !");
         System.exit(-1);
         }
-    ApplyXSLT core = new ApplyXSLT(args[0]);
+    ApplyXSLT core = null;
+
+    if (args.length == 1){
+         core = new ApplyXSLT(args[0]);
+    }
+    else if(args.length == 2){
+         core = new ApplyXSLT(args[0],args[1]); 
+    }
 
     if(core.process())